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 }