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 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 int av_strcasecmp(const char *a, const char *b)
 138 {
 139     uint8_t c1, c2;
 140     do {
 141         c1 = av_tolower(*a++);
 142         c2 = av_tolower(*b++);
 143     } while (c1 && c1 == c2);
 144     return c1 - c2;
 145 }
 146
 147 int av_strncasecmp(const char *a, const char *b, size_t n)
 148 {
 149     const char *end = a + n;
 150     uint8_t c1, c2;
 151     do {
 152         c1 = av_tolower(*a++);
 153         c2 = av_tolower(*b++);
 154     } while (a < end && c1 && c1 == c2);
 155     return c1 - c2;
 156 }
 157
 158 #ifdef TEST
 159
 160 #undef printf
 161
 162 int main(void)
 163 {
 164     int i;
 165
 166     printf("Testing av_get_token()\n");
 167     {
 168         const char *strings[] = {
 169             "''",
 170             "",
 171             ":",
 172             "\\",
 173             "'",
 174             "    ''    :",
 175             "    ''  ''  :",
 176             "foo   '' :",
 177             "'foo'",
 178             "foo     ",
 179             "  '  foo  '  ",
 180             "foo\\",
 181             "foo':  blah:blah",
 182             "foo\\:  blah:blah",
 183             "foo\'",
 184             "'foo :  '  :blahblah",
 185             "\\ :blah",
 186             "     foo",
 187             "      foo       ",
 188             "      foo     \\ ",
 189             "foo ':blah",
 190             " foo   bar    :   blahblah",
 191             "\\f\\o\\o",
 192             "'foo : \\ \\  '   : blahblah",
 193             "'\\fo\\o:': blahblah",
 194             "\\'fo\\o\\:':  foo  '  :blahblah"
 195         };
 196
 197         for (i=0; i < FF_ARRAY_ELEMS(strings); i++) {
 198             const char *p= strings[i];
 199             printf("|%s|", p);
 200             printf(" -> |%s|", av_get_token(&p, ":"));
 201             printf(" + |%s|\n", p);
 202         }
 203     }
 204
 205     return 0;
 206 }
 207
 208 #endif /* TEST */