comparison: src/utils.c
src/utils.c
- changeset 251
- fae240d633fc
- parent 250
- b7d1317b138e
- child 259
- 2f5dea574a75
equal
deleted
inserted
replaced
| 250:b7d1317b138e | 251:fae240d633fc |
|---|---|
| 1 /* | |
| 2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. | |
| 3 * | |
| 4 * Copyright 2017 Olaf Wintermann. All rights reserved. | |
| 5 * | |
| 6 * Redistribution and use in source and binary forms, with or without | |
| 7 * modification, are permitted provided that the following conditions are met: | |
| 8 * | |
| 9 * 1. Redistributions of source code must retain the above copyright | |
| 10 * notice, this list of conditions and the following disclaimer. | |
| 11 * | |
| 12 * 2. Redistributions in binary form must reproduce the above copyright | |
| 13 * notice, this list of conditions and the following disclaimer in the | |
| 14 * documentation and/or other materials provided with the distribution. | |
| 15 * | |
| 16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" | |
| 17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
| 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |
| 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE | |
| 20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR | |
| 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF | |
| 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS | |
| 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN | |
| 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) | |
| 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE | |
| 26 * POSSIBILITY OF SUCH DAMAGE. | |
| 27 */ | |
| 28 | |
| 29 #include "ucx/utils.h" | |
| 30 | |
| 31 #include <math.h> | |
| 32 #include <stdio.h> | |
| 33 #include <limits.h> | |
| 34 #include <errno.h> | |
| 35 | |
| 36 /* COPY FUCNTIONS */ | |
| 37 void* ucx_strcpy(const void* s, void* data) { | |
| 38 const char *str = (const char*) s; | |
| 39 size_t n = 1+strlen(str); | |
| 40 char *cpy = (char*) malloc(n); | |
| 41 memcpy(cpy, str, n); | |
| 42 return cpy; | |
| 43 } | |
| 44 | |
| 45 void* ucx_memcpy(const void* m, void* n) { | |
| 46 size_t k = *((size_t*)n); | |
| 47 void *cpy = malloc(k); | |
| 48 memcpy(cpy, m, k); | |
| 49 return cpy; | |
| 50 } | |
| 51 | |
| 52 size_t ucx_stream_bncopy(void *src, void *dest, read_func readfnc, | |
| 53 write_func writefnc, char* buf, size_t bufsize, size_t n) { | |
| 54 if(n == 0 || bufsize == 0) { | |
| 55 return 0; | |
| 56 } | |
| 57 | |
| 58 char *lbuf; | |
| 59 size_t ncp = 0; | |
| 60 | |
| 61 if(buf) { | |
| 62 lbuf = buf; | |
| 63 } else { | |
| 64 lbuf = (char*)malloc(bufsize); | |
| 65 if(lbuf == NULL) { | |
| 66 return 0; | |
| 67 } | |
| 68 } | |
| 69 | |
| 70 size_t r; | |
| 71 size_t rn = bufsize > n ? n : bufsize; | |
| 72 while((r = readfnc(lbuf, 1, rn, src)) != 0) { | |
| 73 r = writefnc(lbuf, 1, r, dest); | |
| 74 ncp += r; | |
| 75 n -= r; | |
| 76 rn = bufsize > n ? n : bufsize; | |
| 77 if(r == 0 || n == 0) { | |
| 78 break; | |
| 79 } | |
| 80 } | |
| 81 | |
| 82 if (lbuf != buf) { | |
| 83 free(lbuf); | |
| 84 } | |
| 85 | |
| 86 return ncp; | |
| 87 } | |
| 88 | |
| 89 /* COMPARE FUNCTIONS */ | |
| 90 | |
| 91 int ucx_strcmp(const void *s1, const void *s2, void *data) { | |
| 92 return strcmp((const char*)s1, (const char*)s2); | |
| 93 } | |
| 94 | |
| 95 int ucx_strncmp(const void *s1, const void *s2, void *n) { | |
| 96 return strncmp((const char*)s1, (const char*)s2, *((size_t*) n)); | |
| 97 } | |
| 98 | |
| 99 int ucx_intcmp(const void *i1, const void *i2, void *data) { | |
| 100 int a = *((const int*) i1); | |
| 101 int b = *((const int*) i2); | |
| 102 if (a == b) { | |
| 103 return 0; | |
| 104 } else { | |
| 105 return a < b ? -1 : 1; | |
| 106 } | |
| 107 } | |
| 108 | |
| 109 int ucx_floatcmp(const void *f1, const void *f2, void *epsilon) { | |
| 110 float a = *((const float*) f1); | |
| 111 float b = *((const float*) f2); | |
| 112 float e = !epsilon ? 1e-6f : *((float*)epsilon); | |
| 113 if (fabsf(a - b) < e) { | |
| 114 return 0; | |
| 115 } else { | |
| 116 return a < b ? -1 : 1; | |
| 117 } | |
| 118 } | |
| 119 | |
| 120 int ucx_doublecmp(const void *d1, const void *d2, void *epsilon) { | |
| 121 double a = *((const double*) d1); | |
| 122 double b = *((const double*) d2); | |
| 123 double e = !epsilon ? 1e-14 : *((double*)epsilon); | |
| 124 if (fabs(a - b) < e) { | |
| 125 return 0; | |
| 126 } else { | |
| 127 return a < b ? -1 : 1; | |
| 128 } | |
| 129 } | |
| 130 | |
| 131 int ucx_ptrcmp(const void *ptr1, const void *ptr2, void *data) { | |
| 132 const intptr_t p1 = (const intptr_t) ptr1; | |
| 133 const intptr_t p2 = (const intptr_t) ptr2; | |
| 134 if (p1 == p2) { | |
| 135 return 0; | |
| 136 } else { | |
| 137 return p1 < p2 ? -1 : 1; | |
| 138 } | |
| 139 } | |
| 140 | |
| 141 int ucx_memcmp(const void *ptr1, const void *ptr2, void *n) { | |
| 142 return memcmp(ptr1, ptr2, *((size_t*)n)); | |
| 143 } | |
| 144 | |
| 145 /* PRINTF FUNCTIONS */ | |
| 146 | |
| 147 #ifdef va_copy | |
| 148 #define UCX_PRINTF_BUFSIZE 256 | |
| 149 #else | |
| 150 #pragma message("WARNING: C99 va_copy macro not supported by this platform" \ | |
| 151 " - limiting ucx_*printf to 2 KiB") | |
| 152 #define UCX_PRINTF_BUFSIZE 0x800 | |
| 153 #endif | |
| 154 | |
| 155 int ucx_fprintf(void *stream, write_func wfc, const char *fmt, ...) { | |
| 156 int ret; | |
| 157 va_list ap; | |
| 158 va_start(ap, fmt); | |
| 159 ret = ucx_vfprintf(stream, wfc, fmt, ap); | |
| 160 va_end(ap); | |
| 161 return ret; | |
| 162 } | |
| 163 | |
| 164 int ucx_vfprintf(void *stream, write_func wfc, const char *fmt, va_list ap) { | |
| 165 char buf[UCX_PRINTF_BUFSIZE]; | |
| 166 #ifdef va_copy | |
| 167 va_list ap2; | |
| 168 va_copy(ap2, ap); | |
| 169 int ret = vsnprintf(buf, UCX_PRINTF_BUFSIZE, fmt, ap); | |
| 170 if (ret < 0) { | |
| 171 return ret; | |
| 172 } else if (ret < UCX_PRINTF_BUFSIZE) { | |
| 173 return (int)wfc(buf, 1, ret, stream); | |
| 174 } else { | |
| 175 if (ret == INT_MAX) { | |
| 176 errno = ENOMEM; | |
| 177 return -1; | |
| 178 } | |
| 179 | |
| 180 int len = ret + 1; | |
| 181 char *newbuf = (char*)malloc(len); | |
| 182 if (!newbuf) { | |
| 183 return -1; | |
| 184 } | |
| 185 | |
| 186 ret = vsnprintf(newbuf, len, fmt, ap2); | |
| 187 if (ret > 0) { | |
| 188 ret = (int)wfc(newbuf, 1, ret, stream); | |
| 189 } | |
| 190 free(newbuf); | |
| 191 } | |
| 192 return ret; | |
| 193 #else | |
| 194 int ret = vsnprintf(buf, UCX_PRINTF_BUFSIZE, fmt, ap); | |
| 195 if (ret < 0) { | |
| 196 return ret; | |
| 197 } else if (ret < UCX_PRINTF_BUFSIZE) { | |
| 198 return (int)wfc(buf, 1, ret, stream); | |
| 199 } else { | |
| 200 errno = ENOMEM; | |
| 201 return -1; | |
| 202 } | |
| 203 #endif | |
| 204 } | |
| 205 | |
| 206 sstr_t ucx_asprintf(UcxAllocator *allocator, const char *fmt, ...) { | |
| 207 va_list ap; | |
| 208 sstr_t ret; | |
| 209 va_start(ap, fmt); | |
| 210 ret = ucx_vasprintf(allocator, fmt, ap); | |
| 211 va_end(ap); | |
| 212 return ret; | |
| 213 } | |
| 214 | |
| 215 sstr_t ucx_vasprintf(UcxAllocator *a, const char *fmt, va_list ap) { | |
| 216 sstr_t s; | |
| 217 s.ptr = NULL; | |
| 218 s.length = 0; | |
| 219 char buf[UCX_PRINTF_BUFSIZE]; | |
| 220 #ifdef va_copy | |
| 221 va_list ap2; | |
| 222 va_copy(ap2, ap); | |
| 223 int ret = vsnprintf(buf, UCX_PRINTF_BUFSIZE, fmt, ap); | |
| 224 if (ret > 0 && ret < UCX_PRINTF_BUFSIZE) { | |
| 225 s.ptr = (char*)almalloc(a, ret + 1); | |
| 226 if (s.ptr) { | |
| 227 s.length = (size_t)ret; | |
| 228 memcpy(s.ptr, buf, ret); | |
| 229 s.ptr[s.length] = '\0'; | |
| 230 } | |
| 231 } else if (ret == INT_MAX) { | |
| 232 errno = ENOMEM; | |
| 233 } else { | |
| 234 int len = ret + 1; | |
| 235 s.ptr = (char*)almalloc(a, len); | |
| 236 if (s.ptr) { | |
| 237 ret = vsnprintf(s.ptr, len, fmt, ap2); | |
| 238 if (ret < 0) { | |
| 239 free(s.ptr); | |
| 240 s.ptr = NULL; | |
| 241 } else { | |
| 242 s.length = (size_t)ret; | |
| 243 } | |
| 244 } | |
| 245 } | |
| 246 #else | |
| 247 int ret = vsnprintf(buf, UCX_PRINTF_BUFSIZE, fmt, ap); | |
| 248 if (ret > 0 && ret < UCX_PRINTF_BUFSIZE) { | |
| 249 s.ptr = (char*)almalloc(a, ret + 1); | |
| 250 if (s.ptr) { | |
| 251 s.length = (size_t)ret; | |
| 252 memcpy(s.ptr, buf, ret); | |
| 253 s.ptr[s.length] = '\0'; | |
| 254 } | |
| 255 } else { | |
| 256 errno = ENOMEM; | |
| 257 } | |
| 258 #endif | |
| 259 return s; | |
| 260 } |
