admin/SuperMega
1
0
Fork 0
mirror of https://github.com/dobin/SuperMega synced 2026-06-02 17:27:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests: fix some tests

Browse Source
This commit is contained in:
Dobin
2024-03-01 20:06:00 +00:00
parent aed5b168c2
commit adb97e2ad0
4 changed files with 28 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model/exehost.py
+3 -2
View File
@@ -7,6 +7,7 @@ from model.defs import *
import pe.pehelper as pehelper
from pe.superpe import SuperPe
from model.carrier import Carrier
from pe.superpe import PeSection
logger = logging.getLogger("ExeHost")
@@ -129,7 +130,7 @@ class ExeHost():
def get_relocations_for_section(self, section_name: str) -> List[PeRelocEntry]:
section = self.get_section_by_name(section_name)
section: PeSection = self.superpe.get_section_by_name(section_name)
if section is None:
return []
return [reloc for reloc in self.base_relocs if reloc.base_rva == section.virt_addr]
@@ -137,7 +138,7 @@ class ExeHost():
def get_reloc_largest_gap(self, section_name: str = ".rdata"):
tree = IntervalTree()
section = self.get_section_by_name(section_name)
section: PeSection = self.superpe.get_section_by_name(section_name)
print("-- Relocations: {}".format(len(self.base_relocs)))
print("-- section: 0x{:x}".format(section.virt_addr))