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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#import <Foundation/Foundation.h>
#import <Foundation/NSFileManager.h>
#import <sys/snapshot.h>
#import <getopt.h>
typedef char io_string_t[512];
typedef mach_port_t io_object_t;
typedef io_object_t io_registry_entry_t;
io_registry_entry_t IORegistryEntryFromPath(mach_port_t master, const io_string_t path);
CFTypeRef IORegistryEntryCreateCFProperty(io_registry_entry_t entry, CFStringRef key, CFAllocatorRef allocator, uint32_t options);
kern_return_t IOObjectRelease(io_object_t object);
@interface LSApplicationWorkspace : NSObject
+ (id)defaultWorkspace;
- (BOOL)_LSPrivateRebuildApplicationDatabasesForSystemApps:(BOOL)arg1 internal:(BOOL)arg2 user:(BOOL)arg3;
- (BOOL)registerApplicationDictionary:(NSDictionary *)applicationDictionary;
- (BOOL)registerBundleWithInfo:(NSDictionary *)bundleInfo options:(NSDictionary *)options type:(unsigned long long)arg3 progress:(id)arg4 ;
- (BOOL)registerApplication:(NSURL *)url;
- (BOOL)registerPlugin:(NSURL *)url;
- (BOOL)unregisterApplication:(NSURL *)url;
- (NSArray *)installedPlugins;
-(void)_LSPrivateSyncWithMobileInstallation;
@end
void usage(char *name) {
printf(
"Usage: %s [volume] [snapshot]\n", name);
}
NSString *bootsnapshot() {
const io_registry_entry_t chosen = IORegistryEntryFromPath(0, "IODeviceTree:/chosen");
const NSData *data = (__bridge const NSData *)IORegistryEntryCreateCFProperty(chosen, (__bridge CFStringRef)@"boot-manifest-hash", kCFAllocatorDefault, 0);
IOObjectRelease(chosen);
NSMutableString *manifestHash = [NSMutableString stringWithString:@""];
NSUInteger len = [data length];
Byte *buf = (Byte*)malloc(len);
memcpy(buf, [data bytes], len);
int buf2;
for (buf2 = 0; buf2 <= 19; buf2++) {
[manifestHash appendFormat:@"%02X", buf[buf2]];
}
// add com.apple.os.update-
return [NSString stringWithFormat:@"%@%@", @"com.apple.os.update-", manifestHash];
}
int restore(const char *vol, const char *snap) {
int fd = open(vol, O_RDONLY, 0);
int ret = fs_snapshot_revert(fd, snap, 0);
return ret;
}
int mount(const char *vol, const char *snap, const char *mnt) {
int fd = open(vol, O_RDONLY, 0);
BOOL isDir;
NSFileManager *fileManager = [NSFileManager defaultManager];
if(![fileManager fileExistsAtPath:[NSString stringWithUTF8String:mnt] isDirectory:&isDir])
if(![fileManager createDirectoryAtPath:[NSString stringWithUTF8String:mnt] withIntermediateDirectories:YES attributes:nil error:NULL])
NSLog(@"Error: Create folder failed %s", mnt);
int ret = fs_snapshot_mount(fd, mnt, snap, 0);
return ret;
}
NSMutableSet *findApps(const char *root, const char *mnt) {
NSMutableString *rootApplications = [NSMutableString stringWithUTF8String:root];
rootApplications = [[rootApplications stringByAppendingString:@"/Applications"] mutableCopy];
NSMutableString *mntApplications = [NSMutableString stringWithUTF8String:mnt];
mntApplications = [[mntApplications stringByAppendingString:@"/Applications"] mutableCopy];
NSArray *rootApps = [[NSFileManager defaultManager] contentsOfDirectoryAtPath:rootApplications error:nil];
NSArray *mntApps = [[NSFileManager defaultManager] contentsOfDirectoryAtPath:mntApplications error:nil];
NSMutableSet *ret = [[NSMutableSet alloc] init];
for (NSString *app in rootApps) {
if (![mntApps containsObject:app]) {
[ret addObject:[@"/Applications/" stringByAppendingString:app]];
}
}
return ret;
}
int unregisterPath(NSString *path) {
path = [path stringByResolvingSymlinksInPath];
NSURL *url = [NSURL fileURLWithPath:path];
LSApplicationWorkspace *workspace = [LSApplicationWorkspace defaultWorkspace];
return [workspace unregisterApplication:url];
}
int rename(const char *vol, const char *snap) {
int fd = open(vol, O_RDONLY, 0);
int ret = fs_snapshot_rename(fd, snap, [bootsnapshot() UTF8String], 0);
return ret;
}
int main(int argc, char *argv[]) {
if (argc != 3) {
usage(argv[0]);
return 0;
}
char *vol = argv[1];
char *snap = argv[2];
char *mnt = "/tmp/rootfsmnt";
printf("Restoring snapshot %s...\n", snap);
restore(vol, snap);
printf("Restored snapshot...\n");
printf("Mounting rootfs...\n");
mount(vol, snap, mnt);
printf("Mounted %s at %s\n", snap, mnt);
NSMutableSet *appSet = findApps(vol, mnt);
if ([appSet count]) {
for (NSString *app in appSet) {
printf("unregistering %s\n", [app UTF8String]);
unregisterPath(app);
}
}
printf("Renaming snapshot...\n");
rename(vol, snap);
printf("Restoring %s on %s has succeeded\n", snap, vol);
return 0;
}
|