tutil.h 5.4 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 100 101 102 103 104 105 106 107 108 109 110 111 112 113

#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 已提交
114
char *strnchr(char *haystack, char needle, int32_t len, bool skipquote);
H
hzcheng 已提交
115 116 117

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

weixin_48148422's avatar
weixin_48148422 已提交
118
char* strtolower(char *dst, const char *src);
H
hzcheng 已提交
119 120 121 122 123

int64_t strnatoi(char *num, int32_t len);

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

S
slguan 已提交
124 125 126 127
#define POW2(x) ((x) * (x))

int32_t strdequote(char *src);

H
hzcheng 已提交
128 129 130 131 132 133 134 135 136 137 138 139
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 已提交
140
int32_t taosInitTimer(void (*callback)(int), int32_t ms);
H
hzcheng 已提交
141 142 143

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

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

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

H
hjxilinx 已提交
148
bool taosValidateEncodec(const char *encodec);
H
hzcheng 已提交
149

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

152 153 154 155 156 157 158 159
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 已提交
160 161
int taosCheckVersion(char *input_client_version, char *input_server_version, int compared_segments);

S
slguan 已提交
162 163
char *taosIpStr(uint32_t ipInt);

S
slguan 已提交
164 165
uint32_t ip2uint(const char *const ip_addr);

weixin_48148422's avatar
weixin_48148422 已提交
166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189
#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 已提交
190
#define getline(lineptr, n, stream) taos_getline(lineptr, n, stream, __FILE__, __LINE__)
S
slguan 已提交
191

weixin_48148422's avatar
weixin_48148422 已提交
192 193 194
#endif  // TAOS_MEM_CHECK_IMPL

#endif // TAOS_MEM_CHECK
S
slguan 已提交
195

H
hzcheng 已提交
196 197 198 199
#ifdef __cplusplus
}
#endif

S
slguan 已提交
200
#endif  // TDENGINE_TUTIL_H