diff mbox series

[FFmpeg-devel] avformat/riffdec: fix support for WAVEFORMAT

Message ID 20220725093647.4417-1-tom.ty89@gmail.com
State New
Headers show
Series [FFmpeg-devel] avformat/riffdec: fix support for WAVEFORMAT | expand

Checks

Context Check Description
yinshiyou/make_loongarch64 success Make finished
yinshiyou/make_fate_loongarch64 success Make fate finished
andriy/make_x86 success Make finished
andriy/make_fate_x86 success Make fate finished

Commit Message

Tom Yan July 25, 2022, 9:36 a.m. UTC
WAVEFORMAT can be used for 16-bit PCM as well.

Signed-off-by: Tom Yan <tom.ty89@gmail.com>
---
 libavformat/riffdec.c | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)
diff mbox series

Patch

diff --git a/libavformat/riffdec.c b/libavformat/riffdec.c
index 3946ecb72f..8289438f08 100644
--- a/libavformat/riffdec.c
+++ b/libavformat/riffdec.c
@@ -94,7 +94,7 @@  static void parse_waveformatex(AVFormatContext *s, AVIOContext *pb, AVCodecParam
 int ff_get_wav_header(AVFormatContext *s, AVIOContext *pb,
                       AVCodecParameters *par, int size, int big_endian)
 {
-    int id, channels;
+    int id, channels = 0;
     uint64_t bitrate = 0;
 
     if (size < 14) {
@@ -118,15 +118,15 @@  int ff_get_wav_header(AVFormatContext *s, AVIOContext *pb,
         bitrate            = avio_rb32(pb) * 8LL;
         par->block_align = avio_rb16(pb);
     }
-    if (size == 14) {  /* We're dealing with plain vanilla WAVEFORMAT */
-        par->bits_per_coded_sample = 8;
-    } else {
+    if (size >= 16) {  /* We're dealing with PCMWAVEFORMAT */
         if (!big_endian) {
             par->bits_per_coded_sample = avio_rl16(pb);
         } else {
             par->bits_per_coded_sample = avio_rb16(pb);
         }
-    }
+    } else if (channels && par->sample_rate) {  /* We're dealing with WAVEFORMAT */
+        par->bits_per_coded_sample = bitrate / channels / par->sample_rate;
+    }  /* Apparently size would be >= 32 if id == 0x0165; see below */
     if (id == 0xFFFE) {
         par->codec_tag = 0;
     } else {