Add error handling

This commit is contained in:
2020-12-11 13:06:44 +01:00
parent d99eabafb0
commit fe92d3e431
5 changed files with 97 additions and 30 deletions

View File

@@ -0,0 +1,30 @@
/*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
*/
#include "ArchiveBase.hpp"
#include "LibArchiveHelper.hpp"
namespace openVulkanoCpp
{
ArchiveBase::ArchiveBase(archive* arch, archive_entry* archEntry, const std::shared_ptr<spdlog::logger>& logger)
: m_archive(arch), m_archiveEntry(archEntry), m_logger(logger)
{
if (!m_logger)
{
m_logger = Logger::FILESYS;
}
}
ArchiveBase::~ArchiveBase()
{
ChkErr(archive_free(m_archive));
}
bool ArchiveBase::ChkErr(int result) const
{
return LibArchiveHelper::CheckError(result, m_archive, m_logger);
}
}

View File

@@ -0,0 +1,38 @@
/*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
*/
#pragma once
#include "IO/FileDescription.hpp"
#include <memory>
struct archive;
struct archive_entry;
namespace spdlog
{
class logger;
}
namespace openVulkanoCpp
{
class ArchiveBase
{
protected:
archive* m_archive;
archive_entry* m_archiveEntry;
std::shared_ptr<spdlog::logger> m_logger;
public:
ArchiveBase(archive* arch, archive_entry* archEntry, const std::shared_ptr<spdlog::logger>& logger);
virtual ~ArchiveBase();
bool ChkErr(int result) const;
void SetLogger(const std::shared_ptr<spdlog::logger>& logger) { m_logger = logger; }
[[nodiscard]] std::shared_ptr<spdlog::logger> GetLogger() const { return m_logger; }
};
}

View File

@@ -80,30 +80,30 @@ namespace openVulkanoCpp
return ac;
}
ArchiveWriter::ArchiveWriter(const char* fileName) : ArchiveWriter(fileName, ArchiveConfiguration::FromFileName(fileName))
ArchiveWriter::ArchiveWriter(const char* fileName, const std::shared_ptr<spdlog::logger>& logger)
: ArchiveWriter(fileName, ArchiveConfiguration::FromFileName(fileName), logger)
{}
ArchiveWriter::ArchiveWriter(const char* fileName, ArchiveConfiguration archiveConfiguration)
: m_archive(archive_write_new()), m_archiveEntry(archive_entry_new())
ArchiveWriter::ArchiveWriter(const char* fileName, ArchiveConfiguration archiveConfiguration, const std::shared_ptr<spdlog::logger>& logger)
: ArchiveBase(archive_write_new(), archive_entry_new(), logger)
{ //TODO error handling
archive_write_set_format(m_archive, TYPE_MAP[static_cast<int>(archiveConfiguration.type)]);
ChkErr(archive_write_set_format(m_archive, TYPE_MAP[static_cast<int>(archiveConfiguration.type)]));
if (archiveConfiguration.type == ArchiveConfiguration::ArchiveType::TAR)
{
archive_write_add_filter(m_archive, COMP_MAP[static_cast<int>(archiveConfiguration.compression)]);
ChkErr(archive_write_add_filter(m_archive, COMP_MAP[static_cast<int>(archiveConfiguration.compression)]));
}
if (archiveConfiguration.compression != ArchiveConfiguration::CompressionType::NONE &&
archiveConfiguration.compressionLevel > 0)
{
std::string level = "compression-level=" + std::to_string(archiveConfiguration.compressionLevel);
archive_write_set_options(m_archive, level.c_str());
ChkErr(archive_write_set_options(m_archive, level.c_str()));
}
archive_write_open_filename(m_archive, fileName);
ChkErr(archive_write_open_filename(m_archive, fileName));
}
ArchiveWriter::~ArchiveWriter()
{
archive_entry_free(m_archiveEntry);
archive_write_free(m_archive);
}
void ArchiveWriter::Close()
@@ -129,17 +129,16 @@ namespace openVulkanoCpp
{
archive_entry_clear(m_archiveEntry);
std::unique_ptr<archive, decltype(archive_free)*> archiveDisk(archive_read_disk_new(), archive_free);
archive_entry_clear(m_archiveEntry);
archive_read_disk_open(archiveDisk.get(), fileName);
archive_read_next_header2(archiveDisk.get(), m_archiveEntry);
ChkErr(archive_read_disk_open(archiveDisk.get(), fileName));
ChkErr(archive_read_next_header2(archiveDisk.get(), m_archiveEntry));
//archive_read_disk_entry_from_file(archiveDisk, m_archiveEntry, -1, nullptr);
archive_entry_set_pathname(m_archiveEntry, inArchiveName);
archive_write_header(m_archive, m_archiveEntry);
if (LibArchiveHelper::CopyArchiveData(archiveDisk.get(), m_archive) != ARCHIVE_OK)
{
//TODO handle error
}
ChkErr(archive_write_header(m_archive, m_archiveEntry));
int result = LibArchiveHelper::CopyArchiveData(archiveDisk.get(), m_archive);
ChkErr(result);
LibArchiveHelper::CheckError(result, archiveDisk.get(), m_logger);
ChkErr(archive_write_finish_entry(m_archive));
m_bytesWritten += archive_entry_size(m_archiveEntry);
return true;
}
@@ -152,7 +151,9 @@ namespace openVulkanoCpp
bool ArchiveWriter::AddFile(const FileDescription& description, const void* buffer)
{
WriteHeader(description);
return archive_write_data(m_archive, buffer, description.size) == ARCHIVE_OK;
bool ok = ChkErr(archive_write_data(m_archive, buffer, description.size));
ChkErr(archive_write_finish_entry(m_archive));
return ok;
}
bool ArchiveWriter::AddFile(const char* fileName, const std::vector<std::pair<const void*, size_t>>& buffers)
@@ -173,11 +174,12 @@ namespace openVulkanoCpp
for(const auto& buffer : buffers)
{
if (buffer.first == nullptr) continue;
if (archive_write_data(m_archive, buffer.first, buffer.second) != ARCHIVE_OK)
if (!ChkErr(archive_write_data(m_archive, buffer.first, buffer.second)))
{
return false;
}
}
ChkErr(archive_write_finish_entry(m_archive));
return true;
}
@@ -190,7 +192,8 @@ namespace openVulkanoCpp
archive_entry_set_perm(m_archiveEntry, static_cast<mode_t>(fileDescription.permissions));
archive_entry_set_ctime(m_archiveEntry, fileDescription.createTime, 0);
archive_entry_set_mtime(m_archiveEntry, fileDescription.modTime, 0);
archive_write_header(m_archive, m_archiveEntry);
ChkErr(archive_write_header(m_archive, m_archiveEntry));
m_bytesWritten += fileDescription.size;
}
bool ArchiveWriter::AddFiles(const char* dirName)

View File

@@ -6,13 +6,11 @@
#pragma once
#include "ArchiveBase.hpp"
#include "IO/FileDescription.hpp"
#include <string_view>
#include <vector>
struct archive;
struct archive_entry;
namespace openVulkanoCpp
{
struct ArchiveConfiguration
@@ -28,17 +26,15 @@ namespace openVulkanoCpp
static ArchiveConfiguration FromFileName(const char* fileName);
};
class ArchiveWriter
class ArchiveWriter final : public ArchiveBase
{
archive* m_archive;
archive_entry* m_archiveEntry;
size_t m_bytesWritten = 0;
public:
ArchiveWriter(const char* fileName);
ArchiveWriter(const char* fileName, ArchiveConfiguration archiveConfiguration);
ArchiveWriter(const char* fileName, const std::shared_ptr<spdlog::logger>& logger = nullptr);
ArchiveWriter(const char* fileName, ArchiveConfiguration archiveConfiguration, const std::shared_ptr<spdlog::logger>& logger = nullptr);
~ArchiveWriter();
~ArchiveWriter() override;
void Close();

View File

@@ -68,7 +68,7 @@ namespace openVulkanoCpp
if (archiveResult <= ARCHIVE_FATAL) lvl = spdlog::level::level_enum::critical;
else if (archiveResult <= ARCHIVE_FAILED) lvl = spdlog::level::level_enum::err;
const char* errorString = archive_error_string(arch);
logger->log(lvl, errorString);
if (logger) logger->log(lvl, errorString);
if (archiveResult == ARCHIVE_FAILED) throw std::runtime_error(errorString);
return false;
}