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
130
131
132
133
134
135
136
137
138
139
140
|
#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 clean() {
NSArray *extrafiles = @[@"/var/lib", @"/var/cache"];
NSError *error = nil;
for (NSString *path in extrafiles) {
[[NSFileManager defaultManager] removeItemAtPath:path error:&error];
}
return 0;
}
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("Cleaning up /var\n");
clean();
printf("Renaming snapshot...\n");
rename(vol, snap);
printf("Restoring %s on %s has succeeded\n", snap, vol);
return 0;
}
|
