Merge pull request #76 from augustj/iOS8-Fix

Update LSNSURLSessionHook.m
luisobo · Sep 23, 2014 · 761efce · 761efce
2 parents 16c7bd5 + 57c94e7
commit 761efce
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Nocilla/Hooks/NSURLSession/LSNSURLSessionHook.m b/Nocilla/Hooks/NSURLSession/LSNSURLSessionHook.m
@@ -13,11 +13,13 @@
 @implementation LSNSURLSessionHook
 
 - (void)load {
-    [self swizzleSelector:@selector(protocolClasses) fromClass:NSClassFromString(@"__NSCFURLSessionConfiguration") ?: [NSURLSessionConfiguration class] toClass:[self class]];
+    Class cls = NSClassFromString(@"__NSCFURLSessionConfiguration") ?: NSClassFromString(@"NSURLSessionConfiguration");
+    [self swizzleSelector:@selector(protocolClasses) fromClass:cls toClass:[self class]];
 }
 
 - (void)unload {
-    [self swizzleSelector:@selector(protocolClasses) fromClass:NSClassFromString(@"__NSCFURLSessionConfiguration") ?: [NSURLSessionConfiguration class] toClass:[self class]];
+    Class cls = NSClassFromString(@"__NSCFURLSessionConfiguration") ?: NSClassFromString(@"NSURLSessionConfiguration");
+    [self swizzleSelector:@selector(protocolClasses) fromClass:cls toClass:[self class]];
 }
 
 - (void)swizzleSelector:(SEL)selector fromClass:(Class)original toClass:(Class)stub {