avutil/cpu: output cpu data to stdout
[ffmpeg.git] / libavutil / avstring.c
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 "avstring.h"
31 #include "bprint.h"
32
33 int av_strstart(const char *str, const char *pfx, const char **ptr)
34 {
35     while (*pfx && *pfx == *str) {
36         pfx++;
37         str++;
38     }
39     if (!*pfx && ptr)
40         *ptr = str;
41     return !*pfx;
42 }
43
44 int av_stristart(const char *str, const char *pfx, const char **ptr)
45 {
46     while (*pfx && av_toupper((unsigned)*pfx) == av_toupper((unsigned)*str)) {
47         pfx++;
48         str++;
49     }
50     if (!*pfx && ptr)
51         *ptr = str;
52     return !*pfx;
53 }
54
55 char *av_stristr(const char *s1, const char *s2)
56 {
57     if (!*s2)
58         return (char*)(intptr_t)s1;
59
60     do
61         if (av_stristart(s1, s2, NULL))
62             return (char*)(intptr_t)s1;
63     while (*s1++);
64
65     return NULL;
66 }
67
68 char *av_strnstr(const char *haystack, const char *needle, size_t hay_length)
69 {
70     size_t needle_len = strlen(needle);
71     if (!needle_len)
72         return (char*)haystack;
73     while (hay_length >= needle_len) {
74         hay_length--;
75         if (!memcmp(haystack, needle, needle_len))
76             return (char*)haystack;
77         haystack++;
78     }
79     return NULL;
80 }
81
82 size_t av_strlcpy(char *dst, const char *src, size_t size)
83 {
84     size_t len = 0;
85     while (++len < size && *src)
86         *dst++ = *src++;
87     if (len <= size)
88         *dst = 0;
89     return len + strlen(src) - 1;
90 }
91
92 size_t av_strlcat(char *dst, const char *src, size_t size)
93 {
94     size_t len = strlen(dst);
95     if (size <= len + 1)
96         return len + strlen(src);
97     return len + av_strlcpy(dst + len, src, size - len);
98 }
99
100 size_t av_strlcatf(char *dst, size_t size, const char *fmt, ...)
101 {
102     int len = strlen(dst);
103     va_list vl;
104
105     va_start(vl, fmt);
106     len += vsnprintf(dst + len, size > len ? size - len : 0, fmt, vl);
107     va_end(vl);
108
109     return len;
110 }
111
112 char *av_asprintf(const char *fmt, ...)
113 {
114     char *p = NULL;
115     va_list va;
116     int len;
117
118     va_start(va, fmt);
119     len = vsnprintf(NULL, 0, fmt, va);
120     va_end(va);
121     if (len < 0)
122         goto end;
123
124     p = av_malloc(len + 1);
125     if (!p)
126         goto end;
127
128     va_start(va, fmt);
129     len = vsnprintf(p, len + 1, fmt, va);
130     va_end(va);
131     if (len < 0)
132         av_freep(&p);
133
134 end:
135     return p;
136 }
137
138 char *av_d2str(double d)
139 {
140     char *str = av_malloc(16);
141     if (str)
142         snprintf(str, 16, "%f", d);
143     return str;
144 }
145
146 #define WHITESPACES " \n\t"
147
148 char *av_get_token(const char **buf, const char *term)
149 {
150     char *out     = av_malloc(strlen(*buf) + 1);
151     char *ret     = out, *end = out;
152     const char *p = *buf;
153     if (!out)
154         return NULL;
155     p += strspn(p, WHITESPACES);
156
157     while (*p && !strspn(p, term)) {
158         char c = *p++;
159         if (c == '\\' && *p) {
160             *out++ = *p++;
161             end    = out;
162         } else if (c == '\'') {
163             while (*p && *p != '\'')
164                 *out++ = *p++;
165             if (*p) {
166                 p++;
167                 end = out;
168             }
169         } else {
170             *out++ = c;
171         }
172     }
173
174     do
175         *out-- = 0;
176     while (out >= end && strspn(out, WHITESPACES));
177
178     *buf = p;
179
180     return ret;
181 }
182
183 char *av_strtok(char *s, const char *delim, char **saveptr)
184 {
185     char *tok;
186
187     if (!s && !(s = *saveptr))
188         return NULL;
189
190     /* skip leading delimiters */
191     s += strspn(s, delim);
192
193     /* s now points to the first non delimiter char, or to the end of the string */
194     if (!*s) {
195         *saveptr = NULL;
196         return NULL;
197     }
198     tok = s++;
199
200     /* skip non delimiters */
201     s += strcspn(s, delim);
202     if (*s) {
203         *s = 0;
204         *saveptr = s+1;
205     } else {
206         *saveptr = NULL;
207     }
208
209     return tok;
210 }
211
212 int av_strcasecmp(const char *a, const char *b)
213 {
214     uint8_t c1, c2;
215     do {
216         c1 = av_tolower(*a++);
217         c2 = av_tolower(*b++);
218     } while (c1 && c1 == c2);
219     return c1 - c2;
220 }
221
222 int av_strncasecmp(const char *a, const char *b, size_t n)
223 {
224     const char *end = a + n;
225     uint8_t c1, c2;
226     do {
227         c1 = av_tolower(*a++);
228         c2 = av_tolower(*b++);
229     } while (a < end && c1 && c1 == c2);
230     return c1 - c2;
231 }
232
233 const char *av_basename(const char *path)
234 {
235     char *p = strrchr(path, '/');
236
237 #if HAVE_DOS_PATHS
238     char *q = strrchr(path, '\\');
239     char *d = strchr(path, ':');
240
241     p = FFMAX3(p, q, d);
242 #endif
243
244     if (!p)
245         return path;
246
247     return p + 1;
248 }
249
250 const char *av_dirname(char *path)
251 {
252     char *p = strrchr(path, '/');
253
254 #if HAVE_DOS_PATHS
255     char *q = strrchr(path, '\\');
256     char *d = strchr(path, ':');
257
258     d = d ? d + 1 : d;
259
260     p = FFMAX3(p, q, d);
261 #endif
262
263     if (!p)
264         return ".";
265
266     *p = '\0';
267
268     return path;
269 }
270
271 int av_escape(char **dst, const char *src, const char *special_chars,
272               enum AVEscapeMode mode, int flags)
273 {
274     AVBPrint dstbuf;
275
276     av_bprint_init(&dstbuf, 1, AV_BPRINT_SIZE_UNLIMITED);
277     av_bprint_escape(&dstbuf, src, special_chars, mode, flags);
278
279     if (!av_bprint_is_complete(&dstbuf)) {
280         av_bprint_finalize(&dstbuf, NULL);
281         return AVERROR(ENOMEM);
282     } else {
283         av_bprint_finalize(&dstbuf, dst);
284         return dstbuf.len;
285     }
286 }
287
288 int av_isdigit(int c)
289 {
290     return c >= '0' && c <= '9';
291 }
292
293 int av_isgraph(int c)
294 {
295     return c > 32 && c < 127;
296 }
297
298 int av_isspace(int c)
299 {
300     return c == ' ' || c == '\f' || c == '\n' || c == '\r' || c == '\t' ||
301            c == '\v';
302 }
303
304 int av_isxdigit(int c)
305 {
306     c = av_tolower(c);
307     return av_isdigit(c) || (c >= 'a' && c <= 'f');
308 }
309
310 int av_match_name(const char *name, const char *names)
311 {
312     const char *p;
313     int len, namelen;
314
315     if (!name || !names)
316         return 0;
317
318     namelen = strlen(name);
319     while ((p = strchr(names, ','))) {
320         len = FFMAX(p - names, namelen);
321         if (!av_strncasecmp(name, names, len))
322             return 1;
323         names = p + 1;
324     }
325     return !av_strcasecmp(name, names);
326 }
327
328 int av_utf8_decode(int32_t *codep, const uint8_t **bufp, const uint8_t *buf_end,
329                    unsigned int flags)
330 {
331     const uint8_t *p = *bufp;
332     uint32_t top;
333     uint64_t code;
334     int ret = 0;
335
336     if (p >= buf_end)
337         return 0;
338
339     code = *p++;
340
341     /* first sequence byte starts with 10, or is 1111-1110 or 1111-1111,
342        which is not admitted */
343     if ((code & 0xc0) == 0x80 || code >= 0xFE) {
344         ret = AVERROR(EILSEQ);
345         goto end;
346     }
347     top = (code & 128) >> 1;
348
349     while (code & top) {
350         int tmp;
351         if (p >= buf_end) {
352             (*bufp) ++;
353             return AVERROR(EILSEQ); /* incomplete sequence */
354         }
355
356         /* we assume the byte to be in the form 10xx-xxxx */
357         tmp = *p++ - 128;   /* strip leading 1 */
358         if (tmp>>6) {
359             (*bufp) ++;
360             return AVERROR(EILSEQ);
361         }
362         code = (code<<6) + tmp;
363         top <<= 5;
364     }
365     code &= (top << 1) - 1;
366
367     if (code >= 1<<31) {
368         ret = AVERROR(EILSEQ);  /* out-of-range value */
369         goto end;
370     }
371
372     *codep = code;
373
374     if (code > 0x10FFFF &&
375         !(flags & AV_UTF8_FLAG_ACCEPT_INVALID_BIG_CODES))
376         ret = AVERROR(EILSEQ);
377     if (code < 0x20 && code != 0x9 && code != 0xA && code != 0xD &&
378         flags & AV_UTF8_FLAG_EXCLUDE_XML_INVALID_CONTROL_CODES)
379         ret = AVERROR(EILSEQ);
380     if (code >= 0xD800 && code <= 0xDFFF &&
381         !(flags & AV_UTF8_FLAG_ACCEPT_SURROGATES))
382         ret = AVERROR(EILSEQ);
383     if ((code == 0xFFFE || code == 0xFFFF) &&
384         !(flags & AV_UTF8_FLAG_ACCEPT_NON_CHARACTERS))
385         ret = AVERROR(EILSEQ);
386
387 end:
388     *bufp = p;
389     return ret;
390 }
391
392 #ifdef TEST
393
394 int main(void)
395 {
396     int i;
397     static const char * const strings[] = {
398         "''",
399         "",
400         ":",
401         "\\",
402         "'",
403         "    ''    :",
404         "    ''  ''  :",
405         "foo   '' :",
406         "'foo'",
407         "foo     ",
408         "  '  foo  '  ",
409         "foo\\",
410         "foo':  blah:blah",
411         "foo\\:  blah:blah",
412         "foo\'",
413         "'foo :  '  :blahblah",
414         "\\ :blah",
415         "     foo",
416         "      foo       ",
417         "      foo     \\ ",
418         "foo ':blah",
419         " foo   bar    :   blahblah",
420         "\\f\\o\\o",
421         "'foo : \\ \\  '   : blahblah",
422         "'\\fo\\o:': blahblah",
423         "\\'fo\\o\\:':  foo  '  :blahblah"
424     };
425
426     printf("Testing av_get_token()\n");
427     for (i = 0; i < FF_ARRAY_ELEMS(strings); i++) {
428         const char *p = strings[i];
429         char *q;
430         printf("|%s|", p);
431         q = av_get_token(&p, ":");
432         printf(" -> |%s|", q);
433         printf(" + |%s|\n", p);
434         av_free(q);
435     }
436
437     return 0;
438 }
439
440 #endif /* TEST */