#!/usr/bin/env python3

import os

import sys

import pefile

import shutil

if len(sys.argv) < 2:

print("[afl-wine-trace] usage: wine-cov binary [args...]\n")

exit(1)

if os.getenv("AFL_PATH"):

my_dir = os.getenv("AFL_PATH")

else:

my_dir = os.path.dirname(os.path.abspath(__file__))

os.environ["WINELOADERNOEXEC"] = "1"

pe = pefile.PE(sys.argv[1])

if "AFL_ENTRYPOINT" not in os.environ:

os.environ["AFL_ENTRYPOINT"] = "0x%x" % (pe.OPTIONAL_HEADER.ImageBase pe.OPTIONAL_HEADER.AddressOfEntryPoint)

if not os.getenv("AFL_INST_LIBS"):

if "AFL_CODE_START" not in os.environ:

os.environ["AFL_CODE_START"] = "0x%x" % (pe.OPTIONAL_HEADER.ImageBase pe.OPTIONAL_HEADER.BaseOfCode)

if "AFL_CODE_END" not in os.environ:

os.environ["AFL_CODE_END"] = "0x%x" % (pe.OPTIONAL_HEADER.ImageBase pe.OPTIONAL_HEADER.BaseOfCode pe.OPTIONAL_HEADER.SizeOfCode)

if pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_AMD64"] or pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_IA64"]:

os.environ["LD_PRELOAD"] = os.path.join(my_dir, "qemu_mode/unsigaction/unsigaction64.so")

else:

os.environ["LD_PRELOAD"] = os.path.join(my_dir, "qemu_mode/unsigaction/unsigaction32.so")

if os.getenv("WINECOV_QEMU_PATH"):

qemu_path = os.getenv("WINECOV_QEMU_PATH")

elif os.path.exists(os.path.join(my_dir, "afl-qemu-trace")):

qemu_path = os.path.join(my_dir, "afl-qemu-trace")

else:

qemu_path = "qemu-"

if pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_AMD64"] or pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_IA64"]:

qemu_path = "x86_64"

elif pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_I386"]:

qemu_path = "i386"

else:

print ("[wine-cov] unsuppoted architecture\n")

exit(1)

qemu_path = shutil.which(qemu_path)

if os.getenv("WINECOV_WINE_PATH"):

wine_path = os.getenv("WINECOV_WINE_PATH")

else:

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"]:

wine_path = "64"

elif pe.FILE_HEADER.Machine == pefile.MACHINE_TYPE["IMAGE_FILE_MACHINE_I386"]:

pass

else:

print ("[afl-wine-trace] unsopported architecture\n")

exit(1)

os.execve(qemu_path, [qemu_path, wine_path] sys.argv[1:], os.environ)