[FFmpeg-cvslog] avformat/au: improve au_probe()

[Paul B Mahol]

ffmpeg | branch: master | Paul B Mahol <onemda at gmail.com> | Fri Aug 11 23:48:39 2023 +0200| [97741adf6f2f97651698a8c8f3d5fc3587a2d984] | committer: Paul B Mahol

avformat/au: improve au_probe()

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=97741adf6f2f97651698a8c8f3d5fc3587a2d984
---

 libavformat/au.c | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/libavformat/au.c b/libavformat/au.c
index 3bf2150258..63dfd71d0f 100644
--- a/libavformat/au.c
+++ b/libavformat/au.c
@@ -30,6 +30,7 @@
 #include "config_components.h"
 
 #include "libavutil/bprint.h"
+#include "libavutil/intreadwrite.h"
 #include "avformat.h"
 #include "internal.h"
 #include "avio_internal.h"
@@ -63,11 +64,15 @@ static const AVCodecTag *const au_codec_tags[] = { codec_au_tags, NULL };
 
 static int au_probe(const AVProbeData *p)
 {
-    if (p->buf[0] == '.' && p->buf[1] == 's' &&
-        p->buf[2] == 'n' && p->buf[3] == 'd')
-        return AVPROBE_SCORE_MAX;
-    else
+    if (p->buf_size < 24 ||
+        AV_RL32(p->buf) != MKTAG('.', 's', 'n', 'd') ||
+        AV_RN32(p->buf+4)  == 0 ||
+        AV_RN32(p->buf+8)  == 0 ||
+        AV_RN32(p->buf+12) == 0 ||
+        AV_RN32(p->buf+16) == 0 ||
+        AV_RN32(p->buf+20) == 0)
         return 0;
+    return AVPROBE_SCORE_MAX;
 }
 
 static int au_read_annotation(AVFormatContext *s, int size)