FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/get_buffer.c
Date: 2023-09-22 12:20:07
Exec Total Coverage
Lines: 107 155 69.0%
Functions: 6 6 100.0%
Branches: 70 102 68.6%
Line Branch Exec Source
1 /*
2 * The default get_buffer2() implementation
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include <stdint.h>
22
23 #include "libavutil/avassert.h"
24 #include "libavutil/avutil.h"
25 #include "libavutil/buffer.h"
26 #include "libavutil/frame.h"
27 #include "libavutil/hwcontext.h"
28 #include "libavutil/imgutils.h"
29 #include "libavutil/mem.h"
30 #include "libavutil/samplefmt.h"
31 #include "libavutil/version.h"
32
33 #include "avcodec.h"
34 #include "internal.h"
35
36 typedef struct FramePool {
37 /**
38 * Pools for each data plane. For audio all the planes have the same size,
39 * so only pools[0] is used.
40 */
41 AVBufferPool *pools[4];
42
43 /*
44 * Pool parameters
45 */
46 int format;
47 int width, height;
48 int stride_align[AV_NUM_DATA_POINTERS];
49 int linesize[4];
50 int planes;
51 int channels;
52 int samples;
53 } FramePool;
54
55 14949 static void frame_pool_free(void *opaque, uint8_t *data)
56 {
57 14949 FramePool *pool = (FramePool*)data;
58 int i;
59
60
2/2
✓ Branch 0 taken 59796 times.
✓ Branch 1 taken 14949 times.
 74745 for (i = 0; i < FF_ARRAY_ELEMS(pool->pools); i++)
61 59796 av_buffer_pool_uninit(&pool->pools[i]);
62
63 14949 av_freep(&data);
64 14949 }
65
66 14949 static AVBufferRef *frame_pool_alloc(void)
67 {
68 14949 FramePool *pool = av_mallocz(sizeof(*pool));
69 AVBufferRef *buf;
70
71
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14949 times.
 14949 if (!pool)
72 ✗ return NULL;
73
74 14949 buf = av_buffer_create((uint8_t*)pool, sizeof(*pool),
75 frame_pool_free, NULL, 0);
76
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14949 times.
 14949 if (!buf) {
77 ✗ av_freep(&pool);
78 ✗ return NULL;
79 }
80
81 14949 return buf;
82 }
83
84 394224 static int update_frame_pool(AVCodecContext *avctx, AVFrame *frame)
85 {
86 788448 FramePool *pool = avctx->internal->pool ?
87
2/2
✓ Branch 0 taken 386236 times.
✓ Branch 1 taken 7988 times.
 394224 (FramePool*)avctx->internal->pool->data : NULL;
88 AVBufferRef *pool_buf;
89 int i, ret, ch, planes;
90
91
2/2
✓ Branch 0 taken 303126 times.
✓ Branch 1 taken 91098 times.
 394224 if (avctx->codec_type == AVMEDIA_TYPE_AUDIO) {
92 303126 int planar = av_sample_fmt_is_planar(frame->format);
93 303126 ch = frame->ch_layout.nb_channels;
94 #if FF_API_OLD_CHANNEL_LAYOUT
95 FF_DISABLE_DEPRECATION_WARNINGS
96
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 303126 times.
 303126 if (!ch)
97 ✗ ch = frame->channels;
98 FF_ENABLE_DEPRECATION_WARNINGS
99 #endif
100
2/2
✓ Branch 0 taken 162258 times.
✓ Branch 1 taken 140868 times.
 303126 planes = planar ? ch : 1;
101 }
102
103
4/4
✓ Branch 0 taken 386236 times.
✓ Branch 1 taken 7988 times.
✓ Branch 2 taken 386056 times.
✓ Branch 3 taken 180 times.
 394224 if (pool && pool->format == frame->format) {
104
2/2
✓ Branch 0 taken 84785 times.
✓ Branch 1 taken 301271 times.
 386056 if (avctx->codec_type == AVMEDIA_TYPE_VIDEO &&
105
4/4
✓ Branch 0 taken 84562 times.
✓ Branch 1 taken 223 times.
✓ Branch 2 taken 84551 times.
✓ Branch 3 taken 11 times.
 84785 pool->width == frame->width && pool->height == frame->height)
106 84551 return 0;
107
4/4
✓ Branch 0 taken 301271 times.
✓ Branch 1 taken 234 times.
✓ Branch 2 taken 301258 times.
✓ Branch 3 taken 13 times.
 301505 if (avctx->codec_type == AVMEDIA_TYPE_AUDIO && pool->planes == planes &&
108
3/4
✓ Branch 0 taken 301258 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 294724 times.
✓ Branch 3 taken 6534 times.
 301258 pool->channels == ch && frame->nb_samples == pool->samples)
109 294724 return 0;
110 }
111
112 14949 pool_buf = frame_pool_alloc();
113
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14949 times.
 14949 if (!pool_buf)
114 ✗ return AVERROR(ENOMEM);
115 14949 pool = (FramePool*)pool_buf->data;
116
117
2/3
✓ Branch 0 taken 6547 times.
✓ Branch 1 taken 8402 times.
✗ Branch 2 not taken.
 14949 switch (avctx->codec_type) {
118 6547 case AVMEDIA_TYPE_VIDEO: {
119 int linesize[4];
120 6547 int w = frame->width;
121 6547 int h = frame->height;
122 int unaligned;
123 ptrdiff_t linesize1[4];
124 size_t size[4];
125
126 6547 avcodec_align_dimensions2(avctx, &w, &h, pool->stride_align);
127
128 do {
129 // NOTE: do not align linesizes individually, this breaks e.g. assumptions
130 // that linesize[0] == 2*linesize[1] in the MPEG-encoder for 4:2:2
131 13083 ret = av_image_fill_linesizes(linesize, avctx->pix_fmt, w);
132
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13083 times.
 13083 if (ret < 0)
133 ✗ goto fail;
134 // increase alignment of w for next try (rhs gives the lowest bit set in w)
135 13083 w += w & ~(w - 1);
136
137 13083 unaligned = 0;
138
2/2
✓ Branch 0 taken 52332 times.
✓ Branch 1 taken 13083 times.
 65415 for (i = 0; i < 4; i++)
139 52332 unaligned |= linesize[i] % pool->stride_align[i];
140
2/2
✓ Branch 0 taken 6536 times.
✓ Branch 1 taken 6547 times.
 13083 } while (unaligned);
141
142
2/2
✓ Branch 0 taken 26188 times.
✓ Branch 1 taken 6547 times.
 32735 for (i = 0; i < 4; i++)
143 26188 linesize1[i] = linesize[i];
144 6547 ret = av_image_fill_plane_sizes(size, avctx->pix_fmt, h, linesize1);
145
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6547 times.
 6547 if (ret < 0)
146 ✗ goto fail;
147
148
2/2
✓ Branch 0 taken 26188 times.
✓ Branch 1 taken 6547 times.
 32735 for (i = 0; i < 4; i++) {
149 26188 pool->linesize[i] = linesize[i];
150
2/2
✓ Branch 0 taken 18106 times.
✓ Branch 1 taken 8082 times.
 26188 if (size[i]) {
151
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18106 times.
 18106 if (size[i] > INT_MAX - (16 + STRIDE_ALIGN - 1)) {
152 ✗ ret = AVERROR(EINVAL);
153 ✗ goto fail;
154 }
155 18106 pool->pools[i] = av_buffer_pool_init(size[i] + 16 + STRIDE_ALIGN - 1,
156 CONFIG_MEMORY_POISONING ?
157 NULL :
158 av_buffer_allocz);
159
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18106 times.
 18106 if (!pool->pools[i]) {
160 ✗ ret = AVERROR(ENOMEM);
161 ✗ goto fail;
162 }
163 }
164 }
165 6547 pool->format = frame->format;
166 6547 pool->width = frame->width;
167 6547 pool->height = frame->height;
168
169 6547 break;
170 }
171 8402 case AVMEDIA_TYPE_AUDIO: {
172 8402 ret = av_samples_get_buffer_size(&pool->linesize[0], ch,
173 8402 frame->nb_samples, frame->format, 0);
174
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8402 times.
 8402 if (ret < 0)
175 ✗ goto fail;
176
177 8402 pool->pools[0] = av_buffer_pool_init(pool->linesize[0], NULL);
178
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8402 times.
 8402 if (!pool->pools[0]) {
179 ✗ ret = AVERROR(ENOMEM);
180 ✗ goto fail;
181 }
182
183 8402 pool->format = frame->format;
184 8402 pool->planes = planes;
185 8402 pool->channels = ch;
186 8402 pool->samples = frame->nb_samples;
187 8402 break;
188 }
189 ✗ default: av_assert0(0);
190 }
191
192 14949 av_buffer_unref(&avctx->internal->pool);
193 14949 avctx->internal->pool = pool_buf;
194
195 14949 return 0;
196 ✗ fail:
197 ✗ av_buffer_unref(&pool_buf);
198 ✗ return ret;
199 }
200
201 303126 static int audio_get_buffer(AVCodecContext *avctx, AVFrame *frame)
202 {
203 303126 FramePool *pool = (FramePool*)avctx->internal->pool->data;
204 303126 int planes = pool->planes;
205 int i;
206
207 303126 frame->linesize[0] = pool->linesize[0];
208
209
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 303126 times.
 303126 if (planes > AV_NUM_DATA_POINTERS) {
210 ✗ frame->extended_data = av_calloc(planes, sizeof(*frame->extended_data));
211 ✗ frame->nb_extended_buf = planes - AV_NUM_DATA_POINTERS;
212 ✗ frame->extended_buf = av_calloc(frame->nb_extended_buf,
213 sizeof(*frame->extended_buf));
214 ✗ if (!frame->extended_data || !frame->extended_buf) {
215 ✗ av_freep(&frame->extended_data);
216 ✗ av_freep(&frame->extended_buf);
217 ✗ return AVERROR(ENOMEM);
218 }
219 } else {
220 303126 frame->extended_data = frame->data;
221
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 303126 times.
 303126 av_assert0(frame->nb_extended_buf == 0);
222 }
223
224
2/2
✓ Branch 0 taken 447446 times.
✓ Branch 1 taken 303126 times.
 750572 for (i = 0; i < FFMIN(planes, AV_NUM_DATA_POINTERS); i++) {
225 447446 frame->buf[i] = av_buffer_pool_get(pool->pools[0]);
226
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 447446 times.
 447446 if (!frame->buf[i])
227 ✗ goto fail;
228 447446 frame->extended_data[i] = frame->data[i] = frame->buf[i]->data;
229 }
230
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 303126 times.
 303126 for (i = 0; i < frame->nb_extended_buf; i++) {
231 ✗ frame->extended_buf[i] = av_buffer_pool_get(pool->pools[0]);
232 ✗ if (!frame->extended_buf[i])
233 ✗ goto fail;
234 ✗ frame->extended_data[i + AV_NUM_DATA_POINTERS] = frame->extended_buf[i]->data;
235 }
236
237
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 303126 times.
 303126 if (avctx->debug & FF_DEBUG_BUFFERS)
238 ✗ av_log(avctx, AV_LOG_DEBUG, "default_get_buffer called on frame %p", frame);
239
240 303126 return 0;
241 ✗ fail:
242 ✗ av_frame_unref(frame);
243 ✗ return AVERROR(ENOMEM);
244 }
245
246 91098 static int video_get_buffer(AVCodecContext *s, AVFrame *pic)
247 {
248 91098 FramePool *pool = (FramePool*)s->internal->pool->data;
249 int i;
250
251
4/8
✓ Branch 0 taken 91098 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 91098 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 91098 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 91098 times.
 91098 if (pic->data[0] || pic->data[1] || pic->data[2] || pic->data[3]) {
252 ✗ av_log(s, AV_LOG_ERROR, "pic->data[*]!=NULL in avcodec_default_get_buffer\n");
253 ✗ return -1;
254 }
255
256 91098 memset(pic->data, 0, sizeof(pic->data));
257 91098 pic->extended_data = pic->data;
258
259
4/4
✓ Branch 0 taken 349887 times.
✓ Branch 1 taken 889 times.
✓ Branch 2 taken 259678 times.
✓ Branch 3 taken 90209 times.
 350776 for (i = 0; i < 4 && pool->pools[i]; i++) {
260 259678 pic->linesize[i] = pool->linesize[i];
261
262 259678 pic->buf[i] = av_buffer_pool_get(pool->pools[i]);
263
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 259678 times.
 259678 if (!pic->buf[i])
264 ✗ goto fail;
265
266 259678 pic->data[i] = pic->buf[i]->data;
267 }
268
2/2
✓ Branch 0 taken 469106 times.
✓ Branch 1 taken 91098 times.
 560204 for (; i < AV_NUM_DATA_POINTERS; i++) {
269 469106 pic->data[i] = NULL;
270 469106 pic->linesize[i] = 0;
271 }
272
273
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 91098 times.
 91098 if (s->debug & FF_DEBUG_BUFFERS)
274 ✗ av_log(s, AV_LOG_DEBUG, "default_get_buffer called on pic %p\n", pic);
275
276 91098 return 0;
277 ✗ fail:
278 ✗ av_frame_unref(pic);
279 ✗ return AVERROR(ENOMEM);
280 }
281
282 394224 int avcodec_default_get_buffer2(AVCodecContext *avctx, AVFrame *frame, int flags)
283 {
284 int ret;
285
286
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 394224 times.
 394224 if (avctx->hw_frames_ctx) {
287 ✗ ret = av_hwframe_get_buffer(avctx->hw_frames_ctx, frame, 0);
288 ✗ frame->width = avctx->coded_width;
289 ✗ frame->height = avctx->coded_height;
290 ✗ return ret;
291 }
292
293
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 394224 times.
 394224 if ((ret = update_frame_pool(avctx, frame)) < 0)
294 ✗ return ret;
295
296
2/3
✓ Branch 0 taken 91098 times.
✓ Branch 1 taken 303126 times.
✗ Branch 2 not taken.
 394224 switch (avctx->codec_type) {
297 91098 case AVMEDIA_TYPE_VIDEO:
298 91098 return video_get_buffer(avctx, frame);
299 303126 case AVMEDIA_TYPE_AUDIO:
300 303126 return audio_get_buffer(avctx, frame);
301 ✗ default:
302 ✗ return -1;
303 }
304 }
305