objective-CでAES暗号化を行うその２

この前作ったのを拡張してみた。"objective-Cで継承ってどうやるんだろう？"という疑問とともに少し練習した次第です。

====

前のやつ

ファイル操作関係も入れてしまった方が楽じゃないの！ってことです。

cc_crypto.hを継承したcc_file_crypto.h

#import <Foundation/Foundation.h>
#import <CommonCrypto/CommonCrypto.h>
#import "cc_crypto.h"

@interface cc_file_crypto : cc_crypto
{

}

- (int)CCCryptoInPath:(NSString *)inPath OutPath:(NSString*)outPath;
@end

cc_file_crypto.m

#import "cc_file_crypto.h"

@implementation cc_file_crypto
- (int)CCCryptoInPath:(NSString *)inPath OutPath:(NSString*)outPath {
  NSFileHandle *fileHandle = [NSFileHandle fileHandleForReadingAtPath:inPath];

  if (!fileHandle) {
    NSLog(@"ファイルがありません．");
    return 1;
  }
  // ファイルの末尾まで読み込み、暗号化を行う
  NSData *data = [fileHandle readDataToEndOfFile];
  NSData* CryptedData = [super CCCrypto:data];

  /** 復号されたデータをファイルとして書き込む **/
  NSFileManager *fileManager = [NSFileManager defaultManager];
  // ファイルが存在しないか?
  if (![fileManager fileExistsAtPath:outPath]) { // yes
    // 空のファイルを作成する
    BOOL result = [fileManager createFileAtPath:outPath contents:[NSData data] attributes:nil];
    if (!result) {
        NSLog(@"ファイルの作成に失敗");
        return 1;
    }
}
  // ファイルハンドルを作成する
  NSFileHandle *writefileHandle = [NSFileHandle fileHandleForWritingAtPath:outPath];
  // ファイルハンドルの作成に失敗したか?
  if (!writefileHandle) { // no
    NSLog(@"ファイルハンドルの作成に失敗");
    return 1;
  }
  // ファイルに書き込む
  [writefileHandle writeData:CryptedData];

  return 0;
}
@end

こいつは暗号化するファイルのパスと暗号化後のファイルのパスを渡せば良きに計らってくれる感じです。

テスト用のmain

#import "cc_file_crypto.h"
#import <Foundation/Foundation.h>
#import <CommonCrypto/CommonCrypto.h>

/*
 * ファイルの暗号化のテスト
 *
 *
 */

int main(void) {
  unsigned char key[16] = {0x01, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff};
  unsigned char iv[16] = {0x01, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff};
  id cc, cc2;
  NSString *filePath = @"/Users/tokihiro/image/yuuyake03.jpg";
  NSString *encfilePath = @"/Users/tokihiro/image/E1";
  NSString *decfilePath = @"/Users/tokihiro/image/hoge.jpg";

  cc = [[cc_file_crypto alloc] initWithOperation:kCCEncrypt key:key iv:iv];
  cc2 = [[cc_file_crypto alloc] initWithOperation:kCCDecrypt key:key iv:iv];

  int result = [cc CCCryptoInPath:filePath OutPath:encfilePath];
  if(result == 1){
    NSLog(@"encrypt error");
    exit(EXIT_FAILURE);
  }

  result = [cc2 CCCryptoInPath:encfilePath OutPath:decfilePath];
  if(result == 1){
    NSLog(@"decrypt error");
    exit(EXIT_FAILURE);
  }
}

以下実行結果

bash-3.2$ cc cc_file_crypto.m cc_crypto.m main3.m -framework Foundation
cc cc_file_crypto.m cc_crypto.m main3.m -framework Foundation
bash-3.2$ ./a.out 
./a.out 
2013-11-19 22:29:14.915 a.out[62471:507] CCCryptorCreate = 0
2013-11-19 22:29:14.917 a.out[62471:507] CCCryptorUpdate = 0, resultLength = 246592
2013-11-19 22:29:14.917 a.out[62471:507] CCCryptorFinal = 0, resultLength = 16
success
2013-11-19 22:29:14.919 a.out[62471:507] CCCryptorCreate = 0
2013-11-19 22:29:14.920 a.out[62471:507] CCCryptorUpdate = 0, resultLength = 246592
2013-11-19 22:29:14.920 a.out[62471:507] CCCryptorFinal = 0, resultLength = 8
success

うまくできたー。

tokihiro000/CCCrypto · GitHub

RavyPop

=͟͟͞͞(๑=͟͟͞͞(๑•̀=͟͟͞͞(๑•̀д•́=͟͟͞͞(๑•̀д•́๑)=͟͟͞͞

objective-CでAES暗号化を行うその２