diff options
Diffstat (limited to 'src/dynbuf.c')
| -rw-r--r-- | src/dynbuf.c | 129 |
1 files changed, 129 insertions, 0 deletions
diff --git a/src/dynbuf.c b/src/dynbuf.c new file mode 100644 index 0000000..712e334 --- /dev/null +++ b/src/dynbuf.c @@ -0,0 +1,129 @@ +/* + * Buffer management functions. + * + * Copyright 2000-2012 Willy Tarreau <w@1wt.eu> + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version + * 2 of the License, or (at your option) any later version. + * + */ + +#include <ctype.h> +#include <stdio.h> +#include <string.h> + +#include <haproxy/api.h> +#include <haproxy/dynbuf.h> +#include <haproxy/global.h> +#include <haproxy/list.h> +#include <haproxy/pool.h> + +struct pool_head *pool_head_buffer __read_mostly; + +/* perform minimal initializations, report 0 in case of error, 1 if OK. */ +int init_buffer() +{ + void *buffer; + int thr; + int done; + + pool_head_buffer = create_pool("buffer", global.tune.bufsize, MEM_F_SHARED|MEM_F_EXACT); + if (!pool_head_buffer) + return 0; + + for (thr = 0; thr < MAX_THREADS; thr++) + LIST_INIT(&ha_thread_ctx[thr].buffer_wq); + + + /* The reserved buffer is what we leave behind us. Thus we always need + * at least one extra buffer in minavail otherwise we'll end up waking + * up tasks with no memory available, causing a lot of useless wakeups. + * That means that we always want to have at least 3 buffers available + * (2 for current session, one for next session that might be needed to + * release a server connection). + */ + pool_head_buffer->minavail = MAX(global.tune.reserved_bufs, 3); + if (global.tune.buf_limit) + pool_head_buffer->limit = global.tune.buf_limit; + + for (done = 0; done < pool_head_buffer->minavail - 1; done++) { + buffer = pool_alloc_nocache(pool_head_buffer, init_buffer); + if (!buffer) + return 0; + pool_free(pool_head_buffer, buffer); + } + return 1; +} + +/* + * Dumps part or all of a buffer. + */ +void buffer_dump(FILE *o, struct buffer *b, int from, int to) +{ + fprintf(o, "Dumping buffer %p\n", b); + fprintf(o, " orig=%p size=%u head=%u tail=%u data=%u\n", + b_orig(b), (unsigned int)b_size(b), (unsigned int)b_head_ofs(b), (unsigned int)b_tail_ofs(b), (unsigned int)b_data(b)); + + fprintf(o, "Dumping contents from byte %d to byte %d\n", from, to); + fprintf(o, " 0 1 2 3 4 5 6 7 8 9 a b c d e f\n"); + /* dump hexa */ + while (from < to) { + int i; + + fprintf(o, " %04x: ", from); + for (i = 0; ((from + i) < to) && (i < 16) ; i++) { + fprintf(o, "%02x ", (unsigned char)b_orig(b)[from + i]); + if (i == 7) + fprintf(o, "- "); + } + if (to - from < 16) { + int j = 0; + + for (j = 0; j < from + 16 - to; j++) + fprintf(o, " "); + if (j > 8) + fprintf(o, " "); + } + fprintf(o, " "); + for (i = 0; (from + i < to) && (i < 16) ; i++) { + fprintf(o, "%c", isprint((unsigned char)b_orig(b)[from + i]) ? b_orig(b)[from + i] : '.') ; + if ((i == 15) && ((from + i) != to-1)) + fprintf(o, "\n"); + } + from += i; + } + fprintf(o, "\n--\n"); + fflush(o); +} + +/* see offer_buffers() for details */ +void __offer_buffers(void *from, unsigned int count) +{ + struct buffer_wait *wait, *wait_back; + + /* For now, we consider that all objects need 1 buffer, so we can stop + * waking up them once we have enough of them to eat all the available + * buffers. Note that we don't really know if they are streams or just + * other tasks, but that's a rough estimate. Similarly, for each cached + * event we'll need 1 buffer. + */ + list_for_each_entry_safe(wait, wait_back, &th_ctx->buffer_wq, list) { + if (!count) + break; + + if (wait->target == from || !wait->wakeup_cb(wait->target)) + continue; + + LIST_DEL_INIT(&wait->list); + count--; + } +} + +/* + * Local variables: + * c-indent-level: 8 + * c-basic-offset: 8 + * End: + */ |