input/binary.c

   1 /*
   2  * This file is part of the sigrok project.
   3  *
   4  * Copyright (C) 2010-2012 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 <stdlib.h>
  21 #include <sys/types.h>
  22 #include <sys/stat.h>
  23 #include <fcntl.h>
  24 #include <unistd.h>
  25 #include <sys/time.h>
  26 #include "libsigrok.h"
  27 #include "libsigrok-internal.h"
  28
  29 #define CHUNKSIZE             (512 * 1024)
  30 #define DEFAULT_NUM_PROBES    8
  31
  32 struct context {
  33         uint64_t samplerate;
  34 };
  35
  36 static int format_match(const char *filename)
  37 {
  38         /* suppress compiler warning */
  39         (void)filename;
  40
  41         /* this module will handle anything you throw at it */
  42         return TRUE;
  43 }
  44
  45 static int init(struct sr_input *in)
  46 {
  47         struct sr_probe *probe;
  48         int num_probes, i;
  49         char name[SR_MAX_PROBENAME_LEN + 1];
  50         char *param;
  51         struct context *ctx;
  52
  53         if (!(ctx = g_try_malloc0(sizeof(*ctx)))) {
  54                 sr_err("binary in: %s: ctx malloc failed", __func__);
  55                 return SR_ERR_MALLOC;
  56         }
  57
  58         num_probes = DEFAULT_NUM_PROBES;
  59         ctx->samplerate = 0;
  60
  61         if(in->param) {
  62                 param = g_hash_table_lookup(in->param, "numprobes");
  63                 if (param) {
  64                         num_probes = strtoul(param, NULL, 10);
  65                         if (num_probes < 1)
  66                                 return SR_ERR;
  67                 }
  68
  69                 param = g_hash_table_lookup(in->param, "samplerate");
  70                 if (param) {
  71                         if (sr_parse_sizestring(param, &ctx->samplerate) != SR_OK)
  72                                 return SR_ERR;
  73                 }
  74         }
  75
  76         /* Create a virtual device. */
  77         in->sdi = sr_dev_inst_new(0, SR_ST_ACTIVE, NULL, NULL, NULL);
  78         in->internal = ctx;
  79
  80         for (i = 0; i < num_probes; i++) {
  81                 snprintf(name, SR_MAX_PROBENAME_LEN, "%d", i);
  82                 /* TODO: Check return value. */
  83                 if (!(probe = sr_probe_new(i, SR_PROBE_LOGIC, TRUE, name)))
  84                         return SR_ERR;
  85                 in->sdi->probes = g_slist_append(in->sdi->probes, probe);
  86         }
  87
  88         return SR_OK;
  89 }
  90
  91 static int loadfile(struct sr_input *in, const char *filename)
  92 {
  93         struct sr_datafeed_header header;
  94         struct sr_datafeed_packet packet;
  95         struct sr_datafeed_meta_logic meta;
  96         struct sr_datafeed_logic logic;
  97         unsigned char buffer[CHUNKSIZE];
  98         int fd, size, num_probes;
  99         struct context *ctx;
 100
 101         ctx = in->internal;
 102
 103         if ((fd = open(filename, O_RDONLY)) == -1)
 104                 return SR_ERR;
 105
 106         num_probes = g_slist_length(in->sdi->probes);
 107
 108         /* send header */
 109         header.feed_version = 1;
 110         gettimeofday(&header.starttime, NULL);
 111         packet.type = SR_DF_HEADER;
 112         packet.payload = &header;
 113         sr_session_send(in->sdi, &packet);
 114
 115         /* Send metadata about the SR_DF_LOGIC packets to come. */
 116         packet.type = SR_DF_META_LOGIC;
 117         packet.payload = &meta;
 118         meta.samplerate = ctx->samplerate;
 119         meta.num_probes = num_probes;
 120         sr_session_send(in->sdi, &packet);
 121
 122         /* chop up the input file into chunks and feed it into the session bus */
 123         packet.type = SR_DF_LOGIC;
 124         packet.payload = &logic;
 125         logic.unitsize = (num_probes + 7) / 8;
 126         logic.data = buffer;
 127         while ((size = read(fd, buffer, CHUNKSIZE)) > 0) {
 128                 logic.length = size;
 129                 sr_session_send(in->sdi, &packet);
 130         }
 131         close(fd);
 132
 133         /* end of stream */
 134         packet.type = SR_DF_END;
 135         sr_session_send(in->sdi, &packet);
 136
 137         g_free(ctx);
 138         in->internal = NULL;
 139
 140         return SR_OK;
 141 }
 142
 143 SR_PRIV struct sr_input_format input_binary = {
 144         .id = "binary",
 145         .description = "Raw binary",
 146         .format_match = format_match,
 147         .init = init,
 148         .loadfile = loadfile,
 149 };