ucx/string.c

Tue, 31 Mar 2015 10:17:36 +0200

author
Mike Becker <universe@uap-core.de>
date
Tue, 31 Mar 2015 10:17:36 +0200
changeset 189
f43bbd33fec0
parent 185
a48428642b4e
child 192
1e51558b9d09
permissions
-rw-r--r--

sstrtrim now uses isspace()

olaf@20 1 /*
universe@103 2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
olaf@20 3 *
universe@177 4 * Copyright 2014 Olaf Wintermann. All rights reserved.
universe@103 5 *
universe@103 6 * Redistribution and use in source and binary forms, with or without
universe@103 7 * modification, are permitted provided that the following conditions are met:
universe@103 8 *
universe@103 9 * 1. Redistributions of source code must retain the above copyright
universe@103 10 * notice, this list of conditions and the following disclaimer.
universe@103 11 *
universe@103 12 * 2. Redistributions in binary form must reproduce the above copyright
universe@103 13 * notice, this list of conditions and the following disclaimer in the
universe@103 14 * documentation and/or other materials provided with the distribution.
universe@103 15 *
universe@103 16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
universe@103 17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
universe@103 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
universe@103 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
universe@103 20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
universe@103 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
universe@103 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
universe@103 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
universe@103 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
universe@103 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
universe@103 26 * POSSIBILITY OF SUCH DAMAGE.
olaf@20 27 */
olaf@20 28
olaf@20 29 #include <stdlib.h>
universe@69 30 #include <string.h>
olaf@20 31 #include <stdarg.h>
universe@189 32 #include <ctype.h>
olaf@20 33
olaf@20 34 #include "string.h"
olaf@109 35 #include "allocator.h"
olaf@20 36
universe@116 37 sstr_t sstr(char *cstring) {
olaf@20 38 sstr_t string;
universe@116 39 string.ptr = cstring;
universe@116 40 string.length = strlen(cstring);
olaf@20 41 return string;
olaf@20 42 }
olaf@20 43
universe@116 44 sstr_t sstrn(char *cstring, size_t length) {
olaf@20 45 sstr_t string;
universe@116 46 string.ptr = cstring;
universe@116 47 string.length = length;
olaf@20 48 return string;
olaf@20 49 }
olaf@20 50
olaf@68 51 size_t sstrnlen(size_t n, sstr_t s, ...) {
olaf@20 52 va_list ap;
olaf@20 53 size_t size = s.length;
olaf@20 54 va_start(ap, s);
olaf@20 55
universe@116 56 for (size_t i = 1 ; i < n ; i++) {
olaf@20 57 sstr_t str = va_arg(ap, sstr_t);
olaf@20 58 size += str.length;
olaf@20 59 }
universe@24 60 va_end(ap);
olaf@20 61
olaf@20 62 return size;
olaf@20 63 }
olaf@20 64
olaf@180 65 static sstr_t sstrvcat_a(
olaf@180 66 UcxAllocator *a,
olaf@180 67 size_t count,
olaf@180 68 sstr_t s1,
olaf@180 69 sstr_t s2,
olaf@180 70 va_list ap) {
olaf@180 71 sstr_t str;
olaf@180 72 str.ptr = NULL;
olaf@180 73 str.length = 0;
olaf@180 74 if(count < 2) {
olaf@180 75 return str;
olaf@180 76 }
olaf@180 77
universe@185 78 sstr_t *strings = (sstr_t*) calloc(count, sizeof(sstr_t));
olaf@180 79 if(!strings) {
olaf@180 80 return str;
olaf@180 81 }
olaf@180 82
olaf@180 83 // get all args and overall length
olaf@180 84 strings[0] = s1;
olaf@180 85 strings[1] = s2;
olaf@180 86 size_t strlen = s1.length + s2.length;
olaf@180 87 for (size_t i=2;i<count;i++) {
olaf@180 88 sstr_t s = va_arg (ap, sstr_t);
olaf@180 89 strings[i] = s;
olaf@180 90 strlen += s.length;
olaf@180 91 }
olaf@180 92
olaf@180 93 // create new string
universe@185 94 str.ptr = (char*) almalloc(a, strlen + 1);
olaf@180 95 str.length = strlen;
olaf@180 96 if(!str.ptr) {
olaf@180 97 free(strings);
olaf@180 98 str.length = 0;
olaf@180 99 return str;
olaf@180 100 }
olaf@180 101
olaf@180 102 // concatenate strings
olaf@180 103 size_t pos = 0;
olaf@180 104 for (size_t i=0;i<count;i++) {
olaf@180 105 sstr_t s = strings[i];
olaf@180 106 memcpy(str.ptr + pos, s.ptr, s.length);
olaf@180 107 pos += s.length;
olaf@180 108 }
olaf@180 109
olaf@180 110 str.ptr[str.length] = '\0';
olaf@180 111
olaf@180 112 free(strings);
olaf@180 113
olaf@180 114 return str;
olaf@180 115 }
olaf@180 116
olaf@180 117 sstr_t sstrcat(size_t count, sstr_t s1, sstr_t s2, ...) {
olaf@180 118 va_list ap;
olaf@180 119 va_start(ap, s2);
olaf@180 120 sstr_t s = sstrvcat_a(ucx_default_allocator(), count, s1, s2, ap);
olaf@180 121 va_end(ap);
olaf@180 122 return s;
olaf@180 123 }
olaf@180 124
olaf@180 125 sstr_t sstrcat_a(UcxAllocator *a, size_t count, sstr_t s1, sstr_t s2, ...) {
olaf@180 126 va_list ap;
olaf@180 127 va_start(ap, s2);
olaf@180 128 sstr_t s = sstrvcat_a(a, count, s1, s2, ap);
olaf@180 129 va_end(ap);
olaf@180 130 return s;
olaf@180 131 }
olaf@180 132
olaf@68 133 sstr_t sstrsubs(sstr_t s, size_t start) {
olaf@20 134 return sstrsubsl (s, start, s.length-start);
olaf@20 135 }
olaf@20 136
olaf@68 137 sstr_t sstrsubsl(sstr_t s, size_t start, size_t length) {
olaf@20 138 sstr_t new_sstr;
olaf@104 139 if (start >= s.length) {
universe@173 140 new_sstr.ptr = NULL;
universe@173 141 new_sstr.length = 0;
universe@173 142 } else {
universe@173 143 if (length > s.length-start) {
universe@173 144 length = s.length-start;
universe@173 145 }
universe@173 146 new_sstr.ptr = &s.ptr[start];
universe@173 147 new_sstr.length = length;
olaf@20 148 }
olaf@20 149 return new_sstr;
olaf@20 150 }
olaf@20 151
olaf@108 152 sstr_t sstrchr(sstr_t s, int c) {
olaf@108 153 for(size_t i=0;i<s.length;i++) {
olaf@108 154 if(s.ptr[i] == c) {
olaf@108 155 return sstrsubs(s, i);
olaf@108 156 }
olaf@108 157 }
olaf@108 158 sstr_t n;
olaf@108 159 n.ptr = NULL;
olaf@108 160 n.length = 0;
olaf@108 161 return n;
olaf@108 162 }
olaf@108 163
universe@148 164 sstr_t sstrrchr(sstr_t s, int c) {
universe@148 165 if (s.length > 0) {
universe@152 166 for(size_t i=s.length;i>0;i--) {
universe@152 167 if(s.ptr[i-1] == c) {
universe@152 168 return sstrsubs(s, i-1);
universe@148 169 }
universe@148 170 }
universe@148 171 }
universe@148 172 sstr_t n;
universe@148 173 n.ptr = NULL;
universe@148 174 n.length = 0;
universe@148 175 return n;
universe@148 176 }
universe@148 177
universe@173 178 sstr_t* sstrsplit(sstr_t s, sstr_t d, ssize_t *n) {
universe@125 179 return sstrsplit_a(ucx_default_allocator(), s, d, n);
universe@119 180 }
universe@119 181
universe@173 182 sstr_t* sstrsplit_a(UcxAllocator *allocator, sstr_t s, sstr_t d, ssize_t *n) {
universe@119 183 if (s.length == 0 || d.length == 0) {
universe@119 184 *n = -1;
universe@39 185 return NULL;
universe@39 186 }
universe@39 187
universe@39 188 sstr_t* result;
universe@173 189 ssize_t nmax = *n;
universe@39 190 *n = 1;
universe@39 191
universe@39 192 /* special case: exact match - no processing needed */
universe@119 193 if (sstrcmp(s, d) == 0) {
universe@71 194 *n = 0;
universe@71 195 return NULL;
universe@39 196 }
universe@39 197 sstr_t sv = sstrdup(s);
universe@119 198 if (sv.length == 0) {
universe@119 199 *n = -2;
universe@119 200 return NULL;
universe@119 201 }
universe@39 202
universe@95 203 for (size_t i = 0 ; i < s.length ; i++) {
universe@39 204 if (sv.ptr[i] == d.ptr[0]) {
universe@39 205 _Bool match = 1;
universe@95 206 for (size_t j = 1 ; j < d.length ; j++) {
universe@39 207 if (j+i < s.length) {
universe@39 208 match &= (sv.ptr[i+j] == d.ptr[j]);
universe@39 209 } else {
universe@39 210 match = 0;
universe@39 211 break;
universe@39 212 }
universe@39 213 }
universe@39 214 if (match) {
universe@39 215 (*n)++;
universe@95 216 for (size_t j = 0 ; j < d.length ; j++) {
universe@39 217 sv.ptr[i+j] = 0;
universe@39 218 }
universe@39 219 i += d.length;
universe@39 220 }
universe@39 221 }
universe@39 222 if ((*n) == nmax) break;
universe@39 223 }
universe@173 224 result = (sstr_t*) almalloc(allocator, sizeof(sstr_t)*(*n));
universe@39 225
universe@119 226 if (result) {
universe@119 227 char *pptr = sv.ptr;
universe@173 228 for (ssize_t i = 0 ; i < *n ; i++) {
universe@119 229 size_t l = strlen(pptr);
universe@173 230 char* ptr = (char*) almalloc(allocator, l + 1);
universe@173 231 if (ptr) {
universe@173 232 memcpy(ptr, pptr, l);
universe@173 233 ptr[l] = 0;
universe@39 234
universe@173 235 result[i] = sstrn(ptr, l);
universe@173 236 pptr += l + d.length;
universe@173 237 } else {
universe@173 238 for (ssize_t j = i-1 ; j >= 0 ; j--) {
universe@173 239 alfree(allocator, result[j].ptr);
universe@173 240 }
universe@173 241 alfree(allocator, result);
universe@173 242 *n = -2;
universe@173 243 break;
universe@173 244 }
universe@119 245 }
universe@119 246 } else {
universe@119 247 *n = -2;
universe@39 248 }
universe@119 249
universe@39 250 free(sv.ptr);
universe@39 251
universe@39 252 return result;
universe@39 253 }
universe@39 254
olaf@68 255 int sstrcmp(sstr_t s1, sstr_t s2) {
universe@116 256 if (s1.length == s2.length) {
universe@116 257 return memcmp(s1.ptr, s2.ptr, s1.length);
universe@116 258 } else if (s1.length > s2.length) {
universe@116 259 return 1;
universe@116 260 } else {
universe@116 261 return -1;
universe@116 262 }
olaf@20 263 }
olaf@20 264
universe@149 265 int sstrcasecmp(sstr_t s1, sstr_t s2) {
universe@149 266 if (s1.length == s2.length) {
universe@149 267 #ifdef _WIN32
universe@149 268 return _strnicmp(s1.ptr, s2.ptr, s1.length);
universe@149 269 #else
universe@149 270 return strncasecmp(s1.ptr, s2.ptr, s1.length);
universe@149 271 #endif
universe@149 272 } else if (s1.length > s2.length) {
universe@149 273 return 1;
universe@149 274 } else {
universe@149 275 return -1;
universe@149 276 }
universe@149 277 }
universe@149 278
olaf@68 279 sstr_t sstrdup(sstr_t s) {
universe@125 280 return sstrdup_a(ucx_default_allocator(), s);
olaf@109 281 }
olaf@20 282
universe@125 283 sstr_t sstrdup_a(UcxAllocator *allocator, sstr_t s) {
olaf@109 284 sstr_t newstring;
universe@173 285 newstring.ptr = (char*)almalloc(allocator, s.length + 1);
olaf@109 286 if (newstring.ptr) {
olaf@109 287 newstring.length = s.length;
olaf@109 288 newstring.ptr[newstring.length] = 0;
olaf@109 289
olaf@109 290 memcpy(newstring.ptr, s.ptr, s.length);
olaf@109 291 } else {
olaf@109 292 newstring.length = 0;
olaf@109 293 }
olaf@109 294
olaf@20 295 return newstring;
olaf@20 296 }
olaf@96 297
olaf@96 298 sstr_t sstrtrim(sstr_t string) {
olaf@96 299 sstr_t newstr = string;
universe@189 300
universe@189 301 while (newstr.length > 0 && isspace(*newstr.ptr)) {
universe@189 302 newstr.ptr++;
universe@189 303 newstr.length--;
universe@98 304 }
universe@189 305 while (newstr.length > 0 && isspace(newstr.ptr[newstr.length-1])) {
universe@189 306 newstr.length--;
olaf@96 307 }
olaf@96 308
olaf@96 309 return newstr;
olaf@96 310 }
universe@146 311
universe@146 312 int sstrprefix(sstr_t string, sstr_t prefix) {
universe@146 313 if (string.length == 0) {
universe@146 314 return prefix.length == 0;
universe@146 315 }
universe@146 316 if (prefix.length == 0) {
universe@146 317 return 1;
universe@146 318 }
universe@146 319
universe@146 320 if (prefix.length > string.length) {
universe@146 321 return 0;
universe@146 322 } else {
universe@146 323 return memcmp(string.ptr, prefix.ptr, prefix.length) == 0;
universe@146 324 }
universe@146 325 }
universe@146 326
universe@146 327 int sstrsuffix(sstr_t string, sstr_t suffix) {
universe@146 328 if (string.length == 0) {
universe@146 329 return suffix.length == 0;
universe@146 330 }
universe@146 331 if (suffix.length == 0) {
universe@146 332 return 1;
universe@146 333 }
universe@146 334
universe@146 335 if (suffix.length > string.length) {
universe@146 336 return 0;
universe@146 337 } else {
universe@146 338 return memcmp(string.ptr+string.length-suffix.length,
universe@146 339 suffix.ptr, suffix.length) == 0;
universe@146 340 }
universe@146 341 }

mercurial