diff options
| author | Andreas Rheinhardt <andreas.rheinhardt@gmail.com> | 2019-11-18 08:47:58 +0100 |
|---|---|---|
| committer | Mark Thompson <sw@jkqxz.net> | 2020-02-09 22:23:29 +0000 |
| commit | ac5d5046c8ef4988b36734effe42a2fae8547ce1 (patch) | |
| tree | da35e39aebdf8cf76f08ab4df6797e84a0c433fa /libavcodec/cbs.c | |
| parent | 1e6cef686183288a50eb4e45265a3a7f7582b1b5 (diff) | |
avcodec/cbs: Fix potential double-free when adding unit fails
ff_cbs_insert_unit_data() has two modes of operation: It can insert a
unit with a newly created reference to an already existing AVBuffer; or
it can take a buffer and create an AVBuffer for it. Said buffer will
then become owned by the unit lateron.
A potential memleak/double-free exists in the second case, because if
creating the AVBuffer fails, the function immediately returns, but when
it fails lateron, the supplied buffer will be freed. The caller has no
way to distinguish between these two outcomes. The only such caller
(cbs_jpeg_split_fragment() in cbs_jpeg.c) opted for a potential
double-free.
This commit changes this by explicitly stating that a non-refcounted
buffer will be freed on error. The aforementioned caller has been
brought in line with this.
Fixes CID 1452623.
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@gmail.com>
Diffstat (limited to 'libavcodec/cbs.c')
| -rw-r--r-- | libavcodec/cbs.c | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/libavcodec/cbs.c b/libavcodec/cbs.c index 0badb192d9..0bd5e1ac5d 100644 --- a/libavcodec/cbs.c +++ b/libavcodec/cbs.c @@ -775,8 +775,11 @@ int ff_cbs_insert_unit_data(CodedBitstreamContext *ctx, data_ref = av_buffer_ref(data_buf); else data_ref = av_buffer_create(data, data_size, NULL, NULL, 0); - if (!data_ref) + if (!data_ref) { + if (!data_buf) + av_free(data); return AVERROR(ENOMEM); + } err = cbs_insert_unit(ctx, frag, position); if (err < 0) { |