summaryrefslogtreecommitdiffstats
path: root/compiler-rt/test/sanitizer_common/TestCases/NetBSD/paccept.cpp
blob: 7d2eb4e85596f4b35e86588a4d82a7da97071aa5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
// RUN: %clangxx -O0 -g %s -o %t && %run %t

#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <signal.h>
#include <stdlib.h>
#include <unistd.h>

int main(void) {
  int child;
  int fd, sfd;
  socklen_t len;
  struct sockaddr_in server = {}, client = {};
  sigset_t set;

  child = fork();
  if (child == 0) {
    fd = socket(AF_INET, SOCK_STREAM, 0);
    if (fd == -1)
      _exit(1);

    server.sin_family = AF_INET;
    server.sin_addr.s_addr = INADDR_ANY;
    server.sin_port = htons(2222);

    if (connect(fd, (struct sockaddr *)&server, sizeof(server)) == -1)
      _exit(1);

    close(fd);

    _exit(0);
  }

  fd = socket(AF_INET, SOCK_STREAM, 0);
  if (fd == -1) {
    kill(child, SIGKILL);
    wait(NULL);
    exit(1);
  }

  server.sin_family = AF_INET;
  server.sin_addr.s_addr = INADDR_ANY;
  server.sin_port = htons(2222);

  if (bind(fd, (const struct sockaddr *)&server, sizeof(server)) == -1) {
    kill(child, SIGKILL);
    wait(NULL);
    exit(1);
  }

  listen(fd, 3);

  if (sigemptyset(&set) == -1) {
    kill(child, SIGKILL);
    wait(NULL);
    exit(1);
  }

  len = sizeof(client);
  sfd = paccept(fd, (struct sockaddr *)&client, &len, &set, SOCK_NONBLOCK);
  if (sfd == -1) {
    kill(child, SIGKILL);
    wait(NULL);
    exit(1);
  }

  wait(NULL);

  close(sfd);
  close(fd);

  return 0;
}
OpenPOWER on IntegriCloud