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 <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 s1;
  55
  56     do {
  57         if (av_stristart(s1, s2, NULL))
  58             return 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_d2str(double d)
  95 {
  96     char *str= av_malloc(16);
  97     if(str) snprintf(str, 16, "%f", d);
  98     return str;
  99 }
 100
 101 #define WHITESPACES " \n\t"
 102
 103 char *av_get_token(const char **buf, const char *term)
 104 {
 105     char *out = av_malloc(strlen(*buf) + 1);
 106     char *ret= out, *end= out;
 107     const char *p = *buf;
 108     if (!out) return NULL;
 109     p += strspn(p, WHITESPACES);
 110
 111     while(*p && !strspn(p, term)) {
 112         char c = *p++;
 113         if(c == '\\' && *p){
 114             *out++ = *p++;
 115             end= out;
 116         }else if(c == '\''){
 117             while(*p && *p != '\'')
 118                 *out++ = *p++;
 119             if(*p){
 120                 p++;
 121                 end= out;
 122             }
 123         }else{
 124             *out++ = c;
 125         }
 126     }
 127
 128     do{
 129         *out-- = 0;
 130     }while(out >= end && strspn(out, WHITESPACES));
 131
 132     *buf = p;
 133
 134     return ret;
 135 }
 136
 137 #ifdef TEST
 138
 139 #undef printf
 140
 141 int main(void)
 142 {
 143     int i;
 144
 145     printf("Testing av_get_token()\n");
 146     {
 147         const char *strings[] = {
 148             "''",
 149             "",
 150             ":",
 151             "\\",
 152             "'",
 153             "    ''    :",
 154             "    ''  ''  :",
 155             "foo   '' :",
 156             "'foo'",
 157             "foo     ",
 158             "  '  foo  '  ",
 159             "foo\\",
 160             "foo':  blah:blah",
 161             "foo\\:  blah:blah",
 162             "foo\'",
 163             "'foo :  '  :blahblah",
 164             "\\ :blah",
 165             "     foo",
 166             "      foo       ",
 167             "      foo     \\ ",
 168             "foo ':blah",
 169             " foo   bar    :   blahblah",
 170             "\\f\\o\\o",
 171             "'foo : \\ \\  '   : blahblah",
 172             "'\\fo\\o:': blahblah",
 173             "\\'fo\\o\\:':  foo  '  :blahblah"
 174         };
 175
 176         for (i=0; i < FF_ARRAY_ELEMS(strings); i++) {
 177             const char *p= strings[i];
 178             printf("|%s|", p);
 179             printf(" -> |%s|", av_get_token(&p, ":"));
 180             printf(" + |%s|\n", p);
 181         }
 182     }
 183
 184     return 0;
 185 }
 186
 187 #endif /* TEST */