mirror of https://github.com/crytic/slither
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.
35 lines
1.6 KiB
35 lines
1.6 KiB
import re
|
|
from ..exceptions import FormatError
|
|
from ..utils.patches import create_patch
|
|
|
|
def format(slither, patches, elements):
|
|
for element in elements:
|
|
if element['type'] != "function":
|
|
# Skip variable elements
|
|
continue
|
|
target_contract = slither.get_contract_from_name(element['type_specific_fields']['parent']['name'])
|
|
if target_contract:
|
|
for function in target_contract.functions:
|
|
if function.name == element['name']:
|
|
_patch(slither, patches,
|
|
element['source_mapping']['filename_absolute'],
|
|
element['source_mapping']['filename_relative'],
|
|
int(function.parameters_src.source_mapping['start']),
|
|
int(function.returns_src.source_mapping['start']))
|
|
break
|
|
|
|
|
|
def _patch(slither, patches, in_file, in_file_relative, modify_loc_start, modify_loc_end):
|
|
in_file_str = slither.source_code[in_file].encode('utf-8')
|
|
old_str_of_interest = in_file_str[modify_loc_start:modify_loc_end]
|
|
# Find the keywords view|pure|constant and remove them
|
|
m = re.search("(view|pure|constant)", old_str_of_interest.decode('utf-8'))
|
|
if m:
|
|
create_patch(patches,
|
|
in_file,
|
|
modify_loc_start + m.span()[0],
|
|
modify_loc_start + m.span()[1],
|
|
m.groups(0)[0], # this is view|pure|constant
|
|
"")
|
|
else:
|
|
raise FormatError("No view/pure/constant specifier exists. Regex failed to remove specifier!")
|
|
|