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
|
/* $OpenBSD: recvmsg.c,v 1.2 2016/01/27 01:20:10 jca Exp $ */
/*
* Federico G. Schwindt <fgsch@openbsd.org>, 2011. Public Domain.
*/
#include <sys/types.h>
#include <sys/uio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <pthread.h>
#include <signal.h>
#include <unistd.h>
#include "test.h"
volatile sig_atomic_t hits = 0;
void
handler(int sig)
{
hits++;
}
void *
thr_recvmsg(void *arg)
{
struct sockaddr_in sa;
struct msghdr msg;
struct iovec iov;
char buf;
int s;
CHECKe(s = socket(AF_INET, SOCK_DGRAM, 0));
bzero(&sa, sizeof(sa));
sa.sin_family = AF_INET;
sa.sin_port = htons(6543);
CHECKe(bind(s, (const void*)&sa, sizeof(sa)));
bzero(&msg, sizeof(msg));
iov.iov_base = &buf;
iov.iov_len = 1;
msg.msg_iov = &iov;
msg.msg_iovlen = 1;
ASSERT(recvmsg(s, &msg, 0) == -1);
return ((caddr_t)NULL + errno);
}
int
main(int argc, char **argv)
{
struct sigaction sa;
pthread_t tid;
void *retval;
bzero(&sa, sizeof(sa));
sa.sa_handler = handler;
sa.sa_flags = SA_RESTART;
CHECKe(sigaction(SIGUSR1, &sa, NULL));
sa.sa_flags = 0;
CHECKe(sigaction(SIGUSR2, &sa, NULL));
CHECKr(pthread_create(&tid, NULL, thr_recvmsg, NULL));
sleep(2);
/* Should restart it. */
CHECKr(pthread_kill(tid, SIGUSR1));
sleep(1);
/* Should interrupt it. */
CHECKr(pthread_kill(tid, SIGUSR2));
sleep(1);
CHECKr(pthread_join(tid, &retval));
ASSERT(retval == (void *)EINTR);
ASSERT(hits == 2);
SUCCEED;
}
|
