From c8778606b3811da6bd58ca6b73d2446bd430013e Mon Sep 17 00:00:00 2001 From: Anton Khirnov Date: Thu, 8 Apr 2021 11:16:06 +0200 Subject: [PATCH] lavu/video_enc_params: make sure blocks are properly aligned --- libavutil/video_enc_params.c | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/libavutil/video_enc_params.c b/libavutil/video_enc_params.c index b9cdafddbb..7d903af385 100644 --- a/libavutil/video_enc_params.c +++ b/libavutil/video_enc_params.c @@ -29,10 +29,14 @@ AVVideoEncParams *av_video_enc_params_alloc(enum AVVideoEncParamsType type, unsigned int nb_blocks, size_t *out_size) { + const size_t blocks_offset = offsetof( + struct { + AVVideoEncParams p; + AVVideoBlockParams b; + }, b); + size_t size = blocks_offset; AVVideoEncParams *par; - size_t size; - size = sizeof(*par); if (nb_blocks > (SIZE_MAX - size) / sizeof(AVVideoBlockParams)) return NULL; size += sizeof(AVVideoBlockParams) * nb_blocks; @@ -44,7 +48,7 @@ AVVideoEncParams *av_video_enc_params_alloc(enum AVVideoEncParamsType type, par->type = type; par->nb_blocks = nb_blocks; par->block_size = sizeof(AVVideoBlockParams); - par->blocks_offset = sizeof(*par); + par->blocks_offset = blocks_offset; if (out_size) *out_size = size;