mirror of https://github.com/ConsenSys/mythril
blockchainethereumsmart-contractssoliditysecurityprogram-analysissecurity-analysissymbolic-execution
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
2.0 KiB
73 lines
2.0 KiB
#!/usr/bin/env python
|
|
"""mythril.py: Ethereum VM bytecode assembler/ disassembler
|
|
|
|
http://www.github.com/b-mueller/mythril
|
|
"""
|
|
|
|
from ether import asm
|
|
import sys
|
|
import codecs
|
|
import argparse
|
|
import util
|
|
|
|
|
|
def exitWithError(message):
|
|
print(message)
|
|
sys.exit()
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='Ethereum VM bytecode assembler/ disassembler')
|
|
|
|
parser.add_argument('-d', '--disassemble', action='store_true', help='disassemble, use with -c or -t')
|
|
parser.add_argument('-a', '--assemble', help='produce bytecode from easm input file', metavar='INPUT FILE')
|
|
parser.add_argument('-c', '--code', help='bytecode string ("6060604052...")', metavar='BYTECODE')
|
|
parser.add_argument('-t', '--transaction_hash', help='id of contract creation transaction')
|
|
parser.add_argument('-o', '--outfile')
|
|
parser.add_argument('--rpchost', default='127.0.0.1', help='RPC host')
|
|
parser.add_argument('--rpcport', type=int, default=[8545], help='RPC port')
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
if (args.disassemble):
|
|
|
|
if (args.code):
|
|
encoded_bytecode = args.code
|
|
elif (args.transaction_hash):
|
|
|
|
try:
|
|
|
|
encoded_bytecode = util.bytecode_from_blockchain(args.transaction_hash, args.rpchost, args.rpcport)
|
|
|
|
print(encoded_bytecode)
|
|
except Exception as e:
|
|
exitWithError("Exception loading bytecode via RPC" + str(e.message))
|
|
|
|
else:
|
|
exitWithError("Disassembler: Pass either the -c or -t flag to specify the input bytecode")
|
|
|
|
disassembly = asm.disassemble(util.safe_decode(encoded_bytecode))
|
|
|
|
easm_text = asm.disassembly_to_easm(disassembly)
|
|
|
|
if (args.outfile):
|
|
util.string_to_file(args.outfile, easm_text)
|
|
else:
|
|
sys.stdout.write(easm_text)
|
|
|
|
elif (args.assemble):
|
|
|
|
easm = util.file_to_string(args.assemble)
|
|
|
|
disassembly = asm.easm_to_disassembly(easm)
|
|
|
|
assembly = asm.assemble(disassembly)
|
|
|
|
if (args.outfile):
|
|
util.string_to_file(args.outfile, assembly)
|
|
else:
|
|
print("0x" + codecs.encode(assembly, "hex_codec"))
|
|
|
|
else:
|
|
|
|
parser.print_help()
|
|
|