Fix for the UBSan issue reported in #362.

erikdoe · Nov 4, 2018 · d00d8f3 · d00d8f3
1 parent 662bbeb
commit d00d8f3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Source/OCMock/OCPartialMockObject.m b/Source/OCMock/OCPartialMockObject.m
@@ -174,9 +174,9 @@ - (void)setupForwarderForSelector:(SEL)sel
         return;
 
     Method originalMethod = class_getInstanceMethod(mockedClass, sel);
-	IMP originalIMP = method_getImplementation(originalMethod);
-    const char *types = method_getTypeEncoding(originalMethod);
     /* Might be NULL if the selector is forwarded to another class */
+    IMP originalIMP = (originalMethod != NULL) ? method_getImplementation(originalMethod) : NULL;
+    const char *types = (originalMethod != NULL) ? method_getTypeEncoding(originalMethod) : NULL;
     // TODO: check the fallback implementation is actually sufficient
     if(types == NULL)
         types = ([[mockedClass instanceMethodSignatureForSelector:sel] fullObjCTypes]);