From 1355554d1207267ccfb272532917e8085eeb9fb9 Mon Sep 17 00:00:00 2001 From: Yorick van Pelt Date: Sat, 7 Dec 2019 23:23:11 +0700 Subject: [PATCH] code 'cleanup' --- src/libutil/tarfile.cc | 42 ++++++++++++++++++++++++++++-------------- 1 file changed, 28 insertions(+), 14 deletions(-) diff --git a/src/libutil/tarfile.cc b/src/libutil/tarfile.cc index 8e2f7c8cb..b44ebea64 100644 --- a/src/libutil/tarfile.cc +++ b/src/libutil/tarfile.cc @@ -20,7 +20,7 @@ std::shared_ptr archive_read_ptr() { } static void ac(std::shared_ptr a, int r, const char* str = "archive is corrupt (%s)") { if (r == ARCHIVE_EOF) { - throw EndOfFile("tar ended"); + throw EndOfFile("reached end of archive"); } if (r != ARCHIVE_OK) { throw Error(str, archive_error_string(a.get())); @@ -72,16 +72,25 @@ static void copy_data(std::shared_ptr ar, std::shared_ptr a, const Path & destDir) { - char * cwd = getcwd(0, 0); - if (!cwd) throw SysError("getting current directory"); - Finally backCwd([&]() { - int r = chdir(cwd); - free(cwd); - if (r != 0) throw SysError("resetting directory after archive extraction"); - }); - int r = chdir(destDir.c_str()); - if (r != 0) throw SysError("setting directory to tar output path"); + // need to chdir back *after* archive closing + PushD newDir(destDir); struct archive_entry *entry; int flags = 0; auto ext = archive_write_ptr(); @@ -90,7 +99,7 @@ static void extract_archive(std::shared_ptr a, const Path & dest archive_write_disk_set_options(ext.get(), flags); archive_write_disk_set_standard_lookup(ext.get()); for(;;) { - r = archive_read_next_header(a.get(), &entry); + int r = archive_read_next_header(a.get(), &entry); if (r == ARCHIVE_EOF) break; if (r == ARCHIVE_WARN) { std::cerr << "warning: " << archive_error_string(a.get()); @@ -101,22 +110,27 @@ static void extract_archive(std::shared_ptr a, const Path & dest copy_data(a, ext); archive_write_finish_entry(ext.get()); } + // done in dtor, but this error can be 'failed to set permissions' + // so it's important + ac(ext, archive_write_close(ext.get()), "finishing archive extraction"); } void unpackTarfile(Source & source, const Path & destDir) { - auto a = archive_read_ptr(); + auto a = nix::archive_read_ptr(); archive_read_support_filter_all(a.get()); archive_read_support_format_all(a.get()); - archive_read_open_source(a, source); + nix::archive_read_open_source(a, source); + createDirs(destDir); extract_archive(a, destDir); } void unpackTarfile(const Path & tarFile, const Path & destDir) { - auto a = archive_read_ptr(); + auto a = nix::archive_read_ptr(); archive_read_support_filter_all(a.get()); archive_read_support_format_all(a.get()); ac(a, archive_read_open_filename(a.get(), tarFile.c_str(), 16384)); + createDirs(destDir); extract_archive(a, destDir); }