more TDB

source/libs/tdb/src/db/tdbBtree.c

@@ -207,6 +207,7 @@ int tdbBtreeGet(SBTree *pBt, const void *pKey, int kLen, void **ppVal, int *vLen
   SBTC         btc;
   SCell       *pCell;
   int          cret;
+  void        *pVal;
   SCellDecoder cd;
 
   tdbBtreeCursor(&btc, pBt);
@@ -221,12 +222,12 @@ int tdbBtreeGet(SBTree *pBt, const void *pKey, int kLen, void **ppVal, int *vLen
   tdbBtreeDecodeCell(btc.pPage, pCell, &cd);
 
   *vLen = cd.vLen;
-  // TODO: here may have memory leak
-  *ppVal = realloc(*ppVal, *vLen);
-  if (*ppVal == NULL) {
+  pVal = TDB_REALLOC(*ppVal, *vLen);
+  if (pVal == NULL) {
     return -1;
   }
 
+  *ppVal = pVal;
   memcpy(*ppVal, cd.pVal, cd.vLen);
   return 0;
 }

source/libs/tdb/src/inc/tdbUtil.h

@@ -39,6 +39,28 @@ int tdbGetFileSize(const char *fname, int pgSize, SPgno *pSize);
 
 int tdbPRead(int fd, void *pData, int count, i64 offset);
 
+#define TDB_REALLOC(PTR, SIZE)                                                          \
+  ({                                                                                    \
+    void *nPtr;                                                                         \
+    if ((PTR) == NULL || ((int *)(PTR))[-1] < (SIZE)) {                                 \
+      nPtr = realloc((PTR) ? (char *)(PTR) - sizeof(int) : NULL, (SIZE) + sizeof(int)); \
+      if (nPtr) {                                                                       \
+        ((int *)nPtr)[0] = (SIZE);                                                      \
+        nPtr = (char *)nPtr + sizeof(int);                                              \
+      }                                                                                 \
+    } else {                                                                            \
+      nPtr = (PTR);                                                                     \
+    }                                                                                   \
+    nPtr;                                                                               \
+  })
+
+#define TDB_FREE(PTR)                    \
+  do {                                   \
+    if (PTR) {                           \
+      free((char *)(PTR) - sizeof(int)); \
+    }                                    \
+  } while (0)
+
 static inline void *tdbOsMalloc(void *arg, size_t size) {
   void *ptr;
 

source/libs/tdb/test/tdbTest.cpp

@@ -44,6 +44,8 @@ TEST(tdb_test, simple_test) {
         GTEST_ASSERT_EQ(vLen, strlen(val));
         GTEST_ASSERT_EQ(memcmp(val, pVal, vLen), 0);
       }
+
+      TDB_FREE(pVal);
     }
 
     {  // Loop to query the data