Remove incorrect use of ctype.h functions.
[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
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 (char*)(intptr_t)s1;
58
59     do
60         if (av_stristart(s1, s2, NULL))
61             return (char*)(intptr_t)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 (char*)haystack;
72     while (hay_length >= needle_len) {
73         hay_length--;
74         if (!memcmp(haystack, needle, needle_len))
75             return (char*)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_asprintf(const char *fmt, ...)
112 {
113     char *p = NULL;
114     va_list va;
115     int len;
116
117     va_start(va, fmt);
118     len = vsnprintf(NULL, 0, fmt, va);
119     va_end(va);
120     if (len < 0)
121         goto end;
122
123     p = av_malloc(len + 1);
124     if (!p)
125         goto end;
126
127     va_start(va, fmt);
128     len = vsnprintf(p, len + 1, fmt, va);
129     va_end(va);
130     if (len < 0)
131         av_freep(&p);
132
133 end:
134     return p;
135 }
136
137 char *av_d2str(double d)
138 {
139     char *str = av_malloc(16);
140     if (str)
141         snprintf(str, 16, "%f", d);
142     return str;
143 }
144
145 #define WHITESPACES " \n\t"
146
147 char *av_get_token(const char **buf, const char *term)
148 {
149     char *out     = av_malloc(strlen(*buf) + 1);
150     char *ret     = out, *end = out;
151     const char *p = *buf;
152     if (!out)
153         return NULL;
154     p += strspn(p, WHITESPACES);
155
156     while (*p && !strspn(p, term)) {
157         char c = *p++;
158         if (c == '\\' && *p) {
159             *out++ = *p++;
160             end    = out;
161         } else if (c == '\'') {
162             while (*p && *p != '\'')
163                 *out++ = *p++;
164             if (*p) {
165                 p++;
166                 end = out;
167             }
168         } else {
169             *out++ = c;
170         }
171     }
172
173     do
174         *out-- = 0;
175     while (out >= end && strspn(out, WHITESPACES));
176
177     *buf = p;
178
179     return ret;
180 }
181
182 char *av_strtok(char *s, const char *delim, char **saveptr)
183 {
184     char *tok;
185
186     if (!s && !(s = *saveptr))
187         return NULL;
188
189     /* skip leading delimiters */
190     s += strspn(s, delim);
191
192     /* s now points to the first non delimiter char, or to the end of the string */
193     if (!*s) {
194         *saveptr = NULL;
195         return NULL;
196     }
197     tok = s++;
198
199     /* skip non delimiters */
200     s += strcspn(s, delim);
201     if (*s) {
202         *s = 0;
203         *saveptr = s+1;
204     } else {
205         *saveptr = NULL;
206     }
207
208     return tok;
209 }
210
211 int av_strcasecmp(const char *a, const char *b)
212 {
213     uint8_t c1, c2;
214     do {
215         c1 = av_tolower(*a++);
216         c2 = av_tolower(*b++);
217     } while (c1 && c1 == c2);
218     return c1 - c2;
219 }
220
221 int av_strncasecmp(const char *a, const char *b, size_t n)
222 {
223     const char *end = a + n;
224     uint8_t c1, c2;
225     do {
226         c1 = av_tolower(*a++);
227         c2 = av_tolower(*b++);
228     } while (a < end && c1 && c1 == c2);
229     return c1 - c2;
230 }
231
232 const char *av_basename(const char *path)
233 {
234     char *p = strrchr(path, '/');
235
236 #if HAVE_DOS_PATHS
237     char *q = strrchr(path, '\\');
238     char *d = strchr(path, ':');
239
240     p = FFMAX3(p, q, d);
241 #endif
242
243     if (!p)
244         return path;
245
246     return p + 1;
247 }
248
249 const char *av_dirname(char *path)
250 {
251     char *p = strrchr(path, '/');
252
253 #if HAVE_DOS_PATHS
254     char *q = strrchr(path, '\\');
255     char *d = strchr(path, ':');
256
257     d = d ? d + 1 : d;
258
259     p = FFMAX3(p, q, d);
260 #endif
261
262     if (!p)
263         return ".";
264
265     *p = '\0';
266
267     return path;
268 }
269
270 #ifdef TEST
271
272 int main(void)
273 {
274     int i;
275     const char *strings[] = {
276         "''",
277         "",
278         ":",
279         "\\",
280         "'",
281         "    ''    :",
282         "    ''  ''  :",
283         "foo   '' :",
284         "'foo'",
285         "foo     ",
286         "  '  foo  '  ",
287         "foo\\",
288         "foo':  blah:blah",
289         "foo\\:  blah:blah",
290         "foo\'",
291         "'foo :  '  :blahblah",
292         "\\ :blah",
293         "     foo",
294         "      foo       ",
295         "      foo     \\ ",
296         "foo ':blah",
297         " foo   bar    :   blahblah",
298         "\\f\\o\\o",
299         "'foo : \\ \\  '   : blahblah",
300         "'\\fo\\o:': blahblah",
301         "\\'fo\\o\\:':  foo  '  :blahblah"
302     };
303
304     printf("Testing av_get_token()\n");
305     for (i = 0; i < FF_ARRAY_ELEMS(strings); i++) {
306         const char *p = strings[i];
307         char *q;
308         printf("|%s|", p);
309         q = av_get_token(&p, ":");
310         printf(" -> |%s|", q);
311         printf(" + |%s|\n", p);
312         av_free(q);
313     }
314
315     return 0;
316 }
317
318 #endif /* TEST */