150 lines
11 KiB
C
150 lines
11 KiB
C
/*
|
|
* 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_SYSTEM_H_
|
|
#define _TD_OS_SYSTEM_H_
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
// If the error is in a third-party library, place this header file under the third-party library header file.
|
|
// When you want to use this feature, you should find or add the same function in the following section.
|
|
#ifndef ALLOW_FORBID_FUNC
|
|
#define popen POPEN_FUNC_TAOS_FORBID
|
|
#define pclose PCLOSE_FUNC_TAOS_FORBID
|
|
#define tcsetattr TCSETATTR_FUNC_TAOS_FORBID
|
|
#define tcgetattr TCGETATTR_FUNC_TAOS_FORBID
|
|
#endif
|
|
|
|
typedef struct TdCmd* TdCmdPtr;
|
|
|
|
TdCmdPtr taosOpenCmd(const char* cmd);
|
|
int64_t taosGetsCmd(TdCmdPtr pCmd, int32_t maxSize, char* __restrict buf);
|
|
int64_t taosGetLineCmd(TdCmdPtr pCmd, char** __restrict ptrBuf);
|
|
int32_t taosEOFCmd(TdCmdPtr pCmd);
|
|
int64_t taosCloseCmd(TdCmdPtr* ppCmd);
|
|
|
|
void* taosLoadDll(const char* filename);
|
|
void* taosLoadSym(void* handle, char* name);
|
|
void taosCloseDll(void* handle);
|
|
|
|
int32_t taosSetConsoleEcho(bool on);
|
|
void taosSetTerminalMode();
|
|
int32_t taosGetOldTerminalMode();
|
|
void taosResetTerminalMode();
|
|
|
|
#define STACKSIZE 100
|
|
|
|
#if !defined(WINDOWS)
|
|
#define taosLogTraceToBuf(buf, bufSize, ignoreNum) { \
|
|
void* array[STACKSIZE]; \
|
|
int32_t size = backtrace(array, STACKSIZE); \
|
|
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[STACKSIZE]; \
|
|
int32_t size = backtrace(array, STACKSIZE); \
|
|
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); \
|
|
}
|
|
#else
|
|
|
|
#include <windows.h>
|
|
#include <dbghelp.h>
|
|
|
|
#define taosLogTraceToBuf(buf, bufSize, ignoreNum) { \
|
|
unsigned int i; \
|
|
void* stack[STACKSIZE]; \
|
|
unsigned short frames; \
|
|
SYMBOL_INFO* symbol; \
|
|
HANDLE process; \
|
|
int32_t offset = 0; \
|
|
\
|
|
process = GetCurrentProcess(); \
|
|
\
|
|
SymInitialize(process, NULL, TRUE); \
|
|
\
|
|
frames = CaptureStackBackTrace(0, STACKSIZE, stack, NULL); \
|
|
symbol = (SYMBOL_INFO*)calloc(sizeof(SYMBOL_INFO) + 256 * sizeof(char), 1); \
|
|
if (symbol != NULL) { \
|
|
symbol->MaxNameLen = 255; \
|
|
symbol->SizeOfStruct = sizeof(SYMBOL_INFO); \
|
|
\
|
|
if (frames > 0) { \
|
|
offset = snprintf(buf, bufSize - 1, "obtained %d stack frames\n", (ignoreNum > 0) ? frames - ignoreNum : frames); \
|
|
for (i = (ignoreNum > 0) ? ignoreNum : 0; i < frames; i++) { \
|
|
SymFromAddr(process, (DWORD64)(stack[i]), 0, symbol); \
|
|
offset += snprintf(buf + offset, bufSize - 1 - offset, "frame:%i, %s - 0x%0X\n", (ignoreNum > 0) ? i - ignoreNum : i, symbol->Name, symbol->Address); \
|
|
} \
|
|
} \
|
|
free(symbol); \
|
|
} \
|
|
}
|
|
|
|
#define taosPrintTrace(flags, level, dflag, ignoreNum) \
|
|
{ \
|
|
unsigned int i; \
|
|
void* stack[STACKSIZE]; \
|
|
unsigned short frames; \
|
|
SYMBOL_INFO* symbol; \
|
|
HANDLE process; \
|
|
\
|
|
process = GetCurrentProcess(); \
|
|
\
|
|
SymInitialize(process, NULL, TRUE); \
|
|
\
|
|
frames = CaptureStackBackTrace(0, STACKSIZE, stack, NULL); \
|
|
symbol = (SYMBOL_INFO*)calloc(sizeof(SYMBOL_INFO) + 256 * sizeof(char), 1); \
|
|
if (symbol != NULL) { \
|
|
symbol->MaxNameLen = 255; \
|
|
symbol->SizeOfStruct = sizeof(SYMBOL_INFO); \
|
|
\
|
|
if (frames > 0) { \
|
|
taosPrintLog(flags, level, dflag, "obtained %d stack frames\n", (ignoreNum > 0) ? frames - ignoreNum : frames); \
|
|
for (i = (ignoreNum > 0) ? ignoreNum : 0; i < frames; i++) { \
|
|
SymFromAddr(process, (DWORD64)(stack[i]), 0, symbol); \
|
|
taosPrintLog(flags, level, dflag, "frame:%i, %s - 0x%0X\n", (ignoreNum > 0) ? i - ignoreNum : i, symbol->Name, symbol->Address); \
|
|
} \
|
|
} \
|
|
free(symbol); \
|
|
} \
|
|
}
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /*_TD_OS_SYSTEM_H_*/
|