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 | 30760 | int av_dict_count(const AVDictionary *m) | |
38 | { | ||
39 |
2/2✓ Branch 0 taken 21508 times.
✓ Branch 1 taken 9252 times.
|
30760 | return m ? m->count : 0; |
40 | } | ||
41 | |||
42 | 4923814 | const AVDictionaryEntry *av_dict_iterate(const AVDictionary *m, | |
43 | const AVDictionaryEntry *prev) | ||
44 | { | ||
45 | 4923814 | int i = 0; | |
46 | |||
47 |
2/2✓ Branch 0 taken 2341749 times.
✓ Branch 1 taken 2582065 times.
|
4923814 | if (!m) |
48 | 2341749 | return NULL; | |
49 | |||
50 |
2/2✓ Branch 0 taken 1646934 times.
✓ Branch 1 taken 935131 times.
|
2582065 | if (prev) |
51 | 1646934 | i = prev - m->elems + 1; | |
52 | |||
53 | av_assert2(i >= 0); | ||
54 |
2/2✓ Branch 0 taken 694283 times.
✓ Branch 1 taken 1887782 times.
|
2582065 | if (i >= m->count) |
55 | 694283 | return NULL; | |
56 | |||
57 | 1887782 | return &m->elems[i]; | |
58 | } | ||
59 | |||
60 | 1620541 | AVDictionaryEntry *av_dict_get(const AVDictionary *m, const char *key, | |
61 | const AVDictionaryEntry *prev, int flags) | ||
62 | { | ||
63 | 1620541 | const AVDictionaryEntry *entry = prev; | |
64 | unsigned int j; | ||
65 | |||
66 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1620540 times.
|
1620541 | if (!key) |
67 | 1 | return NULL; | |
68 | |||
69 |
2/2✓ Branch 1 taken 1446358 times.
✓ Branch 2 taken 1379479 times.
|
2825837 | while ((entry = av_dict_iterate(m, entry))) { |
70 | 1446358 | const char *s = entry->key; | |
71 |
2/2✓ Branch 0 taken 42222 times.
✓ Branch 1 taken 1404136 times.
|
1446358 | if (flags & AV_DICT_MATCH_CASE) |
72 |
4/4✓ Branch 0 taken 133028 times.
✓ Branch 1 taken 34020 times.
✓ Branch 2 taken 124826 times.
✓ Branch 3 taken 8202 times.
|
167048 | for (j = 0; s[j] == key[j] && key[j]; j++) |
73 | ; | ||
74 | else | ||
75 |
4/4✓ Branch 0 taken 5038730 times.
✓ Branch 1 taken 1171435 times.
✓ Branch 2 taken 4806029 times.
✓ Branch 3 taken 232701 times.
|
6210165 | for (j = 0; av_toupper(s[j]) == av_toupper(key[j]) && key[j]; j++) |
76 | ; | ||
77 |
2/2✓ Branch 0 taken 1200583 times.
✓ Branch 1 taken 245775 times.
|
1446358 | if (key[j]) |
78 | 1200583 | continue; | |
79 |
4/4✓ Branch 0 taken 4872 times.
✓ Branch 1 taken 240903 times.
✓ Branch 2 taken 4714 times.
✓ Branch 3 taken 158 times.
|
245775 | if (s[j] && !(flags & AV_DICT_IGNORE_SUFFIX)) |
80 | 4714 | continue; | |
81 | 241061 | return (AVDictionaryEntry *)entry; | |
82 | } | ||
83 | 1379479 | return NULL; | |
84 | } | ||
85 | |||
86 | 1212927 | int av_dict_set(AVDictionary **pm, const char *key, const char *value, | |
87 | int flags) | ||
88 | { | ||
89 | 1212927 | AVDictionary *m = *pm; | |
90 | 1212927 | AVDictionaryEntry *tag = NULL; | |
91 | 1212927 | char *copy_key = NULL, *copy_value = NULL; | |
92 | int err; | ||
93 | |||
94 |
2/2✓ Branch 0 taken 60665 times.
✓ Branch 1 taken 1152262 times.
|
1212927 | if (flags & AV_DICT_DONT_STRDUP_VAL) |
95 | 60665 | copy_value = (void *)value; | |
96 |
2/2✓ Branch 0 taken 488959 times.
✓ Branch 1 taken 663303 times.
|
1152262 | else if (value) |
97 | 488959 | copy_value = av_strdup(value); | |
98 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 1212923 times.
|
1212927 | if (!key) { |
99 | 4 | err = AVERROR(EINVAL); | |
100 | 4 | goto err_out; | |
101 | } | ||
102 |
2/2✓ Branch 0 taken 30880 times.
✓ Branch 1 taken 1182043 times.
|
1212923 | if (flags & AV_DICT_DONT_STRDUP_KEY) |
103 | 30880 | copy_key = (void *)key; | |
104 | else | ||
105 | 1182043 | copy_key = av_strdup(key); | |
106 |
4/6✓ Branch 0 taken 1212923 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 549621 times.
✓ Branch 3 taken 663302 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 549621 times.
|
1212923 | if (!copy_key || (value && !copy_value)) |
107 | ✗ | goto enomem; | |
108 | |||
109 |
2/2✓ Branch 0 taken 1109017 times.
✓ Branch 1 taken 103906 times.
|
1212923 | if (!(flags & AV_DICT_MULTIKEY)) { |
110 | 1109017 | tag = av_dict_get(m, key, NULL, flags); | |
111 |
2/2✓ Branch 0 taken 5463 times.
✓ Branch 1 taken 98443 times.
|
103906 | } else if (flags & AV_DICT_DEDUP) { |
112 |
2/2✓ Branch 1 taken 9 times.
✓ Branch 2 taken 5455 times.
|
5464 | 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 689067 times.
✓ Branch 1 taken 523848 times.
|
1212915 | if (!m) |
122 | 689067 | m = *pm = av_mallocz(sizeof(*m)); | |
123 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1212915 times.
|
1212915 | if (!m) |
124 | ✗ | goto enomem; | |
125 | |||
126 |
2/2✓ Branch 0 taken 155253 times.
✓ Branch 1 taken 1057662 times.
|
1212915 | if (tag) { |
127 |
2/2✓ Branch 0 taken 780 times.
✓ Branch 1 taken 154473 times.
|
155253 | if (flags & AV_DICT_DONT_OVERWRITE) { |
128 | 780 | av_free(copy_key); | |
129 | 780 | av_free(copy_value); | |
130 | 780 | return 0; | |
131 | } | ||
132 |
4/4✓ Branch 0 taken 35439 times.
✓ Branch 1 taken 119034 times.
✓ Branch 2 taken 15500 times.
✓ Branch 3 taken 19939 times.
|
154473 | if (copy_value && flags & AV_DICT_APPEND) { |
133 | 15500 | size_t oldlen = strlen(tag->value); | |
134 | 15500 | size_t new_part_len = strlen(copy_value); | |
135 | 15500 | size_t len = oldlen + new_part_len + 1; | |
136 | 15500 | char *newval = av_realloc(tag->value, len); | |
137 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15500 times.
|
15500 | if (!newval) |
138 | ✗ | goto enomem; | |
139 | 15500 | memcpy(newval + oldlen, copy_value, new_part_len + 1); | |
140 | 15500 | av_freep(©_value); | |
141 | 15500 | copy_value = newval; | |
142 | } else | ||
143 | 138973 | av_free(tag->value); | |
144 | 154473 | av_free(tag->key); | |
145 | 154473 | *tag = m->elems[--m->count]; | |
146 |
2/2✓ Branch 0 taken 513394 times.
✓ Branch 1 taken 544268 times.
|
1057662 | } else if (copy_value) { |
147 | 513394 | AVDictionaryEntry *tmp = av_realloc_array(m->elems, | |
148 | 513394 | m->count + 1, sizeof(*m->elems)); | |
149 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 513394 times.
|
513394 | if (!tmp) |
150 | ✗ | goto enomem; | |
151 | 513394 | m->elems = tmp; | |
152 | } | ||
153 |
2/2✓ Branch 0 taken 548833 times.
✓ Branch 1 taken 663302 times.
|
1212135 | if (copy_value) { |
154 | 548833 | m->elems[m->count].key = copy_key; | |
155 | 548833 | m->elems[m->count].value = copy_value; | |
156 | 548833 | m->count++; | |
157 | } else { | ||
158 | 663302 | err = 0; | |
159 | 663302 | goto end; | |
160 | } | ||
161 | |||
162 | 548833 | return 0; | |
163 | |||
164 | ✗ | enomem: | |
165 | ✗ | err = AVERROR(ENOMEM); | |
166 | 4 | err_out: | |
167 | 4 | av_free(copy_value); | |
168 | 663306 | end: | |
169 |
4/4✓ Branch 0 taken 663302 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 501680 times.
✓ Branch 3 taken 161622 times.
|
663306 | if (m && !m->count) { |
170 | 501680 | av_freep(&m->elems); | |
171 | 501680 | av_freep(pm); | |
172 | } | ||
173 | 663306 | av_free(copy_key); | |
174 | 663306 | return err; | |
175 | } | ||
176 | |||
177 | 8910 | int av_dict_set_int(AVDictionary **pm, const char *key, int64_t value, | |
178 | int flags) | ||
179 | { | ||
180 | char valuestr[22]; | ||
181 | 8910 | snprintf(valuestr, sizeof(valuestr), "%"PRId64, value); | |
182 | 8910 | flags &= ~AV_DICT_DONT_STRDUP_VAL; | |
183 | 8910 | return av_dict_set(pm, key, valuestr, flags); | |
184 | } | ||
185 | |||
186 | 481 | 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 | 481 | char *key = av_get_token(buf, key_val_sep); | |
191 | 481 | char *val = NULL; | |
192 | int ret; | ||
193 | |||
194 |
3/6✓ Branch 0 taken 481 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 481 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 481 times.
✗ Branch 5 not taken.
|
481 | if (key && *key && strspn(*buf, key_val_sep)) { |
195 | 481 | (*buf)++; | |
196 | 481 | val = av_get_token(buf, pairs_sep); | |
197 | } | ||
198 | |||
199 |
4/8✓ Branch 0 taken 481 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 481 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 481 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 481 times.
✗ Branch 7 not taken.
|
481 | if (key && *key && val && *val) |
200 | 481 | ret = av_dict_set(pm, key, val, flags); | |
201 | else | ||
202 | ✗ | ret = AVERROR(EINVAL); | |
203 | |||
204 | 481 | av_freep(&key); | |
205 | 481 | av_freep(&val); | |
206 | |||
207 | 481 | return ret; | |
208 | } | ||
209 | |||
210 | 231 | 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 226 times.
|
231 | if (!str) |
217 | 5 | return 0; | |
218 | |||
219 | /* ignore STRDUP flags */ | ||
220 | 226 | flags &= ~(AV_DICT_DONT_STRDUP_KEY | AV_DICT_DONT_STRDUP_VAL); | |
221 | |||
222 |
2/2✓ Branch 0 taken 481 times.
✓ Branch 1 taken 226 times.
|
933 | while (*str) { |
223 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 481 times.
|
481 | 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 224 times.
✓ Branch 1 taken 257 times.
|
481 | if (*str) |
227 | 257 | str++; | |
228 | } | ||
229 | |||
230 | 226 | return 0; | |
231 | } | ||
232 | |||
233 | 8107081 | void av_dict_free(AVDictionary **pm) | |
234 | { | ||
235 | 8107081 | AVDictionary *m = *pm; | |
236 | |||
237 |
2/2✓ Branch 0 taken 187387 times.
✓ Branch 1 taken 7919694 times.
|
8107081 | if (m) { |
238 |
2/2✓ Branch 0 taken 394360 times.
✓ Branch 1 taken 187387 times.
|
581747 | while (m->count--) { |
239 | 394360 | av_freep(&m->elems[m->count].key); | |
240 | 394360 | av_freep(&m->elems[m->count].value); | |
241 | } | ||
242 | 187387 | av_freep(&m->elems); | |
243 | } | ||
244 | 8107081 | av_freep(pm); | |
245 | 8107081 | } | |
246 | |||
247 | 1271014 | int av_dict_copy(AVDictionary **dst, const AVDictionary *src, int flags) | |
248 | { | ||
249 | 1271014 | const AVDictionaryEntry *t = NULL; | |
250 | |||
251 |
2/2✓ Branch 1 taken 51616 times.
✓ Branch 2 taken 1271014 times.
|
1322630 | while ((t = av_dict_iterate(src, t))) { |
252 | 51616 | int ret = av_dict_set(dst, t->key, t->value, flags); | |
253 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 51616 times.
|
51616 | if (ret < 0) |
254 | ✗ | return ret; | |
255 | } | ||
256 | |||
257 | 1271014 | return 0; | |
258 | } | ||
259 | |||
260 | 4369 | int av_dict_get_string(const AVDictionary *m, char **buffer, | |
261 | const char key_val_sep, const char pairs_sep) | ||
262 | { | ||
263 | 4369 | const AVDictionaryEntry *t = NULL; | |
264 | AVBPrint bprint; | ||
265 | 4369 | int cnt = 0; | |
266 | 4369 | char special_chars[] = {pairs_sep, key_val_sep, '\0'}; | |
267 | |||
268 |
4/8✓ Branch 0 taken 4369 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4369 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 4369 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 4369 times.
✗ Branch 7 not taken.
|
4369 | if (!buffer || pairs_sep == '\0' || key_val_sep == '\0' || pairs_sep == key_val_sep || |
269 |
2/4✓ Branch 0 taken 4369 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4369 times.
|
4369 | pairs_sep == '\\' || key_val_sep == '\\') |
270 | ✗ | return AVERROR(EINVAL); | |
271 | |||
272 |
2/2✓ Branch 1 taken 5 times.
✓ Branch 2 taken 4364 times.
|
4369 | if (!av_dict_count(m)) { |
273 | 5 | *buffer = av_strdup(""); | |
274 |
1/2✓ Branch 0 taken 5 times.
✗ Branch 1 not taken.
|
5 | return *buffer ? 0 : AVERROR(ENOMEM); |
275 | } | ||
276 | |||
277 | 4364 | av_bprint_init(&bprint, 64, AV_BPRINT_SIZE_UNLIMITED); | |
278 |
2/2✓ Branch 1 taken 4394 times.
✓ Branch 2 taken 4364 times.
|
8758 | while ((t = av_dict_iterate(m, t))) { |
279 |
2/2✓ Branch 0 taken 30 times.
✓ Branch 1 taken 4364 times.
|
4394 | if (cnt++) |
280 | 30 | av_bprint_append_data(&bprint, &pairs_sep, 1); | |
281 | 4394 | av_bprint_escape(&bprint, t->key, special_chars, AV_ESCAPE_MODE_BACKSLASH, 0); | |
282 | 4394 | av_bprint_append_data(&bprint, &key_val_sep, 1); | |
283 | 4394 | av_bprint_escape(&bprint, t->value, special_chars, AV_ESCAPE_MODE_BACKSLASH, 0); | |
284 | } | ||
285 | 4364 | return av_bprint_finalize(&bprint, buffer); | |
286 | } | ||
287 |