#define DEFAULT_OUTPUT_FORMAT "bits:width=64"
+static struct sr_context *sr_ctx = NULL;
+
static uint64_t limit_samples = 0;
static uint64_t limit_frames = 0;
static struct sr_output_format *output_format = NULL;
static int default_output_format = FALSE;
static char *output_format_param = NULL;
static GHashTable *pd_ann_visible = NULL;
-static struct sr_datastore *singleds = NULL;
+static GByteArray *savebuf;
static gboolean opt_version = FALSE;
static gint opt_loglevel = SR_LOG_WARN; /* Show errors+warnings per default. */
return opts;
}
+static void free_hwopt(struct sr_hwopt *hwopt)
+{
+ g_free((void *)hwopt->value);
+ g_free(hwopt);
+}
+
static GSList *device_scan(void)
{
struct sr_dev_driver **drivers, *driver;
return NULL;
}
g_free(drvname);
- if (sr_driver_init(driver) != SR_OK) {
+ if (sr_driver_init(sr_ctx, driver) != SR_OK) {
g_critical("Failed to initialize driver.");
return NULL;
}
/* Unknown options, already logged. */
return NULL;
devices = sr_driver_scan(driver, drvopts);
+ g_slist_free_full(drvopts, (GDestroyNotify)free_hwopt);
} else {
/* No driver specified, let them all scan on their own. */
devices = NULL;
drivers = sr_driver_list();
for (i = 0; drivers[i]; i++) {
driver = drivers[i];
- if (sr_driver_init(driver) != SR_OK) {
+ if (sr_driver_init(sr_ctx, driver) != SR_OK) {
g_critical("Failed to initialize driver.");
return NULL;
}
static void show_version(void)
{
- GSList *l;
+ const GSList *l;
struct sr_dev_driver **drivers;
struct sr_input_format **inputs;
struct sr_output_format **outputs;
if (sdi->probes) {
if (g_slist_length(sdi->probes) == 1) {
probe = sdi->probes->data;
- printf("with 1 probe: %s\n", probe->name);
+ printf("with 1 probe: %s", probe->name);
} else {
- printf("with %d probes:\n", g_slist_length(sdi->probes));
+ printf("with %d probes:", g_slist_length(sdi->probes));
for (l = sdi->probes; l; l = l->next) {
probe = l->data;
- printf(" %s\n", probe->name);
+ printf(" %s", probe->name);
}
}
- } else
- printf("\n");
+ }
+ printf("\n");
}
static void show_dev_list(void)
printf(" %s\n", stropts[i]);
} else if (hwo->hwcap == SR_HWCAP_FILTER) {
- /* Supported trigger sources */
+ /* Supported filters */
printf(" %s", hwo->shortname);
if (sr_info_get(sdi->driver, SR_DI_FILTERS,
(const void **)&stropts, sdi) != SR_OK) {
g_strfreev(pdtokens);
}
+static GArray *get_enabled_logic_probes(const struct sr_dev_inst *sdi)
+{
+ struct sr_probe *probe;
+ GArray *probes;
+ GSList *l;
+
+ probes = g_array_new(FALSE, FALSE, sizeof(int));
+ for (l = sdi->probes; l; l = l->next) {
+ probe = l->data;
+ if (probe->type != SR_PROBE_LOGIC)
+ continue;
+ if (probe->enabled != TRUE)
+ continue;
+ g_array_append_val(probes, probe->index);
+ }
+
+ return probes;
+}
+
static void datafeed_in(const struct sr_dev_inst *sdi,
- struct sr_datafeed_packet *packet)
+ const struct sr_datafeed_packet *packet)
{
+ const struct sr_datafeed_logic *logic;
+ const struct sr_datafeed_analog *analog;
static struct sr_output *o = NULL;
- static int logic_probelist[SR_MAX_NUM_PROBES] = { -1 };
- static struct sr_probe *analog_probelist[SR_MAX_NUM_PROBES];
+ static GArray *logic_probelist = NULL;
static uint64_t received_samples = 0;
static int unitsize = 0;
static int triggered = 0;
static FILE *outfile = NULL;
- static int num_analog_probes = 0;
- struct sr_probe *probe;
- struct sr_datafeed_logic *logic;
- struct sr_datafeed_meta_logic *meta_logic;
- struct sr_datafeed_analog *analog;
- struct sr_datafeed_meta_analog *meta_analog;
- static int num_enabled_analog_probes = 0;
- int num_enabled_probes, sample_size, ret, i;
- uint64_t output_len, filter_out_len;
+ int sample_size, ret;
+ uint64_t *samplerate, output_len, filter_out_len;
uint8_t *output_buf, *filter_out;
GString *out;
exit(1);
}
}
+
+ /* Prepare non-stdout output. */
+ outfile = stdout;
+ if (opt_output_file) {
+ if (default_output_format) {
+ /* output file is in session format, so we'll
+ * keep a copy of everything as it comes in
+ * and save from there after the session. */
+ outfile = NULL;
+ savebuf = g_byte_array_new();
+ } else {
+ /* saving to a file in whatever format was set
+ * with --format, so all we need is a filehandle */
+ outfile = g_fopen(opt_output_file, "wb");
+ }
+ }
+
+ /* Prepare for logic data. */
+ logic_probelist = get_enabled_logic_probes(sdi);
+ /* How many bytes we need to store the packed samples. */
+ unitsize = (logic_probelist->len + 7) / 8;
+
+ if (opt_pds && logic_probelist->len) {
+ if (sr_info_get(sdi->driver, SR_DI_CUR_SAMPLERATE,
+ (const void **)&samplerate, sdi) != SR_OK) {
+ g_critical("Unable to initialize protocol "
+ "decoders: no samplerate found.");
+ break;
+ }
+ srd_session_start(logic_probelist->len, unitsize,
+ *samplerate);
+ }
break;
case SR_DF_END:
if (opt_continuous)
g_warning("Device stopped after %" PRIu64 " samples.",
received_samples);
+
if (outfile && outfile != stdout)
fclose(outfile);
+ if (opt_output_file && default_output_format) {
+ if (sr_session_save(opt_output_file, sdi, savebuf->data,
+ unitsize, savebuf->len / unitsize) != SR_OK)
+ g_critical("Failed to save session.");
+ g_byte_array_free(savebuf, FALSE);
+ }
+
+ g_array_free(logic_probelist, TRUE);
if (o->format->cleanup)
o->format->cleanup(o);
g_free(o);
triggered = 1;
break;
- case SR_DF_META_LOGIC:
- g_message("cli: Received SR_DF_META_LOGIC");
- meta_logic = packet->payload;
- num_enabled_probes = 0;
- for (i = 0; i < meta_logic->num_probes; i++) {
- probe = g_slist_nth_data(sdi->probes, i);
- if (probe->enabled)
- logic_probelist[num_enabled_probes++] = probe->index;
- }
- logic_probelist[num_enabled_probes] = -1;
- /* How many bytes we need to store num_enabled_probes bits */
- unitsize = (num_enabled_probes + 7) / 8;
-
- outfile = stdout;
- if (opt_output_file) {
- if (default_output_format) {
- /* output file is in session format, which means we'll
- * dump everything in the datastore as it comes in,
- * and save from there after the session. */
- outfile = NULL;
- ret = sr_datastore_new(unitsize, &singleds);
- if (ret != SR_OK) {
- g_critical("Failed to create datastore.");
- exit(1);
- }
- } else {
- /* saving to a file in whatever format was set
- * with --format, so all we need is a filehandle */
- outfile = g_fopen(opt_output_file, "wb");
- }
- }
- if (opt_pds)
- srd_session_start(num_enabled_probes, unitsize,
- meta_logic->samplerate);
- break;
-
case SR_DF_LOGIC:
logic = packet->payload;
g_message("cli: received SR_DF_LOGIC, %"PRIu64" bytes", logic->length);
if (ret != SR_OK)
break;
- /* what comes out of the filter is guaranteed to be packed into the
+ /* What comes out of the filter is guaranteed to be packed into the
* minimum size needed to support the number of samples at this sample
* size. however, the driver may have submitted too much -- cut off
* the buffer of the last packet according to the sample limit.
limit_samples * sample_size))
filter_out_len = limit_samples * sample_size - received_samples;
- if (singleds)
- sr_datastore_put(singleds, filter_out,
- filter_out_len, sample_size, logic_probelist);
-
- if (opt_output_file && default_output_format)
- /* saving to a session file, don't need to do anything else
- * to this data for now. */
- goto cleanup;
-
- if (opt_pds) {
- if (srd_session_send(received_samples, (uint8_t*)filter_out,
- filter_out_len) != SRD_OK)
- sr_session_stop();
+ if (opt_output_file && default_output_format) {
+ /* Saving to a session file. */
+ g_byte_array_append(savebuf, filter_out, filter_out_len);
} else {
- output_len = 0;
- if (o->format->data && packet->type == o->format->df_type)
- o->format->data(o, filter_out, filter_out_len, &output_buf, &output_len);
- if (output_buf) {
- fwrite(output_buf, 1, output_len, outfile);
- fflush(outfile);
- g_free(output_buf);
+ if (opt_pds) {
+ if (srd_session_send(received_samples, (uint8_t*)filter_out,
+ filter_out_len) != SRD_OK)
+ sr_session_stop();
+ } else {
+ output_len = 0;
+ if (o->format->data && packet->type == o->format->df_type)
+ o->format->data(o, filter_out, filter_out_len,
+ &output_buf, &output_len);
+ if (output_buf) {
+ fwrite(output_buf, 1, output_len, outfile);
+ fflush(outfile);
+ g_free(output_buf);
+ }
}
}
-
- cleanup:
g_free(filter_out);
- received_samples += logic->length / sample_size;
- break;
- case SR_DF_META_ANALOG:
- g_message("cli: Received SR_DF_META_ANALOG");
- meta_analog = packet->payload;
- num_analog_probes = meta_analog->num_probes;
- num_enabled_analog_probes = 0;
- for (i = 0; i < num_analog_probes; i++) {
- probe = g_slist_nth_data(sdi->probes, i);
- if (probe->enabled)
- analog_probelist[num_enabled_analog_probes++] = probe;
- }
-
- outfile = stdout;
- if (opt_output_file) {
- if (default_output_format) {
- /* output file is in session format, which means we'll
- * dump everything in the datastore as it comes in,
- * and save from there after the session. */
- outfile = NULL;
- ret = sr_datastore_new(unitsize, &singleds);
- if (ret != SR_OK) {
- g_critical("Failed to create datastore.");
- exit(1);
- }
- } else {
- /* saving to a file in whatever format was set
- * with --format, so all we need is a filehandle */
- outfile = g_fopen(opt_output_file, "wb");
- }
- }
+ received_samples += logic->length / sample_size;
break;
case SR_DF_ANALOG:
break;
default:
- g_message("received unknown packet type %d", packet->type);
+ break;
}
if (o && o->format->recv) {
int ret;
char **pdtokens, **pdtok, *pd_name;
- /* Avoid compiler warnings. */
(void)dev;
ret = 0;
static int select_probes(struct sr_dev_inst *sdi)
{
- char **probelist;
- int max_probes, i;
+ struct sr_probe *probe;
+ GSList *selected_probes, *l;
if (!opt_probes)
return SR_OK;
- /*
- * This only works because a device by default initializes
- * and enables all its probes.
- */
- max_probes = g_slist_length(sdi->probes);
- probelist = parse_probestring(max_probes, opt_probes);
- if (!probelist) {
+ if (!(selected_probes = parse_probestring(sdi, opt_probes)))
return SR_ERR;
- }
- for (i = 0; i < max_probes; i++) {
- if (probelist[i]) {
- sr_dev_probe_name_set(sdi, i, probelist[i]);
- g_free(probelist[i]);
- } else {
- sr_dev_probe_enable(sdi, i, FALSE);
- }
+ for (l = sdi->probes; l; l = l->next) {
+ probe = l->data;
+ if (g_slist_find(selected_probes, probe))
+ probe->enabled = TRUE;
+ else
+ probe->enabled = FALSE;
}
- g_free(probelist);
+ g_slist_free(selected_probes);
return SR_OK;
}
}
if (select_probes(in->sdi) > 0)
- return;
+ return;
sr_session_new();
sr_session_datafeed_callback_add(datafeed_in);
}
input_format->loadfile(in, opt_input_file);
- if (opt_output_file && default_output_format) {
- if (sr_session_save(opt_output_file, in->sdi, singleds) != SR_OK)
- g_critical("Failed to save session.");
- }
+
sr_session_destroy();
if (fmtargs)
g_critical("Failed to set device option '%s'.", (char *)key);
return ret;
}
- else
- break;
}
return SR_OK;
if (opt_continuous)
clear_anykey();
- if (opt_output_file && default_output_format) {
- if (sr_session_save(opt_output_file, sdi, singleds) != SR_OK)
- g_critical("Failed to save session.");
- }
sr_session_destroy();
g_slist_free(devices);
static void logger(const gchar *log_domain, GLogLevelFlags log_level,
const gchar *message, gpointer cb_data)
{
- /* Avoid compiler warnings. */
(void)log_domain;
(void)cb_data;
int main(int argc, char **argv)
{
+ int ret = 1;
GOptionContext *context;
GError *error;
if (!g_option_context_parse(context, &argc, &argv, &error)) {
g_critical("%s", error->message);
- return 1;
+ goto done;
}
/* Set the loglevel (amount of messages to output) for libsigrok. */
if (sr_log_loglevel_set(opt_loglevel) != SR_OK)
- return 1;
+ goto done;
/* Set the loglevel (amount of messages to output) for libsigrokdecode. */
if (srd_log_loglevel_set(opt_loglevel) != SRD_OK)
- return 1;
+ goto done;
- if (sr_init() != SR_OK)
- return 1;
+ if (sr_init(&sr_ctx) != SR_OK)
+ goto done;
if (opt_pds) {
if (srd_init(NULL) != SRD_OK)
- return 1;
+ goto done;
if (register_pds(NULL, opt_pds) != 0)
- return 1;
+ goto done;
if (srd_pd_output_callback_add(SRD_OUTPUT_ANN,
show_pd_annotations, NULL) != SRD_OK)
- return 1;
+ goto done;
if (setup_pd_stack() != 0)
- return 1;
+ goto done;
if (setup_pd_annotations() != 0)
- return 1;
+ goto done;
}
if (setup_output_format() != 0)
- return 1;
+ goto done;
if (opt_version)
show_version();
if (opt_pds)
srd_exit();
+ ret = 0;
+
+done:
+ if (sr_ctx)
+ sr_exit(sr_ctx);
+
g_option_context_free(context);
- sr_exit();
- return 0;
+ return ret;
}