Fix various clang warnings.
[sigrok-cli.git] / anykey.c
1 /*
2  * This file is part of the sigrok-cli project.
3  *
4  * Copyright (C) 2011 Bert Vermeulen <bert@biot.com>
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 3 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 <config.h>
21 #include <stdio.h>
22 #ifdef _WIN32
23 #include <windows.h>
24 #else
25 #include <termios.h>
26 #include <unistd.h>
27 #endif
28 #include <string.h>
29 #include <glib.h>
30 #include "sigrok-cli.h"
31
32 #ifdef _WIN32
33 static HANDLE stdin_handle;
34 static DWORD stdin_mode;
35 #else
36 static struct termios term_orig;
37 #endif
38 static unsigned int watch_id = 0;
39
40 static gboolean received_anykey(GIOChannel *source,
41                 GIOCondition condition, void *data)
42 {
43         struct sr_session *session;
44
45         (void)source;
46         (void)condition;
47         session = data;
48
49         watch_id = 0;
50         sr_session_stop(session);
51
52         return G_SOURCE_REMOVE;
53 }
54
55 /* Turn off buffering on stdin and watch for input.
56  */
57 void add_anykey(struct sr_session *session)
58 {
59         GIOChannel *channel;
60
61 #ifdef _WIN32
62         stdin_handle = GetStdHandle(STD_INPUT_HANDLE);
63
64         if (!GetConsoleMode(stdin_handle, &stdin_mode)) {
65                 /* TODO: Error handling. */
66         }
67         SetConsoleMode(stdin_handle, 0);
68
69         channel = g_io_channel_win32_new_fd(0);
70 #else
71         struct termios term;
72
73         tcgetattr(STDIN_FILENO, &term);
74         memcpy(&term_orig, &term, sizeof(struct termios));
75         term.c_lflag &= ~(ECHO | ICANON | ISIG);
76         tcsetattr(STDIN_FILENO, TCSADRAIN, &term);
77
78         channel = g_io_channel_unix_new(STDIN_FILENO);
79 #endif
80         g_io_channel_set_encoding(channel, NULL, NULL);
81         g_io_channel_set_buffered(channel, FALSE);
82
83         watch_id = g_io_add_watch(channel, G_IO_IN, &received_anykey, session);
84         g_io_channel_unref(channel);
85
86         g_message("Press any key to stop acquisition.");
87 }
88
89 /* Remove the event watch and restore stdin attributes.
90  */
91 void clear_anykey(void)
92 {
93         if (watch_id != 0) {
94                 g_source_remove(watch_id);
95                 watch_id = 0;
96         }
97 #ifdef _WIN32
98         SetConsoleMode(stdin_handle, stdin_mode);
99 #else
100         tcflush(STDIN_FILENO, TCIFLUSH);
101         tcsetattr(STDIN_FILENO, TCSANOW, &term_orig);
102 #endif
103 }