using contractTracer

Author: JukLee0ira

src/opCodeFinder.js

@@ -34,8 +34,7 @@ async function validateInputs(args) {
   } else {
     endBlock = parseInt(args[1], 10);
   }
-  opcode = parseInt(args[2], 16);
-
+  opcode = args[2];
   // Validate that the block number is a valid positive integer
   if (
     isNaN(startBlock) ||
@@ -50,142 +49,47 @@ async function validateInputs(args) {
     process.exit(1);
   }
 
-  // Validate that the opcode is a valid hexadecimal string (optionally starting with "0x")
-  if (!/^0x[a-fA-F0-9]+$/i.test(opcode) && !/^[a-fA-F0-9]+$/i.test(opcode)) {
-    console.log(
-      "Opcode must be a valid hexadecimal string, optionally starting with '0x'."
-    );
-    process.exit(1);
-  }
-
   // If all validations pass
   validArgs.push(startBlock, endBlock, opcode, rpcUrl);
   return validArgs;
 }
 
 async function findContracts(startBlockNumber, endBlockNumber) {
-  let contractCount = 0;
-  let failedTxs = 0;
-
   for (let i = startBlockNumber; i <= endBlockNumber; i++) {
     process.stdout.write(
-      `\rSearching for block ${i} ,total found ${contractCount} contracts, ${matchContractCount} of which meet the criteria, ${failedTxs} transactions queried failed`
+      `\rSearching for block ${i} , ${matchContractCount} contracts match the criteria`
     );
-    let block = await web3.eth.getBlock(i);
-    if (block != null) {
-      if (block.transactions != null) {
-        let blockFinishedCount = block.transactions.length;
-        //trace all the transactions in this block
-        for (let txHash of block.transactions) {
-          try {
-            //get the trace object of the transaction
-            const traceObject = await processTransactionsWithRetry(txHash);
-            //check if the transaction created a contract
-            if (
-              traceObject.result.type === "CREATE" ||
-              traceObject.result.type === "CREATE2"
-            ) {
-              getContractBytecode(traceObject.result.to);
-              contractCount++;
-            }
-            // go through any nested calls within the trace.
-            if (
-              traceObject.result.calls &&
-              traceObject.result.calls.length > 0
-            ) {
-              contractCount += isContractCreation(traceObject.result.calls);
-            }
-          } catch (error) {
-            failedTxs++;
-          }
-          blockFinishedCount--;
-        }
-        while (blockFinishedCount > 0) {
-          console.log("waiting...");
-        }
-      }
-    }
-  }
-  console.log();
-}
-
-async function processTransactionsWithRetry(
-  txHash,
-  maxRetries = 3,
-  initialDelay = 1000
-) {
-  let retries = 0;
-  while (retries < maxRetries) {
     try {
       let traceObject = await web3.eth.currentProvider.sendAsync({
-        method: "debug_traceTransaction",
-        params: [`${txHash}`, { tracer: "callTracer" }],
+        method: "debug_traceBlockByNumber",
+        params: [
+          `0x${i.toString(16)}`,
+          { tracer: "contractTracer", tracerConfig: { opCode: `${opCode}` } },
+        ],
         jsonrpc: "2.0",
         id: "1",
       });
-      //if the trace object is not found or there is an error, throw an error
       if (!traceObject || traceObject.error) {
         throw new Error(
-          "Failed to get trace object: " +
-            (traceObject.error ? traceObject.error.message : "unknown error")
+          traceObject.error ? traceObject.error.message : "unknown error"
         );
       }
-      //if the trace object is found, return it
-      return traceObject;
+      //go through traceObject.result
+      if (traceObject.result.length === 0) continue;
+      // Loop through traceObject.result array
+      traceObject.result.forEach((item) => {
+        if (item.result && item.result.length > 0) {
+          item.result.forEach((element) => {
+            if (element && element.length > 0) saveTheContract(element);
+          });
+        }
+      });
     } catch (error) {
-      //if the trace object is not found or there is an error, retry
-      retries++;
-      //if the retries reach the maximum, handle failed transactions
-      if (retries >= maxRetries) {
-        fs.appendFile(
-          "failed-transactions.txt",
-          txHash + "\t" + error.message + "\n",
-          { encoding: "utf8" },
-          (err) => {
-            if (err) {
-              console.error("Error occurred while appending content.");
-              return;
-            }
-          }
-        );
-        throw error;
-      }
-      await new Promise((resolve) =>
-        setTimeout(resolve, initialDelay * Math.pow(2, retries))
-      );
+      //check if the error and throw a specific error message
+      console.error("\nAn error occurred in block " + i + ":" + error.message);
     }
   }
-}
-
-//check if the tx created a contract
-function isContractCreation(traceRes) {
-  let contractCount = 0;
-
-  for (let call of traceRes) {
-    // check if the current call creates a contract.
-    if (call.type === "CREATE" || call.type === "CREATE2") {
-      getContractBytecode(call.to);
-      contractCount++;
-    }
-
-    // if the current call has sub-calls, go through each sub-call.
-    if (call.calls && call.calls.length > 0) {
-      contractCount += isContractCreation(call.calls);
-    }
-  }
-  return contractCount;
-}
-
-async function getContractBytecode(contractArr) {
-  try {
-    let bytecode = await web3.eth.getCode(contractArr);
-    let isContainsTheOpCode = findOpcode(stringToHexArray(bytecode));
-    if (isContainsTheOpCode) {
-      saveTheContract(contractArr); //saveTheContract in a txt file
-    }
-  } catch (error) {
-    console.error("Error fetching contract bytecode:", error);
-  }
+  console.log();
 }
 
 function saveTheContract(contractAddress) {
@@ -203,48 +107,6 @@ function saveTheContract(contractAddress) {
   matchContractCount++;
 }
 
-function stringToHexArray(hexString) {
-  //  Remove the "0x" prefix
-  hexString = hexString.replace(/^0x/, "");
-  // Ensure the string length is even
-  if (hexString.length % 2 !== 0) {
-    throw new Error("Invalid hex string length");
-  }
-  // Convert each pair of characters to hexadecimal values
-  let hexArray = [];
-  for (let i = 0; i < hexString.length; i += 2) {
-    let hexByte = hexString.substring(i, i + 2);
-    hexArray.push("0x" + hexByte);
-  }
-  return hexArray;
-}
-
-function findOpcode(bytecode) {
-  for (let i = 0; i < bytecode.length; ) {
-    let op = bytecode[i];
-    // skip PUSH opcodes
-    if (isPushOpcode(op)) {
-      // increment i, skipping PUSH argument
-      i += skipPush(op, i);
-    } else if (op == opCode) {
-      return true;
-    }
-    i++;
-  }
-  return false;
-}
-
-// Check if the opcode is a PUSH opcode
-function isPushOpcode(op) {
-  return op >= 0x5f && op <= 0x7f; // PUSH1 (0x5f) to PUSH32 (0x7f)
-}
-// Example skipPush function
-function skipPush(op) {
-  // The PUSH opcodes are consecutive, so we can use the opcode to determine how many bytes to skip
-  // For example, PUSH1 (0x5f) skips 1 byte, PUSH32 (0x7f) skips 32 bytes
-  return op - 0x60 + 1;
-}
-
 async function main() {
   const args = process.argv.slice(2);
   const validArgs = await validateInputs(args);