FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
wavdec.c
Go to the documentation of this file.
1 /*
2  * WAV demuxer
3  * Copyright (c) 2001, 2002 Fabrice Bellard
4  *
5  * Sony Wave64 demuxer
6  * RF64 demuxer
7  * Copyright (c) 2009 Daniel Verkamp
8  *
9  * This file is part of FFmpeg.
10  *
11  * FFmpeg is free software; you can redistribute it and/or
12  * modify it under the terms of the GNU Lesser General Public
13  * License as published by the Free Software Foundation; either
14  * version 2.1 of the License, or (at your option) any later version.
15  *
16  * FFmpeg is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
19  * Lesser General Public License for more details.
20  *
21  * You should have received a copy of the GNU Lesser General Public
22  * License along with FFmpeg; if not, write to the Free Software
23  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
24  */
25 
26 #include <stdint.h>
27 
28 #include "libavutil/avassert.h"
29 #include "libavutil/dict.h"
30 #include "libavutil/intreadwrite.h"
31 #include "libavutil/log.h"
32 #include "libavutil/mathematics.h"
33 #include "libavutil/opt.h"
34 #include "avformat.h"
35 #include "avio.h"
36 #include "avio_internal.h"
37 #include "internal.h"
38 #include "metadata.h"
39 #include "pcm.h"
40 #include "riff.h"
41 #include "w64.h"
42 #include "spdif.h"
43 
44 typedef struct WAVDemuxContext {
45  const AVClass *class;
46  int64_t data_end;
47  int w64;
48  int64_t smv_data_ofs;
51  int smv_block;
53  int smv_eof;
54  int audio_eof;
56  int spdif;
59  int unaligned; // e.g. if an odd number of bytes ID3 tag was prepended
60  int rifx; // RIFX: integer byte order for parameters is big endian
62 
63 #if CONFIG_WAV_DEMUXER
64 
65 static int64_t next_tag(AVIOContext *pb, uint32_t *tag, int big_endian)
66 {
67  *tag = avio_rl32(pb);
68  if (!big_endian) {
69  return avio_rl32(pb);
70  } else {
71  return avio_rb32(pb);
72  }
73 }
74 
75 /* RIFF chunks are always at even offsets relative to where they start. */
76 static int64_t wav_seek_tag(WAVDemuxContext * wav, AVIOContext *s, int64_t offset, int whence)
77 {
78  offset += offset < INT64_MAX && offset + wav->unaligned & 1;
79 
80  return avio_seek(s, offset, whence);
81 }
82 
83 /* return the size of the found tag */
84 static int64_t find_tag(WAVDemuxContext * wav, AVIOContext *pb, uint32_t tag1)
85 {
86  unsigned int tag;
87  int64_t size;
88 
89  for (;;) {
90  if (avio_feof(pb))
91  return AVERROR_EOF;
92  size = next_tag(pb, &tag, wav->rifx);
93  if (tag == tag1)
94  break;
95  wav_seek_tag(wav, pb, size, SEEK_CUR);
96  }
97  return size;
98 }
99 
100 static int wav_probe(AVProbeData *p)
101 {
102  /* check file header */
103  if (p->buf_size <= 32)
104  return 0;
105  if (!memcmp(p->buf + 8, "WAVE", 4)) {
106  if (!memcmp(p->buf, "RIFF", 4) || !memcmp(p->buf, "RIFX", 4))
107  /* Since the ACT demuxer has a standard WAV header at the top of
108  * its own, the returned score is decreased to avoid a probe
109  * conflict between ACT and WAV. */
110  return AVPROBE_SCORE_MAX - 1;
111  else if (!memcmp(p->buf, "RF64", 4) &&
112  !memcmp(p->buf + 12, "ds64", 4))
113  return AVPROBE_SCORE_MAX;
114  }
115  return 0;
116 }
117 
118 static void handle_stream_probing(AVStream *st)
119 {
120  if (st->codec->codec_id == AV_CODEC_ID_PCM_S16LE) {
122  st->probe_packets = FFMIN(st->probe_packets, 32);
123  }
124 }
125 
126 static int wav_parse_fmt_tag(AVFormatContext *s, int64_t size, AVStream **st)
127 {
128  AVIOContext *pb = s->pb;
129  WAVDemuxContext *wav = s->priv_data;
130  int ret;
131 
132  /* parse fmt header */
133  *st = avformat_new_stream(s, NULL);
134  if (!*st)
135  return AVERROR(ENOMEM);
136 
137  ret = ff_get_wav_header(pb, (*st)->codec, size, wav->rifx);
138  if (ret < 0)
139  return ret;
140  handle_stream_probing(*st);
141 
142  (*st)->need_parsing = AVSTREAM_PARSE_FULL_RAW;
143 
144  avpriv_set_pts_info(*st, 64, 1, (*st)->codec->sample_rate);
145 
146  return 0;
147 }
148 
149 static inline int wav_parse_bext_string(AVFormatContext *s, const char *key,
150  int length)
151 {
152  char temp[257];
153  int ret;
154 
155  av_assert0(length <= sizeof(temp));
156  if ((ret = avio_read(s->pb, temp, length)) < 0)
157  return ret;
158 
159  temp[length] = 0;
160 
161  if (strlen(temp))
162  return av_dict_set(&s->metadata, key, temp, 0);
163 
164  return 0;
165 }
166 
167 static int wav_parse_bext_tag(AVFormatContext *s, int64_t size)
168 {
169  char temp[131], *coding_history;
170  int ret, x;
171  uint64_t time_reference;
172  int64_t umid_parts[8], umid_mask = 0;
173 
174  if ((ret = wav_parse_bext_string(s, "description", 256)) < 0 ||
175  (ret = wav_parse_bext_string(s, "originator", 32)) < 0 ||
176  (ret = wav_parse_bext_string(s, "originator_reference", 32)) < 0 ||
177  (ret = wav_parse_bext_string(s, "origination_date", 10)) < 0 ||
178  (ret = wav_parse_bext_string(s, "origination_time", 8)) < 0)
179  return ret;
180 
181  time_reference = avio_rl64(s->pb);
182  snprintf(temp, sizeof(temp), "%"PRIu64, time_reference);
183  if ((ret = av_dict_set(&s->metadata, "time_reference", temp, 0)) < 0)
184  return ret;
185 
186  /* check if version is >= 1, in which case an UMID may be present */
187  if (avio_rl16(s->pb) >= 1) {
188  for (x = 0; x < 8; x++)
189  umid_mask |= umid_parts[x] = avio_rb64(s->pb);
190 
191  if (umid_mask) {
192  /* the string formatting below is per SMPTE 330M-2004 Annex C */
193  if (umid_parts[4] == 0 && umid_parts[5] == 0 &&
194  umid_parts[6] == 0 && umid_parts[7] == 0) {
195  /* basic UMID */
196  snprintf(temp, sizeof(temp),
197  "0x%016"PRIX64"%016"PRIX64"%016"PRIX64"%016"PRIX64,
198  umid_parts[0], umid_parts[1],
199  umid_parts[2], umid_parts[3]);
200  } else {
201  /* extended UMID */
202  snprintf(temp, sizeof(temp),
203  "0x%016"PRIX64"%016"PRIX64"%016"PRIX64"%016"PRIX64
204  "%016"PRIX64"%016"PRIX64"%016"PRIX64"%016"PRIX64,
205  umid_parts[0], umid_parts[1],
206  umid_parts[2], umid_parts[3],
207  umid_parts[4], umid_parts[5],
208  umid_parts[6], umid_parts[7]);
209  }
210 
211  if ((ret = av_dict_set(&s->metadata, "umid", temp, 0)) < 0)
212  return ret;
213  }
214 
215  avio_skip(s->pb, 190);
216  } else
217  avio_skip(s->pb, 254);
218 
219  if (size > 602) {
220  /* CodingHistory present */
221  size -= 602;
222 
223  if (!(coding_history = av_malloc(size + 1)))
224  return AVERROR(ENOMEM);
225 
226  if ((ret = avio_read(s->pb, coding_history, size)) < 0)
227  return ret;
228 
229  coding_history[size] = 0;
230  if ((ret = av_dict_set(&s->metadata, "coding_history", coding_history,
232  return ret;
233  }
234 
235  return 0;
236 }
237 
238 static const AVMetadataConv wav_metadata_conv[] = {
239  { "description", "comment" },
240  { "originator", "encoded_by" },
241  { "origination_date", "date" },
242  { "origination_time", "creation_time" },
243  { 0 },
244 };
245 
246 /* wav input */
247 static int wav_read_header(AVFormatContext *s)
248 {
249  int64_t size, av_uninit(data_size);
250  int64_t sample_count = 0;
251  int rf64 = 0;
252  char start_code[32];
253  uint32_t tag;
254  AVIOContext *pb = s->pb;
255  AVStream *st = NULL;
256  WAVDemuxContext *wav = s->priv_data;
257  int ret, got_fmt = 0;
258  int64_t next_tag_ofs, data_ofs = -1;
259 
260  wav->unaligned = avio_tell(s->pb) & 1;
261 
262  wav->smv_data_ofs = -1;
263 
264  /* read chunk ID */
265  tag = avio_rl32(pb);
266  switch (tag) {
267  case MKTAG('R', 'I', 'F', 'F'):
268  break;
269  case MKTAG('R', 'I', 'F', 'X'):
270  wav->rifx = 1;
271  break;
272  case MKTAG('R', 'F', '6', '4'):
273  rf64 = 1;
274  break;
275  default:
276  av_get_codec_tag_string(start_code, sizeof(start_code), tag);
277  av_log(s, AV_LOG_ERROR, "invalid start code %s in RIFF header\n", start_code);
278  return AVERROR_INVALIDDATA;
279  }
280 
281  /* read chunk size */
282  avio_rl32(pb);
283 
284  /* read format */
285  if (avio_rl32(pb) != MKTAG('W', 'A', 'V', 'E')) {
286  av_log(s, AV_LOG_ERROR, "invalid format in RIFF header\n");
287  return AVERROR_INVALIDDATA;
288  }
289 
290  if (rf64) {
291  if (avio_rl32(pb) != MKTAG('d', 's', '6', '4'))
292  return AVERROR_INVALIDDATA;
293  size = avio_rl32(pb);
294  if (size < 24)
295  return AVERROR_INVALIDDATA;
296  avio_rl64(pb); /* RIFF size */
297 
298  data_size = avio_rl64(pb);
299  sample_count = avio_rl64(pb);
300 
301  if (data_size < 0 || sample_count < 0) {
302  av_log(s, AV_LOG_ERROR, "negative data_size and/or sample_count in "
303  "ds64: data_size = %"PRId64", sample_count = %"PRId64"\n",
304  data_size, sample_count);
305  return AVERROR_INVALIDDATA;
306  }
307  avio_skip(pb, size - 24); /* skip rest of ds64 chunk */
308 
309  }
310 
311  for (;;) {
312  AVStream *vst;
313  size = next_tag(pb, &tag, wav->rifx);
314  next_tag_ofs = avio_tell(pb) + size;
315 
316  if (avio_feof(pb))
317  break;
318 
319  switch (tag) {
320  case MKTAG('f', 'm', 't', ' '):
321  /* only parse the first 'fmt ' tag found */
322  if (!got_fmt && (ret = wav_parse_fmt_tag(s, size, &st)) < 0) {
323  return ret;
324  } else if (got_fmt)
325  av_log(s, AV_LOG_WARNING, "found more than one 'fmt ' tag\n");
326 
327  got_fmt = 1;
328  break;
329  case MKTAG('d', 'a', 't', 'a'):
330  if (!got_fmt) {
331  av_log(s, AV_LOG_ERROR,
332  "found no 'fmt ' tag before the 'data' tag\n");
333  return AVERROR_INVALIDDATA;
334  }
335 
336  if (rf64) {
337  next_tag_ofs = wav->data_end = avio_tell(pb) + data_size;
338  } else if (size != 0xFFFFFFFF) {
339  data_size = size;
340  next_tag_ofs = wav->data_end = size ? next_tag_ofs : INT64_MAX;
341  } else {
342  av_log(s, AV_LOG_WARNING, "Ignoring maximum wav data size, "
343  "file may be invalid\n");
344  data_size = 0;
345  next_tag_ofs = wav->data_end = INT64_MAX;
346  }
347 
348  data_ofs = avio_tell(pb);
349 
350  /* don't look for footer metadata if we can't seek or if we don't
351  * know where the data tag ends
352  */
353  if (!pb->seekable || (!rf64 && !size))
354  goto break_loop;
355  break;
356  case MKTAG('f', 'a', 'c', 't'):
357  if (!sample_count)
358  sample_count = (!wav->rifx ? avio_rl32(pb) : avio_rb32(pb));
359  break;
360  case MKTAG('b', 'e', 'x', 't'):
361  if ((ret = wav_parse_bext_tag(s, size)) < 0)
362  return ret;
363  break;
364  case MKTAG('S','M','V','0'):
365  if (!got_fmt) {
366  av_log(s, AV_LOG_ERROR, "found no 'fmt ' tag before the 'SMV0' tag\n");
367  return AVERROR_INVALIDDATA;
368  }
369  // SMV file, a wav file with video appended.
370  if (size != MKTAG('0','2','0','0')) {
371  av_log(s, AV_LOG_ERROR, "Unknown SMV version found\n");
372  goto break_loop;
373  }
374  av_log(s, AV_LOG_DEBUG, "Found SMV data\n");
375  wav->smv_given_first = 0;
376  vst = avformat_new_stream(s, NULL);
377  if (!vst)
378  return AVERROR(ENOMEM);
379  avio_r8(pb);
380  vst->id = 1;
383  vst->codec->width = avio_rl24(pb);
384  vst->codec->height = avio_rl24(pb);
385  if (ff_alloc_extradata(vst->codec, 4)) {
386  av_log(s, AV_LOG_ERROR, "Could not allocate extradata.\n");
387  return AVERROR(ENOMEM);
388  }
389  size = avio_rl24(pb);
390  wav->smv_data_ofs = avio_tell(pb) + (size - 5) * 3;
391  avio_rl24(pb);
392  wav->smv_block_size = avio_rl24(pb);
393  avpriv_set_pts_info(vst, 32, 1, avio_rl24(pb));
394  vst->duration = avio_rl24(pb);
395  avio_rl24(pb);
396  avio_rl24(pb);
397  wav->smv_frames_per_jpeg = avio_rl24(pb);
398  if (wav->smv_frames_per_jpeg > 65536) {
399  av_log(s, AV_LOG_ERROR, "too many frames per jpeg\n");
400  return AVERROR_INVALIDDATA;
401  }
403  wav->smv_cur_pt = 0;
404  goto break_loop;
405  case MKTAG('L', 'I', 'S', 'T'):
406  if (size < 4) {
407  av_log(s, AV_LOG_ERROR, "too short LIST tag\n");
408  return AVERROR_INVALIDDATA;
409  }
410  switch (avio_rl32(pb)) {
411  case MKTAG('I', 'N', 'F', 'O'):
412  ff_read_riff_info(s, size - 4);
413  }
414  break;
415  }
416 
417  /* seek to next tag unless we know that we'll run into EOF */
418  if ((avio_size(pb) > 0 && next_tag_ofs >= avio_size(pb)) ||
419  wav_seek_tag(wav, pb, next_tag_ofs, SEEK_SET) < 0) {
420  break;
421  }
422  }
423 
424 break_loop:
425  if (data_ofs < 0) {
426  av_log(s, AV_LOG_ERROR, "no 'data' tag found\n");
427  return AVERROR_INVALIDDATA;
428  }
429 
430  avio_seek(pb, data_ofs, SEEK_SET);
431 
432  if ( data_size > 0 && sample_count && st->codec->channels
433  && data_size / sample_count / st->codec->channels > 8) {
434  av_log(s, AV_LOG_WARNING, "ignoring wrong sample_count %"PRId64"\n", sample_count);
435  sample_count = 0;
436  }
437 
438  if (!sample_count || av_get_exact_bits_per_sample(st->codec->codec_id) > 0)
439  if ( st->codec->channels
440  && data_size
442  && wav->data_end <= avio_size(pb))
443  sample_count = (data_size << 3)
444  /
445  (st->codec->channels * (uint64_t)av_get_bits_per_sample(st->codec->codec_id));
446 
447  if (sample_count)
448  st->duration = sample_count;
449 
450  ff_metadata_conv_ctx(s, NULL, wav_metadata_conv);
452 
453  return 0;
454 }
455 
456 /**
457  * Find chunk with w64 GUID by skipping over other chunks.
458  * @return the size of the found chunk
459  */
460 static int64_t find_guid(AVIOContext *pb, const uint8_t guid1[16])
461 {
462  uint8_t guid[16];
463  int64_t size;
464 
465  while (!avio_feof(pb)) {
466  avio_read(pb, guid, 16);
467  size = avio_rl64(pb);
468  if (size <= 24)
469  return AVERROR_INVALIDDATA;
470  if (!memcmp(guid, guid1, 16))
471  return size;
472  avio_skip(pb, FFALIGN(size, INT64_C(8)) - 24);
473  }
474  return AVERROR_EOF;
475 }
476 
477 #define MAX_SIZE 4096
478 
479 static int wav_read_packet(AVFormatContext *s, AVPacket *pkt)
480 {
481  int ret, size;
482  int64_t left;
483  AVStream *st;
484  WAVDemuxContext *wav = s->priv_data;
485 
486  if (CONFIG_SPDIF_DEMUXER && wav->spdif == 0 &&
487  s->streams[0]->codec->codec_tag == 1) {
488  enum AVCodecID codec;
489  ret = ff_spdif_probe(s->pb->buffer, s->pb->buf_end - s->pb->buffer,
490  &codec);
491  if (ret > AVPROBE_SCORE_EXTENSION) {
492  s->streams[0]->codec->codec_id = codec;
493  wav->spdif = 1;
494  } else {
495  wav->spdif = -1;
496  }
497  }
498  if (CONFIG_SPDIF_DEMUXER && wav->spdif == 1)
499  return ff_spdif_read_packet(s, pkt);
500 
501  if (wav->smv_data_ofs > 0) {
502  int64_t audio_dts, video_dts;
503 smv_retry:
504  audio_dts = (int32_t)s->streams[0]->cur_dts;
505  video_dts = (int32_t)s->streams[1]->cur_dts;
506 
507  if (audio_dts != AV_NOPTS_VALUE && video_dts != AV_NOPTS_VALUE) {
508  /*We always return a video frame first to get the pixel format first*/
509  wav->smv_last_stream = wav->smv_given_first ?
510  av_compare_ts(video_dts, s->streams[1]->time_base,
511  audio_dts, s->streams[0]->time_base) > 0 : 0;
512  wav->smv_given_first = 1;
513  }
514  wav->smv_last_stream = !wav->smv_last_stream;
515  wav->smv_last_stream |= wav->audio_eof;
516  wav->smv_last_stream &= !wav->smv_eof;
517  if (wav->smv_last_stream) {
518  uint64_t old_pos = avio_tell(s->pb);
519  uint64_t new_pos = wav->smv_data_ofs +
520  wav->smv_block * wav->smv_block_size;
521  if (avio_seek(s->pb, new_pos, SEEK_SET) < 0) {
522  ret = AVERROR_EOF;
523  goto smv_out;
524  }
525  size = avio_rl24(s->pb);
526  ret = av_get_packet(s->pb, pkt, size);
527  if (ret < 0)
528  goto smv_out;
529  pkt->pos -= 3;
530  pkt->pts = wav->smv_block * wav->smv_frames_per_jpeg + wav->smv_cur_pt;
531  wav->smv_cur_pt++;
532  if (wav->smv_frames_per_jpeg > 0)
533  wav->smv_cur_pt %= wav->smv_frames_per_jpeg;
534  if (!wav->smv_cur_pt)
535  wav->smv_block++;
536 
537  pkt->stream_index = 1;
538 smv_out:
539  avio_seek(s->pb, old_pos, SEEK_SET);
540  if (ret == AVERROR_EOF) {
541  wav->smv_eof = 1;
542  goto smv_retry;
543  }
544  return ret;
545  }
546  }
547 
548  st = s->streams[0];
549 
550  left = wav->data_end - avio_tell(s->pb);
551  if (wav->ignore_length)
552  left = INT_MAX;
553  if (left <= 0) {
554  if (CONFIG_W64_DEMUXER && wav->w64)
555  left = find_guid(s->pb, ff_w64_guid_data) - 24;
556  else
557  left = find_tag(wav, s->pb, MKTAG('d', 'a', 't', 'a'));
558  if (left < 0) {
559  wav->audio_eof = 1;
560  if (wav->smv_data_ofs > 0 && !wav->smv_eof)
561  goto smv_retry;
562  return AVERROR_EOF;
563  }
564  wav->data_end = avio_tell(s->pb) + left;
565  }
566 
567  size = MAX_SIZE;
568  if (st->codec->block_align > 1) {
569  if (size < st->codec->block_align)
570  size = st->codec->block_align;
571  size = (size / st->codec->block_align) * st->codec->block_align;
572  }
573  size = FFMIN(size, left);
574  ret = av_get_packet(s->pb, pkt, size);
575  if (ret < 0)
576  return ret;
577  pkt->stream_index = 0;
578 
579  return ret;
580 }
581 
582 static int wav_read_seek(AVFormatContext *s,
583  int stream_index, int64_t timestamp, int flags)
584 {
585  WAVDemuxContext *wav = s->priv_data;
586  AVStream *st;
587  wav->smv_eof = 0;
588  wav->audio_eof = 0;
589  if (wav->smv_data_ofs > 0) {
590  int64_t smv_timestamp = timestamp;
591  if (stream_index == 0)
592  smv_timestamp = av_rescale_q(timestamp, s->streams[0]->time_base, s->streams[1]->time_base);
593  else
594  timestamp = av_rescale_q(smv_timestamp, s->streams[1]->time_base, s->streams[0]->time_base);
595  if (wav->smv_frames_per_jpeg > 0) {
596  wav->smv_block = smv_timestamp / wav->smv_frames_per_jpeg;
597  wav->smv_cur_pt = smv_timestamp % wav->smv_frames_per_jpeg;
598  }
599  }
600 
601  st = s->streams[0];
602  switch (st->codec->codec_id) {
603  case AV_CODEC_ID_MP2:
604  case AV_CODEC_ID_MP3:
605  case AV_CODEC_ID_AC3:
606  case AV_CODEC_ID_DTS:
607  /* use generic seeking with dynamically generated indexes */
608  return -1;
609  default:
610  break;
611  }
612  return ff_pcm_read_seek(s, stream_index, timestamp, flags);
613 }
614 
615 #define OFFSET(x) offsetof(WAVDemuxContext, x)
616 #define DEC AV_OPT_FLAG_DECODING_PARAM
617 static const AVOption demux_options[] = {
618  { "ignore_length", "Ignore length", OFFSET(ignore_length), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, DEC },
619  { NULL },
620 };
621 
622 static const AVClass wav_demuxer_class = {
623  .class_name = "WAV demuxer",
624  .item_name = av_default_item_name,
625  .option = demux_options,
626  .version = LIBAVUTIL_VERSION_INT,
627 };
628 AVInputFormat ff_wav_demuxer = {
629  .name = "wav",
630  .long_name = NULL_IF_CONFIG_SMALL("WAV / WAVE (Waveform Audio)"),
631  .priv_data_size = sizeof(WAVDemuxContext),
632  .read_probe = wav_probe,
633  .read_header = wav_read_header,
634  .read_packet = wav_read_packet,
635  .read_seek = wav_read_seek,
636  .flags = AVFMT_GENERIC_INDEX,
637  .codec_tag = (const AVCodecTag * const []) { ff_codec_wav_tags, 0 },
638  .priv_class = &wav_demuxer_class,
639 };
640 #endif /* CONFIG_WAV_DEMUXER */
641 
642 #if CONFIG_W64_DEMUXER
643 static int w64_probe(AVProbeData *p)
644 {
645  if (p->buf_size <= 40)
646  return 0;
647  if (!memcmp(p->buf, ff_w64_guid_riff, 16) &&
648  !memcmp(p->buf + 24, ff_w64_guid_wave, 16))
649  return AVPROBE_SCORE_MAX;
650  else
651  return 0;
652 }
653 
654 static int w64_read_header(AVFormatContext *s)
655 {
656  int64_t size, data_ofs = 0;
657  AVIOContext *pb = s->pb;
658  WAVDemuxContext *wav = s->priv_data;
659  AVStream *st;
660  uint8_t guid[16];
661  int ret;
662 
663  avio_read(pb, guid, 16);
664  if (memcmp(guid, ff_w64_guid_riff, 16))
665  return AVERROR_INVALIDDATA;
666 
667  /* riff + wave + fmt + sizes */
668  if (avio_rl64(pb) < 16 + 8 + 16 + 8 + 16 + 8)
669  return AVERROR_INVALIDDATA;
670 
671  avio_read(pb, guid, 16);
672  if (memcmp(guid, ff_w64_guid_wave, 16)) {
673  av_log(s, AV_LOG_ERROR, "could not find wave guid\n");
674  return AVERROR_INVALIDDATA;
675  }
676 
677  wav->w64 = 1;
678 
679  st = avformat_new_stream(s, NULL);
680  if (!st)
681  return AVERROR(ENOMEM);
682 
683  while (!avio_feof(pb)) {
684  if (avio_read(pb, guid, 16) != 16)
685  break;
686  size = avio_rl64(pb);
687  if (size <= 24 || INT64_MAX - size < avio_tell(pb))
688  return AVERROR_INVALIDDATA;
689 
690  if (!memcmp(guid, ff_w64_guid_fmt, 16)) {
691  /* subtract chunk header size - normal wav file doesn't count it */
692  ret = ff_get_wav_header(pb, st->codec, size - 24, 0);
693  if (ret < 0)
694  return ret;
695  avio_skip(pb, FFALIGN(size, INT64_C(8)) - size);
696 
697  avpriv_set_pts_info(st, 64, 1, st->codec->sample_rate);
698  } else if (!memcmp(guid, ff_w64_guid_fact, 16)) {
699  int64_t samples;
700 
701  samples = avio_rl64(pb);
702  if (samples > 0)
703  st->duration = samples;
704  } else if (!memcmp(guid, ff_w64_guid_data, 16)) {
705  wav->data_end = avio_tell(pb) + size - 24;
706 
707  data_ofs = avio_tell(pb);
708  if (!pb->seekable)
709  break;
710 
711  avio_skip(pb, size - 24);
712  } else if (!memcmp(guid, ff_w64_guid_summarylist, 16)) {
713  int64_t start, end, cur;
714  uint32_t count, chunk_size, i;
715 
716  start = avio_tell(pb);
717  end = start + FFALIGN(size, INT64_C(8)) - 24;
718  count = avio_rl32(pb);
719 
720  for (i = 0; i < count; i++) {
721  char chunk_key[5], *value;
722 
723  if (avio_feof(pb) || (cur = avio_tell(pb)) < 0 || cur > end - 8 /* = tag + size */)
724  break;
725 
726  chunk_key[4] = 0;
727  avio_read(pb, chunk_key, 4);
728  chunk_size = avio_rl32(pb);
729 
730  value = av_mallocz(chunk_size + 1);
731  if (!value)
732  return AVERROR(ENOMEM);
733 
734  ret = avio_get_str16le(pb, chunk_size, value, chunk_size);
735  avio_skip(pb, chunk_size - ret);
736 
737  av_dict_set(&s->metadata, chunk_key, value, AV_DICT_DONT_STRDUP_VAL);
738  }
739 
740  avio_skip(pb, end - avio_tell(pb));
741  } else {
742  av_log(s, AV_LOG_DEBUG, "unknown guid: "FF_PRI_GUID"\n", FF_ARG_GUID(guid));
743  avio_skip(pb, FFALIGN(size, INT64_C(8)) - 24);
744  }
745  }
746 
747  if (!data_ofs)
748  return AVERROR_EOF;
749 
750  ff_metadata_conv_ctx(s, NULL, wav_metadata_conv);
752 
753  handle_stream_probing(st);
755 
756  avio_seek(pb, data_ofs, SEEK_SET);
757 
758  return 0;
759 }
760 
761 AVInputFormat ff_w64_demuxer = {
762  .name = "w64",
763  .long_name = NULL_IF_CONFIG_SMALL("Sony Wave64"),
764  .priv_data_size = sizeof(WAVDemuxContext),
765  .read_probe = w64_probe,
766  .read_header = w64_read_header,
767  .read_packet = wav_read_packet,
768  .read_seek = wav_read_seek,
769  .flags = AVFMT_GENERIC_INDEX,
770  .codec_tag = (const AVCodecTag * const []) { ff_codec_wav_tags, 0 },
771 };
772 #endif /* CONFIG_W64_DEMUXER */
int ff_read_riff_info(AVFormatContext *s, int64_t size)
Definition: riffdec.c:198
#define NULL
Definition: coverity.c:32
const char * s
Definition: avisynth_c.h:631
Bytestream IO Context.
Definition: avio.h:111
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
void ff_metadata_conv_ctx(AVFormatContext *ctx, const AVMetadataConv *d_conv, const AVMetadataConv *s_conv)
Definition: metadata.c:59
int64_t avio_size(AVIOContext *s)
Get the filesize.
Definition: aviobuf.c:281
const uint8_t ff_w64_guid_wave[16]
Definition: w64.c:28
Buffered I/O operations.
int ff_spdif_probe(const uint8_t *p_buf, int buf_size, enum AVCodecID *codec)
Definition: spdifdec.c:112
AVOption.
Definition: opt.h:255
const uint8_t ff_w64_guid_fact[16]
Definition: w64.c:38
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:182
#define LIBAVUTIL_VERSION_INT
Definition: version.h:62
unsigned char * buf_end
End of the data, may be less than buffer+buffer_size if the read function returned less data than req...
Definition: avio.h:128
int64_t pos
byte position in stream, -1 if unknown
Definition: avcodec.h:1187
else temp
Definition: vf_mcdeint.c:257
#define FF_ARG_GUID(g)
Definition: riff.h:97
void avpriv_set_pts_info(AVStream *s, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: utils.c:4006
int probe_packets
Definition: avformat.h:1027
static int read_seek(AVFormatContext *ctx, int stream_index, int64_t timestamp, int flags)
Definition: libcdio.c:153
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:203
int smv_frames_per_jpeg
Definition: wavdec.c:50
unsigned char * buffer
Start of the buffer.
Definition: avio.h:125
int64_t avio_skip(AVIOContext *s, int64_t offset)
Skip given number of bytes forward.
Definition: aviobuf.c:276
size_t av_get_codec_tag_string(char *buf, size_t buf_size, unsigned int codec_tag)
Put a string representing the codec tag codec_tag in buf.
Definition: utils.c:2990
int ff_get_wav_header(AVIOContext *pb, AVCodecContext *codec, int size, int big_endian)
Definition: riffdec.c:84
static AVPacket pkt
#define MAX_SIZE
Definition: vf_unsharp.c:265
int avio_get_str16le(AVIOContext *pb, int maxlen, char *buf, int buflen)
Read a UTF-16 string from pb and convert it to UTF-8.
int block_align
number of bytes per packet if constant and known or 0 Used by some WAV based audio codecs...
Definition: avcodec.h:2022
int ignore_length
Definition: wavdec.c:55
#define FFALIGN(x, a)
Definition: common.h:71
Format I/O context.
Definition: avformat.h:1272
int64_t cur_dts
Definition: avformat.h:1019
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:72
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
internal metadata API header see avformat.h or the public API!
Public dictionary API.
if()
Definition: avfilter.c:975
uint8_t
#define av_malloc(s)
AVOptions.
unsigned int avio_rb32(AVIOContext *s)
Definition: aviobuf.c:689
static av_cold int end(AVCodecContext *avctx)
Definition: avrndec.c:67
enum AVStreamParseType need_parsing
Definition: avformat.h:1035
int id
Format-specific stream ID.
Definition: avformat.h:849
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:1355
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
Definition: utils.c:3672
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1340
uint32_t tag
Definition: movenc.c:1333
#define AVERROR_EOF
End of file.
Definition: error.h:55
int av_get_packet(AVIOContext *s, AVPacket *pkt, int size)
Allocate and read the payload of a packet and initialize its fields with default values.
Definition: utils.c:241
ptrdiff_t size
Definition: opengl_enc.c:101
uint64_t avio_rb64(AVIOContext *s)
Definition: aviobuf.c:756
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:365
full parsing and repack with timestamp and position generation by parser for raw this assumes that ea...
Definition: avformat.h:778
#define av_log(a,...)
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:537
int64_t av_rescale_q(int64_t a, AVRational bq, AVRational cq)
Rescale a 64-bit integer by 2 rational numbers.
Definition: mathematics.c:140
int ff_spdif_read_packet(AVFormatContext *s, AVPacket *pkt)
Definition: spdifdec.c:169
AVCodecID
Identify the syntax and semantics of the bitstream.
Definition: avcodec.h:102
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
int av_get_bits_per_sample(enum AVCodecID codec_id)
Return codec bits per sample.
Definition: utils.c:3318
AVDictionary * metadata
Metadata that applies to the whole file.
Definition: avformat.h:1482
av_default_item_name
unsigned int avio_rl32(AVIOContext *s)
Definition: aviobuf.c:658
#define AVERROR(e)
Definition: error.h:43
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:175
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:197
preferred ID for decoding MPEG audio layer 1, 2 or 3
Definition: avcodec.h:421
simple assert() macros that are a bit more flexible than ISO C assert().
GLsizei GLsizei * length
Definition: opengl_enc.c:115
int smv_block_size
Definition: wavdec.c:49
int smv_cur_pt
Definition: wavdec.c:57
static const uint8_t offset[127][2]
Definition: vf_spp.c:92
GLsizei count
Definition: opengl_enc.c:109
const AVCodecTag ff_codec_wav_tags[]
Definition: riff.c:376
int av_compare_ts(int64_t ts_a, AVRational tb_a, int64_t ts_b, AVRational tb_b)
Compare 2 timestamps each in its own timebases.
Definition: mathematics.c:145
int avio_r8(AVIOContext *s)
Definition: aviobuf.c:528
return
AVCodecContext * codec
Codec context associated with this stream.
Definition: avformat.h:861
int buf_size
Size of buf except extra allocated bytes.
Definition: avformat.h:451
unsigned char * buf
Buffer must have AVPROBE_PADDING_SIZE of extra allocated bytes filled with zero.
Definition: avformat.h:450
int seekable
A combination of AVIO_SEEKABLE_ flags or 0 when the stream is not seekable.
Definition: avio.h:160
#define FFMIN(a, b)
Definition: common.h:66
int smv_last_stream
Definition: wavdec.c:52
int av_get_exact_bits_per_sample(enum AVCodecID codec_id)
Return codec bits per sample.
Definition: utils.c:3243
#define AV_DICT_DONT_STRDUP_VAL
Take ownership of a value that's been allocated with av_malloc() or another memory allocation functio...
Definition: dict.h:78
static int read_probe(AVProbeData *pd)
Definition: jvdec.c:55
ret
Definition: avfilter.c:974
int width
picture width / height.
Definition: avcodec.h:1414
GLsizei GLboolean const GLfloat * value
Definition: opengl_enc.c:109
const uint8_t ff_w64_guid_data[16]
Definition: w64.c:42
internal header for RIFF based (de)muxers do NOT include this in end user applications ...
int32_t
const uint8_t ff_w64_guid_riff[16]
Definition: w64.c:23
int audio_eof
Definition: wavdec.c:54
int ff_pcm_read_seek(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
Definition: pcm.c:45
static int read_header(FFV1Context *f)
Definition: ffv1dec.c:620
Stream structure.
Definition: avformat.h:842
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
Definition: avio_reading.c:42
enum AVMediaType codec_type
Definition: avcodec.h:1249
enum AVCodecID codec_id
Definition: avcodec.h:1258
int sample_rate
samples per second
Definition: avcodec.h:1985
AVIOContext * pb
I/O context.
Definition: avformat.h:1314
unsigned int codec_tag
fourcc (LSB first, so "ABCD" -> ('D'<<24) + ('C'<<16) + ('B'<<8) + 'A').
Definition: avcodec.h:1273
int ff_alloc_extradata(AVCodecContext *avctx, int size)
Allocate extradata with additional FF_INPUT_BUFFER_PADDING_SIZE at end which is always set to 0...
Definition: utils.c:2864
int av_dict_set(AVDictionary **pm, const char *key, const char *value, int flags)
Set the given entry in *pm, overwriting an existing entry.
Definition: dict.c:69
Describe the class of an AVClass context structure.
Definition: log.h:67
#define AVFMT_GENERIC_INDEX
Use generic index building code.
Definition: avformat.h:473
#define DEC
Definition: tta.c:405
int64_t smv_data_ofs
Definition: wavdec.c:48
#define snprintf
Definition: snprintf.h:34
#define AVPROBE_SCORE_EXTENSION
score for file extension
Definition: avformat.h:458
This structure contains the data a format has to probe a file.
Definition: avformat.h:448
static const uint8_t start_code[]
Definition: h264.c:1264
const AVMetadataConv ff_riff_info_conv[]
Definition: riff.c:453
static int flags
Definition: cpu.c:47
const uint8_t ff_w64_guid_fmt[16]
Definition: w64.c:33
#define OFFSET(x)
Definition: ffmpeg_opt.c:2874
int64_t duration
Decoding: duration of the stream, in stream time base.
Definition: avformat.h:901
#define AVPROBE_SCORE_MAX
maximum score
Definition: avformat.h:460
int smv_given_first
Definition: wavdec.c:58
unsigned int avio_rl16(AVIOContext *s)
Definition: aviobuf.c:642
Main libavformat public API header.
#define FF_PRI_GUID
Definition: riff.h:93
int64_t data_end
Definition: wavdec.c:46
int smv_block
Definition: wavdec.c:51
int channels
number of audio channels
Definition: avcodec.h:1986
void * priv_data
Format private data.
Definition: avformat.h:1300
const uint8_t ff_w64_guid_summarylist[16]
Definition: w64.c:47
#define av_uninit(x)
Definition: attributes.h:141
void INT64 start
Definition: avisynth_c.h:553
const char * name
A comma separated list of short names for the format.
Definition: avformat.h:628
int avio_feof(AVIOContext *s)
feof() equivalent for AVIOContext.
Definition: aviobuf.c:300
unsigned int avio_rl24(AVIOContext *s)
Definition: aviobuf.c:650
int stream_index
Definition: avcodec.h:1164
AVRational time_base
This is the fundamental unit of time (in seconds) in terms of which frame timestamps are represented...
Definition: avformat.h:884
#define MKTAG(a, b, c, d)
Definition: common.h:315
int request_probe
stream probing state -1 -> probing finished 0 -> no probing requested rest -> perform probing with re...
Definition: avformat.h:1081
This structure stores compressed data.
Definition: avcodec.h:1139
uint64_t avio_rl64(AVIOContext *s)
Definition: aviobuf.c:666
void * av_mallocz(size_t size)
Allocate a block of size bytes with alignment suitable for all memory accesses (including vectors if ...
Definition: mem.c:250
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: avcodec.h:1155
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:241
int unaligned
Definition: wavdec.c:59
#define AV_WL32(p, v)
Definition: intreadwrite.h:426