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 /* Message logging helpers with driver-specific prefix string. */
  30 #define DRIVER_LOG_DOMAIN "input/binary: "
  31 #define sr_log(l, s, args...) sr_log(l, DRIVER_LOG_DOMAIN s, ## args)
  32 #define sr_spew(s, args...) sr_spew(DRIVER_LOG_DOMAIN s, ## args)
  33 #define sr_dbg(s, args...) sr_dbg(DRIVER_LOG_DOMAIN s, ## args)
  34 #define sr_info(s, args...) sr_info(DRIVER_LOG_DOMAIN s, ## args)
  35 #define sr_warn(s, args...) sr_warn(DRIVER_LOG_DOMAIN s, ## args)
  36 #define sr_err(s, args...) sr_err(DRIVER_LOG_DOMAIN s, ## args)
  37
  38 #define CHUNKSIZE             (512 * 1024)
  39 #define DEFAULT_NUM_PROBES    8
  40
  41 struct context {
  42         uint64_t samplerate;
  43 };
  44
  45 static int format_match(const char *filename)
  46 {
  47         (void)filename;
  48
  49         /* This module will handle anything you throw at it. */
  50         return TRUE;
  51 }
  52
  53 static int init(struct sr_input *in, const char *filename)
  54 {
  55         struct sr_probe *probe;
  56         int num_probes, i;
  57         char name[SR_MAX_PROBENAME_LEN + 1];
  58         char *param;
  59         struct context *ctx;
  60
  61         (void)filename;
  62
  63         if (!(ctx = g_try_malloc0(sizeof(*ctx)))) {
  64                 sr_err("Input format context malloc failed.");
  65                 return SR_ERR_MALLOC;
  66         }
  67
  68         num_probes = DEFAULT_NUM_PROBES;
  69         ctx->samplerate = 0;
  70
  71         if (in->param) {
  72                 param = g_hash_table_lookup(in->param, "numprobes");
  73                 if (param) {
  74                         num_probes = strtoul(param, NULL, 10);
  75                         if (num_probes < 1)
  76                                 return SR_ERR;
  77                 }
  78
  79                 param = g_hash_table_lookup(in->param, "samplerate");
  80                 if (param) {
  81                         if (sr_parse_sizestring(param, &ctx->samplerate) != SR_OK)
  82                                 return SR_ERR;
  83                 }
  84         }
  85
  86         /* Create a virtual device. */
  87         in->sdi = sr_dev_inst_new(0, SR_ST_ACTIVE, NULL, NULL, NULL);
  88         in->internal = ctx;
  89
  90         for (i = 0; i < num_probes; i++) {
  91                 snprintf(name, SR_MAX_PROBENAME_LEN, "%d", i);
  92                 /* TODO: Check return value. */
  93                 if (!(probe = sr_probe_new(i, SR_PROBE_LOGIC, TRUE, name)))
  94                         return SR_ERR;
  95                 in->sdi->probes = g_slist_append(in->sdi->probes, probe);
  96         }
  97
  98         return SR_OK;
  99 }
 100
 101 static int loadfile(struct sr_input *in, const char *filename)
 102 {
 103         struct sr_datafeed_packet packet;
 104         struct sr_datafeed_meta meta;
 105         struct sr_datafeed_logic logic;
 106         struct sr_config *src;
 107         unsigned char buffer[CHUNKSIZE];
 108         int fd, size, num_probes;
 109         struct context *ctx;
 110
 111         ctx = in->internal;
 112
 113         if ((fd = open(filename, O_RDONLY)) == -1)
 114                 return SR_ERR;
 115
 116         num_probes = g_slist_length(in->sdi->probes);
 117
 118         /* Send header packet to the session bus. */
 119         std_session_send_df_header(in->sdi, DRIVER_LOG_DOMAIN);
 120
 121         if (ctx->samplerate) {
 122                 packet.type = SR_DF_META;
 123                 packet.payload = &meta;
 124                 src = sr_config_new(SR_CONF_SAMPLERATE, (const void *)&ctx->samplerate);
 125                 meta.config = g_slist_append(NULL, src);
 126                 sr_session_send(in->sdi, &packet);
 127         }
 128
 129         /* Chop up the input file into chunks & send it to the session bus. */
 130         packet.type = SR_DF_LOGIC;
 131         packet.payload = &logic;
 132         logic.unitsize = (num_probes + 7) / 8;
 133         logic.data = buffer;
 134         while ((size = read(fd, buffer, CHUNKSIZE)) > 0) {
 135                 logic.length = size;
 136                 sr_session_send(in->sdi, &packet);
 137         }
 138         close(fd);
 139
 140         /* Send end packet to the session bus. */
 141         packet.type = SR_DF_END;
 142         sr_session_send(in->sdi, &packet);
 143
 144         g_free(ctx);
 145         in->internal = NULL;
 146
 147         return SR_OK;
 148 }
 149
 150 SR_PRIV struct sr_input_format input_binary = {
 151         .id = "binary",
 152         .description = "Raw binary",
 153         .format_match = format_match,
 154         .init = init,
 155         .loadfile = loadfile,
 156 };