ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-10 17:21:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

better wine trace with winepath on .cur_input

Browse Source
This commit is contained in:
Andrea Fioraldi 2019-10-31 16:19:12 +01:00
parent 3ad5316dd1
commit 664f603a31
1 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
afl-wine-trace
View File

@ -4,9 +4,10 @@ import os
import sys import sys
import pefile import pefile
import shutil import shutil
import subprocess
if len(sys.argv) < 2: if len(sys.argv) < 2:
print("[afl-wine-trace] usage: wine-cov binary [args...]\n") print("[afl-wine-trace] usage: ./afl-wine-trace binary [args...]\n")
exit(1) exit(1)
if os.getenv("AFL_PATH"): if os.getenv("AFL_PATH"):
@ -42,14 +43,20 @@ else:
elif pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_I386"]: elif pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_I386"]:
qemu_path += "i386" qemu_path += "i386"
else: else:
print ("[wine-cov] unsuppoted architecture\n") print ("[afl-wine-trace] unsuppoted architecture\n")
exit(1) exit(1)
qemu_path = shutil.which(qemu_path) qemu_path = shutil.which(qemu_path)
if os.getenv("WINECOV_WINE_PATH"): wine_path = None
wine_path = os.getenv("WINECOV_WINE_PATH") if os.getenv("AFL_WINE_PATH"):
wine_path = os.getenv("AFL_WINE_PATH")
else: else:
wine_path = "/usr/lib/wine/wine" if not wine_path and shutil.which("wine"):
wine_path = shutil.which("wine")
if not wine_path and os.path.exists("/usr/bin/wine"):
wine_path = "/usr/bin/wine"
if not wine_path and os.path.exists("/usr/lib/wine/wine"):
wine_path = "/usr/lib/wine/wine"
if pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_AMD64"] or pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_IA64"]: if pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_AMD64"] or pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_IA64"]:
wine_path += "64" wine_path += "64"
elif pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_I386"]: elif pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_I386"]:
@ -58,4 +65,11 @@ else:
print ("[afl-wine-trace] unsopported architecture\n") print ("[afl-wine-trace] unsopported architecture\n")
exit(1) exit(1)
os.execve(qemu_path, [qemu_path, wine_path] + sys.argv[1:], os.environ) argv = sys.argv[1:]
for i in range(len(argv)):
if ".cur_input" in argv[i]:
argv[i] = subprocess.run([os.path.join(os.path.dirname(wine_path), "winepath"), "--windows", argv[i]], universal_newlines=True, stdout=subprocess.PIPE).stdout
break
print("[afl-wine-trace] exec:", " ".join([qemu_path, wine_path] + argv))
os.execve(qemu_path, [qemu_path, wine_path] + argv, os.environ)