ucx/string.c

Tue, 27 Oct 2015 15:29:34 +0100

author
Mike Becker <universe@uap-core.de>
date
Tue, 27 Oct 2015 15:29:34 +0100
changeset 214
2bc19726c340
parent 213
fa8b745be7b5
child 225
a1a068c2c4ef
permissions
-rw-r--r--

added sstrstr() function + improved sstrsplit variants by using sstrprefix()

olaf@20 1 /*
universe@103 2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
olaf@20 3 *
universe@192 4 * Copyright 2015 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@214 178 sstr_t sstrstr(sstr_t string, sstr_t match) {
universe@214 179 if (match.length == 0) {
universe@214 180 return string;
universe@214 181 }
universe@214 182
universe@214 183 for (size_t i = 0 ; i < string.length ; i++) {
universe@214 184 sstr_t substr = sstrsubs(string, i);
universe@214 185 if (sstrprefix(substr, match)) {
universe@214 186 return substr;
universe@214 187 }
universe@214 188 }
universe@214 189
universe@214 190 sstr_t emptystr;
universe@214 191 emptystr.length = 0;
universe@214 192 emptystr.ptr = NULL;
universe@214 193 return emptystr;
universe@214 194 }
universe@214 195
universe@173 196 sstr_t* sstrsplit(sstr_t s, sstr_t d, ssize_t *n) {
universe@125 197 return sstrsplit_a(ucx_default_allocator(), s, d, n);
universe@119 198 }
universe@119 199
universe@173 200 sstr_t* sstrsplit_a(UcxAllocator *allocator, sstr_t s, sstr_t d, ssize_t *n) {
universe@119 201 if (s.length == 0 || d.length == 0) {
universe@119 202 *n = -1;
universe@39 203 return NULL;
universe@39 204 }
universe@39 205
universe@39 206 sstr_t* result;
universe@173 207 ssize_t nmax = *n;
universe@39 208 *n = 1;
universe@39 209
universe@39 210 /* special case: exact match - no processing needed */
universe@119 211 if (sstrcmp(s, d) == 0) {
universe@71 212 *n = 0;
universe@71 213 return NULL;
universe@39 214 }
universe@39 215 sstr_t sv = sstrdup(s);
universe@119 216 if (sv.length == 0) {
universe@119 217 *n = -2;
universe@119 218 return NULL;
universe@119 219 }
universe@39 220
universe@95 221 for (size_t i = 0 ; i < s.length ; i++) {
universe@214 222 sstr_t substr = sstrsubs(sv, i);
universe@214 223 if (sstrprefix(substr, d)) {
universe@214 224 (*n)++;
universe@214 225 for (size_t j = 0 ; j < d.length ; j++) {
universe@214 226 sv.ptr[i+j] = 0;
universe@39 227 }
universe@214 228 i += d.length - 1; // -1, because the loop will do a i++
universe@39 229 }
universe@39 230 if ((*n) == nmax) break;
universe@39 231 }
universe@173 232 result = (sstr_t*) almalloc(allocator, sizeof(sstr_t)*(*n));
universe@39 233
universe@119 234 if (result) {
universe@119 235 char *pptr = sv.ptr;
universe@173 236 for (ssize_t i = 0 ; i < *n ; i++) {
universe@119 237 size_t l = strlen(pptr);
universe@173 238 char* ptr = (char*) almalloc(allocator, l + 1);
universe@173 239 if (ptr) {
universe@173 240 memcpy(ptr, pptr, l);
universe@173 241 ptr[l] = 0;
universe@39 242
universe@173 243 result[i] = sstrn(ptr, l);
universe@173 244 pptr += l + d.length;
universe@173 245 } else {
universe@173 246 for (ssize_t j = i-1 ; j >= 0 ; j--) {
universe@173 247 alfree(allocator, result[j].ptr);
universe@173 248 }
universe@173 249 alfree(allocator, result);
universe@173 250 *n = -2;
universe@173 251 break;
universe@173 252 }
universe@119 253 }
universe@119 254 } else {
universe@119 255 *n = -2;
universe@39 256 }
universe@119 257
universe@39 258 free(sv.ptr);
universe@39 259
universe@39 260 return result;
universe@39 261 }
universe@39 262
olaf@68 263 int sstrcmp(sstr_t s1, sstr_t s2) {
universe@116 264 if (s1.length == s2.length) {
universe@116 265 return memcmp(s1.ptr, s2.ptr, s1.length);
universe@116 266 } else if (s1.length > s2.length) {
universe@116 267 return 1;
universe@116 268 } else {
universe@116 269 return -1;
universe@116 270 }
olaf@20 271 }
olaf@20 272
universe@149 273 int sstrcasecmp(sstr_t s1, sstr_t s2) {
universe@149 274 if (s1.length == s2.length) {
universe@149 275 #ifdef _WIN32
universe@149 276 return _strnicmp(s1.ptr, s2.ptr, s1.length);
universe@149 277 #else
universe@149 278 return strncasecmp(s1.ptr, s2.ptr, s1.length);
universe@149 279 #endif
universe@149 280 } else if (s1.length > s2.length) {
universe@149 281 return 1;
universe@149 282 } else {
universe@149 283 return -1;
universe@149 284 }
universe@149 285 }
universe@149 286
olaf@68 287 sstr_t sstrdup(sstr_t s) {
universe@125 288 return sstrdup_a(ucx_default_allocator(), s);
olaf@109 289 }
olaf@20 290
universe@125 291 sstr_t sstrdup_a(UcxAllocator *allocator, sstr_t s) {
olaf@109 292 sstr_t newstring;
universe@173 293 newstring.ptr = (char*)almalloc(allocator, s.length + 1);
olaf@109 294 if (newstring.ptr) {
olaf@109 295 newstring.length = s.length;
olaf@109 296 newstring.ptr[newstring.length] = 0;
olaf@109 297
olaf@109 298 memcpy(newstring.ptr, s.ptr, s.length);
olaf@109 299 } else {
olaf@109 300 newstring.length = 0;
olaf@109 301 }
olaf@109 302
olaf@20 303 return newstring;
olaf@20 304 }
olaf@96 305
olaf@96 306 sstr_t sstrtrim(sstr_t string) {
olaf@96 307 sstr_t newstr = string;
universe@189 308
universe@189 309 while (newstr.length > 0 && isspace(*newstr.ptr)) {
universe@189 310 newstr.ptr++;
universe@189 311 newstr.length--;
universe@98 312 }
universe@189 313 while (newstr.length > 0 && isspace(newstr.ptr[newstr.length-1])) {
universe@189 314 newstr.length--;
olaf@96 315 }
olaf@96 316
olaf@96 317 return newstr;
olaf@96 318 }
universe@146 319
universe@146 320 int sstrprefix(sstr_t string, sstr_t prefix) {
universe@146 321 if (string.length == 0) {
universe@146 322 return prefix.length == 0;
universe@146 323 }
universe@146 324 if (prefix.length == 0) {
universe@146 325 return 1;
universe@146 326 }
universe@146 327
universe@146 328 if (prefix.length > string.length) {
universe@146 329 return 0;
universe@146 330 } else {
universe@146 331 return memcmp(string.ptr, prefix.ptr, prefix.length) == 0;
universe@146 332 }
universe@146 333 }
universe@146 334
universe@146 335 int sstrsuffix(sstr_t string, sstr_t suffix) {
universe@146 336 if (string.length == 0) {
universe@146 337 return suffix.length == 0;
universe@146 338 }
universe@146 339 if (suffix.length == 0) {
universe@146 340 return 1;
universe@146 341 }
universe@146 342
universe@146 343 if (suffix.length > string.length) {
universe@146 344 return 0;
universe@146 345 } else {
universe@146 346 return memcmp(string.ptr+string.length-suffix.length,
universe@146 347 suffix.ptr, suffix.length) == 0;
universe@146 348 }
universe@146 349 }
universe@210 350
universe@210 351 sstr_t sstrlower(sstr_t string) {
universe@210 352 sstr_t ret = sstrdup(string);
universe@210 353 for (size_t i = 0; i < ret.length ; i++) {
universe@210 354 ret.ptr[i] = tolower(ret.ptr[i]);
universe@210 355 }
universe@210 356 return ret;
universe@210 357 }
universe@210 358
universe@210 359 sstr_t sstrlower_a(UcxAllocator *allocator, sstr_t string) {
universe@210 360 sstr_t ret = sstrdup_a(allocator, string);
universe@210 361 for (size_t i = 0; i < ret.length ; i++) {
universe@210 362 ret.ptr[i] = tolower(ret.ptr[i]);
universe@210 363 }
universe@210 364 return ret;
universe@210 365 }
universe@210 366
universe@210 367 sstr_t sstrupper(sstr_t string) {
universe@210 368 sstr_t ret = sstrdup(string);
universe@210 369 for (size_t i = 0; i < ret.length ; i++) {
universe@210 370 ret.ptr[i] = toupper(ret.ptr[i]);
universe@210 371 }
universe@210 372 return ret;
universe@210 373 }
universe@210 374
universe@210 375 sstr_t sstrupper_a(UcxAllocator *allocator, sstr_t string) {
universe@210 376 sstr_t ret = sstrdup_a(allocator, string);
universe@210 377 for (size_t i = 0; i < ret.length ; i++) {
universe@210 378 ret.ptr[i] = toupper(ret.ptr[i]);
universe@210 379 }
universe@210 380 return ret;
universe@210 381 }

mercurial