summaryrefslogtreecommitdiff
path: root/arena-memstream.c
blob: 64ca09048d34610ed4d50b12e936b909fd36a91b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#include <assert.h>
#include <stdio.h>
#include <stddef.h>
#include <string.h>
#include <stdlib.h>
#include <stdint.h>

struct arena {
    FILE *stream;
    char *buffer;
    size_t offset;
    size_t size;
};

void arena_create(struct arena *arena)
{
    *arena = (struct arena){open_memstream(&arena->buffer, &arena->size)};
}

void arena_destroy(struct arena *arena)
{
    fclose(arena->stream);
    free(arena->buffer);
    *arena = (struct arena){0};
}

size_t arena_add(struct arena *arena, void const *data, size_t const size)
{
    size_t const offset = arena->offset;
    if (data) {
        fwrite(data, 1, size, arena->stream);
    }
    arena->offset += size;
    return offset;
}

static inline void *arena_get(struct arena const *arena, size_t offset)
{
    fflush(arena->stream);
    return arena->buffer + offset;
}

int main(int argc, char * const argv[])
{
    (void)argc;
    (void)argv;
    struct arena arena;
    arena_create(&arena);
    unsigned data[4] = {1u, 2u, 3u, 4u};
    for (size_t i = 0; i < 50000000lu; i++) {
        size_t const offset = arena_add(&arena, data, sizeof(data));
        data[0] = (uintptr_t)arena_get(&arena, offset);
    }
    arena_destroy(&arena);
}