1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#include "substitute.h"
#include <objc/runtime.h>
#include <notify.h>
#include <dispatch/dispatch.h>
#import <Foundation/Foundation.h>
#import <UIKit/UIKit.h>
/* TODO: test what happens when both are required */
static bool g_did_say_in_setup_mode;
static bool g_did_exit_safety_dance;
@interface _SBApplicationController
- (id)applicationWithBundleIdentifier:(NSString *)identifier;
@end
@interface _SBSetupManager
- (bool)updateInSetupMode;
- (bool)_setInSetupMode:(bool)inSetupMode;
- (id)applicationWithBundleIdentifier:(NSString *)identifier;
@end
@interface _SpringBoard
- (void)relaunchSpringBoard;
@end
static id (*old_setupApplication)(id, SEL);
static id my_setupApplication(id self, SEL sel) {
if (g_did_say_in_setup_mode)
return [self applicationWithBundleIdentifier:@"com.ex.SafetyDance"];
return old_setupApplication(self, sel);
}
static bool (*old_updateInSetupMode)(id, SEL);
static bool my_updateInSetupMode(id self, SEL sel) {
if (g_did_exit_safety_dance) {
g_did_say_in_setup_mode = false;
return old_updateInSetupMode(self, sel);
} else {
/* take priority over real setup */
g_did_say_in_setup_mode = true;
[self _setInSetupMode:true];
return true;
}
}
static bool (*old__handleSetupExited)(id, SEL, id);
static bool my__handleSetupExited(id self, SEL sel, id app) {
if (g_did_say_in_setup_mode)
g_did_exit_safety_dance = true;
return old__handleSetupExited(self, sel, app);
}
__attribute__((constructor))
static void init() {
#define GET(clsname) \
Class clsname = objc_getClass(#clsname); \
if (!clsname) { \
NSLog(@"substitute safe mode failed to find %s", #clsname); \
return; \
}
GET(SBApplicationController);
GET(SBSetupManager);
GET(SBWorkspace);
int notify_token;
uint32_t notify_status = notify_register_dispatch(
"com.ex.substitute.safemode-restart-springboard-plz",
¬ify_token, dispatch_get_main_queue(), ^(int tok) {
id sb = [UIApplication sharedApplication];
[sb relaunchSpringBoard];
}
);
#define HOOK(cls, sel, selvar) do { \
int ret = substitute_hook_objc_message(cls, @selector(sel), \
my_##selvar, \
&old_##selvar, NULL); \
if (ret) { \
NSLog(@"substitute safe mode '%s' hook failed: %d", #sel, ret); \
return; \
} \
} while(0)
/* note: any of these might fail, leaving the previous ones hooked */
HOOK(SBApplicationController, setupApplication, setupApplication);
HOOK(SBWorkspace, _handleSetupExited:, _handleSetupExited);
HOOK(SBSetupManager, updateInSetupMode, updateInSetupMode);
}
|
