60 lines
1.4 KiB
C
60 lines
1.4 KiB
C
|
|
#include <lighttpd/buffer.h>
|
|
|
|
static void _buffer_init(liBuffer *buf, gsize alloc_size) {
|
|
buf->alloc_size = alloc_size;
|
|
buf->used = 0;
|
|
buf->mptr = li_mempool_alloc(alloc_size);
|
|
buf->addr = buf->mptr.data;
|
|
}
|
|
|
|
static void _buffer_init_slice(liBuffer *buf, gsize alloc_size) {
|
|
buf->alloc_size = alloc_size;
|
|
buf->used = 0;
|
|
buf->mptr.data = NULL;
|
|
buf->addr = g_slice_alloc(alloc_size);
|
|
}
|
|
|
|
static void _buffer_destroy(liBuffer *buf) {
|
|
if (!buf || NULL == buf->addr) return;
|
|
|
|
if (NULL == buf->mptr.data) {
|
|
g_slice_free1(buf->alloc_size, buf->addr);
|
|
} else {
|
|
li_mempool_free(buf->mptr, buf->alloc_size);
|
|
buf->addr = NULL;
|
|
buf->mptr.data = NULL; buf->mptr.priv_data = NULL;
|
|
buf->used = buf->alloc_size = 0;
|
|
}
|
|
|
|
g_slice_free(liBuffer, buf);
|
|
}
|
|
|
|
|
|
liBuffer* li_buffer_new(gsize max_size) {
|
|
liBuffer *buf = g_slice_new0(liBuffer);
|
|
_buffer_init(buf, li_mempool_align_page_size(max_size));
|
|
buf->refcount = 1;
|
|
return buf;
|
|
}
|
|
|
|
liBuffer* li_buffer_new_slice(gsize max_size) {
|
|
liBuffer *buf = g_slice_new0(liBuffer);
|
|
_buffer_init_slice(buf, max_size);
|
|
buf->refcount = 1;
|
|
return buf;
|
|
}
|
|
|
|
void li_buffer_release(liBuffer *buf) {
|
|
if (!buf) return;
|
|
assert(g_atomic_int_get(&buf->refcount) > 0);
|
|
if (g_atomic_int_dec_and_test(&buf->refcount)) {
|
|
_buffer_destroy(buf);
|
|
}
|
|
}
|
|
|
|
void li_buffer_acquire(liBuffer *buf) {
|
|
assert(g_atomic_int_get(&buf->refcount) > 0);
|
|
g_atomic_int_inc(&buf->refcount);
|
|
}
|