2 * This file is part of the sigrok-cli project.
4 * Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
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.
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.
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/>.
21 #include <sys/types.h>
29 #include "sigrok-cli.h"
31 #define CHUNK_SIZE (4 * 1024 * 1024)
33 static void load_input_file_module(struct df_arg_desc *df_arg)
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;
49 g_critical("No supported input formats available.");
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");
58 is_stdin = strcmp(opt_input_file, "-") == 0;
60 buf = g_string_sized_new(CHUNK_SIZE);
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);
71 if (!(in = sr_input_new(imod, mod_opts)))
72 g_critical("Error: failed to initialize input module.");
74 g_hash_table_destroy(mod_opts);
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,
83 * An actual filename: let the input modules try to
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);
92 * Taking input from a pipe: let the input modules try
93 * to identify the stream content.
99 fd = open(opt_input_file, O_RDONLY);
101 g_critical("Failed to load %s: %s.", opt_input_file,
104 if ((len = read(fd, buf->str, CHUNK_SIZE)) < 1)
105 g_critical("Failed to read %s: %s.", opt_input_file,
108 sr_input_scan_buffer(buf, &in);
111 g_critical("Error: no input module found for this file.");
113 sr_session_new(sr_ctx, &session);
114 df_arg->session = session;
115 sr_session_datafeed_callback_add(session, datafeed_in, df_arg);
119 g_string_truncate(buf, 0);
120 len = read(fd, buf->str, CHUNK_SIZE);
122 g_critical("Read failed: %s", g_strerror(errno));
124 /* End of file or stream. */
127 if (sr_input_send(in, buf) != SR_OK)
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)
135 if (sr_session_dev_add(session, sdi) != SR_OK) {
136 g_critical("Failed to use device.");
137 sr_session_destroy(session);
145 g_string_free(buf, TRUE);
147 df_arg->session = NULL;
148 sr_session_destroy(session);
152 void load_input_file(gboolean do_props)
154 struct df_arg_desc df_arg;
155 struct sr_session *session;
156 struct sr_dev_inst *sdi;
158 GMainLoop *main_loop;
161 memset(&df_arg, 0, sizeof(df_arg));
162 df_arg.do_props = do_props;
164 if (!strcmp(opt_input_file, "-")) {
165 /* Input from stdin is never a session file. */
166 load_input_file_module(&df_arg);
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);
179 g_slist_free(devices);
180 if (select_channels(sdi) != SR_OK) {
181 sr_session_destroy(session);
184 main_loop = g_main_loop_new(NULL, FALSE);
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,
192 if (sr_session_start(session) == SR_OK)
193 g_main_loop_run(main_loop);
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.");
202 /* Fall back on input modules. */
203 load_input_file_module(&df_arg);