dirmngr: Fix problems with the getsrv function.
[gnupg.git] / dirmngr / 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                  , stdout);
87           exit (0);
88         }
89       else if (!strcmp (*argv, "--verbose"))
90         {
91           verbose++;
92           argc--; argv++;
93         }
94       else if (!strcmp (*argv, "--debug"))
95         {
96           verbose += 2;
97           debug++;
98           argc--; argv++;
99         }
100       else if (!strcmp (*argv, "--use-tor"))
101         {
102           opt_tor = 1;
103           argc--; argv++;
104         }
105       else if (!strcmp (*argv, "--new-circuit"))
106         {
107           opt_new_circuit = 1;
108           argc--; argv++;
109         }
110       else if (!strcmp (*argv, "--standard-resolver"))
111         {
112           enable_standard_resolver (1);
113           argc--; argv++;
114         }
115       else if (!strcmp (*argv, "--recursive-resolver"))
116         {
117           enable_recursive_resolver (1);
118           argc--; argv++;
119         }
120       else if (!strcmp (*argv, "--bracket"))
121         {
122           opt_bracket = 1;
123           argc--; argv++;
124         }
125       else if (!strcmp (*argv, "--cert"))
126         {
127           any_options = opt_cert = 1;
128           argc--; argv++;
129         }
130       else if (!strcmp (*argv, "--srv"))
131         {
132           any_options = opt_srv = 1;
133           argc--; argv++;
134         }
135       else if (!strcmp (*argv, "--cname"))
136         {
137           any_options = opt_cname = 1;
138           argc--; argv++;
139         }
140       else if (!strncmp (*argv, "--", 2))
141         {
142           fprintf (stderr, PGM ": unknown option '%s'\n", *argv);
143           exit (1);
144         }
145     }
146
147   if (!argc && !any_options)
148     {
149       opt_cert = 1;
150       name = "simon.josefsson.org";
151     }
152   else if (argc == 1)
153     name = *argv;
154   else
155     {
156       fprintf (stderr, PGM ": none or too many host names given\n");
157       exit (1);
158     }
159
160   set_dns_verbose (verbose, debug);
161   init_sockets ();
162
163   if (opt_tor)
164     {
165       err = enable_dns_tormode (opt_new_circuit);
166       if (err)
167         {
168           fprintf (stderr, "error switching into Tor mode: %s\n",
169                    gpg_strerror (err));
170           exit (1);
171         }
172     }
173
174   if (opt_cert)
175     {
176       unsigned char *fpr;
177       size_t fpr_len;
178       char *url;
179       void *key;
180       size_t keylen;
181
182       if (verbose || any_options)
183         printf ("CERT lookup on '%s'\n", name);
184
185       err = get_dns_cert (name, DNS_CERTTYPE_ANY, &key, &keylen,
186                           &fpr, &fpr_len, &url);
187       if (err)
188         printf ("get_dns_cert failed: %s <%s>\n",
189                 gpg_strerror (err), gpg_strsource (err));
190       else if (key)
191         {
192           if (verbose || any_options)
193             printf ("Key found (%u bytes)\n", (unsigned int)keylen);
194         }
195       else
196         {
197           if (fpr)
198             {
199               int i;
200
201               printf ("Fingerprint found (%d bytes): ", (int)fpr_len);
202               for (i = 0; i < fpr_len; i++)
203                 printf ("%02X", fpr[i]);
204               putchar ('\n');
205             }
206           else
207             printf ("No fingerprint found\n");
208
209           if (url)
210             printf ("URL found: %s\n", url);
211           else
212             printf ("No URL found\n");
213
214         }
215
216       xfree (key);
217       xfree (fpr);
218       xfree (url);
219     }
220   else if (opt_cname)
221     {
222       char *cname;
223
224       printf ("CNAME lookup on '%s'\n", name);
225       err = get_dns_cname (name, &cname);
226       if (err)
227         printf ("get_dns_cname failed: %s <%s>\n",
228                 gpg_strerror (err), gpg_strsource (err));
229       else
230         {
231           printf ("CNAME found: '%s'\n", cname);
232         }
233       xfree (cname);
234     }
235   else if (opt_srv)
236     {
237       struct srventry *srv;
238       unsigned int count;
239       int i;
240
241       err = get_dns_srv (name? name : "_hkp._tcp.wwwkeys.pgp.net",
242                          &srv, &count);
243       if (err)
244         printf ("get_dns_srv failed: %s <%s>\n",
245                 gpg_strerror (err), gpg_strsource (err));
246       else
247         {
248           printf ("count=%u\n",count);
249           for (i=0; i < count; i++)
250             {
251               printf("priority=%-8hu  ",srv[i].priority);
252               printf("weight=%-8hu  ",srv[i].weight);
253               printf("port=%-5hu  ",srv[i].port);
254               printf("target=%s\n",srv[i].target);
255             }
256
257           xfree(srv);
258         }
259     }
260   else /* Standard lookup.  */
261     {
262       char *cname;
263       dns_addrinfo_t aibuf, ai;
264       char *host;
265
266       printf ("Lookup on '%s'\n", name);
267
268       err = resolve_dns_name (name, 0, 0, SOCK_STREAM, &aibuf, &cname);
269       if (err)
270         {
271           fprintf (stderr, PGM": resolving '%s' failed: %s\n",
272                    name, gpg_strerror (err));
273           exit (1);
274         }
275
276       if (cname)
277         printf ("cname: %s\n", cname);
278       for (ai = aibuf; ai; ai = ai->next)
279         {
280           printf ("%s %3d %3d   ",
281                   ai->family == AF_INET6? "inet6" :
282                   ai->family == AF_INET?  "inet4" : "?    ",
283                   ai->socktype, ai->protocol);
284
285           err = resolve_dns_addr (ai->addr, ai->addrlen,
286                                   (DNS_NUMERICHOST
287                                    | (opt_bracket? DNS_WITHBRACKET:0)),
288                                   &host);
289           if (err)
290             printf ("[resolve_dns_addr failed: %s]", gpg_strerror (err));
291           else
292             {
293               printf ("%s", host);
294               xfree (host);
295             }
296
297           err = resolve_dns_addr (ai->addr, ai->addrlen,
298                                   (opt_bracket? DNS_WITHBRACKET:0),
299                                   &host);
300           if (err)
301             printf ("  [resolve_dns_addr failed (2): %s]", gpg_strerror (err));
302           else
303             {
304               if (!is_ip_address (host))
305                 printf ("  (%s)", host);
306               xfree (host);
307             }
308           putchar ('\n');
309         }
310       xfree (cname);
311       free_dns_addrinfo (aibuf);
312     }
313
314   reload_dns_stuff (1); /* Release objects.  */
315
316   return 0;
317 }