chiark / gitweb /
b087b5ead6aff56384cd060c20dd93f8c033495f
[gnupg2.git] / t-dns-stuff.c
1 /* t-dns-cert.c - Module test for dns-stuff.c
2  * Copyright (C) 2011 Free Software Foundation, Inc.
3  * Copyright (C) 2011, 2015 Werner Koch
4  *
5  * This file is part of GnuPG.
6  *
7  * GnuPG is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * GnuPG is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, see <https://www.gnu.org/licenses/>.
19  */
20
21 #include <config.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <assert.h>
25
26
27 #include "util.h"
28 #include "dns-stuff.h"
29
30 #define PGM "t-dns-stuff"
31
32 static int verbose;
33 static int debug;
34
35
36 static void
37 init_sockets (void)
38 {
39 #ifdef HAVE_W32_SYSTEM
40   WSADATA wsadat;
41
42   WSAStartup (0x202, &wsadat);
43 #endif
44 }
45
46
47 int
48 main (int argc, char **argv)
49 {
50   int last_argc = -1;
51   gpg_error_t err;
52   int any_options = 0;
53   int opt_tor = 0;
54   int opt_new_circuit = 0;
55   int opt_cert = 0;
56   int opt_srv = 0;
57   int opt_bracket = 0;
58   int opt_cname = 0;
59   char const *name = NULL;
60
61   gpgrt_init ();
62   log_set_prefix (PGM, GPGRT_LOG_WITH_PREFIX);
63   if (argc)
64     { argc--; argv++; }
65   while (argc && last_argc != argc )
66     {
67       last_argc = argc;
68       if (!strcmp (*argv, "--"))
69         {
70           argc--; argv++;
71           break;
72         }
73       else if (!strcmp (*argv, "--help"))
74         {
75           fputs ("usage: " PGM " [HOST]\n"
76                  "Options:\n"
77                  "  --verbose           print timings etc.\n"
78                  "  --debug             flyswatter\n"
79                  "  --standard-resolver use the system's resolver\n"
80                  "  --use-tor           use Tor\n"
81                  "  --new-circuit       use a new Tor circuit\n"
82                  "  --bracket           enclose v6 addresses in brackets\n"
83                  "  --cert              lookup a CERT RR\n"
84                  "  --srv               lookup a SRV RR\n"
85                  "  --cname             lookup a CNAME RR\n"
86                  "  --timeout SECONDS   timeout after SECONDS\n"
87                  , stdout);
88           exit (0);
89         }
90       else if (!strcmp (*argv, "--verbose"))
91         {
92           verbose++;
93           argc--; argv++;
94         }
95       else if (!strcmp (*argv, "--debug"))
96         {
97           verbose += 2;
98           debug++;
99           argc--; argv++;
100         }
101       else if (!strcmp (*argv, "--use-tor"))
102         {
103           opt_tor = 1;
104           argc--; argv++;
105         }
106       else if (!strcmp (*argv, "--new-circuit"))
107         {
108           opt_new_circuit = 1;
109           argc--; argv++;
110         }
111       else if (!strcmp (*argv, "--standard-resolver"))
112         {
113           enable_standard_resolver (1);
114           argc--; argv++;
115         }
116       else if (!strcmp (*argv, "--recursive-resolver"))
117         {
118           enable_recursive_resolver (1);
119           argc--; argv++;
120         }
121       else if (!strcmp (*argv, "--bracket"))
122         {
123           opt_bracket = 1;
124           argc--; argv++;
125         }
126       else if (!strcmp (*argv, "--cert"))
127         {
128           any_options = opt_cert = 1;
129           argc--; argv++;
130         }
131       else if (!strcmp (*argv, "--srv"))
132         {
133           any_options = opt_srv = 1;
134           argc--; argv++;
135         }
136       else if (!strcmp (*argv, "--cname"))
137         {
138           any_options = opt_cname = 1;
139           argc--; argv++;
140         }
141       else if (!strcmp (*argv, "--timeout"))
142         {
143           argc--; argv++;
144           if (argc)
145             {
146               set_dns_timeout (atoi (*argv));
147               argc--; argv++;
148             }
149         }
150       else if (!strncmp (*argv, "--", 2))
151         {
152           fprintf (stderr, PGM ": unknown option '%s'\n", *argv);
153           exit (1);
154         }
155     }
156
157   if (!argc && !any_options)
158     {
159       opt_cert = 1;
160       name = "simon.josefsson.org";
161     }
162   else if (argc == 1)
163     name = *argv;
164   else
165     {
166       fprintf (stderr, PGM ": none or too many host names given\n");
167       exit (1);
168     }
169
170   set_dns_verbose (verbose, debug);
171   init_sockets ();
172
173   if (opt_tor)
174     {
175       err = enable_dns_tormode (opt_new_circuit);
176       if (err)
177         {
178           fprintf (stderr, "error switching into Tor mode: %s\n",
179                    gpg_strerror (err));
180           exit (1);
181         }
182     }
183
184   if (opt_cert)
185     {
186       unsigned char *fpr;
187       size_t fpr_len;
188       char *url;
189       void *key;
190       size_t keylen;
191
192       if (verbose || any_options)
193         printf ("CERT lookup on '%s'\n", name);
194
195       err = get_dns_cert (name, DNS_CERTTYPE_ANY, &key, &keylen,
196                           &fpr, &fpr_len, &url);
197       if (err)
198         printf ("get_dns_cert failed: %s <%s>\n",
199                 gpg_strerror (err), gpg_strsource (err));
200       else if (key)
201         {
202           if (verbose || any_options)
203             printf ("Key found (%u bytes)\n", (unsigned int)keylen);
204         }
205       else
206         {
207           if (fpr)
208             {
209               int i;
210
211               printf ("Fingerprint found (%d bytes): ", (int)fpr_len);
212               for (i = 0; i < fpr_len; i++)
213                 printf ("%02X", fpr[i]);
214               putchar ('\n');
215             }
216           else
217             printf ("No fingerprint found\n");
218
219           if (url)
220             printf ("URL found: %s\n", url);
221           else
222             printf ("No URL found\n");
223
224         }
225
226       xfree (key);
227       xfree (fpr);
228       xfree (url);
229     }
230   else if (opt_cname)
231     {
232       char *cname;
233
234       printf ("CNAME lookup on '%s'\n", name);
235       err = get_dns_cname (name, &cname);
236       if (err)
237         printf ("get_dns_cname failed: %s <%s>\n",
238                 gpg_strerror (err), gpg_strsource (err));
239       else
240         {
241           printf ("CNAME found: '%s'\n", cname);
242         }
243       xfree (cname);
244     }
245   else if (opt_srv)
246     {
247       struct srventry *srv;
248       unsigned int count;
249       int i;
250
251       err = get_dns_srv (name? name : "_hkp._tcp.wwwkeys.pgp.net",
252                          &srv, &count);
253       if (err)
254         printf ("get_dns_srv failed: %s <%s>\n",
255                 gpg_strerror (err), gpg_strsource (err));
256       else
257         {
258           printf ("count=%u\n",count);
259           for (i=0; i < count; i++)
260             {
261               printf("priority=%-8hu  ",srv[i].priority);
262               printf("weight=%-8hu  ",srv[i].weight);
263               printf("port=%-5hu  ",srv[i].port);
264               printf("target=%s\n",srv[i].target);
265             }
266
267           xfree(srv);
268         }
269     }
270   else /* Standard lookup.  */
271     {
272       char *cname;
273       dns_addrinfo_t aibuf, ai;
274       char *host;
275
276       printf ("Lookup on '%s'\n", name);
277
278       err = resolve_dns_name (name, 0, 0, SOCK_STREAM, &aibuf, &cname);
279       if (err)
280         {
281           fprintf (stderr, PGM": resolving '%s' failed: %s\n",
282                    name, gpg_strerror (err));
283           exit (1);
284         }
285
286       if (cname)
287         printf ("cname: %s\n", cname);
288       for (ai = aibuf; ai; ai = ai->next)
289         {
290           printf ("%s %3d %3d   ",
291                   ai->family == AF_INET6? "inet6" :
292                   ai->family == AF_INET?  "inet4" : "?    ",
293                   ai->socktype, ai->protocol);
294
295           err = resolve_dns_addr (ai->addr, ai->addrlen,
296                                   (DNS_NUMERICHOST
297                                    | (opt_bracket? DNS_WITHBRACKET:0)),
298                                   &host);
299           if (err)
300             printf ("[resolve_dns_addr failed: %s]", gpg_strerror (err));
301           else
302             {
303               printf ("%s", host);
304               xfree (host);
305             }
306
307           err = resolve_dns_addr (ai->addr, ai->addrlen,
308                                   (opt_bracket? DNS_WITHBRACKET:0),
309                                   &host);
310           if (err)
311             printf ("  [resolve_dns_addr failed (2): %s]", gpg_strerror (err));
312           else
313             {
314               if (!is_ip_address (host))
315                 printf ("  (%s)", host);
316               xfree (host);
317             }
318           putchar ('\n');
319         }
320       xfree (cname);
321       free_dns_addrinfo (aibuf);
322     }
323
324   reload_dns_stuff (1); /* Release objects.  */
325
326   return 0;
327 }