osDef.h 5.6 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 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
#if defined(_TD_DARWIN_64)
  // specific
  typedef int(*__compar_fn_t)(const void *, const void *);

  // 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)
  typedef int(*__compar_fn_t)(const void *, const void *);
  void  error (int, int, const char *);
  #ifndef PTHREAD_MUTEX_RECURSIVE_NP
    #define  PTHREAD_MUTEX_RECURSIVE_NP PTHREAD_MUTEX_RECURSIVE
  #endif
#endif


52
#if defined(WINDOWS)
S
Shengliang Guan 已提交
53 54 55 56 57 58
  char *stpcpy (char *dest, const char *src);
  char *stpncpy (char *dest, const char *src, size_t n);

  // specific
  typedef int (*__compar_fn_t)(const void *, const void *);
  #define ssize_t int
59
  #define _SSIZE_T_
S
Shengliang Guan 已提交
60 61 62 63 64 65 66
  #define bzero(ptr, size) memset((ptr), 0, (size))
  #define strcasecmp  _stricmp
  #define strncasecmp _strnicmp
  #define wcsncasecmp _wcsnicmp
  #define strtok_r strtok_s
  #define snprintf _snprintf
  #define in_addr_t unsigned long
67
//  #define socklen_t int
S
Shengliang Guan 已提交
68 69 70 71

  char *     strsep(char **stringp, const char *delim);
  char *     getpass(const char *prefix);
  char *     strndup(const char *s, size_t n);
72
  int        gettimeofday(struct timeval *ptv, void *pTimeZone);
S
Shengliang Guan 已提交
73

74 75 76 77 78 79 80 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
  // 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 已提交
107 108 109 110 111 112
#ifndef WINDOWS
  #ifndef O_BINARY
    #define O_BINARY 0
  #endif
#endif

S
Shengliang Guan 已提交
113 114 115 116 117 118 119 120 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
#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 已提交
147 148 149 150 151 152
#if defined(__GNUC__)
#define FORCE_INLINE inline __attribute__((always_inline))
#else
#define FORCE_INLINE
#endif

S
Shengliang Guan 已提交
153 154 155 156 157 158 159 160 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
#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 已提交
194
#ifdef WINDOWS
S
Shengliang Guan 已提交
195 196 197 198
  #define PRIzu "ld"  
#else
  #define PRIzu "zu"  
#endif
199

200
#if !defined(WINDOWS)
201 202 203 204 205 206 207 208 209 210 211 212 213 214
  #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
215
  #define setThreadName(name)
216 217
#endif

S
Shengliang Guan 已提交
218 219 220 221 222 223
#if defined(_WIN32)
#define TD_DIRSEP	"\\"
#else
#define TD_DIRSEP	"/"
#endif

S
os  
Shengliang Guan 已提交
224 225 226 227
#define TD_LOCALE_LEN   64
#define TD_CHARSET_LEN  64
#define TD_TIMEZONE_LEN 96

H
refact  
Hongze Cheng 已提交
228 229 230 231
#ifdef __cplusplus
}
#endif

232
#endif /*_TD_OS_DEF_H_*/