Gnuplot output: More error handling.

This commit is contained in:
Uwe Hermann 2010-05-09 14:35:02 +02:00
parent 2aebf78d9d
commit a821069b34
1 changed files with 21 additions and 16 deletions

View File

@ -41,7 +41,6 @@ const char *gnuplot_header = "\
const char *gnuplot_header_comment = "\ const char *gnuplot_header_comment = "\
# Comment: Acquisition with %d/%d probes at %s\n"; # Comment: Acquisition with %d/%d probes at %s\n";
static int init(struct output *o) static int init(struct output *o)
{ {
/* Maximum header length */ /* Maximum header length */
@ -56,9 +55,9 @@ static int init(struct output *o)
char *c, *samplerate_s; char *c, *samplerate_s;
char wbuf[1000], comment[128]; char wbuf[1000], comment[128];
ctx = malloc(sizeof(struct context)); if (!(ctx = calloc(1, sizeof(struct context))))
if (ctx == NULL)
return SIGROK_ERR_MALLOC; return SIGROK_ERR_MALLOC;
o->internal = ctx; o->internal = ctx;
ctx->num_enabled_probes = 0; ctx->num_enabled_probes = 0;
for (l = o->device->probes; l; l = l->next) { for (l = o->device->probes; l; l = l->next) {
@ -72,22 +71,27 @@ static int init(struct output *o)
/* TODO: Allow for configuration via o->param. */ /* TODO: Allow for configuration via o->param. */
ctx->header = calloc(1, MAX_HEADER_LEN + 1); if (!(ctx->header = calloc(1, MAX_HEADER_LEN + 1))) {
if (ctx->header == NULL) free(ctx);
return SIGROK_ERR_MALLOC; return SIGROK_ERR_MALLOC;
}
num_probes = g_slist_length(o->device->probes); num_probes = g_slist_length(o->device->probes);
/* TODO: Handle num_probes == 0, too many probes, etc. */ /* TODO: Handle num_probes == 0, too many probes, etc. */
comment[0] = '\0';
if (o->device->plugin) { if (o->device->plugin) {
samplerate = *((uint64_t *) o->device->plugin->get_device_info( samplerate = *((uint64_t *) o->device->plugin->get_device_info(
o->device->plugin_index, DI_CUR_SAMPLERATE)); o->device->plugin_index, DI_CUR_SAMPLERATE));
if ((samplerate_s = sigrok_samplerate_string(samplerate)) == NULL) if (!(samplerate_s = sigrok_samplerate_string(samplerate))) {
free(ctx->header);
free(ctx);
return SIGROK_ERR; return SIGROK_ERR;
snprintf(comment, 127, gnuplot_header_comment, ctx->num_enabled_probes, }
num_probes, samplerate_s); snprintf(comment, 127, gnuplot_header_comment,
ctx->num_enabled_probes, num_probes, samplerate_s);
free(samplerate_s); free(samplerate_s);
} }
else
comment[0] = '\0';
/* Columns / channels */ /* Columns / channels */
wbuf[0] = '\0'; wbuf[0] = '\0';
@ -116,6 +120,7 @@ static int event(struct output *o, int event_type, char **data_out,
ctx = o->internal; ctx = o->internal;
switch (event_type) { switch (event_type) {
case DF_TRIGGER: case DF_TRIGGER:
/* TODO */
break; break;
case DF_END: case DF_END:
outbuf = calloc(1, 1); /* FIXME */ outbuf = calloc(1, 1); /* FIXME */
@ -140,20 +145,20 @@ static int data(struct output *o, char *data_in, uint64_t length_in,
char *outbuf, *c; char *outbuf, *c;
ctx = o->internal; ctx = o->internal;
outsize = 0; outsize = 0;
if (ctx->header) if (ctx->header)
outsize = strlen(ctx->header); outsize = strlen(ctx->header);
outbuf = calloc(1, outsize + 1 + 10000); /* FIXME: Use realloc(). */
if (outbuf == NULL) /* FIXME: Use realloc(). */
return SIGROK_ERR_MALLOC; if (!(outbuf = calloc(1, outsize + 1 + 10000)))
return SIGROK_ERR_MALLOC; /* TODO: free()? What to free? */
outbuf[0] = '\0';
if (ctx->header) { if (ctx->header) {
/* The header is still here, this must be the first packet. */ /* The header is still here, this must be the first packet. */
strncpy(outbuf, ctx->header, outsize); strncpy(outbuf, ctx->header, outsize);
free(ctx->header); free(ctx->header);
ctx->header = NULL; ctx->header = NULL;
} else {
outbuf[0] = 0;
} }
/* TODO: Are disabled probes handled correctly? */ /* TODO: Are disabled probes handled correctly? */