osMemory.h

/*
 * 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/>.
 */

#ifndef TDENGINE_OS_MEMORY_H
#define TDENGINE_OS_MEMORY_H

#include "osString.h"

#ifdef __cplusplus
extern "C" {
#endif

typedef enum {
  TAOS_ALLOC_MODE_DEFAULT = 0,
  TAOS_ALLOC_MODE_RANDOM_FAIL = 1,
  TAOS_ALLOC_MODE_DETECT_LEAK = 2
} ETaosMemoryAllocMode;

void   taosSetAllocMode(int mode, const char *path, bool autoDump);
void   taosDumpMemoryLeak();

void * taosTMalloc(size_t size);
void * taosTCalloc(size_t nmemb, size_t size);
void * taosTRealloc(void *ptr, size_t size);
void   taosTZfree(void *ptr);
size_t taosTSizeof(void *ptr);
void   taosTMemset(void *ptr, int c);

#define taosTFree(x)     \
  do {                   \
    if (x) {             \
      free((void *)(x)); \
      x = 0;             \
    }                    \
  } while (0);

#ifndef TAOS_MEM_CHECK
  #define taosMalloc(size) malloc(size)
  #define taosCalloc(num, size) calloc(num, size)
  #define taosRealloc(ptr, size) realloc(ptr, size)
  #define taosFree(ptr) free(ptr)
  #define taosStrdup(str) taosStrdupImp(str)
  #define taosStrndup(str, size) taosStrndupImp(str, size)
  #define taosGetline(lineptr, n, stream) taosGetlineImp(lineptr, n, stream)
#else
  void *  taos_malloc(size_t size, const char *file, uint32_t line);
  void *  taos_calloc(size_t num, size_t size, const char *file, uint32_t line);
  void *  taos_realloc(void *ptr, size_t size, const char *file, uint32_t line);
  void    taos_free(void *ptr, const char *file, uint32_t line);
  char *  taos_strdup(const char *str, const char *file, uint32_t line);
  char *  taos_strndup(const char *str, size_t size, const char *file, uint32_t line);
  ssize_t taos_getline(char **lineptr, size_t *n, FILE *stream, const char *file, uint32_t line);
  #define taosMalloc(size) taos_malloc(size, __FILE__, __LINE__)
  #define taosCalloc(num, size) taos_calloc(num, size, __FILE__, __LINE__)
  #define taosRealloc(ptr, size) taos_realloc(ptr, size, __FILE__, __LINE__)
  #define taosFree(ptr) taos_free(ptr, __FILE__, __LINE__)
  #define taosStrdup(str) taos_strdup(str, __FILE__, __LINE__)
  #define taosStrndup(str, size) taos_strndup(str, size, __FILE__, __LINE__)
  #define taosGetline(lineptr, n, stream) taos_getline(lineptr, n, stream, __FILE__, __LINE__)
#endif 

#ifdef __cplusplus
}
#endif

#endif