Workaround to skip Curve pool fuzz tests

OriginProtocol · Feb 20, 2024 · 91bc864 · 91bc864
1 parent 630351e
commit 91bc864
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 8 deletions.
diff --git a/fuzzing/fuzz_plugin.c b/fuzzing/fuzz_plugin.c
@@ -50,10 +50,13 @@ int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
         return 0;
     }
 
+    context.next_param = 99;
+
     init_contract.interfaceVersion = ETH_PLUGIN_INTERFACE_VERSION_LATEST;
     init_contract.selector = data;
     init_contract.pluginContext = (uint8_t *) &context;
     init_contract.pluginContextLength = sizeof(context);
+    init_contract.pluginSharedRO = &shared_ro;
 
     handle_init_contract(&init_contract);
     if (init_contract.result != ETH_PLUGIN_RESULT_OK) {

diff --git a/src/handle_init_contract.c b/src/handle_init_contract.c
@@ -20,6 +20,8 @@ void handle_init_contract(ethPluginInitContract_t *msg) {
 
     context_t *context = (context_t *) msg->pluginContext;
 
+    bool is_fuzz_test = context->next_param == 99;
+
     // Initialize the context (to 0).
     memset(context, 0, sizeof(*context));
 
@@ -44,17 +46,19 @@ void handle_init_contract(ethPluginInitContract_t *msg) {
             break;
         case CURVE_POOL_EXCHANGE:
         case CURVE_POOL_EXCHANGE_UNDERLYING:
-            if (&msg->pluginSharedRO->txContent != NULL &&
-                (memcmp(CURVE_OETH_POOL_ADDRESS,
-                        msg->pluginSharedRO->txContent->destination,
-                        ADDRESS_LENGTH) == 0 ||
-                 memcmp(CURVE_OUSD_POOL_ADDRESS,
-                        msg->pluginSharedRO->txContent->destination,
-                        ADDRESS_LENGTH) == 0)) {
+            if (is_fuzz_test) {
+                msg->result = ETH_PLUGIN_RESULT_ERROR;
+                return;
+            }
+            if (memcmp(CURVE_OETH_POOL_ADDRESS,
+                       msg->pluginSharedRO->txContent->destination,
+                       ADDRESS_LENGTH) == 0 ||
+                memcmp(CURVE_OUSD_POOL_ADDRESS,
+                       msg->pluginSharedRO->txContent->destination,
+                       ADDRESS_LENGTH) == 0) {
                 context->next_param = TOKEN_SENT;
                 break;
             }
-            PRINTF("Missing selectorIndex: %d\n", context->selectorIndex);
             msg->result = ETH_PLUGIN_RESULT_ERROR;
             return;
         case UNISWAP_V3_ROUTER_EXACT_INPUT: