1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright (C) 2013 Intel Corporation. All rights reserved.
7 Copyright (C) 2014 Tom Gundersen
9 systemd is free software; you can redistribute it and/or modify it
10 under the terms of the GNU Lesser General Public License as published by
11 the Free Software Foundation; either version 2.1 of the License, or
12 (at your option) any later version.
14 systemd is distributed in the hope that it will be useful, but
15 WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 Lesser General Public License for more details.
19 You should have received a copy of the GNU Lesser General Public License
20 along with systemd; If not, see <http://www.gnu.org/licenses/>.
25 #include "sd-dhcp-server.h"
28 #include "dhcp-server-internal.h"
29 #include "event-util.h"
31 static void test_pool(struct in_addr
*address
, unsigned size
, int ret
) {
32 _cleanup_dhcp_server_unref_ sd_dhcp_server
*server
= NULL
;
34 assert_se(sd_dhcp_server_new(&server
, 1) >= 0);
36 assert_se(sd_dhcp_server_configure_pool(server
, address
, 8, 0, size
) == ret
);
39 static int test_basic(sd_event
*event
) {
40 _cleanup_dhcp_server_unref_ sd_dhcp_server
*server
= NULL
;
41 struct in_addr address_lo
= {
42 .s_addr
= htonl(INADDR_LOOPBACK
),
44 struct in_addr address_any
= {
45 .s_addr
= htonl(INADDR_ANY
),
49 /* attach to loopback interface */
50 assert_se(sd_dhcp_server_new(&server
, 1) >= 0);
53 assert_se(sd_dhcp_server_attach_event(server
, event
, 0) >= 0);
54 assert_se(sd_dhcp_server_attach_event(server
, event
, 0) == -EBUSY
);
55 assert_se(sd_dhcp_server_get_event(server
) == event
);
56 assert_se(sd_dhcp_server_detach_event(server
) >= 0);
57 assert_se(!sd_dhcp_server_get_event(server
));
58 assert_se(sd_dhcp_server_attach_event(server
, NULL
, 0) >= 0);
59 assert_se(sd_dhcp_server_attach_event(server
, NULL
, 0) == -EBUSY
);
61 assert_se(sd_dhcp_server_ref(server
) == server
);
62 assert_se(!sd_dhcp_server_unref(server
));
64 assert_se(sd_dhcp_server_start(server
) == -EUNATCH
);
66 assert_se(sd_dhcp_server_configure_pool(server
, &address_any
, 28, 0, 0) == -EINVAL
);
67 assert_se(sd_dhcp_server_configure_pool(server
, &address_lo
, 38, 0, 0) == -ERANGE
);
68 assert_se(sd_dhcp_server_configure_pool(server
, &address_lo
, 8, 0, 0) >= 0);
69 assert_se(sd_dhcp_server_configure_pool(server
, &address_lo
, 8, 0, 0) == -EBUSY
);
71 test_pool(&address_any
, 1, -EINVAL
);
72 test_pool(&address_lo
, 1, 0);
74 r
= sd_dhcp_server_start(server
);
77 return EXIT_TEST_SKIP
;
80 assert_se(sd_dhcp_server_start(server
) == -EBUSY
);
81 assert_se(sd_dhcp_server_stop(server
) >= 0);
82 assert_se(sd_dhcp_server_stop(server
) >= 0);
83 assert_se(sd_dhcp_server_start(server
) >= 0);
88 static void test_message_handler(void) {
89 _cleanup_dhcp_server_unref_ sd_dhcp_server
*server
= NULL
;
96 } _packed_ option_type
;
101 } _packed_ option_requested_ip
;
106 } _packed_ option_server_id
;
111 } _packed_ option_client_id
;
114 .message
.op
= BOOTREQUEST
,
115 .message
.htype
= ARPHRD_ETHER
,
116 .message
.hlen
= ETHER_ADDR_LEN
,
117 .message
.xid
= htobe32(0x12345678),
118 .message
.chaddr
= { 'A', 'B', 'C', 'D', 'E', 'F' },
119 .option_type
.code
= DHCP_OPTION_MESSAGE_TYPE
,
120 .option_type
.length
= 1,
121 .option_type
.type
= DHCP_DISCOVER
,
122 .end
= DHCP_OPTION_END
,
124 struct in_addr address_lo
= {
125 .s_addr
= htonl(INADDR_LOOPBACK
),
128 assert_se(sd_dhcp_server_new(&server
, 1) >= 0);
129 assert_se(sd_dhcp_server_configure_pool(server
, &address_lo
, 8, 0, 0) >= 0);
130 assert_se(sd_dhcp_server_attach_event(server
, NULL
, 0) >= 0);
131 assert_se(sd_dhcp_server_start(server
) >= 0);
133 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_OFFER
);
136 /* TODO, shouldn't this fail? */
137 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_OFFER
);
138 test
.end
= DHCP_OPTION_END
;
139 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_OFFER
);
141 test
.option_type
.code
= 0;
142 test
.option_type
.length
= 0;
143 test
.option_type
.type
= 0;
144 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == 0);
145 test
.option_type
.code
= DHCP_OPTION_MESSAGE_TYPE
;
146 test
.option_type
.length
= 1;
147 test
.option_type
.type
= DHCP_DISCOVER
;
148 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_OFFER
);
151 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == 0);
152 test
.message
.op
= BOOTREQUEST
;
153 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_OFFER
);
155 test
.message
.htype
= 0;
156 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == 0);
157 test
.message
.htype
= ARPHRD_ETHER
;
158 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_OFFER
);
160 test
.message
.hlen
= 0;
161 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == 0);
162 test
.message
.hlen
= ETHER_ADDR_LEN
;
163 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_OFFER
);
165 test
.option_type
.type
= DHCP_REQUEST
;
166 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == 0);
167 test
.option_requested_ip
.code
= DHCP_OPTION_REQUESTED_IP_ADDRESS
;
168 test
.option_requested_ip
.length
= 4;
169 test
.option_requested_ip
.address
= htobe32(0x12345678);
170 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_NAK
);
171 test
.option_server_id
.code
= DHCP_OPTION_SERVER_IDENTIFIER
;
172 test
.option_server_id
.length
= 4;
173 test
.option_server_id
.address
= htobe32(INADDR_LOOPBACK
);
174 test
.option_requested_ip
.address
= htobe32(INADDR_LOOPBACK
+ 3);
175 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_ACK
);
177 test
.option_server_id
.address
= htobe32(0x12345678);
178 test
.option_requested_ip
.address
= htobe32(INADDR_LOOPBACK
+ 3);
179 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == 0);
180 test
.option_server_id
.address
= htobe32(INADDR_LOOPBACK
);
181 test
.option_requested_ip
.address
= htobe32(INADDR_LOOPBACK
+ 4);
182 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == 0);
183 test
.option_requested_ip
.address
= htobe32(INADDR_LOOPBACK
+ 3);
184 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_ACK
);
186 test
.option_client_id
.code
= DHCP_OPTION_CLIENT_IDENTIFIER
;
187 test
.option_client_id
.length
= 7;
188 test
.option_client_id
.id
[0] = 0x01;
189 test
.option_client_id
.id
[1] = 'A';
190 test
.option_client_id
.id
[2] = 'B';
191 test
.option_client_id
.id
[3] = 'C';
192 test
.option_client_id
.id
[4] = 'D';
193 test
.option_client_id
.id
[5] = 'E';
194 test
.option_client_id
.id
[6] = 'F';
195 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == DHCP_ACK
);
197 test
.option_requested_ip
.address
= htobe32(INADDR_LOOPBACK
+ 30);
198 assert_se(dhcp_server_handle_message(server
, (DHCPMessage
*)&test
, sizeof(test
)) == 0);
201 static uint64_t client_id_hash_helper(DHCPClientId
*id
, uint8_t key
[HASH_KEY_SIZE
]) {
202 struct siphash state
;
204 siphash24_init(&state
, key
);
205 client_id_hash_func(id
, &state
);
207 return htole64(siphash24_finalize(&state
));
210 static void test_client_id_hash(void) {
216 uint8_t hash_key
[HASH_KEY_SIZE
] = {
217 '0', '1', '2', '3', '4', '5', '6', '7',
218 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F',
221 a
.data
= (uint8_t*)strdup("abcd");
222 b
.data
= (uint8_t*)strdup("abcd");
224 assert_se(client_id_compare_func(&a
, &b
) == 0);
225 assert_se(client_id_hash_helper(&a
, hash_key
) == client_id_hash_helper(&b
, hash_key
));
227 assert_se(client_id_compare_func(&a
, &b
) != 0);
229 assert_se(client_id_compare_func(&a
, &b
) == 0);
230 assert_se(client_id_hash_helper(&a
, hash_key
) == client_id_hash_helper(&b
, hash_key
));
233 assert_se(client_id_compare_func(&a
, &b
) != 0);
235 assert_se(client_id_compare_func(&a
, &b
) == 0);
236 assert_se(client_id_hash_helper(&a
, hash_key
) == client_id_hash_helper(&b
, hash_key
));
239 b
.data
= (uint8_t*)strdup("abce");
240 assert_se(client_id_compare_func(&a
, &b
) != 0);
246 int main(int argc
, char *argv
[]) {
247 _cleanup_event_unref_ sd_event
*e
;
250 log_set_max_level(LOG_DEBUG
);
251 log_parse_environment();
254 assert_se(sd_event_new(&e
) >= 0);
260 test_message_handler();
261 test_client_id_hash();