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.
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU Lesser General Public License as published by
10 the Free Software Foundation; either version 2.1 of the License, or
11 (at your option) any later version.
13 systemd is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 Lesser General Public License for more details.
18 You should have received a copy of the GNU Lesser General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
29 #include "dhcp-protocol.h"
30 #include "dhcp-internal.h"
31 #include "sd-dhcp-client.h"
33 static struct ether_addr mac_addr = {
34 .ether_addr_octet = {'A', 'B', 'C', '1', '2', '3'}
37 static void test_request_basic(void)
39 sd_dhcp_client *client;
41 client = sd_dhcp_client_new();
45 assert(sd_dhcp_client_set_request_option(NULL, 0) == -EINVAL);
46 assert(sd_dhcp_client_set_request_address(NULL, NULL) == -EINVAL);
47 assert(sd_dhcp_client_set_index(NULL, 0) == -EINVAL);
49 assert(sd_dhcp_client_set_index(client, 15) == 0);
50 assert(sd_dhcp_client_set_index(client, -42) == -EINVAL);
51 assert(sd_dhcp_client_set_index(client, -1) == 0);
53 assert(sd_dhcp_client_set_request_option(client,
54 DHCP_OPTION_SUBNET_MASK) == -EEXIST);
55 assert(sd_dhcp_client_set_request_option(client,
56 DHCP_OPTION_ROUTER) == -EEXIST);
57 assert(sd_dhcp_client_set_request_option(client,
58 DHCP_OPTION_HOST_NAME) == -EEXIST);
59 assert(sd_dhcp_client_set_request_option(client,
60 DHCP_OPTION_DOMAIN_NAME) == -EEXIST);
61 assert(sd_dhcp_client_set_request_option(client,
62 DHCP_OPTION_DOMAIN_NAME_SERVER)
64 assert(sd_dhcp_client_set_request_option(client,
65 DHCP_OPTION_NTP_SERVER) == -EEXIST);
67 assert(sd_dhcp_client_set_request_option(client,
68 DHCP_OPTION_PAD) == -EINVAL);
69 assert(sd_dhcp_client_set_request_option(client,
70 DHCP_OPTION_END) == -EINVAL);
71 assert(sd_dhcp_client_set_request_option(client,
72 DHCP_OPTION_MESSAGE_TYPE) == -EINVAL);
73 assert(sd_dhcp_client_set_request_option(client,
74 DHCP_OPTION_OVERLOAD) == -EINVAL);
75 assert(sd_dhcp_client_set_request_option(client,
76 DHCP_OPTION_PARAMETER_REQUEST_LIST)
79 assert(sd_dhcp_client_set_request_option(client, 33) == 0);
80 assert(sd_dhcp_client_set_request_option(client, 33) == -EEXIST);
81 assert(sd_dhcp_client_set_request_option(client, 44) == 0);
84 static uint16_t client_checksum(void *buf, int len)
94 for (i = 0; i < len / 2 ; i++)
102 return ~((sum & 0xffff) + (sum >> 16));
105 static void test_checksum(void)
108 0x45, 0x00, 0x02, 0x40, 0x00, 0x00, 0x00, 0x00,
109 0x40, 0x11, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
110 0xff, 0xff, 0xff, 0xff
117 uint16_t *val = (uint16_t *)check;
119 assert(client_checksum(&buf, 20) == *val);
122 static int check_options(uint8_t code, uint8_t len, const uint8_t *option,
128 int dhcp_network_send_raw_packet(int index, const void *packet, size_t len)
131 _cleanup_free_ DHCPPacket *discover;
132 uint16_t ip_check, udp_check;
138 size = sizeof(DHCPPacket) + 4;
141 discover = memdup(packet, len);
143 assert(memcmp(discover->dhcp.chaddr,
144 &mac_addr.ether_addr_octet, 6) == 0);
145 assert(discover->ip.ttl == IPDEFTTL);
146 assert(discover->ip.protocol == IPPROTO_UDP);
147 assert(discover->ip.saddr == INADDR_ANY);
148 assert(discover->ip.daddr == INADDR_BROADCAST);
149 assert(discover->udp.source == ntohs(DHCP_PORT_CLIENT));
150 assert(discover->udp.dest == ntohs(DHCP_PORT_SERVER));
152 ip_check = discover->ip.check;
154 discover->ip.ttl = 0;
155 discover->ip.check = discover->udp.len;
157 udp_check = ~client_checksum(&discover->ip.ttl, len - 8);
158 assert(udp_check == 0xffff);
160 discover->ip.ttl = IPDEFTTL;
161 discover->ip.check = ip_check;
163 ip_check = ~client_checksum(&discover->ip, sizeof(discover->ip));
164 assert(ip_check == 0xffff);
166 size = len - sizeof(struct iphdr) - sizeof(struct udphdr);
168 res = dhcp_option_parse(&discover->dhcp, size, check_options, NULL);
175 static void test_discover_message(void)
177 sd_dhcp_client *client;
180 client = sd_dhcp_client_new();
183 assert(sd_dhcp_client_set_index(client, 42) >= 0);
184 assert(sd_dhcp_client_set_mac(client, &mac_addr) >= 0);
186 assert(sd_dhcp_client_set_request_option(client, 248) >= 0);
188 res = sd_dhcp_client_start(client);
190 assert(res == 0 || res == -EINPROGRESS);
193 int main(int argc, char *argv[])
195 test_request_basic();
198 test_discover_message();