service: Simplify nameserver route adding and removing
[framework/connectivity/connman.git] / tools / resolv-test.c
index db5b3fe..9867a4b 100644 (file)
@@ -2,7 +2,7 @@
  *
  *  Connection Manager
  *
- *  Copyright (C) 2007-2010  Intel Corporation. All rights reserved.
+ *  Copyright (C) 2007-2012  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
 #include <config.h>
 #endif
 
-#include <unistd.h>
+#include <stdio.h>
+#include <stdlib.h>
 #include <string.h>
-#include <resolv.h>
-#include <sys/socket.h>
-#include <arpa/inet.h>
-#include <arpa/nameser.h>
+#include <signal.h>
 
-static int do_connect(const char *server)
+#include <gweb/gresolv.h>
+
+static GTimer *timer;
+
+static GMainLoop *main_loop;
+
+static void resolv_debug(const char *str, void *data)
+{
+       g_print("%s: %s\n", (const char *) data, str);
+}
+
+static void sig_term(int sig)
+{
+       g_main_loop_quit(main_loop);
+}
+
+static const char *status2str(GResolvResultStatus status)
 {
-       struct sockaddr_in sin;
-       int sk;
-
-       sk = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
-       //sk = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
-       if (sk < 0)
-               return -1;
-
-       memset(&sin, 0, sizeof(sin));
-       sin.sin_family = AF_INET;
-       sin.sin_port = htons(53);
-       sin.sin_addr.s_addr = inet_addr(server);
-
-       if (connect(sk, (struct sockaddr *) &sin, sizeof(sin)) < 0) {
-               close(sk);
-               return -1;
+       switch (status) {
+       case G_RESOLV_RESULT_STATUS_SUCCESS:
+               return "success";
+       case G_RESOLV_RESULT_STATUS_ERROR:
+               return "error";
+       case G_RESOLV_RESULT_STATUS_NO_RESPONSE:
+               return "no response";
+       case G_RESOLV_RESULT_STATUS_FORMAT_ERROR:
+               return "format error";
+       case G_RESOLV_RESULT_STATUS_SERVER_FAILURE:
+               return "server failure";
+       case G_RESOLV_RESULT_STATUS_NAME_ERROR:
+               return "name error";
+       case G_RESOLV_RESULT_STATUS_NOT_IMPLEMENTED:
+               return "not implemented";
+       case G_RESOLV_RESULT_STATUS_REFUSED:
+               return "refused";
        }
 
-       return sk;
+       return NULL;
 }
 
-int main(int argc, char *argv[])
+static void resolv_result(GResolvResultStatus status,
+                                       char **results, gpointer user_data)
 {
-       ns_msg msg;
-       ns_rr rr;
-       int rcode;
-       const char *nameserver;
-       unsigned char buf[4096];
-       int i, sk, err, len, off = 0;
+       gdouble elapsed;
+       int i;
 
-       if (argc < 2) {
-               printf("missing argument\n");
-               return 1;
-       }
+       elapsed = g_timer_elapsed(timer, NULL);
 
-       if (argc > 2)
-               nameserver = argv[2];
-       else
-               nameserver = "127.0.0.1";
+       g_print("elapse: %f seconds\n", elapsed);
 
-       sk = do_connect(nameserver);
-       if (sk < 0) {
-               printf("Can't connect\n");
-               return 1;
+       g_print("status: %s\n", status2str(status));
+
+       if (results != NULL) {
+               for (i = 0; results[i]; i++)
+                       g_print("result: %s\n", results[i]);
        }
 
-       len = res_mkquery(ns_o_query, argv[1], ns_c_in, ns_t_a,
-                               NULL, 0, NULL, buf + off, sizeof(buf) - off);
-       printf("query len: %d\n", len);
+       g_main_loop_quit(main_loop);
+}
 
-       if (off > 0) {
-               buf[0] = len >> 8;
-               buf[1] = len & 0xff;
-       }
+static gboolean option_debug = FALSE;
 
-       //for (i = 0; i < len + off; i++)
-       //      printf("%02x ", buf[i]);
-       //printf("\n");
+static GOptionEntry options[] = {
+       { "debug", 'd', 0, G_OPTION_ARG_NONE, &option_debug,
+                                       "Enable debug output" },
+       { NULL },
+};
+
+int main(int argc, char *argv[])
+{
+       GOptionContext *context;
+       GError *error = NULL;
+       struct sigaction sa;
+       GResolv *resolv;
+       int index = 0;
+
+       context = g_option_context_new(NULL);
+       g_option_context_add_main_entries(context, options, NULL);
+
+       if (g_option_context_parse(context, &argc, &argv, &error) == FALSE) {
+               if (error != NULL) {
+                       g_printerr("%s\n", error->message);
+                       g_error_free(error);
+               } else
+                       g_printerr("An unknown error occurred\n");
+               exit(1);
+       }
 
-       err = send(sk, buf, len + off, 0);
-       printf("send result: %d\n", err);
+       g_option_context_free(context);
 
-       len = recv(sk, buf, sizeof(buf), 0);
-       printf("answer len: %d\n", len);
+       if (argc < 2) {
+               printf("missing argument\n");
+               return 1;
+       }
 
-       //for (i = 0; i < len + off; i++)
-       //      printf("%02x ", buf[i]);
-       //printf("\n");
+       resolv = g_resolv_new(index);
+       if (resolv == NULL) {
+               printf("failed to create resolver\n");
+               return 1;
+       }
 
-       close(sk);
+       if (option_debug == TRUE)
+               g_resolv_set_debug(resolv, resolv_debug, "RESOLV");
 
-       ns_initparse(buf + off, len - off, &msg);
+       main_loop = g_main_loop_new(NULL, FALSE);
 
-       rcode = ns_msg_getflag(msg, ns_f_rcode);
+       if (argc > 2) {
+               int i;
 
-       printf("msg id: 0x%04x\n", ns_msg_id(msg));
-       printf("msg rcode: %d\n", rcode);
-       printf("msg count: %d\n", ns_msg_count(msg, ns_s_an));
+               for (i = 2; i < argc; i++)
+                       g_resolv_add_nameserver(resolv, argv[i], 53, 0);
+       }
 
-       for (i = 0; i < ns_msg_count(msg, ns_s_an); i++) {
-               char result[100];
+       timer = g_timer_new();
 
-               ns_parserr(&msg, ns_s_an, i, &rr);
+       if (g_resolv_lookup_hostname(resolv, argv[1],
+                                       resolv_result, NULL) == 0) {
+               printf("failed to start lookup\n");
+               return 1;
+       }
 
-               if (ns_rr_class(rr) != ns_c_in)
-                       continue;
+       memset(&sa, 0, sizeof(sa));
+       sa.sa_handler = sig_term;
+       sigaction(SIGINT, &sa, NULL);
+       sigaction(SIGTERM, &sa, NULL);
 
-               if (ns_rr_type(rr) != ns_t_a)
-                       continue;
+       g_main_loop_run(main_loop);
 
-               if (ns_rr_rdlen(rr) != NS_INADDRSZ)
-                       continue;
+       g_timer_destroy(timer);
 
-               inet_ntop(AF_INET, ns_rr_rdata(rr), result, sizeof(result));
+       g_resolv_unref(resolv);
 
-               printf("result: %s\n", result);
-       }
+       g_main_loop_unref(main_loop);
 
        return 0;
 }