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
87
88
89
90
91
92
93
|
#include "term/term.h"
#include <assert.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdlib.h>
/* --wrap redirects calls after LTO has analyzed their side effects. Keep
* these accesses observable so it cannot fold away the wrapper's changes. */
static volatile bool fail_next_realloc;
static volatile int malloc_failure_countdown = -1;
static volatile bool malloc_was_failed;
void *__real_malloc(size_t);
void *__wrap_malloc(size_t size)
{
if (malloc_failure_countdown >= 0) {
if (malloc_failure_countdown-- == 0) {
malloc_failure_countdown = -1;
malloc_was_failed = true;
return NULL;
}
}
return __real_malloc(size);
}
void *__real_realloc(void *, size_t);
void *__wrap_realloc(void *p, size_t size)
{
if (fail_next_realloc) {
fail_next_realloc = false;
return NULL;
}
return __real_realloc(p, size);
}
int main(void)
{
struct term_options options = { .size = { 8, 3 }, .scrollback = 5 };
struct term *term = term_new(&options);
struct term_snapshot *snapshot = term_snapshot_new();
assert(term && snapshot && term_snapshot_update(snapshot, term));
const char accented[] = "e\xcc\x81";
term_feed(term, accented, sizeof(accented) - 1, 0);
fail_next_realloc = true;
assert(!term_snapshot_update(snapshot, term));
assert(!fail_next_realloc);
assert(term_snapshot_view(snapshot)->size.x == 0);
assert(term_snapshot_view(snapshot)->size.y == 0);
assert(term_snapshot_row(snapshot, 0) == NULL);
assert(term_snapshot_update(snapshot, term));
const struct term_cell *row = term_snapshot_row(snapshot, 0);
assert(row);
uint32_t cps[4];
assert(term_snapshot_cluster(snapshot, row[0].text, cps, 4) == 2);
assert(cps[0] == 'e' && cps[1] == 0x301);
term_snapshot_free(snapshot);
term_free(term);
/* More than 256 rows moves both temporary arrays off the stack. */
options.size.y = 300;
term = term_new(&options);
snapshot = term_snapshot_new();
assert(term && snapshot && term_snapshot_update(snapshot, term));
/* Fail new_keys allocation, then make sure retry fills all rows. */
malloc_was_failed = false;
malloc_failure_countdown = 0;
assert(!term_snapshot_update(snapshot, term));
assert(malloc_was_failed);
assert(term_snapshot_view(snapshot)->size.y == 0);
assert(term_snapshot_row(snapshot, 0) == NULL);
assert(term_snapshot_update(snapshot, term));
assert(term_snapshot_view(snapshot)->size.y == 300);
assert(term_snapshot_row(snapshot, 299));
/* The first malloc is new_keys; fail the following selection-span array. */
malloc_was_failed = false;
malloc_failure_countdown = 1;
assert(!term_snapshot_update(snapshot, term));
assert(malloc_was_failed);
assert(term_snapshot_view(snapshot)->size.y == 0);
assert(term_snapshot_row(snapshot, 0) == NULL);
assert(term_snapshot_update(snapshot, term));
assert(term_snapshot_view(snapshot)->size.y == 300);
assert(term_snapshot_row(snapshot, 299));
term_snapshot_free(snapshot);
term_free(term);
return 0;
}
|
