Fix slither-flat external/private options (#964)

The options --convert-external and --convert-private are currently
broken from what looks like some automatic conversion according to git
blame.
Also fixes issue #580
pull/983/head
Marius G 3 years ago committed by GitHub
parent f3de50a8c6
commit 7887f3914d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 6
      slither/tools/flattening/flattening.py

@ -88,7 +88,7 @@ class Flattening:
:return: :return:
""" """
src_mapping = contract.source_mapping src_mapping = contract.source_mapping
content = self._slither.source_code[src_mapping["filename_absolute"]].encode("utf8") content = self._slither.source_code[src_mapping["filename_absolute"]]
start = src_mapping["start"] start = src_mapping["start"]
end = src_mapping["start"] + src_mapping["length"] end = src_mapping["start"] + src_mapping["length"]
@ -171,7 +171,7 @@ class Flattening:
index = index - start index = index - start
if patch_type == "public_to_external": if patch_type == "public_to_external":
content = content[:index] + "public" + content[index + len("external") :] content = content[:index] + "public" + content[index + len("external") :]
if patch_type == "private_to_internal": elif patch_type == "private_to_internal":
content = content[:index] + "internal" + content[index + len("private") :] content = content[:index] + "internal" + content[index + len("private") :]
elif patch_type == "calldata_to_memory": elif patch_type == "calldata_to_memory":
content = content[:index] + "memory" + content[index + len("calldata") :] content = content[:index] + "memory" + content[index + len("calldata") :]
@ -179,7 +179,7 @@ class Flattening:
assert patch_type == "line_removal" assert patch_type == "line_removal"
content = content[:index] + " // " + content[index:] content = content[:index] + " // " + content[index:]
self._source_codes[contract] = content.decode("utf8") self._source_codes[contract] = content
def _pragmas(self) -> str: def _pragmas(self) -> str:
""" """

Loading…
Cancel
Save