osSystem.h 5.6 KB
Newer Older
D
dapan1121 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

S
Shengliang Guan 已提交
16 17
#ifndef _TD_OS_SYSTEM_H_
#define _TD_OS_SYSTEM_H_
D
dapan1121 已提交
18 19 20 21 22

#ifdef __cplusplus
extern "C" {
#endif

wafwerar's avatar
wafwerar 已提交
23
// If the error is in a third-party library, place this header file under the third-party library header file.
wafwerar's avatar
wafwerar 已提交
24
// When you want to use this feature, you should find or add the same function in the following section.
wafwerar's avatar
wafwerar 已提交
25
#ifndef ALLOW_FORBID_FUNC
H
Hongze Cheng 已提交
26 27 28 29
#define popen     POPEN_FUNC_TAOS_FORBID
#define pclose    PCLOSE_FUNC_TAOS_FORBID
#define tcsetattr TCSETATTR_FUNC_TAOS_FORBID
#define tcgetattr TCGETATTR_FUNC_TAOS_FORBID
wafwerar's avatar
wafwerar 已提交
30 31
#endif

H
Hongze Cheng 已提交
32
typedef struct TdCmd* TdCmdPtr;
wafwerar's avatar
wafwerar 已提交
33

34
TdCmdPtr taosOpenCmd(const char* cmd);
H
Hongze Cheng 已提交
35
int64_t  taosGetsCmd(TdCmdPtr pCmd, int32_t maxSize, char* __restrict buf);
36 37 38
int64_t  taosGetLineCmd(TdCmdPtr pCmd, char** __restrict ptrBuf);
int32_t  taosEOFCmd(TdCmdPtr pCmd);
int64_t  taosCloseCmd(TdCmdPtr* ppCmd);
wafwerar's avatar
wafwerar 已提交
39

S
Shengliang Guan 已提交
40
void* taosLoadDll(const char* filename);
D
dapan1121 已提交
41
void* taosLoadSym(void* handle, char* name);
S
Shengliang Guan 已提交
42
void  taosCloseDll(void* handle);
D
dapan1121 已提交
43

S
Shengliang Guan 已提交
44
int32_t taosSetConsoleEcho(bool on);
45 46 47
void    taosSetTerminalMode();
int32_t taosGetOldTerminalMode();
void    taosResetTerminalMode();
48

S
Shengliang Guan 已提交
49
#if !defined(WINDOWS)
D
dapan1121 已提交
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
#define taosLogTraceToBuf(buf, bufSize, ignoreNum) {                                                                      \
  void*   array[100];                                                                                                     \
  int32_t size = backtrace(array, 100);                                                                                   \
  char**  strings = backtrace_symbols(array, size);                                                                       \
  int32_t offset = 0;                                                                                                     \
  if (strings != NULL) {                                                                                                  \
    offset = snprintf(buf, bufSize - 1, "obtained %d stack frames\n", (ignoreNum > 0) ? size - ignoreNum : size);           \
    for (int32_t i = (ignoreNum > 0) ? ignoreNum : 0; i < size; i++) {                                                    \
      offset += snprintf(buf + offset, bufSize - 1 - offset, "frame:%d, %s\n", (ignoreNum > 0) ? i - ignoreNum : i, strings[i]);     \
    }                                                                                                                     \
  }                                                                                                                       \
                                                                                                                          \
  taosMemoryFree(strings);                                                                                                \
}

#define taosPrintTrace(flags, level, dflag, ignoreNum)                                                          \
  {                                                                                                             \
    void*   array[100];                                                                                         \
    int32_t size = backtrace(array, 100);                                                                       \
    char**  strings = backtrace_symbols(array, size);                                                           \
    if (strings != NULL) {                                                                                      \
      taosPrintLog(flags, level, dflag, "obtained %d stack frames", (ignoreNum > 0) ? size - ignoreNum : size); \
      for (int32_t i = (ignoreNum > 0) ? ignoreNum : 0; i < size; i++) {                                        \
        taosPrintLog(flags, level, dflag, "frame:%d, %s", (ignoreNum > 0) ? i - ignoreNum : i, strings[i]);     \
      }                                                                                                         \
    }                                                                                                           \
                                                                                                                \
    taosMemoryFree(strings);                                                                                    \
S
Shengliang Guan 已提交
78 79
  }
#else
D
dapan1121 已提交
80 81 82 83 84 85 86 87 88 89
#define taosLogTraceToBuf(buf, bufSize, ignoreNum) {                                                            \
  snprintf(buf, bufSize - 1,                                                                                    \
               "backtrace not implemented on windows, so detailed stack information cannot be printed");        \
}

#define taosPrintTrace(flags, level, dflag, ignoreNum)                                                         \
  {                                                                                                            \
    taosPrintLog(flags, level, dflag,                                                                          \
                 "backtrace not implemented on windows, so detailed stack information cannot be printed");     \
  }
S
Shengliang Guan 已提交
90 91
#endif

D
dapan1121 已提交
92 93 94 95
#ifdef __cplusplus
}
#endif

S
Shengliang Guan 已提交
96
#endif /*_TD_OS_SYSTEM_H_*/