Fri, 30 Nov 2012 13:10:58 +0100
fixed buffer
universe@60 | 1 | #include "buffer.h" |
universe@56 | 2 | #include <stdarg.h> |
universe@56 | 3 | #include <stdlib.h> |
universe@56 | 4 | #include <string.h> |
universe@56 | 5 | |
olaf@76 | 6 | UcxBuffer *ucx_buffer_new(void *space, size_t size, int flags) { |
universe@60 | 7 | UcxBuffer *buffer = (UcxBuffer*) malloc(sizeof(UcxBuffer)); |
universe@60 | 8 | if (buffer) { |
universe@61 | 9 | buffer->flags = flags; |
universe@56 | 10 | if (!space) { |
olaf@76 | 11 | buffer->space = malloc(size); |
universe@60 | 12 | if (!buffer->space) { |
universe@60 | 13 | free(buffer); |
universe@56 | 14 | return NULL; |
universe@56 | 15 | } |
olaf@76 | 16 | memset(buffer->space, 0, size); |
universe@61 | 17 | buffer->flags |= UCX_BUFFER_AUTOFREE; |
universe@56 | 18 | } else { |
universe@60 | 19 | buffer->space = space; |
universe@56 | 20 | } |
olaf@76 | 21 | buffer->capacity = size; |
olaf@76 | 22 | buffer->size = 0; |
universe@56 | 23 | |
universe@60 | 24 | buffer->pos = 0; |
universe@56 | 25 | } |
universe@56 | 26 | |
universe@60 | 27 | return buffer; |
universe@56 | 28 | } |
universe@56 | 29 | |
universe@60 | 30 | void ucx_buffer_free(UcxBuffer *buffer) { |
universe@63 | 31 | if ((buffer->flags & UCX_BUFFER_AUTOFREE) == UCX_BUFFER_AUTOFREE) { |
universe@60 | 32 | free(buffer->space); |
universe@56 | 33 | } |
universe@60 | 34 | free(buffer); |
universe@56 | 35 | } |
universe@56 | 36 | |
olaf@76 | 37 | UcxBuffer* ucx_buffer_extract( |
olaf@76 | 38 | UcxBuffer *src, size_t start, size_t length, int flags) { |
olaf@76 | 39 | if(src->size == 0) { |
olaf@76 | 40 | return NULL; |
olaf@76 | 41 | } |
universe@62 | 42 | if (length == 0) { |
universe@62 | 43 | length = src->size - start; |
universe@62 | 44 | } |
universe@62 | 45 | if (start+length > src->size) { |
universe@62 | 46 | return NULL; |
universe@62 | 47 | } |
universe@62 | 48 | |
olaf@76 | 49 | UcxBuffer *dst = (UcxBuffer*) malloc(sizeof(UcxBuffer)); |
universe@62 | 50 | if (dst) { |
universe@62 | 51 | dst->space = malloc(length); |
universe@62 | 52 | if (!dst->space) { |
universe@62 | 53 | free(dst); |
universe@62 | 54 | return NULL; |
universe@62 | 55 | } |
olaf@76 | 56 | dst->capacity = length; |
universe@62 | 57 | dst->size = length; |
universe@62 | 58 | dst->flags = flags | UCX_BUFFER_AUTOFREE; |
universe@62 | 59 | dst->pos = 0; |
olaf@76 | 60 | memcpy(dst->space, src->space+start, length); |
universe@62 | 61 | } |
universe@62 | 62 | return dst; |
universe@62 | 63 | } |
universe@62 | 64 | |
universe@60 | 65 | int ucx_buffer_seek(UcxBuffer *buffer, off_t offset, int whence) { |
olaf@57 | 66 | off_t npos; |
universe@56 | 67 | switch (whence) { |
universe@56 | 68 | case SEEK_SET: |
universe@56 | 69 | npos = 0; |
universe@56 | 70 | break; |
universe@56 | 71 | case SEEK_CUR: |
universe@60 | 72 | npos = buffer->pos; |
universe@56 | 73 | break; |
universe@56 | 74 | case SEEK_END: |
olaf@76 | 75 | npos = buffer->size; |
universe@56 | 76 | break; |
universe@56 | 77 | } |
universe@56 | 78 | |
universe@56 | 79 | npos += offset; |
olaf@76 | 80 | |
universe@62 | 81 | if (npos < 0 || npos > buffer->size) { |
universe@56 | 82 | return -1; |
universe@56 | 83 | } else { |
universe@60 | 84 | buffer->pos = npos; |
universe@56 | 85 | return 0; |
universe@56 | 86 | } |
universe@56 | 87 | |
universe@56 | 88 | } |
universe@56 | 89 | |
universe@60 | 90 | int ucx_buffer_eof(UcxBuffer *buffer) { |
universe@62 | 91 | return buffer->pos >= buffer->size; |
universe@56 | 92 | } |
universe@56 | 93 | |
olaf@76 | 94 | int ucx_buffer_extend(UcxBuffer *buffer, size_t len) { |
olaf@76 | 95 | size_t newcap = buffer->capacity; |
olaf@76 | 96 | while (buffer->pos + len > newcap) newcap <<= 1; |
olaf@76 | 97 | |
olaf@76 | 98 | char *newspace = realloc(buffer->space, newcap); |
olaf@76 | 99 | if (newspace) { |
olaf@76 | 100 | memset(newspace+buffer->size, 0, newcap-buffer->size); |
olaf@76 | 101 | buffer->space = newspace; |
olaf@76 | 102 | buffer->capacity = newcap; |
olaf@76 | 103 | } else { |
olaf@76 | 104 | return -1; |
olaf@76 | 105 | } |
olaf@76 | 106 | |
olaf@76 | 107 | return 0; |
olaf@76 | 108 | } |
olaf@76 | 109 | |
olaf@76 | 110 | size_t ucx_buffer_write(const void *ptr, size_t size, size_t nitems, |
olaf@76 | 111 | UcxBuffer *buffer) { |
olaf@76 | 112 | size_t len = size * nitems; |
olaf@76 | 113 | if (buffer->pos + len > buffer->capacity) { |
olaf@76 | 114 | if ((buffer->flags & UCX_BUFFER_AUTOEXTEND) == UCX_BUFFER_AUTOEXTEND) { |
olaf@76 | 115 | if(ucx_buffer_extend(buffer, len)) { |
olaf@76 | 116 | return -1; |
universe@64 | 117 | } |
universe@64 | 118 | } else { |
olaf@76 | 119 | len = buffer->capacity - buffer->pos; |
olaf@76 | 120 | if (size > 1) len -= len%size; |
universe@64 | 121 | } |
universe@56 | 122 | } |
olaf@76 | 123 | |
universe@64 | 124 | if (len <= 0) { |
universe@64 | 125 | return len; |
universe@56 | 126 | } |
olaf@76 | 127 | |
olaf@76 | 128 | memcpy(buffer->space + buffer->pos, ptr, len); |
olaf@76 | 129 | buffer->pos += len; |
olaf@76 | 130 | if(buffer->pos > buffer->size) { |
olaf@76 | 131 | buffer->size = buffer->pos; |
olaf@76 | 132 | } |
olaf@76 | 133 | |
olaf@76 | 134 | return len / size; |
olaf@76 | 135 | } |
universe@56 | 136 | |
olaf@76 | 137 | size_t ucx_buffer_read(void *ptr, size_t size, size_t nitems, |
olaf@76 | 138 | UcxBuffer *buffer) { |
olaf@76 | 139 | size_t len = size * nitems; |
olaf@76 | 140 | if (buffer->pos + len > buffer->size) { |
olaf@76 | 141 | len = buffer->size - buffer->pos; |
olaf@76 | 142 | if (size > 1) len -= len%size; |
universe@56 | 143 | } |
olaf@76 | 144 | |
olaf@76 | 145 | if (len <= 0) { |
olaf@76 | 146 | return len; |
olaf@76 | 147 | } |
olaf@76 | 148 | |
olaf@76 | 149 | memcpy(ptr, buffer->space + buffer->pos, len); |
olaf@76 | 150 | buffer->pos += len; |
olaf@76 | 151 | |
olaf@76 | 152 | return len / size; |
universe@56 | 153 | } |
universe@56 | 154 | |
universe@60 | 155 | int ucx_buffer_putc(UcxBuffer *buffer, int c) { |
olaf@76 | 156 | if(buffer->pos >= buffer->capacity) { |
olaf@76 | 157 | if ((buffer->flags & UCX_BUFFER_AUTOEXTEND) == UCX_BUFFER_AUTOEXTEND) { |
olaf@76 | 158 | if(ucx_buffer_extend(buffer, 1)) { |
olaf@76 | 159 | return EOF; |
olaf@76 | 160 | } |
olaf@76 | 161 | } else { |
olaf@76 | 162 | return EOF; |
olaf@76 | 163 | } |
universe@56 | 164 | } |
olaf@76 | 165 | |
olaf@76 | 166 | c &= 0xFF; |
olaf@76 | 167 | buffer->space[buffer->pos] = (char) c; |
olaf@76 | 168 | buffer->pos++; |
olaf@76 | 169 | if(buffer->pos > buffer->size) { |
olaf@76 | 170 | buffer->size = buffer->pos; |
olaf@76 | 171 | } |
olaf@76 | 172 | return c; |
universe@56 | 173 | } |
universe@56 | 174 | |
universe@60 | 175 | int ucx_buffer_getc(UcxBuffer *buffer) { |
universe@60 | 176 | if (ucx_buffer_eof(buffer)) { |
universe@56 | 177 | return EOF; |
universe@56 | 178 | } else { |
olaf@76 | 179 | int c = buffer->space[buffer->pos]; |
universe@60 | 180 | buffer->pos++; |
universe@56 | 181 | return c; |
universe@56 | 182 | } |
universe@56 | 183 | } |
olaf@76 | 184 | |
olaf@76 | 185 | size_t ucx_buffer_generic_copy(void *s1, void *s2, |
olaf@76 | 186 | read_func readfnc, write_func writefnc, size_t bufsize) { |
olaf@76 | 187 | size_t ncp = 0; |
olaf@76 | 188 | char *buf = malloc(bufsize); |
olaf@76 | 189 | if(buf == NULL) { |
olaf@76 | 190 | return 0; |
olaf@76 | 191 | } |
olaf@76 | 192 | |
olaf@76 | 193 | size_t r; |
olaf@76 | 194 | while((r = readfnc(buf, 1, bufsize, s1)) != 0) { |
olaf@76 | 195 | r = writefnc(buf, 1, r, s2); |
olaf@76 | 196 | ncp += r; |
olaf@76 | 197 | if(r == 0) { |
olaf@76 | 198 | break; |
olaf@76 | 199 | } |
olaf@76 | 200 | } |
olaf@76 | 201 | |
olaf@76 | 202 | free(buf); |
olaf@76 | 203 | return ncp; |
olaf@76 | 204 | } |