Artifact 316847ca638c005f235a65dda3932c08f3cdd699cd3aca981212ca4a968ce23e:
- File
src/stream.c
— part of check-in
[6893b9f5ab]
at
2012-09-30 01:34:46
on branch trunk
— Rename eof to at_end.
This is more fitting as end of file is not really true for a socket. (user: js size: 6676)
/* * Copyright (c) 2012, Jonathan Schleifer <js@webkeks.org> * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE * POSSIBILITY OF SUCH DAMAGE. */ #include <stdlib.h> #include <string.h> #include "stream.h" #define BUFFER_SIZE 4096 static bool ctor(void *ptr, va_list args) { CFWStream *stream = ptr; stream->ops = NULL; stream->cache = NULL; stream->cache_len = 0; return true; } static void dtor(void *ptr) { cfw_stream_close(ptr); } ssize_t cfw_stream_read(void *ptr, void *buf, size_t len) { CFWStream *stream = ptr; ssize_t ret; if (stream == NULL || stream->ops == NULL) return -1; if (stream->cache == NULL) { if ((ret = stream->ops->read(stream, buf, len)) < -1) ret = -1; return ret; } if (len >= stream->cache_len) { ret = stream->cache_len; memcpy(buf, stream->cache, stream->cache_len); free(stream->cache); stream->cache = NULL; stream->cache_len = 0; return ret; } else { char *tmp; if ((tmp = malloc(stream->cache_len - len)) == NULL) return -1; memcpy(tmp, stream->cache + len, stream->cache_len - len); memcpy(buf, stream->cache, len); free(stream->cache); stream->cache = tmp; stream->cache_len -= len; return len; } } CFWString* cfw_stream_read_line(void *ptr) { CFWStream *stream = ptr; CFWString *ret; char *buf, *ret_str, *new_cache; ssize_t buf_len; size_t i, ret_len; /* Look if there is a line or \0 in our cache */ if (stream->cache != NULL) { for (i = 0; i < stream->cache_len; i++) { if (stream->cache[i] == '\n' || stream->cache[i] == '\0') { ret_len = i; if (i > 0 && stream->cache[i - 1] == '\r') ret_len--; ret_str = cfw_strndup(stream->cache, ret_len); if (ret_str == NULL) return NULL; ret = cfw_create(cfw_string, NULL); if (ret == NULL) { free(ret_str); return NULL; } cfw_string_set_nocopy(ret, ret_str, ret_len); new_cache = malloc(stream->cache_len - i - 1); if (new_cache == NULL) return NULL; memcpy(new_cache, stream->cache + i + 1, stream->cache_len - i - 1); free(stream->cache); stream->cache = new_cache; stream->cache_len -= i + 1; return ret; } } } /* Read and see if we get a newline or \0 */ if ((buf = malloc(BUFFER_SIZE)) == NULL) return NULL; for (;;) { if (stream->ops->at_end(stream)) { free(buf); if (stream->cache == NULL) return NULL; ret_len = stream->cache_len; if (ret_len > 0 && stream->cache[ret_len - 1] == '\r') ret_len--; ret_str = cfw_strndup(stream->cache, ret_len); if (ret_str == NULL) return NULL; ret = cfw_create(cfw_string, NULL); if (ret == NULL) { free(ret_str); return NULL; } cfw_string_set_nocopy(ret, ret_str, ret_len); free(stream->cache); stream->cache = NULL; stream->cache_len = 0; return ret; } buf_len = stream->ops->read(stream, buf, BUFFER_SIZE); if (buf_len == -1) { free(buf); return NULL; } /* Look if there's a newline or \0 */ for (i = 0; i < buf_len; i++) { if (buf[i] == '\n' || buf[i] == '\0') { ret_len = stream->cache_len + i; if ((ret_str = malloc(ret_len + 1)) == NULL) { /* * FIXME: We lost the current buffer. * Mark the stream as broken? */ free(buf); return NULL; } memcpy(ret_str, stream->cache, stream->cache_len); memcpy(ret_str + stream->cache_len, buf, i); if (ret_len > 0 && ret_str[ret_len - 1] == '\r') ret_len--; ret_str[ret_len] = '\0'; ret = cfw_create(cfw_string, NULL); if (ret == NULL) { free(buf); free(ret_str); return NULL; } cfw_string_set_nocopy(ret, ret_str, ret_len); new_cache = malloc(buf_len - i - 1); if (new_cache == NULL) { free(buf); return NULL; } memcpy(new_cache, buf + i + 1, buf_len - i - 1); free(stream->cache); stream->cache = new_cache; stream->cache_len = buf_len - i - 1; free(buf); return ret; } } /* There was no newline or \0 */ new_cache = realloc(stream->cache, stream->cache_len + buf_len); if (new_cache == NULL) { free(buf); return NULL; } memcpy(new_cache + stream->cache_len, buf, buf_len); stream->cache = new_cache; stream->cache_len += buf_len; } } bool cfw_stream_write(void *ptr, const void *buf, size_t len) { CFWStream *stream = ptr; if (stream == NULL || stream->ops == NULL) return false; return stream->ops->write(stream, buf, len); } bool cfw_stream_write_string(void *ptr, const char *str) { return cfw_stream_write(ptr, str, strlen(str)); } bool cfw_stream_write_line(void *ptr, const char *str) { char *tmp; size_t len; len = strlen(str); if ((tmp = malloc(len + 2)) == NULL) return false; memcpy(tmp, str, len); tmp[len] = '\n'; tmp[len + 1] = '\0'; if (!cfw_stream_write(ptr, tmp, len + 1)) { free(tmp); return false; } free(tmp); return true; } bool cfw_stream_at_end(void *ptr) { CFWStream *stream = ptr; if (stream == NULL || stream->ops == NULL) return true; if (stream->cache != NULL) return false; return stream->ops->at_end(stream); } void cfw_stream_close(void *ptr) { CFWStream *stream = ptr; if (stream == NULL || stream->ops == NULL) return; stream->ops->close(stream); } static CFWClass class = { .name = "CFWStream", .size = sizeof(CFWStream), .ctor = ctor, .dtor = dtor }; CFWClass *cfw_stream = &class;