mirror of
https://github.com/opa334/TrollStore.git
synced 2024-12-04 15:00:39 +08:00
88 lines
2.6 KiB
Mathematica
88 lines
2.6 KiB
Mathematica
|
#import "unarchive.h"
|
||
|
|
|
||
|
|
#include <libarchive/archive.h>
|
||
|
|
#include <libarchive/archive_entry.h>
|
||
|
|
|
||
|
|
static int
|
||
|
|
copy_data(struct archive *ar, struct archive *aw)
|
||
|
|
{
|
||
|
|
int r;
|
||
|
|
const void *buff;
|
||
|
|
size_t size;
|
||
|
|
la_int64_t offset;
|
||
|
|
|
||
|
|
for (;;) {
|
||
|
|
r = archive_read_data_block(ar, &buff, &size, &offset);
|
||
|
|
if (r == ARCHIVE_EOF)
|
||
|
|
return (ARCHIVE_OK);
|
||
|
|
if (r < ARCHIVE_OK)
|
||
|
|
return (r);
|
||
|
|
r = archive_write_data_block(aw, buff, size, offset);
|
||
|
|
if (r < ARCHIVE_OK) {
|
||
|
|
fprintf(stderr, "%s\n", archive_error_string(aw));
|
||
|
|
return (r);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int extract(NSString* fileToExtract, NSString* extractionPath)
|
||
|
|
{
|
||
|
|
struct archive *a;
|
||
|
|
struct archive *ext;
|
||
|
|
struct archive_entry *entry;
|
||
|
|
int flags;
|
||
|
|
int r;
|
||
|
|
|
||
|
|
/* Select which attributes we want to restore. */
|
||
|
|
flags = ARCHIVE_EXTRACT_TIME;
|
||
|
|
flags |= ARCHIVE_EXTRACT_PERM;
|
||
|
|
flags |= ARCHIVE_EXTRACT_ACL;
|
||
|
|
flags |= ARCHIVE_EXTRACT_FFLAGS;
|
||
|
|
|
||
|
|
a = archive_read_new();
|
||
|
|
archive_read_support_format_all(a);
|
||
|
|
archive_read_support_filter_all(a);
|
||
|
|
ext = archive_write_disk_new();
|
||
|
|
archive_write_disk_set_options(ext, flags);
|
||
|
|
archive_write_disk_set_standard_lookup(ext);
|
||
|
|
if ((r = archive_read_open_filename(a, fileToExtract.UTF8String, 10240)))
|
||
|
|
return 1;
|
||
|
|
for (;;)
|
||
|
|
{
|
||
|
|
r = archive_read_next_header(a, &entry);
|
||
|
|
if (r == ARCHIVE_EOF)
|
||
|
|
break;
|
||
|
|
if (r < ARCHIVE_OK)
|
||
|
|
fprintf(stderr, "%s\n", archive_error_string(a));
|
||
|
|
if (r < ARCHIVE_WARN)
|
||
|
|
return 1;
|
||
|
|
|
||
|
|
NSString* currentFile = [NSString stringWithUTF8String:archive_entry_pathname(entry)];
|
||
|
|
NSString* fullOutputPath = [extractionPath stringByAppendingPathComponent:currentFile];
|
||
|
|
//printf("extracting %s to %s\n", currentFile.UTF8String, fullOutputPath.UTF8String);
|
||
|
|
archive_entry_set_pathname(entry, fullOutputPath.UTF8String);
|
||
|
|
|
||
|
|
r = archive_write_header(ext, entry);
|
||
|
|
if (r < ARCHIVE_OK)
|
||
|
|
fprintf(stderr, "%s\n", archive_error_string(ext));
|
||
|
|
else if (archive_entry_size(entry) > 0) {
|
||
|
|
r = copy_data(a, ext);
|
||
|
|
if (r < ARCHIVE_OK)
|
||
|
|
fprintf(stderr, "%s\n", archive_error_string(ext));
|
||
|
|
if (r < ARCHIVE_WARN)
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
r = archive_write_finish_entry(ext);
|
||
|
|
if (r < ARCHIVE_OK)
|
||
|
|
fprintf(stderr, "%s\n", archive_error_string(ext));
|
||
|
|
if (r < ARCHIVE_WARN)
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
archive_read_close(a);
|
||
|
|
archive_read_free(a);
|
||
|
|
archive_write_close(ext);
|
||
|
|
archive_write_free(ext);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|