show: add support for -i <fn> --show, provide details on input stream
[sigrok-cli.git] / input.c
1 /*
2  * This file is part of the sigrok-cli project.
3  *
4  * Copyright (C) 2013 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 <sys/types.h>
22 #include <sys/stat.h>
23 #include <fcntl.h>
24 #include <unistd.h>
25 #include <errno.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <glib.h>
29 #include "sigrok-cli.h"
30
31 #define CHUNK_SIZE (4 * 1024 * 1024)
32
33 static void load_input_file_module(struct df_arg_desc *df_arg)
34 {
35         struct sr_session *session;
36         const struct sr_input *in;
37         const struct sr_input_module *imod;
38         const struct sr_option **options;
39         struct sr_dev_inst *sdi;
40         GHashTable *mod_args, *mod_opts;
41         GString *buf;
42         gboolean got_sdi;
43         int fd;
44         ssize_t len;
45         char *mod_id;
46         gboolean is_stdin;
47
48         if (!sr_input_list())
49                 g_critical("No supported input formats available.");
50
51         mod_id = NULL;
52         mod_args = NULL;
53         if (opt_input_format) {
54                 mod_args = parse_generic_arg(opt_input_format, TRUE);
55                 mod_id = g_hash_table_lookup(mod_args, "sigrok_key");
56         }
57
58         is_stdin = strcmp(opt_input_file, "-") == 0;
59         fd = 0;
60         buf = g_string_sized_new(CHUNK_SIZE);
61         if (mod_id) {
62                 /* User specified an input module to use. */
63                 if (!(imod = sr_input_find(mod_id)))
64                         g_critical("Error: unknown input module '%s'.", mod_id);
65                 g_hash_table_remove(mod_args, "sigrok_key");
66                 if ((options = sr_input_options_get(imod))) {
67                         mod_opts = generic_arg_to_opt(options, mod_args);
68                         sr_output_options_free(options);
69                 } else
70                         mod_opts = NULL;
71                 if (!(in = sr_input_new(imod, mod_opts)))
72                         g_critical("Error: failed to initialize input module.");
73                 if (mod_opts)
74                         g_hash_table_destroy(mod_opts);
75                 if (mod_args)
76                         g_hash_table_destroy(mod_args);
77                 if (!is_stdin && (fd = open(opt_input_file, O_RDONLY)) < 0)
78                         g_critical("Failed to load %s: %s.", opt_input_file,
79                                         g_strerror(errno));
80         } else {
81                 if (!is_stdin) {
82                         /*
83                          * An actual filename: let the input modules try to
84                          * identify the file.
85                          */
86                         if (sr_input_scan_file(opt_input_file, &in) == SR_OK) {
87                                 /* That worked, reopen the file for reading. */
88                                 fd = open(opt_input_file, O_RDONLY);
89                         }
90                 } else {
91                         /*
92                          * Taking input from a pipe: let the input modules try
93                          * to identify the stream content.
94                          */
95                         if (is_stdin) {
96                                 /* stdin */
97                                 fd = 0;
98                         } else {
99                                 fd = open(opt_input_file, O_RDONLY);
100                                 if (fd == -1)
101                                         g_critical("Failed to load %s: %s.", opt_input_file,
102                                                         g_strerror(errno));
103                         }
104                         if ((len = read(fd, buf->str, CHUNK_SIZE)) < 1)
105                                 g_critical("Failed to read %s: %s.", opt_input_file,
106                                                 g_strerror(errno));
107                         buf->len = len;
108                         sr_input_scan_buffer(buf, &in);
109                 }
110                 if (!in)
111                         g_critical("Error: no input module found for this file.");
112         }
113         sr_session_new(sr_ctx, &session);
114         df_arg->session = session;
115         sr_session_datafeed_callback_add(session, datafeed_in, df_arg);
116
117         got_sdi = FALSE;
118         while (TRUE) {
119                 g_string_truncate(buf, 0);
120                 len = read(fd, buf->str, CHUNK_SIZE);
121                 if (len < 0)
122                         g_critical("Read failed: %s", g_strerror(errno));
123                 if (len == 0)
124                         /* End of file or stream. */
125                         break;
126                 buf->len = len;
127                 if (sr_input_send(in, buf) != SR_OK)
128                         break;
129
130                 sdi = sr_input_dev_inst_get(in);
131                 if (!got_sdi && sdi) {
132                         /* First time we got a valid sdi. */
133                         if (select_channels(sdi) != SR_OK)
134                                 return;
135                         if (sr_session_dev_add(session, sdi) != SR_OK) {
136                                 g_critical("Failed to use device.");
137                                 sr_session_destroy(session);
138                                 return;
139                         }
140                         got_sdi = TRUE;
141                 }
142         }
143         sr_input_end(in);
144         sr_input_free(in);
145         g_string_free(buf, TRUE);
146
147         df_arg->session = NULL;
148         sr_session_destroy(session);
149
150 }
151
152 void load_input_file(gboolean do_props)
153 {
154         struct df_arg_desc df_arg;
155         struct sr_session *session;
156         struct sr_dev_inst *sdi;
157         GSList *devices;
158         GMainLoop *main_loop;
159         int ret;
160
161         memset(&df_arg, 0, sizeof(df_arg));
162         df_arg.do_props = do_props;
163
164         if (!strcmp(opt_input_file, "-")) {
165                 /* Input from stdin is never a session file. */
166                 load_input_file_module(&df_arg);
167         } else {
168                 if ((ret = sr_session_load(sr_ctx, opt_input_file,
169                                 &session)) == SR_OK) {
170                         /* sigrok session file */
171                         ret = sr_session_dev_list(session, &devices);
172                         if (ret != SR_OK || !devices || !devices->data) {
173                                 g_critical("Failed to access session device.");
174                                 g_slist_free(devices);
175                                 sr_session_destroy(session);
176                                 return;
177                         }
178                         sdi = devices->data;
179                         g_slist_free(devices);
180                         if (select_channels(sdi) != SR_OK) {
181                                 sr_session_destroy(session);
182                                 return;
183                         }
184                         main_loop = g_main_loop_new(NULL, FALSE);
185
186                         df_arg.session = session;
187                         sr_session_datafeed_callback_add(session,
188                                 datafeed_in, &df_arg);
189                         sr_session_stopped_callback_set(session,
190                                 (sr_session_stopped_callback)g_main_loop_quit,
191                                 main_loop);
192                         if (sr_session_start(session) == SR_OK)
193                                 g_main_loop_run(main_loop);
194
195                         g_main_loop_unref(main_loop);
196                         df_arg.session = NULL;
197                         sr_session_destroy(session);
198                 } else if (ret != SR_ERR) {
199                         /* It's a session file, but it didn't work out somehow. */
200                         g_critical("Failed to load session file.");
201                 } else {
202                         /* Fall back on input modules. */
203                         load_input_file_module(&df_arg);
204                 }
205         }
206 }