osDef.h 5.7 KB
Newer Older
H
refact  
Hongze Cheng 已提交
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 _TD_OS_DEF_H_
#define _TD_OS_DEF_H_

#ifdef __cplusplus
extern "C" {
#endif

S
Shengliang Guan 已提交
23 24
#if defined(_TD_DARWIN_64)
  // specific
wafwerar's avatar
wafwerar 已提交
25 26
#ifndef __COMPAR_FN_T
#define __COMPAR_FN_T
S
Shengliang Guan 已提交
27
  typedef int(*__compar_fn_t)(const void *, const void *);
wafwerar's avatar
wafwerar 已提交
28
#endif
S
Shengliang Guan 已提交
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46

  // for send function in tsocket.c
  #if defined(MSG_NOSIGNAL)
    #undef MSG_NOSIGNAL
  #endif

  #define MSG_NOSIGNAL             0

  #define SO_NO_CHECK              0x1234
  #define SOL_TCP                  0x1234
  #define TCP_KEEPIDLE             0x1234

  #ifndef PTHREAD_MUTEX_RECURSIVE_NP
    #define PTHREAD_MUTEX_RECURSIVE_NP PTHREAD_MUTEX_RECURSIVE
  #endif
#endif

#if defined(_ALPINE)
wafwerar's avatar
wafwerar 已提交
47 48
#ifndef __COMPAR_FN_T
#define __COMPAR_FN_T
S
Shengliang Guan 已提交
49
  typedef int(*__compar_fn_t)(const void *, const void *);
wafwerar's avatar
wafwerar 已提交
50
#endif
S
Shengliang Guan 已提交
51 52 53 54 55 56 57
  void  error (int, int, const char *);
  #ifndef PTHREAD_MUTEX_RECURSIVE_NP
    #define  PTHREAD_MUTEX_RECURSIVE_NP PTHREAD_MUTEX_RECURSIVE
  #endif
#endif


58
#if defined(WINDOWS)
S
Shengliang Guan 已提交
59
  char *stpcpy (char *dest, const char *src);
wafwerar's avatar
wafwerar 已提交
60
  char *stpncpy (char *dest, const char *src, int n);
S
Shengliang Guan 已提交
61 62

  // specific
wafwerar's avatar
wafwerar 已提交
63 64
#ifndef __COMPAR_FN_T
#define __COMPAR_FN_T
S
Shengliang Guan 已提交
65
  typedef int (*__compar_fn_t)(const void *, const void *);
wafwerar's avatar
wafwerar 已提交
66
#endif
S
Shengliang Guan 已提交
67
  #define ssize_t int
68
  #define _SSIZE_T_
S
Shengliang Guan 已提交
69 70 71 72 73
  #define bzero(ptr, size) memset((ptr), 0, (size))
  #define strcasecmp  _stricmp
  #define strncasecmp _strnicmp
  #define wcsncasecmp _wcsnicmp
  #define strtok_r strtok_s
wafwerar's avatar
wafwerar 已提交
74
  // #define snprintf _snprintf
S
Shengliang Guan 已提交
75
  #define in_addr_t unsigned long
76
//  #define socklen_t int
S
Shengliang Guan 已提交
77 78 79

  char *     strsep(char **stringp, const char *delim);
  char *     getpass(const char *prefix);
wafwerar's avatar
wafwerar 已提交
80
  char *     strndup(const char *s, int n);
S
Shengliang Guan 已提交
81

82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
  // for send function in tsocket.c
  #define MSG_NOSIGNAL             0
  #define SO_NO_CHECK              0x1234
  #define SOL_TCP                  0x1234

  #define SHUT_RDWR                SD_BOTH
  #define SHUT_RD                  SD_RECEIVE
  #define SHUT_WR                  SD_SEND

  #define LOCK_EX 1
  #define LOCK_NB 2
  #define LOCK_UN 3

  #ifndef PATH_MAX
  #define PATH_MAX 256
  #endif

  typedef struct {
    int    we_wordc;
    char  *we_wordv[1];
    int    we_offs;
    char   wordPos[1025];
  } wordexp_t;
  int  wordexp(char *words, wordexp_t *pwordexp, int flags);
  void wordfree(wordexp_t *pwordexp);

  #define openlog(a, b, c)
  #define closelog()
  #define LOG_ERR 0
  #define LOG_INFO 1
  void syslog(int unused, const char *format, ...);
#endif // WINDOWS
  
H
Hongze Cheng 已提交
115 116 117 118 119 120
#ifndef WINDOWS
  #ifndef O_BINARY
    #define O_BINARY 0
  #endif
#endif

S
Shengliang Guan 已提交
121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
#define POINTER_SHIFT(p, b) ((void *)((char *)(p) + (b)))
#define POINTER_DISTANCE(p1, p2) ((char *)(p1) - (char *)(p2)) 

#ifndef NDEBUG
#define ASSERT(x) assert(x)
#else
#define ASSERT(x)
#endif

#ifndef UNUSED
#define UNUSED(x) ((void)(x))
#endif

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

H
refact  
Hongze Cheng 已提交
155 156 157 158 159 160
#if defined(__GNUC__)
#define FORCE_INLINE inline __attribute__((always_inline))
#else
#define FORCE_INLINE
#endif

S
Shengliang Guan 已提交
161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201
#define DEFAULT_UNICODE_ENCODEC "UCS-4LE"

#define DEFAULT_COMP(x, y)       \
  do {                           \
    if ((x) == (y)) {            \
      return 0;                  \
    } else {                     \
      return (x) < (y) ? -1 : 1; \
    }                            \
  } while (0)

#define DEFAULT_DOUBLE_COMP(x, y)           \
  do {                                      \
    if (isnan(x) && isnan(y)) { return 0; } \
    if (isnan(x)) { return -1; }            \
    if (isnan(y)) { return 1; }             \
    if ((x) == (y)) {                       \
      return 0;                             \
    } else {                                \
      return (x) < (y) ? -1 : 1;            \
    }                                       \
  } while (0)

#define DEFAULT_FLOAT_COMP(x, y) DEFAULT_DOUBLE_COMP(x, y)

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

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

#undef threadlocal
#ifdef _ISOC11_SOURCE
  #define threadlocal _Thread_local
#elif defined(__APPLE__)
  #define threadlocal __thread
#elif defined(__GNUC__) && !defined(threadlocal)
  #define threadlocal __thread
#else
  #define threadlocal __declspec( thread )
#endif

wafwerar's avatar
wafwerar 已提交
202
#ifdef WINDOWS
S
Shengliang Guan 已提交
203 204 205 206
  #define PRIzu "ld"  
#else
  #define PRIzu "zu"  
#endif
207

208
#if !defined(WINDOWS)
209 210 211 212 213 214 215 216 217 218 219 220 221 222
  #if defined(_TD_DARWIN_64)
    // MacOS
    #if !defined(_GNU_SOURCE)
      #define setThreadName(name) do { pthread_setname_np((name)); } while (0)
    #else
      // pthread_setname_np not defined
      #define setThreadName(name)
    #endif
  #else
    // Linux, length of name must <= 16 (the last '\0' included)
    #define setThreadName(name) do { prctl(PR_SET_NAME, (name)); } while (0)
  #endif
#else
  // Windows
223
  #define setThreadName(name)
224 225
#endif

S
Shengliang Guan 已提交
226 227 228 229 230 231
#if defined(_WIN32)
#define TD_DIRSEP	"\\"
#else
#define TD_DIRSEP	"/"
#endif

S
os  
Shengliang Guan 已提交
232 233 234 235
#define TD_LOCALE_LEN   64
#define TD_CHARSET_LEN  64
#define TD_TIMEZONE_LEN 96

H
refact  
Hongze Cheng 已提交
236 237 238 239
#ifdef __cplusplus
}
#endif

240
#endif /*_TD_OS_DEF_H_*/