#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
+#include <ctype.h>
#include <string.h>
+#include <stdint.h>
#include "libsigrok.h"
#include "libsigrok-internal.h"
#define LOG_PREFIX "input/wav"
+/* How many bytes at a time to process and send to the session bus. */
#define CHUNK_SIZE 4096
+/* Minimum size of header + 1 8-bit mono PCM sample. */
+#define MIN_DATA_CHUNK_OFFSET 45
+
+/* Expect to find the "data" chunk within this offset from the start. */
+#define MAX_DATA_CHUNK_OFFSET 256
+
+#define WAVE_FORMAT_PCM 0x0001
+#define WAVE_FORMAT_IEEE_FLOAT 0x0003
+#define WAVE_FORMAT_EXTENSIBLE 0xfffe
+
struct context {
+ int fmt_code;
uint64_t samplerate;
int samplesize;
int num_channels;
+ int unitsize;
+ gboolean found_data;
};
-static int get_wav_header(const char *filename, char *buf)
+static int parse_wav_header(GString *buf, struct context *inc)
{
- struct stat st;
- int fd, l;
+ uint64_t samplerate;
+ unsigned int fmt_code, samplesize, num_channels, unitsize;
- l = strlen(filename);
- if (l <= 4 || strcasecmp(filename + l - 4, ".wav"))
- return SR_ERR;
+ if (buf->len < MIN_DATA_CHUNK_OFFSET)
+ return SR_OK_CONTINUE;
- if (stat(filename, &st) == -1)
- return SR_ERR;
- if (st.st_size <= 45)
- /* Minimum size of header + 1 8-bit mono PCM sample. */
- return SR_ERR;
+ fmt_code = RL16(buf->str + 20);
+ samplerate = RL32(buf->str + 24);
- if ((fd = open(filename, O_RDONLY)) == -1)
+ samplesize = RL16(buf->str + 32);
+ num_channels = RL16(buf->str + 22);
+ if (num_channels == 0)
return SR_ERR;
+ unitsize = samplesize / num_channels;
+ if (unitsize != 1 && unitsize != 2 && unitsize != 4) {
+ sr_err("Only 8, 16 or 32 bits per sample supported.");
+ return SR_ERR_DATA;
+ }
- l = read(fd, buf, 40);
- close(fd);
- if (l != 40)
- return SR_ERR;
- return SR_OK;
-}
+ if (fmt_code == WAVE_FORMAT_PCM) {
+ } else if (fmt_code == WAVE_FORMAT_IEEE_FLOAT) {
+ if (unitsize != 4) {
+ sr_err("only 32-bit floats supported.");
+ return SR_ERR_DATA;
+ }
+ } else if (fmt_code == WAVE_FORMAT_EXTENSIBLE) {
+ if (buf->len < 70)
+ /* Not enough for extensible header and next chunk. */
+ return SR_OK_CONTINUE;
-static int format_match(const char *filename)
-{
- char buf[40];
-
- if (get_wav_header(filename, buf) != SR_OK)
- return FALSE;
-
- if (strncmp(buf, "RIFF", 4))
- return FALSE;
- if (strncmp(buf + 8, "WAVE", 4))
- return FALSE;
- if (strncmp(buf + 12, "fmt ", 4))
- return FALSE;
- if (GUINT16_FROM_LE(*(uint16_t *)(buf + 20)) != 1)
- /* Not PCM. */
- return FALSE;
- if (strncmp(buf + 36, "data", 4))
- return FALSE;
-
- return TRUE;
+ if (RL16(buf->str + 16) != 40) {
+ sr_err("WAV extensible format chunk must be 40 bytes.");
+ return SR_ERR;
+ }
+ if (RL16(buf->str + 36) != 22) {
+ sr_err("WAV extension must be 22 bytes.");
+ return SR_ERR;
+ }
+ if (RL16(buf->str + 34) != RL16(buf->str + 38)) {
+ sr_err("Reduced valid bits per sample not supported.");
+ return SR_ERR_DATA;
+ }
+ /* Real format code is the first two bytes of the GUID. */
+ fmt_code = RL16(buf->str + 44);
+ if (fmt_code != WAVE_FORMAT_PCM && fmt_code != WAVE_FORMAT_IEEE_FLOAT) {
+ sr_err("Only PCM and floating point samples are supported.");
+ return SR_ERR_DATA;
+ }
+ if (fmt_code == WAVE_FORMAT_IEEE_FLOAT && unitsize != 4) {
+ sr_err("only 32-bit floats supported.");
+ return SR_ERR_DATA;
+ }
+ } else {
+ sr_err("Only PCM and floating point samples are supported.");
+ return SR_ERR_DATA;
+ }
+
+ if (inc) {
+ inc->fmt_code = fmt_code;
+ inc->samplerate = samplerate;
+ inc->samplesize = samplesize;
+ inc->num_channels = num_channels;
+ inc->unitsize = unitsize;
+ inc->found_data = FALSE;
+ }
+
+ return SR_OK;
}
-static int init(struct sr_input *in, const char *filename)
+static int format_match(GHashTable *metadata)
{
- struct sr_channel *ch;
- struct context *ctx;
- char buf[40], channelname[8];
- int i;
+ GString *buf;
+ int ret;
- if (get_wav_header(filename, buf) != SR_OK)
+ buf = g_hash_table_lookup(metadata, GINT_TO_POINTER(SR_INPUT_META_HEADER));
+ if (strncmp(buf->str, "RIFF", 4))
+ return SR_ERR;
+ if (strncmp(buf->str + 8, "WAVE", 4))
+ return SR_ERR;
+ if (strncmp(buf->str + 12, "fmt ", 4))
return SR_ERR;
+ /*
+ * Only gets called when we already know this is a WAV file, so
+ * this parser can log error messages.
+ */
+ if ((ret = parse_wav_header(buf, NULL)) != SR_OK)
+ return ret;
- if (!(ctx = g_try_malloc0(sizeof(struct context))))
- return SR_ERR_MALLOC;
+ return SR_OK;
+}
- /* Create a virtual device. */
- in->sdi = sr_dev_inst_new(0, SR_ST_ACTIVE, NULL, NULL, NULL);
- in->sdi->priv = ctx;
+static int init(struct sr_input *in, GHashTable *options)
+{
+ (void)options;
- ctx->samplerate = GUINT32_FROM_LE(*(uint32_t *)(buf + 24));
- ctx->samplesize = GUINT16_FROM_LE(*(uint16_t *)(buf + 34)) / 8;
- if (ctx->samplesize != 1 && ctx->samplesize != 2 && ctx->samplesize != 4) {
- sr_err("only 8, 16 or 32 bits per sample supported.");
- return SR_ERR;
- }
+ in->sdi = sr_dev_inst_new(0, SR_ST_ACTIVE, NULL, NULL, NULL);
- if ((ctx->num_channels = GUINT16_FROM_LE(*(uint16_t *)(buf + 22))) > 20) {
- sr_err("%d channels seems crazy.", ctx->num_channels);
- return SR_ERR;
- }
+ return SR_OK;
+}
- for (i = 0; i < ctx->num_channels; i++) {
- snprintf(channelname, 8, "CH%d", i + 1);
- if (!(ch = sr_channel_new(0, SR_CHANNEL_ANALOG, TRUE, channelname)))
- return SR_ERR;
- in->sdi->channels = g_slist_append(in->sdi->channels, ch);
+static int find_data_chunk(GString *buf, int initial_offset)
+{
+ unsigned int offset, i;
+
+ offset = initial_offset;
+ while(offset < MIN(MAX_DATA_CHUNK_OFFSET, buf->len)) {
+ if (!memcmp(buf->str + offset, "data", 4))
+ /* Skip into the samples. */
+ return offset + 8;
+ for (i = 0; i < 4; i++) {
+ if (!isalpha(buf->str[offset + i])
+ && !isascii(buf->str[offset + i])
+ && !isblank(buf->str[offset + i]))
+ /* Doesn't look like a chunk ID. */
+ return -1;
+ }
+ /* Skip past this chunk. */
+ offset += 8 + RL32(buf->str + offset + 4);
}
- return SR_OK;
+ return offset;
}
-static int loadfile(struct sr_input *in, const char *filename)
+static int initial_receive(struct sr_input *in)
{
struct sr_datafeed_packet packet;
struct sr_datafeed_meta meta;
- struct sr_datafeed_analog analog;
+ struct sr_channel *ch;
struct sr_config *src;
- struct context *ctx;
- float fdata[CHUNK_SIZE];
- uint64_t sample;
- int num_samples, chunk_samples, s, c, fd, l;
- char buf[CHUNK_SIZE];
+ struct context *inc;
+ int ret, i;
+ char channelname[8];
- ctx = in->sdi->priv;
+ if (!in->buf)
+ /* Shouldn't happen. */
+ return SR_ERR;
+
+ inc = in->priv = g_malloc(sizeof(struct context));
+ if ((ret = parse_wav_header(in->buf, inc)) != SR_OK)
+ return ret;
+
+ for (i = 0; i < inc->num_channels; i++) {
+ snprintf(channelname, 8, "CH%d", i + 1);
+ ch = sr_channel_new(i, SR_CHANNEL_ANALOG, TRUE, channelname);
+ in->sdi->channels = g_slist_append(in->sdi->channels, ch);
+ }
- /* Send header packet to the session bus. */
std_session_send_df_header(in->sdi, LOG_PREFIX);
packet.type = SR_DF_META;
packet.payload = &meta;
- src = sr_config_new(SR_CONF_SAMPLERATE,
- g_variant_new_uint64(ctx->samplerate));
+ src = sr_config_new(SR_CONF_SAMPLERATE, g_variant_new_uint64(inc->samplerate));
meta.config = g_slist_append(NULL, src);
sr_session_send(in->sdi, &packet);
sr_config_free(src);
- if ((fd = open(filename, O_RDONLY)) == -1)
- return SR_ERR;
+ return SR_OK;
+}
- lseek(fd, 40, SEEK_SET);
- l = read(fd, buf, 4);
- num_samples = GUINT32_FROM_LE((uint32_t)*(buf));
- num_samples /= ctx->samplesize / ctx->num_channels;
- while (TRUE) {
- if ((l = read(fd, buf, CHUNK_SIZE)) < 1)
- break;
- chunk_samples = l / ctx->samplesize / ctx->num_channels;
- for (s = 0; s < chunk_samples; s++) {
- for (c = 0; c < ctx->num_channels; c++) {
- sample = 0;
- memcpy(&sample, buf + s * ctx->samplesize + c, ctx->samplesize);
- switch (ctx->samplesize) {
- case 1:
- /* 8-bit PCM samples are unsigned. */
- fdata[s + c] = (uint8_t)sample / 255.0;
- break;
- case 2:
- fdata[s + c] = GINT16_FROM_LE(sample) / 32767.0;
- break;
- case 4:
- fdata[s + c] = GINT32_FROM_LE(sample) / 65535.0;
- break;
- }
+static void send_chunk(const struct sr_input *in, int offset, int num_samples)
+{
+ struct sr_datafeed_packet packet;
+ struct sr_datafeed_analog analog;
+ struct context *inc;
+ float fdata[CHUNK_SIZE];
+ uint64_t sample;
+ int total_samples, samplenum;
+ char *s, *d;
+
+ inc = in->priv;
+
+ s = in->buf->str + offset;
+ d = (char *)fdata;
+ memset(fdata, 0, CHUNK_SIZE);
+ total_samples = num_samples * inc->num_channels;
+ for (samplenum = 0; samplenum < total_samples; samplenum++) {
+ if (inc->fmt_code == WAVE_FORMAT_PCM) {
+ sample = 0;
+ memcpy(&sample, s, inc->unitsize);
+ switch (inc->samplesize) {
+ case 1:
+ /* 8-bit PCM samples are unsigned. */
+ fdata[samplenum] = (uint8_t)sample / (float)255;
+ break;
+ case 2:
+ fdata[samplenum] = RL16S(&sample) / (float)INT16_MAX;
+ break;
+ case 4:
+ fdata[samplenum] = RL32S(&sample) / (float)INT32_MAX;
+ break;
}
+ } else {
+ /* BINARY32 float */
+#ifdef WORDS_BIGENDIAN
+ int i;
+ for (i = 0; i < inc->unitsize; i++)
+ d[i] = s[inc->unitsize - 1 - i];
+#else
+ memcpy(d, s, inc->unitsize);
+#endif
}
- packet.type = SR_DF_ANALOG;
- packet.payload = &analog;
- analog.channels = in->sdi->channels;
- analog.num_samples = chunk_samples;
- analog.mq = 0;
- analog.unit = 0;
- analog.data = fdata;
- sr_session_send(in->sdi, &packet);
+ s += inc->unitsize;
+ d += inc->unitsize;
}
-
- close(fd);
- packet.type = SR_DF_END;
+ packet.type = SR_DF_ANALOG;
+ packet.payload = &analog;
+ analog.channels = in->sdi->channels;
+ analog.num_samples = num_samples;
+ analog.mq = 0;
+ analog.mqflags = 0;
+ analog.unit = 0;
+ analog.data = fdata;
sr_session_send(in->sdi, &packet);
+}
+
+static int receive(const struct sr_input *in, GString *buf)
+{
+ struct context *inc;
+ int offset, chunk_samples, total_samples, processed, max_chunk_samples, num_samples, i;
+
+ g_string_append_len(in->buf, buf->str, buf->len);
+
+ if (!in->priv) {
+ if (initial_receive((struct sr_input *)in) != SR_OK)
+ return SR_ERR;
+ if (in->buf->len < MIN_DATA_CHUNK_OFFSET) {
+ /*
+ * Don't even get started until there's enough room
+ * for the data segment to start.
+ */
+ return SR_OK;
+ }
+ }
+ inc = in->priv;
+
+ if (!inc->found_data) {
+ /* Skip past size of 'fmt ' chunk. */
+ i = 20 + RL32(in->buf->str + 16);
+ offset = find_data_chunk(in->buf, i);
+ if (offset < 0) {
+ if (in->buf->len > MAX_DATA_CHUNK_OFFSET) {
+ sr_err("Couldn't find data chunk.");
+ return SR_ERR;
+ }
+ }
+ inc->found_data = TRUE;
+ } else
+ offset = 0;
+
+ /* Round off up to the last channels * unitsize boundary. */
+ chunk_samples = (in->buf->len - offset) / inc->num_channels / inc->unitsize;
+ max_chunk_samples = CHUNK_SIZE / inc->num_channels / inc->unitsize;
+ processed = 0;
+ total_samples = chunk_samples;
+ while (processed < total_samples) {
+ if (chunk_samples > max_chunk_samples)
+ num_samples = max_chunk_samples;
+ else
+ num_samples = chunk_samples;
+ send_chunk(in, offset, num_samples);
+ offset += num_samples * inc->unitsize;
+ chunk_samples -= num_samples;
+ processed += num_samples;
+ }
+
+ if ((unsigned int)offset < in->buf->len) {
+ /*
+ * The incoming buffer wasn't processed completely. Stash
+ * the leftover data for next time.
+ */
+ g_string_erase(in->buf, 0, offset);
+ } else
+ g_string_truncate(in->buf, 0);
return SR_OK;
}
+static int cleanup(struct sr_input *in)
+{
+ struct sr_datafeed_packet packet;
+
+ if (in->priv) {
+ /* End of stream. */
+ packet.type = SR_DF_END;
+ sr_session_send(in->sdi, &packet);
+
+ g_free(in->priv);
+ in->priv = NULL;
+ }
+
+ return SR_OK;
+}
-SR_PRIV struct sr_input_format input_wav = {
+SR_PRIV struct sr_input_module input_wav = {
.id = "wav",
- .description = "WAV file",
+ .name = "WAV",
+ .desc = "WAV file",
+ .metadata = { SR_INPUT_META_HEADER | SR_INPUT_META_REQUIRED },
.format_match = format_match,
.init = init,
- .loadfile = loadfile,
+ .receive = receive,
+ .cleanup = cleanup,
};