Safely inspect bundles

parse-community · Jan 3, 2017 · 4cd853c · 4cd853c
1 parent 7a820b7
commit 4cd853c
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/Parse/Internal/Object/Subclassing/PFObjectSubclassingController.m b/Parse/Internal/Object/Subclassing/PFObjectSubclassingController.m
@@ -128,20 +128,12 @@ - (void)scanForUnregisteredSubclasses:(BOOL)shouldSubscribe {
                                                                                             queue:nil
                                                                                        usingBlock:^(NSNotification *note) {
                                                                                            @strongify(self);
-                                                                                           [self _registerSubclassesInBundle:note.object];
+                                                                                           [self _registerSubclassesInLoadedBundle:note.object];
                                                                                        }];
     }
     NSArray *bundles = [[NSBundle allFrameworks] arrayByAddingObjectsFromArray:[NSBundle allBundles]];
     for (NSBundle *bundle in bundles) {
-        // Skip bundles that aren't loaded yet.
-        if (!bundle.loaded || !bundle.executablePath) {
-            continue;
-        }
-        // Filter out any system bundles
-        if ([bundle.bundlePath hasPrefix:@"/System/"] || [bundle.bundlePath hasPrefix:@"/Library/"]) {
-            continue;
-        }
-        [self _registerSubclassesInBundle:bundle];
+        [self _registerSubclassesInLoadedBundle:bundle];
     }
 }
 
@@ -338,6 +330,18 @@ - (void)_rawRegisterSubclass:(Class)kls {
     _registeredSubclasses[[kls parseClassName]] = subclassInfo;
 }
 
+- (void)_registerSubclassesInLoadedBundle:(NSBundle *)bundle {
+    // Skip bundles that aren't loaded yet.
+    if (!bundle.loaded || !bundle.executablePath) {
+        return;
+    }
+    // Filter out any system bundles
+    if ([bundle.bundlePath hasPrefix:@"/System/"] || [bundle.bundlePath hasPrefix:@"/Library/"] || [bundle.bundlePath rangeOfString:@"iPhoneSimulator.sdk"].location != NSNotFound) {
+        return;
+    }
+    [self _registerSubclassesInBundle:bundle];
+}
+
 - (void)_registerSubclassesInBundle:(NSBundle *)bundle {
     PFConsistencyAssert(bundle.loaded, @"Cannot register subclasses in a bundle that hasn't been loaded!");