avformat/flvdec: Check for avio_read() failure in amf_get_string()

Suggested-by: Anton Khirnov <[email protected]>
Signed-off-by: Michael Niedermayer <[email protected]>
(cherry picked from commit cb316676112c01e8d66420908b6b3d06b3b498e3)
Signed-off-by: Michael Niedermayer <[email protected]>

1 files changed, 6 insertions, 1 deletions

diff --git a/libavformat/flvdec.c b/libavformat/flvdec.c
index b88c26f659..f9437b9e81 100644
--- a/libavformat/flvdec.c
+++ b/libavformat/flvdec.c
@@ -287,13 +287,18 @@ static int flv_set_video_codec(AVFormatContext *s, AVStream *vstream,
 
 static int amf_get_string(AVIOContext *ioc, char *buffer, int buffsize)
 {
+    int ret;
     int length = avio_rb16(ioc);
     if (length >= buffsize) {
         avio_skip(ioc, length);
         return -1;
     }
 
-    avio_read(ioc, buffer, length);
+    ret = avio_read(ioc, buffer, length);
+    if (ret < 0)
+        return ret;
+    if (ret < length)
+        return AVERROR_INVALIDDATA;
 
     buffer[length] = '\0';