From 74699b63e77bae3fce7ccb8569fa5de2c7868e6a Mon Sep 17 00:00:00 2001 From: Devaev Maxim Date: Thu, 27 Sep 2018 05:08:43 +0300 Subject: [PATCH] some asserts --- src/jpeg/encoder.c | 1 + src/omx/encoder.c | 1 + 2 files changed, 2 insertions(+) diff --git a/src/jpeg/encoder.c b/src/jpeg/encoder.c index adc2351..005f2ac 100644 --- a/src/jpeg/encoder.c +++ b/src/jpeg/encoder.c @@ -110,6 +110,7 @@ void jpeg_encoder_compress_buffer(struct device_t *dev, int index) { jpeg_destroy_compress(&jpeg); free(line_buffer); assert(dev->run->pictures[index].size > 0); + assert(dev->run->pictures[index].size <= dev->run->max_picture_size); } static void _jpeg_set_dest_picture(j_compress_ptr jpeg, unsigned char *picture, unsigned long *written) { diff --git a/src/omx/encoder.c b/src/omx/encoder.c index 326ef92..ed42463 100644 --- a/src/omx/encoder.c +++ b/src/omx/encoder.c @@ -179,6 +179,7 @@ int omx_encoder_compress_buffer(struct omx_encoder_t *omx, struct device_t *dev, omx->output_buffer->pBuffer, omx->output_buffer->nFilledLen ); + assert(dev->run->pictures[index].size + omx->output_buffer->nFilledLen <= dev->run->max_picture_size); dev->run->pictures[index].size += omx->output_buffer->nFilledLen; if (omx->output_buffer->nFlags & OMX_BUFFERFLAG_ENDOFFRAME) {