tests/test_echo_udp_send_recv.c

   1 #include <stdarg.h>
   2 #include <stddef.h>
   3 #include <setjmp.h>
   4 #include <cmocka.h>
   5
   6 #include "config.h"
   7 #include "torture.h"
   8
   9 #include <errno.h>
  10 #include <sys/types.h>
  11 #include <sys/socket.h>
  12 #include <netinet/in.h>
  13 #include <arpa/inet.h>
  14 #include <stdlib.h>
  15 #include <stdio.h>
  16 #include <unistd.h>
  17
  18 static int setup_echo_srv_udp_ipv4(void **state)
  19 {
  20         torture_setup_echo_srv_udp_ipv4(state);
  21
  22         return 0;
  23 }
  24
  25 #ifdef HAVE_IPV6
  26 static int setup_echo_srv_udp_ipv6(void **state)
  27 {
  28         torture_setup_echo_srv_udp_ipv6(state);
  29
  30         return 0;
  31 }
  32 #endif
  33
  34 static int teardown(void **state)
  35 {
  36         torture_teardown_echo_srv(state);
  37
  38         return 0;
  39 }
  40
  41 static void test_send_recv_ipv4(void **state)
  42 {
  43         struct torture_address addr = {
  44                 .sa_socklen = sizeof(struct sockaddr_in),
  45         };
  46         ssize_t ret;
  47         int rc;
  48         int i;
  49         int s;
  50
  51         (void) state; /* unused */
  52
  53         s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
  54         assert_int_not_equal(s, -1);
  55
  56         addr.sa.in.sin_family = AF_INET;
  57         addr.sa.in.sin_port = htons(torture_server_port());
  58
  59         rc = inet_pton(AF_INET,
  60                        torture_server_address(AF_INET),
  61                        &addr.sa.in.sin_addr);
  62         assert_int_equal(rc, 1);
  63
  64         rc = connect(s, &addr.sa.s, addr.sa_socklen);
  65         assert_int_equal(rc, 0);
  66
  67         for (i = 0; i < 10; i++) {
  68                 char send_buf[64] = {0};
  69                 char recv_buf[64] = {0};
  70
  71                 snprintf(send_buf, sizeof(send_buf), "packet.%d", i);
  72
  73                 ret = send(s,
  74                            send_buf,
  75                            sizeof(send_buf),
  76                            0);
  77                 assert_int_not_equal(ret, -1);
  78
  79                 ret = recv(s,
  80                            recv_buf,
  81                            sizeof(recv_buf),
  82                            0);
  83                 assert_int_not_equal(ret, -1);
  84
  85                 assert_memory_equal(send_buf, recv_buf, sizeof(send_buf));
  86         }
  87
  88         close(s);
  89 }
  90
  91 #ifdef HAVE_IPV6
  92 static void test_send_recv_ipv6(void **state)
  93 {
  94         struct torture_address addr = {
  95                 .sa_socklen = sizeof(struct sockaddr_in6),
  96         };
  97         ssize_t ret;
  98         int rc;
  99         int i;
 100         int s;
 101
 102         (void) state; /* unused */
 103
 104         s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
 105         assert_int_not_equal(s, -1);
 106
 107         addr.sa.in6.sin6_family = AF_INET6;
 108         addr.sa.in6.sin6_port = htons(torture_server_port());
 109
 110         rc = inet_pton(AF_INET6,
 111                        torture_server_address(AF_INET6),
 112                        &addr.sa.in6.sin6_addr);
 113         assert_int_equal(rc, 1);
 114
 115         rc = connect(s, &addr.sa.s, addr.sa_socklen);
 116         assert_int_equal(rc, 0);
 117
 118         for (i = 0; i < 10; i++) {
 119                 char send_buf[64] = {0};
 120                 char recv_buf[64] = {0};
 121
 122                 snprintf(send_buf, sizeof(send_buf), "packet.%d", i);
 123
 124                 ret = send(s,
 125                            send_buf,
 126                            sizeof(send_buf),
 127                            0);
 128                 assert_int_not_equal(ret, -1);
 129
 130                 ret = recv(s,
 131                            recv_buf,
 132                            sizeof(recv_buf),
 133                            0);
 134                 assert_int_not_equal(ret, -1);
 135
 136                 assert_memory_equal(send_buf, recv_buf, sizeof(send_buf));
 137         }
 138
 139         close(s);
 140 }
 141 #endif
 142
 143 int main(void) {
 144         int rc;
 145
 146         const struct CMUnitTest send_tests[] = {
 147                 cmocka_unit_test_setup_teardown(test_send_recv_ipv4,
 148                                                 setup_echo_srv_udp_ipv4,
 149                                                 teardown),
 150 #ifdef HAVE_IPV6
 151                 cmocka_unit_test_setup_teardown(test_send_recv_ipv6,
 152                                                 setup_echo_srv_udp_ipv6,
 153                                                 teardown),
 154 #endif
 155         };
 156
 157         rc = cmocka_run_group_tests(send_tests, NULL, NULL);
 158
 159         return rc;
 160 }