Fixed NullPointerException when exports dir was missing

lowleveldesign · Dec 9, 2023 · b23c8a2 · b23c8a2
1 parent 971cb02
commit b23c8a2
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/withdll/DllInjection.cs b/withdll/DllInjection.cs
@@ -61,17 +61,19 @@ private static bool ExportsOrdinal1(string modulePath)
             using var pereader = new PEReader(File.OpenRead(modulePath));
 
             var exportsDirEntry = pereader.PEHeaders.PEHeader!.ExportTableDirectory;
-            unsafe
+            if (exportsDirEntry.RelativeVirtualAddress != 0)
             {
-                var exportsDir = (IMAGE_EXPORT_DIRECTORY*)pereader.GetSectionData(exportsDirEntry.RelativeVirtualAddress).Pointer;
-
-                if (exportsDir->Base == 1)
+                unsafe
                 {
-                    return ((uint*)pereader.GetSectionData((int)exportsDir->AddressOfFunctions).Pointer)[0] != 0;
-                }
+                    var exportsDir = (IMAGE_EXPORT_DIRECTORY*)pereader.GetSectionData(exportsDirEntry.RelativeVirtualAddress).Pointer;
 
-                return false;
+                    if (exportsDir->Base == 1)
+                    {
+                        return ((uint*)pereader.GetSectionData((int)exportsDir->AddressOfFunctions).Pointer)[0] != 0;
+                    }
+                }
             }
+            return false;
         }
 
         public static void StartProcessWithDlls(List<string> cmdlineArgs, bool debug, List<string> dllPaths)