-
Notifications
You must be signed in to change notification settings - Fork 5
/
FWEncryptorAES.m
111 lines (87 loc) · 3.8 KB
/
FWEncryptorAES.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
//
// FWEncryptorAES.m
// FWEncryptorAES
//
// Created by FrankWu on 2013/12/20.
// Copyright (c) 2013年 FrankWu. All rights reserved.
//
#import "FWEncryptorAES.h"
#import "NSData+CommonCrypto.h"
#import "NSData+Base64.h"
@implementation FWEncryptorAES
+ (NSData *)encrypt:(NSData *)data Key:(id)key IV:(id)iv
{
NSParameterAssert([key isKindOfClass: [NSData class]] || [key isKindOfClass: [NSString class]]);
NSParameterAssert([iv isKindOfClass: [NSData class]] || [iv isKindOfClass: [NSString class]]);
NSData *keyData = [self getKey:key];
NSData *ivData = [self getIV:iv];
CCCryptorStatus status = kCCSuccess;
NSData *encrypted = [data dataEncryptedUsingAlgorithm:kCCAlgorithmAES128 key:keyData initializationVector:ivData options:kCCOptionPKCS7Padding error:&status];
return encrypted;
}
+ (NSData *)decrypt:(NSData *)data Key:(id)key IV:(id)iv
{
NSParameterAssert([key isKindOfClass: [NSData class]] || [key isKindOfClass: [NSString class]]);
NSParameterAssert([iv isKindOfClass: [NSData class]] || [iv isKindOfClass: [NSString class]]);
NSData *keyData = [self getKey:key];
NSData *ivData = [self getIV:iv];
CCCryptorStatus status = kCCSuccess;
NSData *decrypted = [data decryptedDataUsingAlgorithm:kCCAlgorithmAES128 key:keyData initializationVector:ivData options:kCCOptionPKCS7Padding error:&status];
return decrypted;
}
+ (NSString *)encryptToBase64:(NSData *)data Key:(id)key IV:(id)iv
{
NSParameterAssert([key isKindOfClass: [NSData class]] || [key isKindOfClass: [NSString class]]);
NSParameterAssert([iv isKindOfClass: [NSData class]] || [iv isKindOfClass: [NSString class]]);
NSData *encrypted = [self encrypt:data Key:key IV:iv];
return [encrypted base64EncodedString];
}
+ (NSData *)decryptFromBase64:(NSString *)str Key:(id)key IV:(id)iv
{
NSParameterAssert([key isKindOfClass: [NSData class]] || [key isKindOfClass: [NSString class]]);
NSParameterAssert([iv isKindOfClass: [NSData class]] || [iv isKindOfClass: [NSString class]]);
NSData *data = [NSData dataFromBase64String:str];
return [self decrypt:data Key:key IV:iv];
}
+ (NSString *)encryptStrToBase64:(NSString *)str Key:(id)key IV:(id)iv
{
NSParameterAssert([key isKindOfClass: [NSData class]] || [key isKindOfClass: [NSString class]]);
NSParameterAssert([iv isKindOfClass: [NSData class]] || [iv isKindOfClass: [NSString class]]);
NSData *data = [str dataUsingEncoding:NSUTF8StringEncoding];
return [self encryptToBase64:data Key:key IV:iv];
}
+ (NSString *)decryptStrFromBase64:(NSString *)str Key:(id)key IV:(id)iv
{
NSParameterAssert([key isKindOfClass: [NSData class]] || [key isKindOfClass: [NSString class]]);
NSParameterAssert([iv isKindOfClass: [NSData class]] || [iv isKindOfClass: [NSString class]]);
NSData *data = [NSData dataFromBase64String:str];
NSData *decrypted = [self decrypt:data Key:key IV:iv];
return [[NSString alloc] initWithData:decrypted encoding:NSUTF8StringEncoding];
}
+ (NSData *)getKey:(id)key
{
NSParameterAssert([key isKindOfClass: [NSData class]] || [key isKindOfClass: [NSString class]]);
if ([key isKindOfClass: [NSString class]])
return [[key dataUsingEncoding:NSUTF8StringEncoding] SHA256Hash];
else
return (NSData *)key;
}
+ (NSData *)getIV:(id)iv
{
NSParameterAssert([iv isKindOfClass: [NSData class]] || [iv isKindOfClass: [NSString class]]);
if ([iv isKindOfClass: [NSString class]])
return [[iv dataUsingEncoding:NSUTF8StringEncoding] MD5Sum];
else
return (NSData *)iv;
}
+ (NSString*)convertHexStringFromData:(NSData*)data
{
const unsigned char *result = [data bytes];
NSMutableString *str = [NSMutableString stringWithCapacity:[data length]];
for(int i = 0; i<[data length]; i++)
{
[str appendFormat:@"%02x",result[i]];
}
return str;
}
@end