From 49133a657192a7a6768c7a7d381b26b1a06a0146 Mon Sep 17 00:00:00 2001 From: n0p <0x90@n0p.cc> Date: Wed, 2 Oct 2013 19:04:39 +0200 Subject: Raw data of the section is added/removed correctly now if there is raw data after the last section. --- SectionDoubleP.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/SectionDoubleP.py b/SectionDoubleP.py index 00b0ac1..f1a52f1 100644 --- a/SectionDoubleP.py +++ b/SectionDoubleP.py @@ -106,7 +106,9 @@ class SectionDoubleP: # Stripping the data of the section from the file. if self.pe.sections[-1].SizeOfRawData != 0: - self.pe.__data__ = self.pe.__data__[:-self.pe.sections[-1].SizeOfRawData] + self.pe.__data__ = (self.pe.__data__[:self.pe.sections[-1].PointerToRawData] + \ + self.pe.__data__[self.pe.sections[-1].PointerToRawData + \ + self.pe.sections[-1].SizeOfRawData:]) # Overwriting the section header in the binary with nulls. # Getting the address of the section table and manually overwriting @@ -214,7 +216,8 @@ class SectionDoubleP: # Appending the data of the new section to the file. if len(Data) > 0: - self.pe.__data__ = self.pe.__data__[:] + Data + self.pe.__data__ = (self.pe.__data__[:RawAddress] + Data + \ + self.pe.__data__[RawAddress:]) section_offset = section_table_offset + self.pe.FILE_HEADER.NumberOfSections*0x28 -- cgit v1.2.3-54-g00ecf