diff options
Diffstat (limited to 'examples/redis-unstable/deps/jemalloc/src/buf_writer.c')
| -rw-r--r-- | examples/redis-unstable/deps/jemalloc/src/buf_writer.c | 144 |
1 files changed, 0 insertions, 144 deletions
diff --git a/examples/redis-unstable/deps/jemalloc/src/buf_writer.c b/examples/redis-unstable/deps/jemalloc/src/buf_writer.c deleted file mode 100644 index 7c6f794..0000000 --- a/examples/redis-unstable/deps/jemalloc/src/buf_writer.c +++ /dev/null @@ -1,144 +0,0 @@ -#include "jemalloc/internal/jemalloc_preamble.h" -#include "jemalloc/internal/jemalloc_internal_includes.h" - -#include "jemalloc/internal/buf_writer.h" -#include "jemalloc/internal/malloc_io.h" - -static void * -buf_writer_allocate_internal_buf(tsdn_t *tsdn, size_t buf_len) { -#ifdef JEMALLOC_JET - if (buf_len > SC_LARGE_MAXCLASS) { - return NULL; - } -#else - assert(buf_len <= SC_LARGE_MAXCLASS); -#endif - return iallocztm(tsdn, buf_len, sz_size2index(buf_len), false, NULL, - true, arena_get(tsdn, 0, false), true); -} - -static void -buf_writer_free_internal_buf(tsdn_t *tsdn, void *buf) { - if (buf != NULL) { - idalloctm(tsdn, buf, NULL, NULL, true, true); - } -} - -static void -buf_writer_assert(buf_writer_t *buf_writer) { - assert(buf_writer != NULL); - assert(buf_writer->write_cb != NULL); - if (buf_writer->buf != NULL) { - assert(buf_writer->buf_size > 0); - } else { - assert(buf_writer->buf_size == 0); - assert(buf_writer->internal_buf); - } - assert(buf_writer->buf_end <= buf_writer->buf_size); -} - -bool -buf_writer_init(tsdn_t *tsdn, buf_writer_t *buf_writer, write_cb_t *write_cb, - void *cbopaque, char *buf, size_t buf_len) { - if (write_cb != NULL) { - buf_writer->write_cb = write_cb; - } else { - buf_writer->write_cb = je_malloc_message != NULL ? - je_malloc_message : wrtmessage; - } - buf_writer->cbopaque = cbopaque; - assert(buf_len >= 2); - if (buf != NULL) { - buf_writer->buf = buf; - buf_writer->internal_buf = false; - } else { - buf_writer->buf = buf_writer_allocate_internal_buf(tsdn, - buf_len); - buf_writer->internal_buf = true; - } - if (buf_writer->buf != NULL) { - buf_writer->buf_size = buf_len - 1; /* Allowing for '\0'. */ - } else { - buf_writer->buf_size = 0; - } - buf_writer->buf_end = 0; - buf_writer_assert(buf_writer); - return buf_writer->buf == NULL; -} - -void -buf_writer_flush(buf_writer_t *buf_writer) { - buf_writer_assert(buf_writer); - if (buf_writer->buf == NULL) { - return; - } - buf_writer->buf[buf_writer->buf_end] = '\0'; - buf_writer->write_cb(buf_writer->cbopaque, buf_writer->buf); - buf_writer->buf_end = 0; - buf_writer_assert(buf_writer); -} - -void -buf_writer_cb(void *buf_writer_arg, const char *s) { - buf_writer_t *buf_writer = (buf_writer_t *)buf_writer_arg; - buf_writer_assert(buf_writer); - if (buf_writer->buf == NULL) { - buf_writer->write_cb(buf_writer->cbopaque, s); - return; - } - size_t i, slen, n; - for (i = 0, slen = strlen(s); i < slen; i += n) { - if (buf_writer->buf_end == buf_writer->buf_size) { - buf_writer_flush(buf_writer); - } - size_t s_remain = slen - i; - size_t buf_remain = buf_writer->buf_size - buf_writer->buf_end; - n = s_remain < buf_remain ? s_remain : buf_remain; - memcpy(buf_writer->buf + buf_writer->buf_end, s + i, n); - buf_writer->buf_end += n; - buf_writer_assert(buf_writer); - } - assert(i == slen); -} - -void -buf_writer_terminate(tsdn_t *tsdn, buf_writer_t *buf_writer) { - buf_writer_assert(buf_writer); - buf_writer_flush(buf_writer); - if (buf_writer->internal_buf) { - buf_writer_free_internal_buf(tsdn, buf_writer->buf); - } -} - -void -buf_writer_pipe(buf_writer_t *buf_writer, read_cb_t *read_cb, - void *read_cbopaque) { - /* - * A tiny local buffer in case the buffered writer failed to allocate - * at init. - */ - static char backup_buf[16]; - static buf_writer_t backup_buf_writer; - - buf_writer_assert(buf_writer); - assert(read_cb != NULL); - if (buf_writer->buf == NULL) { - buf_writer_init(TSDN_NULL, &backup_buf_writer, - buf_writer->write_cb, buf_writer->cbopaque, backup_buf, - sizeof(backup_buf)); - buf_writer = &backup_buf_writer; - } - assert(buf_writer->buf != NULL); - ssize_t nread = 0; - do { - buf_writer->buf_end += nread; - buf_writer_assert(buf_writer); - if (buf_writer->buf_end == buf_writer->buf_size) { - buf_writer_flush(buf_writer); - } - nread = read_cb(read_cbopaque, - buf_writer->buf + buf_writer->buf_end, - buf_writer->buf_size - buf_writer->buf_end); - } while (nread > 0); - buf_writer_flush(buf_writer); -} |
