123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434 |
- //
- // SSZipArchive.m
- // SSZipArchive
- //
- // Created by Sam Soffes on 7/21/10.
- //
- #import "SSZipArchive.h"
- #include "minizip/mz_compat.h"
- #include "minizip/mz_zip.h"
- #include "minizip/mz_os.h"
- #include <zlib.h>
- #include <sys/stat.h>
- NSString *const SSZipArchiveErrorDomain = @"SSZipArchiveErrorDomain";
- #define CHUNK 16384
- int _zipOpenEntry(zipFile entry, NSString *name, const zip_fileinfo *zipfi, int level, NSString *password, BOOL aes);
- BOOL _fileIsSymbolicLink(const unz_file_info *fileInfo);
- #ifndef API_AVAILABLE
- // Xcode 7- compatibility
- #define API_AVAILABLE(...)
- #endif
- @interface NSData(SSZipArchive)
- - (NSString *)_base64RFC4648 API_AVAILABLE(macos(10.9), ios(7.0), watchos(2.0), tvos(9.0));
- - (NSString *)_hexString;
- @end
- @interface NSString (SSZipArchive)
- - (NSString *)_sanitizedPath;
- @end
- @interface SSZipArchive ()
- - (instancetype)init NS_DESIGNATED_INITIALIZER;
- @end
- @implementation SSZipArchive
- {
- /// path for zip file
- NSString *_path;
- zipFile _zip;
- }
- #pragma mark - Password check
- + (BOOL)isFilePasswordProtectedAtPath:(NSString *)path {
- // Begin opening
- zipFile zip = unzOpen(path.fileSystemRepresentation);
- if (zip == NULL) {
- return NO;
- }
-
- BOOL passwordProtected = NO;
- int ret = unzGoToFirstFile(zip);
- if (ret == UNZ_OK) {
- do {
- ret = unzOpenCurrentFile(zip);
- if (ret != UNZ_OK) {
- // attempting with an arbitrary password to workaround `unzOpenCurrentFile` limitation on AES encrypted files
- ret = unzOpenCurrentFilePassword(zip, "");
- unzCloseCurrentFile(zip);
- if (ret == UNZ_OK || ret == MZ_PASSWORD_ERROR) {
- passwordProtected = YES;
- }
- break;
- }
- unz_file_info fileInfo = {};
- ret = unzGetCurrentFileInfo(zip, &fileInfo, NULL, 0, NULL, 0, NULL, 0);
- unzCloseCurrentFile(zip);
- if (ret != UNZ_OK) {
- break;
- } else if ((fileInfo.flag & MZ_ZIP_FLAG_ENCRYPTED) == 1) {
- passwordProtected = YES;
- break;
- }
-
- ret = unzGoToNextFile(zip);
- } while (ret == UNZ_OK);
- }
-
- unzClose(zip);
- return passwordProtected;
- }
- + (BOOL)isPasswordValidForArchiveAtPath:(NSString *)path password:(NSString *)pw error:(NSError **)error {
- if (error) {
- *error = nil;
- }
- zipFile zip = unzOpen(path.fileSystemRepresentation);
- if (zip == NULL) {
- if (error) {
- *error = [NSError errorWithDomain:SSZipArchiveErrorDomain
- code:SSZipArchiveErrorCodeFailedOpenZipFile
- userInfo:@{NSLocalizedDescriptionKey: @"failed to open zip file"}];
- }
- return NO;
- }
- // Initialize passwordValid to YES (No password required)
- BOOL passwordValid = YES;
- int ret = unzGoToFirstFile(zip);
- if (ret == UNZ_OK) {
- do {
- if (pw.length == 0) {
- ret = unzOpenCurrentFile(zip);
- } else {
- ret = unzOpenCurrentFilePassword(zip, [pw cStringUsingEncoding:NSUTF8StringEncoding]);
- }
- if (ret != UNZ_OK) {
- if (ret != MZ_PASSWORD_ERROR) {
- if (error) {
- *error = [NSError errorWithDomain:SSZipArchiveErrorDomain
- code:SSZipArchiveErrorCodeFailedOpenFileInZip
- userInfo:@{NSLocalizedDescriptionKey: @"failed to open file in zip archive"}];
- }
- }
- passwordValid = NO;
- break;
- }
- unz_file_info fileInfo = {};
- ret = unzGetCurrentFileInfo(zip, &fileInfo, NULL, 0, NULL, 0, NULL, 0);
- if (ret != UNZ_OK) {
- if (error) {
- *error = [NSError errorWithDomain:SSZipArchiveErrorDomain
- code:SSZipArchiveErrorCodeFileInfoNotLoadable
- userInfo:@{NSLocalizedDescriptionKey: @"failed to retrieve info for file"}];
- }
- passwordValid = NO;
- break;
- } else if ((fileInfo.flag & 1) == 1) {
- unsigned char buffer[10] = {0};
- int readBytes = unzReadCurrentFile(zip, buffer, (unsigned)MIN(10UL,fileInfo.uncompressed_size));
- if (readBytes < 0) {
- // Let's assume error Z_DATA_ERROR is caused by an invalid password
- // Let's assume other errors are caused by Content Not Readable
- if (readBytes != Z_DATA_ERROR) {
- if (error) {
- *error = [NSError errorWithDomain:SSZipArchiveErrorDomain
- code:SSZipArchiveErrorCodeFileContentNotReadable
- userInfo:@{NSLocalizedDescriptionKey: @"failed to read contents of file entry"}];
- }
- }
- passwordValid = NO;
- break;
- }
- passwordValid = YES;
- break;
- }
-
- unzCloseCurrentFile(zip);
- ret = unzGoToNextFile(zip);
- } while (ret == UNZ_OK);
- }
-
- unzClose(zip);
- return passwordValid;
- }
- + (NSNumber *)payloadSizeForArchiveAtPath:(NSString *)path error:(NSError **)error {
- if (error) {
- *error = nil;
- }
- zipFile zip = unzOpen(path.fileSystemRepresentation);
- if (zip == NULL) {
- if (error) {
- *error = [NSError errorWithDomain:SSZipArchiveErrorDomain
- code:SSZipArchiveErrorCodeFailedOpenZipFile
- userInfo:@{NSLocalizedDescriptionKey: @"failed to open zip file"}];
- }
- return @0;
- }
- unsigned long long totalSize = 0;
- int ret = unzGoToFirstFile(zip);
- if (ret == UNZ_OK) {
- do {
- ret = unzOpenCurrentFile(zip);
- if (ret != UNZ_OK) {
- if (error) {
- *error = [NSError errorWithDomain:SSZipArchiveErrorDomain
- code:SSZipArchiveErrorCodeFailedOpenFileInZip
- userInfo:@{NSLocalizedDescriptionKey: @"failed to open file in zip archive"}];
- }
- break;
- }
- unz_file_info fileInfo = {};
- ret = unzGetCurrentFileInfo(zip, &fileInfo, NULL, 0, NULL, 0, NULL, 0);
- if (ret != UNZ_OK) {
- if (error) {
- *error = [NSError errorWithDomain:SSZipArchiveErrorDomain
- code:SSZipArchiveErrorCodeFileInfoNotLoadable
- userInfo:@{NSLocalizedDescriptionKey: @"failed to retrieve info for file"}];
- }
- break;
- }
- totalSize += fileInfo.uncompressed_size;
- unzCloseCurrentFile(zip);
- ret = unzGoToNextFile(zip);
- } while (ret == UNZ_OK);
- }
- unzClose(zip);
- return [NSNumber numberWithUnsignedLongLong:totalSize];
- }
- #pragma mark - Unzipping
- + (BOOL)unzipFileAtPath:(NSString *)path toDestination:(NSString *)destination
- {
- return [self unzipFileAtPath:path toDestination:destination delegate:nil];
- }
- + (BOOL)unzipFileAtPath:(NSString *)path toDestination:(NSString *)destination overwrite:(BOOL)overwrite password:(nullable NSString *)password error:(NSError **)error
- {
- return [self unzipFileAtPath:path toDestination:destination preserveAttributes:YES overwrite:overwrite password:password error:error delegate:nil progressHandler:nil completionHandler:nil];
- }
- + (BOOL)unzipFileAtPath:(NSString *)path toDestination:(NSString *)destination delegate:(nullable id<SSZipArchiveDelegate>)delegate
- {
- return [self unzipFileAtPath:path toDestination:destination preserveAttributes:YES overwrite:YES password:nil error:nil delegate:delegate progressHandler:nil completionHandler:nil];
- }
- + (BOOL)unzipFileAtPath:(NSString *)path
- toDestination:(NSString *)destination
- overwrite:(BOOL)overwrite
- password:(nullable NSString *)password
- error:(NSError **)error
- delegate:(nullable id<SSZipArchiveDelegate>)delegate
- {
- return [self unzipFileAtPath:path toDestination:destination preserveAttributes:YES overwrite:overwrite password:password error:error delegate:delegate progressHandler:nil completionHandler:nil];
- }
- + (BOOL)unzipFileAtPath:(NSString *)path
- toDestination:(NSString *)destination
- overwrite:(BOOL)overwrite
- password:(NSString *)password
- progressHandler:(void (^)(NSString *entry, unz_file_info zipInfo, long entryNumber, long total))progressHandler
- completionHandler:(void (^)(NSString *path, BOOL succeeded, NSError * _Nullable error))completionHandler
- {
- return [self unzipFileAtPath:path toDestination:destination preserveAttributes:YES overwrite:overwrite password:password error:nil delegate:nil progressHandler:progressHandler completionHandler:completionHandler];
- }
- + (BOOL)unzipFileAtPath:(NSString *)path
- toDestination:(NSString *)destination
- progressHandler:(void (^_Nullable)(NSString *entry, unz_file_info zipInfo, long entryNumber, long total))progressHandler
- completionHandler:(void (^_Nullable)(NSString *path, BOOL succeeded, NSError * _Nullable error))completionHandler
- {
- return [self unzipFileAtPath:path toDestination:destination preserveAttributes:YES overwrite:YES password:nil error:nil delegate:nil progressHandler:progressHandler completionHandler:completionHandler];
- }
- + (BOOL)unzipFileAtPath:(NSString *)path
- toDestination:(NSString *)destination
- preserveAttributes:(BOOL)preserveAttributes
- overwrite:(BOOL)overwrite
- password:(nullable NSString *)password
- error:(NSError * *)error
- delegate:(nullable id<SSZipArchiveDelegate>)delegate
- {
- return [self unzipFileAtPath:path toDestination:destination preserveAttributes:preserveAttributes overwrite:overwrite password:password error:error delegate:delegate progressHandler:nil completionHandler:nil];
- }
- + (BOOL)unzipFileAtPath:(NSString *)path
- toDestination:(NSString *)destination
- preserveAttributes:(BOOL)preserveAttributes
- overwrite:(BOOL)overwrite
- password:(nullable NSString *)password
- error:(NSError **)error
- delegate:(nullable id<SSZipArchiveDelegate>)delegate
- progressHandler:(void (^_Nullable)(NSString *entry, unz_file_info zipInfo, long entryNumber, long total))progressHandler
- completionHandler:(void (^_Nullable)(NSString *path, BOOL succeeded, NSError * _Nullable error))completionHandler
- {
- return [self unzipFileAtPath:path toDestination:destination preserveAttributes:preserveAttributes overwrite:overwrite nestedZipLevel:0 password:password error:error delegate:delegate progressHandler:progressHandler completionHandler:completionHandler];
- }
- + (BOOL)unzipFileAtPath:(NSString *)path
- toDestination:(NSString *)destination
- preserveAttributes:(BOOL)preserveAttributes
- overwrite:(BOOL)overwrite
- nestedZipLevel:(NSInteger)nestedZipLevel
- password:(nullable NSString *)password
- error:(NSError **)error
- delegate:(nullable id<SSZipArchiveDelegate>)delegate
- progressHandler:(void (^_Nullable)(NSString *entry, unz_file_info zipInfo, long entryNumber, long total))progressHandler
- completionHandler:(void (^_Nullable)(NSString *path, BOOL succeeded, NSError * _Nullable error))completionHandler
- {
- // Guard against empty strings
- if (path.length == 0 || destination.length == 0)
- {
- NSDictionary *userInfo = @{NSLocalizedDescriptionKey: @"received invalid argument(s)"};
- NSError *err = [NSError errorWithDomain:SSZipArchiveErrorDomain code:SSZipArchiveErrorCodeInvalidArguments userInfo:userInfo];
- if (error)
- {
- *error = err;
- }
- if (completionHandler)
- {
- completionHandler(nil, NO, err);
- }
- return NO;
- }
-
- // Begin opening
- zipFile zip = unzOpen(path.fileSystemRepresentation);
- if (zip == NULL)
- {
- NSDictionary *userInfo = @{NSLocalizedDescriptionKey: @"failed to open zip file"};
- NSError *err = [NSError errorWithDomain:SSZipArchiveErrorDomain code:SSZipArchiveErrorCodeFailedOpenZipFile userInfo:userInfo];
- if (error)
- {
- *error = err;
- }
- if (completionHandler)
- {
- completionHandler(nil, NO, err);
- }
- return NO;
- }
-
- NSDictionary * fileAttributes = [[NSFileManager defaultManager] attributesOfItemAtPath:path error:nil];
- unsigned long long fileSize = [[fileAttributes objectForKey:NSFileSize] unsignedLongLongValue];
- unsigned long long currentPosition = 0;
-
- unz_global_info globalInfo = {};
- unzGetGlobalInfo(zip, &globalInfo);
-
- // Begin unzipping
- int ret = 0;
- ret = unzGoToFirstFile(zip);
- if (ret != UNZ_OK && ret != MZ_END_OF_LIST)
- {
- NSDictionary *userInfo = @{NSLocalizedDescriptionKey: @"failed to open first file in zip file"};
- NSError *err = [NSError errorWithDomain:SSZipArchiveErrorDomain code:SSZipArchiveErrorCodeFailedOpenFileInZip userInfo:userInfo];
- if (error)
- {
- *error = err;
- }
- if (completionHandler)
- {
- completionHandler(nil, NO, err);
- }
- unzClose(zip);
- return NO;
- }
-
- BOOL success = YES;
- BOOL canceled = NO;
- int crc_ret = 0;
- unsigned char buffer[4096] = {0};
- NSFileManager *fileManager = [NSFileManager defaultManager];
- NSMutableArray<NSDictionary *> *directoriesModificationDates = [[NSMutableArray alloc] init];
-
- // Message delegate
- if ([delegate respondsToSelector:@selector(zipArchiveWillUnzipArchiveAtPath:zipInfo:)]) {
- [delegate zipArchiveWillUnzipArchiveAtPath:path zipInfo:globalInfo];
- }
- if ([delegate respondsToSelector:@selector(zipArchiveProgressEvent:total:)]) {
- [delegate zipArchiveProgressEvent:currentPosition total:fileSize];
- }
-
- NSInteger currentFileNumber = -1;
- NSError *unzippingError;
- do {
- currentFileNumber++;
- if (ret == MZ_END_OF_LIST) {
- break;
- }
- @autoreleasepool {
- if (password.length == 0) {
- ret = unzOpenCurrentFile(zip);
- } else {
- ret = unzOpenCurrentFilePassword(zip, [password cStringUsingEncoding:NSUTF8StringEncoding]);
- }
-
- if (ret != UNZ_OK) {
- unzippingError = [NSError errorWithDomain:@"SSZipArchiveErrorDomain" code:SSZipArchiveErrorCodeFailedOpenFileInZip userInfo:@{NSLocalizedDescriptionKey: @"failed to open file in zip file"}];
- success = NO;
- break;
- }
-
- // Reading data and write to file
- unz_file_info fileInfo;
- memset(&fileInfo, 0, sizeof(unz_file_info));
-
- ret = unzGetCurrentFileInfo(zip, &fileInfo, NULL, 0, NULL, 0, NULL, 0);
- if (ret != UNZ_OK) {
- unzippingError = [NSError errorWithDomain:@"SSZipArchiveErrorDomain" code:SSZipArchiveErrorCodeFileInfoNotLoadable userInfo:@{NSLocalizedDescriptionKey: @"failed to retrieve info for file"}];
- success = NO;
- unzCloseCurrentFile(zip);
- break;
- }
-
- currentPosition += fileInfo.compressed_size;
-
- // Message delegate
- if ([delegate respondsToSelector:@selector(zipArchiveShouldUnzipFileAtIndex:totalFiles:archivePath:fileInfo:)]) {
- if (![delegate zipArchiveShouldUnzipFileAtIndex:currentFileNumber
- totalFiles:(NSInteger)globalInfo.number_entry
- archivePath:path
- fileInfo:fileInfo]) {
- success = NO;
- canceled = YES;
- break;
- }
- }
- if ([delegate respondsToSelector:@selector(zipArchiveWillUnzipFileAtIndex:totalFiles:archivePath:fileInfo:)]) {
- [delegate zipArchiveWillUnzipFileAtIndex:currentFileNumber totalFiles:(NSInteger)globalInfo.number_entry
- archivePath:path fileInfo:fileInfo];
- }
- if ([delegate respondsToSelector:@selector(zipArchiveProgressEvent:total:)]) {
- [delegate zipArchiveProgressEvent:(NSInteger)currentPosition total:(NSInteger)fileSize];
- }
-
- char *filename = (char *)malloc(fileInfo.size_filename + 1);
- if (filename == NULL)
- {
- success = NO;
- break;
- }
-
- unzGetCurrentFileInfo(zip, &fileInfo, filename, fileInfo.size_filename + 1, NULL, 0, NULL, 0);
- filename[fileInfo.size_filename] = '\0';
-
- BOOL fileIsSymbolicLink = _fileIsSymbolicLink(&fileInfo);
-
- NSString * strPath = [SSZipArchive _filenameStringWithCString:filename
- version_made_by:fileInfo.version
- general_purpose_flag:fileInfo.flag
- size:fileInfo.size_filename];
- if ([strPath hasPrefix:@"__MACOSX/"]) {
- // ignoring resource forks: https://superuser.com/questions/104500/what-is-macosx-folder
- unzCloseCurrentFile(zip);
- ret = unzGoToNextFile(zip);
- free(filename);
- continue;
- }
-
- // Check if it contains directory
- BOOL isDirectory = NO;
- if (filename[fileInfo.size_filename-1] == '/' || filename[fileInfo.size_filename-1] == '\\') {
- isDirectory = YES;
- }
- free(filename);
-
- // Sanitize paths in the file name.
- strPath = [strPath _sanitizedPath];
- if (!strPath.length) {
- // if filename data is unsalvageable, we default to currentFileNumber
- strPath = @(currentFileNumber).stringValue;
- }
-
- NSString *fullPath = [destination stringByAppendingPathComponent:strPath];
- NSError *err = nil;
- NSDictionary *directoryAttr;
- if (preserveAttributes) {
- NSDate *modDate = [[self class] _dateWithMSDOSFormat:(UInt32)fileInfo.mz_dos_date];
- directoryAttr = @{NSFileCreationDate: modDate, NSFileModificationDate: modDate};
- [directoriesModificationDates addObject: @{@"path": fullPath, @"modDate": modDate}];
- }
- if (isDirectory) {
- [fileManager createDirectoryAtPath:fullPath withIntermediateDirectories:YES attributes:directoryAttr error:&err];
- } else {
- [fileManager createDirectoryAtPath:fullPath.stringByDeletingLastPathComponent withIntermediateDirectories:YES attributes:directoryAttr error:&err];
- }
- if (err != nil) {
- if ([err.domain isEqualToString:NSCocoaErrorDomain] &&
- err.code == 640) {
- unzippingError = err;
- unzCloseCurrentFile(zip);
- success = NO;
- break;
- }
- NSLog(@"[SSZipArchive] Error: %@", err.localizedDescription);
- }
-
- if ([fileManager fileExistsAtPath:fullPath] && !isDirectory && !overwrite) {
- //FIXME: couldBe CRC Check?
- unzCloseCurrentFile(zip);
- ret = unzGoToNextFile(zip);
- continue;
- }
-
- if (isDirectory && !fileIsSymbolicLink) {
- // nothing to read/write for a directory
- } else if (!fileIsSymbolicLink) {
- // ensure we are not creating stale file entries
- int readBytes = unzReadCurrentFile(zip, buffer, 4096);
- if (readBytes >= 0) {
- FILE *fp = fopen(fullPath.fileSystemRepresentation, "wb");
- while (fp) {
- if (readBytes > 0) {
- if (0 == fwrite(buffer, readBytes, 1, fp)) {
- if (ferror(fp)) {
- NSString *message = [NSString stringWithFormat:@"Failed to write file (check your free space)"];
- NSLog(@"[SSZipArchive] %@", message);
- success = NO;
- unzippingError = [NSError errorWithDomain:@"SSZipArchiveErrorDomain" code:SSZipArchiveErrorCodeFailedToWriteFile userInfo:@{NSLocalizedDescriptionKey: message}];
- break;
- }
- }
- } else {
- break;
- }
- readBytes = unzReadCurrentFile(zip, buffer, 4096);
- if (readBytes < 0) {
- // Let's assume error Z_DATA_ERROR is caused by an invalid password
- // Let's assume other errors are caused by Content Not Readable
- success = NO;
- }
- }
-
- if (fp) {
- fclose(fp);
-
- if (nestedZipLevel
- && [fullPath.pathExtension.lowercaseString isEqualToString:@"zip"]
- && [self unzipFileAtPath:fullPath
- toDestination:fullPath.stringByDeletingLastPathComponent
- preserveAttributes:preserveAttributes
- overwrite:overwrite
- nestedZipLevel:nestedZipLevel - 1
- password:password
- error:nil
- delegate:nil
- progressHandler:nil
- completionHandler:nil]) {
- [directoriesModificationDates removeLastObject];
- [[NSFileManager defaultManager] removeItemAtPath:fullPath error:nil];
- } else if (preserveAttributes) {
-
- // Set the original datetime property
- if (fileInfo.mz_dos_date != 0) {
- NSDate *orgDate = [[self class] _dateWithMSDOSFormat:(UInt32)fileInfo.mz_dos_date];
- NSDictionary *attr = @{NSFileModificationDate: orgDate};
-
- if (attr) {
- if (![fileManager setAttributes:attr ofItemAtPath:fullPath error:nil]) {
- // Can't set attributes
- NSLog(@"[SSZipArchive] Failed to set attributes - whilst setting modification date");
- }
- }
- }
-
- // Set the original permissions on the file (+read/write to solve #293)
- uLong permissions = fileInfo.external_fa >> 16 | 0b110000000;
- if (permissions != 0) {
- // Store it into a NSNumber
- NSNumber *permissionsValue = @(permissions);
-
- // Retrieve any existing attributes
- NSMutableDictionary *attrs = [[NSMutableDictionary alloc] initWithDictionary:[fileManager attributesOfItemAtPath:fullPath error:nil]];
-
- // Set the value in the attributes dict
- [attrs setObject:permissionsValue forKey:NSFilePosixPermissions];
-
- // Update attributes
- if (![fileManager setAttributes:attrs ofItemAtPath:fullPath error:nil]) {
- // Unable to set the permissions attribute
- NSLog(@"[SSZipArchive] Failed to set attributes - whilst setting permissions");
- }
- }
- }
- }
- else
- {
- // if we couldn't open file descriptor we can validate global errno to see the reason
- int errnoSave = errno;
- BOOL isSeriousError = NO;
- switch (errnoSave) {
- case EISDIR:
- // Is a directory
- // assumed case
- break;
-
- case ENOSPC:
- case EMFILE:
- // No space left on device
- // or
- // Too many open files
- isSeriousError = YES;
- break;
-
- default:
- // ignore case
- // Just log the error
- {
- NSError *errorObject = [NSError errorWithDomain:NSPOSIXErrorDomain
- code:errnoSave
- userInfo:nil];
- NSLog(@"[SSZipArchive] Failed to open file on unzipping.(%@)", errorObject);
- }
- break;
- }
-
- if (isSeriousError) {
- // serious case
- unzippingError = [NSError errorWithDomain:NSPOSIXErrorDomain
- code:errnoSave
- userInfo:nil];
- unzCloseCurrentFile(zip);
- // Log the error
- NSLog(@"[SSZipArchive] Failed to open file on unzipping.(%@)", unzippingError);
- // Break unzipping
- success = NO;
- break;
- }
- }
- } else {
- // Let's assume error Z_DATA_ERROR is caused by an invalid password
- // Let's assume other errors are caused by Content Not Readable
- success = NO;
- break;
- }
- }
- else
- {
- // Assemble the path for the symbolic link
- NSMutableString *destinationPath = [NSMutableString string];
- int bytesRead = 0;
- while ((bytesRead = unzReadCurrentFile(zip, buffer, 4096)) > 0)
- {
- buffer[bytesRead] = 0;
- [destinationPath appendString:@((const char *)buffer)];
- }
- if (bytesRead < 0) {
- // Let's assume error Z_DATA_ERROR is caused by an invalid password
- // Let's assume other errors are caused by Content Not Readable
- success = NO;
- break;
- }
-
- // Check if the symlink exists and delete it if we're overwriting
- if (overwrite)
- {
- if ([fileManager fileExistsAtPath:fullPath])
- {
- NSError *localError = nil;
- BOOL removeSuccess = [fileManager removeItemAtPath:fullPath error:&localError];
- if (!removeSuccess)
- {
- NSString *message = [NSString stringWithFormat:@"Failed to delete existing symbolic link at \"%@\"", localError.localizedDescription];
- NSLog(@"[SSZipArchive] %@", message);
- success = NO;
- unzippingError = [NSError errorWithDomain:SSZipArchiveErrorDomain code:localError.code userInfo:@{NSLocalizedDescriptionKey: message}];
- }
- }
- }
-
- // Create the symbolic link (making sure it stays relative if it was relative before)
- int symlinkError = symlink([destinationPath cStringUsingEncoding:NSUTF8StringEncoding],
- [fullPath cStringUsingEncoding:NSUTF8StringEncoding]);
-
- if (symlinkError != 0)
- {
- // Bubble the error up to the completion handler
- NSString *message = [NSString stringWithFormat:@"Failed to create symbolic link at \"%@\" to \"%@\" - symlink() error code: %d", fullPath, destinationPath, errno];
- NSLog(@"[SSZipArchive] %@", message);
- success = NO;
- unzippingError = [NSError errorWithDomain:NSPOSIXErrorDomain code:symlinkError userInfo:@{NSLocalizedDescriptionKey: message}];
- }
- }
-
- crc_ret = unzCloseCurrentFile(zip);
- if (crc_ret == MZ_CRC_ERROR) {
- // CRC ERROR
- success = NO;
- break;
- }
- ret = unzGoToNextFile(zip);
-
- // Message delegate
- if ([delegate respondsToSelector:@selector(zipArchiveDidUnzipFileAtIndex:totalFiles:archivePath:fileInfo:)]) {
- [delegate zipArchiveDidUnzipFileAtIndex:currentFileNumber totalFiles:(NSInteger)globalInfo.number_entry
- archivePath:path fileInfo:fileInfo];
- } else if ([delegate respondsToSelector: @selector(zipArchiveDidUnzipFileAtIndex:totalFiles:archivePath:unzippedFilePath:)]) {
- [delegate zipArchiveDidUnzipFileAtIndex: currentFileNumber totalFiles: (NSInteger)globalInfo.number_entry
- archivePath:path unzippedFilePath: fullPath];
- }
-
- if (progressHandler)
- {
- progressHandler(strPath, fileInfo, currentFileNumber, globalInfo.number_entry);
- }
- }
- } while (ret == UNZ_OK && success);
-
- // Close
- unzClose(zip);
-
- // The process of decompressing the .zip archive causes the modification times on the folders
- // to be set to the present time. So, when we are done, they need to be explicitly set.
- // set the modification date on all of the directories.
- if (success && preserveAttributes) {
- NSError * err = nil;
- for (NSDictionary * d in directoriesModificationDates) {
- if (![[NSFileManager defaultManager] setAttributes:@{NSFileModificationDate: [d objectForKey:@"modDate"]} ofItemAtPath:[d objectForKey:@"path"] error:&err]) {
- NSLog(@"[SSZipArchive] Set attributes failed for directory: %@.", [d objectForKey:@"path"]);
- }
- if (err) {
- NSLog(@"[SSZipArchive] Error setting directory file modification date attribute: %@", err.localizedDescription);
- }
- }
- }
-
- // Message delegate
- if (success && [delegate respondsToSelector:@selector(zipArchiveDidUnzipArchiveAtPath:zipInfo:unzippedPath:)]) {
- [delegate zipArchiveDidUnzipArchiveAtPath:path zipInfo:globalInfo unzippedPath:destination];
- }
- // final progress event = 100%
- if (!canceled && [delegate respondsToSelector:@selector(zipArchiveProgressEvent:total:)]) {
- [delegate zipArchiveProgressEvent:fileSize total:fileSize];
- }
-
- NSError *retErr = nil;
- if (crc_ret == MZ_CRC_ERROR)
- {
- NSDictionary *userInfo = @{NSLocalizedDescriptionKey: @"crc check failed for file"};
- retErr = [NSError errorWithDomain:SSZipArchiveErrorDomain code:SSZipArchiveErrorCodeFileInfoNotLoadable userInfo:userInfo];
- }
-
- if (error) {
- if (unzippingError) {
- *error = unzippingError;
- }
- else {
- *error = retErr;
- }
- }
- if (completionHandler)
- {
- if (unzippingError) {
- completionHandler(path, success, unzippingError);
- }
- else
- {
- completionHandler(path, success, retErr);
- }
- }
- return success;
- }
- #pragma mark - Zipping
- + (BOOL)createZipFileAtPath:(NSString *)path withFilesAtPaths:(NSArray<NSString *> *)paths
- {
- return [SSZipArchive createZipFileAtPath:path withFilesAtPaths:paths withPassword:nil];
- }
- + (BOOL)createZipFileAtPath:(NSString *)path withContentsOfDirectory:(NSString *)directoryPath {
- return [SSZipArchive createZipFileAtPath:path withContentsOfDirectory:directoryPath withPassword:nil];
- }
- + (BOOL)createZipFileAtPath:(NSString *)path withContentsOfDirectory:(NSString *)directoryPath keepParentDirectory:(BOOL)keepParentDirectory {
- return [SSZipArchive createZipFileAtPath:path withContentsOfDirectory:directoryPath keepParentDirectory:keepParentDirectory withPassword:nil];
- }
- + (BOOL)createZipFileAtPath:(NSString *)path withFilesAtPaths:(NSArray<NSString *> *)paths withPassword:(NSString *)password {
- return [self createZipFileAtPath:path withFilesAtPaths:paths withPassword:password progressHandler:nil];
- }
- + (BOOL)createZipFileAtPath:(NSString *)path withFilesAtPaths:(NSArray<NSString *> *)paths withPassword:(NSString *)password progressHandler:(void(^ _Nullable)(NSUInteger entryNumber, NSUInteger total))progressHandler
- {
- SSZipArchive *zipArchive = [[SSZipArchive alloc] initWithPath:path];
- BOOL success = [zipArchive open];
- if (success) {
- NSUInteger total = paths.count, complete = 0;
- for (NSString *filePath in paths) {
- success &= [zipArchive writeFile:filePath withPassword:password];
- if (progressHandler) {
- complete++;
- progressHandler(complete, total);
- }
- }
- success &= [zipArchive close];
- }
- return success;
- }
- + (BOOL)createZipFileAtPath:(NSString *)path withContentsOfDirectory:(NSString *)directoryPath withPassword:(nullable NSString *)password {
- return [SSZipArchive createZipFileAtPath:path withContentsOfDirectory:directoryPath keepParentDirectory:NO withPassword:password];
- }
- + (BOOL)createZipFileAtPath:(NSString *)path withContentsOfDirectory:(NSString *)directoryPath keepParentDirectory:(BOOL)keepParentDirectory withPassword:(nullable NSString *)password {
- return [SSZipArchive createZipFileAtPath:path
- withContentsOfDirectory:directoryPath
- keepParentDirectory:keepParentDirectory
- withPassword:password
- andProgressHandler:nil
- ];
- }
- + (BOOL)createZipFileAtPath:(NSString *)path
- withContentsOfDirectory:(NSString *)directoryPath
- keepParentDirectory:(BOOL)keepParentDirectory
- withPassword:(nullable NSString *)password
- andProgressHandler:(void(^ _Nullable)(NSUInteger entryNumber, NSUInteger total))progressHandler {
- return [self createZipFileAtPath:path withContentsOfDirectory:directoryPath keepParentDirectory:keepParentDirectory compressionLevel:Z_DEFAULT_COMPRESSION password:password AES:YES progressHandler:progressHandler];
- }
- + (BOOL)createZipFileAtPath:(NSString *)path
- withContentsOfDirectory:(NSString *)directoryPath
- keepParentDirectory:(BOOL)keepParentDirectory
- compressionLevel:(int)compressionLevel
- password:(nullable NSString *)password
- AES:(BOOL)aes
- progressHandler:(void(^ _Nullable)(NSUInteger entryNumber, NSUInteger total))progressHandler {
-
- SSZipArchive *zipArchive = [[SSZipArchive alloc] initWithPath:path];
- BOOL success = [zipArchive open];
- if (success) {
- // use a local fileManager (queue/thread compatibility)
- NSFileManager *fileManager = [[NSFileManager alloc] init];
- NSDirectoryEnumerator *dirEnumerator = [fileManager enumeratorAtPath:directoryPath];
- NSArray<NSString *> *allObjects = dirEnumerator.allObjects;
- NSUInteger total = allObjects.count, complete = 0;
- if (keepParentDirectory && !total) {
- allObjects = @[@""];
- total = 1;
- }
- for (__strong NSString *fileName in allObjects) {
- NSString *fullFilePath = [directoryPath stringByAppendingPathComponent:fileName];
- if ([fullFilePath isEqualToString:path]) {
- NSLog(@"[SSZipArchive] the archive path and the file path: %@ are the same, which is forbidden.", fullFilePath);
- continue;
- }
-
- if (keepParentDirectory) {
- fileName = [directoryPath.lastPathComponent stringByAppendingPathComponent:fileName];
- }
-
- BOOL isDir;
- [fileManager fileExistsAtPath:fullFilePath isDirectory:&isDir];
- if (!isDir) {
- // file
- success &= [zipArchive writeFileAtPath:fullFilePath withFileName:fileName compressionLevel:compressionLevel password:password AES:aes];
- } else {
- // directory
- if (![fileManager enumeratorAtPath:fullFilePath].nextObject) {
- // empty directory
- success &= [zipArchive writeFolderAtPath:fullFilePath withFolderName:fileName withPassword:password];
- }
- }
- if (progressHandler) {
- complete++;
- progressHandler(complete, total);
- }
- }
- success &= [zipArchive close];
- }
- return success;
- }
- + (BOOL)createZipFileAtPath:(NSString *)path withFilesAtPaths:(NSArray<NSString *> *)paths withPassword:(nullable NSString *)password keepSymlinks:(BOOL)keeplinks {
- if (!keeplinks) {
- return [SSZipArchive createZipFileAtPath:path withFilesAtPaths:paths withPassword:password];
- } else {
- SSZipArchive *zipArchive = [[SSZipArchive alloc] initWithPath:path];
- BOOL success = [zipArchive open];
- if (success) {
- for (NSString *filePath in paths) {
- //is symlink
- if (mz_os_is_symlink(filePath.fileSystemRepresentation) == MZ_OK) {
- success &= [zipArchive writeSymlinkFileAtPath:filePath withFileName:nil compressionLevel:Z_DEFAULT_COMPRESSION password:password AES:YES];
- } else {
- success &= [zipArchive writeFile:filePath withPassword:password];
- }
- }
- success &= [zipArchive close];
- }
- return success;
- }
- }
- + (BOOL)createZipFileAtPath:(NSString *)path
- withContentsOfDirectory:(NSString *)directoryPath
- keepParentDirectory:(BOOL)keepParentDirectory
- compressionLevel:(int)compressionLevel
- password:(nullable NSString *)password
- AES:(BOOL)aes
- progressHandler:(void(^ _Nullable)(NSUInteger entryNumber, NSUInteger total))progressHandler
- keepSymlinks:(BOOL)keeplinks {
- if (!keeplinks) {
- return [SSZipArchive createZipFileAtPath:path
- withContentsOfDirectory:directoryPath
- keepParentDirectory:keepParentDirectory
- compressionLevel:compressionLevel
- password:password
- AES:aes
- progressHandler:progressHandler];
- } else {
- SSZipArchive *zipArchive = [[SSZipArchive alloc] initWithPath:path];
- BOOL success = [zipArchive open];
- if (success) {
- // use a local fileManager (queue/thread compatibility)
- NSFileManager *fileManager = [[NSFileManager alloc] init];
- NSDirectoryEnumerator *dirEnumerator = [fileManager enumeratorAtPath:directoryPath];
- NSArray<NSString *> *allObjects = dirEnumerator.allObjects;
- NSUInteger total = allObjects.count, complete = 0;
- if (keepParentDirectory && !total) {
- allObjects = @[@""];
- total = 1;
- }
- for (__strong NSString *fileName in allObjects) {
- NSString *fullFilePath = [directoryPath stringByAppendingPathComponent:fileName];
-
- if (keepParentDirectory) {
- fileName = [directoryPath.lastPathComponent stringByAppendingPathComponent:fileName];
- }
- //is symlink
- BOOL isSymlink = NO;
- if (mz_os_is_symlink(fullFilePath.fileSystemRepresentation) == MZ_OK)
- isSymlink = YES;
- BOOL isDir;
- [fileManager fileExistsAtPath:fullFilePath isDirectory:&isDir];
- if (!isDir || isSymlink) {
- // file or symlink
- if (!isSymlink) {
- success &= [zipArchive writeFileAtPath:fullFilePath withFileName:fileName compressionLevel:compressionLevel password:password AES:aes];
- } else {
- success &= [zipArchive writeSymlinkFileAtPath:fullFilePath withFileName:fileName compressionLevel:compressionLevel password:password AES:aes];
- }
- } else {
- // directory
- if (![fileManager enumeratorAtPath:fullFilePath].nextObject) {
- // empty directory
- success &= [zipArchive writeFolderAtPath:fullFilePath withFolderName:fileName withPassword:password];
- }
- }
- if (progressHandler) {
- complete++;
- progressHandler(complete, total);
- }
- }
- success &= [zipArchive close];
- }
- return success;
- }
- }
- - (BOOL)writeSymlinkFileAtPath:(NSString *)path withFileName:(nullable NSString *)fileName compressionLevel:(int)compressionLevel password:(nullable NSString *)password AES:(BOOL)aes
- {
- NSAssert((_zip != NULL), @"Attempting to write to an archive which was never opened");
- //read symlink
- char link_path[1024];
- int32_t err = MZ_OK;
- err = mz_os_read_symlink(path.fileSystemRepresentation, link_path, sizeof(link_path));
- if (err != MZ_OK) {
- NSLog(@"[SSZipArchive] Failed to read sylink");
- return NO;
- }
- if (!fileName) {
- fileName = path.lastPathComponent;
- }
-
- zip_fileinfo zipInfo = {};
- [SSZipArchive zipInfo:&zipInfo setAttributesOfItemAtPath:path];
-
- //unpdate zipInfo.external_fa
- uint32_t target_attrib = 0;
- uint32_t src_attrib = 0;
- uint32_t src_sys = 0;
- mz_os_get_file_attribs(path.fileSystemRepresentation, &src_attrib);
- src_sys = MZ_HOST_SYSTEM(MZ_VERSION_MADEBY);
- if ((src_sys != MZ_HOST_SYSTEM_MSDOS) && (src_sys != MZ_HOST_SYSTEM_WINDOWS_NTFS)) {
- /* High bytes are OS specific attributes, low byte is always DOS attributes */
- if (mz_zip_attrib_convert(src_sys, src_attrib, MZ_HOST_SYSTEM_MSDOS, &target_attrib) == MZ_OK)
- zipInfo.external_fa = target_attrib;
- zipInfo.external_fa |= (src_attrib << 16);
- } else {
- zipInfo.external_fa = src_attrib;
- }
- uint16_t version_madeby = 3 << 8;//UNIX
- int error = zipOpenNewFileInZip5(_zip, fileName.fileSystemRepresentation, &zipInfo, NULL, 0, NULL, 0, NULL, Z_DEFLATED, compressionLevel, 0, -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, password.UTF8String, aes, version_madeby, 0, 0);
- zipWriteInFileInZip(_zip, link_path, (uint32_t)strlen(link_path));
- zipCloseFileInZip(_zip);
- return error == ZIP_OK;
- }
- // disabling `init` because designated initializer is `initWithPath:`
- - (instancetype)init { @throw nil; }
- // designated initializer
- - (instancetype)initWithPath:(NSString *)path
- {
- if ((self = [super init])) {
- _path = [path copy];
- }
- return self;
- }
- - (BOOL)open
- {
- NSAssert((_zip == NULL), @"Attempting to open an archive which is already open");
- _zip = zipOpen(_path.fileSystemRepresentation, APPEND_STATUS_CREATE);
- return (NULL != _zip);
- }
- - (BOOL)openForAppending
- {
- NSAssert((_zip == NULL), @"Attempting to open an archive which is already open");
- _zip = zipOpen(_path.fileSystemRepresentation, APPEND_STATUS_ADDINZIP);
- return (NULL != _zip);
- }
- - (BOOL)writeFolderAtPath:(NSString *)path withFolderName:(NSString *)folderName withPassword:(nullable NSString *)password
- {
- NSAssert((_zip != NULL), @"Attempting to write to an archive which was never opened");
-
- zip_fileinfo zipInfo = {};
-
- [SSZipArchive zipInfo:&zipInfo setAttributesOfItemAtPath:path];
-
- int error = _zipOpenEntry(_zip, [folderName stringByAppendingString:@"/"], &zipInfo, Z_NO_COMPRESSION, password, NO);
- const void *buffer = NULL;
- zipWriteInFileInZip(_zip, buffer, 0);
- zipCloseFileInZip(_zip);
- return error == ZIP_OK;
- }
- - (BOOL)writeFile:(NSString *)path withPassword:(nullable NSString *)password
- {
- return [self writeFileAtPath:path withFileName:nil withPassword:password];
- }
- - (BOOL)writeFileAtPath:(NSString *)path withFileName:(nullable NSString *)fileName withPassword:(nullable NSString *)password
- {
- return [self writeFileAtPath:path withFileName:fileName compressionLevel:Z_DEFAULT_COMPRESSION password:password AES:YES];
- }
- // supports writing files with logical folder/directory structure
- // *path* is the absolute path of the file that will be compressed
- // *fileName* is the relative name of the file how it is stored within the zip e.g. /folder/subfolder/text1.txt
- - (BOOL)writeFileAtPath:(NSString *)path withFileName:(nullable NSString *)fileName compressionLevel:(int)compressionLevel password:(nullable NSString *)password AES:(BOOL)aes
- {
- NSAssert((_zip != NULL), @"Attempting to write to an archive which was never opened");
-
- FILE *input = fopen(path.fileSystemRepresentation, "r");
- if (NULL == input) {
- return NO;
- }
-
- if (!fileName) {
- fileName = path.lastPathComponent;
- }
-
- zip_fileinfo zipInfo = {};
-
- [SSZipArchive zipInfo:&zipInfo setAttributesOfItemAtPath:path];
-
- void *buffer = malloc(CHUNK);
- if (buffer == NULL)
- {
- fclose(input);
- return NO;
- }
-
- int error = _zipOpenEntry(_zip, fileName, &zipInfo, compressionLevel, password, aes);
-
- while (!feof(input) && !ferror(input))
- {
- unsigned int len = (unsigned int) fread(buffer, 1, CHUNK, input);
- zipWriteInFileInZip(_zip, buffer, len);
- }
-
- zipCloseFileInZip(_zip);
- free(buffer);
- fclose(input);
- return error == ZIP_OK;
- }
- - (BOOL)writeData:(NSData *)data filename:(nullable NSString *)filename withPassword:(nullable NSString *)password
- {
- return [self writeData:data filename:filename compressionLevel:Z_DEFAULT_COMPRESSION password:password AES:YES];
- }
- - (BOOL)writeData:(NSData *)data filename:(nullable NSString *)filename compressionLevel:(int)compressionLevel password:(nullable NSString *)password AES:(BOOL)aes
- {
- if (!_zip) {
- return NO;
- }
- if (!data) {
- return NO;
- }
- zip_fileinfo zipInfo = {};
- [SSZipArchive zipInfo:&zipInfo setDate:[NSDate date]];
-
- int error = _zipOpenEntry(_zip, filename, &zipInfo, compressionLevel, password, aes);
-
- zipWriteInFileInZip(_zip, data.bytes, (unsigned int)data.length);
-
- zipCloseFileInZip(_zip);
- return error == ZIP_OK;
- }
- - (BOOL)close
- {
- NSAssert((_zip != NULL), @"[SSZipArchive] Attempting to close an archive which was never opened");
- int error = zipClose(_zip, NULL);
- _zip = nil;
- return error == ZIP_OK;
- }
- #pragma mark - Private
- + (NSString *)_filenameStringWithCString:(const char *)filename
- version_made_by:(uint16_t)version_made_by
- general_purpose_flag:(uint16_t)flag
- size:(uint16_t)size_filename {
-
- // Respect Language encoding flag only reading filename as UTF-8 when this is set
- // when file entry created on dos system.
- //
- // https://pkware.cachefly.net/webdocs/casestudies/APPNOTE.TXT
- // Bit 11: Language encoding flag (EFS). If this bit is set,
- // the filename and comment fields for this file
- // MUST be encoded using UTF-8. (see APPENDIX D)
- uint16_t made_by = version_made_by >> 8;
- BOOL made_on_dos = made_by == 0;
- BOOL languageEncoding = (flag & (1 << 11)) != 0;
- if (!languageEncoding && made_on_dos) {
- // APPNOTE.TXT D.1:
- // D.2 If general purpose bit 11 is unset, the file name and comment should conform
- // to the original ZIP character encoding. If general purpose bit 11 is set, the
- // filename and comment must support The Unicode Standard, Version 4.1.0 or
- // greater using the character encoding form defined by the UTF-8 storage
- // specification. The Unicode Standard is published by the The Unicode
- // Consortium (www.unicode.org). UTF-8 encoded data stored within ZIP files
- // is expected to not include a byte order mark (BOM).
-
- // Code Page 437 corresponds to kCFStringEncodingDOSLatinUS
- NSStringEncoding encoding = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingDOSLatinUS);
- NSString* strPath = [NSString stringWithCString:filename encoding:encoding];
- if (strPath) {
- return strPath;
- }
- }
-
- // attempting unicode encoding
- NSString * strPath = @(filename);
- if (strPath) {
- return strPath;
- }
-
- // if filename is non-unicode, detect and transform Encoding
- NSData *data = [NSData dataWithBytes:(const void *)filename length:sizeof(unsigned char) * size_filename];
- // Testing availability of @available (https://stackoverflow.com/a/46927445/1033581)
- #if __clang_major__ < 9
- // Xcode 8-
- if (floor(NSFoundationVersionNumber) > NSFoundationVersionNumber10_9_2) {
- #else
- // Xcode 9+
- if (@available(macOS 10.10, iOS 8.0, watchOS 2.0, tvOS 9.0, *)) {
- #endif
- // supported encodings are in [NSString availableStringEncodings]
- [NSString stringEncodingForData:data encodingOptions:nil convertedString:&strPath usedLossyConversion:nil];
- } else {
- // fallback to a simple manual detect for macOS 10.9 or older
- NSArray<NSNumber *> *encodings = @[@(kCFStringEncodingGB_18030_2000), @(kCFStringEncodingShiftJIS)];
- for (NSNumber *encoding in encodings) {
- strPath = [NSString stringWithCString:filename encoding:(NSStringEncoding)CFStringConvertEncodingToNSStringEncoding(encoding.unsignedIntValue)];
- if (strPath) {
- break;
- }
- }
- }
- if (strPath) {
- return strPath;
- }
-
- // if filename encoding is non-detected, we default to something based on data
- // _hexString is more readable than _base64RFC4648 for debugging unknown encodings
- strPath = [data _hexString];
- return strPath;
- }
- + (void)zipInfo:(zip_fileinfo *)zipInfo setAttributesOfItemAtPath:(NSString *)path
- {
- NSDictionary *attr = [[NSFileManager defaultManager] attributesOfItemAtPath:path error: nil];
- if (attr)
- {
- NSDate *fileDate = (NSDate *)[attr objectForKey:NSFileModificationDate];
- if (fileDate)
- {
- [self zipInfo:zipInfo setDate:fileDate];
- }
-
- // Write permissions into the external attributes, for details on this see here: https://unix.stackexchange.com/a/14727
- // Get the permissions value from the files attributes
- NSNumber *permissionsValue = (NSNumber *)[attr objectForKey:NSFilePosixPermissions];
- if (permissionsValue != nil) {
- // Get the short value for the permissions
- short permissionsShort = permissionsValue.shortValue;
-
- // Convert this into an octal by adding 010000, 010000 being the flag for a regular file
- NSInteger permissionsOctal = 0100000 + permissionsShort;
-
- // Convert this into a long value
- uLong permissionsLong = @(permissionsOctal).unsignedLongValue;
-
- // Store this into the external file attributes once it has been shifted 16 places left to form part of the second from last byte
-
- // Casted back to an unsigned int to match type of external_fa in minizip
- zipInfo->external_fa = (unsigned int)(permissionsLong << 16L);
- }
- }
- }
- + (void)zipInfo:(zip_fileinfo *)zipInfo setDate:(NSDate *)date
- {
- NSCalendar *currentCalendar = SSZipArchive._gregorian;
- NSCalendarUnit flags = NSCalendarUnitYear | NSCalendarUnitMonth | NSCalendarUnitDay | NSCalendarUnitHour | NSCalendarUnitMinute | NSCalendarUnitSecond;
- NSDateComponents *components = [currentCalendar components:flags fromDate:date];
- struct tm tmz_date;
- tmz_date.tm_sec = (unsigned int)components.second;
- tmz_date.tm_min = (unsigned int)components.minute;
- tmz_date.tm_hour = (unsigned int)components.hour;
- tmz_date.tm_mday = (unsigned int)components.day;
- // ISO/IEC 9899 struct tm is 0-indexed for January but NSDateComponents for gregorianCalendar is 1-indexed for January
- tmz_date.tm_mon = (unsigned int)components.month - 1;
- // ISO/IEC 9899 struct tm is 0-indexed for AD 1900 but NSDateComponents for gregorianCalendar is 1-indexed for AD 1
- tmz_date.tm_year = (unsigned int)components.year - 1900;
- zipInfo->mz_dos_date = mz_zip_tm_to_dosdate(&tmz_date);
- }
- + (NSCalendar *)_gregorian
- {
- static NSCalendar *gregorian;
- static dispatch_once_t onceToken;
- dispatch_once(&onceToken, ^{
- gregorian = [[NSCalendar alloc] initWithCalendarIdentifier:NSCalendarIdentifierGregorian];
- });
-
- return gregorian;
- }
- // Format from http://newsgroups.derkeiler.com/Archive/Comp/comp.os.msdos.programmer/2009-04/msg00060.html
- // Two consecutive words, or a longword, YYYYYYYMMMMDDDDD hhhhhmmmmmmsssss
- // YYYYYYY is years from 1980 = 0
- // sssss is (seconds/2).
- //
- // 3658 = 0011 0110 0101 1000 = 0011011 0010 11000 = 27 2 24 = 2007-02-24
- // 7423 = 0111 0100 0010 0011 - 01110 100001 00011 = 14 33 3 = 14:33:06
- + (NSDate *)_dateWithMSDOSFormat:(UInt32)msdosDateTime
- {
- // the whole `_dateWithMSDOSFormat:` method is equivalent but faster than this one line,
- // essentially because `mktime` is slow:
- //NSDate *date = [NSDate dateWithTimeIntervalSince1970:dosdate_to_time_t(msdosDateTime)];
- static const UInt32 kYearMask = 0xFE000000;
- static const UInt32 kMonthMask = 0x1E00000;
- static const UInt32 kDayMask = 0x1F0000;
- static const UInt32 kHourMask = 0xF800;
- static const UInt32 kMinuteMask = 0x7E0;
- static const UInt32 kSecondMask = 0x1F;
-
- NSAssert(0xFFFFFFFF == (kYearMask | kMonthMask | kDayMask | kHourMask | kMinuteMask | kSecondMask), @"[SSZipArchive] MSDOS date masks don't add up");
-
- NSDateComponents *components = [[NSDateComponents alloc] init];
- components.year = 1980 + ((msdosDateTime & kYearMask) >> 25);
- components.month = (msdosDateTime & kMonthMask) >> 21;
- components.day = (msdosDateTime & kDayMask) >> 16;
- components.hour = (msdosDateTime & kHourMask) >> 11;
- components.minute = (msdosDateTime & kMinuteMask) >> 5;
- components.second = (msdosDateTime & kSecondMask) * 2;
-
- NSDate *date = [self._gregorian dateFromComponents:components];
- return date;
- }
- @end
- int _zipOpenEntry(zipFile entry, NSString *name, const zip_fileinfo *zipfi, int level, NSString *password, BOOL aes)
- {
- // https://pkware.cachefly.net/webdocs/casestudies/APPNOTE.TXT
- uint16_t made_on_darwin = 19 << 8;
- //MZ_ZIP_FLAG_UTF8
- uint16_t flag_base = 1 << 11;
- return zipOpenNewFileInZip5(entry, name.fileSystemRepresentation, zipfi, NULL, 0, NULL, 0, NULL, Z_DEFLATED, level, 0, -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, password.UTF8String, aes, made_on_darwin, flag_base, 1);
- }
- #pragma mark - Private tools for file info
- BOOL _fileIsSymbolicLink(const unz_file_info *fileInfo)
- {
- //
- // Determine whether this is a symbolic link:
- // - File is stored with 'version made by' value of UNIX (3),
- // as per https://www.pkware.com/documents/casestudies/APPNOTE.TXT
- // in the upper byte of the version field.
- // - BSD4.4 st_mode constants are stored in the high 16 bits of the
- // external file attributes (defacto standard, verified against libarchive)
- //
- // The original constants can be found here:
- // https://minnie.tuhs.org/cgi-bin/utree.pl?file=4.4BSD/usr/include/sys/stat.h
- //
- const uLong ZipUNIXVersion = 3;
- const uLong BSD_SFMT = 0170000;
- const uLong BSD_IFLNK = 0120000;
-
- BOOL fileIsSymbolicLink = ((fileInfo->version >> 8) == ZipUNIXVersion) && BSD_IFLNK == (BSD_SFMT & (fileInfo->external_fa >> 16));
- return fileIsSymbolicLink;
- }
- #pragma mark - Private tools for unreadable encodings
- @implementation NSData (SSZipArchive)
- // `base64EncodedStringWithOptions` uses a base64 alphabet with '+' and '/'.
- // we got those alternatives to make it compatible with filenames: https://en.wikipedia.org/wiki/Base64
- // * modified Base64 encoding for IMAP mailbox names (RFC 3501): uses '+' and ','
- // * modified Base64 for URL and filenames (RFC 4648): uses '-' and '_'
- - (NSString *)_base64RFC4648
- {
- NSString *strName = [self base64EncodedStringWithOptions:0];
- strName = [strName stringByReplacingOccurrencesOfString:@"+" withString:@"-"];
- strName = [strName stringByReplacingOccurrencesOfString:@"/" withString:@"_"];
- return strName;
- }
- // initWithBytesNoCopy from NSProgrammer, Jan 25 '12: https://stackoverflow.com/a/9009321/1033581
- // hexChars from Peter, Aug 19 '14: https://stackoverflow.com/a/25378464/1033581
- // not implemented as too lengthy: a potential mapping improvement from Moose, Nov 3 '15: https://stackoverflow.com/a/33501154/1033581
- - (NSString *)_hexString
- {
- const char *hexChars = "0123456789ABCDEF";
- NSUInteger length = self.length;
- const unsigned char *bytes = self.bytes;
- char *chars = malloc(length * 2);
- if (chars == NULL) {
- // we directly raise an exception instead of using NSAssert to make sure assertion is not disabled as this is irrecoverable
- [NSException raise:@"NSInternalInconsistencyException" format:@"failed malloc" arguments:nil];
- return nil;
- }
- char *s = chars;
- NSUInteger i = length;
- while (i--) {
- *s++ = hexChars[*bytes >> 4];
- *s++ = hexChars[*bytes & 0xF];
- bytes++;
- }
- NSString *str = [[NSString alloc] initWithBytesNoCopy:chars
- length:length * 2
- encoding:NSASCIIStringEncoding
- freeWhenDone:YES];
- return str;
- }
- @end
- #pragma mark Private tools for security
- @implementation NSString (SSZipArchive)
- // One implementation alternative would be to use the algorithm found at mz_path_resolve from https://github.com/nmoinvaz/minizip/blob/dev/mz_os.c,
- // but making sure to work with unichar values and not ascii values to avoid breaking Unicode characters containing 2E ('.') or 2F ('/') in their decomposition
- /// Sanitize path traversal characters to prevent directory backtracking. Ignoring these characters mimicks the default behavior of the Unarchiving tool on macOS.
- - (NSString *)_sanitizedPath
- {
- // Change Windows paths to Unix paths: https://en.wikipedia.org/wiki/Path_(computing)
- // Possible improvement: only do this if the archive was created on a non-Unix system
- NSString *strPath = [self stringByReplacingOccurrencesOfString:@"\\" withString:@"/"];
-
- // Percent-encode file path (where path is defined by https://tools.ietf.org/html/rfc8089)
- // The key part is to allow characters "." and "/" and disallow "%".
- // CharacterSet.urlPathAllowed seems to do the job
- #if (__MAC_OS_X_VERSION_MIN_REQUIRED >= 1090 || __IPHONE_OS_VERSION_MIN_REQUIRED >= 70000 || __WATCH_OS_VERSION_MIN_REQUIRED >= 20000 || __TV_OS_VERSION_MIN_REQUIRED >= 90000)
- strPath = [strPath stringByAddingPercentEncodingWithAllowedCharacters:NSCharacterSet.URLPathAllowedCharacterSet];
- #else
- // Testing availability of @available (https://stackoverflow.com/a/46927445/1033581)
- #if __clang_major__ < 9
- // Xcode 8-
- if (floor(NSFoundationVersionNumber) > NSFoundationVersionNumber10_8_4) {
- #else
- // Xcode 9+
- if (@available(macOS 10.9, iOS 7.0, watchOS 2.0, tvOS 9.0, *)) {
- #endif
- strPath = [strPath stringByAddingPercentEncodingWithAllowedCharacters:NSCharacterSet.URLPathAllowedCharacterSet];
- } else {
- strPath = [strPath stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
- }
- #endif
-
- // `NSString.stringByAddingPercentEncodingWithAllowedCharacters:` may theorically fail: https://stackoverflow.com/questions/33558933/
- // But because we auto-detect encoding using `NSString.stringEncodingForData:encodingOptions:convertedString:usedLossyConversion:`,
- // we likely already prevent UTF-16, UTF-32 and invalid Unicode in the form of unpaired surrogate chars: https://stackoverflow.com/questions/53043876/
- // To be on the safe side, we will still perform a guard check.
- if (strPath == nil) {
- return nil;
- }
-
- // Add scheme "file:///" to support sanitation on names with a colon like "file:a/../../../usr/bin"
- strPath = [@"file:///" stringByAppendingString:strPath];
-
- // Sanitize path traversal characters to prevent directory backtracking. Ignoring these characters mimicks the default behavior of the Unarchiving tool on macOS.
- // "../../../../../../../../../../../tmp/test.txt" -> "tmp/test.txt"
- // "a/b/../c.txt" -> "a/c.txt"
- strPath = [NSURL URLWithString:strPath].standardizedURL.absoluteString;
-
- // Remove the "file:///" scheme
- strPath = [strPath substringFromIndex:8];
-
- // Remove the percent-encoding
- #if (__MAC_OS_X_VERSION_MIN_REQUIRED >= 1090 || __IPHONE_OS_VERSION_MIN_REQUIRED >= 70000 || __WATCH_OS_VERSION_MIN_REQUIRED >= 20000 || __TV_OS_VERSION_MIN_REQUIRED >= 90000)
- strPath = strPath.stringByRemovingPercentEncoding;
- #else
- // Testing availability of @available (https://stackoverflow.com/a/46927445/1033581)
- #if __clang_major__ < 9
- // Xcode 8-
- if (floor(NSFoundationVersionNumber) > NSFoundationVersionNumber10_8_4) {
- #else
- // Xcode 9+
- if (@available(macOS 10.9, iOS 7.0, watchOS 2.0, tvOS 9.0, *)) {
- #endif
- strPath = strPath.stringByRemovingPercentEncoding;
- } else {
- strPath = [strPath stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
- }
- #endif
-
- return strPath;
- }
- @end
|