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 16384
  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         int fd;
  43         ssize_t len;
  44         char *mod_id;
  45
  46         if (!sr_input_list())
  47                 g_critical("No supported input formats available.");
  48
  49         mod_id = NULL;
  50         mod_args = NULL;
  51         if (opt_input_format) {
  52                 mod_args = parse_generic_arg(opt_input_format, TRUE);
  53                 mod_id = g_hash_table_lookup(mod_args, "sigrok_key");
  54         }
  55
  56         buf = g_string_sized_new(BUFSIZE);
  57         if (mod_id) {
  58                 /* User specified an input module to use. */
  59                 if (!(imod = sr_input_find(mod_id)))
  60                         g_critical("Error: unknown input module '%s'.", mod_id);
  61                 g_hash_table_remove(mod_args, "sigrok_key");
  62                 if ((options = sr_input_options_get(imod))) {
  63                         mod_opts = generic_arg_to_opt(options, mod_args);
  64                         sr_output_options_free(options);
  65                 } else
  66                         mod_opts = NULL;
  67                 if (!(in = sr_input_new(imod, mod_opts)))
  68                         g_critical("Error: failed to initialize input module.");
  69                 if (mod_opts)
  70                         g_hash_table_destroy(mod_opts);
  71                 if (mod_args)
  72                         g_hash_table_destroy(mod_args);
  73                 if ((fd = open(opt_input_file, O_RDONLY)) == -1)
  74                         g_critical("Failed to load %s: %s.", opt_input_file,
  75                                         strerror(errno));
  76         } else {
  77                 /* Let the input modules try to identify the file. */
  78                 if ((fd = open(opt_input_file, O_RDONLY)) == -1)
  79                         g_critical("Failed to load %s: %s.", opt_input_file,
  80                                         strerror(errno));
  81                 if ((len = read(fd, buf->str, BUFSIZE)) < 1)
  82                         g_critical("Failed to read %s: %s.", opt_input_file,
  83                                         strerror(errno));
  84                 buf->len = len;
  85                 if (!(in = sr_input_scan_buffer(buf)))
  86                         g_critical("Error: no input module found for this file.");
  87         }
  88         sdi = sr_input_dev_inst_get(in);
  89
  90         if (select_channels(sdi) != SR_OK)
  91                 return;
  92
  93         sr_session_new(&session);
  94         sr_session_datafeed_callback_add(session, &datafeed_in, NULL);
  95         if (sr_session_dev_add(session, sdi) != SR_OK) {
  96                 g_critical("Failed to use device.");
  97                 sr_session_destroy(session);
  98                 return;
  99         }
 100
 101         while(TRUE) {
 102                 g_string_truncate(buf, 0);
 103                 len = read(fd, buf->str, BUFSIZE);
 104                 if (len < 0)
 105                         g_critical("Read failed: %s", strerror(errno));
 106                 buf->len = len;
 107                 if (sr_input_send(in, buf) != SR_OK)
 108                         break;
 109                 if (len < BUFSIZE)
 110                         break;
 111         }
 112         sr_input_free(in);
 113         g_string_free(buf, TRUE);
 114
 115         sr_session_destroy(session);
 116
 117 }
 118
 119 void load_input_file(void)
 120 {
 121         struct sr_session *session;
 122         struct sr_dev_inst *sdi;
 123         GSList *devices;
 124         struct stat st;
 125         int ret;
 126
 127         if (stat(opt_input_file, &st) == -1)
 128                 g_critical("Failed to load %s: %s.", opt_input_file, strerror(errno));
 129
 130         if (sr_session_load(opt_input_file, &session) == SR_OK) {
 131                 /* sigrok session file */
 132                 ret = sr_session_dev_list(session, &devices);
 133                 if (ret != SR_OK || !devices->data) {
 134                         g_critical("Failed to access session device.");
 135                         sr_session_destroy(session);
 136                         return;
 137                 }
 138                 sdi = devices->data;
 139                 if (select_channels(sdi) != SR_OK) {
 140                         sr_session_destroy(session);
 141                         return;
 142                 }
 143                 sr_session_datafeed_callback_add(session, datafeed_in, NULL);
 144                 sr_session_start(session);
 145                 sr_session_run(session);
 146                 sr_session_stop(session);
 147         }
 148         else {
 149                 /* fall back on input modules */
 150                 load_input_file_module();
 151         }
 152 }