Added specific verification of class being UIViewController class

lightory · Mar 28, 2014 · a1f78d7 · a1f78d7
1 parent 6bf4645
commit a1f78d7
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/HHRouter/HHRouter.m b/HHRouter/HHRouter.m
@@ -144,9 +144,13 @@ - (NSDictionary*)paramsInRoute:(NSString*)route
             }
         }
     }
-
-    if (class_isMetaClass(object_getClass(subRoutes[@"_"]))) {
-        params[@"controller_class"] = subRoutes[@"_"];
+    Class class = subRoutes[@"_"];
+    if (class_isMetaClass(object_getClass(class))) {
+        if ([class isSubclassOfClass:[UIViewController class]]) {
+            params[@"controller_class"] = subRoutes[@"_"];
+        } else {
+            return nil;
+        }
     } else {
         if (subRoutes[@"_"]) {
             params[@"block"] = [subRoutes[@"_"] copy];