]> git.saurik.com Git - uikittools.git/blob - uicache.mm
Separate unregisters/register, don't interleave.
[uikittools.git] / uicache.mm
1 /* UIKit Tools - command-line utilities for UIKit
2 * Copyright (C) 2008-2012 Jay Freeman (saurik)
3 */
4
5 /* Modified BSD License {{{ */
6 /*
7 * Redistribution and use in source and binary
8 * forms, with or without modification, are permitted
9 * provided that the following conditions are met:
10 *
11 * 1. Redistributions of source code must retain the
12 * above copyright notice, this list of conditions
13 * and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the
15 * above copyright notice, this list of conditions
16 * and the following disclaimer in the documentation
17 * and/or other materials provided with the
18 * distribution.
19 * 3. The name of the author may not be used to endorse
20 * or promote products derived from this software
21 * without specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
24 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
25 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
26 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
28 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
29 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
30 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
31 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
32 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
34 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
35 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
36 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37 */
38 /* }}} */
39
40 #import <Foundation/Foundation.h>
41
42 #include <notify.h>
43 #include <sys/types.h>
44 #include <sys/stat.h>
45 #include <unistd.h>
46
47 #include <objc/runtime.h>
48
49 #include <MobileCoreServices/LSApplicationWorkspace.h>
50
51 @interface NSMutableArray (Cydia)
52 - (void) addInfoDictionary:(NSDictionary *)info;
53 @end
54
55 @implementation NSMutableArray (Cydia)
56
57 - (void) addInfoDictionary:(NSDictionary *)info {
58 [self addObject:info];
59 }
60
61 @end
62
63 @interface NSMutableDictionary (Cydia)
64 - (void) addInfoDictionary:(NSDictionary *)info;
65 @end
66
67 @implementation NSMutableDictionary (Cydia)
68
69 - (void) addInfoDictionary:(NSDictionary *)info {
70 NSString *bundle = [info objectForKey:@"CFBundleIdentifier"];
71 [self setObject:info forKey:bundle];
72 }
73
74 @end
75
76 int main(int argc, const char *argv[]) {
77 NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
78
79 bool respring(false);
80
81 NSString *home(NSHomeDirectory());
82 NSString *path([NSString stringWithFormat:@"%@/Library/Caches/com.apple.mobile.installation.plist", home]);
83
84 Class $LSApplicationWorkspace(objc_getClass("LSApplicationWorkspace"));
85 LSApplicationWorkspace *workspace($LSApplicationWorkspace == nil ? nil : [$LSApplicationWorkspace defaultWorkspace]);
86
87 if (NSMutableDictionary *cache = [NSMutableDictionary dictionaryWithContentsOfFile:path]) {
88 NSFileManager *manager = [NSFileManager defaultManager];
89 NSError *error = nil;
90
91 NSMutableDictionary *bundles([NSMutableDictionary dictionaryWithCapacity:16]);
92
93 id system = [cache objectForKey:@"System"];
94 if (system == nil)
95 goto error;
96
97 [system removeAllObjects];
98
99 if (NSArray *apps = [manager contentsOfDirectoryAtPath:@"/Applications" error:&error]) {
100 for (NSString *app in apps)
101 if ([app hasSuffix:@".app"]) {
102 NSString *path = [@"/Applications" stringByAppendingPathComponent:app];
103 NSString *plist = [path stringByAppendingPathComponent:@"Info.plist"];
104
105 if (NSMutableDictionary *info = [NSMutableDictionary dictionaryWithContentsOfFile:plist]) {
106 if (NSString *identifier = [info objectForKey:@"CFBundleIdentifier"]) {
107 [bundles setObject:path forKey:identifier];
108 [info setObject:path forKey:@"Path"];
109 [info setObject:@"System" forKey:@"ApplicationType"];
110 [system addInfoDictionary:info];
111 } else
112 fprintf(stderr, "%s missing CFBundleIdentifier", [app UTF8String]);
113 }
114 }
115 } else goto error;
116
117 [cache writeToFile:path atomically:YES];
118
119 if (workspace != nil) {
120 for (NSString *identifier in bundles) {
121 NSString *path([bundles objectForKey:identifier]);
122 [workspace unregisterApplication:[NSURL fileURLWithPath:path]];
123 }
124
125 for (NSString *identifier in bundles)
126 if ([workspace respondsToSelector:@selector(invalidateIconCache:)])
127 [workspace invalidateIconCache:identifier];
128
129 for (NSString *identifier in bundles) {
130 NSString *path([bundles objectForKey:identifier]);
131 [workspace registerApplication:[NSURL fileURLWithPath:path]];
132 }
133 }
134
135 if (false) error:
136 fprintf(stderr, "%s\n", error == nil ? strerror(errno) : [[error localizedDescription] UTF8String]);
137 } else fprintf(stderr, "cannot open cache file. incorrect user?\n");
138
139 if (respring || kCFCoreFoundationVersionNumber >= 550.32) {
140 unlink([[NSString stringWithFormat:@"%@/Library/Caches/com.apple.springboard-imagecache-icons", home] UTF8String]);
141 unlink([[NSString stringWithFormat:@"%@/Library/Caches/com.apple.springboard-imagecache-icons.plist", home] UTF8String]);
142
143 unlink([[NSString stringWithFormat:@"%@/Library/Caches/com.apple.springboard-imagecache-smallicons", home] UTF8String]);
144 unlink([[NSString stringWithFormat:@"%@/Library/Caches/com.apple.springboard-imagecache-smallicons.plist", home] UTF8String]);
145
146 system([[NSString stringWithFormat:@"rm -rf %@/Library/Caches/SpringBoardIconCache", home] UTF8String]);
147 system([[NSString stringWithFormat:@"rm -rf %@/Library/Caches/SpringBoardIconCache-small", home] UTF8String]);
148
149 system([[NSString stringWithFormat:@"rm -rf %@/Library/Caches/com.apple.IconsCache", home] UTF8String]);
150 }
151
152 system("killall installd");
153
154 if (respring)
155 system("launchctl stop com.apple.SpringBoard");
156 else
157 notify_post("com.apple.mobile.application_installed");
158
159 [pool release];
160
161 return 0;
162 }