memcheck/tests/freebsd/sctp.c

   1 /* This is based on /src/tools/test/stress2/misc/sctp.sh
   2  * but it uses fork and the server exits rather than
   3  * loops indefinitely */
   4
   5 #include <sys/types.h>
   6 #include <sys/socket.h>
   7 #include <unistd.h>
   8 #include <arpa/inet.h>
   9 #include <libgen.h>
  10 #include <netinet/in.h>
  11 #include <netinet/sctp.h>
  12 #include <stdio.h>
  13 #include <stdlib.h>
  14 #include <string.h>
  15
  16 static int my_port_num = 62324;
  17
  18 static void
  19 die(const char *s)
  20 {
  21    perror(s);
  22    exit(1);
  23 }
  24
  25 static void
  26 server(void)
  27 {
  28    struct sctp_sndrcvinfo sndrcvinfo;
  29    struct sockaddr_in servaddr = {
  30       .sin_family = AF_INET,
  31       .sin_addr.s_addr = htonl(INADDR_ANY),
  32       .sin_port = htons(my_port_num),
  33    };
  34    struct sctp_initmsg initmsg = {
  35       .sinit_num_ostreams = 5,
  36       .sinit_max_instreams = 5,
  37       .sinit_max_attempts = 4,
  38    };
  39    int listen_fd, conn_fd, flags, ret, in;
  40
  41    listen_fd = socket(AF_INET, SOCK_STREAM, IPPROTO_SCTP);
  42    if (listen_fd < 0)
  43       die("socket");
  44
  45    ret = bind(listen_fd, (struct sockaddr *) &servaddr, sizeof(servaddr));
  46    if (ret < 0)
  47       die("bind");
  48
  49    ret = setsockopt(listen_fd, IPPROTO_SCTP, SCTP_INITMSG, &initmsg,
  50          sizeof(initmsg));
  51    if (ret < 0)
  52       die("setsockopt");
  53
  54    ret = listen(listen_fd, initmsg.sinit_max_instreams);
  55    if (ret < 0)
  56       die("listen");
  57
  58    char buffer[1024];
  59
  60    printf("Waiting for connection\n");
  61    fflush(stdout);
  62
  63    conn_fd = accept(listen_fd, (struct sockaddr *) NULL, NULL);
  64    if(conn_fd < 0)
  65       die("accept()");
  66
  67    printf("New client connected\n");
  68    fflush(stdout);
  69
  70    /* Note that flags is uninitialized here */
  71    in = sctp_recvmsg(conn_fd, buffer, sizeof(buffer), NULL, 0,
  72          &sndrcvinfo, &flags);
  73    if (in > 0) {
  74       printf("Received data: %s\n", buffer);
  75       fflush(stdout);
  76    }
  77
  78    close(conn_fd);
  79 }
  80
  81 static void
  82 client(void)
  83 {
  84    struct sockaddr_in servaddr = {
  85       .sin_family = AF_INET,
  86       .sin_port = htons(my_port_num),
  87       .sin_addr.s_addr = inet_addr("127.0.0.1"),
  88    };
  89    int conn_fd, ret;
  90    const char *msg = "Hello, Server!";
  91
  92    conn_fd = socket(AF_INET, SOCK_STREAM, IPPROTO_SCTP);
  93    if (conn_fd < 0)
  94       die("socket()");
  95
  96    ret = connect(conn_fd, (struct sockaddr *) &servaddr, sizeof(servaddr));
  97    if (ret < 0)
  98       die("connect()");
  99
 100    ret = sctp_sendmsg(conn_fd, (void *) msg, strlen(msg) + 1, NULL, 0, 0, 0, 0, 0, 0 );
 101    if (ret < 0)
 102       die("sctp_sendmsg");
 103
 104    close(conn_fd);
 105 }
 106
 107 int
 108 main(int argc __unused, char *argv[])
 109 {
 110    switch (fork())
 111    {
 112    case 0:
 113       sleep(1);
 114       client();
 115       exit(EXIT_SUCCESS);
 116    default:
 117       server();
 118       exit(EXIT_SUCCESS);
 119    case -1:
 120       exit(EXIT_FAILURE);
 121    }
 122 }