6d7c403324
The advantages are that we save a few lines, and that we can override logging using environment variables in more test executables.
245 lines
9.7 KiB
C
245 lines
9.7 KiB
C
/* SPDX-License-Identifier: LGPL-2.1+ */
|
|
/***
|
|
Copyright © 2013 Intel Corporation. All rights reserved.
|
|
***/
|
|
|
|
#include <errno.h>
|
|
|
|
#include "sd-dhcp-server.h"
|
|
#include "sd-event.h"
|
|
|
|
#include "dhcp-server-internal.h"
|
|
#include "tests.h"
|
|
|
|
static void test_pool(struct in_addr *address, unsigned size, int ret) {
|
|
_cleanup_(sd_dhcp_server_unrefp) sd_dhcp_server *server = NULL;
|
|
|
|
assert_se(sd_dhcp_server_new(&server, 1) >= 0);
|
|
|
|
assert_se(sd_dhcp_server_configure_pool(server, address, 8, 0, size) == ret);
|
|
}
|
|
|
|
static int test_basic(sd_event *event) {
|
|
_cleanup_(sd_dhcp_server_unrefp) sd_dhcp_server *server = NULL;
|
|
struct in_addr address_lo = {
|
|
.s_addr = htonl(INADDR_LOOPBACK),
|
|
};
|
|
struct in_addr address_any = {
|
|
.s_addr = htonl(INADDR_ANY),
|
|
};
|
|
int r;
|
|
|
|
/* attach to loopback interface */
|
|
assert_se(sd_dhcp_server_new(&server, 1) >= 0);
|
|
assert_se(server);
|
|
|
|
assert_se(sd_dhcp_server_attach_event(server, event, 0) >= 0);
|
|
assert_se(sd_dhcp_server_attach_event(server, event, 0) == -EBUSY);
|
|
assert_se(sd_dhcp_server_get_event(server) == event);
|
|
assert_se(sd_dhcp_server_detach_event(server) >= 0);
|
|
assert_se(!sd_dhcp_server_get_event(server));
|
|
assert_se(sd_dhcp_server_attach_event(server, NULL, 0) >= 0);
|
|
assert_se(sd_dhcp_server_attach_event(server, NULL, 0) == -EBUSY);
|
|
|
|
assert_se(sd_dhcp_server_ref(server) == server);
|
|
assert_se(!sd_dhcp_server_unref(server));
|
|
|
|
assert_se(sd_dhcp_server_start(server) == -EUNATCH);
|
|
|
|
assert_se(sd_dhcp_server_configure_pool(server, &address_any, 28, 0, 0) == -EINVAL);
|
|
assert_se(sd_dhcp_server_configure_pool(server, &address_lo, 38, 0, 0) == -ERANGE);
|
|
assert_se(sd_dhcp_server_configure_pool(server, &address_lo, 8, 0, 0) >= 0);
|
|
assert_se(sd_dhcp_server_configure_pool(server, &address_lo, 8, 0, 0) >= 0);
|
|
|
|
test_pool(&address_any, 1, -EINVAL);
|
|
test_pool(&address_lo, 1, 0);
|
|
|
|
r = sd_dhcp_server_start(server);
|
|
if (r == -EPERM)
|
|
return log_info_errno(r, "sd_dhcp_server_start failed: %m");
|
|
assert_se(r >= 0);
|
|
|
|
assert_se(sd_dhcp_server_start(server) == -EBUSY);
|
|
assert_se(sd_dhcp_server_stop(server) >= 0);
|
|
assert_se(sd_dhcp_server_stop(server) >= 0);
|
|
assert_se(sd_dhcp_server_start(server) >= 0);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void test_message_handler(void) {
|
|
_cleanup_(sd_dhcp_server_unrefp) sd_dhcp_server *server = NULL;
|
|
struct {
|
|
DHCPMessage message;
|
|
struct {
|
|
uint8_t code;
|
|
uint8_t length;
|
|
uint8_t type;
|
|
} _packed_ option_type;
|
|
struct {
|
|
uint8_t code;
|
|
uint8_t length;
|
|
be32_t address;
|
|
} _packed_ option_requested_ip;
|
|
struct {
|
|
uint8_t code;
|
|
uint8_t length;
|
|
be32_t address;
|
|
} _packed_ option_server_id;
|
|
struct {
|
|
uint8_t code;
|
|
uint8_t length;
|
|
uint8_t id[7];
|
|
} _packed_ option_client_id;
|
|
uint8_t end;
|
|
} _packed_ test = {
|
|
.message.op = BOOTREQUEST,
|
|
.message.htype = ARPHRD_ETHER,
|
|
.message.hlen = ETHER_ADDR_LEN,
|
|
.message.xid = htobe32(0x12345678),
|
|
.message.chaddr = { 'A', 'B', 'C', 'D', 'E', 'F' },
|
|
.option_type.code = SD_DHCP_OPTION_MESSAGE_TYPE,
|
|
.option_type.length = 1,
|
|
.option_type.type = DHCP_DISCOVER,
|
|
.end = SD_DHCP_OPTION_END,
|
|
};
|
|
struct in_addr address_lo = {
|
|
.s_addr = htonl(INADDR_LOOPBACK),
|
|
};
|
|
|
|
assert_se(sd_dhcp_server_new(&server, 1) >= 0);
|
|
assert_se(sd_dhcp_server_configure_pool(server, &address_lo, 8, 0, 0) >= 0);
|
|
assert_se(sd_dhcp_server_attach_event(server, NULL, 0) >= 0);
|
|
assert_se(sd_dhcp_server_start(server) >= 0);
|
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
|
|
|
test.end = 0;
|
|
/* TODO, shouldn't this fail? */
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
|
test.end = SD_DHCP_OPTION_END;
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
|
|
|
test.option_type.code = 0;
|
|
test.option_type.length = 0;
|
|
test.option_type.type = 0;
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
test.option_type.code = SD_DHCP_OPTION_MESSAGE_TYPE;
|
|
test.option_type.length = 1;
|
|
test.option_type.type = DHCP_DISCOVER;
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
|
|
|
test.message.op = 0;
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
test.message.op = BOOTREQUEST;
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
|
|
|
test.message.htype = 0;
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
test.message.htype = ARPHRD_ETHER;
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
|
|
|
test.message.hlen = 0;
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
test.message.hlen = ETHER_ADDR_LEN;
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
|
|
|
test.option_type.type = DHCP_REQUEST;
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
test.option_requested_ip.code = SD_DHCP_OPTION_REQUESTED_IP_ADDRESS;
|
|
test.option_requested_ip.length = 4;
|
|
test.option_requested_ip.address = htobe32(0x12345678);
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_NAK);
|
|
test.option_server_id.code = SD_DHCP_OPTION_SERVER_IDENTIFIER;
|
|
test.option_server_id.length = 4;
|
|
test.option_server_id.address = htobe32(INADDR_LOOPBACK);
|
|
test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 3);
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_ACK);
|
|
|
|
test.option_server_id.address = htobe32(0x12345678);
|
|
test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 3);
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
test.option_server_id.address = htobe32(INADDR_LOOPBACK);
|
|
test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 4);
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 3);
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_ACK);
|
|
|
|
test.option_client_id.code = SD_DHCP_OPTION_CLIENT_IDENTIFIER;
|
|
test.option_client_id.length = 7;
|
|
test.option_client_id.id[0] = 0x01;
|
|
test.option_client_id.id[1] = 'A';
|
|
test.option_client_id.id[2] = 'B';
|
|
test.option_client_id.id[3] = 'C';
|
|
test.option_client_id.id[4] = 'D';
|
|
test.option_client_id.id[5] = 'E';
|
|
test.option_client_id.id[6] = 'F';
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_ACK);
|
|
|
|
test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 30);
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
}
|
|
|
|
static uint64_t client_id_hash_helper(DHCPClientId *id, uint8_t key[HASH_KEY_SIZE]) {
|
|
struct siphash state;
|
|
|
|
siphash24_init(&state, key);
|
|
client_id_hash_func(id, &state);
|
|
|
|
return htole64(siphash24_finalize(&state));
|
|
}
|
|
|
|
static void test_client_id_hash(void) {
|
|
DHCPClientId a = {
|
|
.length = 4,
|
|
}, b = {
|
|
.length = 4,
|
|
};
|
|
uint8_t hash_key[HASH_KEY_SIZE] = {
|
|
'0', '1', '2', '3', '4', '5', '6', '7',
|
|
'8', '9', 'A', 'B', 'C', 'D', 'E', 'F',
|
|
};
|
|
|
|
a.data = (uint8_t*)strdup("abcd");
|
|
b.data = (uint8_t*)strdup("abcd");
|
|
|
|
assert_se(client_id_compare_func(&a, &b) == 0);
|
|
assert_se(client_id_hash_helper(&a, hash_key) == client_id_hash_helper(&b, hash_key));
|
|
a.length = 3;
|
|
assert_se(client_id_compare_func(&a, &b) != 0);
|
|
a.length = 4;
|
|
assert_se(client_id_compare_func(&a, &b) == 0);
|
|
assert_se(client_id_hash_helper(&a, hash_key) == client_id_hash_helper(&b, hash_key));
|
|
|
|
b.length = 3;
|
|
assert_se(client_id_compare_func(&a, &b) != 0);
|
|
b.length = 4;
|
|
assert_se(client_id_compare_func(&a, &b) == 0);
|
|
assert_se(client_id_hash_helper(&a, hash_key) == client_id_hash_helper(&b, hash_key));
|
|
|
|
free(b.data);
|
|
b.data = (uint8_t*)strdup("abce");
|
|
assert_se(client_id_compare_func(&a, &b) != 0);
|
|
|
|
free(a.data);
|
|
free(b.data);
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
_cleanup_(sd_event_unrefp) sd_event *e;
|
|
int r;
|
|
|
|
test_setup_logging(LOG_DEBUG);
|
|
|
|
assert_se(sd_event_new(&e) >= 0);
|
|
|
|
r = test_basic(e);
|
|
if (r != 0)
|
|
return log_tests_skipped("cannot start dhcp server");
|
|
|
|
test_message_handler();
|
|
test_client_id_hash();
|
|
|
|
return 0;
|
|
}
|