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