fd_socket_connect_norace.cpp 1.39 KB
// 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