Add def for NtQuerySystem + fix in CustomAllocator

This commit is contained in:
Clement Rouault
2015-09-16 16:42:21 +02:00
parent 9f70a58170
commit e02bf24e8c
3 changed files with 38 additions and 2 deletions
+1 -1
View File
@@ -91,7 +91,7 @@ class CustomAllocator(object):
def write_code(self, code):
size = len(code)
if size + self.cur_offset > self.cur_page_size:
self.get_new_page((payload_size + 0x1000) & ~0xfff)
self.get_new_page((size + 0x1000) & ~0xfff)
self.maps[-1][self.cur_offset: self.cur_offset + size] = code
addr = self.maps[-1].addr + self.cur_offset
self.cur_offset += size