avcodec: add ff_lock/unlock_avcodec 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 #include <ctype.h>
27 #include "avstring.h"
28 #include "mem.h"
29
30 int av_strstart(const char *str, const char *pfx, const char **ptr)
31 {
32     while (*pfx && *pfx == *str) {
33         pfx++;
34         str++;
35     }
36     if (!*pfx && ptr)
37         *ptr = str;
38     return !*pfx;
39 }
40
41 int av_stristart(const char *str, const char *pfx, const char **ptr)
42 {
43     while (*pfx && toupper((unsigned)*pfx) == toupper((unsigned)*str)) {
44         pfx++;
45         str++;
46     }
47     if (!*pfx && ptr)
48         *ptr = str;
49     return !*pfx;
50 }
51
52 char *av_stristr(const char *s1, const char *s2)
53 {
54     if (!*s2)
55         return (char*)(intptr_t)s1;
56
57     do {
58         if (av_stristart(s1, s2, NULL))
59             return (char*)(intptr_t)s1;
60     } while (*s1++);
61
62     return NULL;
63 }
64
65 size_t av_strlcpy(char *dst, const char *src, size_t size)
66 {
67     size_t len = 0;
68     while (++len < size && *src)
69         *dst++ = *src++;
70     if (len <= size)
71         *dst = 0;
72     return len + strlen(src) - 1;
73 }
74
75 size_t av_strlcat(char *dst, const char *src, size_t size)
76 {
77     size_t len = strlen(dst);
78     if (size <= len + 1)
79         return len + strlen(src);
80     return len + av_strlcpy(dst + len, src, size - len);
81 }
82
83 size_t av_strlcatf(char *dst, size_t size, const char *fmt, ...)
84 {
85     int len = strlen(dst);
86     va_list vl;
87
88     va_start(vl, fmt);
89     len += vsnprintf(dst + len, size > len ? size - len : 0, fmt, vl);
90     va_end(vl);
91
92     return len;
93 }
94
95 char *av_asprintf(const char *fmt, ...)
96 {
97     char *p = NULL;
98     va_list va;
99     int len;
100
101     va_start(va, fmt);
102     len = vsnprintf(NULL, 0, fmt, va);
103     va_end(va);
104     if (len < 0)
105         goto end;
106
107     p = av_malloc(len + 1);
108     if (!p)
109         goto end;
110
111     va_start(va, fmt);
112     len = vsnprintf(p, len + 1, fmt, va);
113     va_end(va);
114     if (len < 0)
115         av_freep(&p);
116
117 end:
118     return p;
119 }
120
121 char *av_d2str(double d)
122 {
123     char *str= av_malloc(16);
124     if(str) snprintf(str, 16, "%f", d);
125     return str;
126 }
127
128 #define WHITESPACES " \n\t"
129
130 char *av_get_token(const char **buf, const char *term)
131 {
132     char *out = av_malloc(strlen(*buf) + 1);
133     char *ret= out, *end= out;
134     const char *p = *buf;
135     if (!out) return NULL;
136     p += strspn(p, WHITESPACES);
137
138     while(*p && !strspn(p, term)) {
139         char c = *p++;
140         if(c == '\\' && *p){
141             *out++ = *p++;
142             end= out;
143         }else if(c == '\''){
144             while(*p && *p != '\'')
145                 *out++ = *p++;
146             if(*p){
147                 p++;
148                 end= out;
149             }
150         }else{
151             *out++ = c;
152         }
153     }
154
155     do{
156         *out-- = 0;
157     }while(out >= end && strspn(out, WHITESPACES));
158
159     *buf = p;
160
161     return ret;
162 }
163
164 char *av_strtok(char *s, const char *delim, char **saveptr)
165 {
166     char *tok;
167
168     if (!s && !(s = *saveptr))
169         return NULL;
170
171     /* skip leading delimiters */
172     s += strspn(s, delim);
173
174     /* s now points to the first non delimiter char, or to the end of the string */
175     if (!*s) {
176         *saveptr = NULL;
177         return NULL;
178     }
179     tok = s++;
180
181     /* skip non delimiters */
182     s += strcspn(s, delim);
183     if (*s) {
184         *s = 0;
185         *saveptr = s+1;
186     } else {
187         *saveptr = NULL;
188     }
189
190     return tok;
191 }
192
193 int av_strcasecmp(const char *a, const char *b)
194 {
195     uint8_t c1, c2;
196     do {
197         c1 = av_tolower(*a++);
198         c2 = av_tolower(*b++);
199     } while (c1 && c1 == c2);
200     return c1 - c2;
201 }
202
203 int av_strncasecmp(const char *a, const char *b, size_t n)
204 {
205     const char *end = a + n;
206     uint8_t c1, c2;
207     do {
208         c1 = av_tolower(*a++);
209         c2 = av_tolower(*b++);
210     } while (a < end && c1 && c1 == c2);
211     return c1 - c2;
212 }
213
214 #ifdef TEST
215
216 #include "common.h"
217
218 int main(void)
219 {
220     int i;
221
222     printf("Testing av_get_token()\n");
223     {
224         const char *strings[] = {
225             "''",
226             "",
227             ":",
228             "\\",
229             "'",
230             "    ''    :",
231             "    ''  ''  :",
232             "foo   '' :",
233             "'foo'",
234             "foo     ",
235             "  '  foo  '  ",
236             "foo\\",
237             "foo':  blah:blah",
238             "foo\\:  blah:blah",
239             "foo\'",
240             "'foo :  '  :blahblah",
241             "\\ :blah",
242             "     foo",
243             "      foo       ",
244             "      foo     \\ ",
245             "foo ':blah",
246             " foo   bar    :   blahblah",
247             "\\f\\o\\o",
248             "'foo : \\ \\  '   : blahblah",
249             "'\\fo\\o:': blahblah",
250             "\\'fo\\o\\:':  foo  '  :blahblah"
251         };
252
253         for (i=0; i < FF_ARRAY_ELEMS(strings); i++) {
254             const char *p = strings[i];
255             char *q;
256             printf("|%s|", p);
257             q = av_get_token(&p, ":");
258             printf(" -> |%s|", q);
259             printf(" + |%s|\n", p);
260             av_free(q);
261         }
262     }
263
264     return 0;
265 }
266
267 #endif /* TEST */