]> git.sesse.net Git - ffmpeg/blob - libavutil/avstring.c
Merge remote-tracking branch 'mans/dnxhd'
[ffmpeg] / 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 char *av_strtok(char *s, const char *delim, char **saveptr)
164 {
165     char *tok;
166
167     if (!s && !(s = *saveptr))
168         return NULL;
169
170     /* skip leading delimiters */
171     s += strspn(s, delim);
172
173     /* s now points to the first non delimiter char, or to the end of the string */
174     if (!*s) {
175         *saveptr = NULL;
176         return NULL;
177     }
178     tok = s++;
179
180     /* skip non delimiters */
181     s += strcspn(s, delim);
182     if (*s) {
183         *s = 0;
184         *saveptr = s+1;
185     } else {
186         *saveptr = NULL;
187     }
188
189     return tok;
190 }
191
192 #ifdef TEST
193
194 #undef printf
195
196 int main(void)
197 {
198     int i;
199
200     printf("Testing av_get_token()\n");
201     {
202         const char *strings[] = {
203             "''",
204             "",
205             ":",
206             "\\",
207             "'",
208             "    ''    :",
209             "    ''  ''  :",
210             "foo   '' :",
211             "'foo'",
212             "foo     ",
213             "  '  foo  '  ",
214             "foo\\",
215             "foo':  blah:blah",
216             "foo\\:  blah:blah",
217             "foo\'",
218             "'foo :  '  :blahblah",
219             "\\ :blah",
220             "     foo",
221             "      foo       ",
222             "      foo     \\ ",
223             "foo ':blah",
224             " foo   bar    :   blahblah",
225             "\\f\\o\\o",
226             "'foo : \\ \\  '   : blahblah",
227             "'\\fo\\o:': blahblah",
228             "\\'fo\\o\\:':  foo  '  :blahblah"
229         };
230
231         for (i=0; i < FF_ARRAY_ELEMS(strings); i++) {
232             const char *p= strings[i];
233             printf("|%s|", p);
234             printf(" -> |%s|", av_get_token(&p, ":"));
235             printf(" + |%s|\n", p);
236         }
237     }
238
239     return 0;
240 }
241
242 #endif /* TEST */