xref: /openbsd-src/regress/lib/libpthread/restart/accept/accept.c (revision 91f110e064cd7c194e59e019b83bb7496c1c84d4)
1 /*	$OpenBSD: accept.c,v 1.1 2011/09/13 23:50:17 fgsch Exp $	*/
2 /*
3  * Federico G. Schwindt <fgsch@openbsd.org>, 2011. Public Domain.
4  */
5 #include <sys/types.h>
6 #include <sys/socket.h>
7 #include <netinet/in.h>
8 #include <pthread.h>
9 #include <signal.h>
10 #include <unistd.h>
11 #include "test.h"
12 
13 volatile sig_atomic_t hits = 0;
14 
15 void
16 handler(int sig)
17 {
18 	hits++;
19 }
20 
21 void *
22 thr_accept(void *arg)
23 {
24 	struct sockaddr_in sa;
25 	socklen_t salen;
26 	int s;
27 
28 	CHECKe(s = socket(AF_INET, SOCK_STREAM, 0));
29 	bzero(&sa, sizeof(sa));
30 	sa.sin_port = htons(6543);
31 	CHECKe(bind(s, (const void*)&sa, sizeof(sa)));
32 	CHECKe(listen(s, 10));
33 	salen = sizeof(sa);
34 	ASSERT(accept(s, (struct sockaddr *)&sa, &salen) == -1);
35 	return ((caddr_t)NULL + errno);
36 }
37 
38 int
39 main(int argc, char **argv)
40 {
41 	struct sigaction sa;
42 	pthread_t tid;
43 	void *retval;
44 
45 	bzero(&sa, sizeof(sa));
46 	sa.sa_handler = handler;
47 	sa.sa_flags = SA_RESTART;
48 	CHECKe(sigaction(SIGUSR1, &sa, NULL));
49 	sa.sa_flags = 0;
50 	CHECKe(sigaction(SIGUSR2, &sa, NULL));
51 
52 	CHECKr(pthread_create(&tid, NULL, thr_accept, NULL));
53 	sleep(2);
54 
55 	/* Should restart it. */
56 	CHECKr(pthread_kill(tid, SIGUSR1));
57 	sleep(1);
58 
59 	/* Should interrupt it. */
60 	CHECKr(pthread_kill(tid, SIGUSR2));
61 	sleep(1);
62 
63 	CHECKr(pthread_join(tid, &retval));
64 	ASSERT(retval == (void *)EINTR);
65 	ASSERT(hits == 2);
66 	SUCCEED;
67 }
68