]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | * This file is part of the PulseView project. | |
3 | * | |
4 | * Copyright (C) 2013 Adam Reichold | |
5 | * | |
6 | * This program is free software; you can redistribute it and/or modify | |
7 | * it under the terms of the GNU General Public License as published by | |
8 | * the Free Software Foundation; either version 2 of the License, or | |
9 | * (at your option) any later version. | |
10 | * | |
11 | * This program is distributed in the hope that it will be useful, | |
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
14 | * GNU General Public License for more details. | |
15 | * | |
16 | * You should have received a copy of the GNU General Public License | |
17 | * along with this program; if not, see <http://www.gnu.org/licenses/>. | |
18 | */ | |
19 | ||
20 | #include "signalhandler.hpp" | |
21 | ||
22 | #include <assert.h> | |
23 | #include <signal.h> | |
24 | #include <stdlib.h> | |
25 | #include <sys/socket.h> | |
26 | #include <unistd.h> | |
27 | ||
28 | #include <QDebug> | |
29 | #include <QSocketNotifier> | |
30 | ||
31 | int SignalHandler::sockets_[2]; | |
32 | ||
33 | bool SignalHandler::prepare_signals() | |
34 | { | |
35 | if (socketpair(AF_UNIX, SOCK_STREAM, 0, sockets_) != 0) | |
36 | return false; | |
37 | ||
38 | struct sigaction sig_action; | |
39 | ||
40 | sig_action.sa_handler = SignalHandler::handle_signals; | |
41 | sigemptyset(&sig_action.sa_mask); | |
42 | sig_action.sa_flags = SA_RESTART; | |
43 | ||
44 | if (sigaction(SIGINT, &sig_action, nullptr) != 0 || | |
45 | sigaction(SIGTERM, &sig_action, nullptr) != 0) { | |
46 | close(sockets_[0]); | |
47 | close(sockets_[1]); | |
48 | return false; | |
49 | } | |
50 | ||
51 | return true; | |
52 | } | |
53 | ||
54 | SignalHandler::SignalHandler(QObject* parent) : QObject(parent), | |
55 | socket_notifier_(nullptr) | |
56 | { | |
57 | socket_notifier_ = new QSocketNotifier(sockets_[1], | |
58 | QSocketNotifier::Read, this); | |
59 | connect(socket_notifier_, SIGNAL(activated(int)), | |
60 | SLOT(on_socket_notifier_activated())); | |
61 | } | |
62 | ||
63 | void SignalHandler::on_socket_notifier_activated() | |
64 | { | |
65 | socket_notifier_->setEnabled(false); | |
66 | ||
67 | int sig_number; | |
68 | if (read(sockets_[1], &sig_number, sizeof(int)) != sizeof(int)) { | |
69 | qDebug() << "Failed to catch signal"; | |
70 | abort(); | |
71 | } | |
72 | ||
73 | switch(sig_number) | |
74 | { | |
75 | case SIGINT: | |
76 | Q_EMIT int_received(); | |
77 | break; | |
78 | case SIGTERM: | |
79 | Q_EMIT term_received(); | |
80 | break; | |
81 | } | |
82 | ||
83 | socket_notifier_->setEnabled(true); | |
84 | } | |
85 | ||
86 | void SignalHandler::handle_signals(int sig_number) | |
87 | { | |
88 | if (write(sockets_[0], &sig_number, sizeof(int)) != sizeof(int)) { | |
89 | // Failed to handle signal | |
90 | abort(); | |
91 | } | |
92 | } |