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 "error.h" | ||
29 | #include "mem.h" | ||
30 | #include "bprint.h" | ||
31 | |||
32 | struct AVDictionary { | ||
33 | int count; | ||
34 | AVDictionaryEntry *elems; | ||
35 | }; | ||
36 | |||
37 | 31260 | int av_dict_count(const AVDictionary *m) | |
38 | { | ||
39 |
2/2✓ Branch 0 taken 21866 times.
✓ Branch 1 taken 9394 times.
|
31260 | return m ? m->count : 0; |
40 | } | ||
41 | |||
42 | 4998745 | const AVDictionaryEntry *av_dict_iterate(const AVDictionary *m, | |
43 | const AVDictionaryEntry *prev) | ||
44 | { | ||
45 | 4998745 | int i = 0; | |
46 | |||
47 |
2/2✓ Branch 0 taken 2377341 times.
✓ Branch 1 taken 2621404 times.
|
4998745 | if (!m) |
48 | 2377341 | return NULL; | |
49 | |||
50 |
2/2✓ Branch 0 taken 1669582 times.
✓ Branch 1 taken 951822 times.
|
2621404 | if (prev) |
51 | 1669582 | i = prev - m->elems + 1; | |
52 | |||
53 | av_assert2(i >= 0); | ||
54 |
2/2✓ Branch 0 taken 705681 times.
✓ Branch 1 taken 1915723 times.
|
2621404 | if (i >= m->count) |
55 | 705681 | return NULL; | |
56 | |||
57 | 1915723 | return &m->elems[i]; | |
58 | } | ||
59 | |||
60 | 1656157 | AVDictionaryEntry *av_dict_get(const AVDictionary *m, const char *key, | |
61 | const AVDictionaryEntry *prev, int flags) | ||
62 | { | ||
63 | 1656157 | const AVDictionaryEntry *entry = prev; | |
64 | unsigned int j; | ||
65 | |||
66 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1656156 times.
|
1656157 | if (!key) |
67 | 1 | return NULL; | |
68 | |||
69 |
2/2✓ Branch 1 taken 1467188 times.
✓ Branch 2 taken 1409802 times.
|
2876990 | while ((entry = av_dict_iterate(m, entry))) { |
70 | 1467188 | const char *s = entry->key; | |
71 |
2/2✓ Branch 0 taken 42429 times.
✓ Branch 1 taken 1424759 times.
|
1467188 | if (flags & AV_DICT_MATCH_CASE) |
72 |
4/4✓ Branch 0 taken 134568 times.
✓ Branch 1 taken 34117 times.
✓ Branch 2 taken 126256 times.
✓ Branch 3 taken 8312 times.
|
168685 | for (j = 0; s[j] == key[j] && key[j]; j++) |
73 | ; | ||
74 | else | ||
75 |
4/4✓ Branch 0 taken 5148254 times.
✓ Branch 1 taken 1186875 times.
✓ Branch 2 taken 4910370 times.
✓ Branch 3 taken 237884 times.
|
6335129 | for (j = 0; av_toupper(s[j]) == av_toupper(key[j]) && key[j]; j++) |
76 | ; | ||
77 |
2/2✓ Branch 0 taken 1216120 times.
✓ Branch 1 taken 251068 times.
|
1467188 | if (key[j]) |
78 | 1216120 | continue; | |
79 |
4/4✓ Branch 0 taken 4872 times.
✓ Branch 1 taken 246196 times.
✓ Branch 2 taken 4714 times.
✓ Branch 3 taken 158 times.
|
251068 | if (s[j] && !(flags & AV_DICT_IGNORE_SUFFIX)) |
80 | 4714 | continue; | |
81 | 246354 | return (AVDictionaryEntry *)entry; | |
82 | } | ||
83 | 1409802 | return NULL; | |
84 | } | ||
85 | |||
86 | 1240550 | int av_dict_set(AVDictionary **pm, const char *key, const char *value, | |
87 | int flags) | ||
88 | { | ||
89 | 1240550 | AVDictionary *m = *pm; | |
90 | 1240550 | AVDictionaryEntry *tag = NULL; | |
91 | 1240550 | char *copy_key = NULL, *copy_value = NULL; | |
92 | int err; | ||
93 | |||
94 |
2/2✓ Branch 0 taken 61691 times.
✓ Branch 1 taken 1178859 times.
|
1240550 | if (flags & AV_DICT_DONT_STRDUP_VAL) |
95 | 61691 | copy_value = (void *)value; | |
96 |
2/2✓ Branch 0 taken 498107 times.
✓ Branch 1 taken 680752 times.
|
1178859 | else if (value) |
97 | 498107 | copy_value = av_strdup(value); | |
98 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 1240546 times.
|
1240550 | if (!key) { |
99 | 4 | err = AVERROR(EINVAL); | |
100 | 4 | goto err_out; | |
101 | } | ||
102 |
2/2✓ Branch 0 taken 31271 times.
✓ Branch 1 taken 1209275 times.
|
1240546 | if (flags & AV_DICT_DONT_STRDUP_KEY) |
103 | 31271 | copy_key = (void *)key; | |
104 | else | ||
105 | 1209275 | copy_key = av_strdup(key); | |
106 |
4/6✓ Branch 0 taken 1240546 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 559795 times.
✓ Branch 3 taken 680751 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 559795 times.
|
1240546 | if (!copy_key || (value && !copy_value)) |
107 | ✗ | goto enomem; | |
108 | |||
109 |
2/2✓ Branch 0 taken 1135103 times.
✓ Branch 1 taken 105443 times.
|
1240546 | if (!(flags & AV_DICT_MULTIKEY)) { |
110 | 1135103 | tag = av_dict_get(m, key, NULL, flags); | |
111 |
2/2✓ Branch 0 taken 5599 times.
✓ Branch 1 taken 99844 times.
|
105443 | } else if (flags & AV_DICT_DEDUP) { |
112 |
2/2✓ Branch 1 taken 9 times.
✓ Branch 2 taken 5591 times.
|
5600 | while ((tag = av_dict_get(m, key, tag, flags))) { |
113 |
2/6✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 9 times.
✗ Branch 5 not taken.
|
9 | if ((!value && !tag->value) || |
114 |
3/4✓ Branch 0 taken 9 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 8 times.
✓ Branch 3 taken 1 times.
|
9 | (value && tag->value && !strcmp(value, tag->value))) { |
115 | 8 | av_free(copy_key); | |
116 | 8 | av_free(copy_value); | |
117 | 8 | return 0; | |
118 | } | ||
119 | } | ||
120 | } | ||
121 |
2/2✓ Branch 0 taken 705692 times.
✓ Branch 1 taken 534846 times.
|
1240538 | if (!m) |
122 | 705692 | m = *pm = av_mallocz(sizeof(*m)); | |
123 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1240538 times.
|
1240538 | if (!m) |
124 | ✗ | goto enomem; | |
125 | |||
126 |
2/2✓ Branch 0 taken 159192 times.
✓ Branch 1 taken 1081346 times.
|
1240538 | if (tag) { |
127 |
2/2✓ Branch 0 taken 786 times.
✓ Branch 1 taken 158406 times.
|
159192 | if (flags & AV_DICT_DONT_OVERWRITE) { |
128 | 786 | av_free(copy_key); | |
129 | 786 | av_free(copy_value); | |
130 | 786 | return 0; | |
131 | } | ||
132 |
4/4✓ Branch 0 taken 36094 times.
✓ Branch 1 taken 122312 times.
✓ Branch 2 taken 15851 times.
✓ Branch 3 taken 20243 times.
|
158406 | if (copy_value && flags & AV_DICT_APPEND) { |
133 | 15851 | size_t oldlen = strlen(tag->value); | |
134 | 15851 | size_t new_part_len = strlen(copy_value); | |
135 | 15851 | size_t len = oldlen + new_part_len + 1; | |
136 | 15851 | char *newval = av_realloc(tag->value, len); | |
137 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15851 times.
|
15851 | if (!newval) |
138 | ✗ | goto enomem; | |
139 | 15851 | memcpy(newval + oldlen, copy_value, new_part_len + 1); | |
140 | 15851 | av_freep(©_value); | |
141 | 15851 | copy_value = newval; | |
142 | } else | ||
143 | 142555 | av_free(tag->value); | |
144 | 158406 | av_free(tag->key); | |
145 | 158406 | *tag = m->elems[--m->count]; | |
146 |
2/2✓ Branch 0 taken 522907 times.
✓ Branch 1 taken 558439 times.
|
1081346 | } else if (copy_value) { |
147 | 522907 | AVDictionaryEntry *tmp = av_realloc_array(m->elems, | |
148 | 522907 | m->count + 1, sizeof(*m->elems)); | |
149 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 522907 times.
|
522907 | if (!tmp) |
150 | ✗ | goto enomem; | |
151 | 522907 | m->elems = tmp; | |
152 | } | ||
153 |
2/2✓ Branch 0 taken 559001 times.
✓ Branch 1 taken 680751 times.
|
1239752 | if (copy_value) { |
154 | 559001 | m->elems[m->count].key = copy_key; | |
155 | 559001 | m->elems[m->count].value = copy_value; | |
156 | 559001 | m->count++; | |
157 | } else { | ||
158 | 680751 | err = 0; | |
159 | 680751 | goto end; | |
160 | } | ||
161 | |||
162 | 559001 | return 0; | |
163 | |||
164 | ✗ | enomem: | |
165 | ✗ | err = AVERROR(ENOMEM); | |
166 | 4 | err_out: | |
167 | 4 | av_free(copy_value); | |
168 | 680755 | end: | |
169 |
4/4✓ Branch 0 taken 680751 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 515232 times.
✓ Branch 3 taken 165519 times.
|
680755 | if (m && !m->count) { |
170 | 515232 | av_freep(&m->elems); | |
171 | 515232 | av_freep(pm); | |
172 | } | ||
173 | 680755 | av_free(copy_key); | |
174 | 680755 | return err; | |
175 | } | ||
176 | |||
177 | 9024 | int av_dict_set_int(AVDictionary **pm, const char *key, int64_t value, | |
178 | int flags) | ||
179 | { | ||
180 | char valuestr[22]; | ||
181 | 9024 | snprintf(valuestr, sizeof(valuestr), "%"PRId64, value); | |
182 | 9024 | flags &= ~AV_DICT_DONT_STRDUP_VAL; | |
183 | 9024 | return av_dict_set(pm, key, valuestr, flags); | |
184 | } | ||
185 | |||
186 | 485 | static int parse_key_value_pair(AVDictionary **pm, const char **buf, | |
187 | const char *key_val_sep, const char *pairs_sep, | ||
188 | int flags) | ||
189 | { | ||
190 | 485 | char *key = av_get_token(buf, key_val_sep); | |
191 | 485 | char *val = NULL; | |
192 | int ret; | ||
193 | |||
194 |
3/6✓ Branch 0 taken 485 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 485 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 485 times.
✗ Branch 5 not taken.
|
485 | if (key && *key && strspn(*buf, key_val_sep)) { |
195 | 485 | (*buf)++; | |
196 | 485 | val = av_get_token(buf, pairs_sep); | |
197 | } | ||
198 | |||
199 |
4/8✓ Branch 0 taken 485 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 485 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 485 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 485 times.
✗ Branch 7 not taken.
|
485 | if (key && *key && val && *val) |
200 | 485 | ret = av_dict_set(pm, key, val, flags); | |
201 | else | ||
202 | ✗ | ret = AVERROR(EINVAL); | |
203 | |||
204 | 485 | av_freep(&key); | |
205 | 485 | av_freep(&val); | |
206 | |||
207 | 485 | return ret; | |
208 | } | ||
209 | |||
210 | 233 | int av_dict_parse_string(AVDictionary **pm, const char *str, | |
211 | const char *key_val_sep, const char *pairs_sep, | ||
212 | int flags) | ||
213 | { | ||
214 | int ret; | ||
215 | |||
216 |
2/2✓ Branch 0 taken 5 times.
✓ Branch 1 taken 228 times.
|
233 | if (!str) |
217 | 5 | return 0; | |
218 | |||
219 | /* ignore STRDUP flags */ | ||
220 | 228 | flags &= ~(AV_DICT_DONT_STRDUP_KEY | AV_DICT_DONT_STRDUP_VAL); | |
221 | |||
222 |
2/2✓ Branch 0 taken 485 times.
✓ Branch 1 taken 228 times.
|
941 | while (*str) { |
223 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 485 times.
|
485 | if ((ret = parse_key_value_pair(pm, &str, key_val_sep, pairs_sep, flags)) < 0) |
224 | ✗ | return ret; | |
225 | |||
226 |
2/2✓ Branch 0 taken 226 times.
✓ Branch 1 taken 259 times.
|
485 | if (*str) |
227 | 259 | str++; | |
228 | } | ||
229 | |||
230 | 228 | return 0; | |
231 | } | ||
232 | |||
233 | 8183840 | void av_dict_free(AVDictionary **pm) | |
234 | { | ||
235 | 8183840 | AVDictionary *m = *pm; | |
236 | |||
237 |
2/2✓ Branch 0 taken 190460 times.
✓ Branch 1 taken 7993380 times.
|
8183840 | if (m) { |
238 |
2/2✓ Branch 0 taken 400595 times.
✓ Branch 1 taken 190460 times.
|
591055 | while (m->count--) { |
239 | 400595 | av_freep(&m->elems[m->count].key); | |
240 | 400595 | av_freep(&m->elems[m->count].value); | |
241 | } | ||
242 | 190460 | av_freep(&m->elems); | |
243 | } | ||
244 | 8183840 | av_freep(pm); | |
245 | 8183840 | } | |
246 | |||
247 | 1280228 | int av_dict_copy(AVDictionary **dst, const AVDictionary *src, int flags) | |
248 | { | ||
249 | 1280228 | const AVDictionaryEntry *t = NULL; | |
250 | |||
251 |
2/2✓ Branch 1 taken 52258 times.
✓ Branch 2 taken 1280228 times.
|
1332486 | while ((t = av_dict_iterate(src, t))) { |
252 | 52258 | int ret = av_dict_set(dst, t->key, t->value, flags); | |
253 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 52258 times.
|
52258 | if (ret < 0) |
254 | ✗ | return ret; | |
255 | } | ||
256 | |||
257 | 1280228 | return 0; | |
258 | } | ||
259 | |||
260 | 4496 | int av_dict_get_string(const AVDictionary *m, char **buffer, | |
261 | const char key_val_sep, const char pairs_sep) | ||
262 | { | ||
263 | 4496 | const AVDictionaryEntry *t = NULL; | |
264 | AVBPrint bprint; | ||
265 | 4496 | int cnt = 0; | |
266 | 4496 | char special_chars[] = {pairs_sep, key_val_sep, '\0'}; | |
267 | |||
268 |
4/8✓ Branch 0 taken 4496 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4496 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 4496 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 4496 times.
✗ Branch 7 not taken.
|
4496 | if (!buffer || pairs_sep == '\0' || key_val_sep == '\0' || pairs_sep == key_val_sep || |
269 |
2/4✓ Branch 0 taken 4496 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4496 times.
|
4496 | pairs_sep == '\\' || key_val_sep == '\\') |
270 | ✗ | return AVERROR(EINVAL); | |
271 | |||
272 |
2/2✓ Branch 1 taken 26 times.
✓ Branch 2 taken 4470 times.
|
4496 | if (!av_dict_count(m)) { |
273 | 26 | *buffer = av_strdup(""); | |
274 |
1/2✓ Branch 0 taken 26 times.
✗ Branch 1 not taken.
|
26 | return *buffer ? 0 : AVERROR(ENOMEM); |
275 | } | ||
276 | |||
277 | 4470 | av_bprint_init(&bprint, 64, AV_BPRINT_SIZE_UNLIMITED); | |
278 |
2/2✓ Branch 1 taken 4507 times.
✓ Branch 2 taken 4470 times.
|
8977 | while ((t = av_dict_iterate(m, t))) { |
279 |
2/2✓ Branch 0 taken 37 times.
✓ Branch 1 taken 4470 times.
|
4507 | if (cnt++) |
280 | 37 | av_bprint_append_data(&bprint, &pairs_sep, 1); | |
281 | 4507 | av_bprint_escape(&bprint, t->key, special_chars, AV_ESCAPE_MODE_BACKSLASH, 0); | |
282 | 4507 | av_bprint_append_data(&bprint, &key_val_sep, 1); | |
283 | 4507 | av_bprint_escape(&bprint, t->value, special_chars, AV_ESCAPE_MODE_BACKSLASH, 0); | |
284 | } | ||
285 | 4470 | return av_bprint_finalize(&bprint, buffer); | |
286 | } | ||
287 |