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
|
/*
* gensio - A library for abstracting stream I/O
* Copyright (C) 2020 Corey Minyard <minyard@acm.org>
*
* SPDX-License-Identifier: LGPL-2.1-only
*/
#include "errtrig.h"
#ifdef ENABLE_ERRTRIG_TEST
#include "pthread_handler.h"
#include <assert.h>
/*
* Some memory allocation and other failure testing. If the
* GENSIO_ERRTRIG_TEST environment variable is set to number N, the
* Nth call to do_errtrig will return true. The program should call
* gensio_sel_exit (below); it will cause specific values to be
* returned on an exit failure.
*/
static lock_type errtrig_mutex;
static bool errtrig_initialized;
static bool errtrig_ready;
static bool triggered;
static unsigned int errtrig_count;
static unsigned int errtrig_curr;
bool do_errtrig(void)
{
unsigned int curr;
bool triggerit = false;
LOCK(&errtrig_mutex);
if (!errtrig_initialized) {
char *s = getenv("GENSIO_ERRTRIG_TEST");
errtrig_initialized = true;
if (s) {
errtrig_count = strtoul(s, NULL, 0);
errtrig_ready = true;
}
}
if (errtrig_ready) {
curr = errtrig_curr++;
if (curr == errtrig_count) {
triggered = true;
triggerit = true;
}
}
UNLOCK(&errtrig_mutex);
return triggerit;
}
#include <stdio.h>
void errtrig_exit(int rv)
{
if (!errtrig_ready)
exit(rv);
assert (rv == 1 || rv == 0); /* Only these values are allowed. */
/*
* Return an error. The values mean:
*
* 0 - No error occurred and the memory allocation failure didn't happen
* 1 - An error occurred and the memory allocation failure happenned
* 2 - No error occurred and the memory allocation failure happenned
* 3 - An error occurred and the memory allocation failure didn't happen
*/
if (rv == 0 && triggered)
rv = 2;
if (rv == 0 && !triggered)
rv = 0;
if (rv == 1 && triggered)
rv = 1;
if (rv == 1 && !triggered)
rv = 3;
exit(rv);
}
#endif
|
