src/libsystemd-network/test-dhcp-server.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright (C) 2013 Intel Corporation. All rights reserved.
   7   Copyright (C) 2014 Tom Gundersen
   8
   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.
  13
  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.
  18
  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/>.
  21 ***/
  22
  23 #include <netinet/if_ether.h>
  24 #include <assert.h>
  25 #include <errno.h>
  26
  27 #include "sd-event.h"
  28 #include "event-util.h"
  29
  30 #include "sd-dhcp-server.h"
  31 #include "dhcp-server-internal.h"
  32
  33 static void test_basic(sd_event *event) {
  34         _cleanup_dhcp_server_unref_ sd_dhcp_server *server = NULL;
  35
  36         /* attach to loopback interface */
  37         assert_se(sd_dhcp_server_new(&server, 1) >= 0);
  38         assert_se(server);
  39
  40         assert_se(sd_dhcp_server_attach_event(server, event, 0) >= 0);
  41         assert_se(sd_dhcp_server_attach_event(server, event, 0) == -EBUSY);
  42         assert_se(sd_dhcp_server_get_event(server) == event);
  43         assert_se(sd_dhcp_server_detach_event(server) >= 0);
  44         assert_se(!sd_dhcp_server_get_event(server));
  45         assert_se(sd_dhcp_server_attach_event(server, NULL, 0) >= 0);
  46         assert_se(sd_dhcp_server_attach_event(server, NULL, 0) == -EBUSY);
  47
  48         assert_se(sd_dhcp_server_ref(server) == server);
  49         assert_se(!sd_dhcp_server_unref(server));
  50
  51         assert_se(sd_dhcp_server_start(server) >= 0);
  52         assert_se(sd_dhcp_server_start(server) == -EBUSY);
  53         assert_se(sd_dhcp_server_stop(server) >= 0);
  54         assert_se(sd_dhcp_server_stop(server) >= 0);
  55         assert_se(sd_dhcp_server_start(server) >= 0);
  56 }
  57
  58 static void test_message_handler(void) {
  59         _cleanup_dhcp_server_unref_ sd_dhcp_server *server = NULL;
  60         struct {
  61                 DHCPMessage message;
  62                 struct {
  63                         uint8_t code;
  64                         uint8_t length;
  65                         uint8_t type;
  66                 } _packed_ option_type;
  67                 uint8_t end;
  68         } _packed_ test = {
  69                 .message.op = BOOTREQUEST,
  70                 .message.htype = ARPHRD_ETHER,
  71                 .message.hlen = ETHER_ADDR_LEN,
  72                 .option_type.code = DHCP_OPTION_MESSAGE_TYPE,
  73                 .option_type.length = 1,
  74                 .option_type.type = DHCP_DISCOVER,
  75                 .end = DHCP_OPTION_END,
  76         };
  77
  78         assert_se(sd_dhcp_server_new(&server, 1) >= 0);
  79
  80         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 1);
  81
  82         test.end = 0;
  83         /* TODO, shouldn't this fail? */
  84         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 1);
  85         test.end = DHCP_OPTION_END;
  86         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 1);
  87
  88         test.option_type.code = 0;
  89         test.option_type.length = 0;
  90         test.option_type.type = 0;
  91         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
  92         test.option_type.code = DHCP_OPTION_MESSAGE_TYPE;
  93         test.option_type.length = 1;
  94         test.option_type.type = DHCP_DISCOVER;
  95         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 1);
  96
  97         test.message.op = 0;
  98         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
  99         test.message.op = BOOTREQUEST;
 100         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 1);
 101
 102         test.message.htype = 0;
 103         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
 104         test.message.htype = ARPHRD_ETHER;
 105         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 1);
 106
 107         test.message.hlen = 0;
 108         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 0);
 109         test.message.hlen = ETHER_ADDR_LEN;
 110         assert_se(dhcp_server_handle_message(server, (DHCPMessage*)&test, sizeof(test)) == 1);
 111 }
 112
 113 int main(int argc, char *argv[]) {
 114         _cleanup_event_unref_ sd_event *e;
 115
 116         log_set_max_level(LOG_DEBUG);
 117         log_parse_environment();
 118         log_open();
 119
 120         assert_se(sd_event_new(&e) >= 0);
 121
 122         test_basic(e);
 123         test_message_handler();
 124
 125         return 0;
 126 }