]>
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.h" | |
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)) != | |
70 | sizeof(int)) { | |
71 | qDebug() << "Failed to catch signal"; | |
72 | abort(); | |
73 | } | |
74 | ||
75 | switch(sig_number) | |
76 | { | |
77 | case SIGINT: | |
78 | emit int_received(); | |
79 | break; | |
80 | case SIGTERM: | |
81 | emit term_received(); | |
82 | break; | |
83 | } | |
84 | ||
85 | _socket_notifier->setEnabled(true); | |
86 | } | |
87 | ||
88 | void SignalHandler::handle_signals(int sig_number) | |
89 | { | |
90 | if(write(_sockets[0], &sig_number, sizeof(int)) != | |
91 | sizeof(int)) { | |
92 | // Failed to handle signal | |
93 | abort(); | |
94 | } | |
95 | } |