X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=src%2Finput%2Fwav.c;h=8f8a37101200d49488596d7e914d7e8bbd490ad3;hb=0af636bed97c174bea46e61e961eaa1b0b162e0f;hp=5669e8a4453c456195c95af9d92432167f89cb75;hpb=7db0639495cb8d96aadcd17678a535c97449a677;p=libsigrok.git diff --git a/src/input/wav.c b/src/input/wav.c index 5669e8a4..8f8a3710 100644 --- a/src/input/wav.c +++ b/src/input/wav.c @@ -23,6 +23,7 @@ #include #include #include +#include #include "libsigrok.h" #include "libsigrok-internal.h" @@ -37,10 +38,12 @@ /* Expect to find the "data" chunk within this offset from the start. */ #define MAX_DATA_CHUNK_OFFSET 256 -#define WAVE_FORMAT_PCM 1 -#define WAVE_FORMAT_IEEE_FLOAT 3 +#define WAVE_FORMAT_PCM 0x0001 +#define WAVE_FORMAT_IEEE_FLOAT 0x0003 +#define WAVE_FORMAT_EXTENSIBLE 0xfffe struct context { + gboolean started; int fmt_code; uint64_t samplerate; int samplesize; @@ -52,32 +55,61 @@ struct context { static int parse_wav_header(GString *buf, struct context *inc) { uint64_t samplerate; - int fmt_code, samplesize, num_channels, unitsize; + unsigned int fmt_code, samplesize, num_channels, unitsize; - if (buf->len < MIN_DATA_CHUNK_OFFSET) { + if (buf->len < MIN_DATA_CHUNK_OFFSET) + return SR_ERR_NA; + + fmt_code = RL16(buf->str + 20); + samplerate = RL32(buf->str + 24); + + 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; } - fmt_code = GUINT16_FROM_LE(*(uint16_t *)(buf->str + 20)); - samplerate = GUINT32_FROM_LE(*(uint32_t *)(buf->str + 24)); - samplesize = GUINT16_FROM_LE(*(uint16_t *)(buf->str + 32)); - num_channels = GUINT16_FROM_LE(*(uint16_t *)(buf->str + 22)); - /* TODO div0 */ - unitsize = samplesize / num_channels; if (fmt_code == WAVE_FORMAT_PCM) { - if (samplesize != 1 && samplesize != 2 && samplesize != 4) { - sr_err("only 8, 16 or 32 bits per sample supported."); - return SR_ERR; - } } 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_ERR_NA; + + 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; + return SR_ERR_DATA; } if (inc) { @@ -95,29 +127,31 @@ static int parse_wav_header(GString *buf, struct context *inc) static int format_match(GHashTable *metadata) { GString *buf; + int ret; buf = g_hash_table_lookup(metadata, GINT_TO_POINTER(SR_INPUT_META_HEADER)); if (strncmp(buf->str, "RIFF", 4)) - return FALSE; + return SR_ERR; if (strncmp(buf->str + 8, "WAVE", 4)) - return FALSE; + return SR_ERR; if (strncmp(buf->str + 12, "fmt ", 4)) - return FALSE; + return SR_ERR; /* * Only gets called when we already know this is a WAV file, so * this parser can log error messages. */ - if (parse_wav_header(buf, NULL) != SR_OK) - return FALSE; + if ((ret = parse_wav_header(buf, NULL)) != SR_OK) + return ret; - return TRUE; + return SR_OK; } static int init(struct sr_input *in, GHashTable *options) { (void)options; - in->sdi = sr_dev_inst_new(0, SR_ST_ACTIVE, NULL, NULL, NULL); + in->sdi = sr_dev_inst_new(); + in->priv = g_malloc0(sizeof(struct context)); return SR_OK; } @@ -139,48 +173,12 @@ static int find_data_chunk(GString *buf, int initial_offset) return -1; } /* Skip past this chunk. */ - offset += 8 + GUINT32_FROM_LE(*(uint32_t *)(buf->str + offset + 4)); + offset += 8 + RL32(buf->str + offset + 4); } return offset; } -static int initial_receive(struct sr_input *in) -{ - struct sr_datafeed_packet packet; - struct sr_datafeed_meta meta; - struct sr_channel *ch; - struct sr_config *src; - struct context *inc; - int i; - char channelname[8]; - - if (!in->buf) - /* Shouldn't happen. */ - return SR_ERR; - - inc = in->priv = g_malloc(sizeof(struct context)); - if (parse_wav_header(in->buf, inc) != SR_OK) - return SR_ERR; - - 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); - } - - 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(inc->samplerate)); - meta.config = g_slist_append(NULL, src); - sr_session_send(in->sdi, &packet); - sr_config_free(src); - - return SR_OK; -} - static void send_chunk(const struct sr_input *in, int offset, int num_samples) { struct sr_datafeed_packet packet; @@ -204,20 +202,21 @@ static void send_chunk(const struct sr_input *in, int offset, int num_samples) switch (inc->samplesize) { case 1: /* 8-bit PCM samples are unsigned. */ - fdata[samplenum] = (uint8_t)sample / 255.0; + fdata[samplenum] = (uint8_t)sample / (float)255; break; case 2: - fdata[samplenum] = GINT16_FROM_LE(sample) / 32767.0; + fdata[samplenum] = RL16S(&sample) / (float)INT16_MAX; break; case 4: - fdata[samplenum] = GINT32_FROM_LE(sample) / 65535.0; + 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 - i]; + d[i] = s[inc->unitsize - 1 - i]; #else memcpy(d, s, inc->unitsize); #endif @@ -236,37 +235,40 @@ static void send_chunk(const struct sr_input *in, int offset, int num_samples) sr_session_send(in->sdi, &packet); } -static int receive(const struct sr_input *in, GString *buf) +static int process_buffer(struct sr_input *in) { - struct sr_datafeed_packet packet; struct context *inc; - int offset, chunk_samples, total_samples, processed, max_chunk_samples, num_samples, i; + struct sr_datafeed_packet packet; + struct sr_datafeed_meta meta; + struct sr_channel *ch; + struct sr_config *src; + int offset, chunk_samples, total_samples, processed, max_chunk_samples; + int num_samples, i; + char channelname[8]; - if (buf->len == 0) { - /* End of stream. */ - packet.type = SR_DF_END; - sr_session_send(in->sdi, &packet); - return SR_OK; - } + inc = in->priv; + if (!inc->started) { + 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); + } - g_string_append_len(in->buf, buf->str, buf->len); + std_session_send_df_header(in->sdi, LOG_PREFIX); - 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; - } + packet.type = SR_DF_META; + packet.payload = &meta; + 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); + + inc->started = TRUE; } - inc = in->priv; if (!inc->found_data) { /* Skip past size of 'fmt ' chunk. */ - i = 20 + GUINT32_FROM_LE(*(uint32_t *)(in->buf->str + 16)); + 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) { @@ -306,12 +308,57 @@ static int receive(const struct sr_input *in, GString *buf) return SR_OK; } -static int cleanup(struct sr_input *in) +static int receive(struct sr_input *in, GString *buf) { - g_free(in->priv); - in->priv = NULL; + struct context *inc; + int ret; - return SR_OK; + g_string_append_len(in->buf, buf->str, buf->len); + + if (in->buf->len < MIN_DATA_CHUNK_OFFSET) { + /* + * Don't even try until there's enough room + * for the data segment to start. + */ + return SR_OK; + } + + inc = in->priv; + if (!in->sdi_ready) { + if ((ret = parse_wav_header(in->buf, inc)) == SR_ERR_NA) + /* Not enough data yet. */ + return SR_OK; + else if (ret != SR_OK) + return ret; + + /* sdi is ready, notify frontend. */ + in->sdi_ready = TRUE; + return SR_OK; + } + + ret = process_buffer(in); + + return ret; +} + +static int end(struct sr_input *in) +{ + struct sr_datafeed_packet packet; + struct context *inc; + int ret; + + if (in->sdi_ready) + ret = process_buffer(in); + else + ret = SR_OK; + + inc = in->priv; + if (inc->started) { + packet.type = SR_DF_END; + sr_session_send(in->sdi, &packet); + } + + return ret; } SR_PRIV struct sr_input_module input_wav = { @@ -322,6 +369,6 @@ SR_PRIV struct sr_input_module input_wav = { .format_match = format_match, .init = init, .receive = receive, - .cleanup = cleanup, + .end = end, };