Objective-CでZIPアーカイブを読み取る

//
//  ZipFile.h
//  ZippedImagesEtcetera
//
//  Created by Kenji Nishishiro  on 10/05/08.
//  Copyright 2010 Kenji Nishishiro. All rights reserved.
//
 
#import "unzip.h"
 
@interface ZipFile : NSObject {
	NSString *path_;
	unzFile unzipFile_;
}
 
- (id)initWithFileAtPath:(NSString *)path;
- (BOOL)open;
- (void)close;
- (NSData *)readWithFileName:(NSString *)fileName maxLength:(NSUInteger)maxLength;
- (NSArray *)fileNames;
@end
 
//
//  ZipFile.m
//  ZippedImagesEtcetera
//
//  Created by Kenji Nishishiro  on 10/05/08.
//  Copyright 2010 Kenji Nishishiro. All rights reserved.
//
 
#import "ZipFile.h"
 
@implementation ZipFile
 
static const int CASE_SENSITIVITY = 0;
static const unsigned int BUFFER_SIZE = 8192;
 
- (id)initWithFileAtPath:(NSString *)path {
	NSAssert(path, @"path");
 
	if (self = [super init]) {
		path_ = [path retain];
		unzipFile_ = NULL;
	}
	return self;
}
 
- (void)dealloc {
	NSAssert(!unzipFile_, @"!unzipFile_");
 
	[path_ release];
	[super dealloc];
}
 
- (BOOL)open {
	NSAssert(!unzipFile_, @"!unzipFile_");
 
	unzipFile_ = unzOpen64([path_ UTF8String]);
	return unzipFile_ != NULL;
}
 
- (void)close {
	NSAssert(unzipFile_, @"unzipFile_");
 
	unzClose(unzipFile_);
	unzipFile_ = NULL;
}
 
- (NSData *)readWithFileName:(NSString *)fileName maxLength:(NSUInteger)maxLength {
	NSAssert(unzipFile_, @"unzipFile_");
	NSAssert(fileName, @"fileName");
 
	if (unzLocateFile(unzipFile_, [fileName UTF8String], CASE_SENSITIVITY) != UNZ_OK) {
		return nil;
	}
 
	if (unzOpenCurrentFile(unzipFile_) != UNZ_OK) {
		return nil;
	}
 
	NSMutableData *data = [NSMutableData data];
	NSUInteger length = 0;
	void *buffer = (void *)malloc(BUFFER_SIZE);
	while (YES) {
		unsigned size = length + BUFFER_SIZE <= maxLength ? BUFFER_SIZE : maxLength - length;
		int readLength = unzReadCurrentFile(unzipFile_, buffer, size);
		if (readLength  0) {
			[data appendBytes:buffer length:readLength];
			length += readLength;
		}
		if (readLength == 0) {
			break;
		}
	};	
	free(buffer);
 
	unzCloseCurrentFile(unzipFile_); 
 
	return data;
}
 
- (NSArray *)fileNames {
	NSAssert(unzipFile_, @"unzipFile_");
 
	NSMutableArray *results = [NSMutableArray array];
	if (unzGoToFirstFile(unzipFile_) != UNZ_OK) {
		return nil;
	}
	while (YES) {
		unz_file_info64 fileInfo;
		char fileName[PATH_MAX];
		if (unzGetCurrentFileInfo64(unzipFile_, &fileInfo, fileName, PATH_MAX, NULL, 0, NULL, 0) != UNZ_OK) {
			return nil;
		}
		[results addObject:[NSString stringWithUTF8String:fileName]];
 
		int error = unzGoToNextFile(unzipFile_);
		if (error == UNZ_END_OF_LIST_OF_FILE) {
			break;
		}
		if (error != UNZ_OK) {
			return nil;
		}
	}
	return results;
}
 
@end

ZipFile *zipFile = [[[ZipFile alloc] initWithFileAtPath:@"ZIPアーカイブ名"] autorelease];
 
// ファイルを開く。
if (![zipFile open]) {
	// エラー処理
}
 
// ファイルリストを取り出す。パーシャルバスで全ファイルが取得されるので、階層的に扱う必要は無い。
NSArray *fileNames = [zipFile fileNames];
if (!fileNames) {
	// エラー処理
}
 
// ファイルの内容を取り出す。一度にメモリに読み込むので、ストリーム処理が必要な大きいファイルには向かない。安全のために最大サイズを指定できる。
NSData *data = [zipFile readWithFileName:@"ファイル名" maxLength:読み込む最大サイズ];
if (!data) {
	// エラー処理
}
 
// ファイルを閉じる。
[zipFile close];