[FFmpeg-devel] [PATCH 2/2] examples/decoding_encoding: check av_samples_get_buffer_size() return code

[Timothy Gu]

Fixes CID1135756.

Signed-off-by: Timothy Gu <timothygu99 at gmail.com>
---
 doc/examples/decoding_encoding.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/doc/examples/decoding_encoding.c b/doc/examples/decoding_encoding.c
index 8bfc30f..4d2813b 100644
--- a/doc/examples/decoding_encoding.c
+++ b/doc/examples/decoding_encoding.c
@@ -170,6 +170,10 @@ static void audio_encode_example(const char *filename)
      * we calculate the size of the samples buffer in bytes */
     buffer_size = av_samples_get_buffer_size(NULL, c->channels, c->frame_size,
                                              c->sample_fmt, 0);
+    if (!buffer_size) {
+        fprintf(stderr, "Could not get sample buffer size\n");
+        exit(1);
+    }
     samples = av_malloc(buffer_size);
     if (!samples) {
         fprintf(stderr, "Could not allocate %d bytes for samples buffer\n",
-- 
1.8.3.2