andreacavalli/tdlib-fork
Archived
1
0
Fork 0
Code Pull Requests Releases Activity

Better compile-time dispatching on function type.

Browse Source 
GitOrigin-RevId: 2687fa402c9c76ad654bdad0759893dc6df6262e
This commit is contained in:
levlam 2018-12-12 05:26:52 +03:00
parent d0f8450726
commit 3791b38296
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -63,23 +63,24 @@ Status setup_signals_alt_stack() {
} }
#if TD_PORT_POSIX #if TD_PORT_POSIX
static void set_handler(struct sigaction &act, decltype(act.sa_handler) handler) {
act.sa_handler = handler;
}
static void set_handler(struct sigaction &act, decltype(act.sa_sigaction) handler) {
act.sa_sigaction = handler;
act.sa_flags |= SA_SIGINFO;
}
template <class F> template <class F>
static Status set_signal_handler_impl(vector<int> signals, F func, bool is_extended = false) { static Status set_signal_handler_impl(vector<int> signals, F func) {
struct sigaction act; struct sigaction act;
std::memset(&act, '\0', sizeof(act)); std::memset(&act, '\0', sizeof(act));
if (is_extended) { // TODO if constexpr, remove useless reinterpret_cast
act.sa_handler = reinterpret_cast<decltype(act.sa_handler)>(func);
} else {
act.sa_sigaction = reinterpret_cast<decltype(act.sa_sigaction)>(func);
}
sigemptyset(&act.sa_mask); sigemptyset(&act.sa_mask);
for (auto signal : signals) { for (auto signal : signals) {
sigaddset(&act.sa_mask, signal); sigaddset(&act.sa_mask, signal);
} }
act.sa_flags = SA_RESTART | SA_ONSTACK; act.sa_flags = SA_RESTART | SA_ONSTACK;
if (is_extended) { set_handler(act, func);
act.sa_flags |= SA_SIGINFO;
}
for (auto signal : signals) { for (auto signal : signals) {
if (sigaction(signal, &act, nullptr) != 0) { if (sigaction(signal, &act, nullptr) != 0) {
@ -111,7 +112,7 @@ static vector<int> get_native_signals(SignalType type) {
} }
#endif #endif
#if TD_PORT_WINDOWS #if TD_PORT_WINDOWS
static Status set_signal_handler_impl(vector<int> signals, void (*func)(int sig), bool /*unused*/ = true) { static Status set_signal_handler_impl(vector<int> signals, void (*func)(int sig)) {
for (auto signal : signals) { for (auto signal : signals) {
if (std::signal(signal, func) == SIG_ERR) { if (std::signal(signal, func) == SIG_ERR) {
return Status::Error("Failed to set signal handler"); return Status::Error("Failed to set signal handler");
@ -171,7 +172,7 @@ Status set_extended_signal_handler(SignalType type, extended_signal_handler func
UNREACHABLE(); UNREACHABLE();
} }
} }
return set_signal_handler_impl(std::move(signals), siginfo_handler, true); return set_signal_handler_impl(std::move(signals), siginfo_handler);
} }
Status set_runtime_signal_handler(int runtime_signal_number, void (*func)(int)) { Status set_runtime_signal_handler(int runtime_signal_number, void (*func)(int)) {