recon-generate 0.0.26 → 0.0.27

package/dist/info.d.ts

@@ -32,6 +32,7 @@ interface InfoOutput {
     with_fallback: string[];
     with_receive: string[];
     coverage_map: Record<string, Record<string, CoverageBlock>>;
+    exclude_from_fuzzing: string[];
 }
 export declare const runInfo: (foundryRoot: string, contractName: string, options?: {
     outputPath?: string;

package/dist/info.js

@@ -907,6 +907,39 @@ const hasReceive = (contract) => {
     const allFunctions = (0, utils_1.getDefinitions)(contract, 'vFunctions', true);
     return allFunctions.some(fn => fn.kind === solc_typed_ast_1.FunctionKind.Receive);
 };
+/**
+ * Check if a function contains a fuzzFromHere() call
+ */
+const hasFuzzFromHereCall = (fnDef) => {
+    for (const call of fnDef.getChildrenByType(solc_typed_ast_1.FunctionCall)) {
+        const expr = call.vExpression;
+        if (expr instanceof solc_typed_ast_1.MemberAccess && expr.memberName === 'fuzzFromHere') {
+            return true;
+        }
+    }
+    return false;
+};
+/**
+ * Collect public/external functions that contain vm.fuzzFromHere() calls
+ * These functions should be excluded from fuzzing
+ */
+const collectExcludeFromFuzzing = (contract) => {
+    const exclude = [];
+    const allFunctions = (0, utils_1.getDefinitions)(contract, 'vFunctions', true).reverse();
+    for (const fnDef of allFunctions) {
+        // Only check public/external functions
+        if (fnDef.visibility !== solc_typed_ast_1.FunctionVisibility.External && fnDef.visibility !== solc_typed_ast_1.FunctionVisibility.Public) {
+            continue;
+        }
+        if (hasFuzzFromHereCall(fnDef)) {
+            // Use function name (not signature) as per the expected output format
+            if (!exclude.includes(fnDef.name)) {
+                exclude.push(fnDef.name);
+            }
+        }
+    }
+    return exclude;
+};
 /**
  * Load source file contents for source location info
  */
@@ -956,7 +989,10 @@ const runInfo = async (foundryRoot, contractName, options = {}) => {
         with_fallback: [],
         with_receive: [],
         coverage_map: {},
+        exclude_from_fuzzing: [],
     };
+    // Collect functions to exclude from fuzzing (from target contract)
+    output.exclude_from_fuzzing = collectExcludeFromFuzzing(targetContract);
     // Collect info for all related contracts
     for (const contract of relatedContracts) {
         // Skip abstract contracts (their functions are inherited by concrete contracts)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "recon-generate",
-  "version": "0.0.26",
+  "version": "0.0.27",
   "description": "CLI to scaffold Recon fuzzing suite inside Foundry projects",
   "main": "dist/index.js",
   "bin": {