avstring: Expose the simple name match function
[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 Libav.
6  *
7  * Libav 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  * Libav 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 Libav; 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
32 int av_strstart(const char *str, const char *pfx, const char **ptr)
33 {
34     while (*pfx && *pfx == *str) {
35         pfx++;
36         str++;
37     }
38     if (!*pfx && ptr)
39         *ptr = str;
40     return !*pfx;
41 }
42
43 int av_stristart(const char *str, const char *pfx, const char **ptr)
44 {
45     while (*pfx && av_toupper((unsigned)*pfx) == av_toupper((unsigned)*str)) {
46         pfx++;
47         str++;
48     }
49     if (!*pfx && ptr)
50         *ptr = str;
51     return !*pfx;
52 }
53
54 char *av_stristr(const char *s1, const char *s2)
55 {
56     if (!*s2)
57         return s1;
58
59     do
60         if (av_stristart(s1, s2, NULL))
61             return s1;
62     while (*s1++);
63
64     return NULL;
65 }
66
67 char *av_strnstr(const char *haystack, const char *needle, size_t hay_length)
68 {
69     size_t needle_len = strlen(needle);
70     if (!needle_len)
71         return haystack;
72     while (hay_length >= needle_len) {
73         hay_length--;
74         if (!memcmp(haystack, needle, needle_len))
75             return haystack;
76         haystack++;
77     }
78     return NULL;
79 }
80
81 size_t av_strlcpy(char *dst, const char *src, size_t size)
82 {
83     size_t len = 0;
84     while (++len < size && *src)
85         *dst++ = *src++;
86     if (len <= size)
87         *dst = 0;
88     return len + strlen(src) - 1;
89 }
90
91 size_t av_strlcat(char *dst, const char *src, size_t size)
92 {
93     size_t len = strlen(dst);
94     if (size <= len + 1)
95         return len + strlen(src);
96     return len + av_strlcpy(dst + len, src, size - len);
97 }
98
99 size_t av_strlcatf(char *dst, size_t size, const char *fmt, ...)
100 {
101     int len = strlen(dst);
102     va_list vl;
103
104     va_start(vl, fmt);
105     len += vsnprintf(dst + len, size > len ? size - len : 0, fmt, vl);
106     va_end(vl);
107
108     return len;
109 }
110
111 char *av_d2str(double d)
112 {
113     char *str = av_malloc(16);
114     if (str)
115         snprintf(str, 16, "%f", d);
116     return str;
117 }
118
119 #define WHITESPACES " \n\t"
120
121 char *av_get_token(const char **buf, const char *term)
122 {
123     char *out     = av_malloc(strlen(*buf) + 1);
124     char *ret     = out, *end = out;
125     const char *p = *buf;
126     if (!out)
127         return NULL;
128     p += strspn(p, WHITESPACES);
129
130     while (*p && !strspn(p, term)) {
131         char c = *p++;
132         if (c == '\\' && *p) {
133             *out++ = *p++;
134             end    = out;
135         } else if (c == '\'') {
136             while (*p && *p != '\'')
137                 *out++ = *p++;
138             if (*p) {
139                 p++;
140                 end = out;
141             }
142         } else {
143             *out++ = c;
144         }
145     }
146
147     do
148         *out-- = 0;
149     while (out >= end && strspn(out, WHITESPACES));
150
151     *buf = p;
152
153     return ret;
154 }
155
156 int av_strcasecmp(const char *a, const char *b)
157 {
158     uint8_t c1, c2;
159     do {
160         c1 = av_tolower(*a++);
161         c2 = av_tolower(*b++);
162     } while (c1 && c1 == c2);
163     return c1 - c2;
164 }
165
166 int av_strncasecmp(const char *a, const char *b, size_t n)
167 {
168     const char *end = a + n;
169     uint8_t c1, c2;
170     do {
171         c1 = av_tolower(*a++);
172         c2 = av_tolower(*b++);
173     } while (a < end && c1 && c1 == c2);
174     return c1 - c2;
175 }
176
177 const char *av_basename(const char *path)
178 {
179     char *p = strrchr(path, '/');
180
181 #if HAVE_DOS_PATHS
182     char *q = strrchr(path, '\\');
183     char *d = strchr(path, ':');
184
185     p = FFMAX3(p, q, d);
186 #endif
187
188     if (!p)
189         return path;
190
191     return p + 1;
192 }
193
194 const char *av_dirname(char *path)
195 {
196     char *p = strrchr(path, '/');
197
198 #if HAVE_DOS_PATHS
199     char *q = strrchr(path, '\\');
200     char *d = strchr(path, ':');
201
202     d = d ? d + 1 : d;
203
204     p = FFMAX3(p, q, d);
205 #endif
206
207     if (!p)
208         return ".";
209
210     *p = '\0';
211
212     return path;
213 }
214
215 int av_isdigit(int c)
216 {
217     return c >= '0' && c <= '9';
218 }
219
220 int av_isgraph(int c)
221 {
222     return c > 32 && c < 127;
223 }
224
225 int av_isspace(int c)
226 {
227     return c == ' ' || c == '\f' || c == '\n' || c == '\r' || c == '\t' ||
228            c == '\v';
229 }
230
231 int av_isxdigit(int c)
232 {
233     c = av_tolower(c);
234     return av_isdigit(c) || (c >= 'a' && c <= 'f');
235 }
236
237 int av_match_name(const char *name, const char *names)
238 {
239     const char *p;
240     int len, namelen;
241
242     if (!name || !names)
243         return 0;
244
245     namelen = strlen(name);
246     while ((p = strchr(names, ','))) {
247         len = FFMAX(p - names, namelen);
248         if (!av_strncasecmp(name, names, len))
249             return 1;
250         names = p + 1;
251     }
252     return !av_strcasecmp(name, names);
253 }
254
255
256
257 #ifdef TEST
258
259 int main(void)
260 {
261     int i;
262     const char *strings[] = {
263         "''",
264         "",
265         ":",
266         "\\",
267         "'",
268         "    ''    :",
269         "    ''  ''  :",
270         "foo   '' :",
271         "'foo'",
272         "foo     ",
273         "  '  foo  '  ",
274         "foo\\",
275         "foo':  blah:blah",
276         "foo\\:  blah:blah",
277         "foo\'",
278         "'foo :  '  :blahblah",
279         "\\ :blah",
280         "     foo",
281         "      foo       ",
282         "      foo     \\ ",
283         "foo ':blah",
284         " foo   bar    :   blahblah",
285         "\\f\\o\\o",
286         "'foo : \\ \\  '   : blahblah",
287         "'\\fo\\o:': blahblah",
288         "\\'fo\\o\\:':  foo  '  :blahblah"
289     };
290
291     printf("Testing av_get_token()\n");
292     for (i = 0; i < FF_ARRAY_ELEMS(strings); i++) {
293         const char *p = strings[i];
294         char *q;
295         printf("|%s|", p);
296         q = av_get_token(&p, ":");
297         printf(" -> |%s|", q);
298         printf(" + |%s|\n", p);
299         av_free(q);
300     }
301
302     return 0;
303 }
304
305 #endif /* TEST */