/*
* Copyright (c) 2008, 2009, 2010, 2011, 2012, 2013, 2014
* Jonathan Schleifer <js@webkeks.org>
*
* All rights reserved.
*
* This file is part of ObjFW. It may be distributed under the terms of the
* Q Public License 1.0, which can be found in the file LICENSE.QPL included in
* the packaging of this file.
*
* Alternatively, it may be distributed under the terms of the GNU General
* Public License, either version 2 or 3, which can be found in the file
* LICENSE.GPLv2 or LICENSE.GPLv3 respectively included in the packaging of this
* file.
*/
#include <string.h>
#import "OFApplication.h"
#import "OFArray.h"
#import "OFDate.h"
#import "OFDictionary.h"
#import "OFFile.h"
#import "OFOptionsParser.h"
#import "OFSet.h"
#import "OFStdIOStream.h"
#import "OFZIPArchive.h"
#import "OFZIPArchiveEntry.h"
#import "OFCreateDirectoryFailedException.h"
#import "OFInvalidFormatException.h"
#import "OFOpenFileFailedException.h"
#import "autorelease.h"
#import "macros.h"
#define BUFFER_SIZE 4096
@interface OFZIP: OFObject
{
int_fast8_t _override, _outputLevel;
int _exitStatus;
}
- (OFZIPArchive*)openArchiveWithPath: (OFString*)path;
- (void)listFilesInArchive: (OFZIPArchive*)archive;
- (void)extractFiles: (OFArray*)files
fromArchive: (OFZIPArchive*)archive;
@end
OF_APPLICATION_DELEGATE(OFZIP)
static void
help(OFStream *stream, bool full, int status)
{
[stream writeFormat:
@"Usage: %@ -[flnvx] archive.zip [file1 file2 ...]\n",
[OFApplication programName]];
if (full) {
[stream writeString:
@"\nOptions:\n"
@" -f Force / override files\n"
@" -h Show this help\n"
@" -l List all files in the archive\n"
@" -n Never override files\n"
@" -q Quiet mode (no output, except errors)\n"
@" -v Verbose output for file list\n"
@" -x Extract files\n"];
}
[OFApplication terminateWithStatus: status];
}
@implementation OFZIP
- (void)applicationDidFinishLaunching
{
OFOptionsParser *optionsParser =
[OFOptionsParser parserWithOptions: @"fhlnqvx"];
of_unichar_t option, mode = '\0';
OFArray *remainingArguments;
OFZIPArchive *archive;
OFArray *files;
while ((option = [optionsParser nextOption]) != '\0') {
switch (option) {
case 'f':
_override = 1;
break;
case 'n':
_override = -1;
break;
case 'v':
_outputLevel = 1;
break;
case 'q':
_outputLevel = -1;
break;
case 'l':
case 'x':
if (mode != '\0')
help(of_stdout, false, 1);
mode = option;
break;
case 'h':
help(of_stdout, true, 0);
break;
default:
[of_stderr writeFormat: @"%@: Unknown option: -%C\n",
[OFApplication programName],
[optionsParser lastOption]];
[OFApplication terminateWithStatus: 1];
}
}
remainingArguments = [optionsParser remainingArguments];
switch (mode) {
case 'l':
if ([remainingArguments count] != 1)
help(of_stderr, false, 1);
archive = [self openArchiveWithPath:
[remainingArguments firstObject]];
[self listFilesInArchive: archive];
break;
case 'x':
if ([remainingArguments count] < 1)
help(of_stderr, false, 1);
files = [remainingArguments objectsInRange:
of_range(1, [remainingArguments count] - 1)];
archive = [self openArchiveWithPath:
[remainingArguments firstObject]];
@try {
[self extractFiles: files
fromArchive: archive];
} @catch (OFCreateDirectoryFailedException *e) {
[of_stderr writeFormat:
@"\rFailed to create directory %@: %s\n",
[e path], strerror([e errNo])];
_exitStatus = 1;
} @catch (OFOpenFileFailedException *e) {
[of_stderr writeFormat:
@"\rFailed to open file %@: %s\n",
[e path], strerror([e errNo])];
_exitStatus = 1;
}
break;
default:
help(of_stderr, true, 1);
break;
}
[OFApplication terminateWithStatus: _exitStatus];
}
- (OFZIPArchive*)openArchiveWithPath: (OFString*)path
{
OFZIPArchive *archive = nil;
@try {
archive = [OFZIPArchive archiveWithPath: path];
} @catch (OFOpenFileFailedException *e) {
[of_stderr writeFormat: @"Failed to open file %@: %s\n",
[e path], strerror([e errNo])];
[OFApplication terminateWithStatus: 1];
} @catch (OFInvalidFormatException *e) {
[of_stderr writeFormat: @"File %@ is not a valid archive!\n",
path];
[OFApplication terminateWithStatus: 1];
}
return archive;
}
- (void)listFilesInArchive: (OFZIPArchive*)archive
{
OFEnumerator *enumerator = [[archive entries] objectEnumerator];
OFZIPArchiveEntry *entry;
while ((entry = [enumerator nextObject]) != nil) {
void *pool = objc_autoreleasePoolPush();
if (_outputLevel > 0) {
OFString *date = [[entry modificationDate]
localDateStringWithFormat: @"%Y-%m-%d %H:%M:%S"];
[of_stdout writeFormat:
@"%@: %" PRIu64 @" (%" PRIu64 @") bytes; %08X; %@; "
@"%@\n", [entry fileName], [entry uncompressedSize],
[entry compressedSize], [entry CRC32], date,
[entry fileComment]];
} else
[of_stdout writeLine: [entry fileName]];
objc_autoreleasePoolPop(pool);
}
}
- (void)extractFiles: (OFArray*)files
fromArchive: (OFZIPArchive*)archive
{
OFEnumerator *enumerator = [[archive entries] objectEnumerator];
OFZIPArchiveEntry *entry;
bool all = ([files count] == 0);
OFMutableSet *missing = [OFMutableSet setWithArray: files];
while ((entry = [enumerator nextObject]) != nil) {
void *pool = objc_autoreleasePoolPush();
OFString *fileName = [entry fileName];
OFString *outFileName = [fileName stringByStandardizingPath];
OFEnumerator *componentEnumerator;
OFString *component, *directory;
OFStream *stream;
OFFile *output;
char buffer[BUFFER_SIZE];
uint64_t written = 0, size = [entry uncompressedSize];
int_fast8_t percent = -1, newPercent;
if (!all && ![files containsObject: fileName])
continue;
[missing removeObject: fileName];
#ifndef _WIN32
if ([outFileName hasPrefix: @"/"]) {
#else
if ([outFileName hasPrefix: @"/"] ||
[outFileName containsString: @":"]) {
#endif
[of_stderr writeFormat: @"Refusing to extract %@!\n",
fileName];
_exitStatus = 1;
continue;
}
componentEnumerator =
[[outFileName pathComponents] objectEnumerator];
while ((component = [componentEnumerator nextObject]) != nil) {
if ([component isEqual: OF_PATH_PARENT_DIRECTORY]) {
[of_stderr writeFormat:
@"Refusing to extract %@!\n", fileName];
_exitStatus = 1;
continue;
}
}
if (_outputLevel > -1)
[of_stdout writeFormat: @"Extracting %@...", fileName];
if ([fileName hasSuffix: @"/"]) {
[OFFile createDirectoryAtPath: outFileName
createParents: true];
if (_outputLevel > -1)
[of_stdout writeLine: @" done"];
continue;
}
directory = [outFileName stringByDeletingLastPathComponent];
if (![OFFile directoryExistsAtPath: directory])
[OFFile createDirectoryAtPath: directory
createParents: true];
if ([OFFile fileExistsAtPath: outFileName] && _override != 1) {
OFString *line;
if (_override == -1) {
if (_outputLevel > -1)
[of_stdout writeLine: @" skipped"];
continue;
}
do {
[of_stderr writeFormat:
@"\rOverride %@? [ynAN?] ", fileName];
line = [of_stdin readLine];
if ([line isEqual: @"?"])
[of_stderr writeString: @" y: yes\n"
@" n: no\n"
@" A: always\n"
@" N: never\n"];
} while (![line isEqual: @"y"] &&
![line isEqual: @"n"] && ![line isEqual: @"N"] &&
![line isEqual: @"A"]);
if ([line isEqual: @"A"])
_override = 1;
else if ([line isEqual: @"N"])
_override = -1;
if ([line isEqual: @"n"] || [line isEqual: @"N"]) {
[of_stdout writeFormat: @"Skipping %@...\n",
fileName];
continue;
}
[of_stdout writeFormat: @"Extracting %@...", fileName];
}
stream = [archive streamForReadingFile: fileName];
output = [OFFile fileWithPath: outFileName
mode: @"wb"];
while (![stream isAtEndOfStream]) {
size_t length = [stream readIntoBuffer: buffer
length: BUFFER_SIZE];
[output writeBuffer: buffer
length: length];
written += length;
newPercent = (written == size
? 100 : (int_fast8_t)(written * 100 / size));
if (_outputLevel > -1 && percent != newPercent) {
percent = newPercent;
[of_stdout writeFormat:
@"\rExtracting %@... %3u%%",
fileName, percent];
}
}
if (_outputLevel > -1)
[of_stdout writeFormat: @"\rExtracting %@... done\n",
fileName];
objc_autoreleasePoolPop(pool);
}
if ([missing count] > 0) {
OFString *file;
enumerator = [missing objectEnumerator];
while ((file = [enumerator nextObject]) != nil)
[of_stderr writeFormat:
@"File %@ is not in the archive!\n", file];
_exitStatus = 1;
}
}
@end