Clean /varv0.3

3 files changed, 13 insertions, 1 deletions

diff --git a/Makefile b/Makefile
index 8a06256..7819de7 100644
--- a/Makefile
+++ b/Makefile
@@ -9,7 +9,7 @@ DESTDIR         ?=
 
 DEB_MAINTAINER  ?= Cameron Katri <me@cameronkatri.com>
 DEB_ARCH        ?= iphoneos-arm
-SNAPRESTORE_V   := 0.2
+SNAPRESTORE_V   := 0.3
 DEB_SNAPRESTORE := $(SNAPRESTORE_V)
 
 all: build/snaprestore
diff --git a/README.md b/README.md
index f91175b..5a53b9e 100644
--- a/README.md
+++ b/README.md
@@ -5,4 +5,5 @@
 Fair amount of testing  
 Renames snapshot to original name  
 Removes jailbreak apps from icon cache  
+Clean up `/var`
 Works on any jailbreak (hopefully)
diff --git a/src/snaprestore.m b/src/snaprestore.m
index 23b20c8..aef5c5b 100644
--- a/src/snaprestore.m
+++ b/src/snaprestore.m
@@ -99,6 +99,15 @@ int rename(const char *vol, const char *snap) {
 	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]);
@@ -122,6 +131,8 @@ int main(int argc, char *argv[]) {
 			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);