avstring: fix return discards qualifiers from pointer target type.
[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 <stdio.h>
24 #include <string.h>
25 #include <ctype.h>
26 #include "avstring.h"
27 #include "mem.h"
28
29 int av_strstart(const char *str, const char *pfx, const char **ptr)
30 {
31     while (*pfx && *pfx == *str) {
32         pfx++;
33         str++;
34     }
35     if (!*pfx && ptr)
36         *ptr = str;
37     return !*pfx;
38 }
39
40 int av_stristart(const char *str, const char *pfx, const char **ptr)
41 {
42     while (*pfx && toupper((unsigned)*pfx) == toupper((unsigned)*str)) {
43         pfx++;
44         str++;
45     }
46     if (!*pfx && ptr)
47         *ptr = str;
48     return !*pfx;
49 }
50
51 char *av_stristr(const char *s1, const char *s2)
52 {
53     if (!*s2)
54         return (char*)(intptr_t)s1;
55
56     do {
57         if (av_stristart(s1, s2, NULL))
58             return (char*)(intptr_t)s1;
59     } while (*s1++);
60
61     return NULL;
62 }
63
64 size_t av_strlcpy(char *dst, const char *src, size_t size)
65 {
66     size_t len = 0;
67     while (++len < size && *src)
68         *dst++ = *src++;
69     if (len <= size)
70         *dst = 0;
71     return len + strlen(src) - 1;
72 }
73
74 size_t av_strlcat(char *dst, const char *src, size_t size)
75 {
76     size_t len = strlen(dst);
77     if (size <= len + 1)
78         return len + strlen(src);
79     return len + av_strlcpy(dst + len, src, size - len);
80 }
81
82 size_t av_strlcatf(char *dst, size_t size, const char *fmt, ...)
83 {
84     int len = strlen(dst);
85     va_list vl;
86
87     va_start(vl, fmt);
88     len += vsnprintf(dst + len, size > len ? size - len : 0, fmt, vl);
89     va_end(vl);
90
91     return len;
92 }
93
94 char *av_asprintf(const char *fmt, ...)
95 {
96     char *p = NULL;
97     va_list va;
98     int len;
99
100     va_start(va, fmt);
101     len = vsnprintf(NULL, 0, fmt, va);
102     va_end(va);
103     if (len < 0)
104         goto end;
105
106     p = av_malloc(len + 1);
107     if (!p)
108         goto end;
109
110     va_start(va, fmt);
111     len = vsnprintf(p, len + 1, fmt, va);
112     va_end(va);
113     if (len < 0)
114         av_freep(&p);
115
116 end:
117     return p;
118 }
119
120 char *av_d2str(double d)
121 {
122     char *str= av_malloc(16);
123     if(str) snprintf(str, 16, "%f", d);
124     return str;
125 }
126
127 #define WHITESPACES " \n\t"
128
129 char *av_get_token(const char **buf, const char *term)
130 {
131     char *out = av_malloc(strlen(*buf) + 1);
132     char *ret= out, *end= out;
133     const char *p = *buf;
134     if (!out) return NULL;
135     p += strspn(p, WHITESPACES);
136
137     while(*p && !strspn(p, term)) {
138         char c = *p++;
139         if(c == '\\' && *p){
140             *out++ = *p++;
141             end= out;
142         }else if(c == '\''){
143             while(*p && *p != '\'')
144                 *out++ = *p++;
145             if(*p){
146                 p++;
147                 end= out;
148             }
149         }else{
150             *out++ = c;
151         }
152     }
153
154     do{
155         *out-- = 0;
156     }while(out >= end && strspn(out, WHITESPACES));
157
158     *buf = p;
159
160     return ret;
161 }
162
163 #ifdef TEST
164
165 #undef printf
166
167 int main(void)
168 {
169     int i;
170
171     printf("Testing av_get_token()\n");
172     {
173         const char *strings[] = {
174             "''",
175             "",
176             ":",
177             "\\",
178             "'",
179             "    ''    :",
180             "    ''  ''  :",
181             "foo   '' :",
182             "'foo'",
183             "foo     ",
184             "  '  foo  '  ",
185             "foo\\",
186             "foo':  blah:blah",
187             "foo\\:  blah:blah",
188             "foo\'",
189             "'foo :  '  :blahblah",
190             "\\ :blah",
191             "     foo",
192             "      foo       ",
193             "      foo     \\ ",
194             "foo ':blah",
195             " foo   bar    :   blahblah",
196             "\\f\\o\\o",
197             "'foo : \\ \\  '   : blahblah",
198             "'\\fo\\o:': blahblah",
199             "\\'fo\\o\\:':  foo  '  :blahblah"
200         };
201
202         for (i=0; i < FF_ARRAY_ELEMS(strings); i++) {
203             const char *p= strings[i];
204             printf("|%s|", p);
205             printf(" -> |%s|", av_get_token(&p, ":"));
206             printf(" + |%s|\n", p);
207         }
208     }
209
210     return 0;
211 }
212
213 #endif /* TEST */