granicus.if.org Git - strace/blob - tests/inet-accept-connect-send-recv.c

   1 #include <assert.h>
   2 #include <stddef.h>
   3 #include <string.h>
   4 #include <unistd.h>
   5 #include <sys/wait.h>
   6 #include <sys/socket.h>
   7 #include <netinet/in.h>
   8 #include <arpa/inet.h>
   9
  10 int main(void)
  11 {
  12         static const char data[] = "data";
  13         const size_t size = sizeof(data) - 1;
  14         struct sockaddr_in addr;
  15         socklen_t len = sizeof(addr);
  16         pid_t pid;
  17
  18         memset(&addr, 0, sizeof(addr));
  19         addr.sin_family = AF_INET;
  20         addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
  21
  22         close(0);
  23         close(1);
  24
  25         assert(socket(PF_INET, SOCK_STREAM, 0) == 0);
  26         assert(bind(0, (struct sockaddr *) &addr, len) == 0);
  27         assert(listen(0, 5) == 0);
  28
  29         memset(&addr, 0, sizeof(addr));
  30         assert(getsockname(0, (struct sockaddr *) &addr, &len) == 0);
  31
  32         assert((pid = fork()) >= 0);
  33
  34         if (pid) {
  35                 char buf[sizeof(data)];
  36                 int status;
  37
  38                 assert(accept(0, (struct sockaddr *) &addr, &len) == 1);
  39                 assert(close(0) == 0);
  40                 assert(recv(1, buf, sizeof(buf), MSG_WAITALL) == (int) size);
  41                 assert(waitpid(pid, &status, 0) == pid);
  42                 assert(status == 0);
  43                 assert(close(1) == 0);
  44         } else {
  45                 assert(close(0) == 0);
  46                 assert(socket(PF_INET, SOCK_STREAM, 0) == 0);
  47                 assert(connect(0, (struct sockaddr *) &addr, len) == 0);
  48                 assert(send(0, data, size, MSG_DONTROUTE) == (int) size);
  49                 assert(close(0) == 0);
  50         }
  51
  52         return 0;
  53 }