fix linux reading PE mono root

mtgatool · Apr 25, 2024 · bbc9c03 · bbc9c03
1 parent fc28a5e
commit bbc9c03
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/mono_reader.rs b/src/mono_reader.rs
@@ -114,7 +114,8 @@ impl MonoReader {
                         found = true
                     }
                     _ => {
-                        eprintln!("Error: mono_get_root_domain not found");
+                        // This is not the library we are looking for
+                        // eprintln!("Error: mono_get_root_domain not found");
                     }
                 }
             }
@@ -228,7 +229,7 @@ impl MonoReader {
         let val = unsafe {
             match member.read() {
                 Ok(val) => Some(val),
-                Err(_e) => None
+                Err(_e) => None,
             }
         };
 

diff --git a/src/pe_reader.rs b/src/pe_reader.rs
@@ -81,6 +81,10 @@ impl<'a> PEReader<'a> {
             }
         }
 
+        if root_domain_function_address == 0 {
+            return Err(Error::default());
+        }
+
         Ok(root_domain_function_address)
     }
 }