131 lines
2.9 KiB
C
131 lines
2.9 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/>.
|
|
*/
|
|
|
|
#define _DEFAULT_SOURCE
|
|
#include "os.h"
|
|
|
|
#ifdef WINDOWS
|
|
|
|
/*
|
|
* windows implementation
|
|
*/
|
|
|
|
#include <windows.h>
|
|
|
|
typedef void (*FWinSignalHandler)(int32_t signum);
|
|
|
|
int32_t taosSetSignal(int32_t signum, FSignalHandler sigfp) {
|
|
if (signum == SIGUSR1) return 0;
|
|
|
|
// SIGHUP doesn't exist in windows, we handle it in the way of ctrlhandler
|
|
if (signum == SIGHUP) {
|
|
if(SetConsoleCtrlHandler((PHANDLER_ROUTINE)sigfp, TRUE) == 0) {
|
|
terrno = TAOS_SYSTEM_WINAPI_ERROR(GetLastError());
|
|
return terrno;
|
|
}
|
|
} else {
|
|
if(signal(signum, (FWinSignalHandler)sigfp) == SIG_ERR) {
|
|
terrno = TAOS_SYSTEM_ERROR(errno);
|
|
return terrno;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int32_t taosIgnSignal(int32_t signum) {
|
|
if (signum == SIGUSR1 || signum == SIGHUP) return 0;
|
|
if(signal(signum, SIG_IGN) == SIG_ERR) {
|
|
terrno = TAOS_SYSTEM_ERROR(errno);
|
|
return terrno;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int32_t taosDflSignal(int32_t signum) {
|
|
if (signum == SIGUSR1 || signum == SIGHUP) return 0;
|
|
if(signal(signum, SIG_DFL) == SIG_ERR) {
|
|
terrno = TAOS_SYSTEM_ERROR(errno);
|
|
return terrno;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int32_t taosKillChildOnParentStopped() {
|
|
return 0;
|
|
}
|
|
|
|
#else
|
|
|
|
/*
|
|
* linux and darwin implementation
|
|
*/
|
|
|
|
typedef void (*FLinuxSignalHandler)(int32_t signum, siginfo_t *sigInfo, void *context);
|
|
|
|
int32_t taosSetSignal(int32_t signum, FSignalHandler sigfp) {
|
|
struct sigaction act;
|
|
(void)memset(&act, 0, sizeof(act));
|
|
#if 1
|
|
act.sa_flags = SA_SIGINFO | SA_RESTART;
|
|
act.sa_sigaction = (FLinuxSignalHandler)sigfp;
|
|
#else
|
|
act.sa_handler = sigfp;
|
|
#endif
|
|
int32_t code = sigaction(signum, &act, NULL);
|
|
if (-1 == code) {
|
|
terrno = TAOS_SYSTEM_ERROR(errno);
|
|
return terrno;
|
|
}
|
|
|
|
return code;
|
|
}
|
|
|
|
int32_t taosIgnSignal(int32_t signum) {
|
|
sighandler_t h = signal(signum, SIG_IGN);
|
|
if (SIG_ERR == h) {
|
|
terrno = TAOS_SYSTEM_ERROR(errno);
|
|
return terrno;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int32_t taosDflSignal(int32_t signum) {
|
|
sighandler_t h = signal(signum, SIG_DFL);
|
|
if (SIG_ERR == h) {
|
|
terrno = TAOS_SYSTEM_ERROR(errno);
|
|
return terrno;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int32_t taosKillChildOnParentStopped() {
|
|
#ifndef _TD_DARWIN_64
|
|
int32_t code = prctl(PR_SET_PDEATHSIG, SIGKILL);
|
|
if (-1 == code) {
|
|
terrno = TAOS_SYSTEM_ERROR(errno);
|
|
return terrno;
|
|
}
|
|
|
|
return code;
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
#endif
|