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 #ifdef HAVE_MEMORY_H
  13 #include <memory.h>
  14 #endif /* HAVE_MEMORY_H */
  15 #include <sys/types.h>
  16 #include <sys/socket.h>
  17 #include <netinet/in.h>
  18 #include <arpa/inet.h>
  19 #include <netdb.h>
  20 #if defined(STDC_HEADERS)
  21 #include <stdlib.h>
  22 #endif
  23 #if defined(HAVE_UNISTD_H)
  24 #include <unistd.h>
  25 #endif
  26 #if defined(HAVE_STDARG_H)
  27 #include <stdarg.h>
  28 #else
  29 #include <varargs.h>
  30 #endif
  31 #include "socket.h"
  32 #include "fetchmail.h"
  33 #include "i18n.h"
  34
  35 /* We need to define h_errno only if it is not already */
  36 #ifndef h_errno
  37
  38 #ifdef HAVE_RES_SEARCH
  39 /* some versions of FreeBSD should declare this but don't */
  40 extern int h_errno;
  41 #else
  42 /* pretend we have h_errno to avoid some #ifdef's later */
  43 static int h_errno;
  44 #endif
  45
  46 #endif /* ndef h_errno */
  47
  48 #if NET_SECURITY
  49 #include <net/security.h>
  50 #endif /* NET_SECURITY */
  51
  52 #ifdef HAVE_SOCKETPAIR
  53 static int handle_plugin(const char *host,
  54                          const char *service, const char *plugin)
  55 /* get a socket mediated through a given external command */
  56 {
  57     int fds[2];
  58     if (socketpair(AF_UNIX,SOCK_STREAM,0,fds))
  59     {
  60         report(stderr, _("fetchmail: socketpair failed\n"));
  61         return -1;
  62     }
  63     switch (fork()) {
  64         case -1:
  65                 /* error */
  66                 report(stderr, _("fetchmail: fork failed\n"));
  67                 return -1;
  68                 break;
  69         case 0: /* child */
  70                 /* fds[1] is the parent's end; close it for proper EOF
  71                 ** detection */
  72                 (void) close(fds[1]);
  73                 if ( (dup2(fds[0],0) == -1) || (dup2(fds[0],1) == -1) ) {
  74                         report(stderr, _("dup2 failed\n"));
  75                         exit(1);
  76                 }
  77                 /* fds[0] is now connected to 0 and 1; close it */
  78                 (void) close(fds[0]);
  79                 if (outlevel >= O_VERBOSE)
  80                     report(stderr, _("running %s %s %s\n"), plugin, host, service);
  81                 execlp(plugin,plugin,host,service,0);
  82                 report(stderr, _("execl(%s) failed\n"), plugin);
  83                 exit(0);
  84                 break;
  85         default:        /* parent */
  86                 /* NOP */
  87                 break;
  88     }
  89     /* fds[0] is the child's end; close it for proper EOF detection */
  90     (void) close(fds[0]);
  91     return fds[1];
  92 }
  93 #endif /* HAVE_SOCKETPAIR */
  94
  95 #ifdef __UNUSED__
  96 #include <sys/time.h>
  97
  98 int SockCheckOpen(int fd)
  99 /* poll given socket; is it selectable? */
 100 {
 101     fd_set r, w, e;
 102     int rt;
 103     struct timeval tv;
 104
 105     for (;;)
 106     {
 107         FD_ZERO(&r); FD_ZERO(&w); FD_ZERO(&e);
 108         FD_SET(fd, &e);
 109
 110         tv.tv_sec = 0; tv.tv_usec = 0;
 111         rt = select(fd+1, &r, &w, &e, &tv);
 112         if (rt == -1 && (errno != EAGAIN && errno != EINTR))
 113             return 0;
 114         if (rt != -1)
 115             return 1;
 116     }
 117 }
 118 #endif /* __UNUSED__ */
 119
 120 #if INET6
 121 int SockOpen(const char *host, const char *service, const char *options,
 122              const char *plugin)
 123 {
 124     struct addrinfo *ai, req;
 125 #if NET_SECURITY
 126     void *request = NULL;
 127     int requestlen;
 128 #endif /* NET_SECURITY */
 129
 130 #ifdef HAVE_SOCKETPAIR
 131     if (plugin)
 132         return handle_plugin(host,service,plugin);
 133 #endif /* HAVE_SOCKETPAIR */
 134     memset(&req, 0, sizeof(struct addrinfo));
 135     req.ai_socktype = SOCK_STREAM;
 136
 137     if (getaddrinfo(host, service, &req, &ai)) {
 138         report(stderr, _("fetchmail: getaddrinfo(%s.%s)\n"), host,service);
 139         return -1;
 140     };
 141
 142 #if NET_SECURITY
 143     if (!options)
 144         requestlen = 0;
 145     else
 146         if (net_security_strtorequest((char *)options, &request, &requestlen))
 147             goto ret;
 148
 149     i = inner_connect(ai, request, requestlen, NULL, NULL, "fetchmail", NULL);
 150     if (request)
 151         free(request);
 152
 153  ret:
 154 #else /* NET_SECURITY */
 155     i = inner_connect(ai, NULL, 0, NULL, NULL, "fetchmail", NULL);
 156 #endif /* NET_SECURITY */
 157
 158     freeaddrinfo(ai);
 159
 160     return i;
 161 };
 162 #else /* INET6 */
 163 #ifndef HAVE_INET_ATON
 164 #ifndef  INADDR_NONE
 165 #ifdef   INADDR_BROADCAST
 166 #define  INADDR_NONE    INADDR_BROADCAST
 167 #else
 168 #define  INADDR_NONE    -1
 169 #endif
 170 #endif
 171 #endif /* HAVE_INET_ATON */
 172
 173 int SockOpen(const char *host, int clientPort, const char *options,
 174              const char *plugin)
 175 {
 176     int sock = -1;      /* pacify -Wall */
 177 #ifndef HAVE_INET_ATON
 178     unsigned long inaddr;
 179 #endif /* HAVE_INET_ATON */
 180     struct sockaddr_in ad, **pptr;
 181     struct hostent *hp;
 182
 183 #ifdef HAVE_SOCKETPAIR
 184     if (plugin) {
 185       char buf[10];
 186       sprintf(buf,"%d",clientPort);
 187       return handle_plugin(host,buf,plugin);
 188     }
 189 #endif /* HAVE_SOCKETPAIR */
 190
 191     memset(&ad, 0, sizeof(ad));
 192     ad.sin_family = AF_INET;
 193
 194     /* we'll accept a quad address */
 195 #ifndef HAVE_INET_ATON
 196     inaddr = inet_addr(host);
 197     if (inaddr != INADDR_NONE)
 198         memcpy(&ad.sin_addr, &inaddr, sizeof(inaddr));
 199     else
 200 #else
 201     if (!inet_aton(host, &ad.sin_addr))
 202 #endif /* HAVE_INET_ATON */
 203     {
 204         hp = gethostbyname(host);
 205
 206         if (hp == NULL)
 207         {
 208             errno = 0;
 209             return -1;
 210         }
 211         /*
 212          * Add a check to make sure the address has a valid IPv4 or IPv6
 213          * length.  This prevents buffer spamming by a broken DNS.
 214          */
 215         if(hp->h_length != 4 && hp->h_length != 8)
 216         {
 217             h_errno = errno = 0;
 218             report(stderr,
 219                    _("fetchmail: illegal address length received for host %s\n"),host);
 220             return -1;
 221         }
 222         /*
 223          * Try all addresses of a possibly multihomed host until we get
 224          * a successful connect or until we run out of addresses.
 225          */
 226         pptr = (struct sockaddr_in **)hp->h_addr_list;
 227         for(; *pptr != NULL; pptr++)
 228         {
 229             sock = socket(AF_INET, SOCK_STREAM, 0);
 230             if (sock < 0)
 231             {
 232                 h_errno = 0;
 233                 return -1;
 234             }
 235             ad.sin_port = htons(clientPort);
 236             memcpy(&ad.sin_addr, *pptr, sizeof(struct in_addr));
 237             if (connect(sock, (struct sockaddr *) &ad, sizeof(ad)) == 0)
 238                 break; /* success */
 239             close(sock);
 240             memset(&ad, 0, sizeof(ad));
 241             ad.sin_family = AF_INET;
 242         }
 243         if(*pptr == NULL)
 244         {
 245             int olderr = errno;
 246             close(sock);
 247             h_errno = 0;
 248             errno = olderr;
 249             return -1;
 250         }
 251     } else {
 252         ad.sin_port = htons(clientPort);
 253
 254         sock = socket(AF_INET, SOCK_STREAM, 0);
 255         if (sock < 0)
 256         {
 257             h_errno = 0;
 258             return -1;
 259         }
 260         if (connect(sock, (struct sockaddr *) &ad, sizeof(ad)) < 0)
 261         {
 262             int olderr = errno;
 263             close(sock);
 264             h_errno = 0;
 265             errno = olderr;
 266             return -1;
 267         }
 268     }
 269     return(sock);
 270 }
 271 #endif /* INET6 */
 272
 273
 274 #if defined(HAVE_STDARG_H)
 275 int SockPrintf(int sock, const char* format, ...)
 276 {
 277 #else
 278 int SockPrintf(sock,format,va_alist)
 279 int sock;
 280 char *format;
 281 va_dcl {
 282 #endif
 283
 284     va_list ap;
 285     char buf[8192];
 286
 287 #if defined(HAVE_STDARG_H)
 288     va_start(ap, format) ;
 289 #else
 290     va_start(ap);
 291 #endif
 292 #ifdef HAVE_VSNPRINTF
 293     vsnprintf(buf, sizeof(buf), format, ap);
 294 #else
 295     vsprintf(buf, format, ap);
 296 #endif
 297     va_end(ap);
 298     return SockWrite(sock, buf, strlen(buf));
 299
 300 }
 301
 302 #ifdef SSL_ENABLE
 303 #include "ssl.h"
 304 #include "err.h"
 305 #include "pem.h"
 306 #include "x509.h"
 307
 308 static  SSL_CTX *_ctx = NULL;
 309 static  SSL *_ssl_context[FD_SETSIZE];
 310
 311 SSL     *SSLGetContext( int );
 312 #endif /* SSL_ENABLE */
 313
 314 int SockWrite(int sock, char *buf, int len)
 315 {
 316     int n, wrlen = 0;
 317 #ifdef  SSL_ENABLE
 318     SSL *ssl;
 319 #endif
 320
 321     while (len)
 322     {
 323 #ifdef SSL_ENABLE
 324         if( NULL != ( ssl = SSLGetContext( sock ) ) )
 325                 n = SSL_write(ssl, buf, len);
 326         else
 327                 n = write(sock, buf, len);
 328 #else
 329         n = write(sock, buf, len);
 330 #endif
 331         if (n <= 0)
 332             return -1;
 333         len -= n;
 334         wrlen += n;
 335         buf += n;
 336     }
 337     return wrlen;
 338 }
 339
 340 int SockRead(int sock, char *buf, int len)
 341 {
 342     char *newline, *bp = buf;
 343     int n;
 344 #ifdef  SSL_ENABLE
 345     SSL *ssl;
 346 #endif
 347
 348     if (--len < 1)
 349         return(-1);
 350     do {
 351         /*
 352          * The reason for these gymnastics is that we want two things:
 353          * (1) to read \n-terminated lines,
 354          * (2) to return the true length of data read, even if the
 355          *     data coming in has embedded NULS.
 356          */
 357 #ifdef  SSL_ENABLE
 358         if( NULL != ( ssl = SSLGetContext( sock ) ) ) {
 359                 /* Hack alert! */
 360                 /* OK...  SSL_peek works a little different from MSG_PEEK
 361                         Problem is that SSL_peek can return 0 if there
 362                         is no data currently available.  If, on the other
 363                         hand, we loose the socket, we also get a zero, but
 364                         the SSL_read then SEGFAULTS!  To deal with this,
 365                         we'll check the error code any time we get a return
 366                         of zero from SSL_peek.  If we have an error, we bail.
 367                         If we don't, we read one character in SSL_read and
 368                         loop.  This should continue to work even if they
 369                         later change the behavior of SSL_peek
 370                         to "fix" this problem...  :-(   */
 371                 if ((n = SSL_peek(ssl, bp, len)) < 0) {
 372                         return(-1);
 373                 }
 374                 if( 0 == n ) {
 375                         /* SSL_peek says no data...  Does he mean no data
 376                         or did the connection blow up?  If we got an error
 377                         then bail! */
 378                         if( 0 != ( n = ERR_get_error() ) ) {
 379                                 return -1;
 380                         }
 381                         /* We didn't get an error so read at least one
 382                                 character at this point and loop */
 383                         n = 1;
 384                         /* Make sure newline start out NULL!
 385                          * We don't have a string to pass through
 386                          * the strchr at this point yet */
 387                         newline = NULL;
 388                 } else if ((newline = memchr(bp, '\n', n)) != NULL)
 389                         n = newline - bp + 1;
 390                 if ((n = SSL_read(ssl, bp, n)) == -1) {
 391                         return(-1);
 392                 }
 393                 /* Check for case where our single character turned out to
 394                  * be a newline...  (It wasn't going to get caught by
 395                  * the strchr above if it came from the hack...  ). */
 396                 if( NULL == newline && 1 == n && '\n' == *bp ) {
 397                         /* Got our newline - this will break
 398                                 out of the loop now */
 399                         newline = bp;
 400                 }
 401         } else {
 402                 if ((n = recv(sock, bp, len, MSG_PEEK)) <= 0)
 403                         return(-1);
 404                 if ((newline = memchr(bp, '\n', n)) != NULL)
 405                         n = newline - bp + 1;
 406                 if ((n = read(sock, bp, n)) == -1)
 407                         return(-1);
 408         }
 409 #else
 410         if ((n = recv(sock, bp, len, MSG_PEEK)) <= 0)
 411             return(-1);
 412         if ((newline = memchr(bp, '\n', n)) != NULL)
 413             n = newline - bp + 1;
 414         if ((n = read(sock, bp, n)) == -1)
 415             return(-1);
 416 #endif
 417         bp += n;
 418         len -= n;
 419     } while
 420             (!newline && len);
 421     *bp = '\0';
 422     return bp - buf;
 423 }
 424
 425 int SockPeek(int sock)
 426 /* peek at the next socket character without actually reading it */
 427 {
 428     int n;
 429     char ch;
 430 #ifdef  SSL_ENABLE
 431     SSL *ssl;
 432 #endif
 433
 434 #ifdef  SSL_ENABLE
 435         if( NULL != ( ssl = SSLGetContext( sock ) ) ) {
 436                 n = SSL_peek(ssl, &ch, 1);
 437                 if( 0 == n ) {
 438                         /* This code really needs to implement a "hold back"
 439                          * to simulate a functioning SSL_peek()...  sigh...
 440                          * Has to be coordinated with the read code above.
 441                          * Next on the list todo...     */
 442
 443                         /* SSL_peek says no data...  Does he mean no data
 444                         or did the connection blow up?  If we got an error
 445                         then bail! */
 446                         if( 0 != ( n = ERR_get_error() ) ) {
 447                                 return -1;
 448                         }
 449
 450                         /* Haven't seen this case actually occur, but...
 451                            if the problem in SockRead can occur, this should
 452                            be possible...  Just not sure what to do here.
 453                            This should be a safe "punt" the "peek" but don't
 454                            "punt" the "session"... */
 455
 456                         return 0;       /* Give him a '\0' character */
 457                 }
 458         } else {
 459                 n = recv(sock, &ch, 1, MSG_PEEK);
 460         }
 461 #else
 462         n = recv(sock, &ch, 1, MSG_PEEK);
 463 #endif
 464         if (n == -1)
 465                 return -1;
 466         else
 467                 return(ch);
 468 }
 469
 470 #ifdef SSL_ENABLE
 471
 472 static  char *_ssl_server_cname = NULL;
 473
 474 SSL *SSLGetContext( int sock )
 475 {
 476         /* If SSLOpen has never initialized - just return NULL */
 477         if( NULL == _ctx )
 478                 return NULL;
 479
 480         if( sock < 0 || sock > FD_SETSIZE )
 481                 return NULL;
 482         return _ssl_context[sock];
 483 }
 484
 485
 486 int SSL_verify_callback( int ok_return, X509_STORE_CTX *ctx )
 487 {
 488         char buf[260];
 489         char cbuf[260];
 490         char ibuf[260];
 491         char *str_ptr;
 492         X509 *x509_cert;
 493         int err, depth;
 494
 495         x509_cert = X509_STORE_CTX_get_current_cert(ctx);
 496         err = X509_STORE_CTX_get_error(ctx);
 497         depth = X509_STORE_CTX_get_error_depth(ctx);
 498
 499         X509_NAME_oneline(X509_get_subject_name(x509_cert), buf, 256);
 500         X509_NAME_oneline(X509_get_issuer_name(x509_cert), ibuf, 256);
 501
 502         /* Just to be sure those buffers are terminated...  I think the
 503                 X509 libraries do, but... */
 504         buf[256] = ibuf[256] = '\0';
 505
 506         if (depth == 0) {
 507                 if( ( str_ptr = strstr( ibuf, "/O=" ) ) ) {
 508                         str_ptr += 3;
 509                         strcpy( cbuf, str_ptr );
 510                         if( ( str_ptr = strchr(cbuf, '/' ) ) ) {
 511                                 *str_ptr = '\0';
 512                         }
 513                         if (outlevel == O_VERBOSE)
 514                                 report(stdout, "Issuer Organization: %s", cbuf );
 515                 } else {
 516                         if (outlevel == O_VERBOSE)
 517                                 report(stdout, "Unknown Organization", cbuf );
 518                 }
 519                 if( ( str_ptr = strstr( ibuf, "/CN=" ) ) ) {
 520                         str_ptr += 4;
 521                         strcpy( cbuf, str_ptr );
 522                         if( ( str_ptr = strchr(cbuf, '/' ) ) ) {
 523                                 *str_ptr = '\0';
 524                         }
 525                         if (outlevel == O_VERBOSE)
 526                                 report(stdout, "Issuer CommonName: %s", cbuf );
 527                 } else {
 528                         if (outlevel == O_VERBOSE)
 529                                 report(stdout, "Unknown Issuer CommonName", cbuf );
 530                 }
 531                 if( ( str_ptr = strstr( buf, "/CN=" ) ) ) {
 532                         str_ptr += 4;
 533                         strcpy( cbuf, str_ptr );
 534                         if( ( str_ptr = strchr(cbuf, '/' ) ) ) {
 535                                 *str_ptr = '\0';
 536                         }
 537                         if (outlevel == O_VERBOSE)
 538                                 report(stdout, "Server CommonName: %s", cbuf );
 539                         /* Should we have some wildcarding here? */
 540                         if ( NULL != _ssl_server_cname && 0 != strcmp( cbuf, _ssl_server_cname ) ) {
 541                                 report(stdout, "Server CommonName mismatch: %s != %s", cbuf, _ssl_server_cname );
 542                         }
 543                 } else {
 544                         if (outlevel == O_VERBOSE)
 545                                 report(stdout, "Unknown Server CommonName", cbuf );
 546                 }
 547         }
 548
 549         switch (ctx->error) {
 550         case X509_V_ERR_UNABLE_TO_GET_ISSUER_CERT:
 551                 X509_NAME_oneline(X509_get_issuer_name(ctx->current_cert), buf, 256);
 552                 report(stdout, "unknown issuer= %s", buf);
 553                 break;
 554         case X509_V_ERR_CERT_NOT_YET_VALID:
 555         case X509_V_ERR_ERROR_IN_CERT_NOT_BEFORE_FIELD:
 556                 report(stderr, "Server Certificate not yet valid");
 557                 break;
 558         case X509_V_ERR_CERT_HAS_EXPIRED:
 559         case X509_V_ERR_ERROR_IN_CERT_NOT_AFTER_FIELD:
 560                 report(stderr, "Server Certificate expired");
 561                 break;
 562         }
 563         /* We are not requiring or validating server or issuer id's as yet */
 564         /* Always return OK from here */
 565         ok_return = 1;
 566         return( ok_return );
 567 }
 568
 569
 570 /* performs initial SSL handshake over the connected socket
 571  * uses SSL *ssl global variable, which is currently defined
 572  * in this file
 573  */
 574 int SSLOpen(int sock, char *mycert, char *mykey, char *servercname )
 575 {
 576         SSL_load_error_strings();
 577         SSLeay_add_ssl_algorithms();
 578
 579         if( sock < 0 || sock > FD_SETSIZE ) {
 580                 report(stderr, "File descriptor out of range for SSL" );
 581                 return( -1 );
 582         }
 583
 584         if( ! _ctx ) {
 585                 /* Be picky and make sure the memory is cleared */
 586                 memset( _ssl_context, 0, sizeof( _ssl_context ) );
 587                 _ctx = SSL_CTX_new(SSLv23_client_method());
 588                 if(_ctx == NULL) {
 589                         ERR_print_errors_fp(stderr);
 590                         return(-1);
 591                 }
 592         }
 593
 594         _ssl_context[sock] = SSL_new(_ctx);
 595
 596         if(_ssl_context[sock] == NULL) {
 597                 ERR_print_errors_fp(stderr);
 598                 return(-1);
 599         }
 600
 601         /* This static is for the verify callback */
 602         _ssl_server_cname = servercname;
 603
 604         SSL_CTX_set_verify(_ctx, SSL_VERIFY_PEER, SSL_verify_callback);
 605
 606         if( mycert || mykey ) {
 607
 608         /* Ok...  He has a certificate file defined, so lets declare it.  If
 609          * he does NOT have a separate certificate and private key file then
 610          * assume that it's a combined key and certificate file.
 611          */
 612                 if( !mykey )
 613                         mykey = mycert;
 614                 if( !mycert )
 615                         mycert = mykey;
 616                 SSL_use_certificate_file(_ssl_context[sock], mycert, SSL_FILETYPE_PEM);
 617                 SSL_use_RSAPrivateKey_file(_ssl_context[sock], mykey, SSL_FILETYPE_PEM);
 618         }
 619
 620         SSL_set_fd(_ssl_context[sock], sock);
 621
 622         if(SSL_connect(_ssl_context[sock]) == -1) {
 623                 ERR_print_errors_fp(stderr);
 624                 return(-1);
 625         }
 626
 627         return(0);
 628 }
 629 #endif
 630
 631 int SockClose(int sock)
 632 /* close a socket (someday we may do other cleanup here) */
 633 {
 634 #ifdef  SSL_ENABLE
 635     SSL *ssl;
 636
 637     if( NULL != ( ssl = SSLGetContext( sock ) ) ) {
 638         /* Clean up the SSL stack */
 639         SSL_free( _ssl_context[sock] );
 640         _ssl_context[sock] = NULL;
 641     }
 642 #endif
 643     return(close(sock));
 644 }
 645
 646 #ifdef MAIN
 647 /*
 648  * Use the chargen service to test input buffering directly.
 649  * You may have to uncomment the `chargen' service description in your
 650  * inetd.conf (and then SIGHUP inetd) for this to work.
 651  */
 652 main()
 653 {
 654     int         sock = SockOpen("localhost", 19, NULL);
 655     char        buf[80];
 656
 657     while (SockRead(sock, buf, sizeof(buf)-1))
 658         SockWrite(1, buf, strlen(buf));
 659     SockClose(sock);
 660 }
 661 #endif /* MAIN */
 662
 663 /* socket.c ends here */