git.sesse.net Git - ffmpeg/blob - 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 int av_strcasecmp(const char *a, const char *b)
 193 {
 194     uint8_t c1, c2;
 195     do {
 196         c1 = av_tolower(*a++);
 197         c2 = av_tolower(*b++);
 198     } while (c1 && c1 == c2);
 199     return c1 - c2;
 200 }
 201
 202 int av_strncasecmp(const char *a, const char *b, size_t n)
 203 {
 204     const char *end = a + n;
 205     uint8_t c1, c2;
 206     do {
 207         c1 = av_tolower(*a++);
 208         c2 = av_tolower(*b++);
 209     } while (a < end && c1 && c1 == c2);
 210     return c1 - c2;
 211 }
 212
 213 #ifdef TEST
 214
 215 #undef printf
 216
 217 int main(void)
 218 {
 219     int i;
 220
 221     printf("Testing av_get_token()\n");
 222     {
 223         const char *strings[] = {
 224             "''",
 225             "",
 226             ":",
 227             "\\",
 228             "'",
 229             "    ''    :",
 230             "    ''  ''  :",
 231             "foo   '' :",
 232             "'foo'",
 233             "foo     ",
 234             "  '  foo  '  ",
 235             "foo\\",
 236             "foo':  blah:blah",
 237             "foo\\:  blah:blah",
 238             "foo\'",
 239             "'foo :  '  :blahblah",
 240             "\\ :blah",
 241             "     foo",
 242             "      foo       ",
 243             "      foo     \\ ",
 244             "foo ':blah",
 245             " foo   bar    :   blahblah",
 246             "\\f\\o\\o",
 247             "'foo : \\ \\  '   : blahblah",
 248             "'\\fo\\o:': blahblah",
 249             "\\'fo\\o\\:':  foo  '  :blahblah"
 250         };
 251
 252         for (i=0; i < FF_ARRAY_ELEMS(strings); i++) {
 253             const char *p= strings[i];
 254             printf("|%s|", p);
 255             printf(" -> |%s|", av_get_token(&p, ":"));
 256             printf(" + |%s|\n", p);
 257         }
 258     }
 259
 260     return 0;
 261 }
 262
 263 #endif /* TEST */