FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavutil/dict.c
Date: 2024-11-20 23:03:26
Exec Total Coverage
Lines: 145 156 92.9%
Functions: 11 11 100.0%
Branches: 99 122 81.1%
Line Branch Exec Source
1 /*
2 * copyright (c) 2009 Michael Niedermayer
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 <inttypes.h>
22 #include <stdio.h>
23 #include <string.h>
24
25 #include "avassert.h"
26 #include "avstring.h"
27 #include "dict.h"
28 #include "dict_internal.h"
29 #include "error.h"
30 #include "mem.h"
31 #include "time_internal.h"
32 #include "bprint.h"
33
34 struct AVDictionary {
35 int count;
36 AVDictionaryEntry *elems;
37 };
38
39 30313 int av_dict_count(const AVDictionary *m)
40 {
41
2/2
✓ Branch 0 taken 21095 times.
✓ Branch 1 taken 9218 times.
 30313 return m ? m->count : 0;
42 }
43
44 4989298 const AVDictionaryEntry *av_dict_iterate(const AVDictionary *m,
45 const AVDictionaryEntry *prev)
46 {
47 4989298 int i = 0;
48
49
2/2
✓ Branch 0 taken 2463850 times.
✓ Branch 1 taken 2525448 times.
 4989298 if (!m)
50 2463850 return NULL;
51
52
2/2
✓ Branch 0 taken 1611354 times.
✓ Branch 1 taken 914094 times.
 2525448 if (prev)
53 1611354 i = prev - m->elems + 1;
54
55 av_assert2(i >= 0);
56
2/2
✓ Branch 0 taken 678740 times.
✓ Branch 1 taken 1846708 times.
 2525448 if (i >= m->count)
57 678740 return NULL;
58
59 1846708 return &m->elems[i];
60 }
61
62 1597431 AVDictionaryEntry *av_dict_get(const AVDictionary *m, const char *key,
63 const AVDictionaryEntry *prev, int flags)
64 {
65 1597431 const AVDictionaryEntry *entry = prev;
66 unsigned int j;
67
68
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1597430 times.
 1597431 if (!key)
69 1 return NULL;
70
71
2/2
✓ Branch 1 taken 1418095 times.
✓ Branch 2 taken 1361875 times.
 2779970 while ((entry = av_dict_iterate(m, entry))) {
72 1418095 const char *s = entry->key;
73
2/2
✓ Branch 0 taken 42131 times.
✓ Branch 1 taken 1375964 times.
 1418095 if (flags & AV_DICT_MATCH_CASE)
74
4/4
✓ Branch 0 taken 132384 times.
✓ Branch 1 taken 33975 times.
✓ Branch 2 taken 124228 times.
✓ Branch 3 taken 8156 times.
 166359 for (j = 0; s[j] == key[j] && key[j]; j++)
75 ;
76 else
77
4/4
✓ Branch 0 taken 4979313 times.
✓ Branch 1 taken 1148723 times.
✓ Branch 2 taken 4752072 times.
✓ Branch 3 taken 227241 times.
 6128036 for (j = 0; av_toupper(s[j]) == av_toupper(key[j]) && key[j]; j++)
78 ;
79
2/2
✓ Branch 0 taken 1177975 times.
✓ Branch 1 taken 240120 times.
 1418095 if (key[j])
80 1177975 continue;
81
4/4
✓ Branch 0 taken 4723 times.
✓ Branch 1 taken 235397 times.
✓ Branch 2 taken 4565 times.
✓ Branch 3 taken 158 times.
 240120 if (s[j] && !(flags & AV_DICT_IGNORE_SUFFIX))
82 4565 continue;
83 235555 return (AVDictionaryEntry *)entry;
84 }
85 1361875 return NULL;
86 }
87
88 1191462 int av_dict_set(AVDictionary **pm, const char *key, const char *value,
89 int flags)
90 {
91 1191462 AVDictionary *m = *pm;
92 1191462 AVDictionaryEntry *tag = NULL;
93 1191462 char *copy_key = NULL, *copy_value = NULL;
94 int err;
95
96
2/2
✓ Branch 0 taken 55935 times.
✓ Branch 1 taken 1135527 times.
 1191462 if (flags & AV_DICT_DONT_STRDUP_VAL)
97 55935 copy_value = (void *)value;
98
2/2
✓ Branch 0 taken 478719 times.
✓ Branch 1 taken 656808 times.
 1135527 else if (value)
99 478719 copy_value = av_strdup(value);
100
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 1191458 times.
 1191462 if (!key) {
101 4 err = AVERROR(EINVAL);
102 4 goto err_out;
103 }
104
2/2
✓ Branch 0 taken 1098585 times.
✓ Branch 1 taken 92873 times.
 1191458 if (!(flags & AV_DICT_MULTIKEY)) {
105 1098585 tag = av_dict_get(m, key, NULL, flags);
106 }
107
2/2
✓ Branch 0 taken 28245 times.
✓ Branch 1 taken 1163213 times.
 1191458 if (flags & AV_DICT_DONT_STRDUP_KEY)
108 28245 copy_key = (void *)key;
109 else
110 1163213 copy_key = av_strdup(key);
111
2/2
✓ Branch 0 taken 678074 times.
✓ Branch 1 taken 513384 times.
 1191458 if (!m)
112 678074 m = *pm = av_mallocz(sizeof(*m));
113
5/8
✓ Branch 0 taken 1191458 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1191458 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 534651 times.
✓ Branch 5 taken 656807 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 534651 times.
 1191458 if (!m || !copy_key || (value && !copy_value))
114 ✗ goto enomem;
115
116
2/2
✓ Branch 0 taken 152358 times.
✓ Branch 1 taken 1039100 times.
 1191458 if (tag) {
117
2/2
✓ Branch 0 taken 778 times.
✓ Branch 1 taken 151580 times.
 152358 if (flags & AV_DICT_DONT_OVERWRITE) {
118 778 av_free(copy_key);
119 778 av_free(copy_value);
120 778 return 0;
121 }
122
4/4
✓ Branch 0 taken 33686 times.
✓ Branch 1 taken 117894 times.
✓ Branch 2 taken 14573 times.
✓ Branch 3 taken 19113 times.
 151580 if (copy_value && flags & AV_DICT_APPEND) {
123 14573 size_t oldlen = strlen(tag->value);
124 14573 size_t new_part_len = strlen(copy_value);
125 14573 size_t len = oldlen + new_part_len + 1;
126 14573 char *newval = av_realloc(tag->value, len);
127
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14573 times.
 14573 if (!newval)
128 ✗ goto enomem;
129 14573 memcpy(newval + oldlen, copy_value, new_part_len + 1);
130 14573 av_freep(&copy_value);
131 14573 copy_value = newval;
132 } else
133 137007 av_free(tag->value);
134 151580 av_free(tag->key);
135 151580 *tag = m->elems[--m->count];
136
2/2
✓ Branch 0 taken 500187 times.
✓ Branch 1 taken 538913 times.
 1039100 } else if (copy_value) {
137 500187 AVDictionaryEntry *tmp = av_realloc_array(m->elems,
138 500187 m->count + 1, sizeof(*m->elems));
139
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 500187 times.
 500187 if (!tmp)
140 ✗ goto enomem;
141 500187 m->elems = tmp;
142 }
143
2/2
✓ Branch 0 taken 533873 times.
✓ Branch 1 taken 656807 times.
 1190680 if (copy_value) {
144 533873 m->elems[m->count].key = copy_key;
145 533873 m->elems[m->count].value = copy_value;
146 533873 m->count++;
147 } else {
148 656807 err = 0;
149 656807 goto end;
150 }
151
152 533873 return 0;
153
154 ✗ enomem:
155 ✗ err = AVERROR(ENOMEM);
156 4 err_out:
157 4 av_free(copy_value);
158 656811 end:
159
4/4
✓ Branch 0 taken 656807 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 497679 times.
✓ Branch 3 taken 159128 times.
 656811 if (m && !m->count) {
160 497679 av_freep(&m->elems);
161 497679 av_freep(pm);
162 }
163 656811 av_free(copy_key);
164 656811 return err;
165 }
166
167 8839 int av_dict_set_int(AVDictionary **pm, const char *key, int64_t value,
168 int flags)
169 {
170 char valuestr[22];
171 8839 snprintf(valuestr, sizeof(valuestr), "%"PRId64, value);
172 8839 flags &= ~AV_DICT_DONT_STRDUP_VAL;
173 8839 return av_dict_set(pm, key, valuestr, flags);
174 }
175
176 436 static int parse_key_value_pair(AVDictionary **pm, const char **buf,
177 const char *key_val_sep, const char *pairs_sep,
178 int flags)
179 {
180 436 char *key = av_get_token(buf, key_val_sep);
181 436 char *val = NULL;
182 int ret;
183
184
3/6
✓ Branch 0 taken 436 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 436 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 436 times.
✗ Branch 5 not taken.
 436 if (key && *key && strspn(*buf, key_val_sep)) {
185 436 (*buf)++;
186 436 val = av_get_token(buf, pairs_sep);
187 }
188
189
4/8
✓ Branch 0 taken 436 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 436 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 436 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 436 times.
✗ Branch 7 not taken.
 436 if (key && *key && val && *val)
190 436 ret = av_dict_set(pm, key, val, flags);
191 else
192 ✗ ret = AVERROR(EINVAL);
193
194 436 av_freep(&key);
195 436 av_freep(&val);
196
197 436 return ret;
198 }
199
200 214 int av_dict_parse_string(AVDictionary **pm, const char *str,
201 const char *key_val_sep, const char *pairs_sep,
202 int flags)
203 {
204 int ret;
205
206
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 209 times.
 214 if (!str)
207 5 return 0;
208
209 /* ignore STRDUP flags */
210 209 flags &= ~(AV_DICT_DONT_STRDUP_KEY | AV_DICT_DONT_STRDUP_VAL);
211
212
2/2
✓ Branch 0 taken 436 times.
✓ Branch 1 taken 209 times.
 854 while (*str) {
213
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 436 times.
 436 if ((ret = parse_key_value_pair(pm, &str, key_val_sep, pairs_sep, flags)) < 0)
214 ✗ return ret;
215
216
2/2
✓ Branch 0 taken 207 times.
✓ Branch 1 taken 229 times.
 436 if (*str)
217 229 str++;
218 }
219
220 209 return 0;
221 }
222
223 8188890 void av_dict_free(AVDictionary **pm)
224 {
225 8188890 AVDictionary *m = *pm;
226
227
2/2
✓ Branch 0 taken 180395 times.
✓ Branch 1 taken 8008495 times.
 8188890 if (m) {
228
2/2
✓ Branch 0 taken 382293 times.
✓ Branch 1 taken 180395 times.
 562688 while (m->count--) {
229 382293 av_freep(&m->elems[m->count].key);
230 382293 av_freep(&m->elems[m->count].value);
231 }
232 180395 av_freep(&m->elems);
233 }
234 8188890 av_freep(pm);
235 8188890 }
236
237 1405002 int av_dict_copy(AVDictionary **dst, const AVDictionary *src, int flags)
238 {
239 1405002 const AVDictionaryEntry *t = NULL;
240
241
2/2
✓ Branch 1 taken 52561 times.
✓ Branch 2 taken 1405002 times.
 1457563 while ((t = av_dict_iterate(src, t))) {
242 52561 int ret = av_dict_set(dst, t->key, t->value, flags);
243
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 52561 times.
 52561 if (ret < 0)
244 ✗ return ret;
245 }
246
247 1405002 return 0;
248 }
249
250 4338 int av_dict_get_string(const AVDictionary *m, char **buffer,
251 const char key_val_sep, const char pairs_sep)
252 {
253 4338 const AVDictionaryEntry *t = NULL;
254 AVBPrint bprint;
255 4338 int cnt = 0;
256 4338 char special_chars[] = {pairs_sep, key_val_sep, '\0'};
257
258
4/8
✓ Branch 0 taken 4338 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4338 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 4338 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 4338 times.
✗ Branch 7 not taken.
 4338 if (!buffer || pairs_sep == '\0' || key_val_sep == '\0' || pairs_sep == key_val_sep ||
259
2/4
✓ Branch 0 taken 4338 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4338 times.
 4338 pairs_sep == '\\' || key_val_sep == '\\')
260 ✗ return AVERROR(EINVAL);
261
262
2/2
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 4333 times.
 4338 if (!av_dict_count(m)) {
263 5 *buffer = av_strdup("");
264
1/2
✓ Branch 0 taken 5 times.
✗ Branch 1 not taken.
 5 return *buffer ? 0 : AVERROR(ENOMEM);
265 }
266
267 4333 av_bprint_init(&bprint, 64, AV_BPRINT_SIZE_UNLIMITED);
268
2/2
✓ Branch 1 taken 4363 times.
✓ Branch 2 taken 4333 times.
 8696 while ((t = av_dict_iterate(m, t))) {
269
2/2
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 4333 times.
 4363 if (cnt++)
270 30 av_bprint_append_data(&bprint, &pairs_sep, 1);
271 4363 av_bprint_escape(&bprint, t->key, special_chars, AV_ESCAPE_MODE_BACKSLASH, 0);
272 4363 av_bprint_append_data(&bprint, &key_val_sep, 1);
273 4363 av_bprint_escape(&bprint, t->value, special_chars, AV_ESCAPE_MODE_BACKSLASH, 0);
274 }
275 4333 return av_bprint_finalize(&bprint, buffer);
276 }
277
278 564 int avpriv_dict_set_timestamp(AVDictionary **dict, const char *key, int64_t timestamp)
279 {
280 564 time_t seconds = timestamp / 1000000;
281 struct tm *ptm, tmbuf;
282 564 ptm = gmtime_r(&seconds, &tmbuf);
283
1/2
✓ Branch 0 taken 564 times.
✗ Branch 1 not taken.
 564 if (ptm) {
284 char buf[32];
285
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 564 times.
 564 if (!strftime(buf, sizeof(buf), "%Y-%m-%dT%H:%M:%S", ptm))
286 ✗ return AVERROR_EXTERNAL;
287 564 av_strlcatf(buf, sizeof(buf), ".%06dZ", (int)(timestamp % 1000000));
288 564 return av_dict_set(dict, key, buf, 0);
289 } else {
290 ✗ return AVERROR_EXTERNAL;
291 }
292 }
293