tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix signal handler reset on Windows.

Browse Source 
GitOrigin-RevId: 5ca829e660728b054ee6413ac4899e109827ea81
This commit is contained in:
levlam 2020-07-22 23:18:54 +03:00
parent fcd09d5c25
commit f8fd7e410e
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tdutils/td/utils/port/signals.cpp
View File

@ -113,10 +113,23 @@ static vector<int> get_native_signals(SignalType type) {
return {};
}
}
#endif
#if TD_PORT_WINDOWS
#elif TD_PORT_WINDOWS
using signal_handler = void (*)(int sig);
static signal_handler signal_handlers[NSIG] = {};
static void signal_handler_func(int sig) {
std::signal(sig, signal_handler_func);
auto handler = signal_handlers[sig];
handler(sig);
}
static Status set_signal_handler_impl(vector<int> signals, void (*func)(int sig)) {
for (auto signal : signals) {
CHECK(0 <= signal && signal < NSIG);
if (func != SIG_IGN && func != SIG_DFL) {
signal_handlers[signal] = func;
func = signal_handler_func;
}
if (std::signal(signal, func) == SIG_ERR) {
return Status::Error("Failed to set signal handler");
}
@ -146,7 +159,7 @@ static vector<int> get_native_signals(SignalType type) {
}
#endif
Status set_signal_handler(SignalType type, void (*func)(int)) {
Status set_signal_handler(SignalType type, void (*func)(int sig)) {
return set_signal_handler_impl(get_native_signals(type), func == nullptr ? SIG_DFL : func);
}