FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavutil/container_fifo.c
Date: 2025-04-25 22:50:00
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 135611 static int container_fifo_init_entry(AVRefStructOpaque opaque, void *obj)
40 {
41 135611 AVContainerFifo *cf = opaque.nc;
42 135611 void **pobj = obj;
43
44 135611 *pobj = cf->container_alloc(cf->opaque);
45
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 135611 times.
 135611 if (!*pobj)
46 ✗ return AVERROR(ENOMEM);
47
48 135611 return 0;
49 }
50
51 1786595 static void container_fifo_reset_entry(AVRefStructOpaque opaque, void *obj)
52 {
53 1786595 AVContainerFifo *cf = opaque.nc;
54 1786595 cf->container_reset(cf->opaque, *(void**)obj);
55 1786595 }
56
57 135611 static void container_fifo_free_entry(AVRefStructOpaque opaque, void *obj)
58 {
59 135611 AVContainerFifo *cf = opaque.nc;
60 135611 cf->container_free(cf->opaque, *(void**)obj);
61 135611 }
62
63 AVContainerFifo*
64 34564 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 34564 cf = av_mallocz(sizeof(*cf));
74
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 34564 times.
 34564 if (!cf)
75 ✗ return NULL;
76
77 34564 cf->opaque = opaque;
78 34564 cf->container_alloc = container_alloc;
79 34564 cf->container_reset = container_reset;
80 34564 cf->container_free = container_free;
81 34564 cf->fifo_transfer = fifo_transfer;
82
83 34564 cf->fifo = av_fifo_alloc2(1, sizeof(void*), AV_FIFO_FLAG_AUTO_GROW);
84
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 34564 times.
 34564 if (!cf->fifo)
85 ✗ goto fail;
86
87 34564 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 34564 times.
 34564 if (!cf->pool)
93 ✗ goto fail;
94
95 34564 return cf;
96 ✗ fail:
97 ✗ av_container_fifo_free(&cf);
98 ✗ return NULL;
99 }
100
101 34564 void av_container_fifo_free(AVContainerFifo **pcf)
102 {
103 AVContainerFifo *cf;
104
105
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 34564 times.
 34564 if (!*pcf)
106 ✗ return;
107
108 34564 cf = *pcf;
109
110
1/2
✓ Branch 0 taken 34564 times.
✗ Branch 1 not taken.
 34564 if (cf->fifo) {
111 void *obj;
112
2/2
✓ Branch 1 taken 54312 times.
✓ Branch 2 taken 34564 times.
 88876 while (av_fifo_read(cf->fifo, &obj, 1) >= 0)
113 54312 av_refstruct_unref(&obj);
114 34564 av_fifo_freep2(&cf->fifo);
115 }
116
117 34564 av_refstruct_pool_uninit(&cf->pool);
118
119 34564 av_freep(pcf);
120 }
121
122 2537547 int av_container_fifo_read(AVContainerFifo *cf, void *obj, unsigned flags)
123 {
124 void **psrc;
125 int ret;
126
127 2537547 ret = av_fifo_read(cf->fifo, &psrc, 1);
128
2/2
✓ Branch 0 taken 820802 times.
✓ Branch 1 taken 1716745 times.
 2537547 if (ret < 0)
129 820802 return ret;
130
131 1716745 ret = cf->fifo_transfer(cf->opaque, obj, *psrc, flags);
132 1716745 av_refstruct_unref(&psrc);
133
134 1716745 return ret;
135 }
136
137 209707 int av_container_fifo_peek(AVContainerFifo *cf, void **pdst, size_t offset)
138 {
139 void **pobj;
140 int ret;
141
142 209707 ret = av_fifo_peek(cf->fifo, &pobj, 1, offset);
143
2/2
✓ Branch 0 taken 15045 times.
✓ Branch 1 taken 194662 times.
 209707 if (ret < 0)
144 15045 return ret;
145
146 194662 *pdst = *pobj;
147
148 194662 return 0;
149 }
150
151 15538 void av_container_fifo_drain(AVContainerFifo *cf, size_t nb_elems)
152 {
153
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 15538 times.
 15538 av_assert0(nb_elems <= av_fifo_can_read(cf->fifo));
154
2/2
✓ Branch 0 taken 15538 times.
✓ Branch 1 taken 15538 times.
 31076 while (nb_elems--) {
155 void **pobj;
156 15538 int ret = av_fifo_read(cf->fifo, &pobj, 1);
157
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15538 times.
 15538 av_assert0(ret >= 0);
158 15538 av_refstruct_unref(&pobj);
159 }
160 15538 }
161
162 1786595 int av_container_fifo_write(AVContainerFifo *cf, void *obj, unsigned flags)
163 {
164 void **pdst;
165 int ret;
166
167 1786595 pdst = av_refstruct_pool_get(cf->pool);
168
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1786595 times.
 1786595 if (!pdst)
169 ✗ return AVERROR(ENOMEM);
170
171 1786595 ret = cf->fifo_transfer(cf->opaque, *pdst, obj, flags);
172
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1786595 times.
 1786595 if (ret < 0)
173 ✗ goto fail;
174
175 1786595 ret = av_fifo_write(cf->fifo, &pdst, 1);
176
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1786595 times.
 1786595 if (ret < 0)
177 ✗ goto fail;
178
179 1786595 return 0;
180 ✗ fail:
181 ✗ av_refstruct_unref(&pdst);
182 ✗ return ret;
183 }
184
185 5171550 size_t av_container_fifo_can_read(const AVContainerFifo *cf)
186 {
187 5171550 return av_fifo_can_read(cf->fifo);
188 }
189
190 66446 static void *frame_alloc(void *opaque)
191 {
192 66446 return av_frame_alloc();
193 }
194
195 879944 static void frame_reset(void *opaque, void *obj)
196 {
197 879944 av_frame_unref(obj);
198 879944 }
199
200 66446 static void frame_free(void *opaque, void *obj)
201 {
202 66446 AVFrame *frame = obj;
203 66446 av_frame_free(&frame);
204 66446 }
205
206 1717365 static int frame_transfer(void *opaque, void *dst, void *src, unsigned flags)
207 {
208
2/2
✓ Branch 0 taken 10058 times.
✓ Branch 1 taken 1707307 times.
 1717365 if (flags & AV_CONTAINER_FIFO_FLAG_REF)
209 10058 return av_frame_ref(dst, src);
210
211 1707307 av_frame_move_ref(dst, src);
212 1707307 return 0;
213 }
214
215 19491 AVContainerFifo *av_container_fifo_alloc_avframe(unsigned flags)
216 {
217 19491 return av_container_fifo_alloc(NULL, frame_alloc, frame_reset, frame_free,
218 frame_transfer, 0);
219 }
220