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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* Include the implementation to inspect snapshot arena accounting. Link
* against the static library so snapshot.c is compiled only here. */
#include "../src/snapshot.c"
static int fail_realloc;
static int fail_resize_calloc;
void *__real_realloc(void *, size_t);
void *__real_calloc(size_t, size_t);
void *__wrap_calloc(size_t n, size_t size)
{
if (fail_resize_calloc && n == 2 && size == sizeof(struct srow)) {
fail_resize_calloc = 0;
return NULL;
}
return __real_calloc(n, size);
}
void *__wrap_realloc(void *ptr, size_t size)
{
if (fail_realloc) { fail_realloc = 0; return NULL; }
return __real_realloc(ptr, size);
}
int main(void)
{
bool failed = false;
struct wsap_vtcore_options options = { .size = { 8, 3 }, .scrollback = 5 };
struct wsap_vtcore *t = wsap_vtcore_new(&options);
struct wsap_vtcore_snapshot *s = wsap_vtcore_snapshot_new();
assert(t && s && wsap_vtcore_snapshot_update(s, t));
const char accented[] = "e\xcc\x81";
wsap_vtcore_feed(t, accented, sizeof(accented) - 1, 0);
fail_realloc = 1;
bool ok = wsap_vtcore_snapshot_update(s, t);
uint32_t cps[16];
printf("snapshot arena allocation failure: update=%d size=%dx%d (expected false and the previous 8x3 view)\n",
ok, wsap_vtcore_snapshot_view(s)->size.x, wsap_vtcore_snapshot_view(s)->size.y);
failed |= ok || wsap_vtcore_snapshot_view(s)->size.x != 8 || wsap_vtcore_snapshot_view(s)->size.y != 3;
assert(wsap_vtcore_snapshot_update(s, t));
int n = wsap_vtcore_snapshot_cluster(s, wsap_vtcore_snapshot_row(s, 0)[0].text, cps, 16);
printf("retry after allocation recovers: cluster_length=%d (expected 2)\n", n);
failed |= n != 2 || cps[0] != 'e' || cps[1] != 0x301;
wsap_vtcore_snapshot_free(s);
wsap_vtcore_free(t);
options.size.y = 24;
t = wsap_vtcore_new(&options);
s = wsap_vtcore_snapshot_new();
assert(t && s);
const char line[] = "e\xcc\x81\r\n";
for (int i = 1; i <= 10000; i++) {
wsap_vtcore_feed(t, line, sizeof(line) - 1, 0);
assert(wsap_vtcore_snapshot_update(s, t));
if (i == 1000 || i == 10000)
printf("after %d lines: arena_words=%u dead_words=%u history_rows=%d\n",
i, s->arena_len, s->arena_dead, wsap_vtcore_history_rows(t));
}
failed |= s->arena_len >= 8192 || wsap_vtcore_history_rows(t) != 5;
wsap_vtcore_snapshot_free(s);
wsap_vtcore_free(t);
options.size = (struct wsap_vtcore_vector2i){ 8, 4 };
t = wsap_vtcore_new(&options);
s = wsap_vtcore_snapshot_new();
assert(t && s);
const char rows[] = "A\r\nB\r\nC\r\nD";
wsap_vtcore_feed(t, rows, sizeof(rows) - 1, 0);
assert(wsap_vtcore_snapshot_update(s, t));
printf("before failed height shrink: top=%c cursor_y=%d history=%d\n",
(char)wsap_vtcore_snapshot_row(s, 0)[0].text, wsap_vtcore_cursor(t).pos.y, wsap_vtcore_history_rows(t));
fail_resize_calloc = 1;
ok = wsap_vtcore_resize(t, (struct wsap_vtcore_vector2i){ 8, 2 });
assert(!fail_resize_calloc);
assert(wsap_vtcore_snapshot_update(s, t));
printf("after failed height shrink: resize=%d height=%d top=%c cursor_y=%d history=%d\n",
ok, wsap_vtcore_size(t).y, (char)wsap_vtcore_snapshot_row(s, 0)[0].text,
wsap_vtcore_cursor(t).pos.y, wsap_vtcore_history_rows(t));
failed |= ok || wsap_vtcore_size(t).y != 4 || wsap_vtcore_snapshot_row(s, 0)[0].text != 'A' ||
wsap_vtcore_cursor(t).pos.y != 3 || wsap_vtcore_history_rows(t) != 0;
wsap_vtcore_snapshot_free(s);
wsap_vtcore_free(t);
return failed ? 1 : 0;
}
|
