Pileus Git - ~andy/fetchmail/blob - socket.c

   1 /*
   2  * socket.c -- socket library functions
   3  *
   4  * Copyright 1998 by Eric S. Raymond.
   5  * For license terms, see the file COPYING in this directory.
   6  */
   7
   8 #include "config.h"
   9 #include <stdio.h>
  10 #include <errno.h>
  11 #include <string.h>
  12 #include <ctype.h> /* isspace() */
  13 #ifdef HAVE_MEMORY_H
  14 #include <memory.h>
  15 #endif /* HAVE_MEMORY_H */
  16 #include <sys/types.h>
  17 #include <sys/stat.h>
  18 #ifndef HAVE_NET_SOCKET_H
  19 #include <sys/socket.h>
  20 #else
  21 #include <net/socket.h>
  22 #endif
  23 #include <sys/un.h>
  24 #include <netinet/in.h>
  25 #ifdef HAVE_ARPA_INET_H
  26 #include <arpa/inet.h>
  27 #endif
  28 #include <netdb.h>
  29 #if defined(STDC_HEADERS)
  30 #include <stdlib.h>
  31 #endif
  32 #if defined(HAVE_UNISTD_H)
  33 #include <unistd.h>
  34 #endif
  35 #if defined(HAVE_STDARG_H)
  36 #include <stdarg.h>
  37 #else
  38 #include <varargs.h>
  39 #endif
  40 #if TIME_WITH_SYS_TIME
  41 # include <sys/time.h>
  42 # include <time.h>
  43 #else
  44 # if HAVE_SYS_TIME_H
  45 #  include <sys/time.h>
  46 # else
  47 #  include <time.h>
  48 # endif
  49 #endif
  50
  51 #include "socket.h"
  52 #include "fetchmail.h"
  53 #include "i18n.h"
  54
  55 /* Defines to allow BeOS and Cygwin to play nice... */
  56 #ifdef __BEOS__
  57 static char peeked;
  58 #define fm_close(a)  closesocket(a)
  59 #define fm_write(a,b,c)  send(a,b,c,0)
  60 #define fm_peek(a,b,c)   recv(a,b,c,0)
  61 #define fm_read(a,b,c)   recv(a,b,c,0)
  62 #else
  63 #define fm_close(a)  close(a)
  64 #define fm_write(a,b,c)  write(a,b,c)
  65 #define fm_peek(a,b,c)   recv(a,b,c, MSG_PEEK)
  66 #ifdef __CYGWIN__
  67 #define fm_read(a,b,c)   cygwin_read(a,b,c)
  68 static ssize_t cygwin_read(int sock, void *buf, size_t count);
  69 #else /* ! __CYGWIN__ */
  70 #define fm_read(a,b,c)   read(a,b,c)
  71 #endif /* __CYGWIN__ */
  72 #endif
  73
  74 /* We need to define h_errno only if it is not already */
  75 #ifndef h_errno
  76
  77 #ifdef HAVE_RES_SEARCH
  78 /* some versions of FreeBSD should declare this but don't */
  79 extern int h_errno;
  80 #else
  81 /* pretend we have h_errno to avoid some #ifdef's later */
  82 static int h_errno;
  83 #endif
  84
  85 #endif /* ndef h_errno */
  86
  87 #ifdef HAVE_SOCKETPAIR
  88 static char *const *parse_plugin(const char *plugin, const char *host, const char *service)
  89 {       const char **argvec;
  90         const char *c, *p;
  91         char *cp, *plugin_copy;
  92         unsigned int plugin_copy_len;
  93         unsigned int plugin_offset = 0, plugin_copy_offset = 0;
  94         unsigned int i, s = 2 * sizeof(char*), host_count = 0, service_count = 0;
  95         unsigned int plugin_len = strlen(plugin);
  96         unsigned int host_len = strlen(host);
  97         unsigned int service_len = strlen(service);
  98
  99         for (c = p = plugin; *c; c++)
 100         {       if (isspace((unsigned char)*c) && !isspace((unsigned char)*p))
 101                         s += sizeof(char*);
 102                 if (*p == '%' && *c == 'h')
 103                         host_count++;
 104                 if (*p == '%' && *c == 'p')
 105                         service_count++;
 106                 p = c;
 107         }
 108
 109         plugin_copy_len = plugin_len + host_len * host_count + service_len * service_count;
 110         plugin_copy = malloc(plugin_copy_len + 1);
 111         if (!plugin_copy)
 112         {
 113                 report(stderr, GT_("fetchmail: malloc failed\n"));
 114                 return NULL;
 115         }
 116
 117         while (plugin_copy_offset < plugin_copy_len)
 118         {       if ((plugin[plugin_offset] == '%') && (plugin[plugin_offset + 1] == 'h'))
 119                 {       strcpy(plugin_copy + plugin_copy_offset, host);
 120                         plugin_offset += 2;
 121                         plugin_copy_offset += host_len;
 122                 }
 123                 else if ((plugin[plugin_offset] == '%') && (plugin[plugin_offset + 1] == 'p'))
 124                 {       strcpy(plugin_copy + plugin_copy_offset, service);
 125                         plugin_offset += 2;
 126                         plugin_copy_offset += service_len;
 127                 }
 128                 else
 129                 {       plugin_copy[plugin_copy_offset] = plugin[plugin_offset];
 130                         plugin_offset++;
 131                         plugin_copy_offset++;
 132                 }
 133         }
 134         plugin_copy[plugin_copy_len] = 0;
 135
 136         argvec = malloc(s);
 137         if (!argvec)
 138         {
 139                 report(stderr, GT_("fetchmail: malloc failed\n"));
 140                 return NULL;
 141         }
 142         memset(argvec, 0, s);
 143         for (c = p = plugin_copy, i = 0; *c; c++)
 144         {       if ((!isspace((unsigned char)*c)) && (c == p ? 1 : isspace((unsigned char)*p))) {
 145                         argvec[i] = c;
 146                         i++;
 147                 }
 148                 p = c;
 149         }
 150         for (cp = plugin_copy; *cp; cp++)
 151         {       if (isspace((unsigned char)*cp))
 152                         *cp = 0;
 153         }
 154         return (char *const*)argvec;
 155 }
 156
 157 static int handle_plugin(const char *host,
 158                          const char *service, const char *plugin)
 159 /* get a socket mediated through a given external command */
 160 {
 161     int fds[2];
 162     char *const *argvec;
 163
 164     /*
 165      * The author of this code, Felix von Leitner <felix@convergence.de>, says:
 166      * he chose socketpair() instead of pipe() because socketpair creates
 167      * bidirectional sockets while allegedly some pipe() implementations don't.
 168      */
 169     if (socketpair(AF_UNIX,SOCK_STREAM,0,fds))
 170     {
 171         report(stderr, GT_("fetchmail: socketpair failed\n"));
 172         return -1;
 173     }
 174     switch (fork()) {
 175         case -1:
 176                 /* error */
 177                 report(stderr, GT_("fetchmail: fork failed\n"));
 178                 return -1;
 179         case 0: /* child */
 180                 /* fds[1] is the parent's end; close it for proper EOF
 181                 ** detection */
 182                 (void) close(fds[1]);
 183                 if ( (dup2(fds[0],0) == -1) || (dup2(fds[0],1) == -1) ) {
 184                         report(stderr, GT_("dup2 failed\n"));
 185                         exit(1);
 186                 }
 187                 /* fds[0] is now connected to 0 and 1; close it */
 188                 (void) close(fds[0]);
 189                 if (outlevel >= O_VERBOSE)
 190                     report(stderr, GT_("running %s (host %s service %s)\n"), plugin, host, service);
 191                 argvec = parse_plugin(plugin,host,service);
 192                 execvp(*argvec, argvec);
 193                 report(stderr, GT_("execvp(%s) failed\n"), *argvec);
 194                 exit(0);
 195                 break;
 196         default:        /* parent */
 197                 /* NOP */
 198                 break;
 199     }
 200     /* fds[0] is the child's end; close it for proper EOF detection */
 201     (void) close(fds[0]);
 202     return fds[1];
 203 }
 204 #endif /* HAVE_SOCKETPAIR */
 205
 206 #ifdef __UNUSED__
 207
 208 int SockCheckOpen(int fd)
 209 /* poll given socket; is it selectable? */
 210 {
 211     fd_set r, w, e;
 212     int rt;
 213     struct timeval tv;
 214
 215     for (;;)
 216     {
 217         FD_ZERO(&r); FD_ZERO(&w); FD_ZERO(&e);
 218         FD_SET(fd, &e);
 219
 220         tv.tv_sec = 0; tv.tv_usec = 0;
 221         rt = select(fd+1, &r, &w, &e, &tv);
 222         if (rt == -1 && (errno != EAGAIN && errno != EINTR))
 223             return 0;
 224         if (rt != -1)
 225             return 1;
 226     }
 227 }
 228 #endif /* __UNUSED__ */
 229
 230 int UnixOpen(const char *path)
 231 {
 232     int sock = -1;
 233     struct sockaddr_un ad;
 234     memset(&ad, 0, sizeof(ad));
 235     ad.sun_family = AF_UNIX;
 236     strncpy(ad.sun_path, path, sizeof(ad.sun_path)-1);
 237
 238     sock = socket( AF_UNIX, SOCK_STREAM, 0 );
 239     if (sock < 0)
 240     {
 241         h_errno = 0;
 242         return -1;
 243     }
 244
 245         /* Socket opened saved. Usefull if connect timeout
 246          * because it can be closed.
 247          */
 248         mailserver_socket_temp = sock;
 249
 250         if (connect(sock, (struct sockaddr *) &ad, sizeof(ad)) < 0)
 251     {
 252         int olderr = errno;
 253         fm_close(sock); /* don't use SockClose, no traffic yet */
 254         h_errno = 0;
 255         errno = olderr;
 256         sock = -1;
 257     }
 258
 259         /* No connect timeout, then no need to set mailserver_socket_temp */
 260         mailserver_socket_temp = -1;
 261
 262     return sock;
 263 }
 264
 265 #ifdef INET6_ENABLE
 266 int SockOpen(const char *host, const char *service,
 267              const char *plugin)
 268 {
 269     struct addrinfo *ai, *ai0, req;
 270     int i;
 271
 272 #ifdef HAVE_SOCKETPAIR
 273     if (plugin)
 274         return handle_plugin(host,service,plugin);
 275 #endif /* HAVE_SOCKETPAIR */
 276     memset(&req, 0, sizeof(struct addrinfo));
 277     req.ai_socktype = SOCK_STREAM;
 278
 279     if (getaddrinfo(host, service, &req, &ai0)) {
 280         report(stderr, GT_("fetchmail: getaddrinfo(%s.%s)\n"), host,service);
 281         return -1;
 282     }
 283
 284 #ifdef HAVE_INNER_CONNECT
 285     i = inner_connect(ai0, NULL, 0, NULL, NULL, "fetchmail", NULL);
 286     if (i >= 0)
 287         break;
 288 #else
 289
 290     i = -1;
 291     for (ai = ai0; ai; ai = ai->ai_next) {
 292         i = socket(ai->ai_family, ai->ai_socktype, 0);
 293         if (i < 0)
 294             continue;
 295
 296         /* Socket opened saved. Usefull if connect timeout
 297          * because it can be closed.
 298          */
 299         mailserver_socket_temp = i;
 300
 301         if (connect(i, (struct sockaddr *) ai->ai_addr, ai->ai_addrlen) < 0) {
 302             fm_close(i);
 303             i = -1;
 304             continue;
 305         }
 306
 307         /* No connect timeout, then no need to set mailserver_socket_temp */
 308         mailserver_socket_temp = -1;
 309
 310         break;
 311     }
 312
 313 #endif
 314
 315     freeaddrinfo(ai0);
 316
 317     return i;
 318 }
 319 #else /* INET6_ENABLE */
 320 #ifndef HAVE_INET_ATON
 321 #ifndef  INADDR_NONE
 322 #ifdef   INADDR_BROADCAST
 323 #define  INADDR_NONE    INADDR_BROADCAST
 324 #else
 325 #define  INADDR_NONE    -1
 326 #endif
 327 #endif
 328 #endif /* HAVE_INET_ATON */
 329
 330 int SockOpen(const char *host, int clientPort,
 331              const char *plugin)
 332 {
 333     int sock = -1;      /* pacify -Wall */
 334 #ifndef HAVE_INET_ATON
 335     unsigned long inaddr;
 336 #endif /* HAVE_INET_ATON */
 337     struct sockaddr_in ad, **pptr;
 338     struct hostent *hp;
 339
 340 #ifdef HAVE_SOCKETPAIR
 341     if (plugin) {
 342       char buf[10];
 343       snprintf(buf, sizeof(buf), "%d", clientPort);
 344       return handle_plugin(host,buf,plugin);
 345     }
 346 #endif /* HAVE_SOCKETPAIR */
 347
 348     memset(&ad, 0, sizeof(ad));
 349     ad.sin_family = AF_INET;
 350
 351     /* we'll accept a quad address */
 352 #ifndef HAVE_INET_ATON
 353     inaddr = inet_addr((char*)host);
 354     if (inaddr != INADDR_NONE)
 355     {
 356         memcpy(&ad.sin_addr, &inaddr, sizeof(inaddr));
 357 #else
 358     if (inet_aton(host, &ad.sin_addr))
 359     {
 360 #endif /* HAVE_INET_ATON */
 361         ad.sin_port = htons(clientPort);
 362
 363         sock = socket(AF_INET, SOCK_STREAM, 0);
 364         if (sock < 0)
 365         {
 366             h_errno = 0;
 367             return -1;
 368         }
 369
 370                 /* Socket opened saved. Usefull if connect timeout because
 371                  * it can be closed
 372                  */
 373                 mailserver_socket_temp = sock;
 374
 375         if (connect(sock, (struct sockaddr *) &ad, sizeof(ad)) < 0)
 376         {
 377             int olderr = errno;
 378             fm_close(sock);     /* don't use SockClose, no traffic yet */
 379             h_errno = 0;
 380             errno = olderr;
 381             return -1;
 382         }
 383
 384                 /* No connect timeout, then no need to set mailserver_socket_temp */
 385                 mailserver_socket_temp = -1;
 386
 387 #ifndef HAVE_INET_ATON
 388     }
 389 #else
 390     }
 391 #endif /* HAVE_INET_ATON */
 392     else {
 393         hp = gethostbyname((char*)host);
 394
 395         if (hp == NULL)
 396         {
 397             errno = 0;
 398             return -1;
 399         }
 400         /*
 401          * Add a check to make sure the address has a valid IPv4 or IPv6
 402          * length.  This prevents buffer spamming by a broken DNS.
 403          */
 404         if(hp->h_length != 4 && hp->h_length != 8)
 405         {
 406             h_errno = errno = 0;
 407             report(stderr,
 408                    GT_("fetchmail: illegal address length received for host %s\n"),host);
 409             return -1;
 410         }
 411         /*
 412          * Try all addresses of a possibly multihomed host until we get
 413          * a successful connect or until we run out of addresses.
 414          */
 415         pptr = (struct sockaddr_in **)hp->h_addr_list;
 416         for(; *pptr != NULL; pptr++)
 417         {
 418             sock = socket(AF_INET, SOCK_STREAM, 0);
 419             if (sock < 0)
 420             {
 421                 h_errno = 0;
 422                 return -1;
 423             }
 424
 425                 /* Socket opened saved. Usefull if connect timeout because
 426                  * it can be closed
 427                  */
 428                 mailserver_socket_temp = sock;
 429
 430             ad.sin_port = htons(clientPort);
 431             memcpy(&ad.sin_addr, *pptr, sizeof(struct in_addr));
 432             if (connect(sock, (struct sockaddr *) &ad, sizeof(ad)) == 0) {
 433                         /* No connect timeout, then no need to set mailserver_socket_temp */
 434                         mailserver_socket_temp = -1;
 435                         break; /* success */
 436                 }
 437             fm_close(sock);     /* don't use SockClose, no traffic yet */
 438             memset(&ad, 0, sizeof(ad));
 439             ad.sin_family = AF_INET;
 440         }
 441         if(*pptr == NULL)
 442         {
 443             int olderr = errno;
 444             fm_close(sock);     /* don't use SockClose, no traffic yet */
 445             h_errno = 0;
 446             errno = olderr;
 447             return -1;
 448         }
 449     }
 450
 451     return(sock);
 452 }
 453 #endif /* INET6_ENABLE */
 454
 455
 456 #if defined(HAVE_STDARG_H)
 457 int SockPrintf(int sock, const char* format, ...)
 458 {
 459 #else
 460 int SockPrintf(sock,format,va_alist)
 461 int sock;
 462 char *format;
 463 va_dcl {
 464 #endif
 465
 466     va_list ap;
 467     char buf[8192];
 468
 469 #if defined(HAVE_STDARG_H)
 470     va_start(ap, format) ;
 471 #else
 472     va_start(ap);
 473 #endif
 474     vsnprintf(buf, sizeof(buf), format, ap);
 475     va_end(ap);
 476     return SockWrite(sock, buf, strlen(buf));
 477
 478 }
 479
 480 #ifdef SSL_ENABLE
 481 #include <openssl/ssl.h>
 482 #include <openssl/err.h>
 483 #include <openssl/pem.h>
 484 #include <openssl/x509.h>
 485 #include <openssl/rand.h>
 486
 487 static  SSL_CTX *_ctx = NULL;
 488 static  SSL *_ssl_context[FD_SETSIZE];
 489
 490 static SSL      *SSLGetContext( int );
 491 #endif /* SSL_ENABLE */
 492
 493 int SockWrite(int sock, char *buf, int len)
 494 {
 495     int n, wrlen = 0;
 496 #ifdef  SSL_ENABLE
 497     SSL *ssl;
 498 #endif
 499
 500     while (len)
 501     {
 502 #ifdef SSL_ENABLE
 503         if( NULL != ( ssl = SSLGetContext( sock ) ) )
 504                 n = SSL_write(ssl, buf, len);
 505         else
 506 #endif /* SSL_ENABLE */
 507             n = fm_write(sock, buf, len);
 508         if (n <= 0)
 509             return -1;
 510         len -= n;
 511         wrlen += n;
 512         buf += n;
 513     }
 514     return wrlen;
 515 }
 516
 517 int SockRead(int sock, char *buf, int len)
 518 {
 519     char *newline, *bp = buf;
 520     int n;
 521 #ifdef  FORCE_STUFFING
 522     int maxavailable = 0;
 523 #endif
 524 #ifdef  SSL_ENABLE
 525     SSL *ssl;
 526 #endif
 527
 528     if (--len < 1)
 529         return(-1);
 530 #ifdef __BEOS__
 531     if (peeked != 0){
 532         (*bp) = peeked;
 533         bp++;
 534         len--;
 535         peeked = 0;
 536     }
 537 #endif
 538     do {
 539         /*
 540          * The reason for these gymnastics is that we want two things:
 541          * (1) to read \n-terminated lines,
 542          * (2) to return the true length of data read, even if the
 543          *     data coming in has embedded NULS.
 544          */
 545 #ifdef  SSL_ENABLE
 546         if( NULL != ( ssl = SSLGetContext( sock ) ) ) {
 547                 /* Hack alert! */
 548                 /* OK...  SSL_peek works a little different from MSG_PEEK
 549                         Problem is that SSL_peek can return 0 if there
 550                         is no data currently available.  If, on the other
 551                         hand, we loose the socket, we also get a zero, but
 552                         the SSL_read then SEGFAULTS!  To deal with this,
 553                         we'll check the error code any time we get a return
 554                         of zero from SSL_peek.  If we have an error, we bail.
 555                         If we don't, we read one character in SSL_read and
 556                         loop.  This should continue to work even if they
 557                         later change the behavior of SSL_peek
 558                         to "fix" this problem...  :-(   */
 559                 if ((n = SSL_peek(ssl, bp, len)) < 0) {
 560                         (void)SSL_get_error(ssl, n);
 561                         return(-1);
 562                 }
 563 #ifdef FORCE_STUFFING
 564                 maxavailable = n;
 565 #endif
 566                 if( 0 == n ) {
 567                         /* SSL_peek says no data...  Does he mean no data
 568                         or did the connection blow up?  If we got an error
 569                         then bail! */
 570                         if( 0 != ( n = SSL_get_error(ssl, n) ) ) {
 571                                 return -1;
 572                         }
 573                         /* We didn't get an error so read at least one
 574                                 character at this point and loop */
 575                         n = 1;
 576                         /* Make sure newline start out NULL!
 577                          * We don't have a string to pass through
 578                          * the strchr at this point yet */
 579                         newline = NULL;
 580                 } else if ((newline = memchr(bp, '\n', n)) != NULL)
 581                         n = newline - bp + 1;
 582                 /* Matthias Andree: SSL_read can return 0, in that case
 583                  * we must cal SSL_get_error to figure if there was
 584                  * an error or just a "no data" condition */
 585                 if ((n = SSL_read(ssl, bp, n)) <= 0) {
 586                         if ((n = SSL_get_error(ssl, n))) {
 587                                 return(-1);
 588                         }
 589                 }
 590                 /* Check for case where our single character turned out to
 591                  * be a newline...  (It wasn't going to get caught by
 592                  * the strchr above if it came from the hack...  ). */
 593                 if( NULL == newline && 1 == n && '\n' == *bp ) {
 594                         /* Got our newline - this will break
 595                                 out of the loop now */
 596                         newline = bp;
 597                 }
 598         }
 599         else
 600 #endif /* SSL_ENABLE */
 601         {
 602
 603 #ifdef __BEOS__
 604             if ((n = fm_read(sock, bp, 1)) <= 0)
 605 #else
 606             if ((n = fm_peek(sock, bp, len)) <= 0)
 607 #endif
 608                 return (-1);
 609 #ifdef FORCE_STUFFING
 610             maxavailable = n;
 611 #endif
 612             if ((newline = memchr(bp, '\n', n)) != NULL)
 613                 n = newline - bp + 1;
 614 #ifndef __BEOS__
 615             if ((n = fm_read(sock, bp, n)) == -1)
 616                 return(-1);
 617 #endif /* __BEOS__ */
 618         }
 619         bp += n;
 620         len -= n;
 621     } while
 622             (!newline && len);
 623     *bp = '\0';
 624
 625 #ifdef FORCE_STUFFING           /* too ugly to live -- besides, there's IMAP */
 626     /* OK, very weird hack coming up here:
 627      * When POP3 servers send us a message, they're supposed to
 628      * terminate the message with a line containing only a dot. To protect
 629      * against lines in the real message that might contain only a dot,
 630      * they're supposed to preface any line that starts with a dot with
 631      * an additional dot, which will be removed on the client side. That
 632      * process, called byte-stuffing (and unstuffing) is really not the
 633      * concern of this low-level routine, ordinarily, but there are some
 634      * POP servers (and maybe IMAP servers too, who knows) that fail to
 635      * do the byte-stuffing, and this routine is the best place to try to
 636      * identify and fix that fault.
 637      *
 638      * Since the DOT line is supposed to come only at the end of a
 639      * message, the implication is that right after we see it, the server
 640      * is supposed to go back to waiting for the next command. There
 641      * isn't supposed to be any more data to read after we see the dot.
 642      * THEREFORE, if we see more data to be read after something that
 643      * looks like the dot line, then probably the server is failing to
 644      * do byte-stuffing. In that case, we'll byte-pack it for them so
 645      * that the higher-level routines see things as hunky-dorey.
 646      * This is not a perfect test or fix by any means (it has an
 647      * obvious race condition, for one thing), but it should at least
 648      * reduce the nastiness that ensues when people don't know how
 649      * to write POP servers.
 650      */
 651     if ((maxavailable > (bp-buf)) &&
 652             ((((bp-buf) == 3) &&
 653               (buf[0] == '.') &&
 654               (buf[1] == '\r') &&
 655               (buf[2] == '\n')) ||
 656              (((bp-buf) == 2) &&
 657               (buf[0] == '.') &&
 658               (buf[1] == '\n')))) {
 659
 660         memmove(buf+1, buf, (bp-buf)+1);
 661         buf[0] = '.';
 662         bp++;
 663     }
 664 #endif /* FORCE_STUFFING */
 665     return bp - buf;
 666 }
 667
 668 int SockPeek(int sock)
 669 /* peek at the next socket character without actually reading it */
 670 {
 671     int n;
 672     char ch;
 673 #ifdef  SSL_ENABLE
 674     SSL *ssl;
 675 #endif
 676
 677 #ifdef  SSL_ENABLE
 678         if( NULL != ( ssl = SSLGetContext( sock ) ) ) {
 679                 n = SSL_peek(ssl, &ch, 1);
 680                 if (n < 0) {
 681                         (void)SSL_get_error(ssl, n);
 682                         return -1;
 683                 }
 684                 if( 0 == n ) {
 685                         /* This code really needs to implement a "hold back"
 686                          * to simulate a functioning SSL_peek()...  sigh...
 687                          * Has to be coordinated with the read code above.
 688                          * Next on the list todo...     */
 689
 690                         /* SSL_peek says 0...  Does that mean no data
 691                         or did the connection blow up?  If we got an error
 692                         then bail! */
 693                         if( 0 != ( n = SSL_get_error(ssl, n) ) ) {
 694                                 return -1;
 695                         }
 696
 697                         /* Haven't seen this case actually occur, but...
 698                            if the problem in SockRead can occur, this should
 699                            be possible...  Just not sure what to do here.
 700                            This should be a safe "punt" the "peek" but don't
 701                            "punt" the "session"... */
 702
 703                         return 0;       /* Give him a '\0' character */
 704                 }
 705         }
 706         else
 707 #endif /* SSL_ENABLE */
 708             n = fm_peek(sock, &ch, 1);
 709         if (n == -1)
 710                 return -1;
 711
 712 #ifdef __BEOS__
 713     peeked = ch;
 714 #endif
 715     return(ch);
 716 }
 717
 718 #ifdef SSL_ENABLE
 719
 720 static  char *_ssl_server_cname = NULL;
 721 static  int _check_fp;
 722 static  char *_check_digest;
 723 static  char *_server_label;
 724 static  int _depth0ck;
 725 static  int _prev_err;
 726
 727 SSL *SSLGetContext( int sock )
 728 {
 729         /* If SSLOpen has never initialized - just return NULL */
 730         if( NULL == _ctx )
 731                 return NULL;
 732
 733         if( sock < 0 || sock > FD_SETSIZE )
 734                 return NULL;
 735         return _ssl_context[sock];
 736 }
 737
 738
 739 /* ok_return (preverify_ok) is 1 if this stage of certificate verification
 740    passed, or 0 if it failed. This callback lets us display informative
 741    errors, and perform additional validation (e.g. CN matches) */
 742 static int SSL_verify_callback( int ok_return, X509_STORE_CTX *ctx, int strict )
 743 {
 744         char buf[257];
 745         X509 *x509_cert;
 746         int err, depth;
 747         unsigned char digest[EVP_MAX_MD_SIZE];
 748         char text[EVP_MAX_MD_SIZE * 3 + 1], *tp, *te;
 749         const EVP_MD *digest_tp;
 750         unsigned int dsz, i, esz;
 751         X509_NAME *subj, *issuer;
 752
 753         x509_cert = X509_STORE_CTX_get_current_cert(ctx);
 754         err = X509_STORE_CTX_get_error(ctx);
 755         depth = X509_STORE_CTX_get_error_depth(ctx);
 756
 757         subj = X509_get_subject_name(x509_cert);
 758         issuer = X509_get_issuer_name(x509_cert);
 759
 760         if (depth == 0 && !_depth0ck) {
 761                 _depth0ck = 1;
 762
 763                 if (outlevel == O_VERBOSE) {
 764                         if ((i = X509_NAME_get_text_by_NID(issuer, NID_organizationName, buf, sizeof(buf))) != -1) {
 765                                 report(stdout, GT_("Issuer Organization: %s\n"), buf);
 766                                 if (i >= sizeof(buf) - 1)
 767                                         report(stdout, GT_("Warning: Issuer Organization Name too long (possibly truncated).\n"));
 768                         } else
 769                                 report(stdout, GT_("Unknown Organization\n"));
 770                         if ((i = X509_NAME_get_text_by_NID(issuer, NID_commonName, buf, sizeof(buf))) != -1) {
 771                                 report(stdout, GT_("Issuer CommonName: %s\n"), buf);
 772                                 if (i >= sizeof(buf) - 1)
 773                                         report(stdout, GT_("Warning: Issuer CommonName too long (possibly truncated).\n"));
 774                         } else
 775                                 report(stdout, GT_("Unknown Issuer CommonName\n"));
 776                 }
 777                 if ((i = X509_NAME_get_text_by_NID(subj, NID_commonName, buf, sizeof(buf))) != -1) {
 778                         if (outlevel == O_VERBOSE)
 779                                 report(stdout, GT_("Server CommonName: %s\n"), buf);
 780                         if (i >= sizeof(buf) - 1) {
 781                                 /* Possible truncation. In this case, this is a DNS name, so this
 782                                  * is really bad. We do not tolerate this even in the non-strict case. */
 783                                 report(stderr, GT_("Bad certificate: Subject CommonName too long!\n"));
 784                                 return (0);
 785                         }
 786                         if (_ssl_server_cname != NULL) {
 787                                 char *p1 = buf;
 788                                 char *p2 = _ssl_server_cname;
 789                                 int n;
 790
 791                                 if (*p1 == '*') {
 792                                         ++p1;
 793                                         n = strlen(p2) - strlen(p1);
 794                                         if (n >= 0)
 795                                                 p2 += n;
 796                                 }
 797                                 if (0 != strcasecmp(p1, p2)) {
 798                                         report(stderr,
 799                                             GT_("Server CommonName mismatch: %s != %s\n"),
 800                                             buf, _ssl_server_cname );
 801                                         if (ok_return && strict)
 802                                                 return (0);
 803                                 }
 804                         } else if (ok_return) {
 805                                 report(stderr, GT_("Server name not set, could not verify certificate!\n"));
 806                                 if (strict) return (0);
 807                         }
 808                 } else {
 809                         if (outlevel == O_VERBOSE)
 810                                 report(stdout, GT_("Unknown Server CommonName\n"));
 811                         if (ok_return && strict) {
 812                                 report(stderr, GT_("Server name not specified in certificate!\n"));
 813                                 return (0);
 814                         }
 815                 }
 816                 /* Print the finger print. Note that on errors, we might print it more than once
 817                  * normally; we kluge around that by using a global variable. */
 818                 if (_check_fp) {
 819                         _check_fp = 0;
 820                         digest_tp = EVP_md5();
 821                         if (digest_tp == NULL) {
 822                                 report(stderr, GT_("EVP_md5() failed!\n"));
 823                                 return (0);
 824                         }
 825                         if (!X509_digest(x509_cert, digest_tp, digest, &dsz)) {
 826                                 report(stderr, GT_("Out of memory!\n"));
 827                                 return (0);
 828                         }
 829                         tp = text;
 830                         te = text + sizeof(text);
 831                         for (i = 0; i < dsz; i++) {
 832                                 esz = snprintf(tp, te - tp, i > 0 ? ":%02X" : "%02X", digest[i]);
 833                                 if (esz >= te - tp) {
 834                                         report(stderr, GT_("Digest text buffer too small!\n"));
 835                                         return (0);
 836                                 }
 837                                 tp += esz;
 838                         }
 839                         if (outlevel > O_NORMAL)
 840                             report(stdout, GT_("%s key fingerprint: %s\n"), _server_label, text);
 841                         if (_check_digest != NULL) {
 842                                 if (strcmp(text, _check_digest) == 0) {
 843                                     if (outlevel > O_NORMAL)
 844                                         report(stdout, GT_("%s fingerprints match.\n"), _server_label);
 845                                 } else {
 846                                     if (outlevel > O_SILENT)
 847                                         report(stderr, GT_("%s fingerprints do not match!\n"), _server_label);
 848                                     return (0);
 849                                 }
 850                         }
 851                 }
 852         }
 853
 854         if (err != X509_V_OK && err != _prev_err) {
 855                 _prev_err = err;
 856                 report(stderr, GT_("Server certificate verification error: %s\n"), X509_verify_cert_error_string(err));
 857                 /* We gave the error code, but maybe we can add some more details for debugging */
 858                 switch (err) {
 859                 case X509_V_ERR_UNABLE_TO_GET_ISSUER_CERT:
 860                         X509_NAME_oneline(issuer, buf, sizeof(buf));
 861                         buf[sizeof(buf) - 1] = '\0';
 862                         report(stderr, GT_("unknown issuer (first %d characters): %s\n"), sizeof(buf)-1, buf);
 863                         break;
 864                 }
 865         }
 866         if (!strict)
 867                 ok_return = 1;
 868         return (ok_return);
 869 }
 870
 871 static int SSL_nock_verify_callback( int ok_return, X509_STORE_CTX *ctx )
 872 {
 873         return SSL_verify_callback(ok_return, ctx, 0);
 874 }
 875
 876 static int SSL_ck_verify_callback( int ok_return, X509_STORE_CTX *ctx )
 877 {
 878         return SSL_verify_callback(ok_return, ctx, 1);
 879 }
 880
 881 /* performs initial SSL handshake over the connected socket
 882  * uses SSL *ssl global variable, which is currently defined
 883  * in this file
 884  */
 885 int SSLOpen(int sock, char *mycert, char *mykey, char *myproto, int certck, char *certpath,
 886     char *fingerprint, char *servercname, char *label)
 887 {
 888         struct stat randstat;
 889         int i;
 890
 891         SSL_load_error_strings();
 892         SSLeay_add_ssl_algorithms();
 893
 894 #ifdef SSL_ENABLE
 895         if (stat("/dev/random", &randstat)  &&
 896             stat("/dev/urandom", &randstat)) {
 897           /* Neither /dev/random nor /dev/urandom are present, so add
 898              entropy to the SSL PRNG a hard way. */
 899           for (i = 0; i < 10000  &&  ! RAND_status (); ++i) {
 900             char buf[4];
 901             struct timeval tv;
 902             gettimeofday (&tv, 0);
 903             buf[0] = tv.tv_usec & 0xF;
 904             buf[2] = (tv.tv_usec & 0xF0) >> 4;
 905             buf[3] = (tv.tv_usec & 0xF00) >> 8;
 906             buf[1] = (tv.tv_usec & 0xF000) >> 12;
 907             RAND_add (buf, sizeof buf, 0.1);
 908           }
 909         }
 910 #endif /* SSL_ENABLE */
 911
 912
 913         if( sock < 0 || sock > FD_SETSIZE ) {
 914                 report(stderr, GT_("File descriptor out of range for SSL") );
 915                 return( -1 );
 916         }
 917
 918         if( ! _ctx ) {
 919                 /* Be picky and make sure the memory is cleared */
 920                 memset( _ssl_context, 0, sizeof( _ssl_context ) );
 921                 if(myproto) {
 922                         if(!strcmp("ssl2",myproto)) {
 923                                 _ctx = SSL_CTX_new(SSLv2_client_method());
 924                         } else if(!strcmp("ssl3",myproto)) {
 925                                 _ctx = SSL_CTX_new(SSLv3_client_method());
 926                         } else if(!strcmp("tls1",myproto)) {
 927                                 _ctx = SSL_CTX_new(TLSv1_client_method());
 928                         } else if (!strcmp("ssl23",myproto)) {
 929                                 myproto = NULL;
 930                         } else {
 931                                 fprintf(stderr,GT_("Invalid SSL protocol '%s' specified, using default (SSLv23).\n"), myproto);
 932                                 myproto = NULL;
 933                         }
 934                 }
 935                 if(!myproto) {
 936                         _ctx = SSL_CTX_new(SSLv23_client_method());
 937                 }
 938                 if(_ctx == NULL) {
 939                         ERR_print_errors_fp(stderr);
 940                         return(-1);
 941                 }
 942         }
 943
 944         if (certck) {
 945                 SSL_CTX_set_verify(_ctx, SSL_VERIFY_PEER, SSL_ck_verify_callback);
 946         } else {
 947                 /* In this case, we do not fail if verification fails. However,
 948                  *  we provide the callback for output and possible fingerprint checks. */
 949                 SSL_CTX_set_verify(_ctx, SSL_VERIFY_PEER, SSL_nock_verify_callback);
 950         }
 951         if (certpath)
 952                 SSL_CTX_load_verify_locations(_ctx, NULL, certpath);
 953
 954         _ssl_context[sock] = SSL_new(_ctx);
 955
 956         if(_ssl_context[sock] == NULL) {
 957                 ERR_print_errors_fp(stderr);
 958                 return(-1);
 959         }
 960
 961         /* This static is for the verify callback */
 962         _ssl_server_cname = servercname;
 963         _server_label = label;
 964         _check_fp = 1;
 965         _check_digest = fingerprint;
 966         _depth0ck = 0;
 967         _prev_err = -1;
 968
 969         if( mycert || mykey ) {
 970
 971         /* Ok...  He has a certificate file defined, so lets declare it.  If
 972          * he does NOT have a separate certificate and private key file then
 973          * assume that it's a combined key and certificate file.
 974          */
 975                 if( !mykey )
 976                         mykey = mycert;
 977                 if( !mycert )
 978                         mycert = mykey;
 979                 SSL_use_certificate_file(_ssl_context[sock], mycert, SSL_FILETYPE_PEM);
 980                 SSL_use_RSAPrivateKey_file(_ssl_context[sock], mykey, SSL_FILETYPE_PEM);
 981         }
 982
 983         SSL_set_fd(_ssl_context[sock], sock);
 984
 985         if(SSL_connect(_ssl_context[sock]) < 1) {
 986                 ERR_print_errors_fp(stderr);
 987                 return(-1);
 988         }
 989
 990         /* Paranoia: was the callback not called as we expected? */
 991         if (!_depth0ck) {
 992                 report(stderr, GT_("Certificate/fingerprint verification was somehow skipped!\n"));
 993
 994                 if (fingerprint != NULL || certck) {
 995                         if( NULL != SSLGetContext( sock ) ) {
 996                                 /* Clean up the SSL stack */
 997                                 SSL_free( _ssl_context[sock] );
 998                                 _ssl_context[sock] = NULL;
 999                         }
1000                         return(-1);
1001                 }
1002         }
1003
1004         return(0);
1005 }
1006 #endif
1007
1008 int SockClose(int sock)
1009 /* close a socket gracefully */
1010 {
1011 #ifdef  SSL_ENABLE
1012     if( NULL != SSLGetContext( sock ) ) {
1013         /* Clean up the SSL stack */
1014         SSL_free( _ssl_context[sock] );
1015         _ssl_context[sock] = NULL;
1016     }
1017 #endif
1018
1019 #ifdef __UNUSED__
1020     /*
1021      * This hangs in RedHat 6.2 after fetchmail runs for a while a
1022      * FIN_WAIT2 comes up in netstat and fetchmail never returns from
1023      * the recv system call. (Reported from jtnews
1024      * <jtnews@bellatlantic.net>, Wed, 24 May 2000 21:26:02.)
1025      *
1026      * Half-close the connection first so the other end gets notified.
1027      *
1028      * This stops sends but allows receives (effectively, it sends a
1029      * TCP <FIN>).  */
1030     if (shutdown(sock, 1) == 0) {
1031         char ch;
1032         /* If there is any data still waiting in the queue, discard it.
1033          * Call recv() until either it returns 0 (meaning we received a FIN)
1034          * or any error occurs.  This makes sure all data sent by the other
1035          * side is acknowledged at the TCP level.
1036          */
1037         if (fm_peek(sock, &ch, 1) > 0)
1038             while (fm_read(sock, &ch, 1) > 0)
1039                 continue;
1040     }
1041 #endif /* __UNUSED__ */
1042
1043     /* if there's an error closing at this point, not much we can do */
1044     return(fm_close(sock));     /* this is guarded */
1045 }
1046
1047 #ifdef __CYGWIN__
1048 /*
1049  * Workaround Microsoft Winsock recv/WSARecv(..., MSG_PEEK) bug.
1050  * See http://sources.redhat.com/ml/cygwin/2001-08/msg00628.html
1051  * for more details.
1052  */
1053 static ssize_t cygwin_read(int sock, void *buf, size_t count)
1054 {
1055     char *bp = buf;
1056     int n = 0;
1057
1058     if ((n = read(sock, bp, count)) == -1)
1059         return(-1);
1060
1061     if (n != count) {
1062         int n2 = 0;
1063         if (outlevel >= O_VERBOSE)
1064             report(stdout, GT_("Cygwin socket read retry\n"));
1065         n2 = read(sock, bp + n, count - n);
1066         if (n2 == -1 || n + n2 != count) {
1067             report(stderr, GT_("Cygwin socket read retry failed!\n"));
1068             return(-1);
1069         }
1070     }
1071
1072     return count;
1073 }
1074 #endif /* __CYGWIN__ */
1075
1076 #ifdef MAIN
1077 /*
1078  * Use the chargen service to test input buffering directly.
1079  * You may have to uncomment the `chargen' service description in your
1080  * inetd.conf (and then SIGHUP inetd) for this to work.  */
1081 main()
1082 {
1083     int         sock = SockOpen("localhost", 19, NULL);
1084     char        buf[80];
1085
1086     while (SockRead(sock, buf, sizeof(buf)-1))
1087         SockWrite(1, buf, strlen(buf));
1088     SockClose(sock);
1089 }
1090 #endif /* MAIN */
1091
1092 /* socket.c ends here */