Merge branch 'master' of ssh://gnunet.org/gnunet
[oweals/gnunet.git] / src / util / gnunet-resolver.c
1 /*
2      This file is part of GNUnet.
3      Copyright (C) 2010 GNUnet e.V.
4
5      GNUnet is free software; you can redistribute it and/or modify
6      it under the terms of the GNU General Public License as published
7      by the Free Software Foundation; either version 3, or (at your
8      option) any later version.
9
10      GNUnet is distributed in the hope that it will be useful, but
11      WITHOUT ANY WARRANTY; without even the implied warranty of
12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13      General Public License for more details.
14
15      You should have received a copy of the GNU General Public License
16      along with GNUnet; see the file COPYING.  If not, write to the
17      Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
18      Boston, MA 02110-1301, USA.
19 */
20
21 /**
22  * @file util/gnunet-resolver.c
23  * @brief tool to test resolver
24  * @author Christian Grothoff
25  */
26 #include "platform.h"
27 #include "gnunet_util_lib.h"
28 #include "gnunet_resolver_service.h"
29
30 #define GET_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
31
32 /**
33  * Flag for reverse lookup.
34  */
35 static int reverse;
36
37
38 /**
39  * Prints each hostname obtained from DNS.
40  *
41  * @param cls closure (unused)
42  * @param hostname one of the names for the host, NULL
43  *        on the last call to the callback
44  */
45 static void
46 print_hostname (void *cls,
47                 const char *hostname)
48 {
49   if (NULL == hostname)
50     return;
51   FPRINTF (stdout, "%s\n", hostname);
52 }
53
54
55 /**
56  * Callback function to display address.
57  *
58  * @param cls closure (unused)
59  * @param addr one of the addresses of the host, NULL for the last address
60  * @param addrlen length of the address
61  */
62 static void
63 print_sockaddr (void *cls, const struct sockaddr *addr, socklen_t addrlen)
64 {
65   if (NULL == addr)
66     return;
67   FPRINTF (stdout, "%s\n", GNUNET_a2s (addr, addrlen));
68 }
69
70
71 /**
72  * Main function that will be run by the scheduler.
73  *
74  * @param cls closure
75  * @param args remaining command-line arguments
76  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
77  * @param cfg configuration
78  */
79 static void
80 run (void *cls, char *const *args, const char *cfgfile,
81      const struct GNUNET_CONFIGURATION_Handle *cfg)
82 {
83   const struct sockaddr *sa;
84   socklen_t salen;
85   struct sockaddr_in v4;
86   struct sockaddr_in6 v6;
87
88   if (args[0] == NULL)
89     return;
90   if (! reverse)
91   {
92     GNUNET_RESOLVER_ip_get (args[0], AF_UNSPEC, GET_TIMEOUT, &print_sockaddr, NULL);
93     return;
94   }
95
96   sa = NULL;
97   memset (&v4, 0, sizeof (v4));
98   v4.sin_family = AF_INET;
99 #if HAVE_SOCKADDR_IN_SIN_LEN
100   v4.sin_len = sizeof (v4);
101 #endif
102   if (1 == inet_pton (AF_INET,
103                       args[0],
104                       &v4.sin_addr))
105   {
106     sa = (struct sockaddr *) &v4;
107     salen = sizeof (v4);
108   }
109   memset (&v6, 0, sizeof (v6));
110   v6.sin6_family = AF_INET6;
111 #if HAVE_SOCKADDR_IN_SIN_LEN
112   v6.sin6_len = sizeof (v6);
113 #endif
114   if (1 == inet_pton (AF_INET6,
115                       args[0],
116                       &v6.sin6_addr))
117   {
118     sa = (struct sockaddr *) &v6;
119     salen = sizeof (v6);
120   }
121   if (NULL == sa)
122   {
123     fprintf (stderr,
124              "`%s' is not a valid IP: %s\n",
125              args[0],
126              strerror (errno));
127     return;
128   }
129   GNUNET_RESOLVER_hostname_get (sa, salen,
130                                 GNUNET_YES,
131                                 GET_TIMEOUT,
132                                 &print_hostname,
133                                 NULL);
134 }
135
136
137 /**
138  * The main function to access GNUnet's DNS resolver.
139  *
140  * @param argc number of arguments from the command line
141  * @param argv command line arguments
142  * @return 0 ok, 1 on error
143  */
144 int
145 main (int argc, char *const *argv)
146 {
147   struct GNUNET_GETOPT_CommandLineOption options[] = {
148     GNUNET_GETOPT_OPTION_SET_ONE ('r',
149                                   "reverse",
150                                   gettext_noop ("perform a reverse lookup"),
151                                   &reverse),
152     GNUNET_GETOPT_OPTION_END
153   };
154   int ret;
155
156   if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
157     return 2;
158
159   ret = (GNUNET_OK ==
160          GNUNET_PROGRAM_run (argc, argv, "gnunet-resolver [hostname]",
161                              gettext_noop ("Use build-in GNUnet stub resolver"),
162                              options, &run, NULL)) ? 0 : 1;
163   GNUNET_free ((void*) argv);
164   return ret;
165 }
166
167 /* end of gnunet-resolver.c */