void EnableArchiveFormats(struct archive *p_archive)
{
// archive_read_support_filter_bzip2(p_archive);
// archive_read_support_filter_compress(p_archive);
// archive_read_support_filter_gzip(p_archive);
// archive_read_support_filter_grzip(p_archive);
// archive_read_support_filter_lrzip(p_archive);
// archive_read_support_filter_lzip(p_archive);
archive_read_support_filter_lzma(p_archive);
archive_read_support_filter_lzop(p_archive);
archive_read_support_filter_none(p_archive);
archive_read_support_filter_rpm(p_archive);
archive_read_support_filter_uu(p_archive);
archive_read_support_filter_xz(p_archive);
// archive_read_support_format_7zip(p_archive);
archive_read_support_format_ar(p_archive);
archive_read_support_format_cab(p_archive);
archive_read_support_format_cpio(p_archive);
archive_read_support_format_gnutar(p_archive);
// archive_read_support_format_iso9660(p_archive);
archive_read_support_format_lha(p_archive);
archive_read_support_format_mtree(p_archive);
archive_read_support_format_rar(p_archive);
archive_read_support_format_raw(p_archive);
archive_read_support_format_tar(p_archive);
archive_read_support_format_xar(p_archive);
// archive_read_support_format_zip(p_archive);
}
int
do_extract_mtree(char *mtree, const char *prefix)
{
struct archive *a = NULL;
struct archive_entry *ae;
char path[MAXPATHLEN];
const char *fpath;
int retcode = EPKG_OK;
int ret;
if (mtree == NULL || *mtree == '\0')
return EPKG_OK;
a = archive_read_new();
archive_read_support_filter_none(a);
archive_read_support_format_mtree(a);
if (archive_read_open_memory(a, mtree, strlen(mtree)) != ARCHIVE_OK) {
pkg_emit_error("Fail to extract the mtree: %s",
archive_error_string(a));
retcode = EPKG_FATAL;
goto cleanup;
}
while ((ret = archive_read_next_header(a, &ae)) != ARCHIVE_EOF) {
if (ret != ARCHIVE_OK) {
pkg_emit_error("Skipping unsupported mtree line: %s",
archive_error_string(a));
continue;
}
fpath = archive_entry_pathname(ae);
if (*fpath != '/') {
snprintf(path, sizeof(path), "%s/%s", prefix, fpath);
archive_entry_set_pathname(ae, path);
}
/* Ignored failed extraction on purpose */
archive_read_extract(a, ae, EXTRACT_ARCHIVE_FLAGS);
}
cleanup:
if (a != NULL)
archive_read_free(a);
return (retcode);
}
/* Deprecated; remove in libarchive 4.0 */
int
archive_read_support_compression_none(struct archive *a)
{
return archive_read_support_filter_none(a);
}
static void
test_format(int (*set_format)(struct archive *))
{
char filedata[64];
struct archive_entry *ae;
struct archive *a;
size_t used;
size_t buffsize = 1000000;
char *buff;
const char *err;
buff = malloc(buffsize);
/* Create a new archive in memory. */
assert((a = archive_write_new()) != NULL);
assertEqualIntA(a, ARCHIVE_OK, (*set_format)(a));
assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_none(a));
assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, buffsize, &used));
/*
* Write a file to it.
*/
assert((ae = archive_entry_new()) != NULL);
archive_entry_set_pathname(ae, "test");
archive_entry_set_filetype(ae, AE_IFREG);
assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
archive_entry_free(ae);
assertEqualIntA(a, 9, archive_write_data(a, "12345678", 9));
assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
assertEqualInt(ARCHIVE_OK, archive_write_free(a));
/*
* Read from it.
*/
assert((a = archive_read_new()) != NULL);
assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_raw(a));
assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_none(a));
assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used));
assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
assertEqualIntA(a, 9, archive_read_data(a, filedata, 10));
assertEqualMem(filedata, "12345678", 9);
assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
assertEqualInt(ARCHIVE_OK, archive_read_free(a));
/* Create a new archive */
assert((a = archive_write_new()) != NULL);
assertEqualIntA(a, ARCHIVE_OK, (*set_format)(a));
assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_none(a));
assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, buffsize, &used));
/* write first file: that should succeed */
assert((ae = archive_entry_new()) != NULL);
archive_entry_set_pathname(ae, "test");
archive_entry_set_filetype(ae, AE_IFREG);
assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
archive_entry_free(ae);
assertEqualIntA(a, 9, archive_write_data(a, "12345678", 9));
/* write second file: this should fail */
assert((ae = archive_entry_new()) != NULL);
archive_entry_set_pathname(ae, "test2");
archive_entry_set_filetype(ae, AE_IFREG);
assertEqualIntA(a, ARCHIVE_FATAL, archive_write_header(a, ae));
err = archive_error_string(a);
assertEqualMem(err, "Raw format only supports one entry per archive", 47);
archive_entry_free(ae);
assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
assertEqualInt(ARCHIVE_OK, archive_write_free(a));
/* Create a new archive */
assert((a = archive_write_new()) != NULL);
assertEqualIntA(a, ARCHIVE_OK, (*set_format)(a));
assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_none(a));
assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, buffsize, &used));
/* write a directory: this should fail */
assert((ae = archive_entry_new()) != NULL);
archive_entry_copy_pathname(ae, "dir");
archive_entry_set_filetype(ae, AE_IFDIR);
archive_entry_set_size(ae, 512);
assertEqualIntA(a, ARCHIVE_FATAL, archive_write_header(a, ae));
err = archive_error_string(a);
assertEqualMem(err, "Raw format only supports filetype AE_IFREG", 43);
archive_entry_free(ae);
assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
assertEqualInt(ARCHIVE_OK, archive_write_free(a));
free(buff);
}
static int
pkg_read_archive_v2(FILE *fs, struct pkg *pkg,
struct archive **a, struct archive_entry **ae)
{
struct pkg_section_hdr sec;
int retcode = EPKG_OK;
struct pkg_archive_stdio_cookie *cookie;
FILE *cookied_fs;
static const int rd_chunk = 8192;
if (pkg_skip_to_section(fs, PKG_FORMAT_SECTION_PAYLOAD, &sec) != EPKG_OK) {
return (EPKG_END);
}
*a = archive_read_new();
/* We don't support anything but (zstd) + tar here */
archive_read_support_filter_none(*a);
archive_read_support_format_tar(*a);
cookie = xmalloc(sizeof(*cookie));
cookie->fs = fs;
cookie->remain = sec.size;
if (sec.flags & PKG_FORMAT_FLAGS_ZSTD) {
cookie->zstream = ZSTD_createDStream();
cookie->rdbuf = xmalloc(rd_chunk);
cookie->rd_len = rd_chunk;
cookie->rd_offset = 0;
}
else {
cookie->zstream = NULL;
}
cookied_fs = funopen(cookie, pkg_archive_cookie_read, NULL, NULL,
pkg_archive_cookie_close);
if (archive_read_open_FILE(*a, cookied_fs) != ARCHIVE_OK) {
pkg_emit_error("archive_read_open_fd: %s",
archive_error_string(*a));
retcode = EPKG_FATAL;
goto cleanup;
}
if (archive_read_next_header(*a, ae) != ARCHIVE_OK) {
pkg_emit_error("archive_read_next_header: %s",
archive_error_string(*a));
retcode = EPKG_FATAL;
goto cleanup;
}
cleanup:
if (retcode != EPKG_OK) {
if (*a != NULL) {
archive_read_close(*a);
archive_read_free(*a);
}
}
return (retcode);
}
