xref: /openbsd-src/regress/lib/libpthread/restart/connect/connect.c (revision 5e8284b2689a3a73fc7763a8fe4743d6d3544a7d)
1 /*	$OpenBSD: connect.c,v 1.2 2016/08/09 02:25:35 guenther 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_connect(void *arg)
23 {
24 	struct sockaddr_in sa;
25 	int s;
26 
27 	CHECKe(s = socket(AF_INET, SOCK_STREAM, 0));
28 	bzero(&sa, sizeof(sa));
29 	sa.sin_family = AF_INET;
30 	sa.sin_port = htons(23);
31 	sa.sin_addr.s_addr = htonl(0xc7b98903);	/* cvs.openbsd.org */
32 	ASSERT(connect(s, (struct sockaddr *)&sa, sizeof(sa)) == -1);
33 	int err = errno;
34 	ASSERT(connect(s, (struct sockaddr *)&sa, sizeof(sa)) == -1);
35 	ASSERT(errno == EALREADY);
36 	return ((caddr_t)NULL + err);
37 }
38 
39 int
40 main(int argc, char **argv)
41 {
42 	struct sigaction sa;
43 	pthread_t tid;
44 	void *retval;
45 
46 	bzero(&sa, sizeof(sa));
47 	sa.sa_handler = handler;
48 	sa.sa_flags = SA_RESTART;
49 	CHECKe(sigaction(SIGUSR1, &sa, NULL));
50 
51 	CHECKr(pthread_create(&tid, NULL, thr_connect, NULL));
52 	sleep(2);
53 
54 	/* Should interrupt it. */
55 	CHECKr(pthread_kill(tid, SIGUSR1));
56 	sleep(1);
57 
58 	CHECKr(pthread_join(tid, &retval));
59 	ASSERT(retval == (void *)EINTR);
60 	ASSERT(hits == 1);
61 	SUCCEED;
62 }
63