keep ref

trim21 · May 20, 2024 · 40874a3 · 40874a3
1 parent 77bcb69
commit 40874a3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/bencode_c/encode.h b/src/bencode_c/encode.h
@@ -1,3 +1,4 @@
+#include "object.h"
 #include "stdio.h"
 #include "string.h"
 
@@ -252,16 +253,19 @@ static HPy bencode(HPy self, HPy obj) {
   // self is the module object
 
   struct buffer *buf = newBuffer();
+  Py_IncRef(obj);
 
   if (encodeAny(buf, obj)) {
     freeBuffer(buf);
+    Py_DecRef(obj);
     // error when encoding
     return NULL;
   }
 
   HPy res = PyBytes_FromStringAndSize(buf->buf, buf->len);
 
   freeBuffer(buf);
+  Py_DecRef(obj);
 
   return res;
 };