FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavutil/container_fifo.c
Date: 2026-04-22 13:46:37
Exec Total Coverage
Lines: 82 96 85.4%
Functions: 15 15 100.0%
Branches: 21 32 65.6%
Line Branch Exec Source
1 /*
2 * This file is part of FFmpeg.
3 *
4 * FFmpeg is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * FFmpeg is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with FFmpeg; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19 #include "avassert.h"
20 #include "container_fifo.h"
21 #include "error.h"
22 #include "fifo.h"
23 #include "frame.h"
24 #include "mem.h"
25 #include "refstruct.h"
26
27 struct AVContainerFifo {
28 AVFifo *fifo;
29 AVRefStructPool *pool;
30
31 void *opaque;
32 void* (*container_alloc)(void *opaque);
33 void (*container_reset)(void *opaque, void *obj);
34 void (*container_free) (void *opaque, void *obj);
35 int (*fifo_transfer) (void *opaque, void *dst, void *src, unsigned flags);
36
37 };
38
39 96605 static int container_fifo_init_entry(AVRefStructOpaque opaque, void *obj)
40 {
41 96605 AVContainerFifo *cf = opaque.nc;
42 96605 void **pobj = obj;
43
44 96605 *pobj = cf->container_alloc(cf->opaque);
45
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 96605 times.
 96605 if (!*pobj)
46 ✗ return AVERROR(ENOMEM);
47
48 96605 return 0;
49 }
50
51 2017145 static void container_fifo_reset_entry(AVRefStructOpaque opaque, void *obj)
52 {
53 2017145 AVContainerFifo *cf = opaque.nc;
54 2017145 cf->container_reset(cf->opaque, *(void**)obj);
55 2017145 }
56
57 96605 static void container_fifo_free_entry(AVRefStructOpaque opaque, void *obj)
58 {
59 96605 AVContainerFifo *cf = opaque.nc;
60 96605 cf->container_free(cf->opaque, *(void**)obj);
61 96605 }
62
63 AVContainerFifo*
64 35787 av_container_fifo_alloc(void *opaque,
65 void* (*container_alloc)(void *opaque),
66 void (*container_reset)(void *opaque, void *obj),
67 void (*container_free) (void *opaque, void *obj),
68 int (*fifo_transfer) (void *opaque, void *dst, void *src, unsigned flags),
69 unsigned flags)
70 {
71 AVContainerFifo *cf;
72
73 35787 cf = av_mallocz(sizeof(*cf));
74
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 35787 times.
 35787 if (!cf)
75 ✗ return NULL;
76
77 35787 cf->opaque = opaque;
78 35787 cf->container_alloc = container_alloc;
79 35787 cf->container_reset = container_reset;
80 35787 cf->container_free = container_free;
81 35787 cf->fifo_transfer = fifo_transfer;
82
83 35787 cf->fifo = av_fifo_alloc2(1, sizeof(void*), AV_FIFO_FLAG_AUTO_GROW);
84
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 35787 times.
 35787 if (!cf->fifo)
85 ✗ goto fail;
86
87 35787 cf->pool = av_refstruct_pool_alloc_ext(sizeof(void*), 0, cf,
88 container_fifo_init_entry,
89 container_fifo_reset_entry,
90 container_fifo_free_entry,
91 NULL);
92
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 35787 times.
 35787 if (!cf->pool)
93 ✗ goto fail;
94
95 35787 return cf;
96 ✗ fail:
97 ✗ av_container_fifo_free(&cf);
98 ✗ return NULL;
99 }
100
101 35787 void av_container_fifo_free(AVContainerFifo **pcf)
102 {
103 AVContainerFifo *cf;
104
105
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 35787 times.
 35787 if (!*pcf)
106 ✗ return;
107
108 35787 cf = *pcf;
109
110
1/2
✓ Branch 0 taken 35787 times.
✗ Branch 1 not taken.
 35787 if (cf->fifo) {
111 void *obj;
112
2/2
✓ Branch 1 taken 35068 times.
✓ Branch 2 taken 35787 times.
 70855 while (av_fifo_read(cf->fifo, &obj, 1) >= 0)
113 35068 av_refstruct_unref(&obj);
114 35787 av_fifo_freep2(&cf->fifo);
115 }
116
117 35787 av_refstruct_pool_uninit(&cf->pool);
118
119 35787 av_freep(pcf);
120 }
121
122 3028697 int av_container_fifo_read(AVContainerFifo *cf, void *obj, unsigned flags)
123 {
124 void **psrc;
125 int ret;
126
127 3028697 ret = av_fifo_read(cf->fifo, &psrc, 1);
128
2/2
✓ Branch 0 taken 1063074 times.
✓ Branch 1 taken 1965623 times.
 3028697 if (ret < 0)
129 1063074 return ret;
130
131 1965623 ret = cf->fifo_transfer(cf->opaque, obj, *psrc, flags);
132 1965623 av_refstruct_unref(&psrc);
133
134 1965623 return ret;
135 }
136
137 282279 int av_container_fifo_peek(AVContainerFifo *cf, void **pdst, size_t offset)
138 {
139 void **pobj;
140 int ret;
141
142 282279 ret = av_fifo_peek(cf->fifo, &pobj, 1, offset);
143
2/2
✓ Branch 0 taken 15518 times.
✓ Branch 1 taken 266761 times.
 282279 if (ret < 0)
144 15518 return ret;
145
146 266761 *pdst = *pobj;
147
148 266761 return 0;
149 }
150
151 16454 void av_container_fifo_drain(AVContainerFifo *cf, size_t nb_elems)
152 {
153
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 16454 times.
 16454 av_assert0(nb_elems <= av_fifo_can_read(cf->fifo));
154
2/2
✓ Branch 0 taken 16454 times.
✓ Branch 1 taken 16454 times.
 32908 while (nb_elems--) {
155 void **pobj;
156 16454 int ret = av_fifo_read(cf->fifo, &pobj, 1);
157
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16454 times.
 16454 av_assert0(ret >= 0);
158 16454 av_refstruct_unref(&pobj);
159 }
160 16454 }
161
162 2017145 int av_container_fifo_write(AVContainerFifo *cf, void *obj, unsigned flags)
163 {
164 void **pdst;
165 int ret;
166
167 2017145 pdst = av_refstruct_pool_get(cf->pool);
168
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2017145 times.
 2017145 if (!pdst)
169 ✗ return AVERROR(ENOMEM);
170
171 2017145 ret = cf->fifo_transfer(cf->opaque, *pdst, obj, flags);
172
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2017145 times.
 2017145 if (ret < 0)
173 ✗ goto fail;
174
175 2017145 ret = av_fifo_write(cf->fifo, &pdst, 1);
176
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2017145 times.
 2017145 if (ret < 0)
177 ✗ goto fail;
178
179 2017145 return 0;
180 ✗ fail:
181 ✗ av_refstruct_unref(&pdst);
182 ✗ return ret;
183 }
184
185 6206958 size_t av_container_fifo_can_read(const AVContainerFifo *cf)
186 {
187 6206958 return av_fifo_can_read(cf->fifo);
188 }
189
190 27625 static void *frame_alloc(void *opaque)
191 {
192 27625 return av_frame_alloc();
193 }
194
195 994309 static void frame_reset(void *opaque, void *obj)
196 {
197 994309 av_frame_unref(obj);
198 994309 }
199
200 27625 static void frame_free(void *opaque, void *obj)
201 {
202 27625 AVFrame *frame = obj;
203 27625 av_frame_free(&frame);
204 27625 }
205
206 1965269 static int frame_transfer(void *opaque, void *dst, void *src, unsigned flags)
207 {
208
2/2
✓ Branch 0 taken 10140 times.
✓ Branch 1 taken 1955129 times.
 1965269 if (flags & AV_CONTAINER_FIFO_FLAG_REF)
209 10140 return av_frame_ref(dst, src);
210
211 1955129 av_frame_move_ref(dst, src);
212 1955129 return 0;
213 }
214
215 20194 AVContainerFifo *av_container_fifo_alloc_avframe(unsigned flags)
216 {
217 20194 return av_container_fifo_alloc(NULL, frame_alloc, frame_reset, frame_free,
218 frame_transfer, 0);
219 }
220