Handle full name for symbol in PatcherGetSymbolAddress.

CloverHackyColor · Jun 19, 2022 · 8aa646b · 8aa646b
1 parent 58da8df
commit 8aa646b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Library/OcAppleKernelLib/KextPatcher.c b/Library/OcAppleKernelLib/KextPatcher.c
@@ -222,6 +222,9 @@ PatcherGetSymbolAddress (
   UINT64          SymbolAddress;
   UINT32          Offset;
   UINT32          Index;
+  BOOLEAN        FullName = FALSE;
+
+  FullName = (Name[0] == '_' && Name[1] == '_');
 
   Index  = 0;
   Offset = 0;
@@ -256,7 +259,10 @@ PatcherGetSymbolAddress (
     }
 
     SymbolName = MachoGetSymbolName (&Context->MachContext, Symbol);
-    if ((SymbolName != NULL) && (AsciiStrCmp (Name, SymbolName) == 0)) {
+//    if ((SymbolName != NULL) && (AsciiStrCmp (Name, SymbolName) == 0)) {
+    if (SymbolName != NULL &&
+        (( FullName && AsciiStrCmp (Name, SymbolName) == 0) ||
+         (!FullName && AsciiStrStr (SymbolName, Name) != NULL))) {
       //
       // Once we have a symbol, get its ondisk offset.
       //