Git
/
mgeeky-Penetration-Testing-Tools
mirror of https://github.com/mgeeky/Penetration-Testing-Tools.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix

This commit is contained in:
Mariusz B. / mgeeky 2021-10-23 12:30:53 +02:00
parent dbe602a1ec
commit c61fab4dc1
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
windows/findSymbols.py
View File

@ -245,14 +245,17 @@ def processDir(args, regexes, path, results, uniqueSymbols, filesProcessed, symb
filePaths = []
for file in glob.glob(os.path.join(path, '**'), recursive=args.recurse):
if os.path.isfile(file):
looks_like_pe = False
with open(file, 'rb') as f:
mz = f.read(2)
if len(mz) == 2:
looks_like_pe = (mz[0] == ord('M') and mz[1] == ord('Z')) or (mz[1] == ord('M') and mz[0] == ord('Z'))
try:
if os.path.isfile(file):
looks_like_pe = False
with open(file, 'rb') as f:
mz = f.read(2)
if len(mz) == 2:
looks_like_pe = (mz[0] == ord('M') and mz[1] == ord('Z')) or (mz[1] == ord('M') and mz[0] == ord('Z'))
if looks_like_pe: filePaths.append(file)
if looks_like_pe: filePaths.append(file)
except:
continue
cpu_count = multiprocessing.cpu_count()