146 lines
3.9 KiB
C
146 lines
3.9 KiB
C
#define JEMALLOC_BUF_WRITER_C_
|
|
#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);
|
|
}
|