Mirrors/wavestone-cdt-edrsandblast
1
0
Fork 0
mirror of https://github.com/wavestone-cdt/EDRSandblast.git synced 2026-06-08 16:37:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

ExtractOffsets.py: detect invalid PDB

Browse Source
This commit is contained in:
Maxime Meignan
2023-11-03 15:06:34 +01:00
parent 43b159e2b1
commit b1321850c1
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Offsets/ExtractOffsets.py
+15 -1
View File
@@ -225,7 +225,21 @@ def get_pdb(pe: PE, pe_path, keep_ondisk=True, verbose=False):
print(f"[!] ERROR : Could not download PDB of {pe_path} (URL: {pdb_url}): {str(e)}.") print(f"[!] ERROR : Could not download PDB of {pe_path} (URL: {pdb_url}): {str(e)}.")
return None, None return None, None
elif os.path.isfile(pdb_file_path): elif os.path.isfile(pdb_file_path):
# todo: check the PDB and PE GUID are identical pdb = Pdb(path=pdb_file_path)
pe.parse_data_directories(directories=[DIRECTORY_ENTRY["IMAGE_DIRECTORY_ENTRY_DEBUG"]])
guid_string = (
f"{pe.DIRECTORY_ENTRY_DEBUG[0].entry.Signature_Data1:08x}-"
+ f"{pe.DIRECTORY_ENTRY_DEBUG[0].entry.Signature_Data2:04x}-"
+ f"{pe.DIRECTORY_ENTRY_DEBUG[0].entry.Signature_Data3:04x}-"
+ f"{pe.DIRECTORY_ENTRY_DEBUG[0].entry.Signature_Data4:02x}"
+ f"{pe.DIRECTORY_ENTRY_DEBUG[0].entry.Signature_Data5:02x}-"
+ pe.DIRECTORY_ENTRY_DEBUG[0].entry.Signature_Data6.hex().lower()
)
if str(pdb.PDBStream.Guid).lower() != guid_string:
print(f"[!] ERROR : PDB {pdb_file_path} does match {pe_path}. Redownloading")
del pdb
os.remove(pdb_file_path)
return get_pdb(pe, pe_path, keep_ondisk, verbose)
return pdb_file_path, None return pdb_file_path, None