1/* $NetBSD: string.c,v 1.3.4.1 2012/06/05 21:15:07 bouyer Exp $ */ 2 3/* 4 * Copyright (C) 2004-2007, 2011, 2012 Internet Systems Consortium, Inc. ("ISC") 5 * Copyright (C) 1999-2001, 2003 Internet Software Consortium. 6 * 7 * Permission to use, copy, modify, and/or distribute this software for any 8 * purpose with or without fee is hereby granted, provided that the above 9 * copyright notice and this permission notice appear in all copies. 10 * 11 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH 12 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY 13 * AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT, 14 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM 15 * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE 16 * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR 17 * PERFORMANCE OF THIS SOFTWARE. 18 */ 19 20/* Id */ 21 22/*! \file */ 23 24#include <config.h> 25 26#include <ctype.h> 27 28#include <isc/mem.h> 29#include <isc/print.h> 30#include <isc/region.h> 31#include <isc/string.h> 32#include <isc/util.h> 33 34static char digits[] = "0123456789abcdefghijklmnoprstuvwxyz"; 35 36isc_uint64_t 37isc_string_touint64(char *source, char **end, int base) { 38 isc_uint64_t tmp; 39 isc_uint64_t overflow; 40 char *s = source; 41 char *o; 42 char c; 43 44 if ((base < 0) || (base == 1) || (base > 36)) { 45 *end = source; 46 return (0); 47 } 48 49 while (*s != 0 && isascii(*s&0xff) && isspace(*s&0xff)) 50 s++; 51 if (*s == '+' /* || *s == '-' */) 52 s++; 53 if (base == 0) { 54 if (*s == '0' && (*(s+1) == 'X' || *(s+1) == 'x')) { 55 s += 2; 56 base = 16; 57 } else if (*s == '0') 58 base = 8; 59 else 60 base = 10; 61 } 62 if (*s == 0) { 63 *end = source; 64 return (0); 65 } 66 overflow = ~0; 67 overflow /= base; 68 tmp = 0; 69 70 while ((c = *s) != 0) { 71 c = tolower(c&0xff); 72 /* end ? */ 73 if ((o = strchr(digits, c)) == NULL) { 74 *end = s; 75 return (tmp); 76 } 77 /* end ? */ 78 if ((o - digits) >= base) { 79 *end = s; 80 return (tmp); 81 } 82 /* overflow ? */ 83 if (tmp > overflow) { 84 *end = source; 85 return (0); 86 } 87 tmp *= base; 88 /* overflow ? */ 89 if ((tmp + (o - digits)) < tmp) { 90 *end = source; 91 return (0); 92 } 93 tmp += o - digits; 94 s++; 95 } 96 *end = s; 97 return (tmp); 98} 99 100isc_result_t 101isc_string_copy(char *target, size_t size, const char *source) { 102 REQUIRE(size > 0U); 103 104 if (strlcpy(target, source, size) >= size) { 105 memset(target, ISC_STRING_MAGIC, size); 106 return (ISC_R_NOSPACE); 107 } 108 109 ENSURE(strlen(target) < size); 110 111 return (ISC_R_SUCCESS); 112} 113 114void 115isc_string_copy_truncate(char *target, size_t size, const char *source) { 116 REQUIRE(size > 0U); 117 118 strlcpy(target, source, size); 119 120 ENSURE(strlen(target) < size); 121} 122 123isc_result_t 124isc_string_append(char *target, size_t size, const char *source) { 125 REQUIRE(size > 0U); 126 REQUIRE(strlen(target) < size); 127 128 if (strlcat(target, source, size) >= size) { 129 memset(target, ISC_STRING_MAGIC, size); 130 return (ISC_R_NOSPACE); 131 } 132 133 ENSURE(strlen(target) < size); 134 135 return (ISC_R_SUCCESS); 136} 137 138void 139isc_string_append_truncate(char *target, size_t size, const char *source) { 140 REQUIRE(size > 0U); 141 REQUIRE(strlen(target) < size); 142 143 strlcat(target, source, size); 144 145 ENSURE(strlen(target) < size); 146} 147 148isc_result_t 149isc_string_printf(char *target, size_t size, const char *format, ...) { 150 va_list args; 151 size_t n; 152 153 REQUIRE(size > 0U); 154 155 va_start(args, format); 156 n = vsnprintf(target, size, format, args); 157 va_end(args); 158 159 if (n >= size) { 160 memset(target, ISC_STRING_MAGIC, size); 161 return (ISC_R_NOSPACE); 162 } 163 164 ENSURE(strlen(target) < size); 165 166 return (ISC_R_SUCCESS); 167} 168 169void 170isc_string_printf_truncate(char *target, size_t size, const char *format, ...) 171{ 172 va_list args; 173 174 REQUIRE(size > 0U); 175 176 va_start(args, format); 177 /* check return code? */ 178 (void)vsnprintf(target, size, format, args); 179 va_end(args); 180 181 ENSURE(strlen(target) < size); 182} 183 184char * 185isc_string_regiondup(isc_mem_t *mctx, const isc_region_t *source) { 186 char *target; 187 188 REQUIRE(mctx != NULL); 189 REQUIRE(source != NULL); 190 191 target = (char *) isc_mem_allocate(mctx, source->length + 1); 192 if (target != NULL) { 193 memcpy(source->base, target, source->length); 194 target[source->length] = '\0'; 195 } 196 197 return (target); 198} 199 200char * 201isc_string_separate(char **stringp, const char *delim) { 202 char *string = *stringp; 203 char *s; 204 const char *d; 205 char sc, dc; 206 207 if (string == NULL) 208 return (NULL); 209 210 for (s = string; (sc = *s) != '\0'; s++) 211 for (d = delim; (dc = *d) != '\0'; d++) 212 if (sc == dc) { 213 *s++ = '\0'; 214 *stringp = s; 215 return (string); 216 } 217 *stringp = NULL; 218 return (string); 219} 220 221size_t 222isc_string_strlcpy(char *dst, const char *src, size_t size) 223{ 224 char *d = dst; 225 const char *s = src; 226 size_t n = size; 227 228 /* Copy as many bytes as will fit */ 229 if (n != 0U && --n != 0U) { 230 do { 231 if ((*d++ = *s++) == 0) 232 break; 233 } while (--n != 0U); 234 } 235 236 /* Not enough room in dst, add NUL and traverse rest of src */ 237 if (n == 0U) { 238 if (size != 0U) 239 *d = '\0'; /* NUL-terminate dst */ 240 while (*s++) 241 ; 242 } 243 244 return(s - src - 1); /* count does not include NUL */ 245} 246 247size_t 248isc_string_strlcat(char *dst, const char *src, size_t size) 249{ 250 char *d = dst; 251 const char *s = src; 252 size_t n = size; 253 size_t dlen; 254 255 /* Find the end of dst and adjust bytes left but don't go past end */ 256 while (n-- != 0U && *d != '\0') 257 d++; 258 dlen = d - dst; 259 n = size - dlen; 260 261 if (n == 0U) 262 return(dlen + strlen(s)); 263 while (*s != '\0') { 264 if (n != 1U) { 265 *d++ = *s; 266 n--; 267 } 268 s++; 269 } 270 *d = '\0'; 271 272 return(dlen + (s - src)); /* count does not include NUL */ 273} 274