Change the variable name of full_contract_source

pull/1031/head
Nikhil Parasaram 6 years ago
parent 584064d150
commit 58dad6c23f
  1. 22
      mythril/solidity/soliditycontract.py

@ -22,16 +22,16 @@ class SourceMapping:
class SolidityFile:
"""Representation of a file containing Solidity code."""
def __init__(self, filename: str, data: str, full_contract_source: Set[str]):
def __init__(self, filename: str, data: str, full_contract_src_maps: Set[str]):
"""
Metadata class containing data regarding a specific solidity file
:param filename: The filename of the solidity file
:param data: The code of the solidity file
:param full_contract_source: The set of contract source mappings of all the contracts in the file
:param full_contract_src_maps: The set of contract source mappings of all the contracts in the file
"""
self.filename = filename
self.data = data
self.full_contract_source = full_contract_source
self.full_contract_src_maps = full_contract_src_maps
class SourceCodeInfo:
@ -78,11 +78,11 @@ class SolidityContract(EVMContract):
for filename in data["sourceList"]:
with open(filename, "r", encoding="utf-8") as file:
code = file.read()
full_contract_sources = self.get_full_contract_sources(
full_contract_src_maps = self.get_full_contract_src_maps(
data["sources"][filename]["AST"]
)
self.solidity_files.append(
SolidityFile(filename, code, full_contract_sources)
SolidityFile(filename, code, full_contract_src_maps)
)
has_contract = False
@ -132,17 +132,17 @@ class SolidityContract(EVMContract):
super().__init__(code, creation_code, name=name)
@staticmethod
def get_full_contract_sources(ast: Dict) -> Set[str]:
def get_full_contract_src_maps(ast: Dict) -> Set[str]:
"""
Takes a solc AST and gets the src mappings for all the contracts defined in the top level of the ast
:param ast: AST of the contract
:return: The source map
:return: The source maps
"""
source_map = set()
source_maps = set()
for child in ast["children"]:
if "contractKind" in child["attributes"]:
source_map.add(child["src"])
return source_map
source_maps.add(child["src"])
return source_maps
def get_source_info(self, address, constructor=False):
"""
@ -181,7 +181,7 @@ class SolidityContract(EVMContract):
# Handle the common code src map for the entire code.
if (
"{}:{}:{}".format(offset, length, file_index)
in self.solidity_files[file_index].full_contract_source
in self.solidity_files[file_index].full_contract_src_maps
):
return True

Loading…
Cancel
Save