tutil.h 6.3 KB
Newer Older
H
hzcheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
/*
 * 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_TUTIL_H
#define TDENGINE_TUTIL_H

#ifdef __cplusplus
extern "C" {
#endif

S
slguan 已提交
23 24
#include "os.h"
#include "tmd5.h"
H
hzcheng 已提交
25
#include "tcrc32c.h"
S
slguan 已提交
26
#include "taosdef.h"
H
hzcheng 已提交
27

S
slguan 已提交
28
#ifndef STDERR_FILENO
H
hjxilinx 已提交
29
#define STDERR_FILENO (2)
S
slguan 已提交
30
#endif
H
hzcheng 已提交
31

H
hjxilinx 已提交
32 33 34
#define FD_VALID(x) ((x) > STDERR_FILENO)
#define FD_INITIALIZER  ((int32_t)-1)

H
hzcheng 已提交
35
#define WCHAR wchar_t
H
hjxilinx 已提交
36

H
hzcheng 已提交
37 38 39
#define tfree(x) \
  {              \
    if (x) {     \
S
slguan 已提交
40 41
      free((void*)(x));   \
      x = 0;  \
H
hzcheng 已提交
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85
    }            \
  }

#define tclose(x) taosCloseSocket(x)

#ifdef ASSERTION
#define ASSERT(x) assert(x)
#else
#define ASSERT(x)
#endif

#ifdef UNUSED
#undefine UNUSED
#endif
#define UNUSED(x) ((void)(x))

#ifdef UNUSED_FUNC
#undefine UNUSED_FUNC
#endif

#ifdef UNUSED_PARAM
#undef UNUSED_PARAM
#endif

#if defined(__GNUC__)
#define UNUSED_PARAM(x) _UNUSED##x __attribute__((unused))
#define UNUSED_FUNC __attribute__((unused))
#else
#define UNUSED_PARAM(x) x
#define UNUSED_FUNC
#endif

#ifdef tListLen
#undefine tListLen
#endif
#define tListLen(x) (sizeof(x) / sizeof((x)[0]))

#if defined(__GNUC__)
#define FORCE_INLINE inline __attribute__((always_inline))
#else
#define FORCE_INLINE
#endif

#define DEFAULT_UNICODE_ENCODEC "UCS-4LE"
S
slguan 已提交
86
  
H
hzcheng 已提交
87 88 89 90 91 92 93
#define DEFAULT_COMP(x, y)       \
  do {                           \
    if ((x) == (y)) {            \
      return 0;                  \
    } else {                     \
      return (x) < (y) ? -1 : 1; \
    }                            \
94
  } while (0)
H
hzcheng 已提交
95 96 97 98 99

#define GET_INT8_VAL(x)   (*(int8_t *)(x))
#define GET_INT16_VAL(x)  (*(int16_t *)(x))
#define GET_INT32_VAL(x)  (*(int32_t *)(x))
#define GET_INT64_VAL(x)  (*(int64_t *)(x))
S
slguan 已提交
100 101 102 103 104

#ifdef _TD_ARM_32_
  #define GET_FLOAT_VAL(x)  taos_align_get_float(x)
  #define GET_DOUBLE_VAL(x) taos_align_get_double(x)

105 106
  float  taos_align_get_float(const char* pBuf);
  double taos_align_get_double(const char* pBuf);
S
slguan 已提交
107 108 109 110 111 112 113 114 115 116 117

  //#define __float_align_declear()  float __underlyFloat = 0.0;
  //#define __float_align_declear()
  //#define GET_FLOAT_VAL_ALIGN(x) (*(int32_t*)&(__underlyFloat) = *(int32_t*)(x); __underlyFloat);
  // notes: src must be float or double type variable !!!
  #define SET_FLOAT_VAL_ALIGN(dst, src) (*(int32_t*) dst = *(int32_t*)src);
  #define SET_DOUBLE_VAL_ALIGN(dst, src) (*(int64_t*) dst = *(int64_t*)src);
#else
  #define GET_FLOAT_VAL(x)  (*(float *)(x))
  #define GET_DOUBLE_VAL(x) (*(double *)(x))
#endif
H
hzcheng 已提交
118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136

#define ALIGN_NUM(n, align) (((n) + ((align)-1)) & (~((align)-1)))

// align to 8bytes
#define ALIGN8(n) ALIGN_NUM(n, 8)

#define MILLISECOND_PER_MINUTE (MILLISECOND_PER_SECOND * 60)
#define MILLISECOND_PER_HOUR   (MILLISECOND_PER_MINUTE * 60)
#define MILLISECOND_PER_DAY    (MILLISECOND_PER_HOUR * 24)
#define MILLISECOND_PER_WEEK   (MILLISECOND_PER_DAY * 7)
#define MILLISECOND_PER_MONTH  (MILLISECOND_PER_DAY * 30)
#define MILLISECOND_PER_YEAR   (MILLISECOND_PER_DAY * 365)

#define POW2(x) ((x) * (x))

int32_t strdequote(char *src);

void strtrim(char *src);

S
slguan 已提交
137
char *strnchr(char *haystack, char needle, int32_t len, bool skipquote);
H
hzcheng 已提交
138 139 140

char **strsplit(char *src, const char *delim, int32_t *num);

weixin_48148422's avatar
weixin_48148422 已提交
141
char* strtolower(char *dst, const char *src);
H
hzcheng 已提交
142 143 144 145 146

int64_t strnatoi(char *num, int32_t len);

char* strreplace(const char* str, const char* pattern, const char* rep);

S
slguan 已提交
147 148 149 150
#define POW2(x) ((x) * (x))

int32_t strdequote(char *src);

H
hzcheng 已提交
151 152 153 154 155 156 157 158 159 160 161 162
char *paGetToken(char *src, char **token, int32_t *tokenLen);

void taosMsleep(int32_t mseconds);

int32_t taosByteArrayToHexStr(char bytes[], int32_t len, char hexstr[]);

int32_t taosHexStrToByteArray(char hexstr[], char bytes[]);

int64_t str2int64(char *str);

int32_t taosFileRename(char *fullPath, char *suffix, char delimiter, char **dstPath);

S
slguan 已提交
163
int32_t taosInitTimer(void (*callback)(int), int32_t ms);
H
hzcheng 已提交
164 165 166

bool taosMbsToUcs4(char *mbs, int32_t mbs_len, char *ucs4, int32_t ucs4_max_len);

S
slguan 已提交
167 168
int tasoUcs4Compare(void* f1_ucs4, void *f2_ucs4, int bytes);

H
hzcheng 已提交
169 170
bool taosUcs4ToMbs(void *ucs4, int32_t ucs4_max_len, char *mbs);

H
hjxilinx 已提交
171
bool taosValidateEncodec(const char *encodec);
H
hzcheng 已提交
172

S
slguan 已提交
173 174
bool taosGetVersionNumber(char *versionStr, int *versionNubmer);

175 176 177 178 179 180 181 182
static FORCE_INLINE void taosEncryptPass(uint8_t *inBuf, unsigned int inLen, char *target) {
  MD5_CTX context;
  MD5Init(&context);
  MD5Update(&context, inBuf, inLen);
  MD5Final(&context);
  memcpy(target, context.digest, TSDB_KEY_LEN);
}

S
slguan 已提交
183 184
int taosCheckVersion(char *input_client_version, char *input_server_version, int compared_segments);

S
slguan 已提交
185 186
char *taosIpStr(uint32_t ipInt);

S
slguan 已提交
187 188
uint32_t ip2uint(const char *const ip_addr);

weixin_48148422's avatar
weixin_48148422 已提交
189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212
#define TAOS_ALLOC_MODE_DEFAULT 0
#define TAOS_ALLOC_MODE_RANDOM_FAIL 1
#define TAOS_ALLOC_MODE_DETECT_LEAK 2
void taosSetAllocMode(int mode, const char* path, bool autoDump);
void taosDumpMemoryLeak();

#ifdef TAOS_MEM_CHECK

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);

#ifndef TAOS_MEM_CHECK_IMPL

#define malloc(size) taos_malloc(size, __FILE__, __LINE__)
#define calloc(num, size) taos_calloc(num, size, __FILE__, __LINE__)
#define realloc(ptr, size) taos_realloc(ptr, size, __FILE__, __LINE__)
#define free(ptr) taos_free(ptr, __FILE__, __LINE__)
#define strdup(str) taos_strdup(str, __FILE__, __LINE__)
#define strndup(str, size) taos_strndup(str, size, __FILE__, __LINE__)
weixin_48148422's avatar
weixin_48148422 已提交
213
#define getline(lineptr, n, stream) taos_getline(lineptr, n, stream, __FILE__, __LINE__)
S
slguan 已提交
214

weixin_48148422's avatar
weixin_48148422 已提交
215 216 217
#endif  // TAOS_MEM_CHECK_IMPL

#endif // TAOS_MEM_CHECK
S
slguan 已提交
218

H
hzcheng 已提交
219 220 221 222
#ifdef __cplusplus
}
#endif

S
slguan 已提交
223
#endif  // TDENGINE_TUTIL_H