lix/src/libutil/tarfile.cc

126 lines
3.2 KiB
C++
Raw Normal View History

2019-12-07 15:35:14 +00:00
#include <archive.h>
#include <archive_entry.h>
2019-12-09 10:21:46 +00:00
#include "serialise.hh"
2019-03-27 13:12:20 +00:00
2019-09-11 11:10:46 +00:00
namespace nix {
2019-12-07 18:08:33 +00:00
struct TarArchive {
2019-12-19 14:01:58 +00:00
struct archive * archive;
Source * source;
2019-12-09 10:21:46 +00:00
std::vector<unsigned char> buffer;
2019-12-07 18:08:33 +00:00
2019-12-19 14:01:58 +00:00
void check(int err, const char * reason = "failed to extract archive: %s")
{
2019-12-07 18:08:33 +00:00
if (err == ARCHIVE_EOF)
throw EndOfFile("reached end of archive");
else if (err != ARCHIVE_OK)
throw Error(reason, archive_error_string(this->archive));
}
2019-12-07 18:08:33 +00:00
2019-12-19 14:01:58 +00:00
TarArchive(Source & source) : buffer(4096)
{
2019-12-07 18:08:33 +00:00
this->archive = archive_read_new();
this->source = &source;
archive_read_support_filter_all(archive);
archive_read_support_format_all(archive);
2019-12-19 14:01:58 +00:00
check(archive_read_open(archive,
(void *)this,
TarArchive::callback_open,
TarArchive::callback_read,
TarArchive::callback_close),
"failed to open archive: %s");
}
2019-12-07 18:08:33 +00:00
2019-12-19 14:01:58 +00:00
TarArchive(const Path & path)
{
2019-12-07 18:08:33 +00:00
this->archive = archive_read_new();
archive_read_support_filter_all(archive);
archive_read_support_format_all(archive);
2019-12-19 14:01:58 +00:00
check(archive_read_open_filename(archive, path.c_str(), 16384), "failed to open archive: %s");
2019-12-07 18:08:33 +00:00
}
2019-12-19 14:01:58 +00:00
TarArchive(const TarArchive &) = delete;
2019-12-09 10:21:46 +00:00
2019-12-19 14:01:58 +00:00
void close()
{
check(archive_read_close(archive), "failed to close archive: %s");
2019-12-07 18:08:33 +00:00
}
2019-12-19 14:01:58 +00:00
~TarArchive()
{
2019-12-07 18:08:33 +00:00
if (this->archive) archive_read_free(this->archive);
}
private:
2019-12-19 14:01:58 +00:00
static int callback_open(struct archive *, void * self) {
2019-12-07 18:08:33 +00:00
return ARCHIVE_OK;
}
2019-12-09 10:21:46 +00:00
2019-12-19 14:01:58 +00:00
static ssize_t callback_read(struct archive * archive, void * _self, const void * * buffer)
{
auto self = (TarArchive *)_self;
2019-12-09 10:21:46 +00:00
*buffer = self->buffer.data();
2019-12-07 18:08:33 +00:00
try {
return self->source->read((char *) self->buffer.data(), 4096);
2019-12-07 18:08:33 +00:00
} catch (EndOfFile &) {
return 0;
2019-12-19 14:01:58 +00:00
} catch (std::exception & err) {
archive_set_error(archive, EIO, "source threw exception: %s", err.what());
2019-12-07 18:08:33 +00:00
return -1;
}
}
2019-12-19 14:01:58 +00:00
static int callback_close(struct archive *, void * self) {
2019-12-07 18:08:33 +00:00
return ARCHIVE_OK;
}
};
2019-12-19 14:01:58 +00:00
static void extract_archive(TarArchive & archive, const Path & destDir)
{
2019-12-09 10:21:46 +00:00
int flags = ARCHIVE_EXTRACT_FFLAGS
| ARCHIVE_EXTRACT_PERM
| ARCHIVE_EXTRACT_TIME
2019-12-09 10:21:46 +00:00
| ARCHIVE_EXTRACT_SECURE_SYMLINKS
| ARCHIVE_EXTRACT_SECURE_NODOTDOT;
2019-12-07 18:08:33 +00:00
2019-12-19 14:01:58 +00:00
for (;;) {
struct archive_entry * entry;
2019-12-07 18:08:33 +00:00
int r = archive_read_next_header(archive.archive, &entry);
2019-12-07 15:35:14 +00:00
if (r == ARCHIVE_EOF) break;
2019-12-07 18:08:33 +00:00
else if (r == ARCHIVE_WARN)
2019-12-19 14:01:58 +00:00
warn(archive_error_string(archive.archive));
2019-12-07 18:08:33 +00:00
else
archive.check(r);
archive_entry_set_pathname(entry,
(destDir + "/" + archive_entry_pathname(entry)).c_str());
2019-12-07 18:08:33 +00:00
archive.check(archive_read_extract(archive.archive, entry, flags));
2019-12-07 15:35:14 +00:00
}
2019-12-07 18:08:33 +00:00
archive.close();
2019-12-07 15:35:14 +00:00
}
2019-12-07 18:08:33 +00:00
2019-12-07 15:35:14 +00:00
void unpackTarfile(Source & source, const Path & destDir)
{
2019-12-07 18:08:33 +00:00
auto archive = TarArchive(source);
2019-12-07 16:23:11 +00:00
2019-12-07 15:35:14 +00:00
createDirs(destDir);
2019-12-07 18:08:33 +00:00
extract_archive(archive, destDir);
2019-12-07 15:35:14 +00:00
}
2019-12-07 18:08:33 +00:00
2019-12-07 15:35:14 +00:00
void unpackTarfile(const Path & tarFile, const Path & destDir)
{
2019-12-07 18:08:33 +00:00
auto archive = TarArchive(tarFile);
2019-12-07 16:23:11 +00:00
2019-12-07 15:35:14 +00:00
createDirs(destDir);
2019-12-07 18:08:33 +00:00
extract_archive(archive, destDir);
}
2019-09-11 11:10:46 +00:00
}