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_cert = 0;
  55   int opt_srv = 0;
  56   int opt_bracket = 0;
  57   int opt_cname = 0;
  58   char const *name = NULL;
  59
  60   gpgrt_init ();
  61   log_set_prefix (PGM, GPGRT_LOG_WITH_PREFIX);
  62   if (argc)
  63     { argc--; argv++; }
  64   while (argc && last_argc != argc )
  65     {
  66       last_argc = argc;
  67       if (!strcmp (*argv, "--"))
  68         {
  69           argc--; argv++;
  70           break;
  71         }
  72       else if (!strcmp (*argv, "--help"))
  73         {
  74           fputs ("usage: " PGM " [HOST]\n"
  75                  "Options:\n"
  76                  "  --verbose           print timings etc.\n"
  77                  "  --debug             flyswatter\n"
  78                  "  --standard-resolver use the system's resolver\n"
  79                  "  --use-tor           use Tor\n"
  80                  "  --new-circuit       use a new Tor circuit\n"
  81                  "  --bracket           enclose v6 addresses in brackets\n"
  82                  "  --cert              lookup a CERT RR\n"
  83                  "  --srv               lookup a SRV RR\n"
  84                  "  --cname             lookup a CNAME RR\n"
  85                  "  --timeout SECONDS   timeout after SECONDS\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, "--standard-resolver"))
 106         {
 107           enable_standard_resolver (1);
 108           argc--; argv++;
 109         }
 110       else if (!strcmp (*argv, "--recursive-resolver"))
 111         {
 112           enable_recursive_resolver (1);
 113           argc--; argv++;
 114         }
 115       else if (!strcmp (*argv, "--bracket"))
 116         {
 117           opt_bracket = 1;
 118           argc--; argv++;
 119         }
 120       else if (!strcmp (*argv, "--cert"))
 121         {
 122           any_options = opt_cert = 1;
 123           argc--; argv++;
 124         }
 125       else if (!strcmp (*argv, "--srv"))
 126         {
 127           any_options = opt_srv = 1;
 128           argc--; argv++;
 129         }
 130       else if (!strcmp (*argv, "--cname"))
 131         {
 132           any_options = opt_cname = 1;
 133           argc--; argv++;
 134         }
 135       else if (!strcmp (*argv, "--timeout"))
 136         {
 137           argc--; argv++;
 138           if (argc)
 139             {
 140               set_dns_timeout (atoi (*argv));
 141               argc--; argv++;
 142             }
 143         }
 144       else if (!strncmp (*argv, "--", 2))
 145         {
 146           fprintf (stderr, PGM ": unknown option '%s'\n", *argv);
 147           exit (1);
 148         }
 149     }
 150
 151   if (!argc && !any_options)
 152     {
 153       opt_cert = 1;
 154       name = "simon.josefsson.org";
 155     }
 156   else if (argc == 1)
 157     name = *argv;
 158   else
 159     {
 160       fprintf (stderr, PGM ": none or too many host names given\n");
 161       exit (1);
 162     }
 163
 164   set_dns_verbose (verbose, debug);
 165   init_sockets ();
 166
 167   if (opt_tor)
 168     enable_dns_tormode (0);
 169
 170   if (opt_cert)
 171     {
 172       unsigned char *fpr;
 173       size_t fpr_len;
 174       char *url;
 175       void *key;
 176       size_t keylen;
 177
 178       if (verbose || any_options)
 179         printf ("CERT lookup on '%s'\n", name);
 180
 181       err = get_dns_cert (name, DNS_CERTTYPE_ANY, &key, &keylen,
 182                           &fpr, &fpr_len, &url);
 183       if (err)
 184         printf ("get_dns_cert failed: %s <%s>\n",
 185                 gpg_strerror (err), gpg_strsource (err));
 186       else if (key)
 187         {
 188           if (verbose || any_options)
 189             printf ("Key found (%u bytes)\n", (unsigned int)keylen);
 190         }
 191       else
 192         {
 193           if (fpr)
 194             {
 195               int i;
 196
 197               printf ("Fingerprint found (%d bytes): ", (int)fpr_len);
 198               for (i = 0; i < fpr_len; i++)
 199                 printf ("%02X", fpr[i]);
 200               putchar ('\n');
 201             }
 202           else
 203             printf ("No fingerprint found\n");
 204
 205           if (url)
 206             printf ("URL found: %s\n", url);
 207           else
 208             printf ("No URL found\n");
 209
 210         }
 211
 212       xfree (key);
 213       xfree (fpr);
 214       xfree (url);
 215     }
 216   else if (opt_cname)
 217     {
 218       char *cname;
 219
 220       printf ("CNAME lookup on '%s'\n", name);
 221       err = get_dns_cname (name, &cname);
 222       if (err)
 223         printf ("get_dns_cname failed: %s <%s>\n",
 224                 gpg_strerror (err), gpg_strsource (err));
 225       else
 226         {
 227           printf ("CNAME found: '%s'\n", cname);
 228         }
 229       xfree (cname);
 230     }
 231   else if (opt_srv)
 232     {
 233       struct srventry *srv;
 234       unsigned int count;
 235       int i;
 236
 237       err = get_dns_srv (name? name : "_hkp._tcp.wwwkeys.pgp.net",
 238                          &srv, &count);
 239       if (err)
 240         printf ("get_dns_srv failed: %s <%s>\n",
 241                 gpg_strerror (err), gpg_strsource (err));
 242       else
 243         {
 244           printf ("count=%u\n",count);
 245           for (i=0; i < count; i++)
 246             {
 247               printf("priority=%-8hu  ",srv[i].priority);
 248               printf("weight=%-8hu  ",srv[i].weight);
 249               printf("port=%-5hu  ",srv[i].port);
 250               printf("target=%s\n",srv[i].target);
 251             }
 252
 253           xfree(srv);
 254         }
 255     }
 256   else /* Standard lookup.  */
 257     {
 258       char *cname;
 259       dns_addrinfo_t aibuf, ai;
 260       char *host;
 261
 262       printf ("Lookup on '%s'\n", name);
 263
 264       err = resolve_dns_name (name, 0, 0, SOCK_STREAM, &aibuf, &cname);
 265       if (err)
 266         {
 267           fprintf (stderr, PGM": resolving '%s' failed: %s\n",
 268                    name, gpg_strerror (err));
 269           exit (1);
 270         }
 271
 272       if (cname)
 273         printf ("cname: %s\n", cname);
 274       for (ai = aibuf; ai; ai = ai->next)
 275         {
 276           printf ("%s %3d %3d   ",
 277                   ai->family == AF_INET6? "inet6" :
 278                   ai->family == AF_INET?  "inet4" : "?    ",
 279                   ai->socktype, ai->protocol);
 280
 281           err = resolve_dns_addr (ai->addr, ai->addrlen,
 282                                   (DNS_NUMERICHOST
 283                                    | (opt_bracket? DNS_WITHBRACKET:0)),
 284                                   &host);
 285           if (err)
 286             printf ("[resolve_dns_addr failed: %s]", gpg_strerror (err));
 287           else
 288             {
 289               printf ("%s", host);
 290               xfree (host);
 291             }
 292
 293           err = resolve_dns_addr (ai->addr, ai->addrlen,
 294                                   (opt_bracket? DNS_WITHBRACKET:0),
 295                                   &host);
 296           if (err)
 297             printf ("  [resolve_dns_addr failed (2): %s]", gpg_strerror (err));
 298           else
 299             {
 300               if (!is_ip_address (host))
 301                 printf ("  (%s)", host);
 302               xfree (host);
 303             }
 304           putchar ('\n');
 305         }
 306       xfree (cname);
 307       free_dns_addrinfo (aibuf);
 308     }
 309
 310   reload_dns_stuff (1); /* Release objects.  */
 311
 312   return 0;
 313 }