]>
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, write to the Free Software | |
18 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA | |
19 | */ | |
20 | ||
21 | #include "signalhandler.hpp" | |
22 | ||
23 | #include <assert.h> | |
24 | #include <signal.h> | |
25 | #include <stdlib.h> | |
26 | #include <sys/socket.h> | |
27 | #include <unistd.h> | |
28 | ||
29 | #include <QDebug> | |
30 | #include <QSocketNotifier> | |
31 | ||
32 | int SignalHandler::sockets_[2]; | |
33 | ||
34 | bool SignalHandler::prepare_signals() | |
35 | { | |
36 | if (socketpair(AF_UNIX, SOCK_STREAM, 0, sockets_) != 0) | |
37 | return false; | |
38 | ||
39 | struct sigaction sig_action; | |
40 | ||
41 | sig_action.sa_handler = SignalHandler::handle_signals; | |
42 | sigemptyset(&sig_action.sa_mask); | |
43 | sig_action.sa_flags = SA_RESTART; | |
44 | ||
45 | if (sigaction(SIGINT, &sig_action, 0) != 0 || | |
46 | sigaction(SIGTERM, &sig_action, 0) != 0) { | |
47 | close(sockets_[0]); | |
48 | close(sockets_[1]); | |
49 | return false; | |
50 | } | |
51 | ||
52 | return true; | |
53 | } | |
54 | ||
55 | SignalHandler::SignalHandler(QObject* parent) : QObject(parent), | |
56 | socket_notifier_(0) | |
57 | { | |
58 | socket_notifier_ = new QSocketNotifier(sockets_[1], | |
59 | QSocketNotifier::Read, this); | |
60 | connect(socket_notifier_, SIGNAL(activated(int)), | |
61 | SLOT(on_socket_notifier_activated())); | |
62 | } | |
63 | ||
64 | void SignalHandler::on_socket_notifier_activated() | |
65 | { | |
66 | socket_notifier_->setEnabled(false); | |
67 | ||
68 | int sig_number; | |
69 | if (read(sockets_[1], &sig_number, sizeof(int)) != sizeof(int)) { | |
70 | qDebug() << "Failed to catch signal"; | |
71 | abort(); | |
72 | } | |
73 | ||
74 | switch(sig_number) | |
75 | { | |
76 | case SIGINT: | |
77 | Q_EMIT int_received(); | |
78 | break; | |
79 | case SIGTERM: | |
80 | Q_EMIT term_received(); | |
81 | break; | |
82 | } | |
83 | ||
84 | socket_notifier_->setEnabled(true); | |
85 | } | |
86 | ||
87 | void SignalHandler::handle_signals(int sig_number) | |
88 | { | |
89 | if (write(sockets_[0], &sig_number, sizeof(int)) != sizeof(int)) { | |
90 | // Failed to handle signal | |
91 | abort(); | |
92 | } | |
93 | } |