a0b054af337aa2c83d284b17ab707f6c59a0e2f2
[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 "sigrok-cli.h"
21 #include "config.h"
22 #include <sys/types.h>
23 #include <sys/stat.h>
24 #include <fcntl.h>
25 #include <unistd.h>
26 #include <errno.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <glib.h>
30
31 #define BUFSIZE (16 * 1024)
32
33 static void load_input_file_module(void)
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
47         if (!sr_input_list())
48                 g_critical("No supported input formats available.");
49
50         mod_id = NULL;
51         mod_args = NULL;
52         if (opt_input_format) {
53                 mod_args = parse_generic_arg(opt_input_format, TRUE);
54                 mod_id = g_hash_table_lookup(mod_args, "sigrok_key");
55         }
56
57         fd = 0;
58         buf = g_string_sized_new(BUFSIZE);
59         if (mod_id) {
60                 /* User specified an input module to use. */
61                 if (!(imod = sr_input_find(mod_id)))
62                         g_critical("Error: unknown input module '%s'.", mod_id);
63                 g_hash_table_remove(mod_args, "sigrok_key");
64                 if ((options = sr_input_options_get(imod))) {
65                         mod_opts = generic_arg_to_opt(options, mod_args);
66                         sr_output_options_free(options);
67                 } else
68                         mod_opts = NULL;
69                 if (!(in = sr_input_new(imod, mod_opts)))
70                         g_critical("Error: failed to initialize input module.");
71                 if (mod_opts)
72                         g_hash_table_destroy(mod_opts);
73                 if (mod_args)
74                         g_hash_table_destroy(mod_args);
75                 if ((fd = open(opt_input_file, O_RDONLY)) == -1)
76                         g_critical("Failed to load %s: %s.", opt_input_file,
77                                         strerror(errno));
78         } else {
79                 if (strcmp(opt_input_file, "-")) {
80                         /*
81                          * An actual filename: let the input modules try to
82                          * identify the file.
83                          */
84                         if (sr_input_scan_file(opt_input_file, &in) == SR_OK) {
85                                 /* That worked, reopen the file for reading. */
86                                 fd = open(opt_input_file, O_RDONLY);
87                         }
88                 } else {
89                         /*
90                          * Taking input from a pipe: let the input modules try
91                          * to identify the stream content.
92                          */
93                         if (!strcmp(opt_input_file, "-")) {
94                                 /* stdin */
95                                 fd = 0;
96                         } else {
97                                 if ((fd = open(opt_input_file, O_RDONLY)) == -1)
98                                         g_critical("Failed to load %s: %s.", opt_input_file,
99                                                         strerror(errno));
100                         }
101                         if ((len = read(fd, buf->str, BUFSIZE)) < 1)
102                                 g_critical("Failed to read %s: %s.", opt_input_file,
103                                                 strerror(errno));
104                         buf->len = len;
105                         sr_input_scan_buffer(buf, &in);
106                 }
107                 if (!in)
108                         g_critical("Error: no input module found for this file.");
109         }
110         sr_session_new(sr_ctx, &session);
111         sr_session_datafeed_callback_add(session, &datafeed_in, NULL);
112
113         got_sdi = FALSE;
114         while (TRUE) {
115                 g_string_truncate(buf, 0);
116                 len = read(fd, buf->str, BUFSIZE);
117                 if (len < 0)
118                         g_critical("Read failed: %s", strerror(errno));
119                 if (len == 0)
120                         /* End of file or stream. */
121                         break;
122                 buf->len = len;
123                 if (sr_input_send(in, buf) != SR_OK)
124                         break;
125
126                 sdi = sr_input_dev_inst_get(in);
127                 if (!got_sdi && sdi) {
128                         /* First time we got a valid sdi. */
129                         if (select_channels(sdi) != SR_OK)
130                                 return;
131                         if (sr_session_dev_add(session, sdi) != SR_OK) {
132                                 g_critical("Failed to use device.");
133                                 sr_session_destroy(session);
134                                 return;
135                         }
136                         got_sdi = TRUE;
137                 }
138         }
139         sr_input_end(in);
140         sr_input_free(in);
141         g_string_free(buf, TRUE);
142
143         sr_session_destroy(session);
144
145 }
146
147 void load_input_file(void)
148 {
149         struct sr_session *session;
150         struct sr_dev_inst *sdi;
151         GSList *devices;
152         int ret;
153
154         if (!strcmp(opt_input_file, "-")) {
155                 /* Input from stdin is never a session file. */
156                 load_input_file_module();
157         } else {
158                 if ((ret = sr_session_load(sr_ctx, opt_input_file,
159                                 &session)) == SR_OK) {
160                         /* sigrok session file */
161                         ret = sr_session_dev_list(session, &devices);
162                         if (ret != SR_OK || !devices || !devices->data) {
163                                 g_critical("Failed to access session device.");
164                                 sr_session_destroy(session);
165                                 return;
166                         }
167                         sdi = devices->data;
168                         if (select_channels(sdi) != SR_OK) {
169                                 sr_session_destroy(session);
170                                 return;
171                         }
172                         sr_session_datafeed_callback_add(session, datafeed_in, NULL);
173                         sr_session_start(session);
174                         sr_session_run(session);
175                         sr_session_stop(session);
176                 } else if (ret != SR_ERR) {
177                         /* It's a session file, but it didn't work out somehow. */
178                         g_critical("Failed to load session file.");
179                 } else {
180                         /* Fall back on input modules. */
181                         load_input_file_module();
182                 }
183         }
184 }