From 32d1817b7b5e7f9a3044327df36433ae35a532a7 Mon Sep 17 00:00:00 2001 From: Martin Xu Date: Wed, 21 Jul 2010 23:14:27 -0700 Subject: [PATCH] Add initial version of DHCP client test tool --- Makefile.am | 9 ++- tools/dhcp-test.c | 185 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 191 insertions(+), 3 deletions(-) create mode 100644 tools/dhcp-test.c diff --git a/Makefile.am b/Makefile.am index 9ad6951..f3c21b6 100644 --- a/Makefile.am +++ b/Makefile.am @@ -118,10 +118,10 @@ client_cm_LDADD = @DBUS_LIBS@ endif if TOOLS -noinst_PROGRAMS += tools/wifi-scan tools/supplicant-test tools/dbus-test \ +noinst_PROGRAMS += tools/wifi-scan tools/supplicant-test tools/dhcp-test \ tools/addr-test tools/tap-test tools/resolv-test \ - tools/polkit-test tools/portal-test \ - tools/iptables-test + tools/dbus-test tools/polkit-test tools/portal-test \ + tools/iptables-test tools_wifi_scan_LDADD = @GLIB_LIBS@ @NETLINK_LIBS@ @@ -130,6 +130,9 @@ tools_supplicant_test_SOURCES = $(gdbus_sources) tools/supplicant-test.c \ tools/supplicant.h tools/supplicant.c tools_supplicant_test_LDADD = @GLIB_LIBS@ @DBUS_LIBS@ +tools_dhcp_test_SOURCES = $(gdhcp_sources) tools/dhcp-test.c +tools_dhcp_test_LDADD = @GLIB_LIBS@ + tools_dbus_test_SOURCES = $(gdbus_sources) tools/dbus-test.c tools_dbus_test_LDADD = @GLIB_LIBS@ @DBUS_LIBS@ diff --git a/tools/dhcp-test.c b/tools/dhcp-test.c new file mode 100644 index 0000000..325123d --- /dev/null +++ b/tools/dhcp-test.c @@ -0,0 +1,185 @@ +/* + * + * Connection Manager + * + * Copyright (C) 2007-2010 Intel Corporation. All rights reserved. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 2 as + * published by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + * + */ + +#ifdef HAVE_CONFIG_H +#include +#endif + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include + +static GTimer *timer; + +static GMainLoop *main_loop = NULL; + +static void sig_term(int sig) +{ + g_main_loop_quit(main_loop); +} + +static void print_timestamp(void) +{ + GTimeVal timestamp; + gdouble elapsed; + char *str; + + g_get_current_time(×tamp); + str = g_time_val_to_iso8601(×tamp); + printf("=== %s ===\n", str); + g_free(str); + + elapsed = g_timer_elapsed(timer, NULL); + + printf("elapsed: %f seconds\n", elapsed); +} + +static void handle_error(GDHCPClientError error) +{ + switch (error) { + case G_DHCP_CLIENT_ERROR_NONE: + printf("dhcp client ok\n"); + break; + case G_DHCP_CLIENT_ERROR_INERFACE_UNAVAILABLE: + printf("Interface unavailable\n"); + break; + case G_DHCP_CLIENT_ERROR_INTERFACE_IN_USE: + printf("Interface in use\n"); + break; + case G_DHCP_CLIENT_ERROR_INTERFACE_DOWN: + printf("Interface down\n"); + break; + case G_DHCP_CLIENT_ERROR_NOMEM: + printf("No memory\n"); + break; + case G_DHCP_CLIENT_ERROR_INVALID_INDEX: + printf("Invalid index\n"); + break; + case G_DHCP_CLIENT_ERROR_INVALID_OPTION: + printf("Invalid option\n"); + break; + } +} + +static void no_lease_cb(GDHCPClient *dhcp_client, gpointer user_data) +{ + printf("No Lease Available!\n"); + + print_timestamp(); + + g_main_loop_quit(main_loop); +} + +static void lease_available_cb(GDHCPClient *dhcp_client, gpointer user_data) +{ + GList *list, *option_value = NULL; + char *address; + + address = g_dhcp_client_get_address(dhcp_client); + printf("address %s\n", address); + if (address == NULL) + return; + + option_value = g_dhcp_client_get_option(dhcp_client, G_DHCP_SUBNET); + for (list = option_value; list; list = list->next) + printf("sub-mask %s\n", (char *) list->data); + + option_value = g_dhcp_client_get_option(dhcp_client, G_DHCP_DNS_SERVER); + for (list = option_value; list; list = list->next) + printf("domain-name-servers %s\n", (char *) list->data); + + option_value = g_dhcp_client_get_option(dhcp_client, G_DHCP_ROUTER); + for (list = option_value; list; list = list->next) + printf("routers %s\n", (char *) list->data); + + print_timestamp(); +} + +int main(int argc, char *argv[]) +{ + struct sigaction sa; + GDHCPClientError error; + GDHCPClient *dhcp_client; + int index; + + if (argc < 2) { + printf("Usage: dhcp-test \n"); + exit(0); + } + + index = atoi(argv[1]); + + printf("Create DHCP client for interface %d\n", index); + + timer = g_timer_new(); + + dhcp_client = g_dhcp_client_new(G_DHCP_IPV4, index, &error); + if (dhcp_client == NULL) { + handle_error(error); + exit(0); + } + + g_dhcp_client_set_send(dhcp_client, G_DHCP_HOST_NAME, ""); + + g_dhcp_client_set_request(dhcp_client, G_DHCP_HOST_NAME); + g_dhcp_client_set_request(dhcp_client, G_DHCP_SUBNET); + g_dhcp_client_set_request(dhcp_client, G_DHCP_DNS_SERVER); + g_dhcp_client_set_request(dhcp_client, G_DHCP_NTP_SERVER); + g_dhcp_client_set_request(dhcp_client, G_DHCP_ROUTER); + + g_dhcp_client_register_event(dhcp_client, + G_DHCP_CLIENT_EVENT_LEASE_AVAILABLE, + lease_available_cb, NULL); + + g_dhcp_client_register_event(dhcp_client, + G_DHCP_CLIENT_EVENT_NO_LEASE, no_lease_cb, NULL); + + main_loop = g_main_loop_new(NULL, FALSE); + + print_timestamp(); + + printf("Start DHCP operation\n"); + + g_dhcp_client_start(dhcp_client); + + memset(&sa, 0, sizeof(sa)); + sa.sa_handler = sig_term; + sigaction(SIGINT, &sa, NULL); + sigaction(SIGTERM, &sa, NULL); + + g_main_loop_run(main_loop); + + g_dhcp_client_unref(dhcp_client); + + g_main_loop_unref(main_loop); + + return 0; +} -- 2.7.4