You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
1.5 KiB
81 lines
1.5 KiB
2 months ago
|
/*
|
||
|
|
* UCW Library -- Fast Buffered I/O on Memory Pools
|
||
|
|
*
|
||
|
|
* (c) 2007 Pavel Charvat <pchar@ucw.cz>
|
||
|
|
*
|
||
|
|
* This software may be freely distributed and used according to the terms
|
||
|
|
* of the GNU Lesser General Public License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <ucw/lib.h>
|
||
|
|
#include <ucw/mempool.h>
|
||
|
|
#include <ucw/fastbuf.h>
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
|
||
|
|
#define FB_POOL(f) ((struct fbpool *)(f))
|
||
|
|
|
||
|
|
static void
|
||
|
|
fbpool_spout(struct fastbuf *b)
|
||
|
|
{
|
||
|
|
if (b->bptr == b->bufend)
|
||
|
|
{
|
||
|
|
uint len = b->bufend - b->buffer;
|
||
|
|
b->bstop = b->buffer = mp_expand(FB_POOL(b)->mp);
|
||
|
|
b->bufend = b->buffer + mp_avail(FB_POOL(b)->mp);
|
||
|
|
b->bptr = b->buffer + len;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
fbpool_start(struct fbpool *b, struct mempool *mp, size_t init_size)
|
||
|
|
{
|
||
|
|
b->mp = mp;
|
||
|
|
b->fb.buffer = b->fb.bstop = b->fb.bptr = mp_start(mp, init_size);
|
||
|
|
b->fb.bufend = b->fb.buffer + mp_avail(mp);
|
||
|
|
}
|
||
|
|
|
||
|
|
void *
|
||
|
|
fbpool_end(struct fbpool *b)
|
||
|
|
{
|
||
|
|
return mp_end(b->mp, b->fb.bptr);
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
fbpool_init(struct fbpool *b)
|
||
|
|
{
|
||
|
|
bzero(b, sizeof(*b));
|
||
|
|
b->fb.name = "<fbpool>";
|
||
|
|
b->fb.spout = fbpool_spout;
|
||
|
|
b->fb.can_overwrite_buffer = 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef TEST
|
||
|
|
|
||
|
|
int main(void)
|
||
|
|
{
|
||
|
|
struct mempool *mp;
|
||
|
|
struct fbpool fb;
|
||
|
|
byte *p;
|
||
|
|
uint l;
|
||
|
|
|
||
|
|
mp = mp_new(64);
|
||
|
|
fbpool_init(&fb);
|
||
|
|
fbpool_start(&fb, mp, 16);
|
||
|
|
for (uint i = 0; i < 1024; i++)
|
||
|
|
bprintf(&fb.fb, "<hello>");
|
||
|
|
p = fbpool_end(&fb);
|
||
|
|
l = mp_size(mp, p);
|
||
|
|
if (l != 1024 * 7)
|
||
|
|
ASSERT(0);
|
||
|
|
for (uint i = 0; i < 1024; i++)
|
||
|
|
if (memcmp(p + i * 7, "<hello>", 7))
|
||
|
|
ASSERT(0);
|
||
|
|
mp_delete(mp);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|