src/util/u_string.h

   1 /**************************************************************************
   2  *
   3  * Copyright 2008 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27
  28 /**
  29  * @file
  30  * Platform independent functions for string manipulation.
  31  *
  32  * @author Jose Fonseca <jfonseca@vmware.com>
  33  */
  34
  35 #ifndef U_STRING_H_
  36 #define U_STRING_H_
  37
  38 #if !defined(XF86_LIBC_H)
  39 #include <stdio.h>
  40 #endif
  41 #include <stdlib.h>
  42 #include <stddef.h>
  43 #include <stdarg.h>
  44 #include <string.h>
  45
  46 #include "util/macros.h" // PRINTFLIKE
  47
  48
  49 #ifdef __cplusplus
  50 extern "C" {
  51 #endif
  52
  53 #ifdef _GNU_SOURCE
  54
  55 #define util_strchrnul strchrnul
  56
  57 #else
  58
  59 static inline char *
  60 util_strchrnul(const char *s, char c)
  61 {
  62    for (; *s && *s != c; ++s);
  63
  64    return (char *)s;
  65 }
  66
  67 #endif
  68
  69 #ifdef _WIN32
  70
  71 static inline int
  72 util_vsnprintf(char *str, size_t size, const char *format, va_list ap)
  73 {
  74    /* We need to use _vscprintf to calculate the length as vsnprintf returns -1
  75     * if the number of characters to write is greater than count.
  76     */
  77    va_list ap_copy;
  78    int ret;
  79    va_copy(ap_copy, ap);
  80    ret = _vsnprintf(str, size, format, ap);
  81    if (ret < 0) {
  82       ret = _vscprintf(format, ap_copy);
  83    }
  84    va_end(ap_copy);
  85    return ret;
  86 }
  87
  88 static inline int
  89    PRINTFLIKE(3, 4)
  90 util_snprintf(char *str, size_t size, const char *format, ...)
  91 {
  92    va_list ap;
  93    int ret;
  94    va_start(ap, format);
  95    ret = util_vsnprintf(str, size, format, ap);
  96    va_end(ap);
  97    return ret;
  98 }
  99
 100 static inline void
 101 util_vsprintf(char *str, const char *format, va_list ap)
 102 {
 103    util_vsnprintf(str, (size_t)-1, format, ap);
 104 }
 105
 106 static inline void
 107    PRINTFLIKE(2, 3)
 108 util_sprintf(char *str, const char *format, ...)
 109 {
 110    va_list ap;
 111    va_start(ap, format);
 112    util_vsnprintf(str, (size_t)-1, format, ap);
 113    va_end(ap);
 114 }
 115
 116 static inline int
 117 util_vasprintf(char **ret, const char *format, va_list ap)
 118 {
 119    va_list ap_copy;
 120
 121    /* Compute length of output string first */
 122    va_copy(ap_copy, ap);
 123    int r = util_vsnprintf(NULL, 0, format, ap_copy);
 124    va_end(ap_copy);
 125
 126    if (r < 0)
 127       return -1;
 128
 129    *ret = (char *) malloc(r + 1);
 130    if (!*ret)
 131       return -1;
 132
 133    /* Print to buffer */
 134    return util_vsnprintf(*ret, r + 1, format, ap);
 135 }
 136
 137 static inline char *
 138 util_strchr(const char *s, char c)
 139 {
 140    char *p = util_strchrnul(s, c);
 141
 142    return *p ? p : NULL;
 143 }
 144
 145 static inline char*
 146 util_strncat(char *dst, const char *src, size_t n)
 147 {
 148    char *p = dst + strlen(dst);
 149    const char *q = src;
 150    size_t i;
 151
 152    for (i = 0; i < n && *q != '\0'; ++i)
 153        *p++ = *q++;
 154    *p = '\0';
 155
 156    return dst;
 157 }
 158
 159 static inline int
 160 util_strcmp(const char *s1, const char *s2)
 161 {
 162    unsigned char u1, u2;
 163
 164    while (1) {
 165       u1 = (unsigned char) *s1++;
 166       u2 = (unsigned char) *s2++;
 167       if (u1 != u2)
 168          return u1 - u2;
 169       if (u1 == '\0')
 170          return 0;
 171    }
 172    return 0;
 173 }
 174
 175 static inline int
 176 util_strncmp(const char *s1, const char *s2, size_t n)
 177 {
 178    unsigned char u1, u2;
 179
 180    while (n-- > 0) {
 181       u1 = (unsigned char) *s1++;
 182       u2 = (unsigned char) *s2++;
 183       if (u1 != u2)
 184          return u1 - u2;
 185       if (u1 == '\0')
 186          return 0;
 187    }
 188    return 0;
 189 }
 190
 191 static inline char *
 192 util_strstr(const char *haystack, const char *needle)
 193 {
 194    const char *p = haystack;
 195    size_t len = strlen(needle);
 196
 197    for (; (p = util_strchr(p, *needle)) != 0; p++) {
 198       if (util_strncmp(p, needle, len) == 0) {
 199          return (char *)p;
 200       }
 201    }
 202    return NULL;
 203 }
 204
 205
 206 #define util_strcasecmp stricmp
 207 #define util_strdup _strdup
 208
 209 #else
 210
 211 #define util_vsnprintf vsnprintf
 212 #define util_snprintf snprintf
 213 #define util_vsprintf vsprintf
 214 #define util_vasprintf vasprintf
 215 #define util_sprintf sprintf
 216 #define util_strchr strchr
 217 #define util_strcmp strcmp
 218 #define util_strncmp strncmp
 219 #define util_strncat strncat
 220 #define util_strstr strstr
 221 #define util_strcasecmp strcasecmp
 222 #define util_strdup strdup
 223
 224 #endif
 225
 226
 227 #ifdef __cplusplus
 228 }
 229 #endif
 230
 231 #endif /* U_STRING_H_ */