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

Change-Id: Ib16eefc17ae53874b295476fae515305525a7f2b
This commit is contained in:
Jade Lovelace
2025-12-10 14:29:35 -08:00
parent 36a61d922c
commit 6f483e5f52
3 changed files with 36 additions and 38 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");
+29 -30
View File
@@ -45,48 +45,45 @@ 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(), archive_read_free}
, 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);
check(archive_read_open(archive, (void *)this, callback_open, callback_read, callback_close), "Failed to open archive (%s)");
archive_read_set_option(archive.get(), nullptr, "mac-ext", nullptr);
check(
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(), archive_read_free}
{
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)
@@ -99,13 +96,15 @@ 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));
throw Error(
"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);
@@ -125,7 +124,7 @@ static void extract_archive(TarArchive & archive, const Path & destDir)
(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
@@ -8,7 +8,7 @@
namespace nix {
struct TarArchive {
struct archive * archive;
std::unique_ptr<struct archive, decltype(&archive_read_free)> archive;
Source * source;
std::vector<unsigned char> buffer;
@@ -22,8 +22,6 @@ struct TarArchive {
TarArchive(const TarArchive &) = delete;
void close();
~TarArchive();
};
kj::Promise<Result<void>> unpackTarfile(AsyncInputStream & source, const Path & destDir);