addded stamp cost for exceeding max_frames

xian-network · Sep 30, 2024 · 9c3c783 · 9c3c783
1 parent 345fad8
commit 9c3c783
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/contracting/execution/metering/tracer.c b/src/contracting/execution/metering/tracer.c
@@ -146,6 +146,7 @@ Tracer_trace(Tracer * self, PyFrameObject * frame, int what, PyObject * arg) {
     self->call_count++;
 
     if (self->call_count > 800000) {
+        self->cost += 100000; // Add a stamp cost
         PyErr_SetString(PyExc_AssertionError, "Call count exceeded threshold! Infinite Loop?");
         PyEval_SetTrace(NULL, NULL); // Stop tracing
         self->started = 0; // Mark tracer as stopped
@@ -164,6 +165,7 @@ Tracer_trace(Tracer * self, PyFrameObject * frame, int what, PyObject * arg) {
     const char *current_function_name = PyUnicode_AsUTF8(code->co_name);
     if (current_function_name == NULL) {
         Py_DECREF(code);
+        Py_DECREF(globals);
         return RET_OK;
     }
     PyObject *globals = PyFrame_GetGlobals(frame);