Branch data Line data Source code
1 : : /* SPDX-License-Identifier: LGPL-2.1+ */
2 : : /***
3 : : Copyright © 2013 Intel Corporation. All rights reserved.
4 : : ***/
5 : :
6 : : #include <errno.h>
7 : :
8 : : #include "sd-dhcp-server.h"
9 : : #include "sd-event.h"
10 : :
11 : : #include "dhcp-server-internal.h"
12 : : #include "tests.h"
13 : :
14 : 8 : static void test_pool(struct in_addr *address, unsigned size, int ret) {
15 : 8 : _cleanup_(sd_dhcp_server_unrefp) sd_dhcp_server *server = NULL;
16 : :
17 [ - + ]: 8 : assert_se(sd_dhcp_server_new(&server, 1) >= 0);
18 : :
19 [ - + ]: 8 : assert_se(sd_dhcp_server_configure_pool(server, address, 8, 0, size) == ret);
20 : 8 : }
21 : :
22 : 4 : static int test_basic(sd_event *event) {
23 : 4 : _cleanup_(sd_dhcp_server_unrefp) sd_dhcp_server *server = NULL;
24 : 4 : struct in_addr address_lo = {
25 : 4 : .s_addr = htonl(INADDR_LOOPBACK),
26 : : };
27 : 4 : struct in_addr address_any = {
28 : 4 : .s_addr = htonl(INADDR_ANY),
29 : : };
30 : : int r;
31 : :
32 : : /* attach to loopback interface */
33 [ - + ]: 4 : assert_se(sd_dhcp_server_new(&server, 1) >= 0);
34 [ - + ]: 4 : assert_se(server);
35 : :
36 [ - + ]: 4 : assert_se(sd_dhcp_server_attach_event(server, event, 0) >= 0);
37 [ - + ]: 4 : assert_se(sd_dhcp_server_attach_event(server, event, 0) == -EBUSY);
38 [ - + ]: 4 : assert_se(sd_dhcp_server_get_event(server) == event);
39 [ - + ]: 4 : assert_se(sd_dhcp_server_detach_event(server) >= 0);
40 [ - + ]: 4 : assert_se(!sd_dhcp_server_get_event(server));
41 [ - + ]: 4 : assert_se(sd_dhcp_server_attach_event(server, NULL, 0) >= 0);
42 [ - + ]: 4 : assert_se(sd_dhcp_server_attach_event(server, NULL, 0) == -EBUSY);
43 : :
44 [ - + ]: 4 : assert_se(sd_dhcp_server_ref(server) == server);
45 [ - + ]: 4 : assert_se(!sd_dhcp_server_unref(server));
46 : :
47 [ - + ]: 4 : assert_se(sd_dhcp_server_start(server) == -EUNATCH);
48 : :
49 [ - + ]: 4 : assert_se(sd_dhcp_server_configure_pool(server, &address_any, 28, 0, 0) == -EINVAL);
50 [ - + ]: 4 : assert_se(sd_dhcp_server_configure_pool(server, &address_lo, 38, 0, 0) == -ERANGE);
51 [ - + ]: 4 : assert_se(sd_dhcp_server_configure_pool(server, &address_lo, 8, 0, 0) >= 0);
52 [ - + ]: 4 : assert_se(sd_dhcp_server_configure_pool(server, &address_lo, 8, 0, 0) >= 0);
53 : :
54 : 4 : test_pool(&address_any, 1, -EINVAL);
55 : 4 : test_pool(&address_lo, 1, 0);
56 : :
57 : 4 : r = sd_dhcp_server_start(server);
58 [ + - ]: 4 : if (r == -EPERM)
59 [ + - ]: 4 : return log_info_errno(r, "sd_dhcp_server_start failed: %m");
60 [ # # ]: 0 : assert_se(r >= 0);
61 : :
62 [ # # ]: 0 : assert_se(sd_dhcp_server_start(server) == -EBUSY);
63 [ # # ]: 0 : assert_se(sd_dhcp_server_stop(server) >= 0);
64 [ # # ]: 0 : assert_se(sd_dhcp_server_stop(server) >= 0);
65 [ # # ]: 0 : assert_se(sd_dhcp_server_start(server) >= 0);
66 : :
67 : 0 : return 0;
68 : : }
69 : :
70 : 0 : static void test_message_handler(void) {
71 : 0 : _cleanup_(sd_dhcp_server_unrefp) sd_dhcp_server *server = NULL;
72 : : struct {
73 : : DHCPMessage message;
74 : : struct {
75 : : uint8_t code;
76 : : uint8_t length;
77 : : uint8_t type;
78 : : } _packed_ option_type;
79 : : struct {
80 : : uint8_t code;
81 : : uint8_t length;
82 : : be32_t address;
83 : : } _packed_ option_requested_ip;
84 : : struct {
85 : : uint8_t code;
86 : : uint8_t length;
87 : : be32_t address;
88 : : } _packed_ option_server_id;
89 : : struct {
90 : : uint8_t code;
91 : : uint8_t length;
92 : : uint8_t id[7];
93 : : } _packed_ option_client_id;
94 : : uint8_t end;
95 : 0 : } _packed_ test = {
96 : : .message.op = BOOTREQUEST,
97 : : .message.htype = ARPHRD_ETHER,
98 : : .message.hlen = ETHER_ADDR_LEN,
99 : 0 : .message.xid = htobe32(0x12345678),
100 : : .message.chaddr = { 'A', 'B', 'C', 'D', 'E', 'F' },
101 : : .option_type.code = SD_DHCP_OPTION_MESSAGE_TYPE,
102 : : .option_type.length = 1,
103 : : .option_type.type = DHCP_DISCOVER,
104 : : .end = SD_DHCP_OPTION_END,
105 : : };
106 : 0 : struct in_addr address_lo = {
107 : 0 : .s_addr = htonl(INADDR_LOOPBACK),
108 : : };
109 : :
110 [ # # ]: 0 : assert_se(sd_dhcp_server_new(&server, 1) >= 0);
111 [ # # ]: 0 : assert_se(sd_dhcp_server_configure_pool(server, &address_lo, 8, 0, 0) >= 0);
112 [ # # ]: 0 : assert_se(sd_dhcp_server_attach_event(server, NULL, 0) >= 0);
113 [ # # ]: 0 : assert_se(sd_dhcp_server_start(server) >= 0);
114 : :
115 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
116 : :
117 : 0 : test.end = 0;
118 : : /* TODO, shouldn't this fail? */
119 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
120 : 0 : test.end = SD_DHCP_OPTION_END;
121 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
122 : :
123 : 0 : test.option_type.code = 0;
124 : 0 : test.option_type.length = 0;
125 : 0 : test.option_type.type = 0;
126 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
127 : 0 : test.option_type.code = SD_DHCP_OPTION_MESSAGE_TYPE;
128 : 0 : test.option_type.length = 1;
129 : 0 : test.option_type.type = DHCP_DISCOVER;
130 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
131 : :
132 : 0 : test.message.op = 0;
133 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
134 : 0 : test.message.op = BOOTREQUEST;
135 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
136 : :
137 : 0 : test.message.htype = 0;
138 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
139 : 0 : test.message.htype = ARPHRD_ETHER;
140 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
141 : :
142 : 0 : test.message.hlen = 0;
143 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
144 : 0 : test.message.hlen = ETHER_ADDR_LEN;
145 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
146 : :
147 : 0 : test.option_type.type = DHCP_REQUEST;
148 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
149 : 0 : test.option_requested_ip.code = SD_DHCP_OPTION_REQUESTED_IP_ADDRESS;
150 : 0 : test.option_requested_ip.length = 4;
151 : 0 : test.option_requested_ip.address = htobe32(0x12345678);
152 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_NAK);
153 : 0 : test.option_server_id.code = SD_DHCP_OPTION_SERVER_IDENTIFIER;
154 : 0 : test.option_server_id.length = 4;
155 : 0 : test.option_server_id.address = htobe32(INADDR_LOOPBACK);
156 : 0 : test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 3);
157 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_ACK);
158 : :
159 : 0 : test.option_server_id.address = htobe32(0x12345678);
160 : 0 : test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 3);
161 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
162 : 0 : test.option_server_id.address = htobe32(INADDR_LOOPBACK);
163 : 0 : test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 4);
164 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
165 : 0 : test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 3);
166 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_ACK);
167 : :
168 : 0 : test.option_client_id.code = SD_DHCP_OPTION_CLIENT_IDENTIFIER;
169 : 0 : test.option_client_id.length = 7;
170 : 0 : test.option_client_id.id[0] = 0x01;
171 : 0 : test.option_client_id.id[1] = 'A';
172 : 0 : test.option_client_id.id[2] = 'B';
173 : 0 : test.option_client_id.id[3] = 'C';
174 : 0 : test.option_client_id.id[4] = 'D';
175 : 0 : test.option_client_id.id[5] = 'E';
176 : 0 : test.option_client_id.id[6] = 'F';
177 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_ACK);
178 : :
179 : 0 : test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 30);
180 [ # # ]: 0 : assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
181 : 0 : }
182 : :
183 : 0 : static uint64_t client_id_hash_helper(DHCPClientId *id, uint8_t key[HASH_KEY_SIZE]) {
184 : : struct siphash state;
185 : :
186 : 0 : siphash24_init(&state, key);
187 : 0 : client_id_hash_func(id, &state);
188 : :
189 : 0 : return htole64(siphash24_finalize(&state));
190 : : }
191 : :
192 : 0 : static void test_client_id_hash(void) {
193 : 0 : DHCPClientId a = {
194 : : .length = 4,
195 : 0 : }, b = {
196 : : .length = 4,
197 : : };
198 : 0 : uint8_t hash_key[HASH_KEY_SIZE] = {
199 : : '0', '1', '2', '3', '4', '5', '6', '7',
200 : : '8', '9', 'A', 'B', 'C', 'D', 'E', 'F',
201 : : };
202 : :
203 : 0 : a.data = (uint8_t*)strdup("abcd");
204 : 0 : b.data = (uint8_t*)strdup("abcd");
205 : :
206 [ # # ]: 0 : assert_se(client_id_compare_func(&a, &b) == 0);
207 [ # # ]: 0 : assert_se(client_id_hash_helper(&a, hash_key) == client_id_hash_helper(&b, hash_key));
208 : 0 : a.length = 3;
209 [ # # ]: 0 : assert_se(client_id_compare_func(&a, &b) != 0);
210 : 0 : a.length = 4;
211 [ # # ]: 0 : assert_se(client_id_compare_func(&a, &b) == 0);
212 [ # # ]: 0 : assert_se(client_id_hash_helper(&a, hash_key) == client_id_hash_helper(&b, hash_key));
213 : :
214 : 0 : b.length = 3;
215 [ # # ]: 0 : assert_se(client_id_compare_func(&a, &b) != 0);
216 : 0 : b.length = 4;
217 [ # # ]: 0 : assert_se(client_id_compare_func(&a, &b) == 0);
218 [ # # ]: 0 : assert_se(client_id_hash_helper(&a, hash_key) == client_id_hash_helper(&b, hash_key));
219 : :
220 : 0 : free(b.data);
221 : 0 : b.data = (uint8_t*)strdup("abce");
222 [ # # ]: 0 : assert_se(client_id_compare_func(&a, &b) != 0);
223 : :
224 : 0 : free(a.data);
225 : 0 : free(b.data);
226 : 0 : }
227 : :
228 : 4 : int main(int argc, char *argv[]) {
229 : 4 : _cleanup_(sd_event_unrefp) sd_event *e;
230 : : int r;
231 : :
232 : 4 : test_setup_logging(LOG_DEBUG);
233 : :
234 [ - + ]: 4 : assert_se(sd_event_new(&e) >= 0);
235 : :
236 : 4 : r = test_basic(e);
237 [ + - ]: 4 : if (r != 0)
238 : 4 : return log_tests_skipped("cannot start dhcp server");
239 : :
240 : 0 : test_message_handler();
241 : 0 : test_client_id_hash();
242 : :
243 : 0 : return 0;
244 : : }
|