Improve support of slithir conversion for Event with explicit base contract

pull/194/head
Josselin 6 years ago
parent ea24e614be
commit 8a94a6e66b
  1. 22
      slither/slithir/convert.py
  2. 2
      slither/visitors/slithir/expression_to_slithir.py

@ -82,7 +82,7 @@ def is_gas(ins):
return True
return False
def get_sig(ir):
def get_sig(ir, name):
'''
Return a list of potential signature
It is a list, as Constant variables can be converted to int256
@ -92,11 +92,14 @@ def get_sig(ir):
list(str)
'''
sig = '{}({})'
name = ir.function_name
# list of list of arguments
argss = convert_arguments(ir.arguments)
return [sig.format(name, ','.join(args)) for args in argss]
def convert_arguments(arguments):
argss = [[]]
for arg in ir.arguments:
for arg in arguments:
if isinstance(arg, (list,)):
type_arg = '{}[{}]'.format(get_type(arg[0].type), len(arg))
elif isinstance(arg, Function):
@ -119,7 +122,7 @@ def get_sig(ir):
else:
for args in argss:
args.append(type_arg)
return [sig.format(name, ','.join(args)) for args in argss]
return argss
def is_temporary(ins):
return isinstance(ins, (Argument,
@ -445,11 +448,16 @@ def extract_tmp_call(ins, contract):
call.call_id = ins.call_id
return call
if isinstance(ins.ori, Member):
# If there is a call on an inherited contract, it is an internal call
# If there is a call on an inherited contract, it is an internal call or an event
if ins.ori.variable_left in contract.inheritance + [contract]:
if str(ins.ori.variable_right) in [f.name for f in contract.functions]:
internalcall = InternalCall(ins.ori.variable_right, ins.ori.variable_left, ins.nbr_arguments, ins.lvalue, ins.type_call)
internalcall.call_id = ins.call_id
return internalcall
if str(ins.ori.variable_right) in [f.name for f in contract.events]:
eventcall = EventCall(ins.ori.variable_right)
eventcall.call_id = ins.call_id
return eventcall
if isinstance(ins.ori.variable_left, Contract):
st = ins.ori.variable_left.get_structure_from_name(ins.ori.variable_right)
if st:
@ -645,7 +653,7 @@ def get_type(t):
return str(t)
def convert_type_library_call(ir, lib_contract):
sigs = get_sig(ir)
sigs = get_sig(ir, ir.function_name)
func = None
for sig in sigs:
func = lib_contract.get_function_from_signature(sig)
@ -680,7 +688,7 @@ def convert_type_library_call(ir, lib_contract):
def convert_type_of_high_and_internal_level_call(ir, contract):
func = None
sigs = get_sig(ir)
sigs = get_sig(ir, ir.function_name)
for sig in sigs:
func = contract.get_function_from_signature(sig)
if not func:

@ -5,7 +5,7 @@ from slither.core.expressions import (AssignmentOperationType,
UnaryOperationType)
from slither.slithir.operations import (Assignment, Binary, BinaryType, Delete,
Index, InitArray, InternalCall, Member,
NewArray, NewContract, NewStructure,
NewArray, NewContract,
TypeConversion, Unary, Unpack, Return)
from slither.slithir.tmp_operations.argument import Argument
from slither.slithir.tmp_operations.tmp_call import TmpCall

Loading…
Cancel
Save