Imported Debian version 2.5.0~trusty1.1
[deb_ffmpeg.git] / ffmpeg / libavutil / avstring.c
CommitLineData
2ba45a60
DM
1/*
2 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
3 * Copyright (c) 2007 Mans Rullgard
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22#include <stdarg.h>
23#include <stdint.h>
24#include <stdio.h>
25#include <string.h>
26
27#include "config.h"
28#include "common.h"
29#include "mem.h"
30#include "avassert.h"
31#include "avstring.h"
32#include "bprint.h"
33
34int av_strstart(const char *str, const char *pfx, const char **ptr)
35{
36 while (*pfx && *pfx == *str) {
37 pfx++;
38 str++;
39 }
40 if (!*pfx && ptr)
41 *ptr = str;
42 return !*pfx;
43}
44
45int av_stristart(const char *str, const char *pfx, const char **ptr)
46{
47 while (*pfx && av_toupper((unsigned)*pfx) == av_toupper((unsigned)*str)) {
48 pfx++;
49 str++;
50 }
51 if (!*pfx && ptr)
52 *ptr = str;
53 return !*pfx;
54}
55
56char *av_stristr(const char *s1, const char *s2)
57{
58 if (!*s2)
59 return (char*)(intptr_t)s1;
60
61 do
62 if (av_stristart(s1, s2, NULL))
63 return (char*)(intptr_t)s1;
64 while (*s1++);
65
66 return NULL;
67}
68
69char *av_strnstr(const char *haystack, const char *needle, size_t hay_length)
70{
71 size_t needle_len = strlen(needle);
72 if (!needle_len)
73 return (char*)haystack;
74 while (hay_length >= needle_len) {
75 hay_length--;
76 if (!memcmp(haystack, needle, needle_len))
77 return (char*)haystack;
78 haystack++;
79 }
80 return NULL;
81}
82
83size_t av_strlcpy(char *dst, const char *src, size_t size)
84{
85 size_t len = 0;
86 while (++len < size && *src)
87 *dst++ = *src++;
88 if (len <= size)
89 *dst = 0;
90 return len + strlen(src) - 1;
91}
92
93size_t av_strlcat(char *dst, const char *src, size_t size)
94{
95 size_t len = strlen(dst);
96 if (size <= len + 1)
97 return len + strlen(src);
98 return len + av_strlcpy(dst + len, src, size - len);
99}
100
101size_t av_strlcatf(char *dst, size_t size, const char *fmt, ...)
102{
103 int len = strlen(dst);
104 va_list vl;
105
106 va_start(vl, fmt);
107 len += vsnprintf(dst + len, size > len ? size - len : 0, fmt, vl);
108 va_end(vl);
109
110 return len;
111}
112
113char *av_asprintf(const char *fmt, ...)
114{
115 char *p = NULL;
116 va_list va;
117 int len;
118
119 va_start(va, fmt);
120 len = vsnprintf(NULL, 0, fmt, va);
121 va_end(va);
122 if (len < 0)
123 goto end;
124
125 p = av_malloc(len + 1);
126 if (!p)
127 goto end;
128
129 va_start(va, fmt);
130 len = vsnprintf(p, len + 1, fmt, va);
131 va_end(va);
132 if (len < 0)
133 av_freep(&p);
134
135end:
136 return p;
137}
138
139char *av_d2str(double d)
140{
141 char *str = av_malloc(16);
142 if (str)
143 snprintf(str, 16, "%f", d);
144 return str;
145}
146
147#define WHITESPACES " \n\t"
148
149char *av_get_token(const char **buf, const char *term)
150{
151 char *out = av_malloc(strlen(*buf) + 1);
152 char *ret = out, *end = out;
153 const char *p = *buf;
154 if (!out)
155 return NULL;
156 p += strspn(p, WHITESPACES);
157
158 while (*p && !strspn(p, term)) {
159 char c = *p++;
160 if (c == '\\' && *p) {
161 *out++ = *p++;
162 end = out;
163 } else if (c == '\'') {
164 while (*p && *p != '\'')
165 *out++ = *p++;
166 if (*p) {
167 p++;
168 end = out;
169 }
170 } else {
171 *out++ = c;
172 }
173 }
174
175 do
176 *out-- = 0;
177 while (out >= end && strspn(out, WHITESPACES));
178
179 *buf = p;
180
181 return ret;
182}
183
184char *av_strtok(char *s, const char *delim, char **saveptr)
185{
186 char *tok;
187
188 if (!s && !(s = *saveptr))
189 return NULL;
190
191 /* skip leading delimiters */
192 s += strspn(s, delim);
193
194 /* s now points to the first non delimiter char, or to the end of the string */
195 if (!*s) {
196 *saveptr = NULL;
197 return NULL;
198 }
199 tok = s++;
200
201 /* skip non delimiters */
202 s += strcspn(s, delim);
203 if (*s) {
204 *s = 0;
205 *saveptr = s+1;
206 } else {
207 *saveptr = NULL;
208 }
209
210 return tok;
211}
212
213int av_strcasecmp(const char *a, const char *b)
214{
215 uint8_t c1, c2;
216 do {
217 c1 = av_tolower(*a++);
218 c2 = av_tolower(*b++);
219 } while (c1 && c1 == c2);
220 return c1 - c2;
221}
222
223int av_strncasecmp(const char *a, const char *b, size_t n)
224{
225 const char *end = a + n;
226 uint8_t c1, c2;
227 do {
228 c1 = av_tolower(*a++);
229 c2 = av_tolower(*b++);
230 } while (a < end && c1 && c1 == c2);
231 return c1 - c2;
232}
233
234const char *av_basename(const char *path)
235{
236 char *p = strrchr(path, '/');
237
238#if HAVE_DOS_PATHS
239 char *q = strrchr(path, '\\');
240 char *d = strchr(path, ':');
241
242 p = FFMAX3(p, q, d);
243#endif
244
245 if (!p)
246 return path;
247
248 return p + 1;
249}
250
251const char *av_dirname(char *path)
252{
253 char *p = strrchr(path, '/');
254
255#if HAVE_DOS_PATHS
256 char *q = strrchr(path, '\\');
257 char *d = strchr(path, ':');
258
259 d = d ? d + 1 : d;
260
261 p = FFMAX3(p, q, d);
262#endif
263
264 if (!p)
265 return ".";
266
267 *p = '\0';
268
269 return path;
270}
271
272int av_escape(char **dst, const char *src, const char *special_chars,
273 enum AVEscapeMode mode, int flags)
274{
275 AVBPrint dstbuf;
276
277 av_bprint_init(&dstbuf, 1, AV_BPRINT_SIZE_UNLIMITED);
278 av_bprint_escape(&dstbuf, src, special_chars, mode, flags);
279
280 if (!av_bprint_is_complete(&dstbuf)) {
281 av_bprint_finalize(&dstbuf, NULL);
282 return AVERROR(ENOMEM);
283 } else {
284 av_bprint_finalize(&dstbuf, dst);
285 return dstbuf.len;
286 }
287}
288
289int av_isdigit(int c)
290{
291 return c >= '0' && c <= '9';
292}
293
294int av_isgraph(int c)
295{
296 return c > 32 && c < 127;
297}
298
299int av_isspace(int c)
300{
301 return c == ' ' || c == '\f' || c == '\n' || c == '\r' || c == '\t' ||
302 c == '\v';
303}
304
305int av_isxdigit(int c)
306{
307 c = av_tolower(c);
308 return av_isdigit(c) || (c >= 'a' && c <= 'f');
309}
310
311int av_match_name(const char *name, const char *names)
312{
313 const char *p;
314 int len, namelen;
315
316 if (!name || !names)
317 return 0;
318
319 namelen = strlen(name);
320 while ((p = strchr(names, ','))) {
321 len = FFMAX(p - names, namelen);
322 if (!av_strncasecmp(name, names, len))
323 return 1;
324 names = p + 1;
325 }
326 return !av_strcasecmp(name, names);
327}
328
329int av_utf8_decode(int32_t *codep, const uint8_t **bufp, const uint8_t *buf_end,
330 unsigned int flags)
331{
332 const uint8_t *p = *bufp;
333 uint32_t top;
334 uint64_t code;
335 int ret = 0, tail_len;
336 uint32_t overlong_encoding_mins[6] = {
337 0x00000000, 0x00000080, 0x00000800, 0x00010000, 0x00200000, 0x04000000,
338 };
339
340 if (p >= buf_end)
341 return 0;
342
343 code = *p++;
344
345 /* first sequence byte starts with 10, or is 1111-1110 or 1111-1111,
346 which is not admitted */
347 if ((code & 0xc0) == 0x80 || code >= 0xFE) {
348 ret = AVERROR(EILSEQ);
349 goto end;
350 }
351 top = (code & 128) >> 1;
352
353 tail_len = 0;
354 while (code & top) {
355 int tmp;
356 tail_len++;
357 if (p >= buf_end) {
358 (*bufp) ++;
359 return AVERROR(EILSEQ); /* incomplete sequence */
360 }
361
362 /* we assume the byte to be in the form 10xx-xxxx */
363 tmp = *p++ - 128; /* strip leading 1 */
364 if (tmp>>6) {
365 (*bufp) ++;
366 return AVERROR(EILSEQ);
367 }
368 code = (code<<6) + tmp;
369 top <<= 5;
370 }
371 code &= (top << 1) - 1;
372
373 /* check for overlong encodings */
374 av_assert0(tail_len <= 5);
375 if (code < overlong_encoding_mins[tail_len]) {
376 ret = AVERROR(EILSEQ);
377 goto end;
378 }
379
380 if (code >= 1<<31) {
381 ret = AVERROR(EILSEQ); /* out-of-range value */
382 goto end;
383 }
384
385 *codep = code;
386
387 if (code > 0x10FFFF &&
388 !(flags & AV_UTF8_FLAG_ACCEPT_INVALID_BIG_CODES))
389 ret = AVERROR(EILSEQ);
390 if (code < 0x20 && code != 0x9 && code != 0xA && code != 0xD &&
391 flags & AV_UTF8_FLAG_EXCLUDE_XML_INVALID_CONTROL_CODES)
392 ret = AVERROR(EILSEQ);
393 if (code >= 0xD800 && code <= 0xDFFF &&
394 !(flags & AV_UTF8_FLAG_ACCEPT_SURROGATES))
395 ret = AVERROR(EILSEQ);
396 if ((code == 0xFFFE || code == 0xFFFF) &&
397 !(flags & AV_UTF8_FLAG_ACCEPT_NON_CHARACTERS))
398 ret = AVERROR(EILSEQ);
399
400end:
401 *bufp = p;
402 return ret;
403}
404
f6fa7814
DM
405int av_match_list(const char *name, const char *list, char separator)
406{
407 const char *p, *q;
408
409 for (p = name; p && *p; ) {
410 for (q = list; q && *q; ) {
411 int k;
412 for (k = 0; p[k] == q[k] || (p[k]*q[k] == 0 && p[k]+q[k] == separator); k++)
413 if (k && (!p[k] || p[k] == separator))
414 return 1;
415 q = strchr(q, separator);
416 q += !!q;
417 }
418 p = strchr(p, separator);
419 p += !!p;
420 }
421
422 return 0;
423}
424
2ba45a60
DM
425#ifdef TEST
426
427int main(void)
428{
429 int i;
430 static const char * const strings[] = {
431 "''",
432 "",
433 ":",
434 "\\",
435 "'",
436 " '' :",
437 " '' '' :",
438 "foo '' :",
439 "'foo'",
440 "foo ",
441 " ' foo ' ",
442 "foo\\",
443 "foo': blah:blah",
444 "foo\\: blah:blah",
445 "foo\'",
446 "'foo : ' :blahblah",
447 "\\ :blah",
448 " foo",
449 " foo ",
450 " foo \\ ",
451 "foo ':blah",
452 " foo bar : blahblah",
453 "\\f\\o\\o",
454 "'foo : \\ \\ ' : blahblah",
455 "'\\fo\\o:': blahblah",
456 "\\'fo\\o\\:': foo ' :blahblah"
457 };
458
459 printf("Testing av_get_token()\n");
460 for (i = 0; i < FF_ARRAY_ELEMS(strings); i++) {
461 const char *p = strings[i];
462 char *q;
463 printf("|%s|", p);
464 q = av_get_token(&p, ":");
465 printf(" -> |%s|", q);
466 printf(" + |%s|\n", p);
467 av_free(q);
468 }
469
470 return 0;
471}
472
473#endif /* TEST */