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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <cmocka.h>
#include "config.h"
#include "torture.h"
#include <errno.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <stdio.h>
static void setup_echo_srv_udp_ipv4(void **state)
{
torture_setup_echo_srv_udp_ipv4(state);
}
static void setup_echo_srv_udp_ipv6(void **state)
{
torture_setup_echo_srv_udp_ipv6(state);
}
static void teardown(void **state)
{
torture_teardown_echo_srv(state);
}
static void test_sendto_recvfrom_ipv4(void **state)
{
struct sockaddr_in sin;
socklen_t slen = sizeof(struct sockaddr_in);
ssize_t ret;
int rc;
int i;
int s;
(void) state; /* unused */
s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
assert_int_not_equal(s, -1);
ZERO_STRUCT(sin);
sin.sin_family = AF_INET;
sin.sin_port = htons(TORTURE_ECHO_SRV_PORT);
rc = inet_aton(TORTURE_ECHO_SRV_IPV4, &sin.sin_addr);
assert_int_equal(rc, 1);
for (i = 0; i < 10; i++) {
char send_buf[64] = {0};
char recv_buf[64] = {0};
struct sockaddr_in cli_in;
socklen_t clen = sizeof(cli_in);
snprintf(send_buf, sizeof(send_buf), "packet.%d", i);
ret = sendto(s,
send_buf,
sizeof(send_buf),
0,
(struct sockaddr *)(void *)&sin,
slen);
assert_int_not_equal(ret, -1);
ZERO_STRUCT(cli_in);
ret = recvfrom(s,
recv_buf,
sizeof(recv_buf),
0,
(struct sockaddr *)&cli_in,
&clen);
assert_memory_equal(send_buf, recv_buf, sizeof(send_buf));
}
}
#ifdef HAVE_IPV6
static void test_sendto_recvfrom_ipv6(void **state)
{
struct sockaddr_in6 sin6;
socklen_t slen = sizeof(struct sockaddr_in6);
ssize_t ret;
int rc;
int i;
int s;
(void) state; /* unused */
s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
assert_int_not_equal(s, -1);
ZERO_STRUCT(sin6);
sin6.sin6_family = AF_INET6;
sin6.sin6_port = htons(TORTURE_ECHO_SRV_PORT);
rc = inet_pton(AF_INET6, TORTURE_ECHO_SRV_IPV6, &sin6.sin6_addr);
assert_int_equal(rc, 1);
for (i = 0; i < 10; i++) {
char send_buf[64] = {0};
char recv_buf[64] = {0};
struct sockaddr_in6 cli_in6;
socklen_t clen = sizeof(cli_in6);
snprintf(send_buf, sizeof(send_buf), "packet.%d", i);
ret = sendto(s,
send_buf,
sizeof(send_buf),
0,
(struct sockaddr *)(void *)&sin6,
slen);
assert_int_not_equal(ret, -1);
ZERO_STRUCT(cli_in6);
ret = recvfrom(s,
recv_buf,
sizeof(recv_buf),
0,
(struct sockaddr *)&cli_in6,
&clen);
assert_memory_equal(send_buf, recv_buf, sizeof(send_buf));
}
}
#endif
int main(void) {
int rc;
const UnitTest tests[] = {
unit_test_setup_teardown(test_sendto_recvfrom_ipv4, setup_echo_srv_udp_ipv4, teardown),
#ifdef HAVE_IPV6
unit_test_setup_teardown(test_sendto_recvfrom_ipv6, setup_echo_srv_udp_ipv6, teardown),
#endif
};
rc = run_tests(tests);
return rc;
}
|
