comparison: ucx/utils.c
ucx/utils.c
- changeset 17
- 11dffb40cd91
- parent 5
- 88625853ae74
- child 39
- 3e55bed345f9
equal
deleted
inserted
replaced
| 16:5dbef9e07376 | 17:11dffb40cd91 |
|---|---|
| 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE | 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
| 26 * POSSIBILITY OF SUCH DAMAGE. | 26 * POSSIBILITY OF SUCH DAMAGE. |
| 27 */ | 27 */ |
| 28 | 28 |
| 29 #include "utils.h" | 29 #include "utils.h" |
| 30 #include "math.h" | 30 #include <math.h> |
| 31 #include <stdio.h> | |
| 32 #include <limits.h> | |
| 33 #include <errno.h> | |
| 31 | 34 |
| 32 /* COPY FUCNTIONS */ | 35 /* COPY FUCNTIONS */ |
| 33 void* ucx_strcpy(void* s, void* data) { | 36 void* ucx_strcpy(void* s, void* data) { |
| 34 char *str = (char*) s; | 37 char *str = (char*) s; |
| 35 size_t n = 1+strlen(str); | 38 size_t n = 1+strlen(str); |
| 43 void *cpy = malloc(k); | 46 void *cpy = malloc(k); |
| 44 memcpy(cpy, m, k); | 47 memcpy(cpy, m, k); |
| 45 return cpy; | 48 return cpy; |
| 46 } | 49 } |
| 47 | 50 |
| 48 /* COMPARE FUNCTION */ | 51 size_t ucx_stream_copy(void *src, void *dest, read_func readfnc, |
| 52 write_func writefnc, char* buf, size_t bufsize, size_t n) { | |
| 53 if(n == 0 || bufsize == 0) { | |
| 54 return 0; | |
| 55 } | |
| 56 | |
| 57 size_t ncp = 0; | |
| 58 if (!buf) { | |
| 59 buf = (char*)malloc(bufsize); | |
| 60 if(buf == NULL) { | |
| 61 return 0; | |
| 62 } | |
| 63 } | |
| 64 | |
| 65 size_t r; | |
| 66 size_t rn = bufsize > n ? n : bufsize; | |
| 67 while((r = readfnc(buf, 1, rn, src)) != 0) { | |
| 68 r = writefnc(buf, 1, r, dest); | |
| 69 ncp += r; | |
| 70 n -= r; | |
| 71 rn = bufsize > n ? n : bufsize; | |
| 72 if(r == 0 || n == 0) { | |
| 73 break; | |
| 74 } | |
| 75 } | |
| 76 | |
| 77 free(buf); | |
| 78 return ncp; | |
| 79 } | |
| 80 | |
| 81 /* COMPARE FUNCTIONS */ | |
| 49 | 82 |
| 50 int ucx_strcmp(void *s1, void *s2, void *data) { | 83 int ucx_strcmp(void *s1, void *s2, void *data) { |
| 51 return strcmp((char*)s1, (char*)s2); | 84 return strcmp((char*)s1, (char*)s2); |
| 52 } | 85 } |
| 53 | 86 |
| 96 } | 129 } |
| 97 | 130 |
| 98 int ucx_memcmp(void *ptr1, void *ptr2, void *n) { | 131 int ucx_memcmp(void *ptr1, void *ptr2, void *n) { |
| 99 return memcmp(ptr1, ptr2, *((size_t*)n)); | 132 return memcmp(ptr1, ptr2, *((size_t*)n)); |
| 100 } | 133 } |
| 134 | |
| 135 /* PRINTF FUNCTIONS */ | |
| 136 | |
| 137 #define UCX_PRINTF_BUFSIZE 256 | |
| 138 | |
| 139 int ucx_fprintf(void *stream, write_func wfc, const char *fmt, ...) { | |
| 140 va_list ap; | |
| 141 int ret; | |
| 142 va_start(ap, fmt); | |
| 143 ret = ucx_vfprintf(stream, wfc, fmt, ap); | |
| 144 va_end(ap); | |
| 145 return ret; | |
| 146 } | |
| 147 | |
| 148 int ucx_vfprintf(void *stream, write_func wfc, const char *fmt, va_list ap) { | |
| 149 char buf[UCX_PRINTF_BUFSIZE]; | |
| 150 va_list ap2; | |
| 151 va_copy(ap2, ap); | |
| 152 int ret = vsnprintf(buf, UCX_PRINTF_BUFSIZE, fmt, ap); | |
| 153 if (ret < 0) { | |
| 154 return ret; | |
| 155 } else if (ret < UCX_PRINTF_BUFSIZE) { | |
| 156 return (int)wfc(buf, 1, ret, stream); | |
| 157 } else { | |
| 158 if (ret == INT_MAX) { | |
| 159 errno = ENOMEM; | |
| 160 return -1; | |
| 161 } | |
| 162 | |
| 163 int len = ret + 1; | |
| 164 char *newbuf = (char*)malloc(len); | |
| 165 if (!newbuf) { | |
| 166 return -1; | |
| 167 } | |
| 168 | |
| 169 ret = vsnprintf(newbuf, len, fmt, ap2); | |
| 170 va_end(ap2); | |
| 171 if (ret > 0) { | |
| 172 ret = (int)wfc(newbuf, 1, ret, stream); | |
| 173 } | |
| 174 free(newbuf); | |
| 175 } | |
| 176 return ret; | |
| 177 } | |
| 178 | |
| 179 sstr_t ucx_asprintf(UcxAllocator *allocator, const char *fmt, ...) { | |
| 180 va_list ap; | |
| 181 sstr_t ret; | |
| 182 va_start(ap, fmt); | |
| 183 ret = ucx_vasprintf(allocator, fmt, ap); | |
| 184 va_end(ap); | |
| 185 return ret; | |
| 186 } | |
| 187 | |
| 188 sstr_t ucx_vasprintf(UcxAllocator *a, const char *fmt, va_list ap) { | |
| 189 sstr_t s; | |
| 190 s.ptr = NULL; | |
| 191 s.length = 0; | |
| 192 va_list ap2; | |
| 193 va_copy(ap2, ap); | |
| 194 char buf[UCX_PRINTF_BUFSIZE]; | |
| 195 int ret = vsnprintf(buf, UCX_PRINTF_BUFSIZE, fmt, ap); | |
| 196 if (ret > 0 && ret < UCX_PRINTF_BUFSIZE) { | |
| 197 s.ptr = (char*)a->malloc(a->pool, ret + 1); | |
| 198 s.length = (size_t)ret; | |
| 199 memcpy(s.ptr, buf, ret); | |
| 200 s.ptr[s.length] = '\0'; | |
| 201 } else if (ret == INT_MAX) { | |
| 202 errno = ENOMEM; | |
| 203 } else { | |
| 204 int len = ret + 1; | |
| 205 s.ptr = (char*)a->malloc(a->pool, len); | |
| 206 ret = vsnprintf(s.ptr, len, fmt, ap2); | |
| 207 va_end(ap2); | |
| 208 if (ret < 0) { | |
| 209 free(s.ptr); | |
| 210 s.ptr = NULL; | |
| 211 } else { | |
| 212 s.length = (size_t)ret; | |
| 213 } | |
| 214 } | |
| 215 return s; | |
| 216 } |
