Add tool for standalone dhclient testing
authorMarcel Holtmann <marcel@holtmann.org>
Wed, 28 Jul 2010 20:25:22 +0000 (13:25 -0700)
committerMarcel Holtmann <marcel@holtmann.org>
Wed, 28 Jul 2010 20:25:22 +0000 (13:25 -0700)
.gitignore
Makefile.am
tools/dhclient-test.c [new file with mode: 0644]

index e46f5aa..5ec713d 100644 (file)
@@ -47,6 +47,7 @@ tools/resolv-test
 tools/polkit-test
 tools/portal-test
 tools/iptables-test
+tools/dhclient-test
 tools/supplicant-test
 tools/dbus-test
 doc/*.bak
index 50d80de..68f5c11 100644 (file)
@@ -124,7 +124,7 @@ if TOOLS
 noinst_PROGRAMS += tools/wifi-scan tools/supplicant-test tools/dhcp-test \
                        tools/addr-test tools/tap-test tools/resolv-test \
                        tools/dbus-test tools/polkit-test tools/portal-test \
-                                                       tools/iptables-test
+                       tools/iptables-test tools/dhclient-test
 
 tools_wifi_scan_LDADD = @GLIB_LIBS@ @NETLINK_LIBS@
 
@@ -147,6 +147,9 @@ tools_polkit_test_LDADD = @DBUS_LIBS@
 tools_portal_test_LDADD = @GLIB_LIBS@
 
 tools_iptables_test_LDADD = @IPTC_LIBS@ -lip4tc -lxtables
+
+tools_dhclient_test_SOURCES = $(gdbus_sources) tools/dhclient-test.c
+tools_dhclient_test_LDADD = @GLIB_LIBS@ @DBUS_LIBS@
 endif
 
 test_scripts = test/get-state test/list-profiles test/list-services \
diff --git a/tools/dhclient-test.c b/tools/dhclient-test.c
new file mode 100644 (file)
index 0000000..a1bdf3c
--- /dev/null
@@ -0,0 +1,263 @@
+/*
+ *
+ *  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 <config.h>
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <signal.h>
+
+#include <gdbus.h>
+
+#define INTERFACE "isc.dhclient"
+#define PATH "/dhclient"
+
+static GTimer *timer;
+
+static GMainLoop *main_loop;
+
+static guint child_watch = 0;
+static pid_t pid = 0;
+
+static void sig_term(int sig)
+{
+       g_main_loop_quit(main_loop);
+}
+
+static void free_pointer(gpointer data, gpointer user_data)
+{
+       g_free(data);
+}
+
+static void dhclient_died(GPid pid, gint status, gpointer user_data)
+{
+       if (WIFEXITED(status))
+               printf("exit status %d\n", WEXITSTATUS(status));
+       else
+               printf("signal %d\n", WTERMSIG(status));
+
+       g_spawn_close_pid(pid);
+
+       child_watch = 0;
+}
+
+static void dhclient_setup(gpointer user_data)
+{
+       printf("dhclient process setup\n");
+}
+
+static void add_argument(GPtrArray *array, const char *name,
+                                               const char *format, ...)
+{
+       va_list ap;
+       char *str;
+
+       str = g_strdup(name);
+       g_ptr_array_add(array, str);
+
+       va_start(ap, format);
+
+       if (format != NULL) {
+               str = g_strdup_vprintf(format, ap);
+               g_ptr_array_add(array, str);
+       }
+
+       va_end(ap);
+}
+
+static void start_dhclient(DBusConnection *conn, const char *ifname)
+{
+       GSpawnFlags flags = G_SPAWN_DO_NOT_REAP_CHILD;
+       GPtrArray *argv;
+       GPtrArray *envp;
+       gboolean result;
+       const char *busname;
+
+       busname = dbus_bus_get_unique_name(conn);
+       busname = "org.moblin.connman";
+
+       argv = g_ptr_array_new();
+       add_argument(argv, DHCLIENT, NULL);
+       add_argument(argv, "-d", NULL);
+       add_argument(argv, "-q", NULL);
+       add_argument(argv, "-e", "BUSNAME=%s", busname);
+       add_argument(argv, "-e", "BUSINTF=%s", INTERFACE);
+       add_argument(argv, "-e", "BUSPATH=%s", PATH);
+       add_argument(argv, "-pf", "%s/dhclient.%s.pid", STATEDIR, ifname);
+       add_argument(argv, "-lf", "%s/dhclient.%s.leases", STATEDIR, ifname);
+       add_argument(argv, "-cf", "%s/dhclient.conf", SCRIPTDIR);
+       add_argument(argv, "-sf", "%s/dhclient-script", SCRIPTDIR);
+       add_argument(argv, ifname, NULL);
+       add_argument(argv, "-n", NULL);
+       g_ptr_array_add(argv, NULL);
+
+       envp = g_ptr_array_new();
+       g_ptr_array_add(envp, NULL);
+
+       result = g_spawn_async_with_pipes(NULL, (char **) argv->pdata,
+                                               (char **) envp->pdata,
+                                               flags, dhclient_setup, NULL,
+                                               &pid, NULL, NULL, NULL, NULL);
+
+       child_watch = g_child_watch_add(pid, dhclient_died, NULL);
+
+       g_ptr_array_foreach(envp, free_pointer, NULL);
+       g_ptr_array_free(envp, TRUE);
+
+       g_ptr_array_foreach(argv, free_pointer, NULL);
+       g_ptr_array_free(argv, TRUE);
+}
+
+static void parse_notification(DBusMessage *msg)
+{
+       DBusMessageIter iter, dict;
+       dbus_uint32_t pid;
+       gdouble elapsed;
+       const char *text, *key, *value;
+
+       dbus_message_iter_init(msg, &iter);
+
+       dbus_message_iter_get_basic(&iter, &pid);
+       dbus_message_iter_next(&iter);
+
+       dbus_message_iter_get_basic(&iter, &text);
+       dbus_message_iter_next(&iter);
+
+       printf("change %d to %s\n", pid, text);
+
+       dbus_message_iter_recurse(&iter, &dict);
+
+       while (dbus_message_iter_get_arg_type(&dict) == DBUS_TYPE_DICT_ENTRY) {
+               DBusMessageIter entry;
+
+               dbus_message_iter_recurse(&dict, &entry);
+               dbus_message_iter_get_basic(&entry, &key);
+               dbus_message_iter_next(&entry);
+               dbus_message_iter_get_basic(&entry, &value);
+
+               printf("%s = %s\n", key, value);
+
+               dbus_message_iter_next(&dict);
+       }
+
+       if (g_strcmp0(text, "PREINIT") == 0)
+               return;
+
+       elapsed = g_timer_elapsed(timer, NULL);
+
+       g_print("elapsed %f seconds\n", elapsed);
+
+       g_main_loop_quit(main_loop);
+}
+
+static DBusHandlerResult notify_filter(DBusConnection *conn,
+                                       DBusMessage *msg, void *user_data)
+{
+       if (dbus_message_get_type(msg) != DBUS_MESSAGE_TYPE_METHOD_CALL)
+               return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
+
+       if (dbus_message_has_interface(msg, INTERFACE) == FALSE)
+               return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
+
+       if (dbus_message_has_path(msg, PATH) == FALSE)
+               return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
+
+       if (dbus_message_has_member(msg, "Notify") == FALSE)
+               return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
+
+       parse_notification(msg);
+
+       return DBUS_HANDLER_RESULT_HANDLED;
+}
+
+static const char *notify_rule = "type=method_call"
+                                       ",interface=" INTERFACE;
+
+int main(int argc, char *argv[])
+{
+       DBusConnection *conn;
+       DBusError err;
+       struct sigaction sa;
+       const char *ifname;
+
+       if (argc < 2) {
+               printf("Usage: dhclient-test <interface name>\n");
+               exit(0);
+       }
+
+       ifname = argv[1];
+
+       printf("Create DHCP client for interface %s\n", ifname);
+
+       main_loop = g_main_loop_new(NULL, FALSE);
+
+       dbus_error_init(&err);
+
+       conn = g_dbus_setup_bus(DBUS_BUS_SYSTEM, "org.moblin.connman", &err);
+       if (conn == NULL) {
+               if (dbus_error_is_set(&err) == TRUE) {
+                       fprintf(stderr, "%s\n", err.message);
+                       dbus_error_free(&err);
+               } else
+                       fprintf(stderr, "Can't register with system bus\n");
+               exit(1);
+       }
+
+       dbus_connection_add_filter(conn, notify_filter, NULL, NULL);
+
+       dbus_bus_add_match(conn, notify_rule, NULL);
+       dbus_connection_flush(conn);
+
+       printf("Start DHCP operation\n");
+
+       timer = g_timer_new();
+
+       start_dhclient(conn, ifname);
+
+       memset(&sa, 0, sizeof(sa));
+       sa.sa_handler = sig_term;
+       sigaction(SIGINT, &sa, NULL);
+       sigaction(SIGTERM, &sa, NULL);
+
+       g_main_loop_run(main_loop);
+
+       if (pid > 0)
+               kill(pid, SIGTERM);
+
+       if (child_watch > 0)
+               g_source_remove(child_watch);
+
+       g_timer_destroy(timer);
+
+       dbus_bus_remove_match(conn, notify_rule, NULL);
+       dbus_connection_flush(conn);
+
+       dbus_connection_remove_filter(conn, notify_filter, NULL);
+
+       dbus_connection_unref(conn);
+
+       g_main_loop_unref(main_loop);
+
+       return 0;
+}