keyserver/gpgkeys_finger.c

   1 /* gpgkeys_finger.c - fetch a key via finger
   2  * Copyright (C) 2004, 2005 Free Software Foundation, Inc.
   3  *
   4  * This file is part of GnuPG.
   5  *
   6  * GnuPG is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * GnuPG is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
  19  * USA.
  20  */
  21
  22 #include <config.h>
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include <stdlib.h>
  26 #include <errno.h>
  27 #include <unistd.h>
  28 #ifdef HAVE_GETOPT_H
  29 #include <getopt.h>
  30 #endif
  31
  32 #ifdef _WIN32
  33 #include <windows.h>
  34 #else
  35 #include <unistd.h>
  36 #include <sys/types.h>
  37 #include <sys/socket.h>
  38 #include <sys/time.h>
  39 #include <time.h>
  40 #include <netinet/in.h>
  41 #include <arpa/inet.h>
  42 #include <netdb.h>
  43 #endif
  44
  45 #define INCLUDED_BY_MAIN_MODULE 1
  46 #include "util.h"
  47 #include "keyserver.h"
  48 #include "ksutil.h"
  49 #include "iobuf.h"
  50
  51 #ifdef _WIN32
  52 #define sock_close(a)  closesocket(a)
  53 #else
  54 #define sock_close(a)  close(a)
  55 #endif
  56
  57 extern char *optarg;
  58 extern int optind;
  59
  60 static FILE *input,*output,*console;
  61 static struct ks_options *opt;
  62
  63 #ifdef _WIN32
  64 static void
  65 deinit_sockets (void)
  66 {
  67   WSACleanup();
  68 }
  69
  70 static void
  71 init_sockets (void)
  72 {
  73   static int initialized;
  74   static WSADATA wsdata;
  75
  76   if (initialized)
  77     return;
  78
  79   if (WSAStartup (0x0101, &wsdata) )
  80     {
  81       fprintf (console, "error initializing socket library: ec=%d\n",
  82                (int)WSAGetLastError () );
  83       return;
  84     }
  85   if (wsdata.wVersion < 0x0001)
  86     {
  87       fprintf (console, "socket library version is %x.%x - but 1.1 needed\n",
  88                LOBYTE(wsdata.wVersion), HIBYTE(wsdata.wVersion));
  89       WSACleanup();
  90       return;
  91     }
  92   atexit  (deinit_sockets);
  93   initialized = 1;
  94 }
  95 #endif /*_WIN32*/
  96
  97
  98 /* Connect to SERVER at PORT and return a file descriptor or -1 on
  99    error. */
 100 static int
 101 connect_server (const char *server, unsigned short port)
 102 {
 103   int sock = -1;
 104
 105 #ifdef _WIN32
 106   struct hostent *hp;
 107   struct sockaddr_in addr;
 108   unsigned long l;
 109
 110   init_sockets ();
 111
 112   memset (&addr, 0, sizeof addr);
 113   addr.sin_family = AF_INET;
 114   addr.sin_port = htons (port);
 115
 116   /* Win32 gethostbyname doesn't handle IP addresses internally, so we
 117      try inet_addr first on that platform only. */
 118   if ((l = inet_addr (server)) != INADDR_NONE)
 119     memcpy (&addr.sin_addr, &l, sizeof l);
 120   else if ((hp = gethostbyname (server)))
 121     {
 122       if (hp->h_addrtype != AF_INET)
 123         {
 124           fprintf (console, "gpgkeys: unknown address family for `%s'\n",
 125                    server);
 126           return -1;
 127         }
 128       if (hp->h_length != 4)
 129         {
 130           fprintf (console, "gpgkeys: illegal address length for `%s'\n",
 131                    server);
 132           return -1;
 133         }
 134       memcpy (&addr.sin_addr, hp->h_addr, hp->h_length);
 135     }
 136   else
 137     {
 138       fprintf (console, "gpgkeys: host `%s' not found: ec=%d\n",
 139                server, (int)WSAGetLastError ());
 140       return -1;
 141     }
 142
 143   sock = socket (AF_INET, SOCK_STREAM, 0);
 144   if (sock == INVALID_SOCKET)
 145     {
 146       fprintf (console, "gpgkeys: error creating socket: ec=%d\n",
 147                (int)WSAGetLastError ());
 148       return -1;
 149     }
 150
 151   if (connect (sock, (struct sockaddr *)&addr, sizeof addr))
 152     {
 153       fprintf (console, "gpgkeys: error connecting `%s': ec=%d\n",
 154                server, (int)WSAGetLastError ());
 155       sock_close (sock);
 156       return -1;
 157     }
 158
 159 #else
 160
 161   struct sockaddr_in addr;
 162   struct hostent *host;
 163
 164   addr.sin_family = AF_INET;
 165   addr.sin_port = htons (port);
 166   host = gethostbyname ((char*)server);
 167   if (!host)
 168     {
 169       fprintf (console, "gpgkeys: host `%s' not found: %s\n",
 170                server, strerror (errno));
 171       return -1;
 172     }
 173
 174   addr.sin_addr = *(struct in_addr*)host->h_addr;
 175
 176   sock = socket (AF_INET, SOCK_STREAM, 0);
 177   if (sock == -1)
 178     {
 179       fprintf (console, "gpgkeys: error creating socket: %s\n",
 180                strerror (errno));
 181       return -1;
 182     }
 183
 184   if (connect (sock, (struct sockaddr *)&addr, sizeof addr) == -1)
 185     {
 186       fprintf (console, "gpgkeys: error connecting `%s': %s\n",
 187                server, strerror (errno));
 188       close (sock);
 189       return -1;
 190     }
 191 #endif
 192
 193   return sock;
 194 }
 195
 196 static int
 197 write_server (int sock, const char *data, size_t length)
 198 {
 199   int nleft;
 200
 201   nleft = length;
 202   while (nleft > 0)
 203     {
 204       int nwritten;
 205
 206 #ifdef _WIN32
 207       nwritten = send (sock, data, nleft, 0);
 208       if ( nwritten == SOCKET_ERROR )
 209         {
 210           fprintf (console, "gpgkeys: write failed: ec=%d\n",
 211                    (int)WSAGetLastError ());
 212           return -1;
 213         }
 214 #else
 215       nwritten = write (sock, data, nleft);
 216       if (nwritten == -1)
 217         {
 218           if (errno == EINTR)
 219             continue;
 220           if (errno == EAGAIN)
 221             {
 222               struct timeval tv;
 223
 224               tv.tv_sec =  0;
 225               tv.tv_usec = 50000;
 226               select(0, NULL, NULL, NULL, &tv);
 227               continue;
 228             }
 229           fprintf (console, "gpgkeys: write failed: %s\n", strerror(errno));
 230           return -1;
 231         }
 232 #endif
 233       nleft -=nwritten;
 234       data += nwritten;
 235     }
 236
 237   return 0;
 238 }
 239
 240
 241 /* Send the finger REQUEST to the server.  Returns 0 and a file descriptor
 242    in R_SOCK if the request was sucessful. */
 243 static int
 244 send_request (const char *request, int *r_sock)
 245 {
 246   char *server;
 247   char *name;
 248   int sock;
 249
 250   *r_sock = -1;
 251   name = strdup (request);
 252   if (!name)
 253     {
 254       fprintf(console,"gpgkeys: out of memory\n");
 255       return KEYSERVER_NO_MEMORY;
 256     }
 257
 258   server = strchr (name, '@');
 259   if (!server)
 260     {
 261       fprintf (console, "gpgkeys: no name included in request\n");
 262       free (name);
 263       return KEYSERVER_GENERAL_ERROR;
 264     }
 265   *server++ = 0;
 266
 267   sock = connect_server (server, 79);
 268   if (sock == -1)
 269     {
 270       free (name);
 271       return KEYSERVER_UNREACHABLE;
 272     }
 273
 274   if (write_server (sock, name, strlen (name))
 275       || write_server (sock, "\r\n", 2))
 276     {
 277       free (name);
 278       sock_close (sock);
 279       return KEYSERVER_GENERAL_ERROR;
 280     }
 281   free (name);
 282   *r_sock = sock;
 283   return 0;
 284 }
 285
 286
 287
 288 static int
 289 get_key (char *getkey)
 290 {
 291   int rc;
 292   int sock;
 293   iobuf_t fp_read;
 294   unsigned int maxlen, buflen, gotit=0;
 295   byte *line = NULL;
 296
 297   if (strncmp (getkey,"0x",2)==0)
 298     getkey+=2;
 299
 300   /* Frankly we don't know what keys the server will return; we
 301      indicated the requested key anyway. */
 302   fprintf(output,"KEY 0x%s BEGIN\n",getkey);
 303
 304   rc=send_request(opt->opaque,&sock);
 305   if(rc)
 306     {
 307       fprintf(output,"KEY 0x%s FAILED %d\n",getkey, rc);
 308       sock_close (sock);
 309       return KEYSERVER_OK;
 310     }
 311
 312   /* Hmmm, we use iobuf here only to cope with Windows socket
 313      peculiarities (we can't used fdopen).  */
 314   fp_read = iobuf_sockopen (sock , "r");
 315   if (!fp_read)
 316     {
 317       fprintf(output,"KEY 0x%s FAILED %d\n",getkey, KEYSERVER_INTERNAL_ERROR);
 318       sock_close (sock);
 319       return KEYSERVER_OK;
 320     }
 321
 322   while ( iobuf_read_line ( fp_read, &line, &buflen, &maxlen))
 323     {
 324       maxlen=1024;
 325
 326       if(gotit)
 327         {
 328           print_nocr(output, (const char*)line);
 329           if (!strncmp((char*)line,END,strlen(END)))
 330             break;
 331         }
 332       else if(!strncmp((char*)line,BEGIN,strlen(BEGIN)))
 333         {
 334           print_nocr(output, (const char*)line);
 335           gotit=1;
 336         }
 337     }
 338
 339   if(gotit)
 340     fprintf (output,"KEY 0x%s END\n", getkey);
 341   else
 342     {
 343       fprintf(console,"gpgkeys: no key data found for finger:%s\n",
 344               opt->opaque);
 345       fprintf(output,"KEY 0x%s FAILED %d\n",getkey,KEYSERVER_KEY_NOT_FOUND);
 346     }
 347
 348   xfree(line);
 349   iobuf_close (fp_read);
 350
 351   return KEYSERVER_OK;
 352 }
 353
 354
 355 static void
 356 show_help (FILE *fp)
 357 {
 358   fprintf (fp,"-h\thelp\n");
 359   fprintf (fp,"-V\tversion\n");
 360   fprintf (fp,"-o\toutput to this file\n");
 361 }
 362
 363 int
 364 main(int argc,char *argv[])
 365 {
 366   int arg,ret=KEYSERVER_INTERNAL_ERROR;
 367   char line[MAX_LINE];
 368   char *thekey=NULL;
 369
 370   console=stderr;
 371
 372   /* Kludge to implement standard GNU options.  */
 373   if (argc > 1 && !strcmp (argv[1], "--version"))
 374     {
 375       fputs ("gpgkeys_finger (GnuPG) " VERSION"\n", stdout);
 376       return 0;
 377     }
 378   else if (argc > 1 && !strcmp (argv[1], "--help"))
 379     {
 380       show_help (stdout);
 381       return 0;
 382     }
 383
 384   while((arg=getopt(argc,argv,"hVo:"))!=-1)
 385     switch(arg)
 386       {
 387       default:
 388       case 'h':
 389         show_help (console);
 390         return KEYSERVER_OK;
 391
 392       case 'V':
 393         fprintf(stdout,"%d\n%s\n",KEYSERVER_PROTO_VERSION,VERSION);
 394         return KEYSERVER_OK;
 395
 396       case 'o':
 397         output=fopen(optarg,"w");
 398         if(output==NULL)
 399           {
 400             fprintf(console,"gpgkeys: Cannot open output file `%s': %s\n",
 401                     optarg,strerror(errno));
 402             return KEYSERVER_INTERNAL_ERROR;
 403           }
 404
 405         break;
 406       }
 407
 408   if(argc>optind)
 409     {
 410       input=fopen(argv[optind],"r");
 411       if(input==NULL)
 412         {
 413           fprintf(console,"gpgkeys: Cannot open input file `%s': %s\n",
 414                   argv[optind],strerror(errno));
 415           return KEYSERVER_INTERNAL_ERROR;
 416         }
 417     }
 418
 419   if(input==NULL)
 420     input=stdin;
 421
 422   if(output==NULL)
 423     output=stdout;
 424
 425   opt=init_ks_options();
 426   if(!opt)
 427     return KEYSERVER_NO_MEMORY;
 428
 429   /* Get the command and info block */
 430
 431   while(fgets(line,MAX_LINE,input)!=NULL)
 432     {
 433       int err;
 434
 435       if(line[0]=='\n')
 436         break;
 437
 438       err=parse_ks_options(line,opt);
 439       if(err>0)
 440         {
 441           ret=err;
 442           goto fail;
 443         }
 444       else if(err==0)
 445         continue;
 446     }
 447
 448   if(opt->host)
 449     {
 450       fprintf(console,"gpgkeys: finger://relay/user syntax is not"
 451               " supported.  Use finger:user instead.\n");
 452       ret=KEYSERVER_NOT_SUPPORTED;
 453       goto fail;
 454     }
 455
 456   if(opt->timeout && register_timeout()==-1)
 457     {
 458       fprintf(console,"gpgkeys: unable to register timeout handler\n");
 459       return KEYSERVER_INTERNAL_ERROR;
 460     }
 461
 462   /* If it's a GET or a SEARCH, the next thing to come in is the
 463      keyids.  If it's a SEND, then there are no keyids. */
 464
 465   if(opt->action==KS_GET)
 466     {
 467       /* Eat the rest of the file */
 468       for(;;)
 469         {
 470           if(fgets(line,MAX_LINE,input)==NULL)
 471             break;
 472           else
 473             {
 474               if(line[0]=='\n' || line[0]=='\0')
 475                 break;
 476
 477               if(!thekey)
 478                 {
 479                   thekey=strdup(line);
 480                   if(!thekey)
 481                     {
 482                       fprintf(console,"gpgkeys: out of memory while "
 483                               "building key list\n");
 484                       ret=KEYSERVER_NO_MEMORY;
 485                       goto fail;
 486                     }
 487
 488                   /* Trim the trailing \n */
 489                   thekey[strlen(line)-1]='\0';
 490                 }
 491             }
 492         }
 493     }
 494   else
 495     {
 496       fprintf(console,
 497               "gpgkeys: this keyserver type only supports key retrieval\n");
 498       goto fail;
 499     }
 500
 501   if(!thekey || !opt->opaque)
 502     {
 503       fprintf(console,"gpgkeys: invalid keyserver instructions\n");
 504       goto fail;
 505     }
 506
 507   /* Send the response */
 508
 509   fprintf(output,"VERSION %d\n",KEYSERVER_PROTO_VERSION);
 510   fprintf(output,"PROGRAM %s\n\n",VERSION);
 511
 512   if(opt->verbose>1)
 513     {
 514       fprintf(console,"User:\t\t%s\n",opt->opaque);
 515       fprintf(console,"Command:\tGET\n");
 516     }
 517
 518   set_timeout(opt->timeout);
 519
 520   ret=get_key(thekey);
 521
 522  fail:
 523
 524   free(thekey);
 525
 526   if(input!=stdin)
 527     fclose(input);
 528
 529   if(output!=stdout)
 530     fclose(output);
 531
 532   free_ks_options(opt);
 533
 534   return ret;
 535 }