src/input/binary.c

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