]> Pileus Git - ~andy/fetchmail/blob - socket.c
Actually set default SSL certificate path if --sslcertpath is unset.
[~andy/fetchmail] / 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_("getaddrinfo(\"%s\",\"%s\") error: %s\n"),
282                 host, service, gai_strerror(i));
283         if (i == EAI_SERVICE)
284             report(stderr, GT_("Try adding the --service option (see also FAQ item R12).\n"));
285         return -1;
286     }
287
288     i = -1;
289     for (ai = ai0; ai; ai = ai->ai_next) {
290         i = socket(ai->ai_family, ai->ai_socktype, 0);
291         if (i < 0)
292             continue;
293
294         /* Socket opened saved. Usefull if connect timeout 
295          * because it can be closed.
296          */
297         mailserver_socket_temp = i;
298
299         if (connect(i, (struct sockaddr *) ai->ai_addr, ai->ai_addrlen) < 0) {
300             fm_close(i);
301             i = -1;
302             continue;
303         }
304         
305         /* No connect timeout, then no need to set mailserver_socket_temp */
306         mailserver_socket_temp = -1;
307         
308         break;
309     }
310
311     freeaddrinfo(ai0);
312
313     return i;
314 }
315
316
317 #if defined(HAVE_STDARG_H)
318 int SockPrintf(int sock, const char* format, ...)
319 {
320 #else
321 int SockPrintf(sock,format,va_alist)
322 int sock;
323 char *format;
324 va_dcl {
325 #endif
326
327     va_list ap;
328     char buf[8192];
329
330 #if defined(HAVE_STDARG_H)
331     va_start(ap, format) ;
332 #else
333     va_start(ap);
334 #endif
335     vsnprintf(buf, sizeof(buf), format, ap);
336     va_end(ap);
337     return SockWrite(sock, buf, strlen(buf));
338
339 }
340
341 #ifdef SSL_ENABLE
342 #include <openssl/ssl.h>
343 #include <openssl/err.h>
344 #include <openssl/pem.h>
345 #include <openssl/x509v3.h>
346 #include <openssl/rand.h>
347
348 static  SSL_CTX *_ctx = NULL;
349 static  SSL *_ssl_context[FD_SETSIZE];
350
351 static SSL      *SSLGetContext( int );
352 #endif /* SSL_ENABLE */
353
354 int SockWrite(int sock, char *buf, int len)
355 {
356     int n, wrlen = 0;
357 #ifdef  SSL_ENABLE
358     SSL *ssl;
359 #endif
360
361     while (len)
362     {
363 #ifdef SSL_ENABLE
364         if( NULL != ( ssl = SSLGetContext( sock ) ) )
365                 n = SSL_write(ssl, buf, len);
366         else
367 #endif /* SSL_ENABLE */
368             n = fm_write(sock, buf, len);
369         if (n <= 0)
370             return -1;
371         len -= n;
372         wrlen += n;
373         buf += n;
374     }
375     return wrlen;
376 }
377
378 int SockRead(int sock, char *buf, int len)
379 {
380     char *newline, *bp = buf;
381     int n;
382 #ifdef  FORCE_STUFFING
383     int maxavailable = 0;
384 #endif
385 #ifdef  SSL_ENABLE
386     SSL *ssl;
387 #endif
388
389     if (--len < 1)
390         return(-1);
391 #ifdef __BEOS__
392     if (peeked != 0){
393         (*bp) = peeked;
394         bp++;
395         len--;
396         peeked = 0;
397     }
398 #endif        
399     do {
400         /* 
401          * The reason for these gymnastics is that we want two things:
402          * (1) to read \n-terminated lines,
403          * (2) to return the true length of data read, even if the
404          *     data coming in has embedded NULS.
405          */
406 #ifdef  SSL_ENABLE
407         if( NULL != ( ssl = SSLGetContext( sock ) ) ) {
408                 /* Hack alert! */
409                 /* OK...  SSL_peek works a little different from MSG_PEEK
410                         Problem is that SSL_peek can return 0 if there
411                         is no data currently available.  If, on the other
412                         hand, we loose the socket, we also get a zero, but
413                         the SSL_read then SEGFAULTS!  To deal with this,
414                         we'll check the error code any time we get a return
415                         of zero from SSL_peek.  If we have an error, we bail.
416                         If we don't, we read one character in SSL_read and
417                         loop.  This should continue to work even if they
418                         later change the behavior of SSL_peek
419                         to "fix" this problem...  :-(   */
420                 if ((n = SSL_peek(ssl, bp, len)) < 0) {
421                         (void)SSL_get_error(ssl, n);
422                         return(-1);
423                 }
424 #ifdef FORCE_STUFFING
425                 maxavailable = n;
426 #endif
427                 if( 0 == n ) {
428                         /* SSL_peek says no data...  Does he mean no data
429                         or did the connection blow up?  If we got an error
430                         then bail! */
431                         if( 0 != ( n = SSL_get_error(ssl, n) ) ) {
432                                 return -1;
433                         }
434                         /* We didn't get an error so read at least one
435                                 character at this point and loop */
436                         n = 1;
437                         /* Make sure newline start out NULL!
438                          * We don't have a string to pass through
439                          * the strchr at this point yet */
440                         newline = NULL;
441                 } else if ((newline = memchr(bp, '\n', n)) != NULL)
442                         n = newline - bp + 1;
443                 /* Matthias Andree: SSL_read can return 0, in that case
444                  * we must cal SSL_get_error to figure if there was
445                  * an error or just a "no data" condition */
446                 if ((n = SSL_read(ssl, bp, n)) <= 0) {
447                         if ((n = SSL_get_error(ssl, n))) {
448                                 return(-1);
449                         }
450                 }
451                 /* Check for case where our single character turned out to
452                  * be a newline...  (It wasn't going to get caught by
453                  * the strchr above if it came from the hack...  ). */
454                 if( NULL == newline && 1 == n && '\n' == *bp ) {
455                         /* Got our newline - this will break
456                                 out of the loop now */
457                         newline = bp;
458                 }
459         }
460         else
461 #endif /* SSL_ENABLE */
462         {
463
464 #ifdef __BEOS__
465             if ((n = fm_read(sock, bp, 1)) <= 0)
466 #else
467             if ((n = fm_peek(sock, bp, len)) <= 0)
468 #endif
469                 return (-1);
470 #ifdef FORCE_STUFFING
471             maxavailable = n;
472 #endif
473             if ((newline = memchr(bp, '\n', n)) != NULL)
474                 n = newline - bp + 1;
475 #ifndef __BEOS__
476             if ((n = fm_read(sock, bp, n)) == -1)
477                 return(-1);
478 #endif /* __BEOS__ */
479         }
480         bp += n;
481         len -= n;
482     } while 
483             (!newline && len);
484     *bp = '\0';
485
486 #ifdef FORCE_STUFFING           /* too ugly to live -- besides, there's IMAP */
487     /* OK, very weird hack coming up here:
488      * When POP3 servers send us a message, they're supposed to
489      * terminate the message with a line containing only a dot. To protect
490      * against lines in the real message that might contain only a dot,
491      * they're supposed to preface any line that starts with a dot with
492      * an additional dot, which will be removed on the client side. That
493      * process, called byte-stuffing (and unstuffing) is really not the
494      * concern of this low-level routine, ordinarily, but there are some
495      * POP servers (and maybe IMAP servers too, who knows) that fail to
496      * do the byte-stuffing, and this routine is the best place to try to
497      * identify and fix that fault.
498      *
499      * Since the DOT line is supposed to come only at the end of a
500      * message, the implication is that right after we see it, the server
501      * is supposed to go back to waiting for the next command. There
502      * isn't supposed to be any more data to read after we see the dot.
503      * THEREFORE, if we see more data to be read after something that
504      * looks like the dot line, then probably the server is failing to
505      * do byte-stuffing. In that case, we'll byte-pack it for them so
506      * that the higher-level routines see things as hunky-dorey.
507      * This is not a perfect test or fix by any means (it has an
508      * obvious race condition, for one thing), but it should at least
509      * reduce the nastiness that ensues when people don't know how
510      * to write POP servers.
511      */
512     if ((maxavailable > (bp-buf)) &&
513             ((((bp-buf) == 3) &&
514               (buf[0] == '.') &&
515               (buf[1] == '\r') &&
516               (buf[2] == '\n')) ||
517              (((bp-buf) == 2) &&
518               (buf[0] == '.') &&
519               (buf[1] == '\n')))) {
520
521         memmove(buf+1, buf, (bp-buf)+1);
522         buf[0] = '.';
523         bp++;
524     }
525 #endif /* FORCE_STUFFING */
526     return bp - buf;
527 }
528
529 int SockPeek(int sock)
530 /* peek at the next socket character without actually reading it */
531 {
532     int n;
533     char ch;
534 #ifdef  SSL_ENABLE
535     SSL *ssl;
536 #endif
537
538 #ifdef  SSL_ENABLE
539         if( NULL != ( ssl = SSLGetContext( sock ) ) ) {
540                 n = SSL_peek(ssl, &ch, 1);
541                 if (n < 0) {
542                         (void)SSL_get_error(ssl, n);
543                         return -1;
544                 }
545                 if( 0 == n ) {
546                         /* This code really needs to implement a "hold back"
547                          * to simulate a functioning SSL_peek()...  sigh...
548                          * Has to be coordinated with the read code above.
549                          * Next on the list todo...     */
550
551                         /* SSL_peek says 0...  Does that mean no data
552                         or did the connection blow up?  If we got an error
553                         then bail! */
554                         if( 0 != ( n = SSL_get_error(ssl, n) ) ) {
555                                 return -1;
556                         }
557
558                         /* Haven't seen this case actually occur, but...
559                            if the problem in SockRead can occur, this should
560                            be possible...  Just not sure what to do here.
561                            This should be a safe "punt" the "peek" but don't
562                            "punt" the "session"... */
563
564                         return 0;       /* Give him a '\0' character */
565                 }
566         }
567         else
568 #endif /* SSL_ENABLE */
569             n = fm_peek(sock, &ch, 1);
570         if (n == -1)
571                 return -1;
572
573 #ifdef __BEOS__
574     peeked = ch;
575 #endif
576     return(ch);
577 }
578
579 #ifdef SSL_ENABLE
580
581 static  char *_ssl_server_cname = NULL;
582 static  int _check_fp;
583 static  char *_check_digest;
584 static  char *_server_label;
585 static  int _depth0ck;
586 static  int _prev_err;
587
588 SSL *SSLGetContext( int sock )
589 {
590         /* If SSLOpen has never initialized - just return NULL */
591         if( NULL == _ctx )
592                 return NULL;
593
594         if( sock < 0 || sock > FD_SETSIZE )
595                 return NULL;
596         return _ssl_context[sock];
597 }
598
599
600 /* ok_return (preverify_ok) is 1 if this stage of certificate verification
601    passed, or 0 if it failed. This callback lets us display informative
602    errors, and perform additional validation (e.g. CN matches) */
603 static int SSL_verify_callback( int ok_return, X509_STORE_CTX *ctx, int strict )
604 {
605         char buf[257];
606         X509 *x509_cert;
607         int err, depth;
608         unsigned char digest[EVP_MAX_MD_SIZE];
609         char text[EVP_MAX_MD_SIZE * 3 + 1], *tp, *te;
610         const EVP_MD *digest_tp;
611         unsigned int dsz, i, esz;
612         X509_NAME *subj, *issuer;
613
614         x509_cert = X509_STORE_CTX_get_current_cert(ctx);
615         err = X509_STORE_CTX_get_error(ctx);
616         depth = X509_STORE_CTX_get_error_depth(ctx);
617
618         subj = X509_get_subject_name(x509_cert);
619         issuer = X509_get_issuer_name(x509_cert);
620
621         if (depth == 0 && !_depth0ck) {
622                 _depth0ck = 1;
623                 
624                 if (outlevel == O_VERBOSE) {
625                         if ((i = X509_NAME_get_text_by_NID(issuer, NID_organizationName, buf, sizeof(buf))) != -1) {
626                                 report(stdout, GT_("Issuer Organization: %s\n"), buf);
627                                 if (i >= sizeof(buf) - 1)
628                                         report(stdout, GT_("Warning: Issuer Organization Name too long (possibly truncated).\n"));
629                         } else
630                                 report(stdout, GT_("Unknown Organization\n"));
631                         if ((i = X509_NAME_get_text_by_NID(issuer, NID_commonName, buf, sizeof(buf))) != -1) {
632                                 report(stdout, GT_("Issuer CommonName: %s\n"), buf);
633                                 if (i >= sizeof(buf) - 1)
634                                         report(stdout, GT_("Warning: Issuer CommonName too long (possibly truncated).\n"));
635                         } else
636                                 report(stdout, GT_("Unknown Issuer CommonName\n"));
637                 }
638                 if ((i = X509_NAME_get_text_by_NID(subj, NID_commonName, buf, sizeof(buf))) != -1) {
639                         if (outlevel == O_VERBOSE)
640                                 report(stdout, GT_("Server CommonName: %s\n"), buf);
641                         if (i >= sizeof(buf) - 1) {
642                                 /* Possible truncation. In this case, this is a DNS name, so this
643                                  * is really bad. We do not tolerate this even in the non-strict case. */
644                                 report(stderr, GT_("Bad certificate: Subject CommonName too long!\n"));
645                                 return (0);
646                         }
647                         if (_ssl_server_cname != NULL) {
648                                 char *p1 = buf;
649                                 char *p2 = _ssl_server_cname;
650                                 int n;
651                                 int matched = 0;
652                                 STACK_OF(GENERAL_NAME) *gens;
653                                 
654                                 /* RFC 2595 section 2.4: find a matching name
655                                  * first find a match among alternative names */
656                                 gens = X509_get_ext_d2i(x509_cert, NID_subject_alt_name, NULL, NULL);
657                                 if (gens) {
658                                         int i, r;
659                                         for (i = 0, r = sk_GENERAL_NAME_num(gens); i < r; ++i) {
660                                                 const GENERAL_NAME *gn = sk_GENERAL_NAME_value(gens, i);
661                                                 if (gn->type == GEN_DNS) {
662                                                         char *p1 = gn->d.ia5->data;
663                                                         char *p2 = _ssl_server_cname;
664                                                         if (outlevel == O_VERBOSE)
665                                                                 report(stderr, "Subject Alternative Name: %s\n", p1);
666                                                         if (*p1 == '*') {
667                                                                 ++p1;
668                                                                 n = strlen(p2) - strlen(p1);
669                                                                 if (n >= 0)
670                                                                         p2 += n;
671                                                         }
672                                                         if (0 == strcasecmp(p1, p2)) {
673                                                                 matched = 1;
674                                                         }
675                                                 }
676                                         }
677                                         sk_GENERAL_NAME_free(gens);
678                                 }
679                                 if (*p1 == '*') {
680                                         ++p1;
681                                         n = strlen(p2) - strlen(p1);
682                                         if (n >= 0)
683                                                 p2 += n;
684                                 }       
685                                 if (0 == strcasecmp(p1, p2)) {
686                                   matched = 1;
687                                 }
688                                 if (!matched) {
689                                         report(stderr,
690                                             GT_("Server CommonName mismatch: %s != %s\n"),
691                                             buf, _ssl_server_cname );
692                                         if (ok_return && strict)
693                                                 return (0);
694                                 }
695                         } else if (ok_return) {
696                                 report(stderr, GT_("Server name not set, could not verify certificate!\n"));
697                                 if (strict) return (0);
698                         }
699                 } else {
700                         if (outlevel == O_VERBOSE)
701                                 report(stdout, GT_("Unknown Server CommonName\n"));
702                         if (ok_return && strict) {
703                                 report(stderr, GT_("Server name not specified in certificate!\n"));
704                                 return (0);
705                         }
706                 }
707                 /* Print the finger print. Note that on errors, we might print it more than once
708                  * normally; we kluge around that by using a global variable. */
709                 if (_check_fp) {
710                         _check_fp = 0;
711                         digest_tp = EVP_md5();
712                         if (digest_tp == NULL) {
713                                 report(stderr, GT_("EVP_md5() failed!\n"));
714                                 return (0);
715                         }
716                         if (!X509_digest(x509_cert, digest_tp, digest, &dsz)) {
717                                 report(stderr, GT_("Out of memory!\n"));
718                                 return (0);
719                         }
720                         tp = text;
721                         te = text + sizeof(text);
722                         for (i = 0; i < dsz; i++) {
723                                 esz = snprintf(tp, te - tp, i > 0 ? ":%02X" : "%02X", digest[i]);
724                                 if (esz >= te - tp) {
725                                         report(stderr, GT_("Digest text buffer too small!\n"));
726                                         return (0);
727                                 }
728                                 tp += esz;
729                         }
730                         if (outlevel > O_NORMAL)
731                             report(stdout, GT_("%s key fingerprint: %s\n"), _server_label, text);
732                         if (_check_digest != NULL) {
733                                 if (strcmp(text, _check_digest) == 0) {
734                                     if (outlevel > O_NORMAL)
735                                         report(stdout, GT_("%s fingerprints match.\n"), _server_label);
736                                 } else {
737                                     if (outlevel > O_SILENT)
738                                         report(stderr, GT_("%s fingerprints do not match!\n"), _server_label);
739                                     return (0);
740                                 }
741                         }
742                 }
743         }
744
745         if (err != X509_V_OK && err != _prev_err) {
746                 _prev_err = err;
747                 report(stderr, GT_("Server certificate verification error: %s\n"), X509_verify_cert_error_string(err));
748                 /* We gave the error code, but maybe we can add some more details for debugging */
749                 switch (err) {
750                 case X509_V_ERR_UNABLE_TO_GET_ISSUER_CERT:
751                         X509_NAME_oneline(issuer, buf, sizeof(buf));
752                         buf[sizeof(buf) - 1] = '\0';
753                         report(stderr, GT_("unknown issuer (first %d characters): %s\n"), sizeof(buf)-1, buf);
754                         break;
755                 }
756         }
757         if (!strict)
758                 ok_return = 1;
759         return (ok_return);
760 }
761
762 static int SSL_nock_verify_callback( int ok_return, X509_STORE_CTX *ctx )
763 {
764         return SSL_verify_callback(ok_return, ctx, 0);
765 }
766
767 static int SSL_ck_verify_callback( int ok_return, X509_STORE_CTX *ctx )
768 {
769         return SSL_verify_callback(ok_return, ctx, 1);
770 }
771
772 /* performs initial SSL handshake over the connected socket
773  * uses SSL *ssl global variable, which is currently defined
774  * in this file
775  */
776 int SSLOpen(int sock, char *mycert, char *mykey, char *myproto, int certck, char *certpath,
777     char *fingerprint, char *servercname, char *label)
778 {
779         struct stat randstat;
780         int i;
781
782         SSL_load_error_strings();
783         SSLeay_add_ssl_algorithms();
784         
785 #ifdef SSL_ENABLE
786         if (stat("/dev/random", &randstat)  &&
787             stat("/dev/urandom", &randstat)) {
788           /* Neither /dev/random nor /dev/urandom are present, so add
789              entropy to the SSL PRNG a hard way. */
790           for (i = 0; i < 10000  &&  ! RAND_status (); ++i) {
791             char buf[4];
792             struct timeval tv;
793             gettimeofday (&tv, 0);
794             buf[0] = tv.tv_usec & 0xF;
795             buf[2] = (tv.tv_usec & 0xF0) >> 4;
796             buf[3] = (tv.tv_usec & 0xF00) >> 8;
797             buf[1] = (tv.tv_usec & 0xF000) >> 12;
798             RAND_add (buf, sizeof buf, 0.1);
799           }
800         }
801 #endif /* SSL_ENABLE */
802
803
804         if( sock < 0 || sock > FD_SETSIZE ) {
805                 report(stderr, GT_("File descriptor out of range for SSL") );
806                 return( -1 );
807         }
808
809         if( ! _ctx ) {
810                 /* Be picky and make sure the memory is cleared */
811                 memset( _ssl_context, 0, sizeof( _ssl_context ) );
812                 if(myproto) {
813                         if(!strcmp("ssl2",myproto)) {
814                                 _ctx = SSL_CTX_new(SSLv2_client_method());
815                         } else if(!strcmp("ssl3",myproto)) {
816                                 _ctx = SSL_CTX_new(SSLv3_client_method());
817                         } else if(!strcmp("tls1",myproto)) {
818                                 _ctx = SSL_CTX_new(TLSv1_client_method());
819                         } else if (!strcmp("ssl23",myproto)) {
820                                 myproto = NULL;
821                         } else {
822                                 fprintf(stderr,GT_("Invalid SSL protocol '%s' specified, using default (SSLv23).\n"), myproto);
823                                 myproto = NULL;
824                         }
825                 }
826                 if(!myproto) {
827                         _ctx = SSL_CTX_new(SSLv23_client_method());
828                 }
829                 if(_ctx == NULL) {
830                         ERR_print_errors_fp(stderr);
831                         return(-1);
832                 }
833         }
834
835         if (certck) {
836                 SSL_CTX_set_verify(_ctx, SSL_VERIFY_PEER, SSL_ck_verify_callback);
837         } else {
838                 /* In this case, we do not fail if verification fails. However,
839                  *  we provide the callback for output and possible fingerprint checks. */
840                 SSL_CTX_set_verify(_ctx, SSL_VERIFY_PEER, SSL_nock_verify_callback);
841         }
842         if (certpath)
843                 SSL_CTX_load_verify_locations(_ctx, NULL, certpath);
844         else
845                 SSL_CTX_set_default_verify_paths(_ctx);
846         
847         _ssl_context[sock] = SSL_new(_ctx);
848         
849         if(_ssl_context[sock] == NULL) {
850                 ERR_print_errors_fp(stderr);
851                 return(-1);
852         }
853         
854         /* This static is for the verify callback */
855         _ssl_server_cname = servercname;
856         _server_label = label;
857         _check_fp = 1;
858         _check_digest = fingerprint;
859         _depth0ck = 0;
860         _prev_err = -1;
861
862         if( mycert || mykey ) {
863
864         /* Ok...  He has a certificate file defined, so lets declare it.  If
865          * he does NOT have a separate certificate and private key file then
866          * assume that it's a combined key and certificate file.
867          */
868                 if( !mykey )
869                         mykey = mycert;
870                 if( !mycert )
871                         mycert = mykey;
872                 SSL_use_certificate_file(_ssl_context[sock], mycert, SSL_FILETYPE_PEM);
873                 SSL_use_RSAPrivateKey_file(_ssl_context[sock], mykey, SSL_FILETYPE_PEM);
874         }
875
876         SSL_set_fd(_ssl_context[sock], sock);
877         
878         if(SSL_connect(_ssl_context[sock]) < 1) {
879                 ERR_print_errors_fp(stderr);
880                 return(-1);
881         }
882
883         /* Paranoia: was the callback not called as we expected? */
884         if (!_depth0ck) {
885                 report(stderr, GT_("Certificate/fingerprint verification was somehow skipped!\n"));
886
887                 if (fingerprint != NULL || certck) {
888                         if( NULL != SSLGetContext( sock ) ) {
889                                 /* Clean up the SSL stack */
890                                 SSL_shutdown( _ssl_context[sock] );
891                                 SSL_free( _ssl_context[sock] );
892                                 _ssl_context[sock] = NULL;
893                         }
894                         return(-1);
895                 }
896         }
897
898         return(0);
899 }
900 #endif
901
902 int SockClose(int sock)
903 /* close a socket gracefully */
904 {
905 #ifdef  SSL_ENABLE
906     if( NULL != SSLGetContext( sock ) ) {
907         /* Clean up the SSL stack */
908         SSL_shutdown( _ssl_context[sock] );
909         SSL_free( _ssl_context[sock] );
910         _ssl_context[sock] = NULL;
911     }
912 #endif
913
914 #ifdef __UNUSED__
915     /* 
916      * This hangs in RedHat 6.2 after fetchmail runs for a while a
917      * FIN_WAIT2 comes up in netstat and fetchmail never returns from
918      * the recv system call. (Reported from jtnews
919      * <jtnews@bellatlantic.net>, Wed, 24 May 2000 21:26:02.)
920      *
921      * Half-close the connection first so the other end gets notified.
922      *
923      * This stops sends but allows receives (effectively, it sends a
924      * TCP <FIN>).  */
925     if (shutdown(sock, 1) == 0) {
926         char ch;
927         /* If there is any data still waiting in the queue, discard it.
928          * Call recv() until either it returns 0 (meaning we received a FIN)
929          * or any error occurs.  This makes sure all data sent by the other
930          * side is acknowledged at the TCP level.
931          */
932         if (fm_peek(sock, &ch, 1) > 0)
933             while (fm_read(sock, &ch, 1) > 0)
934                 continue;
935     }
936 #endif /* __UNUSED__ */
937
938     /* if there's an error closing at this point, not much we can do */
939     return(fm_close(sock));     /* this is guarded */
940 }
941
942 #ifdef __CYGWIN__
943 /*
944  * Workaround Microsoft Winsock recv/WSARecv(..., MSG_PEEK) bug.
945  * See http://sources.redhat.com/ml/cygwin/2001-08/msg00628.html
946  * for more details.
947  */
948 static ssize_t cygwin_read(int sock, void *buf, size_t count)
949 {
950     char *bp = buf;
951     int n = 0;
952
953     if ((n = read(sock, bp, count)) == -1)
954         return(-1);
955
956     if (n != count) {
957         int n2 = 0;
958         if (outlevel >= O_VERBOSE)
959             report(stdout, GT_("Cygwin socket read retry\n"));
960         n2 = read(sock, bp + n, count - n);
961         if (n2 == -1 || n + n2 != count) {
962             report(stderr, GT_("Cygwin socket read retry failed!\n"));
963             return(-1);
964         }
965     }
966
967     return count;
968 }
969 #endif /* __CYGWIN__ */
970
971 #ifdef MAIN
972 /*
973  * Use the chargen service to test input buffering directly.
974  * You may have to uncomment the `chargen' service description in your
975  * inetd.conf (and then SIGHUP inetd) for this to work.  */
976 main()
977 {
978     int         sock = SockOpen("localhost", "chargen", NULL);
979     char        buf[80];
980
981     while (SockRead(sock, buf, sizeof(buf)-1))
982         SockWrite(1, buf, strlen(buf));
983     SockClose(sock);
984 }
985 #endif /* MAIN */
986
987 /* socket.c ends here */