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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
|
/*
* Copyright (C) 2015 Cyril Hrubis <chrubis@suse.cz>
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of version 2 of the GNU General Public License as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it would be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
*
* Further, this software is distributed without any warranty that it is
* free of the rightful claim of any third person regarding infringement
* or the like. Any license provided herein, whether implied or
* otherwise, applies only to this software file. Patent licenses, if
* any, provided herein do not apply to combinations of this program with
* other software, or any other product whatsoever.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include <stdint.h>
#include <limits.h>
#include <errno.h>
#include <sys/syscall.h>
#include <linux/futex.h>
#include "test.h"
#include "safe_macros.h"
#include "lapi/futex.h"
#define DEFAULT_MSEC_TIMEOUT 10000
futex_t *tst_futexes;
unsigned int tst_max_futexes;
void tst_checkpoint_init(const char *file, const int lineno,
void (*cleanup_fn)(void))
{
int fd;
unsigned int page_size;
if (tst_futexes) {
tst_brkm(TBROK, cleanup_fn,
"%s: %d checkpoints already initialized",
file, lineno);
return;
}
/*
* The parent test process is responsible for creating the temporary
* directory and therefore must pass non-zero cleanup (to remove the
* directory if something went wrong).
*
* We cannot do this check unconditionally because if we need to init
* the checkpoint from a binary that was started by exec() the
* tst_tmpdir_created() will return false because the tmpdir was
* created by parent. In this case we expect the subprogram can call
* the init as a first function with NULL as cleanup function.
*/
if (cleanup_fn && !tst_tmpdir_created()) {
tst_brkm(TBROK, cleanup_fn,
"%s:%d You have to create test temporary directory "
"first (call tst_tmpdir())", file, lineno);
return;
}
page_size = getpagesize();
fd = SAFE_OPEN(cleanup_fn, "checkpoint_futex_base_file",
O_RDWR | O_CREAT, 0666);
SAFE_FTRUNCATE(cleanup_fn, fd, page_size);
tst_futexes = SAFE_MMAP(cleanup_fn, NULL, page_size,
PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
tst_max_futexes = page_size / sizeof(uint32_t);
SAFE_CLOSE(cleanup_fn, fd);
}
int tst_checkpoint_wait(unsigned int id, unsigned int msec_timeout)
{
struct timespec timeout;
int ret;
if (id >= tst_max_futexes) {
errno = EOVERFLOW;
return -1;
}
timeout.tv_sec = msec_timeout/1000;
timeout.tv_nsec = (msec_timeout%1000) * 1000000;
do {
ret = syscall(SYS_futex, &tst_futexes[id], FUTEX_WAIT,
tst_futexes[id], &timeout);
} while (ret == -1 && errno == EINTR);
return ret;
}
int tst_checkpoint_wake(unsigned int id, unsigned int nr_wake,
unsigned int msec_timeout)
{
unsigned int msecs = 0, waked = 0;
if (id >= tst_max_futexes) {
errno = EOVERFLOW;
return -1;
}
for (;;) {
waked += syscall(SYS_futex, &tst_futexes[id], FUTEX_WAKE,
INT_MAX, NULL);
if (waked == nr_wake)
break;
usleep(1000);
msecs++;
if (msecs >= msec_timeout) {
errno = ETIMEDOUT;
return -1;
}
}
return 0;
}
void tst_safe_checkpoint_wait(const char *file, const int lineno,
void (*cleanup_fn)(void), unsigned int id,
unsigned int msec_timeout)
{
int ret;
if (!msec_timeout)
msec_timeout = DEFAULT_MSEC_TIMEOUT;
ret = tst_checkpoint_wait(id, msec_timeout);
if (ret) {
tst_brkm(TBROK | TERRNO, cleanup_fn,
"%s:%d: tst_checkpoint_wait(%u, %i)",
file, lineno, id, msec_timeout);
}
}
void tst_safe_checkpoint_wake(const char *file, const int lineno,
void (*cleanup_fn)(void), unsigned int id,
unsigned int nr_wake)
{
int ret = tst_checkpoint_wake(id, nr_wake, DEFAULT_MSEC_TIMEOUT);
if (ret) {
tst_brkm(TBROK | TERRNO, cleanup_fn,
"%s:%d: tst_checkpoint_wake(%u, %u, %i)",
file, lineno, id, nr_wake, DEFAULT_MSEC_TIMEOUT);
}
}
|
