FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
dict.c
Go to the documentation of this file.
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 <string.h>
22 
23 #include "avstring.h"
24 #include "dict.h"
25 #include "internal.h"
26 #include "mem.h"
27 #include "bprint.h"
28 
29 struct AVDictionary {
30  int count;
32 };
33 
35 {
36  return m ? m->count : 0;
37 }
38 
39 AVDictionaryEntry *av_dict_get(const AVDictionary *m, const char *key,
40  const AVDictionaryEntry *prev, int flags)
41 {
42  unsigned int i, j;
43 
44  if (!m)
45  return NULL;
46 
47  if (prev)
48  i = prev - m->elems + 1;
49  else
50  i = 0;
51 
52  for (; i < m->count; i++) {
53  const char *s = m->elems[i].key;
54  if (flags & AV_DICT_MATCH_CASE)
55  for (j = 0; s[j] == key[j] && key[j]; j++)
56  ;
57  else
58  for (j = 0; av_toupper(s[j]) == av_toupper(key[j]) && key[j]; j++)
59  ;
60  if (key[j])
61  continue;
62  if (s[j] && !(flags & AV_DICT_IGNORE_SUFFIX))
63  continue;
64  return &m->elems[i];
65  }
66  return NULL;
67 }
68 
69 int av_dict_set(AVDictionary **pm, const char *key, const char *value,
70  int flags)
71 {
72  AVDictionary *m = *pm;
74  char *oldval = NULL, *copy_key = NULL, *copy_value = NULL;
75 
76  if (!(flags & AV_DICT_MULTIKEY)) {
77  tag = av_dict_get(m, key, NULL, flags);
78  }
79  if (flags & AV_DICT_DONT_STRDUP_KEY)
80  copy_key = (void *)key;
81  else
82  copy_key = av_strdup(key);
83  if (flags & AV_DICT_DONT_STRDUP_VAL)
84  copy_value = (void *)value;
85  else if (copy_key)
86  copy_value = av_strdup(value);
87  if (!m)
88  m = *pm = av_mallocz(sizeof(*m));
89  if (!m || (key && !copy_key) || (value && !copy_value))
90  goto err_out;
91 
92  if (tag) {
93  if (flags & AV_DICT_DONT_OVERWRITE) {
94  av_free(copy_key);
95  av_free(copy_value);
96  return 0;
97  }
98  if (flags & AV_DICT_APPEND)
99  oldval = tag->value;
100  else
101  av_free(tag->value);
102  av_free(tag->key);
103  *tag = m->elems[--m->count];
104  } else if (copy_value) {
106  (m->count + 1) * sizeof(*m->elems));
107  if (!tmp)
108  goto err_out;
109  m->elems = tmp;
110  }
111  if (copy_value) {
112  m->elems[m->count].key = copy_key;
113  m->elems[m->count].value = copy_value;
114  if (oldval && flags & AV_DICT_APPEND) {
115  size_t len = strlen(oldval) + strlen(copy_value) + 1;
116  char *newval = av_mallocz(len);
117  if (!newval)
118  goto err_out;
119  av_strlcat(newval, oldval, len);
120  av_freep(&oldval);
121  av_strlcat(newval, copy_value, len);
122  m->elems[m->count].value = newval;
123  av_freep(&copy_value);
124  }
125  m->count++;
126  } else {
127  av_freep(&copy_key);
128  }
129  if (!m->count) {
130  av_freep(&m->elems);
131  av_freep(pm);
132  }
133 
134  return 0;
135 
136 err_out:
137  if (m && !m->count) {
138  av_freep(&m->elems);
139  av_freep(pm);
140  }
141  av_free(copy_key);
142  av_free(copy_value);
143  return AVERROR(ENOMEM);
144 }
145 
146 int av_dict_set_int(AVDictionary **pm, const char *key, int64_t value,
147  int flags)
148 {
149  char valuestr[22];
150  snprintf(valuestr, sizeof(valuestr), "%"PRId64, value);
151  flags &= ~AV_DICT_DONT_STRDUP_VAL;
152  return av_dict_set(pm, key, valuestr, flags);
153 }
154 
155 static int parse_key_value_pair(AVDictionary **pm, const char **buf,
156  const char *key_val_sep, const char *pairs_sep,
157  int flags)
158 {
159  char *key = av_get_token(buf, key_val_sep);
160  char *val = NULL;
161  int ret;
162 
163  if (key && *key && strspn(*buf, key_val_sep)) {
164  (*buf)++;
165  val = av_get_token(buf, pairs_sep);
166  }
167 
168  if (key && *key && val && *val)
169  ret = av_dict_set(pm, key, val, flags);
170  else
171  ret = AVERROR(EINVAL);
172 
173  av_freep(&key);
174  av_freep(&val);
175 
176  return ret;
177 }
178 
179 int av_dict_parse_string(AVDictionary **pm, const char *str,
180  const char *key_val_sep, const char *pairs_sep,
181  int flags)
182 {
183  int ret;
184 
185  if (!str)
186  return 0;
187 
188  /* ignore STRDUP flags */
190 
191  while (*str) {
192  if ((ret = parse_key_value_pair(pm, &str, key_val_sep, pairs_sep, flags)) < 0)
193  return ret;
194 
195  if (*str)
196  str++;
197  }
198 
199  return 0;
200 }
201 
203 {
204  AVDictionary *m = *pm;
205 
206  if (m) {
207  while (m->count--) {
208  av_freep(&m->elems[m->count].key);
209  av_freep(&m->elems[m->count].value);
210  }
211  av_freep(&m->elems);
212  }
213  av_freep(pm);
214 }
215 
217 {
218  AVDictionaryEntry *t = NULL;
219 
220  while ((t = av_dict_get(src, "", t, AV_DICT_IGNORE_SUFFIX))) {
221  int ret = av_dict_set(dst, t->key, t->value, flags);
222  if (ret < 0)
223  return ret;
224  }
225 
226  return 0;
227 }
228 
230  const char key_val_sep, const char pairs_sep)
231 {
232  AVDictionaryEntry *t = NULL;
233  AVBPrint bprint;
234  int cnt = 0;
235  char special_chars[] = {pairs_sep, key_val_sep, '\0'};
236 
237  if (!buffer || pairs_sep == '\0' || key_val_sep == '\0' || pairs_sep == key_val_sep ||
238  pairs_sep == '\\' || key_val_sep == '\\')
239  return AVERROR(EINVAL);
240 
241  if (!av_dict_count(m)) {
242  *buffer = av_strdup("");
243  return *buffer ? 0 : AVERROR(ENOMEM);
244  }
245 
247  while ((t = av_dict_get(m, "", t, AV_DICT_IGNORE_SUFFIX))) {
248  if (cnt++)
249  av_bprint_append_data(&bprint, &pairs_sep, 1);
250  av_bprint_escape(&bprint, t->key, special_chars, AV_ESCAPE_MODE_BACKSLASH, 0);
251  av_bprint_append_data(&bprint, &key_val_sep, 1);
252  av_bprint_escape(&bprint, t->value, special_chars, AV_ESCAPE_MODE_BACKSLASH, 0);
253  }
254  return av_bprint_finalize(&bprint, buffer);
255 }
#define NULL
Definition: coverity.c:32
const char const char void * val
Definition: avisynth_c.h:634
int count
Definition: dict.c:30
const char * s
Definition: avisynth_c.h:631
memory handling functions
#define AV_DICT_DONT_OVERWRITE
Don't overwrite existing entries.
Definition: dict.h:79
int av_dict_count(const AVDictionary *m)
Get number of entries in dictionary.
Definition: dict.c:34
int av_dict_get_string(const AVDictionary *m, char **buffer, const char key_val_sep, const char pairs_sep)
Get dictionary entries as a string.
Definition: dict.c:229
int av_dict_copy(AVDictionary **dst, const AVDictionary *src, int flags)
Copy entries from one AVDictionary struct into another.
Definition: dict.c:216
Use backslash escaping.
Definition: avstring.h:310
int av_bprint_finalize(AVBPrint *buf, char **ret_str)
Finalize a print buffer.
Definition: bprint.c:235
#define AV_DICT_DONT_STRDUP_KEY
Take ownership of a key that's been allocated with av_malloc() or another memory allocation function...
Definition: dict.h:73
Public dictionary API.
#define AV_DICT_MULTIKEY
Allow to store several equal keys in the dictionary.
Definition: dict.h:83
void av_bprint_append_data(AVBPrint *buf, const char *data, unsigned size)
Append data to a print buffer.
Definition: bprint.c:158
AVDictionaryEntry * av_dict_get(const AVDictionary *m, const char *key, const AVDictionaryEntry *prev, int flags)
Get a dictionary entry with matching key.
Definition: dict.c:39
uint32_t tag
Definition: movenc.c:1376
unsigned m
Definition: audioconvert.c:187
#define AV_DICT_MATCH_CASE
Only get an entry with exact-case key match.
Definition: dict.h:69
#define AV_BPRINT_SIZE_UNLIMITED
#define AVERROR(e)
Definition: error.h:43
void av_bprint_init(AVBPrint *buf, unsigned size_init, unsigned size_max)
Definition: bprint.c:69
void av_dict_free(AVDictionary **pm)
Free all the memory allocated for an AVDictionary struct and all keys and values. ...
Definition: dict.c:202
char * av_get_token(const char **buf, const char *term)
Unescape the given string until a non escaped terminating char, and return the token corresponding to...
Definition: avstring.c:149
common internal API header
AVDictionaryEntry * elems
Definition: dict.c:31
#define AV_DICT_DONT_STRDUP_VAL
Take ownership of a value that's been allocated with av_malloc() or another memory allocation functio...
Definition: dict.h:76
GLsizei GLboolean const GLfloat * value
Definition: opengl_enc.c:109
#define AV_DICT_APPEND
If the entry already exists, append to it.
Definition: dict.h:80
#define src
Definition: vp9dsp.c:530
int av_dict_parse_string(AVDictionary **pm, const char *str, const char *key_val_sep, const char *pairs_sep, int flags)
Parse the key/value pairs list and add the parsed entries to a dictionary.
Definition: dict.c:179
char * av_strdup(const char *s)
Duplicate the string s.
Definition: mem.c:267
void * buf
Definition: avisynth_c.h:553
int av_dict_set(AVDictionary **pm, const char *key, const char *value, int flags)
Set the given entry in *pm, overwriting an existing entry.
Definition: dict.c:69
static av_const int av_toupper(int c)
Locale-independent conversion of ASCII characters to uppercase.
Definition: avstring.h:231
#define snprintf
Definition: snprintf.h:34
size_t av_strlcat(char *dst, const char *src, size_t size)
Append the string src to the string dst, but to a total length of no more than size - 1 bytes...
Definition: avstring.c:93
static int flags
Definition: cpu.c:47
void av_bprint_escape(AVBPrint *dstbuf, const char *src, const char *special_chars, enum AVEscapeMode mode, int flags)
Escape the content in src and append it to dstbuf.
Definition: bprint.c:265
void * av_realloc(void *ptr, size_t size)
Allocate or reallocate a block of memory.
Definition: mem.c:145
int av_dict_set_int(AVDictionary **pm, const char *key, int64_t value, int flags)
Convenience wrapper for av_dict_set that converts the value to a string and stores it...
Definition: dict.c:146
char * key
Definition: dict.h:86
#define av_free(p)
char * value
Definition: dict.h:87
int len
static uint8_t tmp[8]
Definition: des.c:38
#define av_freep(p)
#define AV_DICT_IGNORE_SUFFIX
Return first entry in a dictionary whose first part corresponds to the search key, ignoring the suffix of the found key string.
Definition: dict.h:70
void * av_mallocz(size_t size)
Allocate a block of size bytes with alignment suitable for all memory accesses (including vectors if ...
Definition: mem.c:252
static int parse_key_value_pair(AVDictionary **pm, const char **buf, const char *key_val_sep, const char *pairs_sep, int flags)
Definition: dict.c:155
GLuint buffer
Definition: opengl_enc.c:102