llvm-project/compiler-rt/test/tsan/fd_socket_connect_norace.cpp

63 lines
1.4 KiB
C++

// RUN: %clangxx_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s
#include <arpa/inet.h>
#include <assert.h>
#include <netinet/in.h>
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
struct sockaddr_in addr4;
struct sockaddr_in6 addr6;
struct sockaddr *addr;
socklen_t addrlen;
int X;
void *ClientThread(void *x) {
int c = socket(addr->sa_family, SOCK_STREAM, IPPROTO_TCP);
X = 42;
if (connect(c, addr, addrlen)) {
perror("connect");
exit(1);
}
close(c);
return NULL;
}
int main() {
addr4.sin_family = AF_INET;
addr4.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
addr4.sin_port = INADDR_ANY;
addr = (struct sockaddr *)&addr4;
addrlen = sizeof(addr4);
int s = socket(addr->sa_family, SOCK_STREAM, IPPROTO_TCP);
if (s <= 0) {
// Try to fall-back to IPv6
addr6.sin6_family = AF_INET6;
addr6.sin6_addr = in6addr_loopback;
addr6.sin6_port = INADDR_ANY;
addr = (struct sockaddr *)&addr6;
addrlen = sizeof(addr6);
s = socket(addr->sa_family, SOCK_STREAM, IPPROTO_TCP);
}
assert(s > 0);
bind(s, addr, addrlen);
getsockname(s, addr, &addrlen);
listen(s, 10);
pthread_t t;
pthread_create(&t, 0, ClientThread, 0);
int c = accept(s, 0, 0);
X = 42;
pthread_join(t, 0);
close(c);
close(s);
fprintf(stderr, "OK\n");
}
// CHECK-NOT: WARNING: ThreadSanitizer: data race