-
Notifications
You must be signed in to change notification settings - Fork 21
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
3 changed files
with
131 additions
and
111 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,103 @@ | ||
| #import <YouTubeHeader/YTAppDelegate.h> | ||
| #import <YouTubeHeader/YTGlobalConfig.h> | ||
| #import <YouTubeHeader/YTColdConfig.h> | ||
| #import <YouTubeHeader/YTHotConfig.h> | ||
| #import <substrate.h> | ||
|
|
||
| NSMutableDictionary <NSString *, NSMutableDictionary <NSString *, NSNumber *> *> *cache; | ||
|
|
||
| extern void SearchHook(); | ||
|
|
||
| extern BOOL tweakEnabled(); | ||
| extern BOOL groupedSettings(); | ||
|
|
||
| extern void updateAllKeys(); | ||
| extern NSString *getKey(NSString *method, NSString *classKey); | ||
| extern BOOL getValue(NSString *methodKey); | ||
|
|
||
| static BOOL returnFunction(id const self, SEL _cmd) { | ||
| NSString *method = NSStringFromSelector(_cmd); | ||
| NSString *methodKey = getKey(method, NSStringFromClass([self class])); | ||
| return getValue(methodKey); | ||
| } | ||
|
|
||
| static BOOL getValueFromInvocation(id target, SEL selector) { | ||
| NSInvocationOperation *i = [[NSInvocationOperation alloc] initWithTarget:target selector:selector object:nil]; | ||
| [i start]; | ||
| BOOL result = NO; | ||
| [i.result getValue:&result]; | ||
| return result; | ||
| } | ||
|
|
||
| static NSMutableArray <NSString *> *getBooleanMethods(Class clz) { | ||
| NSMutableArray *allMethods = [NSMutableArray array]; | ||
| unsigned int methodCount = 0; | ||
| Method *methods = class_copyMethodList(clz, &methodCount); | ||
| for (unsigned int i = 0; i < methodCount; ++i) { | ||
| Method method = methods[i]; | ||
| const char *name = sel_getName(method_getName(method)); | ||
| if (strstr(name, "ndroid") || strstr(name, "musicClient") || strstr(name, "amsterdam") || strstr(name, "unplugged")) continue; | ||
| const char *encoding = method_getTypeEncoding(method); | ||
| if (strcmp(encoding, "B16@0:8")) continue; | ||
| NSString *selector = [NSString stringWithUTF8String:name]; | ||
| if (![allMethods containsObject:selector]) | ||
| [allMethods addObject:selector]; | ||
| } | ||
| free(methods); | ||
| return allMethods; | ||
| } | ||
|
|
||
| static void hookClass(NSObject *instance) { | ||
| if (!instance) [NSException raise:@"hookClass Invalid argument exception" format:@"Hooking the class of a non-existing instance"]; | ||
| Class instanceClass = [instance class]; | ||
| NSMutableArray <NSString *> *methods = getBooleanMethods(instanceClass); | ||
| NSString *classKey = NSStringFromClass(instanceClass); | ||
| NSMutableDictionary *classCache = cache[classKey] = [NSMutableDictionary new]; | ||
| for (NSString *method in methods) { | ||
| SEL selector = NSSelectorFromString(method); | ||
| BOOL result = getValueFromInvocation(instance, selector); | ||
| classCache[method] = @(result); | ||
| MSHookMessageEx(instanceClass, selector, (IMP)returnFunction, NULL); | ||
| } | ||
| } | ||
|
|
||
| %hook YTAppDelegate | ||
|
|
||
| - (BOOL)application:(id)arg1 didFinishLaunchingWithOptions:(id)arg2 { | ||
| if (tweakEnabled()) { | ||
| updateAllKeys(); | ||
| YTGlobalConfig *globalConfig; | ||
| YTColdConfig *coldConfig; | ||
| YTHotConfig *hotConfig; | ||
| @try { | ||
| globalConfig = [self valueForKey:@"_globalConfig"]; | ||
| coldConfig = [self valueForKey:@"_coldConfig"]; | ||
| hotConfig = [self valueForKey:@"_hotConfig"]; | ||
| } @catch (id ex) { | ||
| id settings = [self valueForKey:@"_settings"]; | ||
| globalConfig = [settings valueForKey:@"_globalConfig"]; | ||
| coldConfig = [settings valueForKey:@"_coldConfig"]; | ||
| hotConfig = [settings valueForKey:@"_hotConfig"]; | ||
| } | ||
| hookClass(globalConfig); | ||
| hookClass(coldConfig); | ||
| hookClass(hotConfig); | ||
| if (!groupedSettings()) { | ||
| SearchHook(); | ||
| } | ||
| } | ||
| return %orig; | ||
| } | ||
|
|
||
| %end | ||
|
|
||
| %ctor { | ||
| NSBundle *bundle = [NSBundle bundleWithPath:[NSString stringWithFormat:@"%@/Frameworks/Module_Framework.framework", [[NSBundle mainBundle] bundlePath]]]; | ||
| if (!bundle.loaded) [bundle load]; | ||
| cache = [NSMutableDictionary new]; | ||
| %init; | ||
| } | ||
|
|
||
| %dtor { | ||
| [cache removeAllObjects]; | ||
| } |