tutil.h 5.7 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
    }            \
  }

#define tclose(x) taosCloseSocket(x)

H
TD-166  
hzcheng 已提交
47
#ifndef NDEBUG
H
hzcheng 已提交
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 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 100 101 102 103 104 105 106 107 108 109

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

110 111 112 113 114
typedef struct SPair {
  void* first;
  void* sec;
} SPair;

H
hzcheng 已提交
115 116 117 118
int32_t strdequote(char *src);

void strtrim(char *src);

S
slguan 已提交
119
char *strnchr(char *haystack, char needle, int32_t len, bool skipquote);
H
hzcheng 已提交
120 121 122

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

weixin_48148422's avatar
weixin_48148422 已提交
123
char* strtolower(char *dst, const char *src);
H
hzcheng 已提交
124 125 126 127 128

int64_t strnatoi(char *num, int32_t len);

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

H
hjxilinx 已提交
129
char *paGetToken(char *src, char **token, int32_t *tokenLen);
H
hzcheng 已提交
130 131 132 133 134 135 136 137 138

void taosMsleep(int32_t mseconds);

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

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

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

H
hjxilinx 已提交
139 140 141 142 143 144 145
/**
 *
 * @param fileNamePattern
 * @param dstPath
 */
void getTmpfilePath(const char *fileNamePattern, char *dstPath);

S
slguan 已提交
146
int32_t taosInitTimer(void (*callback)(int), int32_t ms);
weixin_48148422's avatar
weixin_48148422 已提交
147
void taosUninitTimer();
H
hzcheng 已提交
148 149 150

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

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

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

H
hjxilinx 已提交
155
bool taosValidateEncodec(const char *encodec);
H
hzcheng 已提交
156

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

159 160 161 162 163 164 165 166
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 已提交
167 168
int taosCheckVersion(char *input_client_version, char *input_server_version, int compared_segments);

S
slguan 已提交
169 170
char *taosIpStr(uint32_t ipInt);

S
slguan 已提交
171 172
uint32_t ip2uint(const char *const ip_addr);

S
slguan 已提交
173 174
void taosRemoveDir(char *rootDir);

weixin_48148422's avatar
weixin_48148422 已提交
175 176 177 178 179 180
#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();

H
TD-100  
hzcheng 已提交
181 182 183 184 185 186 187
void * tmalloc(size_t size);
void * tcalloc(size_t nmemb, size_t size);
size_t tsizeof(void *ptr);
void   tmemset(void *ptr, int c);
void * trealloc(void *ptr, size_t size);
void   tzfree(void *ptr);

weixin_48148422's avatar
weixin_48148422 已提交
188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205
#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 已提交
206
#define getline(lineptr, n, stream) taos_getline(lineptr, n, stream, __FILE__, __LINE__)
S
slguan 已提交
207

weixin_48148422's avatar
weixin_48148422 已提交
208 209 210
#endif  // TAOS_MEM_CHECK_IMPL

#endif // TAOS_MEM_CHECK
S
slguan 已提交
211

H
hzcheng 已提交
212 213 214 215
#ifdef __cplusplus
}
#endif

S
slguan 已提交
216
#endif  // TDENGINE_TUTIL_H