more

include/util/mallocator.h

@@ -27,7 +27,7 @@ typedef struct SMemAllocator SMemAllocator;
 #define MALLOCATOR_APIS                                        \
   void *(*malloc)(SMemAllocator *, size_t size);               \
   void *(*calloc)(SMemAllocator *, size_t nmemb, size_t size); \
-  void *(*realloc)(SMemAllocator *, size_t size);              \
+  void *(*realloc)(SMemAllocator *, void *ptr, size_t size);   \
   void (*free)(SMemAllocator *, void *ptr);                    \
   size_t (*usage)(SMemAllocator *);
 
@@ -50,6 +50,12 @@ void           tdDestroyHeapAllocator(SMemAllocator *pMemAllocator);
 SMemAllocator *tdCreateArenaAllocator(size_t size);
 void           tdDestroyArenaAllocator(SMemAllocator *);
 
+#define mMalloc(pMemAllocator, size) (*(pMemAllocator->malloc))(pMemAllocator, size)
+#define mCalloc(pMemAllocator, nmemb, size) (*(pMemAllocator->calloc))(pMemAllocator, nmemb, size)
+#define mRealloc(pMemAllocator, ptr, size) (*(pMemAllocator->realloc))(pMemAllocator, ptr, size)
+#define mFree(pMemAllocator, ptr) (*(pMemAllocator->free))(pMemAllocator, ptr)
+#define mUsage(pMemAllocator) (*(pMemAllocator->usage))(pMemAllocator)
+
 #ifdef __cplusplus
 }
 #endif

source/util/src/arenaAllocator.c

-/*
- * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
- *
- * This program is free software: you can use, redistribute, and/or modify
- * it under the terms of the GNU Affero General Public License, version 3
- * or later ("AGPL"), as published by the Free Software Foundation.
- *
- * This program is distributed in the hope that it will be useful, but WITHOUT
- * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
- * FITNESS FOR A PARTICULAR PURPOSE.
- *
- * You should have received a copy of the GNU Affero General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- */
\ No newline at end of file

source/util/src/heapAllocator.c → source/util/src/mallocator.c

@@ -15,6 +15,100 @@
 
 #include "mallocator.h"
 
+/* ------------------------ HEAP ALLOCATOR ------------------------ */
+typedef struct {
+  size_t tusage;
+} SHeapAllocator;
+
+static void * haMalloc(SMemAllocator *pma, size_t size);
+static void * haCalloc(SMemAllocator *pma, size_t nmemb, size_t size);
+static void * haRealloc(SMemAllocator *pma, void *ptr, size_t size);
+static void   haFree(SMemAllocator *pma, void *ptr);
+static size_t haUsage(SMemAllocator *pma);
+
+SMemAllocator *tdCreateHeapAllocator() {
+  SMemAllocator *pma = NULL;
+
+  pma = calloc(1, sizeof(SMemAllocator) + sizeof(SHeapAllocator));
+  if (pma) {
+    pma->impl = POINTER_SHIFT(pma, sizeof(SMemAllocator));
+    pma->malloc = haMalloc;
+    pma->calloc = haCalloc;
+    pma->realloc = haRealloc;
+    pma->free = haFree;
+    pma->usage = haUsage;
+  }
+
+  return pma;
+}
+
+void tdDestroyHeapAllocator(SMemAllocator *pMemAllocator) {
+  // TODO
+}
+
+static void *haMalloc(SMemAllocator *pma, size_t size) {
+  void *          ptr;
+  size_t          tsize = size + sizeof(size_t);
+  SHeapAllocator *pha = (SHeapAllocator *)(pma->impl);
+
+  ptr = malloc(tsize);
+  if (ptr) {
+    *(size_t *)ptr = size;
+    ptr = POINTER_SHIFT(ptr, sizeof(size_t));
+    atomic_fetch_add_64(&(pha->tusage), tsize);
+  }
+
+  return ptr;
+}
+
+static void *haCalloc(SMemAllocator *pma, size_t nmemb, size_t size) {
+  void * ptr;
+  size_t tsize = nmemb * size;
+
+  ptr = haMalloc(pma, tsize);
+  if (ptr) {
+    memset(ptr, 0, tsize);
+  }
+
+  return ptr;
+}
+
+static void *haRealloc(SMemAllocator *pma, void *ptr, size_t size) {
+  size_t psize;
+  size_t tsize = size + sizeof(size_t);
+
+  if (ptr == NULL) {
+    psize = 0;
+  } else {
+    psize = *(size_t *)POINTER_SHIFT(ptr, -sizeof(size_t));
+  }
+
+  if (psize < size) {
+    // TODO
+  } else {
+    return ptr;
+  }
+}
+
+static void haFree(SMemAllocator *pma, void *ptr) { /* TODO */
+  SHeapAllocator *pha = (SHeapAllocator *)(pma->impl);
+  if (ptr) {
+    size_t tsize = *(size_t *)POINTER_SHIFT(ptr, -sizeof(size_t)) + sizeof(size_t);
+    atomic_fetch_sub_64(&(pha->tusage), tsize);
+    free(POINTER_SHIFT(ptr, -sizeof(size_t)));
+  }
+}
+
+static size_t haUsage(SMemAllocator *pma) { return ((SHeapAllocator *)(pma->impl))->tusage; }
+
+/* ------------------------ ARENA ALLOCATOR ------------------------ */
+typedef struct {
+  size_t usage;
+} SArenaAllocator;
+
+#if 0
+SMemAllocator *pDefaultMA;
+
 typedef struct {
   char name[64];
 } SHeapAllocator;
@@ -82,4 +176,5 @@ void *haCalloc(SMemAllocator *pMemAllocator, size_t nmemb, size_t size) {
 }
 
 static void haFree(SMemAllocator *pMemAllocator, void *ptr) { /* TODO */
-}
\ No newline at end of file
+}
+#endif
\ No newline at end of file