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
|
/*
* Nsock regression test suite
* Same license as nmap -- see https://nmap.org/book/man-legal.html
*/
#include "test-common.h"
struct connect_test_data {
nsock_pool nsp;
nsock_iod nsi;
int connect_result;
};
static void connect_handler(nsock_pool nsp, nsock_event nse, void *udata) {
struct connect_test_data *ctd;
ctd = (struct connect_test_data *)nsock_pool_get_udata(nsp);
switch(nse_status(nse)) {
case NSE_STATUS_SUCCESS:
ctd->connect_result = 0;
break;
case NSE_STATUS_ERROR:
ctd->connect_result = -(nse_errorcode(nse));
break;
case NSE_STATUS_TIMEOUT:
ctd->connect_result = -ETIMEDOUT;
break;
default:
ctd->connect_result = -EINVAL;
break;
}
}
static int connect_setup(void **tdata) {
struct connect_test_data *ctd;
ctd = calloc(1, sizeof(struct connect_test_data));
if (ctd == NULL)
return -ENOMEM;
ctd->nsp = nsock_pool_new(ctd);
AssertNonNull(ctd->nsp);
ctd->nsi = nsock_iod_new(ctd->nsp, NULL);
AssertNonNull(ctd->nsi);
*tdata = ctd;
return 0;
}
static int connect_teardown(void *tdata) {
struct connect_test_data *ctd = (struct connect_test_data *)tdata;
if (tdata) {
nsock_iod_delete(ctd->nsi, NSOCK_PENDING_SILENT); /* nsock_pool_delete would also handle it */
nsock_pool_delete(ctd->nsp);
free(tdata);
}
return 0;
}
static int connect_tcp(void *tdata) {
struct connect_test_data *ctd = (struct connect_test_data *)tdata;
struct sockaddr_in peer;
memset(&peer, 0, sizeof(peer));
peer.sin_family = AF_INET;
inet_aton("127.0.0.1", &peer.sin_addr);
nsock_connect_tcp(ctd->nsp, ctd->nsi, connect_handler, 4000, NULL,
(struct sockaddr *)&peer, sizeof(peer), PORT_TCP);
nsock_loop(ctd->nsp, 4000);
return ctd->connect_result;
}
static int connect_tcp_failure(void *tdata) {
struct connect_test_data *ctd = (struct connect_test_data *)tdata;
struct sockaddr_in peer;
memset(&peer, 0, sizeof(peer));
peer.sin_family = AF_INET;
inet_aton("127.0.0.1", &peer.sin_addr);
/* pass in addrlen == 0 to force connect(2) to fail */
nsock_connect_tcp(ctd->nsp, ctd->nsi, connect_handler, 4000, NULL,
(struct sockaddr *)&peer, 0, PORT_TCP);
nsock_loop(ctd->nsp, 4000);
AssertEqual(ctd->connect_result, -EINVAL);
return 0;
}
const struct test_case TestConnectTCP = {
.t_name = "simple tcp connection",
.t_setup = connect_setup,
.t_run = connect_tcp,
.t_teardown = connect_teardown
};
const struct test_case TestConnectFailure = {
.t_name = "tcp connection failure case",
.t_setup = connect_setup,
.t_run = connect_tcp_failure,
.t_teardown = connect_teardown
};
|
