41 lines
1.2 KiB
Mathematica
41 lines
1.2 KiB
Mathematica
|
//
|
||
|
|
// LSNSURLSessionHook.m
|
||
|
|
// Nocilla
|
||
|
|
//
|
||
|
|
// Created by Luis Solano Bonet on 08/01/14.
|
||
|
|
// Copyright (c) 2014 Luis Solano Bonet. All rights reserved.
|
||
|
|
//
|
||
|
|
|
||
|
|
#import "LSNSURLSessionHook.h"
|
||
|
|
#import "LSHTTPStubURLProtocol.h"
|
||
|
|
#import <objc/runtime.h>
|
||
|
|
|
||
|
|
@implementation LSNSURLSessionHook
|
||
|
|
|
||
|
|
- (void)load {
|
||
|
|
Class cls = NSClassFromString(@"__NSCFURLSessionConfiguration") ?: NSClassFromString(@"NSURLSessionConfiguration");
|
||
|
|
[self swizzleSelector:@selector(protocolClasses) fromClass:cls toClass:[self class]];
|
||
|
|
}
|
||
|
|
|
||
|
|
- (void)unload {
|
||
|
|
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 {
|
||
|
|
|
||
|
|
Method originalMethod = class_getInstanceMethod(original, selector);
|
||
|
|
Method stubMethod = class_getInstanceMethod(stub, selector);
|
||
|
|
if (!originalMethod || !stubMethod) {
|
||
|
|
[NSException raise:NSInternalInconsistencyException format:@"Couldn't load NSURLSession hook."];
|
||
|
|
}
|
||
|
|
method_exchangeImplementations(originalMethod, stubMethod);
|
||
|
|
}
|
||
|
|
|
||
|
|
- (NSArray *)protocolClasses {
|
||
|
|
return @[[LSHTTPStubURLProtocol class]];
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
@end
|