2017-11-18 17:09:20 +01:00
|
|
|
/* SPDX-License-Identifier: LGPL-2.1+ */
|
2014-05-24 18:48:41 +02:00
|
|
|
/***
|
2018-06-12 17:15:23 +02:00
|
|
|
Copyright © 2013 Intel Corporation. All rights reserved.
|
2014-05-24 18:48:41 +02:00
|
|
|
***/
|
|
|
|
|
|
|
|
#include <errno.h>
|
|
|
|
|
2015-10-24 22:58:24 +02:00
|
|
|
#include "sd-dhcp-server.h"
|
2014-05-24 18:48:41 +02:00
|
|
|
#include "sd-event.h"
|
|
|
|
|
|
|
|
#include "dhcp-server-internal.h"
|
2018-09-13 13:34:12 +02:00
|
|
|
#include "tests.h"
|
2014-05-24 18:48:41 +02:00
|
|
|
|
2015-08-28 20:29:10 +02:00
|
|
|
static void test_pool(struct in_addr *address, unsigned size, int ret) {
|
tree-wide: expose "p"-suffix unref calls in public APIs to make gcc cleanup easy
GLIB has recently started to officially support the gcc cleanup
attribute in its public API, hence let's do the same for our APIs.
With this patch we'll define an xyz_unrefp() call for each public
xyz_unref() call, to make it easy to use inside a
__attribute__((cleanup())) expression. Then, all code is ported over to
make use of this.
The new calls are also documented in the man pages, with examples how to
use them (well, I only added docs where the _unref() call itself already
had docs, and the examples, only cover sd_bus_unrefp() and
sd_event_unrefp()).
This also renames sd_lldp_free() to sd_lldp_unref(), since that's how we
tend to call our destructors these days.
Note that this defines no public macro that wraps gcc's attribute and
makes it easier to use. While I think it's our duty in the library to
make our stuff easy to use, I figure it's not our duty to make gcc's own
features easy to use on its own. Most likely, client code which wants to
make use of this should define its own:
#define _cleanup_(function) __attribute__((cleanup(function)))
Or similar, to make the gcc feature easier to use.
Making this logic public has the benefit that we can remove three header
files whose only purpose was to define these functions internally.
See #2008.
2015-11-27 19:13:45 +01:00
|
|
|
_cleanup_(sd_dhcp_server_unrefp) sd_dhcp_server *server = NULL;
|
2015-08-28 20:29:10 +02:00
|
|
|
|
|
|
|
assert_se(sd_dhcp_server_new(&server, 1) >= 0);
|
|
|
|
|
|
|
|
assert_se(sd_dhcp_server_configure_pool(server, address, 8, 0, size) == ret);
|
|
|
|
}
|
|
|
|
|
2014-06-13 19:11:31 +02:00
|
|
|
static int test_basic(sd_event *event) {
|
tree-wide: expose "p"-suffix unref calls in public APIs to make gcc cleanup easy
GLIB has recently started to officially support the gcc cleanup
attribute in its public API, hence let's do the same for our APIs.
With this patch we'll define an xyz_unrefp() call for each public
xyz_unref() call, to make it easy to use inside a
__attribute__((cleanup())) expression. Then, all code is ported over to
make use of this.
The new calls are also documented in the man pages, with examples how to
use them (well, I only added docs where the _unref() call itself already
had docs, and the examples, only cover sd_bus_unrefp() and
sd_event_unrefp()).
This also renames sd_lldp_free() to sd_lldp_unref(), since that's how we
tend to call our destructors these days.
Note that this defines no public macro that wraps gcc's attribute and
makes it easier to use. While I think it's our duty in the library to
make our stuff easy to use, I figure it's not our duty to make gcc's own
features easy to use on its own. Most likely, client code which wants to
make use of this should define its own:
#define _cleanup_(function) __attribute__((cleanup(function)))
Or similar, to make the gcc feature easier to use.
Making this logic public has the benefit that we can remove three header
files whose only purpose was to define these functions internally.
See #2008.
2015-11-27 19:13:45 +01:00
|
|
|
_cleanup_(sd_dhcp_server_unrefp) sd_dhcp_server *server = NULL;
|
2014-05-25 00:29:13 +02:00
|
|
|
struct in_addr address_lo = {
|
|
|
|
.s_addr = htonl(INADDR_LOOPBACK),
|
|
|
|
};
|
|
|
|
struct in_addr address_any = {
|
|
|
|
.s_addr = htonl(INADDR_ANY),
|
|
|
|
};
|
2014-06-13 19:11:31 +02:00
|
|
|
int r;
|
2014-05-24 18:48:41 +02:00
|
|
|
|
2014-05-24 19:38:17 +02:00
|
|
|
/* attach to loopback interface */
|
|
|
|
assert_se(sd_dhcp_server_new(&server, 1) >= 0);
|
2014-05-24 18:48:41 +02:00
|
|
|
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));
|
2014-05-24 19:27:20 +02:00
|
|
|
|
2014-05-25 00:29:13 +02:00
|
|
|
assert_se(sd_dhcp_server_start(server) == -EUNATCH);
|
|
|
|
|
2015-08-28 20:29:10 +02:00
|
|
|
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);
|
2017-06-09 17:18:25 +02:00
|
|
|
assert_se(sd_dhcp_server_configure_pool(server, &address_lo, 8, 0, 0) >= 0);
|
2015-08-28 20:29:10 +02:00
|
|
|
|
|
|
|
test_pool(&address_any, 1, -EINVAL);
|
|
|
|
test_pool(&address_lo, 1, 0);
|
2014-05-25 18:28:03 +02:00
|
|
|
|
2014-06-13 19:11:31 +02:00
|
|
|
r = sd_dhcp_server_start(server);
|
|
|
|
if (r == -EPERM)
|
2018-09-12 13:25:45 +02:00
|
|
|
return log_info_errno(r, "sd_dhcp_server_start failed: %m");
|
2014-06-13 19:11:31 +02:00
|
|
|
assert_se(r >= 0);
|
|
|
|
|
2014-05-24 19:27:20 +02:00
|
|
|
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);
|
2014-06-13 19:11:31 +02:00
|
|
|
|
|
|
|
return 0;
|
2014-05-24 18:48:41 +02:00
|
|
|
}
|
|
|
|
|
2014-05-24 21:04:27 +02:00
|
|
|
static void test_message_handler(void) {
|
tree-wide: expose "p"-suffix unref calls in public APIs to make gcc cleanup easy
GLIB has recently started to officially support the gcc cleanup
attribute in its public API, hence let's do the same for our APIs.
With this patch we'll define an xyz_unrefp() call for each public
xyz_unref() call, to make it easy to use inside a
__attribute__((cleanup())) expression. Then, all code is ported over to
make use of this.
The new calls are also documented in the man pages, with examples how to
use them (well, I only added docs where the _unref() call itself already
had docs, and the examples, only cover sd_bus_unrefp() and
sd_event_unrefp()).
This also renames sd_lldp_free() to sd_lldp_unref(), since that's how we
tend to call our destructors these days.
Note that this defines no public macro that wraps gcc's attribute and
makes it easier to use. While I think it's our duty in the library to
make our stuff easy to use, I figure it's not our duty to make gcc's own
features easy to use on its own. Most likely, client code which wants to
make use of this should define its own:
#define _cleanup_(function) __attribute__((cleanup(function)))
Or similar, to make the gcc feature easier to use.
Making this logic public has the benefit that we can remove three header
files whose only purpose was to define these functions internally.
See #2008.
2015-11-27 19:13:45 +01:00
|
|
|
_cleanup_(sd_dhcp_server_unrefp) sd_dhcp_server *server = NULL;
|
2014-05-24 21:04:27 +02:00
|
|
|
struct {
|
|
|
|
DHCPMessage message;
|
|
|
|
struct {
|
|
|
|
uint8_t code;
|
|
|
|
uint8_t length;
|
|
|
|
uint8_t type;
|
|
|
|
} _packed_ option_type;
|
2014-05-25 18:28:03 +02:00
|
|
|
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;
|
2014-05-25 22:07:53 +02:00
|
|
|
struct {
|
|
|
|
uint8_t code;
|
|
|
|
uint8_t length;
|
|
|
|
uint8_t id[7];
|
|
|
|
} _packed_ option_client_id;
|
2014-05-24 21:04:27 +02:00
|
|
|
uint8_t end;
|
|
|
|
} _packed_ test = {
|
|
|
|
.message.op = BOOTREQUEST,
|
|
|
|
.message.htype = ARPHRD_ETHER,
|
|
|
|
.message.hlen = ETHER_ADDR_LEN,
|
2014-05-25 18:28:03 +02:00
|
|
|
.message.xid = htobe32(0x12345678),
|
|
|
|
.message.chaddr = { 'A', 'B', 'C', 'D', 'E', 'F' },
|
2016-01-20 14:44:24 +01:00
|
|
|
.option_type.code = SD_DHCP_OPTION_MESSAGE_TYPE,
|
2014-05-24 21:04:27 +02:00
|
|
|
.option_type.length = 1,
|
|
|
|
.option_type.type = DHCP_DISCOVER,
|
2016-01-20 14:44:24 +01:00
|
|
|
.end = SD_DHCP_OPTION_END,
|
2014-05-24 21:04:27 +02:00
|
|
|
};
|
2014-05-25 00:29:13 +02:00
|
|
|
struct in_addr address_lo = {
|
|
|
|
.s_addr = htonl(INADDR_LOOPBACK),
|
|
|
|
};
|
2014-05-24 21:04:27 +02:00
|
|
|
|
|
|
|
assert_se(sd_dhcp_server_new(&server, 1) >= 0);
|
2015-08-28 20:29:10 +02:00
|
|
|
assert_se(sd_dhcp_server_configure_pool(server, &address_lo, 8, 0, 0) >= 0);
|
2014-05-25 00:29:13 +02:00
|
|
|
assert_se(sd_dhcp_server_attach_event(server, NULL, 0) >= 0);
|
|
|
|
assert_se(sd_dhcp_server_start(server) >= 0);
|
2014-05-24 21:04:27 +02:00
|
|
|
|
2014-05-25 17:31:17 +02:00
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
2014-05-24 21:04:27 +02:00
|
|
|
|
|
|
|
test.end = 0;
|
|
|
|
/* TODO, shouldn't this fail? */
|
2014-05-25 17:31:17 +02:00
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
2016-01-20 14:44:24 +01:00
|
|
|
test.end = SD_DHCP_OPTION_END;
|
2014-05-25 17:31:17 +02:00
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
2014-05-24 21:04:27 +02:00
|
|
|
|
|
|
|
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);
|
2016-01-20 14:44:24 +01:00
|
|
|
test.option_type.code = SD_DHCP_OPTION_MESSAGE_TYPE;
|
2014-05-24 21:04:27 +02:00
|
|
|
test.option_type.length = 1;
|
|
|
|
test.option_type.type = DHCP_DISCOVER;
|
2014-05-25 17:31:17 +02:00
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
2014-05-24 21:04:27 +02:00
|
|
|
|
|
|
|
test.message.op = 0;
|
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
|
|
test.message.op = BOOTREQUEST;
|
2014-05-25 17:31:17 +02:00
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
2014-05-24 21:04:27 +02:00
|
|
|
|
|
|
|
test.message.htype = 0;
|
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
|
|
test.message.htype = ARPHRD_ETHER;
|
2014-05-25 17:31:17 +02:00
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
2014-05-24 21:04:27 +02:00
|
|
|
|
|
|
|
test.message.hlen = 0;
|
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
|
|
|
test.message.hlen = ETHER_ADDR_LEN;
|
2014-05-25 17:31:17 +02:00
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_OFFER);
|
2014-05-25 18:28:03 +02:00
|
|
|
|
|
|
|
test.option_type.type = DHCP_REQUEST;
|
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
2016-01-20 14:44:24 +01:00
|
|
|
test.option_requested_ip.code = SD_DHCP_OPTION_REQUESTED_IP_ADDRESS;
|
2014-05-25 18:28:03 +02:00
|
|
|
test.option_requested_ip.length = 4;
|
|
|
|
test.option_requested_ip.address = htobe32(0x12345678);
|
2014-05-25 20:39:02 +02:00
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == DHCP_NAK);
|
2016-01-20 14:44:24 +01:00
|
|
|
test.option_server_id.code = SD_DHCP_OPTION_SERVER_IDENTIFIER;
|
2014-05-25 18:28:03 +02:00
|
|
|
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);
|
2014-05-25 22:07:53 +02:00
|
|
|
|
2014-05-25 18:28:03 +02:00
|
|
|
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);
|
2014-05-25 20:39:02 +02:00
|
|
|
test.option_server_id.address = htobe32(INADDR_LOOPBACK);
|
2014-05-25 22:07:53 +02:00
|
|
|
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);
|
|
|
|
|
2016-01-20 14:44:24 +01:00
|
|
|
test.option_client_id.code = SD_DHCP_OPTION_CLIENT_IDENTIFIER;
|
2014-05-25 22:07:53 +02:00
|
|
|
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);
|
|
|
|
|
2014-05-25 20:39:02 +02:00
|
|
|
test.option_requested_ip.address = htobe32(INADDR_LOOPBACK + 30);
|
2014-05-25 18:28:03 +02:00
|
|
|
assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
|
2014-05-24 21:04:27 +02:00
|
|
|
}
|
|
|
|
|
2015-10-04 00:22:41 +02:00
|
|
|
static uint64_t client_id_hash_helper(DHCPClientId *id, uint8_t key[HASH_KEY_SIZE]) {
|
|
|
|
struct siphash state;
|
|
|
|
|
2015-10-06 15:04:42 +02:00
|
|
|
siphash24_init(&state, key);
|
2015-10-04 00:22:41 +02:00
|
|
|
client_id_hash_func(id, &state);
|
2015-10-06 15:04:42 +02:00
|
|
|
|
2015-11-16 23:17:52 +01:00
|
|
|
return htole64(siphash24_finalize(&state));
|
2015-10-04 00:22:41 +02:00
|
|
|
}
|
|
|
|
|
2014-05-25 22:07:53 +02:00
|
|
|
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);
|
2015-10-04 00:22:41 +02:00
|
|
|
assert_se(client_id_hash_helper(&a, hash_key) == client_id_hash_helper(&b, hash_key));
|
2014-05-25 22:07:53 +02:00
|
|
|
a.length = 3;
|
|
|
|
assert_se(client_id_compare_func(&a, &b) != 0);
|
|
|
|
a.length = 4;
|
|
|
|
assert_se(client_id_compare_func(&a, &b) == 0);
|
2015-10-04 00:22:41 +02:00
|
|
|
assert_se(client_id_hash_helper(&a, hash_key) == client_id_hash_helper(&b, hash_key));
|
2014-05-25 22:07:53 +02:00
|
|
|
|
|
|
|
b.length = 3;
|
|
|
|
assert_se(client_id_compare_func(&a, &b) != 0);
|
|
|
|
b.length = 4;
|
|
|
|
assert_se(client_id_compare_func(&a, &b) == 0);
|
2015-10-04 00:22:41 +02:00
|
|
|
assert_se(client_id_hash_helper(&a, hash_key) == client_id_hash_helper(&b, hash_key));
|
2014-05-25 22:07:53 +02:00
|
|
|
|
|
|
|
free(b.data);
|
|
|
|
b.data = (uint8_t*)strdup("abce");
|
|
|
|
assert_se(client_id_compare_func(&a, &b) != 0);
|
|
|
|
|
|
|
|
free(a.data);
|
|
|
|
free(b.data);
|
|
|
|
}
|
|
|
|
|
2014-05-24 18:48:41 +02:00
|
|
|
int main(int argc, char *argv[]) {
|
tree-wide: expose "p"-suffix unref calls in public APIs to make gcc cleanup easy
GLIB has recently started to officially support the gcc cleanup
attribute in its public API, hence let's do the same for our APIs.
With this patch we'll define an xyz_unrefp() call for each public
xyz_unref() call, to make it easy to use inside a
__attribute__((cleanup())) expression. Then, all code is ported over to
make use of this.
The new calls are also documented in the man pages, with examples how to
use them (well, I only added docs where the _unref() call itself already
had docs, and the examples, only cover sd_bus_unrefp() and
sd_event_unrefp()).
This also renames sd_lldp_free() to sd_lldp_unref(), since that's how we
tend to call our destructors these days.
Note that this defines no public macro that wraps gcc's attribute and
makes it easier to use. While I think it's our duty in the library to
make our stuff easy to use, I figure it's not our duty to make gcc's own
features easy to use on its own. Most likely, client code which wants to
make use of this should define its own:
#define _cleanup_(function) __attribute__((cleanup(function)))
Or similar, to make the gcc feature easier to use.
Making this logic public has the benefit that we can remove three header
files whose only purpose was to define these functions internally.
See #2008.
2015-11-27 19:13:45 +01:00
|
|
|
_cleanup_(sd_event_unrefp) sd_event *e;
|
2014-06-13 19:11:31 +02:00
|
|
|
int r;
|
2014-05-24 18:48:41 +02:00
|
|
|
|
2018-09-13 14:31:13 +02:00
|
|
|
test_setup_logging(LOG_DEBUG);
|
2014-05-24 18:48:41 +02:00
|
|
|
|
|
|
|
assert_se(sd_event_new(&e) >= 0);
|
|
|
|
|
2014-06-13 19:11:31 +02:00
|
|
|
r = test_basic(e);
|
2018-09-13 13:34:12 +02:00
|
|
|
if (r != 0)
|
|
|
|
return log_tests_skipped("cannot start dhcp server");
|
2014-06-13 19:11:31 +02:00
|
|
|
|
2014-05-24 21:04:27 +02:00
|
|
|
test_message_handler();
|
2014-05-25 22:07:53 +02:00
|
|
|
test_client_id_hash();
|
2014-05-24 18:48:41 +02:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|