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
|
/* FIXME: Do something tricky to ensure we really do loop in read_all. */
#include <ccan/read_write_all/read_write_all.h>
#include <ccan/read_write_all/read_write_all.c>
#include <ccan/tap/tap.h>
#include <unistd.h>
#include <sys/types.h>
#include <signal.h>
#include <sys/wait.h>
#include <limits.h>
#include <err.h>
#include <stdlib.h>
#include <string.h>
static volatile int sigcount;
static int p2c[2], c2p[2];
static void got_signal(int sig)
{
char c = 0;
if (write(p2c[1], &c, 1) == 1)
sigcount++;
}
/* < PIPE_BUF *will* be atomic. But > PIPE_BUF only *might* be non-atomic. */
#define BUFSZ (1024*1024)
int main(int argc, char *argv[])
{
char *buffer;
char c = 0;
int status;
pid_t child;
buffer = calloc(BUFSZ, 2);
plan_tests(6);
/* We fork and torture parent. */
if (pipe(p2c) != 0 || pipe(c2p) != 0)
err(1, "pipe");
child = fork();
if (!child) {
close(p2c[1]);
close(c2p[0]);
/* Child. Make sure parent ready, then write in two parts. */
if (read(p2c[0], &c, 1) != 1)
exit(1);
memset(buffer, 0xff, BUFSZ*2);
if (!write_all(c2p[1], buffer, BUFSZ))
exit(2);
if (kill(getppid(), SIGUSR1) != 0)
exit(3);
/* Make sure they get signal. */
if (read(p2c[0], &c, 1) != 1)
exit(4);
if (write(c2p[1], buffer, BUFSZ) != BUFSZ)
exit(5);
exit(0);
}
if (child == -1)
err(1, "forking");
close(p2c[0]);
close(c2p[1]);
signal(SIGUSR1, got_signal);
ok1(write(p2c[1], &c, 1) == 1);
ok1(read_all(c2p[0], buffer, BUFSZ*2));
ok1(memchr(buffer, 0, BUFSZ*2) == NULL);
ok1(sigcount == 1);
ok1(wait(&status) == child);
ok(WIFEXITED(status) && WEXITSTATUS(status) == 0,
"WIFEXITED(status) = %u, WEXITSTATUS(status) = %u",
WIFEXITED(status), WEXITSTATUS(status));
free(buffer);
return exit_status();
}
|
