refactor: use std::unique_ptr for libarchive state, remove destructor

Change-Id: Ib16eefc17ae53874b295476fae515305525a7f2b
This commit is contained in:
Jade Lovelace
2025-12-15 21:21:04 +01:00
parent 0ebb1c524d
commit 88cea98b82
3 changed files with 31 additions and 39 deletions
+6 -5
View File
@@ -16,7 +16,6 @@
#include <archive.h>
#include <archive_entry.h>
#include <cerrno>
#include <chrono>
#include <cstdio>
#include <cstring>
@@ -63,13 +62,15 @@ struct ArchiveDecompressionSource : Source
struct archive_entry * ae;
if (!archive) {
archive = std::make_unique<TarArchive>(*src, true);
this->archive->check(archive_read_next_header(this->archive->archive, &ae),
"failed to read header (%s)");
if (archive_filter_count(this->archive->archive) < 2) {
this->archive->check(
archive_read_next_header(this->archive->archive.get(), &ae),
"failed to read header (%s)"
);
if (archive_filter_count(this->archive->archive.get()) < 2) {
throw CompressionError("input compression not recognized");
}
}
ssize_t result = archive_read_data(this->archive->archive, data, len);
ssize_t result = archive_read_data(this->archive->archive.get(), data, len);
if (result > 0) return result;
if (result == 0) {
throw EndOfFile("reached end of compressed file");
+24 -31
View File
@@ -45,53 +45,46 @@ void TarArchive::check(int err, const std::string & reason)
if (err == ARCHIVE_EOF) {
throw EndOfFile("reached end of archive");
} else if (err != ARCHIVE_OK) {
throw Error(reason, archive_error_string(this->archive));
throw Error(reason, archive_error_string(this->archive.get()));
}
}
TarArchive::TarArchive(Source & source, bool raw) : buffer(65536)
TarArchive::TarArchive(Source & source, bool raw)
: archive{archive_read_new()}
, source(&source)
, buffer(65536)
{
this->archive = archive_read_new();
this->source = &source;
if (!raw) {
archive_read_support_filter_all(archive);
archive_read_support_format_all(archive);
archive_read_support_filter_all(archive.get());
archive_read_support_format_all(archive.get());
} else {
archive_read_support_filter_all(archive);
archive_read_support_format_raw(archive);
archive_read_support_format_empty(archive);
archive_read_support_filter_all(archive.get());
archive_read_support_format_raw(archive.get());
archive_read_support_format_empty(archive.get());
}
archive_read_set_option(archive, nullptr, "mac-ext", nullptr);
archive_read_set_option(archive.get(), nullptr, "mac-ext", nullptr);
check(
archive_read_open(archive, (void *) this, callback_open, callback_read, callback_close),
archive_read_open(
archive.get(), (void *) this, callback_open, callback_read, callback_close
),
"Failed to open archive (%s)"
);
}
TarArchive::TarArchive(const Path & path)
TarArchive::TarArchive(const Path & path) : archive{archive_read_new()}
{
this->archive = archive_read_new();
archive_read_support_filter_all(archive);
archive_read_support_format_all(archive);
archive_read_set_option(archive, nullptr, "mac-ext", nullptr);
archive_read_support_filter_all(archive.get());
archive_read_support_format_all(archive.get());
archive_read_set_option(archive.get(), nullptr, "mac-ext", nullptr);
check(
archive_read_open_filename(archive, requireCString(path), 16384),
archive_read_open_filename(archive.get(), requireCString(path), 16384),
"failed to open archive: %s"
);
}
void TarArchive::close()
{
check(archive_read_close(this->archive), "Failed to close archive (%s)");
}
TarArchive::~TarArchive()
{
if (this->archive) {
archive_read_free(this->archive);
}
check(archive_read_close(this->archive.get()), "Failed to close archive (%s)");
}
static void extract_archive(TarArchive & archive, const Path & destDir)
@@ -103,18 +96,18 @@ static void extract_archive(TarArchive & archive, const Path & destDir)
for (;;) {
struct archive_entry * entry;
int r = archive_read_next_header(archive.archive, &entry);
int r = archive_read_next_header(archive.archive.get(), &entry);
if (r == ARCHIVE_EOF) {
break;
}
auto name = archive_entry_pathname(entry);
if (!name) {
throw Error(
"cannot get archive member name: %s", archive_error_string(archive.archive)
"cannot get archive member name: %s", archive_error_string(archive.archive.get())
);
}
if (r == ARCHIVE_WARN) {
printTaggedWarning("%1%", Uncolored(archive_error_string(archive.archive)));
printTaggedWarning("%1%", Uncolored(archive_error_string(archive.archive.get())));
} else {
archive.check(r);
}
@@ -135,7 +128,7 @@ static void extract_archive(TarArchive & archive, const Path & destDir)
archive_entry_copy_hardlink(entry, (destDir + "/" + original_hardlink).c_str());
}
archive.check(archive_read_extract(archive.archive, entry, flags));
archive.check(archive_read_extract(archive.archive.get(), entry, flags));
}
archive.close();
+1 -3
View File
@@ -9,7 +9,7 @@ namespace nix {
struct TarArchive
{
struct archive * archive;
std::unique_ptr<struct archive, decltype([](auto * p) { archive_read_free(p); })> archive;
Source * source;
std::vector<unsigned char> buffer;
@@ -23,8 +23,6 @@ struct TarArchive
TarArchive(const TarArchive &) = delete;
void close();
~TarArchive();
};
kj::Promise<Result<void>> unpackTarfile(AsyncInputStream & source, const Path & destDir);