lavu: Add av_strnstr()
[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 #include <ctype.h>
27
28 #include "config.h"
29 #include "common.h"
30 #include "mem.h"
31 #include "avstring.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 && toupper((unsigned)*pfx) == 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 s1;
59
60     do
61         if (av_stristart(s1, s2, NULL))
62             return 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 haystack;
73     while (hay_length >= needle_len) {
74         hay_length--;
75         if (!memcmp(haystack, needle, needle_len))
76             return 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_d2str(double d)
113 {
114     char *str = av_malloc(16);
115     if (str)
116         snprintf(str, 16, "%f", d);
117     return str;
118 }
119
120 #define WHITESPACES " \n\t"
121
122 char *av_get_token(const char **buf, const char *term)
123 {
124     char *out     = av_malloc(strlen(*buf) + 1);
125     char *ret     = out, *end = out;
126     const char *p = *buf;
127     if (!out)
128         return NULL;
129     p += strspn(p, WHITESPACES);
130
131     while (*p && !strspn(p, term)) {
132         char c = *p++;
133         if (c == '\\' && *p) {
134             *out++ = *p++;
135             end    = out;
136         } else if (c == '\'') {
137             while (*p && *p != '\'')
138                 *out++ = *p++;
139             if (*p) {
140                 p++;
141                 end = out;
142             }
143         } else {
144             *out++ = c;
145         }
146     }
147
148     do
149         *out-- = 0;
150     while (out >= end && strspn(out, WHITESPACES));
151
152     *buf = p;
153
154     return ret;
155 }
156
157 int av_strcasecmp(const char *a, const char *b)
158 {
159     uint8_t c1, c2;
160     do {
161         c1 = av_tolower(*a++);
162         c2 = av_tolower(*b++);
163     } while (c1 && c1 == c2);
164     return c1 - c2;
165 }
166
167 int av_strncasecmp(const char *a, const char *b, size_t n)
168 {
169     const char *end = a + n;
170     uint8_t c1, c2;
171     do {
172         c1 = av_tolower(*a++);
173         c2 = av_tolower(*b++);
174     } while (a < end && c1 && c1 == c2);
175     return c1 - c2;
176 }
177
178 const char *av_basename(const char *path)
179 {
180     char *p = strrchr(path, '/');
181
182 #if HAVE_DOS_PATHS
183     char *q = strrchr(path, '\\');
184     char *d = strchr(path, ':');
185
186     p = FFMAX3(p, q, d);
187 #endif
188
189     if (!p)
190         return path;
191
192     return p + 1;
193 }
194
195 const char *av_dirname(char *path)
196 {
197     char *p = strrchr(path, '/');
198
199 #if HAVE_DOS_PATHS
200     char *q = strrchr(path, '\\');
201     char *d = strchr(path, ':');
202
203     d = d ? d + 1 : d;
204
205     p = FFMAX3(p, q, d);
206 #endif
207
208     if (!p)
209         return ".";
210
211     *p = '\0';
212
213     return path;
214 }
215
216 #ifdef TEST
217
218 int main(void)
219 {
220     int i;
221     const char *strings[] = {
222         "''",
223         "",
224         ":",
225         "\\",
226         "'",
227         "    ''    :",
228         "    ''  ''  :",
229         "foo   '' :",
230         "'foo'",
231         "foo     ",
232         "  '  foo  '  ",
233         "foo\\",
234         "foo':  blah:blah",
235         "foo\\:  blah:blah",
236         "foo\'",
237         "'foo :  '  :blahblah",
238         "\\ :blah",
239         "     foo",
240         "      foo       ",
241         "      foo     \\ ",
242         "foo ':blah",
243         " foo   bar    :   blahblah",
244         "\\f\\o\\o",
245         "'foo : \\ \\  '   : blahblah",
246         "'\\fo\\o:': blahblah",
247         "\\'fo\\o\\:':  foo  '  :blahblah"
248     };
249
250     printf("Testing av_get_token()\n");
251     for (i = 0; i < FF_ARRAY_ELEMS(strings); i++) {
252         const char *p = strings[i], *q;
253         printf("|%s|", p);
254         q = av_get_token(&p, ":");
255         printf(" -> |%s|", q);
256         printf(" + |%s|\n", p);
257         av_free(q);
258     }
259
260     return 0;
261 }
262
263 #endif /* TEST */