lib/util/util_str.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4
   5    Copyright (C) Andrew Tridgell 1992-2001
   6    Copyright (C) Simo Sorce      2001-2002
   7    Copyright (C) Martin Pool     2003
   8    Copyright (C) James Peach     2005
   9
  10    This program is free software; you can redistribute it and/or modify
  11    it under the terms of the GNU General Public License as published by
  12    the Free Software Foundation; either version 3 of the License, or
  13    (at your option) any later version.
  14
  15    This program is distributed in the hope that it will be useful,
  16    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18    GNU General Public License for more details.
  19
  20    You should have received a copy of the GNU General Public License
  21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22 */
  23
  24 #include "includes.h"
  25 #include "system/locale.h"
  26 #undef strncasecmp
  27 #undef strcasemp
  28
  29 /**
  30  * @file
  31  * @brief String utilities.
  32  **/
  33
  34 /**
  35  Safe string copy into a known length string. maxlength does not
  36  include the terminating zero.
  37 **/
  38 _PUBLIC_ char *safe_strcpy(char *dest,const char *src, size_t maxlength)
  39 {
  40         size_t len;
  41
  42         if (!dest) {
  43                 DEBUG(0,("ERROR: NULL dest in safe_strcpy\n"));
  44                 return NULL;
  45         }
  46
  47 #ifdef DEVELOPER
  48         /* We intentionally write out at the extremity of the destination
  49          * string.  If the destination is too short (e.g. pstrcpy into mallocd
  50          * or fstring) then this should cause an error under a memory
  51          * checker. */
  52         dest[maxlength] = '\0';
  53         if (PTR_DIFF(&len, dest) > 0) {  /* check if destination is on the stack, ok if so */
  54                 log_suspicious_usage("safe_strcpy", src);
  55         }
  56 #endif
  57
  58         if (!src) {
  59                 *dest = 0;
  60                 return dest;
  61         }
  62
  63         len = strlen(src);
  64
  65         if (len > maxlength) {
  66                 DEBUG(0,("ERROR: string overflow by %u (%u - %u) in safe_strcpy [%.50s]\n",
  67                          (uint_t)(len-maxlength), (unsigned)len, (unsigned)maxlength, src));
  68                 len = maxlength;
  69         }
  70
  71         memmove(dest, src, len);
  72         dest[len] = 0;
  73         return dest;
  74 }
  75
  76 /**
  77  Safe string cat into a string. maxlength does not
  78  include the terminating zero.
  79 **/
  80 _PUBLIC_ char *safe_strcat(char *dest, const char *src, size_t maxlength)
  81 {
  82         size_t src_len, dest_len;
  83
  84         if (!dest) {
  85                 DEBUG(0,("ERROR: NULL dest in safe_strcat\n"));
  86                 return NULL;
  87         }
  88
  89         if (!src)
  90                 return dest;
  91
  92 #ifdef DEVELOPER
  93         if (PTR_DIFF(&src_len, dest) > 0) {  /* check if destination is on the stack, ok if so */
  94                 log_suspicious_usage("safe_strcat", src);
  95         }
  96 #endif
  97         src_len = strlen(src);
  98         dest_len = strlen(dest);
  99
 100         if (src_len + dest_len > maxlength) {
 101                 DEBUG(0,("ERROR: string overflow by %d in safe_strcat [%.50s]\n",
 102                          (int)(src_len + dest_len - maxlength), src));
 103                 if (maxlength > dest_len) {
 104                         memcpy(&dest[dest_len], src, maxlength - dest_len);
 105                 }
 106                 dest[maxlength] = 0;
 107                 return NULL;
 108         }
 109
 110         memcpy(&dest[dest_len], src, src_len);
 111         dest[dest_len + src_len] = 0;
 112         return dest;
 113 }
 114
 115 #ifdef VALGRIND
 116 size_t valgrind_strlen(const char *s)
 117 {
 118         size_t count;
 119         for(count = 0; *s++; count++)
 120                 ;
 121         return count;
 122 }
 123 #endif
 124
 125
 126 /**
 127   format a string into length-prefixed dotted domain format, as used in NBT
 128   and in some ADS structures
 129 **/
 130 _PUBLIC_ const char *str_format_nbt_domain(TALLOC_CTX *mem_ctx, const char *s)
 131 {
 132         char *ret;
 133         int i;
 134         if (!s || !*s) {
 135                 return talloc_strdup(mem_ctx, "");
 136         }
 137         ret = talloc_array(mem_ctx, char, strlen(s)+2);
 138         if (!ret) {
 139                 return ret;
 140         }
 141
 142         memcpy(ret+1, s, strlen(s)+1);
 143         ret[0] = '.';
 144
 145         for (i=0;ret[i];i++) {
 146                 if (ret[i] == '.') {
 147                         char *p = strchr(ret+i+1, '.');
 148                         if (p) {
 149                                 ret[i] = p-(ret+i+1);
 150                         } else {
 151                                 ret[i] = strlen(ret+i+1);
 152                         }
 153                 }
 154         }
 155
 156         return ret;
 157 }
 158
 159 /**
 160  * Add a string to an array of strings.
 161  *
 162  * num should be a pointer to an integer that holds the current
 163  * number of elements in strings. It will be updated by this function.
 164  */
 165 _PUBLIC_ bool add_string_to_array(TALLOC_CTX *mem_ctx,
 166                          const char *str, const char ***strings, int *num)
 167 {
 168         char *dup_str = talloc_strdup(mem_ctx, str);
 169
 170         *strings = talloc_realloc(mem_ctx,
 171                                     *strings,
 172                                     const char *, ((*num)+1));
 173
 174         if ((*strings == NULL) || (dup_str == NULL))
 175                 return false;
 176
 177         (*strings)[*num] = dup_str;
 178         *num += 1;
 179
 180         return true;
 181 }
 182
 183 /**
 184  * Parse a string containing a boolean value.
 185  *
 186  * val will be set to the read value.
 187  *
 188  * @retval true if a boolean value was parsed, false otherwise.
 189  */
 190 _PUBLIC_ bool conv_str_bool(const char * str, bool * val)
 191 {
 192         char *  end = NULL;
 193         long    lval;
 194
 195         if (str == NULL || *str == '\0') {
 196                 return false;
 197         }
 198
 199         lval = strtol(str, &end, 10 /* base */);
 200         if (end == NULL || *end != '\0' || end == str) {
 201                 return set_boolean(str, val);
 202         }
 203
 204         *val = (lval) ? true : false;
 205         return true;
 206 }
 207
 208 /**
 209  * Convert a size specification like 16K into an integral number of bytes.
 210  **/
 211 _PUBLIC_ bool conv_str_size(const char * str, uint64_t * val)
 212 {
 213         char *              end = NULL;
 214         unsigned long long  lval;
 215
 216         if (str == NULL || *str == '\0') {
 217                 return false;
 218         }
 219
 220         lval = strtoull(str, &end, 10 /* base */);
 221         if (end == NULL || end == str) {
 222                 return false;
 223         }
 224
 225         if (*end) {
 226                 if (strwicmp(end, "K") == 0) {
 227                         lval *= 1024ULL;
 228                 } else if (strwicmp(end, "M") == 0) {
 229                         lval *= (1024ULL * 1024ULL);
 230                 } else if (strwicmp(end, "G") == 0) {
 231                         lval *= (1024ULL * 1024ULL * 1024ULL);
 232                 } else if (strwicmp(end, "T") == 0) {
 233                         lval *= (1024ULL * 1024ULL * 1024ULL * 1024ULL);
 234                 } else if (strwicmp(end, "P") == 0) {
 235                         lval *= (1024ULL * 1024ULL * 1024ULL * 1024ULL * 1024ULL);
 236                 } else {
 237                         return false;
 238                 }
 239         }
 240
 241         *val = (uint64_t)lval;
 242         return true;
 243 }
 244
 245 /**
 246  * Parse a uint64_t value from a string
 247  *
 248  * val will be set to the value read.
 249  *
 250  * @retval true if parsing was successful, false otherwise
 251  */
 252 _PUBLIC_ bool conv_str_u64(const char * str, uint64_t * val)
 253 {
 254         char *              end = NULL;
 255         unsigned long long  lval;
 256
 257         if (str == NULL || *str == '\0') {
 258                 return false;
 259         }
 260
 261         lval = strtoull(str, &end, 10 /* base */);
 262         if (end == NULL || *end != '\0' || end == str) {
 263                 return false;
 264         }
 265
 266         *val = (uint64_t)lval;
 267         return true;
 268 }
 269
 270 /**
 271 Do a case-insensitive, whitespace-ignoring string compare.
 272 **/
 273 _PUBLIC_ int strwicmp(const char *psz1, const char *psz2)
 274 {
 275         /* if BOTH strings are NULL, return TRUE, if ONE is NULL return */
 276         /* appropriate value. */
 277         if (psz1 == psz2)
 278                 return (0);
 279         else if (psz1 == NULL)
 280                 return (-1);
 281         else if (psz2 == NULL)
 282                 return (1);
 283
 284         /* sync the strings on first non-whitespace */
 285         while (1) {
 286                 while (isspace((int)*psz1))
 287                         psz1++;
 288                 while (isspace((int)*psz2))
 289                         psz2++;
 290                 if (toupper((unsigned char)*psz1) != toupper((unsigned char)*psz2)
 291                     || *psz1 == '\0'
 292                     || *psz2 == '\0')
 293                         break;
 294                 psz1++;
 295                 psz2++;
 296         }
 297         return (*psz1 - *psz2);
 298 }
 299
 300 /**
 301  * Compare 2 strings.
 302  *
 303  * @note The comparison is case-insensitive.
 304  **/
 305 _PUBLIC_ bool strequal(const char *s1, const char *s2)
 306 {
 307         if (s1 == s2)
 308                 return true;
 309         if (!s1 || !s2)
 310                 return false;
 311
 312         return strcasecmp(s1,s2) == 0;
 313 }
 314
 315 _PUBLIC_ size_t ucs2_align(const void *base_ptr, const void *p, int flags)
 316 {
 317         if (flags & (STR_NOALIGN|STR_ASCII))
 318                 return 0;
 319         return PTR_DIFF(p, base_ptr) & 1;
 320 }
 321
 322 /**
 323  String replace.
 324 **/
 325 _PUBLIC_ void string_replace(char *s, char oldc, char newc)
 326 {
 327         while (*s) {
 328                 if (*s == oldc) *s = newc;
 329                 s++;
 330         }
 331 }
 332
 333