From c8e92cdbd6c38bb771ab0b9c88c92114f1b0315e Mon Sep 17 00:00:00 2001 From: hniksic Date: Wed, 5 Nov 2003 17:12:03 -0800 Subject: [PATCH] [svn] Get rid of ssl_iread/ssl_iwrite. Have the SSL code register its routines for reading and writing data. --- src/ChangeLog | 31 ++++++ src/connect.c | 252 +++++++++++++++++++++++++++++++++++++--------- src/connect.h | 40 ++++---- src/ftp-basic.c | 50 ++++----- src/ftp.c | 140 +++++++++++++------------- src/gen_sslfunc.c | 158 +++++++++++++---------------- src/gen_sslfunc.h | 5 +- src/hash.c | 25 +---- src/hash.h | 3 - src/http.c | 151 +++++++-------------------- src/mswindows.h | 4 - src/rbuf.c | 31 +----- src/rbuf.h | 10 -- src/retr.c | 7 +- src/sysdep.h | 27 ----- 15 files changed, 466 insertions(+), 468 deletions(-) diff --git a/src/ChangeLog b/src/ChangeLog index 330ec2b2..fa420de9 100644 --- a/src/ChangeLog +++ b/src/ChangeLog @@ -1,3 +1,34 @@ +2003-11-06 Hrvoje Niksic + + * connect.c: Updated all callers of + iread/ssl_iread/iwrite/ssl_iwrite to use xread and xwrite instead. + + * rbuf.h (struct rbuf): Removed the SSL member because SSL is + handled automatically by xread. + + * hash.c (ptrhash): Made private. + (ptrcmp): Ditto. + (inthash): Removed. + + * connect.c (select_fd): Don't set errno, leave it to the caller. + + * gen_sslfunc.c (connect_ssl): Use register_extended to register + SSL callbacks for communication with SSL-enabled endpoints. + (ssl_read): New function. + (ssl_write): Ditto. + (ssl_poll): Ditto. + (ssl_close): Ditto. + + * connect.c (register_extended): New function -- register + callbacks for basic socket operations. + (xread): Ditto. + (xwrite): Ditto. + (xclose): Ditto. + (sock_read): New function, default implementation for reading. + (sock_write): Ditto for writing. + (sock_poll): Ditto for polling. + (sock_close): Ditto for closing. + 2003-11-05 Hrvoje Niksic * connect.c (bindport): Fix compilation under pre-C99 compilers. diff --git a/src/connect.c b/src/connect.c index 62ce2af9..c24e3e95 100644 --- a/src/connect.c +++ b/src/connect.c @@ -60,6 +60,7 @@ so, delete this exception statement from your version. */ #include "utils.h" #include "host.h" #include "connect.h" +#include "hash.h" #ifndef errno extern int errno; @@ -304,7 +305,7 @@ connect_to_ip (const ip_address *ip, int port, const char *print) logprintf. */ int save_errno = errno; if (sock >= 0) - CLOSE (sock); + xclose (sock); if (print) logprintf (LOG_VERBOSE, "failed: %s.\n", strerror (errno)); errno = save_errno; @@ -426,7 +427,7 @@ bindport (const ip_address *bind_address, int *port, int *local_sock) sockaddr_set_data (sa, bind_address, *port); if (bind (sock, sa, sockaddr_size (sa)) < 0) { - CLOSE (sock); + xclose (sock); return BINDERR; } DEBUGP (("Local socket fd %d bound.\n", sock)); @@ -435,7 +436,7 @@ bindport (const ip_address *bind_address, int *port, int *local_sock) socklen_t sa_len = sockaddr_size (sa); if (getsockname (sock, sa, &sa_len) < 0) { - CLOSE (sock); + xclose (sock); return CONPORTERR; } sockaddr_get_data (sa, NULL, port); @@ -444,7 +445,7 @@ bindport (const ip_address *bind_address, int *port, int *local_sock) } if (listen (sock, 1) < 0) { - CLOSE (sock); + xclose (sock); return LISTENERR; } *local_sock = sock; @@ -454,14 +455,14 @@ bindport (const ip_address *bind_address, int *port, int *local_sock) #ifdef HAVE_SELECT /* Wait for file descriptor FD to be readable or writable or both, timing out after MAXTIME seconds. Returns 1 if FD is available, 0 - for timeout and -1 for error. The argument WHAT can be a + for timeout and -1 for error. The argument WAIT_FOR can be a combination of WAIT_READ and WAIT_WRITE. This is a mere convenience wrapper around the select call, and should be taken as such. */ int -select_fd (int fd, double maxtime, int wait) +select_fd (int fd, double maxtime, int wait_for) { fd_set fdset; fd_set *rd = NULL, *wr = NULL; @@ -470,23 +471,18 @@ select_fd (int fd, double maxtime, int wait) FD_ZERO (&fdset); FD_SET (fd, &fdset); - if (wait & WAIT_READ) + if (wait_for & WAIT_FOR_READ) rd = &fdset; - if (wait & WAIT_WRITE) + if (wait_for & WAIT_FOR_WRITE) wr = &fdset; - tmout.tv_sec = (long)maxtime; - tmout.tv_usec = 1000000L * (maxtime - (long)maxtime); + tmout.tv_sec = (long) maxtime; + tmout.tv_usec = 1000000L * (maxtime - (long) maxtime); do result = select (fd + 1, rd, wr, NULL, &tmout); while (result < 0 && errno == EINTR); - /* When we've timed out, set errno to ETIMEDOUT for the convenience - of the caller. */ - if (result == 0) - errno = ETIMEDOUT; - return result; } #endif /* HAVE_SELECT */ @@ -505,7 +501,7 @@ acceptport (int local_sock, int *sock) #ifdef HAVE_SELECT if (opt.connect_timeout) - if (select_fd (local_sock, opt.connect_timeout, WAIT_READ) <= 0) + if (select_fd (local_sock, opt.connect_timeout, WAIT_FOR_READ) <= 0) return ACCEPTERR; #endif if ((*sock = accept (local_sock, sa, &addrlen)) < 0) @@ -555,58 +551,214 @@ conaddr (int fd, ip_address *ip) return 0; } + +/* Basic socket operations, mostly EINTR wrappers. */ + +#ifdef WINDOWS +# define read(fd, buf, cnt) recv (fd, buf, cnt, 0) +# define write(fd, buf, cnt) send (fd, buf, cnt, 0) +# define close(fd) closesocket (fd) +#endif -/* Read at most LEN bytes from FD, storing them to BUF. This is - virtually the same as read(), but takes care of EINTR braindamage - and uses select() to timeout the stale connections (a connection is - stale if more than OPT.READ_TIMEOUT time is spent in select() or - read()). */ +#ifdef __BEOS__ +# define read(fd, buf, cnt) recv (fd, buf, cnt, 0) +# define write(fd, buf, cnt) send (fd, buf, cnt, 0) +#endif -int -iread (int fd, char *buf, int len) +static int +sock_read (int fd, char *buf, int bufsize) { int res; - -#ifdef HAVE_SELECT - if (opt.read_timeout) - if (select_fd (fd, opt.read_timeout, WAIT_READ) <= 0) - return -1; -#endif do - res = READ (fd, buf, len); + res = read (fd, buf, bufsize); while (res == -1 && errno == EINTR); + return res; +} +static int +sock_write (int fd, char *buf, int bufsize) +{ + int res = 0; + do + res = write (fd, buf, bufsize); + while (res == -1 && errno == EINTR); return res; } -/* Write LEN bytes from BUF to FD. This is similar to iread(), but - unlike iread(), it makes sure that all of BUF is actually written - to FD, so callers needn't bother with checking that the return - value equals to LEN. Instead, you should simply check for -1. */ +static int +sock_poll (int fd, double timeout, int wait_for) +{ +#ifdef HAVE_SELECT + return select_fd (fd, timeout, wait_for); +#else + return 1; +#endif +} + +static void +sock_close (int fd) +{ + close (fd); + DEBUGP (("Closed fd %d\n", fd)); +} +#undef read +#undef write +#undef close + +/* Reading and writing from the network. We build around the socket + (file descriptor) API, but support "extended" operations for things + that are not mere file descriptors under the hood, such as SSL + sockets. + + That way the user code can call xread(fd, ...) and we'll run read + or SSL_read or whatever is necessary. */ + +static struct hash_table *extended_map; +static long extended_map_modified_tick; + +struct extended_info { + xreader_t reader; + xwriter_t writer; + xpoller_t poller; + xcloser_t closer; + void *ctx; +}; + +void +register_extended (int fd, xreader_t reader, xwriter_t writer, + xpoller_t poller, xcloser_t closer, void *ctx) +{ + struct extended_info *info = xnew (struct extended_info); + info->reader = reader; + info->writer = writer; + info->poller = poller; + info->closer = closer; + info->ctx = ctx; + if (!extended_map) + extended_map = hash_table_new (0, NULL, NULL); + hash_table_put (extended_map, (void *) fd, info); + ++extended_map_modified_tick; +} + +/* When xread/xwrite are called multiple times in a loop, they should + remember the INFO pointer instead of fetching it every time. It is + not enough to compare FD to LAST_FD because FD might have been + closed and reopened. modified_tick ensures that changes to + extended_map will not be unnoticed. + + This is a macro because we want the static storage variables to be + per-function. */ + +#define LAZY_RETRIEVE_INFO(info) do { \ + static struct extended_info *last_info; \ + static int last_fd = -1, last_tick; \ + if (!extended_map) \ + info = NULL; \ + else if (last_fd == fd && last_tick == extended_map_modified_tick) \ + info = last_info; \ + else \ + { \ + info = hash_table_get (extended_map, (void *) fd); \ + last_fd = fd; \ + last_tick = extended_map_modified_tick; \ + } \ +} while (0) + +/* Read no more than BUFSIZE bytes of data from FD, storing them to + BUF. If TIMEOUT is non-zero, the operation aborts if no data is + received after that many seconds. If TIMEOUT is -1, the value of + opt.timeout is used for TIMEOUT. */ int -iwrite (int fd, char *buf, int len) +xread (int fd, char *buf, int bufsize, double timeout) { - int res = 0; + struct extended_info *info; + LAZY_RETRIEVE_INFO (info); + if (timeout == -1) + timeout = opt.read_timeout; + if (timeout) + { + int test; + if (info && info->poller) + test = info->poller (fd, timeout, WAIT_FOR_READ, info->ctx); + else + test = sock_poll (fd, timeout, WAIT_FOR_READ); + if (test == 0) + errno = ETIMEDOUT; + if (test <= 0) + return -1; + } + if (info && info->reader) + return info->reader (fd, buf, bufsize, info->ctx); + else + return sock_read (fd, buf, bufsize); +} - /* `write' may write less than LEN bytes, thus the outward loop - keeps trying it until all was written, or an error occurred. The - inner loop is reserved for the usual EINTR f*kage, and the - innermost loop deals with the same during select(). */ - while (len > 0) +/* Write the entire contents of BUF to FD. If TIMEOUT is non-zero, + the operation aborts if no data is received after that many + seconds. If TIMEOUT is -1, the value of opt.timeout is used for + TIMEOUT. */ + +int +xwrite (int fd, char *buf, int bufsize, double timeout) +{ + int res; + struct extended_info *info; + LAZY_RETRIEVE_INFO (info); + if (timeout == -1) + timeout = opt.read_timeout; + + /* `write' may write less than LEN bytes, thus the loop keeps trying + it until all was written, or an error occurred. */ + res = 0; + while (bufsize > 0) { -#ifdef HAVE_SELECT - if (opt.read_timeout) - if (select_fd (fd, opt.read_timeout, WAIT_WRITE) <= 0) - return -1; -#endif - do - res = WRITE (fd, buf, len); - while (res == -1 && errno == EINTR); + if (timeout) + { + int test; + if (info && info->poller) + test = info->poller (fd, timeout, WAIT_FOR_WRITE, info->ctx); + else + test = sock_poll (fd, timeout, WAIT_FOR_WRITE); + if (test == 0) + errno = ETIMEDOUT; + if (test <= 0) + return -1; + } + if (info && info->writer) + res = info->writer (fd, buf, bufsize, info->ctx); + else + res = sock_write (fd, buf, bufsize); if (res <= 0) break; buf += res; - len -= res; + bufsize -= res; } return res; } + +/* Close the file descriptor FD. */ + +void +xclose (int fd) +{ + struct extended_info *info; + if (fd < 0) + return; + + /* We don't need to be extra-fast here, so save some code by + avoiding LAZY_RETRIEVE_INFO. */ + info = NULL; + if (extended_map) + info = hash_table_get (extended_map, (void *) fd); + + if (info && info->closer) + { + info->closer (fd, info->ctx); + hash_table_remove (extended_map, (void *) fd); + xfree (info); + ++extended_map_modified_tick; + } + else + sock_close (fd); +} diff --git a/src/connect.h b/src/connect.h index 8b0ed4b3..271f32ed 100644 --- a/src/connect.h +++ b/src/connect.h @@ -32,17 +32,6 @@ so, delete this exception statement from your version. */ #include "host.h" /* for definition of ip_address */ -/* Returned by connect_to_host when host name cannot be resolved. */ -enum { - E_HOST = -100 -}; - -/* Flags for select_fd's WAIT argument. */ -enum { - WAIT_READ = 1, - WAIT_WRITE = 2 -}; - /* bindport flags */ #define BIND_ON_IPV4_ONLY LH_IPV4_ONLY #define BIND_ON_IPV6_ONLY LH_IPV6_ONLY @@ -55,18 +44,35 @@ enum { /* Function declarations */ -int connect_to_ip PARAMS ((const ip_address *, int, const char *)); +/* Returned by connect_to_host when host name cannot be resolved. */ +enum { + E_HOST = -100 +}; int connect_to_host PARAMS ((const char *, int)); +int connect_to_ip PARAMS ((const ip_address *, int, const char *)); void sockaddr_get_data PARAMS ((const struct sockaddr *, ip_address *, int *)); - -int test_socket_open PARAMS ((int)); -int select_fd PARAMS ((int, double, int)); uerr_t bindport PARAMS ((const ip_address *, int *, int *)); uerr_t acceptport PARAMS ((int, int *)); int conaddr PARAMS ((int, ip_address *)); -int iread PARAMS ((int, char *, int)); -int iwrite PARAMS ((int, char *, int)); +/* Flags for select_fd's WAIT_FOR argument. */ +enum { + WAIT_FOR_READ = 1, + WAIT_FOR_WRITE = 2 +}; +int select_fd PARAMS ((int, double, int)); +int test_socket_open PARAMS ((int)); +typedef int (*xreader_t) PARAMS ((int, char *, int, void *)); +typedef int (*xwriter_t) PARAMS ((int, char *, int, void *)); +typedef int (*xpoller_t) PARAMS ((int, double, int, void *)); +typedef void (*xcloser_t) PARAMS ((int, void *)); +void register_extended PARAMS ((int, + xreader_t, xwriter_t, xpoller_t, xcloser_t, + void *)); + +int xread PARAMS ((int, char *, int, double)); +int xwrite PARAMS ((int, char *, int, double)); +void xclose PARAMS ((int)); #endif /* CONNECT_H */ diff --git a/src/ftp-basic.c b/src/ftp-basic.c index 8ae3cd9d..469263be 100644 --- a/src/ftp-basic.c +++ b/src/ftp-basic.c @@ -158,7 +158,7 @@ ftp_login (struct rbuf *rbuf, const char *acc, const char *pass) xfree (respline); /* Send USER username. */ request = ftp_request ("USER", acc); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -225,7 +225,7 @@ ftp_login (struct rbuf *rbuf, const char *acc, const char *pass) xfree (respline); /* Send PASS password. */ request = ftp_request ("PASS", pass); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -303,11 +303,11 @@ ftp_port (struct rbuf *rbuf, int *local_sock) /* Send PORT request. */ request = ftp_request ("PORT", bytes); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); - CLOSE (*local_sock); + xclose (*local_sock); return WRITEFAILED; } xfree (request); @@ -317,13 +317,13 @@ ftp_port (struct rbuf *rbuf, int *local_sock) if (err != FTPOK) { xfree (respline); - CLOSE (*local_sock); + xclose (*local_sock); return err; } if (*respline != '2') { xfree (respline); - CLOSE (*local_sock); + xclose (*local_sock); return FTPPORTERR; } xfree (respline); @@ -400,11 +400,11 @@ ftp_lprt (struct rbuf *rbuf, int *local_sock) /* Send PORT request. */ request = ftp_request ("LPRT", bytes); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); - CLOSE (*local_sock); + xclose (*local_sock); return WRITEFAILED; } xfree (request); @@ -413,13 +413,13 @@ ftp_lprt (struct rbuf *rbuf, int *local_sock) if (err != FTPOK) { xfree (respline); - CLOSE (*local_sock); + xclose (*local_sock); return err; } if (*respline != '2') { xfree (respline); - CLOSE (*local_sock); + xclose (*local_sock); return FTPPORTERR; } xfree (respline); @@ -484,11 +484,11 @@ ftp_eprt (struct rbuf *rbuf, int *local_sock) /* Send PORT request. */ request = ftp_request ("EPRT", bytes); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); - CLOSE (*local_sock); + xclose (*local_sock); return WRITEFAILED; } xfree (request); @@ -497,13 +497,13 @@ ftp_eprt (struct rbuf *rbuf, int *local_sock) if (err != FTPOK) { xfree (respline); - CLOSE (*local_sock); + xclose (*local_sock); return err; } if (*respline != '2') { xfree (respline); - CLOSE (*local_sock); + xclose (*local_sock); return FTPPORTERR; } xfree (respline); @@ -532,7 +532,7 @@ ftp_pasv (struct rbuf *rbuf, ip_address *addr, int *port) /* Form the request. */ request = ftp_request ("PASV", NULL); /* And send it. */ - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -603,7 +603,7 @@ ftp_lpsv (struct rbuf *rbuf, ip_address *addr, int *port) request = ftp_request ("LPSV", NULL); /* And send it. */ - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -779,7 +779,7 @@ ftp_epsv (struct rbuf *rbuf, ip_address *ip, int *port) request = ftp_request ("EPSV", (sa->sa_family == AF_INET ? "1" : "2")); /* And send it. */ - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -879,7 +879,7 @@ ftp_type (struct rbuf *rbuf, int type) stype[1] = 0; /* Send TYPE request. */ request = ftp_request ("TYPE", stype); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -914,7 +914,7 @@ ftp_cwd (struct rbuf *rbuf, const char *dir) /* Send CWD request. */ request = ftp_request ("CWD", dir); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -954,7 +954,7 @@ ftp_rest (struct rbuf *rbuf, long offset) number_to_string (numbuf, offset); request = ftp_request ("REST", numbuf); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -988,7 +988,7 @@ ftp_retr (struct rbuf *rbuf, const char *file) /* Send RETR request. */ request = ftp_request ("RETR", file); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -1028,7 +1028,7 @@ ftp_list (struct rbuf *rbuf, const char *file) /* Send LIST request. */ request = ftp_request ("LIST", file); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -1067,7 +1067,7 @@ ftp_syst (struct rbuf *rbuf, enum stype *server_type) /* Send SYST request. */ request = ftp_request ("SYST", NULL); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -1123,7 +1123,7 @@ ftp_pwd (struct rbuf *rbuf, char **pwd) /* Send PWD request. */ request = ftp_request ("PWD", NULL); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); @@ -1169,7 +1169,7 @@ ftp_size (struct rbuf *rbuf, const char *file, long int *size) /* Send PWD request. */ request = ftp_request ("SIZE", file); - nwritten = iwrite (RBUF_FD (rbuf), request, strlen (request)); + nwritten = xwrite (RBUF_FD (rbuf), request, strlen (request), -1); if (nwritten < 0) { xfree (request); diff --git a/src/ftp.c b/src/ftp.c index 1dbc1d00..87bd8668 100644 --- a/src/ftp.c +++ b/src/ftp.c @@ -327,14 +327,14 @@ getftp (struct url *u, long *len, long restval, ccon *con) logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; case FTPSRVERR: logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Error in server greeting.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -342,21 +342,21 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Write failed, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; case FTPLOGREFUSED: logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("The server refuses login.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return FTPLOGREFUSED; break; case FTPLOGINC: logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Login incorrect.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return FTPLOGINC; break; @@ -380,7 +380,7 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -411,7 +411,7 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -465,7 +465,7 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -473,7 +473,7 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Write failed, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -482,7 +482,7 @@ Error in server response, closing control connection.\n")); logprintf (LOG_NOTQUIET, _("Unknown type `%c', closing control connection.\n"), type_char); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; case FTPOK: @@ -580,7 +580,7 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -588,7 +588,7 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Write failed, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -596,7 +596,7 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logprintf (LOG_NOTQUIET, _("No such directory `%s'.\n\n"), u->dir); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -631,7 +631,7 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -661,7 +661,7 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -669,7 +669,7 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Write failed, closing control connection.\n")); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -697,7 +697,7 @@ Error in server response, closing control connection.\n")); if (dtsock < 0) { int save_errno = errno; - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); logprintf (LOG_VERBOSE, _("couldn't connect to %s port %hu: %s\n"), pretty_print_address (&passive_addr), passive_port, @@ -722,9 +722,9 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -732,18 +732,18 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Write failed, closing control connection.\n")); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return err; break; case CONSOCKERR: logputs (LOG_VERBOSE, "\n"); logprintf (LOG_NOTQUIET, "socket: %s\n", strerror (errno)); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -752,16 +752,16 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logprintf (LOG_NOTQUIET, _("Bind error (%s).\n"), strerror (errno)); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (dtsock); + xclose (local_sock); return err; break; case FTPPORTERR: logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Invalid PORT.\n")); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -791,9 +791,9 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -801,9 +801,9 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Write failed, closing control connection.\n")); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -816,9 +816,9 @@ Error in server response, closing control connection.\n")); logprintf (LOG_NOTQUIET, _("\nREST failed; will not truncate `%s'.\n"), con->target); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return CONTNOTSUPPORTED; } @@ -844,9 +844,9 @@ Error in server response, closing control connection.\n")); request. */ if (opt.spider) { - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return RETRFINISHED; } @@ -869,9 +869,9 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -879,17 +879,17 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Write failed, closing control connection.\n")); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return err; break; case FTPNSFOD: logputs (LOG_VERBOSE, "\n"); logprintf (LOG_NOTQUIET, _("No such file `%s'.\n\n"), u->file); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (dtsock); + xclose (local_sock); return err; break; case FTPOK: @@ -920,9 +920,9 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("\ Error in server response, closing control connection.\n")); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -930,9 +930,9 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logputs (LOG_NOTQUIET, _("Write failed, closing control connection.\n")); - CLOSE (csock); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (csock); + xclose (dtsock); + xclose (local_sock); rbuf_uninitialize (&con->rbuf); return err; break; @@ -940,8 +940,8 @@ Error in server response, closing control connection.\n")); logputs (LOG_VERBOSE, "\n"); logprintf (LOG_NOTQUIET, _("No such file or directory `%s'.\n\n"), "."); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (dtsock); + xclose (local_sock); return err; break; case FTPOK: @@ -995,10 +995,10 @@ Error in server response, closing control connection.\n")); if (!fp) { logprintf (LOG_NOTQUIET, "%s: %s\n", con->target, strerror (errno)); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); - CLOSE (dtsock); - CLOSE (local_sock); + xclose (dtsock); + xclose (local_sock); return FOPENERR; } } @@ -1045,8 +1045,8 @@ Error in server response, closing control connection.\n")); tms = time_str (NULL); tmrate = retr_rate (*len - restval, con->dltime, 0); /* Close data connection socket. */ - CLOSE (dtsock); - CLOSE (local_sock); + xclose (dtsock); + xclose (local_sock); /* Close the local file. */ { /* Close or flush the file. We have to be careful to check for @@ -1066,7 +1066,7 @@ Error in server response, closing control connection.\n")); { logprintf (LOG_NOTQUIET, _("%s: %s, closing control connection.\n"), con->target, strerror (errno)); - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return FWRITEERR; } @@ -1094,7 +1094,7 @@ Error in server response, closing control connection.\n")); return FTPRETRINT, since there is a possibility that the whole file was retrieved nevertheless (but that is for ftp_loop_internal to decide). */ - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); return FTPRETRINT; } /* err != FTPOK */ @@ -1123,7 +1123,7 @@ Error in server response, closing control connection.\n")); { /* I should probably send 'QUIT' and check for a reply, but this is faster. #### Is it OK, though? */ - CLOSE (csock); + xclose (csock); rbuf_uninitialize (&con->rbuf); } /* If it was a listing, and opt.server_response is true, @@ -1304,7 +1304,7 @@ ftp_loop_internal (struct url *u, struct fileinfo *f, ccon *con) if (con->st & ON_YOUR_OWN) { - CLOSE (RBUF_FD (&con->rbuf)); + xclose (RBUF_FD (&con->rbuf)); rbuf_uninitialize (&con->rbuf); } if (!opt.spider) @@ -1366,7 +1366,7 @@ ftp_loop_internal (struct url *u, struct fileinfo *f, ccon *con) if (rbuf_initialized_p (&con->rbuf) && (con->st & ON_YOUR_OWN)) { - CLOSE (RBUF_FD (&con->rbuf)); + xclose (RBUF_FD (&con->rbuf)); rbuf_uninitialize (&con->rbuf); } return TRYLIMEXC; @@ -1924,7 +1924,7 @@ ftp_loop (struct url *u, int *dt, struct url *proxy) *dt |= RETROKF; /* If a connection was left, quench it. */ if (rbuf_initialized_p (&con.rbuf)) - CLOSE (RBUF_FD (&con.rbuf)); + xclose (RBUF_FD (&con.rbuf)); xfree_null (con.id); con.id = NULL; xfree_null (con.target); diff --git a/src/gen_sslfunc.c b/src/gen_sslfunc.c index 1e3ecffc..2fe2e386 100644 --- a/src/gen_sslfunc.c +++ b/src/gen_sslfunc.c @@ -30,8 +30,6 @@ so, delete this exception statement from your version. */ #include -#ifdef HAVE_SSL - #include #include #ifdef HAVE_UNISTD_H @@ -256,105 +254,91 @@ init_ssl (SSL_CTX **ctx) return 0; /* Succeded */ } -void -shutdown_ssl (SSL* con) +static +int ssl_read (int fd, char *buf, int bufsize, void *ctx) { - if (con == NULL) - return; - if (0==SSL_shutdown (con)) - SSL_shutdown (con); - SSL_free (con); + int res; + SSL *ssl = (SSL *) ctx; + /* #### Does SSL_read actually set EINTR? */ + do + res = SSL_read (ssl, buf, bufsize); + while (res == -1 && errno == EINTR); + return res; } -/* Sets up a SSL structure and performs the handshake on fd - Returns 0 if everything went right - Returns 1 if something went wrong ----- TODO: More exit codes -*/ -int -connect_ssl (SSL **con, SSL_CTX *ctx, int fd) +static int +ssl_write (int fd, char *buf, int bufsize, void *ctx) { - if (NULL == (*con = SSL_new (ctx))) - { - ssl_printerrors (); - return 1; - } - if (!SSL_set_fd (*con, fd)) - { - ssl_printerrors (); - return 1; - } - SSL_set_connect_state (*con); - switch (SSL_connect (*con)) - { - case 1 : - return (*con)->state != SSL_ST_OK; - default: - ssl_printerrors (); - shutdown_ssl (*con); - *con = NULL; - return 1; - case 0 : - ssl_printerrors (); - SSL_free (*con); - *con = NULL; - return 1; - } - return 0; + int res = 0; + SSL *ssl = (SSL *) ctx; + /* #### Does SSL_write actually set EINTR? */ + do + res = SSL_write (ssl, buf, bufsize); + while (res == -1 && errno == EINTR); + return res; } -void -free_ssl_ctx (SSL_CTX * ctx) +static int +ssl_poll (int fd, double timeout, int wait_for, void *ctx) { - SSL_CTX_free (ctx); + SSL *ssl = (SSL *) ctx; + if (timeout == 0) + return 1; + if (SSL_pending (ssl)) + return 1; +#ifdef HAVE_SELECT + return select_fd (fd, timeout, wait_for); +#else + return 1; +#endif } -/* SSL version of iread. Only exchanged read for SSL_read Read at - most LEN bytes from FD, storing them to BUF. */ - -int -ssl_iread (SSL *con, char *buf, int len) +static void +ssl_close (int fd, void *ctx) { - int res, fd; - BIO_get_fd (con->rbio, &fd); -#ifdef HAVE_SELECT - if (opt.read_timeout && !SSL_pending (con)) - if (select_fd (fd, opt.read_timeout, WAIT_READ) <= 0) - return -1; + SSL *ssl = (SSL *) ctx; + SSL_shutdown (ssl); + SSL_free (ssl); + +#ifdef WINDOWS + closesocket (fd); +#else + close (fd); #endif - do - res = SSL_read (con, buf, len); - while (res == -1 && errno == EINTR); - return res; + DEBUGP (("Closed %d/SSL 0x%0lx\n", fd, (unsigned long) ssl)); } -/* SSL version of iwrite. Only exchanged write for SSL_write Write - LEN bytes from BUF to FD. */ +/* Sets up a SSL structure and performs the handshake on fd. */ -int -ssl_iwrite (SSL *con, char *buf, int len) +SSL * +connect_ssl (int fd, SSL_CTX *ctx) { - int res = 0, fd; - BIO_get_fd (con->rbio, &fd); - /* `write' may write less than LEN bytes, thus the outward loop - keeps trying it until all was written, or an error occurred. The - inner loop is reserved for the usual EINTR f*kage, and the - innermost loop deals with the same during select(). */ - while (len > 0) - { -#ifdef HAVE_SELECT - if (opt.read_timeout) - if (select_fd (fd, opt.read_timeout, WAIT_WRITE) <= 0) - return -1; -#endif - do - res = SSL_write (con, buf, len); - while (res == -1 && errno == EINTR); - if (res <= 0) - break; - buf += res; - len -= res; - } - return res; + SSL *ssl = SSL_new (ctx); + if (!ssl) + goto err; + if (!SSL_set_fd (ssl, fd)) + goto err; + SSL_set_connect_state (ssl); + if (SSL_connect (ssl) <= 0 || ssl->state != SSL_ST_OK) + goto err; + + /* Register the FD to use our functions for read, write, etc. That + way the rest of Wget can keep using xread, xwrite, and + friends. */ + register_extended (fd, ssl_read, ssl_write, ssl_poll, ssl_close, ssl); + DEBUGP (("Connected %d to SSL 0x%0lx\n", fd, (unsigned long) ssl)); + return ssl; + + err: + ssl_printerrors (); + if (ssl) + SSL_free (ssl); + return NULL; +} + +void +free_ssl_ctx (SSL_CTX * ctx) +{ + SSL_CTX_free (ctx); } -#endif /* HAVE_SSL */ diff --git a/src/gen_sslfunc.h b/src/gen_sslfunc.h index acd70211..ece2249d 100644 --- a/src/gen_sslfunc.h +++ b/src/gen_sslfunc.h @@ -38,12 +38,9 @@ so, delete this exception statement from your version. */ void ssl_init_prng PARAMS ((void)); int init_ssl PARAMS ((SSL_CTX **)); -int connect_ssl PARAMS ((SSL **, SSL_CTX *, int)); -void shutdown_ssl PARAMS ((SSL*)); +SSL *connect_ssl PARAMS ((int, SSL_CTX *)); void free_ssl_ctx PARAMS ((SSL_CTX *)); int verify_callback PARAMS ((int, X509_STORE_CTX *)); -int ssl_iread PARAMS ((SSL *, char *, int)); -int ssl_iwrite PARAMS ((SSL *, char *, int)); int ssl_printerrors PARAMS ((void)); #endif /* GEN_SSLFUNC_H */ diff --git a/src/hash.c b/src/hash.c index faa7d6da..b72fac71 100644 --- a/src/hash.c +++ b/src/hash.c @@ -238,6 +238,9 @@ prime_size (int size, int *prime_offset) return 0; } +static unsigned long ptrhash PARAMS ((const void *)); +static int ptrcmp PARAMS ((const void *, const void *)); + /* Create a hash table with hash function HASH_FUNCTION and test function TEST_FUNCTION. The table is empty (its count is 0), but pre-allocated to store at least ITEMS items. @@ -612,7 +615,7 @@ make_nocase_string_hash_table (int items) pointer identity. (Common Lisp calls them EQ hash tables, and Java calls them IdentityHashMaps.) */ -unsigned long +static unsigned long ptrhash (const void *ptr) { unsigned long key = (unsigned long)ptr; @@ -637,29 +640,11 @@ ptrhash (const void *ptr) return key; } -int +static int ptrcmp (const void *ptr1, const void *ptr2) { return ptr1 == ptr2; } - -#if 0 -/* Currently unused: hashing of integers. */ - -unsigned long -inthash (unsigned int key) -{ - key += (key << 12); - key ^= (key >> 22); - key += (key << 4); - key ^= (key >> 9); - key += (key << 10); - key ^= (key >> 2); - key += (key << 7); - key ^= (key >> 12); - return key; -} -#endif #ifdef STANDALONE diff --git a/src/hash.h b/src/hash.h index a9b9aca2..d900d88d 100644 --- a/src/hash.h +++ b/src/hash.h @@ -69,7 +69,4 @@ int string_cmp PARAMS ((const void *, const void *)); struct hash_table *make_string_hash_table PARAMS ((int)); struct hash_table *make_nocase_string_hash_table PARAMS ((int)); -unsigned long ptrhash PARAMS ((const void *)); -int ptrcmp PARAMS ((const void *, const void *)); - #endif /* HASH_H */ diff --git a/src/http.c b/src/http.c index 2d27dbb5..d7b66a3a 100644 --- a/src/http.c +++ b/src/http.c @@ -192,17 +192,13 @@ parse_http_status_line (const char *line, const char **reason_phrase_ptr) longer, read only that much; if the file is shorter, report an error. */ static int -post_file (int sock, void *ssl, const char *file_name, long promised_size) +post_file (int sock, const char *file_name, long promised_size) { static char chunk[8192]; long written = 0; int write_error; FILE *fp; - /* Only one of SOCK and SSL may be active at the same time. */ - assert (sock > -1 || ssl != NULL); - assert (sock == -1 || ssl == NULL); - DEBUGP (("[writing POST file %s ... ", file_name)); fp = fopen (file_name, "rb"); @@ -215,12 +211,7 @@ post_file (int sock, void *ssl, const char *file_name, long promised_size) if (length == 0) break; towrite = WMIN (promised_size - written, length); -#ifdef HAVE_SSL - if (ssl) - write_error = ssl_iwrite (ssl, chunk, towrite); - else -#endif - write_error = iwrite (sock, chunk, towrite); + write_error = xwrite (sock, chunk, towrite, -1); if (write_error < 0) { fclose (fp); @@ -359,6 +350,7 @@ http_process_set_cookie (const char *hdr, void *arg) /* Whether a persistent connection is active. */ static int pc_active_p; + /* Host and port of currently active persistent connection. */ static struct address_list *pc_last_host_ip; static unsigned short pc_last_port; @@ -366,12 +358,8 @@ static unsigned short pc_last_port; /* File descriptor of the currently active persistent connection. */ static int pc_last_fd; -#ifdef HAVE_SSL /* Whether a ssl handshake has occoured on this connection */ -static int pc_active_ssl; -/* SSL connection of the currently active persistent connection. */ -static SSL *pc_last_ssl; -#endif /* HAVE_SSL */ +static int pc_last_ssl_p; /* Mark the persistent connection as invalid. This is used by the CLOSE_* macros after they forcefully close a registered persistent @@ -382,9 +370,7 @@ static void invalidate_persistent (void) { pc_active_p = 0; -#ifdef HAVE_SSL - pc_active_ssl = 0; -#endif /* HAVE_SSL */ + pc_last_ssl_p = 0; if (pc_last_host_ip != NULL) { address_list_release (pc_last_host_ip); @@ -401,15 +387,9 @@ invalidate_persistent (void) If a previous connection was persistent, it is closed. */ -#ifdef HAVE_SSL static void -register_persistent (const char *host, unsigned short port, int fd, SSL *ssl) +register_persistent (const char *host, unsigned short port, int fd, int ssl) { -#else -static void -register_persistent (const char *host, unsigned short port, int fd) -{ -#endif if (pc_active_p) { if (pc_last_fd == fd) @@ -425,13 +405,7 @@ register_persistent (const char *host, unsigned short port, int fd) persistent connection exists, but we then connect to a different host, and try to register a persistent connection to that one. */ -#ifdef HAVE_SSL - /* The ssl disconnect has to take place before the closing - of pc_last_fd. */ - if (pc_last_ssl) - shutdown_ssl(pc_last_ssl); -#endif - CLOSE (pc_last_fd); + xclose (pc_last_fd); invalidate_persistent (); } } @@ -446,34 +420,16 @@ register_persistent (const char *host, unsigned short port, int fd) pc_last_port = port; pc_last_fd = fd; pc_active_p = 1; -#ifdef HAVE_SSL - pc_last_ssl = ssl; - pc_active_ssl = ssl ? 1 : 0; -#endif + pc_last_ssl_p = ssl; DEBUGP (("Registered fd %d for persistent reuse.\n", fd)); } -#ifdef HAVE_SSL -# define SHUTDOWN_SSL(ssl) do { \ - if (ssl) \ - shutdown_ssl (ssl); \ -} while (0) -#else -# define SHUTDOWN_SSL(ssl) -#endif - /* Return non-zero if a persistent connection is available for connecting to HOST:PORT. */ -#ifdef HAVE_SSL static int persistent_available_p (const char *host, unsigned short port, int ssl) { -#else -static int -persistent_available_p (const char *host, unsigned short port) -{ -#endif int success; struct address_list *this_host_ip; @@ -485,15 +441,13 @@ persistent_available_p (const char *host, unsigned short port) if (port != pc_last_port) return 0; -#ifdef HAVE_SSL /* Second, a): check if current connection is (not) ssl, too. This test is unlikely to fail because HTTP and HTTPS typicaly use different ports. Yet it is possible, or so I [Christian Fraenkel] have been told, to run HTTPS and HTTP simultaneus on the same port. */ - if (ssl != pc_active_ssl) + if (ssl != pc_last_ssl_p) return 0; -#endif /* HAVE_SSL */ this_host_ip = lookup_host (host, LH_SILENT); if (!this_host_ip) @@ -518,11 +472,7 @@ persistent_available_p (const char *host, unsigned short port) /* Oops, the socket is no longer open. Now that we know that, let's invalidate the persistent connection before returning 0. */ - CLOSE (pc_last_fd); -#ifdef HAVE_SSL - SHUTDOWN_SSL (pc_last_ssl); - pc_last_ssl = NULL; -#endif + xclose (pc_last_fd); invalidate_persistent (); return 0; } @@ -547,16 +497,14 @@ persistent_available_p (const char *host, unsigned short port) #define CLOSE_FINISH(fd) do { \ if (!keep_alive) \ { \ - SHUTDOWN_SSL (ssl); \ - CLOSE (fd); \ + xclose (fd); \ if (pc_active_p && (fd) == pc_last_fd) \ invalidate_persistent (); \ } \ } while (0) #define CLOSE_INVALIDATE(fd) do { \ - SHUTDOWN_SSL (ssl); \ - CLOSE (fd); \ + xclose (fd); \ if (pc_active_p && (fd) == pc_last_fd) \ invalidate_persistent (); \ } while (0) @@ -637,8 +585,8 @@ gethttp (struct url *u, struct http_stat *hs, int *dt, struct url *proxy) struct rbuf rbuf; #ifdef HAVE_SSL static SSL_CTX *ssl_ctx = NULL; - SSL *ssl = NULL; #endif + int using_ssl = 0; char *cookies = NULL; /* Whether this connection will be kept alive after the HTTP request @@ -734,14 +682,13 @@ gethttp (struct url *u, struct http_stat *hs, int *dt, struct url *proxy) /* First: establish the connection. */ if (inhibit_keep_alive - || -#ifndef HAVE_SSL - !persistent_available_p (conn->host, conn->port) + || !persistent_available_p (conn->host, conn->port, +#ifdef HAVE_SSL + u->scheme == SCHEME_HTTPS #else - !persistent_available_p (conn->host, conn->port, - u->scheme == SCHEME_HTTPS) -#endif /* HAVE_SSL */ - ) + 0 +#endif + )) { sock = connect_to_host (conn->host, conn->port); if (sock == E_HOST) @@ -751,14 +698,17 @@ gethttp (struct url *u, struct http_stat *hs, int *dt, struct url *proxy) #ifdef HAVE_SSL if (conn->scheme == SCHEME_HTTPS) - if (connect_ssl (&ssl, ssl_ctx,sock) != 0) - { - logputs (LOG_VERBOSE, "\n"); - logprintf (LOG_NOTQUIET, _("Unable to establish SSL connection.\n")); - ssl_printerrors (); - CLOSE (sock); - return CONSSLERR; - } + { + if (!connect_ssl (sock, ssl_ctx)) + { + logputs (LOG_VERBOSE, "\n"); + logprintf (LOG_NOTQUIET, + _("Unable to establish SSL connection.\n")); + xclose (sock); + return CONSSLERR; + } + using_ssl = 1; + } #endif /* HAVE_SSL */ } else @@ -768,9 +718,7 @@ gethttp (struct url *u, struct http_stat *hs, int *dt, struct url *proxy) /* #### pc_last_fd should be accessed through an accessor function. */ sock = pc_last_fd; -#ifdef HAVE_SSL - ssl = pc_last_ssl; -#endif /* HAVE_SSL */ + using_ssl = pc_last_ssl_p; DEBUGP (("Reusing fd %d.\n", sock)); } @@ -991,36 +939,17 @@ Accept: %s\r\n\ xfree (full_path); /* Send the request to server. */ -#ifdef HAVE_SSL - if (conn->scheme == SCHEME_HTTPS) - write_error = ssl_iwrite (ssl, request, strlen (request)); - else -#endif - write_error = iwrite (sock, request, strlen (request)); + write_error = xwrite (sock, request, strlen (request), -1); if (write_error >= 0) { if (opt.post_data) { DEBUGP (("[POST data: %s]\n", opt.post_data)); -#ifdef HAVE_SSL - if (conn->scheme == SCHEME_HTTPS) - write_error = ssl_iwrite (ssl, opt.post_data, post_data_size); - else -#endif - write_error = iwrite (sock, opt.post_data, post_data_size); + write_error = xwrite (sock, opt.post_data, post_data_size, -1); } else if (opt.post_file_name && post_data_size != 0) - { -#ifdef HAVE_SSL - if (conn->scheme == SCHEME_HTTPS) - write_error = post_file (-1, ssl, opt.post_file_name, - post_data_size); - else -#endif - write_error = post_file (sock, NULL, opt.post_file_name, - post_data_size); - } + write_error = post_file (sock, opt.post_file_name, post_data_size); } DEBUGP (("---request end---\n")); @@ -1040,12 +969,6 @@ Accept: %s\r\n\ /* Before reading anything, initialize the rbuf. */ rbuf_initialize (&rbuf, sock); -#ifdef HAVE_SSL - if (conn->scheme == SCHEME_HTTPS) - rbuf.ssl = ssl; - else - rbuf.ssl = NULL; -#endif /* HAVE_SSL */ all_headers = NULL; all_length = 0; /* Header-fetching loop. */ @@ -1239,11 +1162,7 @@ Accept: %s\r\n\ if (keep_alive) /* The server has promised that it will not close the connection when we're done. This means that we can register it. */ -#ifndef HAVE_SSL - register_persistent (conn->host, conn->port, sock); -#else - register_persistent (conn->host, conn->port, sock, ssl); -#endif /* HAVE_SSL */ + register_persistent (conn->host, conn->port, sock, using_ssl); if ((statcode == HTTP_STATUS_UNAUTHORIZED) && authenticate_h) diff --git a/src/mswindows.h b/src/mswindows.h index 7d6f36ab..21f4b50f 100644 --- a/src/mswindows.h +++ b/src/mswindows.h @@ -98,10 +98,6 @@ so, delete this exception statement from your version. */ #define REALCLOSE(x) closesocket (x) -/* read & write don't work with sockets on Windows 95. */ -#define READ(fd, buf, cnt) recv ((fd), (buf), (cnt), 0) -#define WRITE(fd, buf, cnt) send ((fd), (buf), (cnt), 0) - /* #### Do we need this? */ #include diff --git a/src/rbuf.c b/src/rbuf.c index 7b11d67f..d9091711 100644 --- a/src/rbuf.c +++ b/src/rbuf.c @@ -42,24 +42,10 @@ so, delete this exception statement from your version. */ #include "rbuf.h" #include "connect.h" -#ifdef HAVE_SSL -#include -#include -#include -#include -#include -#include -#include "gen_sslfunc.h" /* for ssl_iread */ -#endif /* HAVE_SSL */ - void rbuf_initialize (struct rbuf *rbuf, int fd) { rbuf->fd = fd; -#ifdef HAVE_SSL -/* pointing ssl to NULL results in an unchanged behaviour */ - rbuf->ssl = NULL; -#endif /* HAVE_SSL */ rbuf->buffer_pos = rbuf->buffer; rbuf->buffer_left = 0; } @@ -79,12 +65,7 @@ rbuf_uninitialize (struct rbuf *rbuf) int rbuf_read_bufferful (struct rbuf *rbuf) { -#ifdef HAVE_SSL - if (rbuf->ssl) - return ssl_iread (rbuf->ssl, rbuf->buffer, sizeof (rbuf->buffer)); - else -#endif - return iread (rbuf->fd, rbuf->buffer, sizeof (rbuf->buffer)); + return xread (rbuf->fd, rbuf->buffer, sizeof (rbuf->buffer), -1); } /* Currently unused -- see RBUF_READCHAR. */ @@ -106,15 +87,7 @@ rbuf_peek (struct rbuf *rbuf, char *store) int res; rbuf->buffer_pos = rbuf->buffer; rbuf->buffer_left = 0; -#ifdef HAVE_SSL - if (rbuf->ssl != NULL) { - res = ssl_iread (rbuf->ssl, rbuf->buffer, sizeof (rbuf->buffer)); - } else { -#endif /* HAVE_SSL */ - res = iread (rbuf->fd, rbuf->buffer, sizeof (rbuf->buffer)); -#ifdef HAVE_SSL - } -#endif /* HAVE_SSL */ + res = xread (rbuf->fd, rbuf->buffer, sizeof (rbuf->buffer), -1); if (res <= 0) return res; rbuf->buffer_left = res; diff --git a/src/rbuf.h b/src/rbuf.h index 2b61bcf2..b0426377 100644 --- a/src/rbuf.h +++ b/src/rbuf.h @@ -30,17 +30,10 @@ so, delete this exception statement from your version. */ #ifndef RBUF_H #define RBUF_H -#ifdef HAVE_SSL -# include -#endif - /* Retrieval stream */ struct rbuf { int fd; -#ifdef HAVE_SSL - SSL *ssl; /* the ssl structure -- replaces fd for ssl connections */ -#endif /* HAVE_SSL */ char buffer[4096]; /* the input buffer */ char *buffer_pos; /* current position in the buffer */ size_t buffer_left; /* number of bytes left in the buffer: @@ -75,9 +68,6 @@ struct rbuf /* Return the file descriptor of RBUF. */ #define RBUF_FD(rbuf) ((rbuf)->fd) -/* Return the file descriptor of RBUF. */ -#define RBUF_SSL(rbuf) ((rbuf)->ssl) - /* Function declarations */ void rbuf_initialize PARAMS ((struct rbuf *, int)); int rbuf_initialized_p PARAMS ((struct rbuf *)); diff --git a/src/retr.c b/src/retr.c index c3da1641..db8b1c18 100644 --- a/src/retr.c +++ b/src/retr.c @@ -213,12 +213,7 @@ get_contents (int fd, FILE *fp, long *len, long restval, long expected, { int amount_to_read = (use_expected ? MIN (expected - *len, dlbufsize) : dlbufsize); -#ifdef HAVE_SSL - if (rbuf->ssl!=NULL) - res = ssl_iread (rbuf->ssl, dlbuf, amount_to_read); - else -#endif /* HAVE_SSL */ - res = iread (fd, dlbuf, amount_to_read); + res = xread (fd, dlbuf, amount_to_read, -1); if (res <= 0) break; diff --git a/src/sysdep.h b/src/sysdep.h index b888caaf..9b875e09 100644 --- a/src/sysdep.h +++ b/src/sysdep.h @@ -108,33 +108,6 @@ so, delete this exception statement from your version. */ #endif #endif -#ifdef __BEOS__ -# undef READ -# undef WRITE -# define READ(fd, buf, cnt) recv ((fd), (buf), (cnt), 0) -# define WRITE(fd, buf, cnt) send ((fd), (buf), (cnt), 0) -#endif - -/* mswindows.h defines these. */ -#ifndef READ -# define READ(fd, buf, cnt) read ((fd), (buf), (cnt)) -#endif -#ifndef WRITE -# define WRITE(fd, buf, cnt) write ((fd), (buf), (cnt)) -#endif -#ifndef REALCLOSE -# define REALCLOSE(x) close (x) -#endif - -#define CLOSE(x) do { \ - int C_sock = (x); \ - if (C_sock >= 0) \ - { \ - REALCLOSE (C_sock); \ - DEBUGP (("Closing fd %d\n", C_sock)); \ - } \ -} while (0) - /* Define a large integral type useful for storing large sizes that exceed sizes of one download, such as when printing the sum of all downloads. Note that this has nothing to do with large file -- 2.39.2