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