xref: /llvm-project/compiler-rt/test/tsan/fd_socket_connect_norace.cpp (revision bcaeed49cb063de9fe504aa29e1cadff8a7be710)
1*bcaeed49SFangrui Song // RUN: %clangxx_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s
2*bcaeed49SFangrui Song #include <arpa/inet.h>
3*bcaeed49SFangrui Song #include <assert.h>
4*bcaeed49SFangrui Song #include <netinet/in.h>
5*bcaeed49SFangrui Song #include <pthread.h>
6*bcaeed49SFangrui Song #include <stdio.h>
7*bcaeed49SFangrui Song #include <stdlib.h>
8*bcaeed49SFangrui Song #include <sys/socket.h>
9*bcaeed49SFangrui Song #include <sys/types.h>
10*bcaeed49SFangrui Song #include <unistd.h>
11*bcaeed49SFangrui Song 
12*bcaeed49SFangrui Song struct sockaddr_in addr4;
13*bcaeed49SFangrui Song struct sockaddr_in6 addr6;
14*bcaeed49SFangrui Song struct sockaddr *addr;
15*bcaeed49SFangrui Song socklen_t addrlen;
16*bcaeed49SFangrui Song int X;
17*bcaeed49SFangrui Song 
ClientThread(void * x)18*bcaeed49SFangrui Song void *ClientThread(void *x) {
19*bcaeed49SFangrui Song   int c = socket(addr->sa_family, SOCK_STREAM, IPPROTO_TCP);
20*bcaeed49SFangrui Song   X = 42;
21*bcaeed49SFangrui Song   if (connect(c, addr, addrlen)) {
22*bcaeed49SFangrui Song     perror("connect");
23*bcaeed49SFangrui Song     exit(1);
24*bcaeed49SFangrui Song   }
25*bcaeed49SFangrui Song   close(c);
26*bcaeed49SFangrui Song   return NULL;
27*bcaeed49SFangrui Song }
28*bcaeed49SFangrui Song 
main()29*bcaeed49SFangrui Song int main() {
30*bcaeed49SFangrui Song   addr4.sin_family = AF_INET;
31*bcaeed49SFangrui Song   addr4.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
32*bcaeed49SFangrui Song   addr4.sin_port = INADDR_ANY;
33*bcaeed49SFangrui Song   addr = (struct sockaddr *)&addr4;
34*bcaeed49SFangrui Song   addrlen = sizeof(addr4);
35*bcaeed49SFangrui Song 
36*bcaeed49SFangrui Song   int s = socket(addr->sa_family, SOCK_STREAM, IPPROTO_TCP);
37*bcaeed49SFangrui Song   if (s <= 0) {
38*bcaeed49SFangrui Song     // Try to fall-back to IPv6
39*bcaeed49SFangrui Song     addr6.sin6_family = AF_INET6;
40*bcaeed49SFangrui Song     addr6.sin6_addr = in6addr_loopback;
41*bcaeed49SFangrui Song     addr6.sin6_port = INADDR_ANY;
42*bcaeed49SFangrui Song     addr = (struct sockaddr *)&addr6;
43*bcaeed49SFangrui Song     addrlen = sizeof(addr6);
44*bcaeed49SFangrui Song     s = socket(addr->sa_family, SOCK_STREAM, IPPROTO_TCP);
45*bcaeed49SFangrui Song   }
46*bcaeed49SFangrui Song   assert(s > 0);
47*bcaeed49SFangrui Song 
48*bcaeed49SFangrui Song   bind(s, addr, addrlen);
49*bcaeed49SFangrui Song   getsockname(s, addr, &addrlen);
50*bcaeed49SFangrui Song   listen(s, 10);
51*bcaeed49SFangrui Song   pthread_t t;
52*bcaeed49SFangrui Song   pthread_create(&t, 0, ClientThread, 0);
53*bcaeed49SFangrui Song   int c = accept(s, 0, 0);
54*bcaeed49SFangrui Song   X = 42;
55*bcaeed49SFangrui Song   pthread_join(t, 0);
56*bcaeed49SFangrui Song   close(c);
57*bcaeed49SFangrui Song   close(s);
58*bcaeed49SFangrui Song   fprintf(stderr, "OK\n");
59*bcaeed49SFangrui Song }
60*bcaeed49SFangrui Song 
61*bcaeed49SFangrui Song // CHECK-NOT: WARNING: ThreadSanitizer: data race
62*bcaeed49SFangrui Song 
63