diff options
Diffstat (limited to 'miasm2/analysis/sandbox.py')
| -rw-r--r-- | miasm2/analysis/sandbox.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/miasm2/analysis/sandbox.py b/miasm2/analysis/sandbox.py index eef83e85..5132406c 100644 --- a/miasm2/analysis/sandbox.py +++ b/miasm2/analysis/sandbox.py @@ -164,7 +164,7 @@ class OS_Win(OS): ALL_IMP_DLL = ["ntdll.dll", "kernel32.dll", "user32.dll", "ole32.dll", "urlmon.dll", "ws2_32.dll", 'advapi32.dll', "psapi.dll", - ] + ] modules_path = "win_dll" def __init__(self, custom_methods, *args, **kwargs): @@ -187,7 +187,9 @@ class OS_Win(OS): # Load main pe with open(self.fname) as fstream: self.pe = vm_load_pe(self.jitter.vm, fstream.read(), - load_hdr=self.options.load_hdr, **kwargs) + load_hdr=self.options.load_hdr, + name=self.fname, + **kwargs) self.name2module[fname_basename] = self.pe # Load library @@ -258,7 +260,8 @@ class OS_Linux(OS): self.libs = libimp_elf() with open(self.fname) as fstream: - self.elf = vm_load_elf(self.jitter.vm, fstream.read(), **kwargs) + self.elf = vm_load_elf(self.jitter.vm, fstream.read(), + name=self.fname, **kwargs) preload_elf(self.jitter.vm, self.elf, self.libs) self.entry_point = self.elf.Ehdr.entry @@ -284,7 +287,8 @@ class OS_Linux_str(OS): data = open(self.fname).read() self.options.load_base_addr = int(self.options.load_base_addr, 0) self.jitter.vm.add_memory_page( - self.options.load_base_addr, PAGE_READ | PAGE_WRITE, data) + self.options.load_base_addr, PAGE_READ | PAGE_WRITE, data, + "Initial Str") # Library calls handler self.jitter.add_lib_handler(libs, methods) |