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 <http://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
  37 int
  38 main (int argc, char **argv)
  39 {
  40   int last_argc = -1;
  41   gpg_error_t err;
  42   int any_options = 0;
  43   int opt_tor = 0;
  44   int opt_cert = 0;
  45   int opt_srv = 0;
  46   int opt_bracket = 0;
  47   int opt_cname = 0;
  48   char const *name = NULL;
  49
  50   gpgrt_init ();
  51   log_set_prefix (PGM, GPGRT_LOG_WITH_PREFIX);
  52   if (argc)
  53     { argc--; argv++; }
  54   while (argc && last_argc != argc )
  55     {
  56       last_argc = argc;
  57       if (!strcmp (*argv, "--"))
  58         {
  59           argc--; argv++;
  60           break;
  61         }
  62       else if (!strcmp (*argv, "--help"))
  63         {
  64           fputs ("usage: " PGM " [HOST]\n"
  65                  "Options:\n"
  66                  "  --verbose         print timings etc.\n"
  67                  "  --debug           flyswatter\n"
  68                  "  --use-tor         use Tor\n"
  69                  "  --bracket         enclose v6 addresses in brackets\n"
  70                  "  --cert            lookup a CERT RR\n"
  71                  "  --srv             lookup a SRV RR\n"
  72                  "  --cname           lookup a CNAME RR\n"
  73                  , stdout);
  74           exit (0);
  75         }
  76       else if (!strcmp (*argv, "--verbose"))
  77         {
  78           verbose++;
  79           argc--; argv++;
  80         }
  81       else if (!strcmp (*argv, "--debug"))
  82         {
  83           verbose += 2;
  84           debug++;
  85           argc--; argv++;
  86         }
  87       else if (!strcmp (*argv, "--use-tor"))
  88         {
  89           opt_tor = 1;
  90           argc--; argv++;
  91         }
  92       else if (!strcmp (*argv, "--bracket"))
  93         {
  94           opt_bracket = 1;
  95           argc--; argv++;
  96         }
  97       else if (!strcmp (*argv, "--cert"))
  98         {
  99           any_options = opt_cert = 1;
 100           argc--; argv++;
 101         }
 102       else if (!strcmp (*argv, "--srv"))
 103         {
 104           any_options = opt_srv = 1;
 105           argc--; argv++;
 106         }
 107       else if (!strcmp (*argv, "--cname"))
 108         {
 109           any_options = opt_cname = 1;
 110           argc--; argv++;
 111         }
 112       else if (!strncmp (*argv, "--", 2))
 113         {
 114           fprintf (stderr, PGM ": unknown option '%s'\n", *argv);
 115           exit (1);
 116         }
 117     }
 118
 119   if (!argc && !any_options)
 120     {
 121       opt_cert = 1;
 122       name = "simon.josefsson.org";
 123     }
 124   else if (argc == 1)
 125     name = *argv;
 126   else
 127     {
 128       fprintf (stderr, PGM ": none or too many host names given\n");
 129       exit (1);
 130     }
 131
 132   if (opt_tor)
 133     {
 134       err = enable_dns_tormode ();
 135       if (err)
 136         {
 137           fprintf (stderr, "error switching into Tor mode: %s\n",
 138                    gpg_strerror (err));
 139           exit (1);
 140         }
 141     }
 142
 143   if (opt_cert)
 144     {
 145       unsigned char *fpr;
 146       size_t fpr_len;
 147       char *url;
 148       void *key;
 149       size_t keylen;
 150
 151       printf ("CERT lookup on '%s'\n", name);
 152
 153       err = get_dns_cert (name, DNS_CERTTYPE_ANY, &key, &keylen,
 154                           &fpr, &fpr_len, &url);
 155       if (err)
 156         printf ("get_dns_cert failed: %s <%s>\n",
 157                 gpg_strerror (err), gpg_strsource (err));
 158       else if (key)
 159         {
 160           printf ("Key found (%u bytes)\n", (unsigned int)keylen);
 161         }
 162       else
 163         {
 164           if (fpr)
 165             {
 166               int i;
 167
 168               printf ("Fingerprint found (%d bytes): ", (int)fpr_len);
 169               for (i = 0; i < fpr_len; i++)
 170                 printf ("%02X", fpr[i]);
 171               putchar ('\n');
 172             }
 173           else
 174             printf ("No fingerprint found\n");
 175
 176           if (url)
 177             printf ("URL found: %s\n", url);
 178           else
 179             printf ("No URL found\n");
 180
 181         }
 182
 183       xfree (key);
 184       xfree (fpr);
 185       xfree (url);
 186     }
 187   else if (opt_cname)
 188     {
 189       char *cname;
 190
 191       printf ("CNAME lookup on '%s'\n", name);
 192       err = get_dns_cname (name, &cname);
 193       if (err)
 194         printf ("get_dns_cname failed: %s <%s>\n",
 195                 gpg_strerror (err), gpg_strsource (err));
 196       else
 197         {
 198           printf ("CNAME found: '%s'\n", cname);
 199         }
 200
 201       xfree (cname);
 202     }
 203   else if (opt_srv)
 204     {
 205       struct srventry *srv;
 206       int rc,i;
 207
 208       rc=getsrv("_hkp._tcp.wwwkeys.pgp.net",&srv);
 209       printf("Count=%d\n\n",rc);
 210       for(i=0;i<rc;i++)
 211         {
 212           printf("priority=%hu\n",srv[i].priority);
 213           printf("weight=%hu\n",srv[i].weight);
 214           printf("port=%hu\n",srv[i].port);
 215           printf("target=%s\n",srv[i].target);
 216           printf("\n");
 217         }
 218
 219       xfree(srv);
 220     }
 221   else /* Standard lookup.  */
 222     {
 223       char *cname;
 224       dns_addrinfo_t aibuf, ai;
 225       char *host;
 226
 227       printf ("Lookup on '%s'\n", name);
 228
 229       err = resolve_dns_name (name, 0, 0, SOCK_STREAM, &aibuf, &cname);
 230       if (err)
 231         {
 232           fprintf (stderr, PGM": resolving '%s' failed: %s\n",
 233                    name, gpg_strerror (err));
 234           exit (1);
 235         }
 236
 237       if (cname)
 238         printf ("cname: %s\n", cname);
 239       for (ai = aibuf; ai; ai = ai->next)
 240         {
 241           printf ("%s %3d %3d   ",
 242                   ai->family == AF_INET6? "inet6" :
 243                   ai->family == AF_INET?  "inet4" : "?    ",
 244                   ai->socktype, ai->protocol);
 245
 246           err = resolve_dns_addr (ai->addr, ai->addrlen,
 247                                   (DNS_NUMERICHOST
 248                                    | (opt_bracket? DNS_WITHBRACKET:0)),
 249                                   &host);
 250           if (err)
 251             printf ("[resolve_dns_addr failed: %s]", gpg_strerror (err));
 252           else
 253             {
 254               printf ("%s", host);
 255               xfree (host);
 256             }
 257
 258           err = resolve_dns_addr (ai->addr, ai->addrlen,
 259                                   (opt_bracket? DNS_WITHBRACKET:0),
 260                                   &host);
 261           if (err)
 262             printf ("[resolve_dns_addr failed (2): %s]", gpg_strerror (err));
 263           else
 264             {
 265               if (!is_ip_address (host))
 266                 printf ("  (%s)", host);
 267               xfree (host);
 268             }
 269           putchar ('\n');
 270         }
 271       xfree (cname);
 272       free_dns_addrinfo (aibuf);
 273     }
 274
 275
 276   return 0;
 277 }