Merge pull request #487 from vikramarun/master

Adding EVM chain support

Author: 0xalpharush

crytic_compile/cryticparser/cryticparser.py

@@ -385,6 +385,30 @@ def _init_etherscan(parser: ArgumentParser) -> None:
         default=DEFAULTS_FLAG_IN_CONFIG["etherscan_api_key"],
     )
 
+    group_etherscan.add_argument(
+        "--base-apikey",
+        help="Basescan API key.",
+        action="store",
+        dest="base_api_key",
+        default=DEFAULTS_FLAG_IN_CONFIG["etherscan_api_key"],
+    )
+
+    group_etherscan.add_argument(
+        "--gno-apikey",
+        help="Gnosisscan API key.",
+        action="store",
+        dest="gno_api_key",
+        default=DEFAULTS_FLAG_IN_CONFIG["etherscan_api_key"],
+    )
+
+    group_etherscan.add_argument(
+        "--polyzk-apikey",
+        help="zkEVM Polygonscan API key.",
+        action="store",
+        dest="polyzk_api_key",
+        default=DEFAULTS_FLAG_IN_CONFIG["etherscan_api_key"],
+    )
+
     group_etherscan.add_argument(
         "--etherscan-export-directory",
         help="Directory in which to save the analyzed contracts.",

crytic_compile/platform/etherscan.py

@@ -50,6 +50,9 @@
     "testnet.avax:": ("-testnet.snowtrace.io", "testnet.snowtrace.io"),
     "ftm:": (".ftmscan.com", "ftmscan.com"),
     "goerli.base:": ("-goerli.basescan.org", "goerli.basescan.org"),
+    "base:": (".basescan.org", "basescan.org"),
+    "gno:": (".gnosisscan.io", "gnosisscan.io"),
+    "polyzk:": ("-zkevm.polygonscan.com", "zkevm.polygonscan.com"),
 }
 
 
@@ -237,6 +240,9 @@ def compile(self, crytic_compile: "CryticCompile", **kwargs: str) -> None:
         ftmscan_api_key = kwargs.get("ftmscan_api_key", None)
         bscan_api_key = kwargs.get("bscan_api_key", None)
         optim_api_key = kwargs.get("optim_api_key", None)
+        base_api_key = kwargs.get("base_api_key", None)
+        gno_api_key = kwargs.get("gno_api_key", None)
+        polyzk_api_key = kwargs.get("polyzk_api_key", None)
 
         export_dir = kwargs.get("export_dir", "crytic-export")
         export_dir = os.path.join(
@@ -267,6 +273,15 @@ def compile(self, crytic_compile: "CryticCompile", **kwargs: str) -> None:
         if optim_api_key and "optim" in etherscan_url:
             etherscan_url += f"&apikey={optim_api_key}"
             etherscan_bytecode_url += f"&apikey={optim_api_key}"
+        if base_api_key and "base" in etherscan_url:
+            etherscan_url += f"&apikey={base_api_key}"
+            etherscan_bytecode_url += f"&apikey={base_api_key}"
+        if gno_api_key and "gno" in etherscan_url:
+            etherscan_url += f"&apikey={gno_api_key}"
+            etherscan_bytecode_url += f"&apikey={gno_api_key}"
+        if polyzk_api_key and "zkevm" in etherscan_url:
+            etherscan_url += f"&apikey={polyzk_api_key}"
+            etherscan_bytecode_url += f"&apikey={polyzk_api_key}"
 
         source_code: str = ""
         result: Dict[str, Union[bool, str, int]] = {}
@@ -299,6 +314,10 @@ def compile(self, crytic_compile: "CryticCompile", **kwargs: str) -> None:
                 LOGGER.error("Incorrect etherscan request")
                 raise InvalidCompilation("Incorrect etherscan request " + etherscan_url)
 
+            if not info["message"].startswith("OK") and "Invalid API Key" in info["result"]:
+                LOGGER.error("Invalid etherscan API Key")
+                raise InvalidCompilation("Invalid etherscan API Key: " + etherscan_url)
+
             if not info["message"].startswith("OK"):
                 LOGGER.error("Contract has no public source code")
                 raise InvalidCompilation("Contract has no public source code: " + etherscan_url)

setup.py

@@ -11,10 +11,10 @@
     description="Util to facilitate smart contracts compilation.",
     url="https://github.com/crytic/crytic-compile",
     author="Trail of Bits",
-    version="0.3.2",
+    version="0.3.4",
     packages=find_packages(),
     python_requires=">=3.8",
-    install_requires=["pycryptodome>=3.4.6", "cbor2", "solc-select>=v1.0.2"],
+    install_requires=["pycryptodome>=3.4.6", "cbor2", "solc-select>=v1.0.4"],
     extras_require={
         "test": [
             "pytest",