Skip to content

hook class method #33

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open

hook class method #33

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
6db81ff
hook class method
May 22, 2020
0111e1d
some fix
May 26, 2020
7ea716a
repeat hook
May 26, 2020
59adc62
appendAttributedString
Jun 15, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
84 changes: 80 additions & 4 deletions NSObjectSafe/NSObjectSafe.m
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,9 @@ + (void)load
swizzleInstanceMethod([NSObject class], @selector(removeObserver:forKeyPath:), @selector(hookRemoveObserver:forKeyPath:));
swizzleInstanceMethod([NSObject class], @selector(methodSignatureForSelector:), @selector(hookMethodSignatureForSelector:));
swizzleInstanceMethod([NSObject class], @selector(forwardInvocation:), @selector(hookForwardInvocation:));

swizzleClassMethod(object_getClass([NSObject class]), @selector(methodSignatureForSelector:), @selector(hookClassMethodSignatureForSelector:));
swizzleClassMethod(object_getClass([NSObject class]), @selector(forwardInvocation:), @selector(hookClassForwardInvocation:));
});
}
- (void) hookAddObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(NSKeyValueObservingOptions)options context:(void *)context
Expand Down Expand Up @@ -196,6 +199,29 @@ - (void)hookForwardInvocation:(NSInvocation*)invocation
[[[NSSafeProxy new] autorelease] dealException:info];
}

+ (NSMethodSignature*)hookClassMethodSignatureForSelector:(SEL)aSelector {

NSMethodSignature* sig = [self hookClassMethodSignatureForSelector:aSelector];
if (!sig){
if (class_getMethodImplementation(object_getClass([NSObject class]), @selector(methodSignatureForSelector:)) != class_getMethodImplementation(object_getClass(self.class), @selector(methodSignatureForSelector:))){
return nil;
}
return [NSMethodSignature signatureWithObjCTypes:"v@:@"];
}
return sig;
}

+ (void)hookClassForwardInvocation:(NSInvocation*)invocation {

NSString* info = [NSString stringWithFormat:@"unrecognized class selector [%@] sent to %@", NSStringFromSelector(invocation.selector), NSStringFromClass(self.class)];
[[NSNotificationCenter defaultCenter] postNotificationName:NSSafeNotification object:self userInfo:@{@"invocation":invocation}];
[[[NSSafeProxy new] autorelease] dealException:info];
}

- (void)setValue:(id)value forUndefinedKey:(NSString *)key{ SFAssert(NO, @"setValue:%@ forUndefinedKey:%@", value, key); }
- (id)valueForUndefinedKey:(NSString *)key { SFAssert(NO, @"valueForUndefinedKey: %@", key); return nil; }
- (void)setNilValueForKey:(NSString *)key { SFAssert(NO, @"setNilValueForKey: %@", key); }

@end

#pragma mark - NSString
Expand Down Expand Up @@ -538,9 +564,6 @@ + (void)load
swizzleInstanceMethod(NSClassFromString(@"NSConcreteMutableAttributedString"),
@selector(addAttributes:range:), @selector(hookAddAttributes:range:));

swizzleInstanceMethod(NSClassFromString(@"NSConcreteMutableAttributedString"),
@selector(addAttributes:range:), @selector(hookAddAttributes:range:));

swizzleInstanceMethod(NSClassFromString(@"NSConcreteMutableAttributedString"),
@selector(setAttributes:range:), @selector(hookSetAttributes:range:));

Expand All @@ -562,6 +585,9 @@ + (void)load
swizzleInstanceMethod(NSClassFromString(@"NSConcreteMutableAttributedString"),
@selector(enumerateAttributesInRange:options:usingBlock:), @selector(hookEnumerateAttributesInRange:options:usingBlock:));

swizzleInstanceMethod(NSClassFromString(@"NSConcreteMutableAttributedString"),
@selector(appendAttributedString:),
@selector(hookAppendAttributedString:));
});
}
- (id)hookInitWithString:(NSString*)str {
Expand Down Expand Up @@ -711,6 +737,16 @@ - (void)hookEnumerateAttributesInRange:(NSRange)range options:(NSAttributedStrin
}
}
}
- (void)hookAppendAttributedString:(NSAttributedString *)string {
@synchronized (self) {
if ([string isKindOfClass:[NSAttributedString class]] == NO ||
string.length == 0) {
SFAssert(NO, @"hookAppendAttributedString:string is nil or string is not AttributedString!");
return;
}
[self hookAppendAttributedString:string];
}
}
@end

#pragma mark - NSArray
Expand Down Expand Up @@ -752,6 +788,8 @@ + (void)load
swizzleInstanceMethod(NSClassFromString(@"__NSArrayReversed"), @selector(objectAtIndex:), @selector(hookObjectAtIndex:));
swizzleInstanceMethod(NSClassFromString(@"__NSArrayReversed"), @selector(subarrayWithRange:), @selector(hookSubarrayWithRange:));
swizzleInstanceMethod(NSClassFromString(@"__NSArrayReversed"), @selector(objectAtIndexedSubscript:), @selector(hookObjectAtIndexedSubscript:));

swizzleInstanceMethod(NSClassFromString(@"__NSPlaceholderArray"), @selector(initWithObjects:count:), @selector(hookInitWithObjects:count:));
});
}
+ (instancetype) hookArrayWithObject:(id)anObject
Expand Down Expand Up @@ -807,6 +845,21 @@ + (instancetype)hookArrayWithObjects:(const id [])objects count:(NSUInteger)cnt
return [self hookArrayWithObjects:objs count:index];
}
}
- (instancetype)hookInitWithObjects:(id _Nonnull const [])objects count:(NSUInteger)cnt
{
id safeObjects[cnt];
NSUInteger j = 0;
for (NSUInteger i = 0; i < cnt; i++) {
id obj = objects[i];
if (!obj) {
NSLog(@"error: anObject is nil");
continue;
}
safeObjects[j] = obj;
j++;
}
return [self hookInitWithObjects:safeObjects count:j];
}
@end

@implementation NSMutableArray(Safe)
Expand Down Expand Up @@ -930,7 +983,7 @@ - (NSArray *)hookSubarrayWithRange:(NSRange)range

@end

#pragma mark - NSDictionary
#pragma mark - NSData
@implementation NSData (Safe)
+ (void)load
{
Expand Down Expand Up @@ -1046,6 +1099,8 @@ + (void)load
/* 类方法 */
[NSDictionary swizzleClassMethod:@selector(dictionaryWithObject:forKey:) withMethod:@selector(hookDictionaryWithObject:forKey:)];
[NSDictionary swizzleClassMethod:@selector(dictionaryWithObjects:forKeys:count:) withMethod:@selector(hookDictionaryWithObjects:forKeys:count:)];

swizzleInstanceMethod(NSClassFromString(@"__NSPlaceholderDictionary"), @selector(initWithObjects:forKeys:count:), @selector(hookInitWithObjects:forKeys:count:));
});
}
+ (instancetype) hookDictionaryWithObject:(id)object forKey:(id)key
Expand All @@ -1070,6 +1125,23 @@ + (instancetype) hookDictionaryWithObjects:(const id [])objects forKeys:(const i
}
return [self hookDictionaryWithObjects:objs forKeys:ks count:index];
}
- (instancetype)hookInitWithObjects:(const id [])objects forKeys:(const id<NSCopying> [])keys count:(NSUInteger)cnt {
id safeObjects[cnt];
id safeKeys[cnt];
NSUInteger j = 0;
for (NSUInteger i = 0; i < cnt; i++) {
id key = keys[i];
id obj = objects[i];
if (!key || ! obj) {
NSLog(@"error: anObject is nil or anyKey is nil");
continue;
}
safeKeys[j] = key;
safeObjects[j] = obj;
j++;
}
return [self hookInitWithObjects:safeObjects forKeys:safeKeys count:j];
}
@end

@implementation NSMutableDictionary (Safe)
Expand Down Expand Up @@ -1297,6 +1369,10 @@ - (id) hookObjectForKey:(NSString *)defaultName
- (void) hookSetObject:(id)value forKey:(NSString*)aKey
{
if (aKey) {
if (value && !([value isKindOfClass:NSString.class] || [value isKindOfClass:NSData.class] || [value isKindOfClass:NSNumber.class] || [value isKindOfClass:NSDate.class] || [value isKindOfClass:NSArray.class] || [value isKindOfClass:NSDictionary.class])) {
SFAssert(NO, @"NSUserDefaults stores Property List objects (NSString, NSData, NSNumber, NSDate, NSArray, and NSDictionary) ----- NSUserDefaults invalid args hookSetObject:[%@] forKey:[%@]", value, aKey);
return;
}
[self hookSetObject:value forKey:aKey];
} else {
SFAssert(NO, @"NSUserDefaults invalid args hookSetObject:[%@] forKey:[%@]", value, aKey);
Expand Down