diff mbox

[FFmpeg-devel] avformat/fitsenc: validate input pixel format

Message ID 20171113101609.4224-1-timo@rothenpieler.org
State Accepted
Commit 284b432662b6e137148ff9d13ef2b554cb14b4ae
Headers show

Commit Message

Timo Rothenpieler Nov. 13, 2017, 10:16 a.m. UTC 
Fixes CID #1416961 and #1416962
---
 libavformat/fitsenc.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

Comments

Timo Rothenpieler Nov. 13, 2017, 7:51 p.m. UTC | #1 
applied
diff mbox

Patch

diff --git a/libavformat/fitsenc.c b/libavformat/fitsenc.c
index 7cb171596c..cc3999aa8a 100644
--- a/libavformat/fitsenc.c
+++ b/libavformat/fitsenc.c
@@ -106,6 +106,8 @@  static int write_image_header(AVFormatContext *s)
             }
             bzero = 32768;
             break;
+        default:
+            return AVERROR(EINVAL);
     }
 
     if (fitsctx->first_image) {
@@ -166,7 +168,9 @@  static int write_image_header(AVFormatContext *s)
 
 static int fits_write_packet(AVFormatContext *s, AVPacket *pkt)
 {
-    write_image_header(s);
+    int ret = write_image_header(s);
+    if (ret < 0)
+        return ret;
     avio_write(s->pb, pkt->data, pkt->size);
     return 0;
 }