dict: add AV_DICT_APPEND flag.
[libav.git] / libavutil / dict.c
1 /*
2 * copyright (c) 2009 Michael Niedermayer
3 *
4 * This file is part of Libav.
5 *
6 * Libav 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 * Libav 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 Libav; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include <strings.h>
22 #include "avstring.h"
23 #include "dict.h"
24 #include "internal.h"
25 #include "mem.h"
26
27 AVDictionaryEntry *
28 av_dict_get(AVDictionary *m, const char *key, const AVDictionaryEntry *prev, int flags)
29 {
30 unsigned int i, j;
31
32 if(!m)
33 return NULL;
34
35 if(prev) i= prev - m->elems + 1;
36 else i= 0;
37
38 for(; i<m->count; i++){
39 const char *s= m->elems[i].key;
40 if(flags & AV_DICT_MATCH_CASE) for(j=0; s[j] == key[j] && key[j]; j++);
41 else for(j=0; toupper(s[j]) == toupper(key[j]) && key[j]; j++);
42 if(key[j])
43 continue;
44 if(s[j] && !(flags & AV_DICT_IGNORE_SUFFIX))
45 continue;
46 return &m->elems[i];
47 }
48 return NULL;
49 }
50
51 int av_dict_set(AVDictionary **pm, const char *key, const char *value, int flags)
52 {
53 AVDictionary *m = *pm;
54 AVDictionaryEntry *tag = av_dict_get(m, key, NULL, flags);
55 char *oldval = NULL;
56
57 if(!m)
58 m = *pm = av_mallocz(sizeof(*m));
59
60 if(tag) {
61 if (flags & AV_DICT_DONT_OVERWRITE)
62 return 0;
63 if (flags & AV_DICT_APPEND)
64 oldval = tag->value;
65 else
66 av_free(tag->value);
67 av_free(tag->key);
68 *tag = m->elems[--m->count];
69 } else {
70 AVDictionaryEntry *tmp = av_realloc(m->elems, (m->count+1) * sizeof(*m->elems));
71 if(tmp) {
72 m->elems = tmp;
73 } else
74 return AVERROR(ENOMEM);
75 }
76 if (value) {
77 if (flags & AV_DICT_DONT_STRDUP_KEY) {
78 m->elems[m->count].key = key;
79 } else
80 m->elems[m->count].key = av_strdup(key );
81 if (flags & AV_DICT_DONT_STRDUP_VAL) {
82 m->elems[m->count].value = value;
83 } else if (oldval && flags & AV_DICT_APPEND) {
84 int len = strlen(oldval) + strlen(value) + 1;
85 if (!(oldval = av_realloc(oldval, len)))
86 return AVERROR(ENOMEM);
87 av_strlcat(oldval, value, len);
88 m->elems[m->count].value = oldval;
89 } else
90 m->elems[m->count].value = av_strdup(value);
91 m->count++;
92 }
93 if (!m->count) {
94 av_free(m->elems);
95 av_freep(pm);
96 }
97
98 return 0;
99 }
100
101 void av_dict_free(AVDictionary **pm)
102 {
103 AVDictionary *m = *pm;
104
105 if (m) {
106 while(m->count--) {
107 av_free(m->elems[m->count].key);
108 av_free(m->elems[m->count].value);
109 }
110 av_free(m->elems);
111 }
112 av_freep(pm);
113 }
114
115 void av_dict_copy(AVDictionary **dst, AVDictionary *src, int flags)
116 {
117 AVDictionaryEntry *t = NULL;
118
119 while ((t = av_dict_get(src, "", t, AV_DICT_IGNORE_SUFFIX)))
120 av_dict_set(dst, t->key, t->value, flags);
121 }