1*ec9a4a96Sbluhm /* $OpenBSD: connect.c,v 1.3 2017/05/27 16:04:38 bluhm Exp $ */
22d9d7a1eSfgsch /*
32d9d7a1eSfgsch * Federico G. Schwindt <fgsch@openbsd.org>, 2011. Public Domain.
42d9d7a1eSfgsch */
52d9d7a1eSfgsch #include <sys/types.h>
62d9d7a1eSfgsch #include <sys/socket.h>
72d9d7a1eSfgsch #include <netinet/in.h>
82d9d7a1eSfgsch #include <pthread.h>
92d9d7a1eSfgsch #include <signal.h>
102d9d7a1eSfgsch #include <unistd.h>
112d9d7a1eSfgsch #include "test.h"
122d9d7a1eSfgsch
132d9d7a1eSfgsch volatile sig_atomic_t hits = 0;
142d9d7a1eSfgsch
152d9d7a1eSfgsch void
handler(int sig)162d9d7a1eSfgsch handler(int sig)
172d9d7a1eSfgsch {
182d9d7a1eSfgsch hits++;
192d9d7a1eSfgsch }
202d9d7a1eSfgsch
212d9d7a1eSfgsch void *
thr_connect(void * arg)222d9d7a1eSfgsch thr_connect(void *arg)
232d9d7a1eSfgsch {
242d9d7a1eSfgsch struct sockaddr_in sa;
25*ec9a4a96Sbluhm socklen_t len;
26*ec9a4a96Sbluhm int l, s;
272d9d7a1eSfgsch
28*ec9a4a96Sbluhm /* Create a bound TCP socket without listen on loopback. */
29*ec9a4a96Sbluhm CHECKe(l = socket(AF_INET, SOCK_STREAM, 0));
302d9d7a1eSfgsch bzero(&sa, sizeof(sa));
312d9d7a1eSfgsch sa.sin_family = AF_INET;
32*ec9a4a96Sbluhm sa.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
33*ec9a4a96Sbluhm CHECKe(bind(l, (struct sockaddr *)&sa, sizeof(sa)));
34*ec9a4a96Sbluhm len = sizeof(sa);
35*ec9a4a96Sbluhm CHECKe(getsockname(l, (struct sockaddr *)&sa, &len));
36*ec9a4a96Sbluhm
37*ec9a4a96Sbluhm /* Connect to the non listen socket will not reply to SYN. */
38*ec9a4a96Sbluhm CHECKe(s = socket(AF_INET, SOCK_STREAM, 0));
392d9d7a1eSfgsch ASSERT(connect(s, (struct sockaddr *)&sa, sizeof(sa)) == -1);
405e8284b2Sguenther int err = errno;
415e8284b2Sguenther ASSERT(connect(s, (struct sockaddr *)&sa, sizeof(sa)) == -1);
425e8284b2Sguenther ASSERT(errno == EALREADY);
435e8284b2Sguenther return ((caddr_t)NULL + err);
442d9d7a1eSfgsch }
452d9d7a1eSfgsch
462d9d7a1eSfgsch int
main(int argc,char ** argv)472d9d7a1eSfgsch main(int argc, char **argv)
482d9d7a1eSfgsch {
492d9d7a1eSfgsch struct sigaction sa;
502d9d7a1eSfgsch pthread_t tid;
512d9d7a1eSfgsch void *retval;
522d9d7a1eSfgsch
532d9d7a1eSfgsch bzero(&sa, sizeof(sa));
542d9d7a1eSfgsch sa.sa_handler = handler;
552d9d7a1eSfgsch sa.sa_flags = SA_RESTART;
562d9d7a1eSfgsch CHECKe(sigaction(SIGUSR1, &sa, NULL));
572d9d7a1eSfgsch
582d9d7a1eSfgsch CHECKr(pthread_create(&tid, NULL, thr_connect, NULL));
592d9d7a1eSfgsch sleep(2);
602d9d7a1eSfgsch
612d9d7a1eSfgsch /* Should interrupt it. */
622d9d7a1eSfgsch CHECKr(pthread_kill(tid, SIGUSR1));
632d9d7a1eSfgsch sleep(1);
642d9d7a1eSfgsch
652d9d7a1eSfgsch CHECKr(pthread_join(tid, &retval));
662d9d7a1eSfgsch ASSERT(retval == (void *)EINTR);
672d9d7a1eSfgsch ASSERT(hits == 1);
682d9d7a1eSfgsch SUCCEED;
692d9d7a1eSfgsch }
70