def _load_from_compile_legacy2(crytic_compile: "CryticCompile", loaded_json: Dict) -> None:
"""Load from old (old) export
Args:
crytic_compile (CryticCompile): CryticCompile object to populate
loaded_json (Dict): Json representation of the CryticCompile object
"""
for key, compilation_unit_json in loaded_json["compilation_units"].items():
compilation_unit = CompilationUnit(crytic_compile, key)
compilation_unit.compiler_version = CompilerVersion(
compiler=compilation_unit_json["compiler"]["compiler"],
version=compilation_unit_json["compiler"]["version"],
optimized=compilation_unit_json["compiler"]["optimized"],
)
for contract_name, contract in compilation_unit_json["contracts"].items():
compilation_unit.contracts_names.add(contract_name)
filename = Filename(
absolute=contract["filenames"]["absolute"],
relative=contract["filenames"]["relative"],
short=contract["filenames"]["short"],
used=contract["filenames"]["used"],
)
compilation_unit.filename_to_contracts[filename].add(contract_name)
compilation_unit.abis[contract_name] = contract["abi"]
compilation_unit.bytecodes_init[contract_name] = contract["bin"]
compilation_unit.bytecodes_runtime[contract_name] = contract["bin-runtime"]
compilation_unit.srcmaps_init[contract_name] = contract["srcmap"].split(";")
compilation_unit.srcmaps_runtime[contract_name] = contract["srcmap-runtime"].split(";")
compilation_unit.libraries[contract_name] = contract["libraries"]
userdoc = contract.get("userdoc", {})
devdoc = contract.get("devdoc", {})
compilation_unit.natspec[contract_name] = Natspec(userdoc, devdoc)
if contract["is_dependency"]:
crytic_compile.dependencies.add(filename.absolute)
crytic_compile.dependencies.add(filename.relative)
crytic_compile.dependencies.add(filename.short)
crytic_compile.dependencies.add(filename.used)
compilation_unit.asts = compilation_unit_json["asts"]
if "filenames" in compilation_unit_json:
compilation_unit.filenames = {
_convert_dict_to_filename(filename)
for filename in compilation_unit_json["filenames"]
}
else:
# For legay code, we recover the filenames from the contracts list
# This is not perfect, as a filename might not be associated to any contract
for contract_name, contract in compilation_unit_json["contracts"].items():
filename = _convert_dict_to_filename(contract["filenames"])
compilation_unit.filenames.add(filename)
def _load_from_compile_current(crytic_compile: "CryticCompile", loaded_json: Dict) -> None:
for key, compilation_unit_json in loaded_json["compilation_units"].items():
compilation_unit = CompilationUnit(crytic_compile, key)
compilation_unit.compiler_version = CompilerVersion(
compiler=compilation_unit_json["compiler"]["compiler"],
version=compilation_unit_json["compiler"]["version"],
optimized=compilation_unit_json["compiler"]["optimized"],
)
for contracts_data in compilation_unit_json["contracts"].values():
for contract_name, contract in contracts_data.items():
compilation_unit.contracts_names.add(contract_name)
filename = Filename(
absolute=contract["filenames"]["absolute"],
relative=contract["filenames"]["relative"],
short=contract["filenames"]["short"],
used=contract["filenames"]["used"],
)
compilation_unit.filename_to_contracts[filename].add(contract_name)
compilation_unit.abis[contract_name] = contract["abi"]
compilation_unit.bytecodes_init[contract_name] = contract["bin"]
compilation_unit.bytecodes_runtime[contract_name] = contract["bin-runtime"]
compilation_unit.srcmaps_init[contract_name] = contract["srcmap"].split(";")
compilation_unit.srcmaps_runtime[contract_name] = contract["srcmap-runtime"].split(
";"
)
compilation_unit.libraries[contract_name] = contract["libraries"]
userdoc = contract.get("userdoc", {})
devdoc = contract.get("devdoc", {})
compilation_unit.natspec[contract_name] = Natspec(userdoc, devdoc)
if contract["is_dependency"]:
crytic_compile.dependencies.add(filename.absolute)
crytic_compile.dependencies.add(filename.relative)
crytic_compile.dependencies.add(filename.short)
crytic_compile.dependencies.add(filename.used)
compilation_unit.asts = compilation_unit_json["asts"]
compilation_unit.filenames = {
_convert_dict_to_filename(filename)
for filename in compilation_unit_json["filenames"]
}
def _load_from_compile_legacy(crytic_compile: "CryticCompile",
loaded_json: Dict):
compilation_unit = CompilationUnit(crytic_compile, "legacy")
compilation_unit.asts = loaded_json["asts"]
compilation_unit.compiler_version = CompilerVersion(
compiler=loaded_json["compiler"]["compiler"],
version=loaded_json["compiler"]["version"],
optimized=loaded_json["compiler"]["optimized"],
)
for contract_name, contract in loaded_json["contracts"].items():
compilation_unit.contracts_names.add(contract_name)
filename = Filename(
absolute=contract["filenames"]["absolute"],
relative=contract["filenames"]["relative"],
short=contract["filenames"]["short"],
used=contract["filenames"]["used"],
)
compilation_unit.contracts_filenames[contract_name] = filename
compilation_unit.abis[contract_name] = contract["abi"]
compilation_unit.bytecodes_init[contract_name] = contract["bin"]
compilation_unit.bytecodes_runtime[contract_name] = contract[
"bin-runtime"]
compilation_unit.srcmaps_init[contract_name] = contract[
"srcmap"].split(";")
compilation_unit.srcmaps_runtime[contract_name] = contract[
"srcmap-runtime"].split(";")
compilation_unit.libraries[contract_name] = contract["libraries"]
userdoc = contract.get("userdoc", {})
devdoc = contract.get("devdoc", {})
compilation_unit.natspec[contract_name] = Natspec(userdoc, devdoc)
if contract["is_dependency"]:
compilation_unit.crytic_compile.dependencies.add(filename.absolute)
compilation_unit.crytic_compile.dependencies.add(filename.relative)
compilation_unit.crytic_compile.dependencies.add(filename.short)
compilation_unit.crytic_compile.dependencies.add(filename.used)
def load_from_compile(crytic_compile: "CryticCompile",
loaded_json: Dict) -> Tuple[int, List[str]]:
"""
Load from json
:param crytic_compile:
:param loaded_json:
:return:
"""
crytic_compile.package_name = loaded_json.get("package", None)
if "compilation_units" not in loaded_json:
_load_from_compile_legacy(crytic_compile, loaded_json)
else:
for key, compilation_unit_json in loaded_json[
"compilation_units"].items():
compilation_unit = CompilationUnit(crytic_compile, key)
compilation_unit.compiler_version = CompilerVersion(
compiler=compilation_unit_json["compiler"]["compiler"],
version=compilation_unit_json["compiler"]["version"],
optimized=compilation_unit_json["compiler"]["optimized"],
)
for contract_name, contract in compilation_unit_json[
"contracts"].items():
compilation_unit.contracts_names.add(contract_name)
filename = Filename(
absolute=contract["filenames"]["absolute"],
relative=contract["filenames"]["relative"],
short=contract["filenames"]["short"],
used=contract["filenames"]["used"],
)
compilation_unit.contracts_filenames[contract_name] = filename
compilation_unit.abis[contract_name] = contract["abi"]
compilation_unit.bytecodes_init[contract_name] = contract[
"bin"]
compilation_unit.bytecodes_runtime[contract_name] = contract[
"bin-runtime"]
compilation_unit.srcmaps_init[contract_name] = contract[
"srcmap"].split(";")
compilation_unit.srcmaps_runtime[contract_name] = contract[
"srcmap-runtime"].split(";")
compilation_unit.libraries[contract_name] = contract[
"libraries"]
userdoc = contract.get("userdoc", {})
devdoc = contract.get("devdoc", {})
compilation_unit.natspec[contract_name] = Natspec(
userdoc, devdoc)
if contract["is_dependency"]:
crytic_compile.dependencies.add(filename.absolute)
crytic_compile.dependencies.add(filename.relative)
crytic_compile.dependencies.add(filename.short)
crytic_compile.dependencies.add(filename.used)
compilation_unit.asts = compilation_unit_json["asts"]
# Set our filenames
for compilation_unit in crytic_compile.compilation_units.values():
crytic_compile.filenames |= set(
compilation_unit.contracts_filenames.values())
crytic_compile.working_dir = loaded_json["working_dir"]
return loaded_json["type"], loaded_json.get("unit_tests", [])
