提交

SweetParty/常用工具/获取设备UDID/XYKeyChain.m

@@ -0,0 +1,71 @@
+//
+//  XYKeyChain.m
+//  XYUUID
+//
+//  Created by steve on 2016/7/21.
+//  Copyright © 2020 guojunliu.github.io. All rights reserved.
+//
+
+#import "XYKeyChain.h"
+#import <Security/Security.h>
+
+@implementation XYKeyChain
+
++ (NSMutableDictionary *)getKeychainQueryMap:(NSString *)serviceDomain {
+    
+    NSMutableDictionary *queryMap = [[NSMutableDictionary alloc] init];
+    [queryMap setObject:serviceDomain forKey:(__bridge id<NSCopying>)kSecAttrAccount];
+    [queryMap setObject:serviceDomain forKey:(__bridge id<NSCopying>)kSecAttrService];
+    [queryMap setObject:(__bridge id)(kSecClassGenericPassword) forKey:(__bridge id<NSCopying>)kSecClass];
+    [queryMap setObject:(__bridge id)kSecAttrAccessibleAfterFirstUnlock forKey:(__bridge id<NSCopying>)kSecAttrAccessible];
+    return queryMap;
+}
+
++ (BOOL)setData:(id)data serviceDomain:(NSString *)serviceDomain {
+    NSMutableDictionary *queryMap = [self getKeychainQueryMap:serviceDomain];
+    SecItemDelete((__bridge CFDictionaryRef)(queryMap));
+    NSData *archivedData;
+    if (@available(iOS 13.0, macOS 13.0, *)) {
+        NSError *error;
+        archivedData = [NSKeyedArchiver archivedDataWithRootObject:data requiringSecureCoding:YES error:&error];
+        if (error) {
+            return NO;
+        }
+    }
+    else {
+        archivedData = [NSKeyedArchiver archivedDataWithRootObject:data];
+    }
+    [queryMap setObject:archivedData forKey:(__bridge id<NSCopying>)(kSecValueData)];
+    OSStatus status = SecItemAdd((__bridge CFDictionaryRef)(queryMap), NULL);
+    return status == errSecSuccess?YES:NO;
+}
+
++ (id)getDataWithServiceDomain:(NSString *)serviceDomain {
+    id data = nil;
+    NSMutableDictionary *queryMap = [self getKeychainQueryMap:serviceDomain];
+    [queryMap setObject:(id)kCFBooleanTrue forKey:(__bridge id<NSCopying>)(kSecReturnData)];
+    [queryMap setObject:(__bridge id)(kSecMatchLimitOne) forKey:(__bridge id<NSCopying>)(kSecMatchLimit)];
+
+    CFTypeRef result = NULL;
+    CFDictionaryRef cfDicRef = (__bridge_retained CFDictionaryRef)queryMap;
+    if (SecItemCopyMatching(cfDicRef, &result) == noErr) {
+        
+        if (@available(iOS 13.0, *)) {
+            NSError *error;
+            data = [NSKeyedUnarchiver unarchivedObjectOfClass:[NSString class] fromData:(__bridge NSData*)result error:&error];
+        }
+        else {
+            data = [NSKeyedUnarchiver unarchiveObjectWithData:(__bridge NSData*)result];
+        }
+        CFRelease(result);
+    }
+    return data;
+}
+
++ (BOOL)deleteDataWithServiceDomain:(NSString *)serviceDomain {
+    NSMutableDictionary *queryMap = [self getKeychainQueryMap:serviceDomain];
+    OSStatus status = SecItemDelete((__bridge CFDictionaryRef)(queryMap));
+    return status == errSecSuccess?YES:NO;
+}
+
+@end