-For external classes, use the class name as a fallback if the symbol…

… can’t be found (for some strange reason) -OCD cleanup
boljonggo · Jan 8, 2014 · ef5465f · ef5465f
1 parent 5f1e0e3
commit ef5465f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/Source/CDLCSymbolTable.m b/Source/CDLCSymbolTable.m
@@ -116,9 +116,8 @@ - (void)loadSymbols;
         if (className) {
             if (symbol.value != 0)
                 classSymbols[className] = symbol;
-            else {
+            else
                 externalClassSymbols[className] = symbol;
-            }
         }
     };
 

diff --git a/Source/CDObjectiveC2Processor.m b/Source/CDObjectiveC2Processor.m
@@ -191,6 +191,8 @@ - (CDOCCategory *)loadCategoryAtAddress:(uint64_t)address;
             CDSymbol *classSymbol = [[self.machOFile symbolTable] symbolForExternalClassName:externalClassName];
             if (classSymbol)
                 category.classRef = [[CDOCClassReference alloc] initWithClassSymbol:classSymbol];
+            else
+                category.classRef = [[CDOCClassReference alloc] initWithClassName:externalClassName];
         }
     }
 
@@ -280,6 +282,8 @@ - (CDOCClass *)loadClassAtAddress:(uint64_t)address;
             CDSymbol *superClassSymbol = [[self.machOFile symbolTable] symbolForExternalClassName:superClassName];
             if (superClassSymbol)
                 aClass.superClassRef = [[CDOCClassReference alloc] initWithClassSymbol:superClassSymbol];
+            else
+                aClass.superClassRef = [[CDOCClassReference alloc] initWithClassName:superClassName];
         }
     }