cmdline keystone assembler

keystone_assembler.py

#!/usr/bin/env python3

from keystone import *
import sys
import argparse
import binascii


def main(filename=None, raw=False):
    if filename is None:
        print("No file specified.")
        exit()

    if filename != None:
        try:
            # Initialize engine in X86-32bit mode
            with open(filename, "r") as f:
                buf = f.read()
                ks = Ks(KS_ARCH_X86, KS_MODE_32)
                encoding, count = ks.asm(buf)

                if raw:
                    with open(sys.stdout.fileno(), "wb") as output:
                        output.write(bytes(encoding))
                else:
                    print(f"Assembling file: {filename}")
                    print(f"hex: {binascii.hexlify(bytes(encoding))}")

        except KsError as e:
            print("ERROR: %s" % e)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-f",
        "--filename",
        type=str,
        help="The path to the assembly file to be assembled.",
    )
    parser.add_argument(
        "-r",
        "--raw",
        action="store_true",
        help="Whether to output the raw encoded binary.",
    )
    args = parser.parse_args()
    main(**vars(args))