Merge pull request #345 from PLatDT/uikit-fix

Remove UiKit import and use selectors to make sure we can compile for multiple platforms

Author: uroboro

TrustKit.podspec

@@ -1,6 +1,6 @@
 Pod::Spec.new do |s|
   s.name         = "TrustKit"
-  s.version      = "3.0.6"
+  s.version      = "3.0.7"
   s.summary      = 'TrustKit is an open source framework that makes it easy to deploy SSL pinning in any iOS, macOS, tvOS or watchOS App.'
   s.homepage     = "https://datatheorem.github.io/TrustKit"
   s.documentation_url = 'https://datatheorem.github.io/TrustKit/documentation/'

TrustKit/Pinning/TSKSPKIHashCache.m

@@ -8,8 +8,8 @@
  See AUTHORS file for the list of project authors.
  
  */
-
-#import <UIKit/UIKit.h>
+#import <objc/runtime.h>
+#import <objc/message.h>
 #import "TSKSPKIHashCache.h"
 #import "../TSKLog.h"
 #import <CommonCrypto/CommonDigest.h>
@@ -148,6 +148,21 @@ - (SPKICacheDictionnary *)loadSPKICacheFromFileSystem;
 
 @implementation TSKSPKIHashCache
 
+static BOOL isProtectedDataAvailable(void)
+{
+    Class uiApplicationClass = objc_getClass("UIApplication");
+    if (uiApplicationClass) {
+        SEL sharedApplicationSelector = sel_registerName("sharedApplication");
+        SEL isProtectedDataAvailableSelector = sel_registerName("isProtectedDataAvailable");
+        id sharedApp = ((id (*)(id, SEL))objc_msgSend)(uiApplicationClass, sharedApplicationSelector);
+        if (sharedApp && [sharedApp respondsToSelector:isProtectedDataAvailableSelector]) {
+            return (((BOOL (*)(id, SEL))objc_msgSend)(sharedApp, isProtectedDataAvailableSelector));
+        }
+    }
+    
+    return YES;
+}
+
 - (instancetype)initWithIdentifier:(NSString *)uniqueIdentifier
 {
     self = [super init];
@@ -249,7 +264,8 @@ - (NSData *)hashSubjectPublicKeyInfoFromCertificate:(SecCertificateRef)certifica
 
         __weak typeof(self) weakSelf = self;
         void (^updateCacheBlock)(void) = ^{
-            if ([[UIApplication sharedApplication] isProtectedDataAvailable]) {
+            
+            if (isProtectedDataAvailable()) {
                 NSData *serializedSpkiCache = [NSKeyedArchiver archivedDataWithRootObject:weakSelf.spkiCache requiringSecureCoding:YES error:nil];
                 if ([serializedSpkiCache writeToURL:[weakSelf SPKICachePath] atomically:YES] == NO) {
                     NSAssert(false, @"Failed to write cache");

TrustKitTests/TSKEndToEndSwizzlingTests.m

@@ -167,7 +167,7 @@ - (void)test
               // Valid pinning configuration
               @"www.datatheorem.com" : @{
                       kTSKEnforcePinning : @YES,
-                      kTSKPublicKeyHashes : @[@"a4FoHyEnsFhauIx0w/gB7ywslD6tWGk83J2F6Pv1phA=", // CA key
+                      kTSKPublicKeyHashes : @[@"F6jTih9VkkYZS8yuYqeU/4DUGehJ+niBGkkQ1yg8H3U=", // CA key
                                               @"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=" // Fake key
                                               ]},
               // Invalid pinning configuration