handle_established_connection(): Add parentheses to clarify an expression
[privoxy.git] / jcc.c
1 /*********************************************************************
2  *
3  * File        :  $Source: /cvsroot/ijbswa/current/jcc.c,v $
4  *
5  * Purpose     :  Main file.  Contains main() method, main loop, and
6  *                the main connection-handling function.
7  *
8  * Copyright   :  Written by and Copyright (C) 2001-2021 the
9  *                Privoxy team. https://www.privoxy.org/
10  *
11  *                Based on the Internet Junkbuster originally written
12  *                by and Copyright (C) 1997 Anonymous Coders and
13  *                Junkbusters Corporation.  http://www.junkbusters.com
14  *
15  *                This program is free software; you can redistribute it
16  *                and/or modify it under the terms of the GNU General
17  *                Public License as published by the Free Software
18  *                Foundation; either version 2 of the License, or (at
19  *                your option) any later version.
20  *
21  *                This program is distributed in the hope that it will
22  *                be useful, but WITHOUT ANY WARRANTY; without even the
23  *                implied warranty of MERCHANTABILITY or FITNESS FOR A
24  *                PARTICULAR PURPOSE.  See the GNU General Public
25  *                License for more details.
26  *
27  *                The GNU General Public License should be included with
28  *                this file.  If not, you can view it at
29  *                http://www.gnu.org/copyleft/gpl.html
30  *                or write to the Free Software Foundation, Inc., 59
31  *                Temple Place - Suite 330, Boston, MA  02111-1307, USA.
32  *
33  *********************************************************************/
34
35
36 #include "config.h"
37
38 #include <stdio.h>
39 #include <sys/types.h>
40 #include <stdlib.h>
41 #include <string.h>
42 #include <signal.h>
43 #include <fcntl.h>
44 #include <errno.h>
45 #include <assert.h>
46
47 #ifdef _WIN32
48 # ifndef FEATURE_PTHREAD
49 #  ifndef STRICT
50 #   define STRICT
51 #  endif
52 #  include <winsock2.h>
53 #  include <windows.h>
54 #  include <process.h>
55 # endif /* ndef FEATURE_PTHREAD */
56
57 # include "win32.h"
58 # ifndef _WIN_CONSOLE
59 #  include "w32log.h"
60 # endif /* ndef _WIN_CONSOLE */
61 # include "w32svrapi.h"
62
63 #else /* ifndef _WIN32 */
64
65 # include <unistd.h>
66 # include <sys/wait.h>
67 # include <sys/time.h>
68 # include <sys/stat.h>
69 # include <sys/ioctl.h>
70
71 #ifdef sun
72 #include <sys/termios.h>
73 #endif /* sun */
74
75 #ifdef unix
76 #include <pwd.h>
77 #include <grp.h>
78 #endif
79
80 # include <signal.h>
81
82 # ifdef __BEOS__
83 #  include <socket.h>  /* BeOS has select() for sockets only. */
84 #  include <OS.h>      /* declarations for threads and stuff. */
85 # endif
86
87 #ifdef HAVE_POLL
88 #ifdef __GLIBC__
89 #include <sys/poll.h>
90 #else
91 #include <poll.h>
92 #endif /* def __GLIBC__ */
93 #else
94 # ifndef FD_ZERO
95 #  include <select.h>
96 # endif
97 #warning poll() appears to be unavailable. Your platform will become unsupported in the future.
98 #endif /* HAVE_POLL */
99
100 #endif
101
102 #include "project.h"
103 #include "list.h"
104 #include "jcc.h"
105 #ifdef FEATURE_HTTPS_INSPECTION
106 #include "ssl.h"
107 #endif
108 #include "filters.h"
109 #include "loaders.h"
110 #include "parsers.h"
111 #include "miscutil.h"
112 #include "errlog.h"
113 #include "jbsockets.h"
114 #include "gateway.h"
115 #include "actions.h"
116 #include "cgi.h"
117 #include "loadcfg.h"
118 #include "urlmatch.h"
119 #ifdef FEATURE_CLIENT_TAGS
120 #include "client-tags.h"
121 #endif
122
123 int daemon_mode = 1;
124 struct client_states clients[1];
125 struct file_list     files[1];
126
127 #ifdef FEATURE_STATISTICS
128 int urls_read     = 0;     /* total nr of urls read inc rejected */
129 int urls_rejected = 0;     /* total nr of urls rejected */
130 #endif /* def FEATURE_STATISTICS */
131
132 #ifdef FEATURE_GRACEFUL_TERMINATION
133 int g_terminate = 0;
134 #endif
135
136 #if !defined(_WIN32)
137 static void sig_handler(int the_signal);
138 #endif
139 static int client_protocol_is_unsupported(struct client_state *csp, char *req);
140 static jb_err get_request_destination_elsewhere(struct client_state *csp, struct list *headers);
141 static jb_err get_server_headers(struct client_state *csp);
142 static const char *crunch_reason(const struct http_response *rsp);
143 static void send_crunch_response(struct client_state *csp, struct http_response *rsp);
144 static char *get_request_line(struct client_state *csp);
145 static jb_err receive_client_request(struct client_state *csp);
146 static jb_err parse_client_request(struct client_state *csp);
147 static void build_request_line(struct client_state *csp, const struct forward_spec *fwd, char **request_line);
148 static jb_err change_request_destination(struct client_state *csp);
149 static void handle_established_connection(struct client_state *csp);
150 static void chat(struct client_state *csp);
151 static void serve(struct client_state *csp);
152 #if !defined(_WIN32) || defined(_WIN_CONSOLE)
153 static void usage(const char *myname);
154 #endif
155 static void initialize_mutexes(void);
156 static jb_socket bind_port_helper(const char *haddr, int hport, int backlog);
157 static void bind_ports_helper(struct configuration_spec *config, jb_socket sockets[]);
158 static void close_ports_helper(jb_socket sockets[]);
159 static void listen_loop(void);
160 static void serve(struct client_state *csp);
161
162 #ifdef __BEOS__
163 static int32 server_thread(void *data);
164 #endif /* def __BEOS__ */
165
166 #ifdef _WIN32
167 #define sleep(N)  Sleep(((N) * 1000))
168 #endif
169
170 #ifdef FUZZ
171 int process_fuzzed_input(char *fuzz_input_type, char *fuzz_input_file);
172 void show_fuzz_usage(const char *name);
173 #endif
174
175 #ifdef MUTEX_LOCKS_AVAILABLE
176 /*
177  * XXX: Does the locking stuff really belong in this file?
178  */
179 privoxy_mutex_t log_mutex;
180 privoxy_mutex_t log_init_mutex;
181 privoxy_mutex_t connection_reuse_mutex;
182
183 #ifdef FEATURE_HTTPS_INSPECTION
184 privoxy_mutex_t certificate_mutex;
185 privoxy_mutex_t ssl_init_mutex;
186 #endif
187
188 #ifdef FEATURE_EXTERNAL_FILTERS
189 privoxy_mutex_t external_filter_mutex;
190 #endif
191 #ifdef FEATURE_CLIENT_TAGS
192 privoxy_mutex_t client_tags_mutex;
193 #endif
194 #ifdef FEATURE_EXTENDED_STATISTICS
195 privoxy_mutex_t filter_statistics_mutex;
196 privoxy_mutex_t block_statistics_mutex;
197 #endif
198
199 #if !defined(HAVE_GETHOSTBYADDR_R) || !defined(HAVE_GETHOSTBYNAME_R)
200 privoxy_mutex_t resolver_mutex;
201 #endif /* !defined(HAVE_GETHOSTBYADDR_R) || !defined(HAVE_GETHOSTBYNAME_R) */
202
203 #ifndef HAVE_GMTIME_R
204 privoxy_mutex_t gmtime_mutex;
205 #endif /* ndef HAVE_GMTIME_R */
206
207 #ifndef HAVE_LOCALTIME_R
208 privoxy_mutex_t localtime_mutex;
209 #endif /* ndef HAVE_GMTIME_R */
210
211 #if !defined(HAVE_ARC4RANDOM) && !defined(HAVE_RANDOM)
212 privoxy_mutex_t rand_mutex;
213 #endif /* !defined(HAVE_ARC4RANDOM) && !defined(HAVE_RANDOM) */
214
215 #endif /* def MUTEX_LOCKS_AVAILABLE */
216
217 #if defined(unix)
218 const char *basedir = NULL;
219 const char *pidfile = NULL;
220 static int received_hup_signal = 0;
221 #endif /* defined unix */
222
223 /* HTTP snipplets. */
224 static const char CSUCCEED[] =
225    "HTTP/1.1 200 Connection established\r\n\r\n";
226
227 static const char CHEADER[] =
228    "HTTP/1.1 400 Invalid header received from client\r\n"
229    "Content-Type: text/plain\r\n"
230    "Connection: close\r\n\r\n"
231    "Invalid header received from client.\r\n";
232
233 static const char FTP_RESPONSE[] =
234    "HTTP/1.1 400 Invalid request received from client\r\n"
235    "Content-Type: text/plain\r\n"
236    "Connection: close\r\n\r\n"
237    "Invalid request. Privoxy doesn't support FTP.\r\n";
238
239 static const char GOPHER_RESPONSE[] =
240    "HTTP/1.1 400 Invalid request received from client\r\n"
241    "Content-Type: text/plain\r\n"
242    "Connection: close\r\n\r\n"
243    "Invalid request. Privoxy doesn't support gopher.\r\n";
244
245 /* XXX: should be a template */
246 static const char MISSING_DESTINATION_RESPONSE[] =
247    "HTTP/1.1 400 Bad request received from client\r\n"
248    "Content-Type: text/plain\r\n"
249    "Connection: close\r\n\r\n"
250    "Bad request. Privoxy was unable to extract the destination.\r\n";
251
252 /* XXX: should be a template */
253 static const char INVALID_SERVER_HEADERS_RESPONSE[] =
254    "HTTP/1.1 502 Server or forwarder response invalid\r\n"
255    "Content-Type: text/plain\r\n"
256    "Connection: close\r\n\r\n"
257    "Bad response. The server or forwarder response doesn't look like HTTP.\r\n";
258
259 /* XXX: should be a template */
260 static const char MESSED_UP_REQUEST_RESPONSE[] =
261    "HTTP/1.1 400 Malformed request after rewriting\r\n"
262    "Content-Type: text/plain\r\n"
263    "Connection: close\r\n\r\n"
264    "Bad request. Messed up with header filters.\r\n";
265
266 static const char TOO_MANY_CONNECTIONS_RESPONSE[] =
267    "HTTP/1.1 503 Too many open connections\r\n"
268    "Content-Type: text/plain\r\n"
269    "Connection: close\r\n\r\n"
270    "Maximum number of open connections reached.\r\n";
271
272 static const char CLIENT_CONNECTION_TIMEOUT_RESPONSE[] =
273    "HTTP/1.1 504 Connection timeout\r\n"
274    "Content-Type: text/plain\r\n"
275    "Connection: close\r\n\r\n"
276    "The connection timed out because the client request didn't arrive in time.\r\n";
277
278 static const char CLIENT_BODY_PARSE_ERROR_RESPONSE[] =
279    "HTTP/1.1 400 Failed reading client body\r\n"
280    "Content-Type: text/plain\r\n"
281    "Connection: close\r\n\r\n"
282    "Failed parsing or buffering the chunk-encoded client body.\r\n";
283
284 static const char UNSUPPORTED_CLIENT_EXPECTATION_ERROR_RESPONSE[] =
285    "HTTP/1.1 417 Expecting too much\r\n"
286    "Content-Type: text/plain\r\n"
287    "Connection: close\r\n\r\n"
288    "Privoxy detected an unsupported Expect header value.\r\n";
289
290 /* A function to crunch a response */
291 typedef struct http_response *(*crunch_func_ptr)(struct client_state *);
292
293 /* Crunch function flags */
294 #define CF_NO_FLAGS        0
295 /* Cruncher applies to forced requests as well */
296 #define CF_IGNORE_FORCE    1
297 /* Crunched requests are counted for the block statistics */
298 #define CF_COUNT_AS_REJECT 2
299
300 /* A crunch function and its flags */
301 struct cruncher
302 {
303    const crunch_func_ptr cruncher;
304    const int flags;
305 };
306
307 static int crunch_response_triggered(struct client_state *csp, const struct cruncher crunchers[]);
308
309 /* Complete list of cruncher functions */
310 static const struct cruncher crunchers_all[] = {
311    { direct_response, CF_COUNT_AS_REJECT|CF_IGNORE_FORCE},
312    { block_url,       CF_COUNT_AS_REJECT },
313 #ifdef FEATURE_TRUST
314    { trust_url,       CF_COUNT_AS_REJECT },
315 #endif /* def FEATURE_TRUST */
316    { redirect_url,    CF_NO_FLAGS  },
317    { dispatch_cgi,    CF_IGNORE_FORCE},
318    { NULL,            0 }
319 };
320
321 /* Light version, used after tags are applied */
322 static const struct cruncher crunchers_light[] = {
323    { block_url,       CF_COUNT_AS_REJECT },
324    { redirect_url,    CF_NO_FLAGS },
325    { NULL,            0 }
326 };
327
328
329 /*
330  * XXX: Don't we really mean
331  *
332  * #if defined(unix)
333  *
334  * here?
335  */
336 #if !defined(_WIN32)
337 /*********************************************************************
338  *
339  * Function    :  sig_handler
340  *
341  * Description :  Signal handler for different signals.
342  *                Exit gracefully on TERM and INT
343  *                or set a flag that will cause the errlog
344  *                to be reopened by the main thread on HUP.
345  *
346  * Parameters  :
347  *          1  :  the_signal = the signal cause this function to call
348  *
349  * Returns     :  -
350  *
351  *********************************************************************/
352 static void sig_handler(int the_signal)
353 {
354    switch(the_signal)
355    {
356       case SIGTERM:
357       case SIGINT:
358          log_error(LOG_LEVEL_INFO, "exiting by signal %d .. bye", the_signal);
359 #if defined(unix)
360          if (pidfile)
361          {
362             unlink(pidfile);
363          }
364 #endif /* unix */
365          exit(the_signal);
366          break;
367
368       case SIGHUP:
369 #if defined(unix)
370          received_hup_signal = 1;
371 #endif
372          break;
373
374       default:
375          /*
376           * We shouldn't be here, unless we catch signals
377           * in main() that we can't handle here!
378           */
379          log_error(LOG_LEVEL_FATAL,
380             "sig_handler: exiting on unexpected signal %d", the_signal);
381    }
382    return;
383
384 }
385 #endif
386
387
388 /*********************************************************************
389  *
390  * Function    :  get_write_delay
391  *
392  * Description :  Parse the delay-response parameter.
393  *
394  * Parameters  :
395  *          1  :  csp = Current client state (buffers, headers, etc...)
396  *
397  * Returns     :  Number of milliseconds to delay writes.
398  *
399  *********************************************************************/
400 static unsigned int get_write_delay(const struct client_state *csp)
401 {
402    unsigned int delay;
403    char *endptr;
404    char *newval;
405
406    if ((csp->action->flags & ACTION_DELAY_RESPONSE) == 0)
407    {
408       return 0;
409    }
410    newval = csp->action->string[ACTION_STRING_DELAY_RESPONSE];
411
412    delay = (unsigned)strtol(newval, &endptr, 0);
413    if (*endptr != '\0')
414    {
415       log_error(LOG_LEVEL_FATAL,
416          "Invalid delay-response{} parameter: '%s'", newval);
417    }
418
419    return delay;
420
421 }
422
423
424 /*********************************************************************
425  *
426  * Function    :  client_protocol_is_unsupported
427  *
428  * Description :  Checks if the client used a known unsupported
429  *                protocol and deals with it by sending an error
430  *                response.
431  *
432  * Parameters  :
433  *          1  :  csp = Current client state (buffers, headers, etc...)
434  *          2  :  req = the first request line send by the client
435  *
436  * Returns     :  TRUE if an error response has been generated, or
437  *                FALSE if the request doesn't look invalid.
438  *
439  *********************************************************************/
440 static int client_protocol_is_unsupported(struct client_state *csp, char *req)
441 {
442    /*
443     * If it's a FTP or gopher request, we don't support it.
444     *
445     * These checks are better than nothing, but they might
446     * not work in all configurations and some clients might
447     * have problems digesting the answer.
448     *
449     * They should, however, never cause more problems than
450     * Privoxy's old behaviour (returning the misleading HTML
451     * error message:
452     *
453     * "Could not resolve http://(ftp|gopher)://example.org").
454     */
455    if (!strncmpic(req, "GET ftp://", 10) || !strncmpic(req, "GET gopher://", 13))
456    {
457       const char *response = NULL;
458       const char *protocol = NULL;
459
460       if (!strncmpic(req, "GET ftp://", 10))
461       {
462          response = FTP_RESPONSE;
463          protocol = "FTP";
464       }
465       else
466       {
467          response = GOPHER_RESPONSE;
468          protocol = "GOPHER";
469       }
470       log_error(LOG_LEVEL_ERROR,
471          "%s tried to use Privoxy as %s proxy: %s",
472          csp->ip_addr_str, protocol, req);
473       log_error(LOG_LEVEL_CLF,
474          "%s - - [%T] \"%s\" 400 0", csp->ip_addr_str, req);
475       freez(req);
476
477 #ifdef FEATURE_HTTPS_INSPECTION
478       if (client_use_ssl(csp))
479       {
480          ssl_send_data_delayed(&(csp->ssl_client_attr),
481             (const unsigned char *)response, strlen(response),
482             get_write_delay(csp));
483       }
484       else
485 #endif
486       {
487          write_socket_delayed(csp->cfd, response, strlen(response),
488             get_write_delay(csp));
489       }
490
491       return TRUE;
492    }
493
494    return FALSE;
495 }
496
497
498 /*********************************************************************
499  *
500  * Function    :  client_has_unsupported_expectations
501  *
502  * Description :  Checks if the client used an unsupported expectation
503  *                in which case an error message is delivered.
504  *
505  * Parameters  :
506  *          1  :  csp = Current client state (buffers, headers, etc...)
507  *
508  * Returns     :  TRUE if an error response has been generated, or
509  *                FALSE if the request doesn't look invalid.
510  *
511  *********************************************************************/
512 static int client_has_unsupported_expectations(const struct client_state *csp)
513 {
514    if ((csp->flags & CSP_FLAG_UNSUPPORTED_CLIENT_EXPECTATION))
515    {
516       log_error(LOG_LEVEL_ERROR,
517          "Rejecting request from client %s with unsupported Expect header value",
518          csp->ip_addr_str);
519       log_error(LOG_LEVEL_CLF,
520          "%s - - [%T] \"%s\" 417 0", csp->ip_addr_str, csp->http->cmd);
521       write_socket_delayed(csp->cfd,
522          UNSUPPORTED_CLIENT_EXPECTATION_ERROR_RESPONSE,
523          strlen(UNSUPPORTED_CLIENT_EXPECTATION_ERROR_RESPONSE),
524          get_write_delay(csp));
525
526       return TRUE;
527    }
528
529    return FALSE;
530
531 }
532
533
534 /*********************************************************************
535  *
536  * Function    :  get_request_destination_elsewhere
537  *
538  * Description :  If the client's request was redirected into
539  *                Privoxy without the client's knowledge,
540  *                the request line lacks the destination host.
541  *
542  *                This function tries to get it elsewhere,
543  *                provided accept-intercepted-requests is enabled.
544  *
545  *                "Elsewhere" currently only means "Host: header",
546  *                but in the future we may ask the redirecting
547  *                packet filter to look the destination up.
548  *
549  *                If the destination stays unknown, an error
550  *                response is send to the client and headers
551  *                are freed so that chat() can return directly.
552  *
553  * Parameters  :
554  *          1  :  csp = Current client state (buffers, headers, etc...)
555  *          2  :  headers = a header list
556  *
557  * Returns     :  JB_ERR_OK if the destination is now known, or
558  *                JB_ERR_PARSE if it isn't.
559  *
560  *********************************************************************/
561 static jb_err get_request_destination_elsewhere(struct client_state *csp, struct list *headers)
562 {
563    if (!(csp->config->feature_flags & RUNTIME_FEATURE_ACCEPT_INTERCEPTED_REQUESTS))
564    {
565       log_error(LOG_LEVEL_ERROR, "%s's request: \'%s\' is invalid."
566          " Privoxy isn't configured to accept intercepted requests.",
567          csp->ip_addr_str, csp->http->cmd);
568       /* XXX: Use correct size */
569       log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 400 0",
570          csp->ip_addr_str, csp->http->cmd);
571
572       write_socket_delayed(csp->cfd, CHEADER, strlen(CHEADER),
573          get_write_delay(csp));
574       destroy_list(headers);
575
576       return JB_ERR_PARSE;
577    }
578    else if (JB_ERR_OK == get_destination_from_headers(headers, csp->http))
579    {
580       /* Split the domain we just got for pattern matching */
581       init_domain_components(csp->http);
582
583       return JB_ERR_OK;
584    }
585    else
586    {
587       /* We can't work without destination. Go spread the news.*/
588
589       /* XXX: Use correct size */
590       log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 400 0",
591          csp->ip_addr_str, csp->http->cmd);
592       log_error(LOG_LEVEL_ERROR,
593          "Privoxy was unable to get the destination for %s's request: %s",
594          csp->ip_addr_str, csp->http->cmd);
595
596       write_socket_delayed(csp->cfd, MISSING_DESTINATION_RESPONSE,
597          strlen(MISSING_DESTINATION_RESPONSE), get_write_delay(csp));
598       destroy_list(headers);
599
600       return JB_ERR_PARSE;
601    }
602    /*
603     * TODO: If available, use PF's ioctl DIOCNATLOOK as last resort
604     * to get the destination IP address, use it as host directly
605     * or do a reverse DNS lookup first.
606     */
607 }
608
609
610 /*********************************************************************
611  *
612  * Function    :  get_server_headers
613  *
614  * Description :  Parses server headers in iob and fills them
615  *                into csp->headers so that they can later be
616  *                handled by sed().
617  *
618  * Parameters  :
619  *          1  :  csp = Current client state (buffers, headers, etc...)
620  *
621  * Returns     :  JB_ERR_OK if everything went fine, or
622  *                JB_ERR_PARSE if the headers were incomplete.
623  *
624  *********************************************************************/
625 static jb_err get_server_headers(struct client_state *csp)
626 {
627    int continue_hack_in_da_house = 0;
628    char * header;
629
630    while (((header = get_header(csp->iob)) != NULL) || continue_hack_in_da_house)
631    {
632       if (header == NULL)
633       {
634          /*
635           * continue hack in da house. Ignore the ending of
636           * this head and continue enlisting header lines.
637           * The reason is described below.
638           */
639          enlist(csp->headers, "");
640          continue_hack_in_da_house = 0;
641          continue;
642       }
643       else if (0 == strncmpic(header, "HTTP/1.1 100", 12))
644       {
645          /*
646           * It's a bodyless continue response, don't
647           * stop header parsing after reaching its end.
648           *
649           * As a result Privoxy will concatenate the
650           * next response's head and parse and deliver
651           * the headers as if they belonged to one request.
652           *
653           * The client will separate them because of the
654           * empty line between them.
655           *
656           * XXX: What we're doing here is clearly against
657           * the intended purpose of the continue header,
658           * and under some conditions (HTTP/1.0 client request)
659           * it's a standard violation.
660           *
661           * Anyway, "sort of against the spec" is preferable
662           * to "always getting confused by Continue responses"
663           * (Privoxy's behaviour before this hack was added)
664           */
665          log_error(LOG_LEVEL_HEADER, "Continue hack in da house.");
666          continue_hack_in_da_house = 1;
667       }
668       else if (*header == '\0')
669       {
670          /*
671           * If the header is empty, but the Continue hack
672           * isn't active, we can assume that we reached the
673           * end of the buffer before we hit the end of the
674           * head.
675           *
676           * Inform the caller an let it decide how to handle it.
677           */
678          return JB_ERR_PARSE;
679       }
680
681       if (JB_ERR_MEMORY == enlist(csp->headers, header))
682       {
683          /*
684           * XXX: Should we quit the request and return a
685           * out of memory error page instead?
686           */
687          log_error(LOG_LEVEL_ERROR,
688             "Out of memory while enlisting server headers. %s lost.",
689             header);
690       }
691       freez(header);
692    }
693
694    return JB_ERR_OK;
695 }
696
697
698 /*********************************************************************
699  *
700  * Function    :  crunch_reason
701  *
702  * Description :  Translates the crunch reason code into a string.
703  *
704  * Parameters  :
705  *          1  :  rsp = a http_response
706  *
707  * Returns     :  A string with the crunch reason or an error description.
708  *
709  *********************************************************************/
710 static const char *crunch_reason(const struct http_response *rsp)
711 {
712    char * reason = NULL;
713
714    assert(rsp != NULL);
715    if (rsp == NULL)
716    {
717       return "Internal error while searching for crunch reason";
718    }
719
720    switch (rsp->crunch_reason)
721    {
722       case UNSUPPORTED:
723          reason = "Unsupported HTTP feature";
724          break;
725       case BLOCKED:
726          reason = "Blocked";
727          break;
728       case UNTRUSTED:
729          reason = "Untrusted";
730          break;
731       case REDIRECTED:
732          reason = "Redirected";
733          break;
734       case CGI_CALL:
735          reason = "CGI Call";
736          break;
737       case NO_SUCH_DOMAIN:
738          reason = "DNS failure";
739          break;
740       case FORWARDING_FAILED:
741          reason = "Forwarding failed";
742          break;
743       case CONNECT_FAILED:
744          reason = "Connection failure";
745          break;
746       case OUT_OF_MEMORY:
747          reason = "Out of memory (may mask other reasons)";
748          break;
749       case CONNECTION_TIMEOUT:
750          reason = "Connection timeout";
751          break;
752       case NO_SERVER_DATA:
753          reason = "No server data received";
754          break;
755       default:
756          reason = "No reason recorded";
757          break;
758    }
759
760    return reason;
761 }
762
763
764 /*********************************************************************
765  *
766  * Function    :  log_applied_actions
767  *
768  * Description :  Logs the applied actions if LOG_LEVEL_ACTIONS is
769  *                enabled.
770  *
771  * Parameters  :
772  *          1  :  actions = Current action spec to log
773  *
774  * Returns     :  Nothing.
775  *
776  *********************************************************************/
777 static void log_applied_actions(const struct current_action_spec *actions)
778 {
779    /*
780     * The conversion to text requires lots of memory allocations so
781     * we only do the conversion if the user is actually interested.
782     */
783    if (debug_level_is_enabled(LOG_LEVEL_ACTIONS))
784    {
785       char *actions_as_text = actions_to_line_of_text(actions);
786       log_error(LOG_LEVEL_ACTIONS, "%s", actions_as_text);
787       freez(actions_as_text);
788    }
789 }
790
791
792 /*********************************************************************
793  *
794  * Function    :  send_crunch_response
795  *
796  * Description :  Delivers already prepared response for
797  *                intercepted requests, logs the interception
798  *                and frees the response.
799  *
800  * Parameters  :
801  *          1  :  csp = Current client state (buffers, headers, etc...)
802  *          2  :  rsp = Fully prepared response. Will be freed on exit.
803  *
804  * Returns     :  Nothing.
805  *
806  *********************************************************************/
807 static void send_crunch_response(struct client_state *csp, struct http_response *rsp)
808 {
809       const struct http_request *http = csp->http;
810       char status_code[4];
811
812       assert(rsp != NULL);
813       assert(rsp->head != NULL);
814
815       if (rsp == NULL)
816       {
817          log_error(LOG_LEVEL_FATAL, "NULL response in send_crunch_response.");
818       }
819
820       /*
821        * Extract the status code from the actual head
822        * that will be send to the client. It is the only
823        * way to get it right for all requests, including
824        * the fixed ones for out-of-memory problems.
825        *
826        * A head starts like this: 'HTTP/1.1 200...'
827        *                           0123456789|11
828        *                                     10
829        */
830       status_code[0] = rsp->head[9];
831       status_code[1] = rsp->head[10];
832       status_code[2] = rsp->head[11];
833       status_code[3] = '\0';
834
835       /* Log that the request was crunched and why. */
836       log_applied_actions(csp->action);
837 #ifdef FEATURE_HTTPS_INSPECTION
838       if (client_use_ssl(csp))
839       {
840          log_error(LOG_LEVEL_CRUNCH, "%s: https://%s%s", crunch_reason(rsp),
841             http->hostport, http->path);
842          log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s https://%s%s %s\" %s %lu",
843             csp->ip_addr_str, http->gpc, http->hostport, http->path,
844             http->version, status_code, rsp->content_length);
845       }
846       else
847 #endif
848       {
849          log_error(LOG_LEVEL_CRUNCH, "%s: %s", crunch_reason(rsp), http->url);
850          log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" %s %lu",
851             csp->ip_addr_str, http->ocmd, status_code, rsp->content_length);
852       }
853       /* Write the answer to the client */
854 #ifdef FEATURE_HTTPS_INSPECTION
855       if (client_use_ssl(csp))
856       {
857          if ((ssl_send_data_delayed(&(csp->ssl_client_attr),
858                 (const unsigned char *)rsp->head, rsp->head_length,
859                 get_write_delay(csp)) < 0)
860           || (ssl_send_data_delayed(&(csp->ssl_client_attr),
861                 (const unsigned char *)rsp->body, rsp->content_length,
862                 get_write_delay(csp)) < 0))
863          {
864             /* There is nothing we can do about it. */
865             log_error(LOG_LEVEL_CONNECT, "Couldn't deliver the error message "
866                "for https://%s%s through client socket %d using TLS/SSL",
867                http->hostport, http->url, csp->cfd);
868          }
869       }
870       else
871 #endif
872       {
873          if (write_socket_delayed(csp->cfd, rsp->head, rsp->head_length,
874                 get_write_delay(csp))
875           || write_socket_delayed(csp->cfd, rsp->body, rsp->content_length,
876                 get_write_delay(csp)))
877          {
878             /* There is nothing we can do about it. */
879             log_error(LOG_LEVEL_CONNECT,
880                "Couldn't deliver the error message for %s through client socket %d: %E",
881                http->url, csp->cfd);
882          }
883       }
884
885       /* Clean up and return */
886       if (cgi_error_memory() != rsp)
887       {
888          free_http_response(rsp);
889       }
890       return;
891 }
892
893
894 /*********************************************************************
895  *
896  * Function    :  crunch_response_triggered
897  *
898  * Description :  Checks if the request has to be crunched,
899  *                and delivers the crunch response if necessary.
900  *
901  * Parameters  :
902  *          1  :  csp = Current client state (buffers, headers, etc...)
903  *          2  :  crunchers = list of cruncher functions to run
904  *
905  * Returns     :  TRUE if the request was answered with a crunch response
906  *                FALSE otherwise.
907  *
908  *********************************************************************/
909 static int crunch_response_triggered(struct client_state *csp, const struct cruncher crunchers[])
910 {
911    struct http_response *rsp = NULL;
912    const struct cruncher *c;
913
914    /*
915     * If CGI request crunching is disabled,
916     * check the CGI dispatcher out of order to
917     * prevent unintentional blocks or redirects.
918     */
919    if (!(csp->config->feature_flags & RUNTIME_FEATURE_CGI_CRUNCHING)
920        && (NULL != (rsp = dispatch_cgi(csp))))
921    {
922       /* Deliver, log and free the interception response. */
923       send_crunch_response(csp, rsp);
924       csp->flags |= CSP_FLAG_CRUNCHED;
925       return TRUE;
926    }
927
928    for (c = crunchers; c->cruncher != NULL; c++)
929    {
930       /*
931        * Check the cruncher if either Privoxy is toggled
932        * on and the request isn't forced, or if the cruncher
933        * applies to forced requests as well.
934        */
935       if (((csp->flags & CSP_FLAG_TOGGLED_ON) &&
936           !(csp->flags & CSP_FLAG_FORCED)) ||
937           (c->flags & CF_IGNORE_FORCE))
938       {
939          rsp = c->cruncher(csp);
940          if (NULL != rsp)
941          {
942             /* Deliver, log and free the interception response. */
943             send_crunch_response(csp, rsp);
944             csp->flags |= CSP_FLAG_CRUNCHED;
945 #ifdef FEATURE_STATISTICS
946             if (c->flags & CF_COUNT_AS_REJECT)
947             {
948                csp->flags |= CSP_FLAG_REJECTED;
949             }
950 #endif /* def FEATURE_STATISTICS */
951
952             return TRUE;
953          }
954       }
955    }
956
957    return FALSE;
958 }
959
960
961 /*********************************************************************
962  *
963  * Function    :  build_request_line
964  *
965  * Description :  Builds the HTTP request line.
966  *
967  *                If a HTTP forwarder is used it expects the whole URL,
968  *                web servers only get the path.
969  *
970  * Parameters  :
971  *          1  :  csp = Current client state (buffers, headers, etc...)
972  *          2  :  fwd = The forwarding spec used for the request.
973  *                      Can be NULL.
974  *          3  :  request_line = The old request line which will be replaced.
975  *
976  * Returns     :  Nothing. Terminates in case of memory problems.
977  *
978  *********************************************************************/
979 static void build_request_line(struct client_state *csp, const struct forward_spec *fwd, char **request_line)
980 {
981    struct http_request *http = csp->http;
982
983    /*
984     * Downgrade http version from 1.1 to 1.0
985     * if +downgrade action applies.
986     */
987    if ((csp->action->flags & ACTION_DOWNGRADE)
988      && (!strcmpic(http->version, "HTTP/1.1")))
989    {
990       freez(http->version);
991       http->version = strdup_or_die("HTTP/1.0");
992    }
993
994    /*
995     * Rebuild the request line.
996     */
997    freez(*request_line);
998    *request_line = strdup(http->gpc);
999    string_append(request_line, " ");
1000
1001    if (fwd != NULL && fwd->forward_host && fwd->type != FORWARD_WEBSERVER)
1002    {
1003       string_append(request_line, http->url);
1004    }
1005    else
1006    {
1007       string_append(request_line, http->path);
1008    }
1009    string_append(request_line, " ");
1010    string_append(request_line, http->version);
1011
1012    if (*request_line == NULL)
1013    {
1014       log_error(LOG_LEVEL_FATAL, "Out of memory writing HTTP command");
1015    }
1016    log_error(LOG_LEVEL_HEADER, "New HTTP Request-Line: %s", *request_line);
1017 }
1018
1019
1020 /*********************************************************************
1021  *
1022  * Function    :  change_request_destination
1023  *
1024  * Description :  Parse a (rewritten) request line and regenerate
1025  *                the http request data.
1026  *
1027  * Parameters  :
1028  *          1  :  csp = Current client state (buffers, headers, etc...)
1029  *
1030  * Returns     :  Forwards the parse_http_request() return code.
1031  *                Terminates in case of memory problems.
1032  *
1033  *********************************************************************/
1034 static jb_err change_request_destination(struct client_state *csp)
1035 {
1036    struct http_request *http = csp->http;
1037    jb_err err;
1038
1039    log_error(LOG_LEVEL_REDIRECTS, "Rewrite detected: %s",
1040       csp->headers->first->str);
1041    free_http_request(http);
1042    err = parse_http_request(csp->headers->first->str, http);
1043    if (JB_ERR_OK != err)
1044    {
1045       log_error(LOG_LEVEL_ERROR, "Couldn't parse rewritten request: %s.",
1046          jb_err_to_string(err));
1047    }
1048    if (http->ssl && strcmpic(csp->http->gpc, "CONNECT"))
1049    {
1050       /*
1051        * A client header filter changed the request URL from
1052        * http:// to https:// which we currently don't support.
1053        */
1054       log_error(LOG_LEVEL_ERROR, "Changing the request destination from http "
1055          "to https behind the client's back currently isn't supported.");
1056       return JB_ERR_PARSE;
1057    }
1058
1059    return err;
1060 }
1061
1062
1063 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
1064 /*********************************************************************
1065  *
1066  * Function    :  server_response_is_complete
1067  *
1068  * Description :  Determines whether we should stop reading
1069  *                from the server socket.
1070  *
1071  * Parameters  :
1072  *          1  :  csp = Current client state (buffers, headers, etc...)
1073  *          2  :  content_length = Length of content received so far.
1074  *
1075  * Returns     :  TRUE if the response is complete,
1076  *                FALSE otherwise.
1077  *
1078  *********************************************************************/
1079 static int server_response_is_complete(struct client_state *csp,
1080    unsigned long long content_length)
1081 {
1082    int content_length_known = !!(csp->flags & CSP_FLAG_CONTENT_LENGTH_SET);
1083
1084    if (!strcmpic(csp->http->gpc, "HEAD"))
1085    {
1086       /*
1087        * "HEAD" implies no body, we are thus expecting
1088        * no content. XXX: incomplete "list" of methods?
1089        */
1090       csp->expected_content_length = 0;
1091       content_length_known = TRUE;
1092       csp->flags |= CSP_FLAG_SERVER_CONTENT_LENGTH_SET;
1093    }
1094
1095    if (csp->http->status == 204 || csp->http->status == 304)
1096    {
1097       /*
1098        * Expect no body. XXX: incomplete "list" of status codes?
1099        */
1100       csp->expected_content_length = 0;
1101       content_length_known = TRUE;
1102       csp->flags |= CSP_FLAG_SERVER_CONTENT_LENGTH_SET;
1103    }
1104
1105    return (content_length_known && ((0 == csp->expected_content_length)
1106             || (csp->expected_content_length <= content_length)));
1107 }
1108
1109
1110 #ifdef FEATURE_CONNECTION_SHARING
1111 /*********************************************************************
1112  *
1113  * Function    :  wait_for_alive_connections
1114  *
1115  * Description :  Waits for alive connections to timeout.
1116  *
1117  * Parameters  :  N/A
1118  *
1119  * Returns     :  N/A
1120  *
1121  *********************************************************************/
1122 static void wait_for_alive_connections(void)
1123 {
1124    int connections_alive = close_unusable_connections();
1125
1126    while (0 < connections_alive)
1127    {
1128       log_error(LOG_LEVEL_CONNECT,
1129          "Waiting for %d connections to timeout.",
1130          connections_alive);
1131       sleep(60);
1132       connections_alive = close_unusable_connections();
1133    }
1134
1135    log_error(LOG_LEVEL_CONNECT, "No connections to wait for left.");
1136
1137 }
1138 #endif /* def FEATURE_CONNECTION_SHARING */
1139
1140
1141 /*********************************************************************
1142  *
1143  * Function    :  save_connection_destination
1144  *
1145  * Description :  Remembers a connection for reuse later on.
1146  *
1147  * Parameters  :
1148  *          1  :  sfd  = Open socket to remember.
1149  *          2  :  http = The destination for the connection.
1150  *          3  :  fwd  = The forwarder settings used.
1151  *          4  :  server_connection  = storage.
1152  *
1153  * Returns     : void
1154  *
1155  *********************************************************************/
1156 void save_connection_destination(jb_socket sfd,
1157                                  const struct http_request *http,
1158                                  const struct forward_spec *fwd,
1159                                  struct reusable_connection *server_connection)
1160 {
1161    assert(sfd != JB_INVALID_SOCKET);
1162    assert(NULL != http->host);
1163
1164    server_connection->sfd = sfd;
1165    server_connection->host = strdup_or_die(http->host);
1166    server_connection->port = http->port;
1167
1168    assert(NULL != fwd);
1169    assert(server_connection->gateway_host == NULL);
1170    assert(server_connection->gateway_port == 0);
1171    assert(server_connection->forwarder_type == 0);
1172    assert(server_connection->forward_host == NULL);
1173    assert(server_connection->forward_port == 0);
1174
1175    server_connection->forwarder_type = fwd->type;
1176    if (NULL != fwd->gateway_host)
1177    {
1178       server_connection->gateway_host = strdup_or_die(fwd->gateway_host);
1179    }
1180    else
1181    {
1182       server_connection->gateway_host = NULL;
1183    }
1184    server_connection->gateway_port = fwd->gateway_port;
1185    if (NULL != fwd->auth_username)
1186    {
1187       server_connection->auth_username = strdup_or_die(fwd->auth_username);
1188    }
1189    else
1190    {
1191       server_connection->auth_username = NULL;
1192    }
1193    if (NULL != fwd->auth_password)
1194    {
1195       server_connection->auth_password = strdup_or_die(fwd->auth_password);
1196    }
1197    else
1198    {
1199       server_connection->auth_password = NULL;
1200    }
1201
1202    if (NULL != fwd->forward_host)
1203    {
1204       server_connection->forward_host = strdup_or_die(fwd->forward_host);
1205    }
1206    else
1207    {
1208       server_connection->forward_host = NULL;
1209    }
1210    server_connection->forward_port = fwd->forward_port;
1211 }
1212 #endif /* FEATURE_CONNECTION_KEEP_ALIVE */
1213
1214
1215 /*********************************************************************
1216  *
1217  * Function    : verify_request_length
1218  *
1219  * Description : Checks if we already got the whole client requests
1220  *               and sets CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ if
1221  *               we do.
1222  *
1223  *               Data that doesn't belong to the current request is
1224  *               either thrown away to let the client retry on a clean
1225  *               socket, or stashed to be dealt with after the current
1226  *               request is served.
1227  *
1228  * Parameters  :
1229  *          1  :  csp = Current client state (buffers, headers, etc...)
1230  *
1231  * Returns     :  void
1232  *
1233  *********************************************************************/
1234 static void verify_request_length(struct client_state *csp)
1235 {
1236    unsigned long long buffered_request_bytes =
1237       (unsigned long long)(csp->client_iob->eod - csp->client_iob->cur);
1238
1239    if ((csp->expected_client_content_length != 0)
1240       && (buffered_request_bytes != 0))
1241    {
1242       if (csp->expected_client_content_length >= buffered_request_bytes)
1243       {
1244          csp->expected_client_content_length -= buffered_request_bytes;
1245          log_error(LOG_LEVEL_CONNECT, "Reduced expected bytes to %llu "
1246             "to account for the %llu ones we already got.",
1247             csp->expected_client_content_length, buffered_request_bytes);
1248       }
1249       else
1250       {
1251          assert(csp->client_iob->eod > csp->client_iob->cur + csp->expected_client_content_length);
1252          csp->client_iob->eod = csp->client_iob->cur + csp->expected_client_content_length;
1253          log_error(LOG_LEVEL_CONNECT, "Reducing expected bytes to 0. "
1254             "Marking the server socket tainted after throwing %llu bytes away.",
1255             buffered_request_bytes - csp->expected_client_content_length);
1256          csp->expected_client_content_length = 0;
1257          csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED;
1258       }
1259
1260       if (csp->expected_client_content_length == 0)
1261       {
1262          csp->flags |= CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ;
1263       }
1264    }
1265
1266    if (!(csp->flags & CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ)
1267       && ((csp->client_iob->cur < csp->client_iob->eod)
1268          || (csp->expected_client_content_length != 0)))
1269    {
1270       if (strcmpic(csp->http->gpc, "GET")
1271          && strcmpic(csp->http->gpc, "HEAD")
1272          && strcmpic(csp->http->gpc, "TRACE")
1273          && strcmpic(csp->http->gpc, "OPTIONS")
1274          && strcmpic(csp->http->gpc, "DELETE"))
1275       {
1276          /* XXX: this is an incomplete hack */
1277          csp->flags &= ~CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ;
1278          log_error(LOG_LEVEL_CONNECT, "There better be a request body.");
1279       }
1280       else
1281       {
1282          csp->flags |= CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ;
1283
1284          if ((csp->config->feature_flags & RUNTIME_FEATURE_TOLERATE_PIPELINING) == 0)
1285          {
1286             csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED;
1287             log_error(LOG_LEVEL_CONNECT,
1288                "Possible pipeline attempt detected. The connection will not "
1289                "be kept alive and we will only serve the first request.");
1290             /* Nuke the pipelined requests from orbit, just to be sure. */
1291             clear_iob(csp->client_iob);
1292          }
1293          else
1294          {
1295             /*
1296              * Keep the pipelined data around for now, we'll deal with
1297              * it once we're done serving the current request.
1298              */
1299             csp->flags |= CSP_FLAG_PIPELINED_REQUEST_WAITING;
1300             assert(csp->client_iob->eod >= csp->client_iob->cur);
1301             log_error(LOG_LEVEL_CONNECT, "Complete client request followed by "
1302                "%d bytes of pipelined data received.",
1303                (int)(csp->client_iob->eod - csp->client_iob->cur));
1304          }
1305       }
1306    }
1307    else
1308    {
1309       csp->flags |= CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ;
1310       log_error(LOG_LEVEL_CONNECT, "Complete client request received.");
1311    }
1312 }
1313
1314
1315 /*********************************************************************
1316  *
1317  * Function    :  mark_server_socket_tainted
1318  *
1319  * Description :  Makes sure we don't reuse a server socket
1320  *                (if we didn't read everything the server sent
1321  *                us reusing the socket would lead to garbage).
1322  *
1323  * Parameters  :
1324  *          1  :  csp = Current client state (buffers, headers, etc...)
1325  *
1326  * Returns     :  void.
1327  *
1328  *********************************************************************/
1329 static void mark_server_socket_tainted(struct client_state *csp)
1330 {
1331    /*
1332     * For consistency we always mark the server socket
1333     * tainted, however, to reduce the log noise we only
1334     * emit a log message if the server socket could have
1335     * actually been reused.
1336     */
1337    if ((csp->flags & CSP_FLAG_SERVER_CONNECTION_KEEP_ALIVE)
1338       && !(csp->flags & CSP_FLAG_SERVER_SOCKET_TAINTED))
1339    {
1340       log_error(LOG_LEVEL_CONNECT,
1341          "Marking the server socket %d tainted.",
1342          csp->server_connection.sfd);
1343    }
1344    csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED;
1345 }
1346
1347 /*********************************************************************
1348  *
1349  * Function    :  get_request_line
1350  *
1351  * Description : Read the client request line.
1352  *
1353  * Parameters  :
1354  *          1  :  csp = Current client state (buffers, headers, etc...)
1355  *
1356  * Returns     :  Pointer to request line or NULL in case of errors.
1357  *
1358  *********************************************************************/
1359 static char *get_request_line(struct client_state *csp)
1360 {
1361    char buf[BUFFER_SIZE];
1362    char *request_line = NULL;
1363    int len;
1364
1365    memset(buf, 0, sizeof(buf));
1366
1367    if ((csp->flags & CSP_FLAG_PIPELINED_REQUEST_WAITING) != 0)
1368    {
1369       /*
1370        * If there are multiple pipelined requests waiting,
1371        * the flag will be set again once the next request
1372        * has been parsed.
1373        */
1374       csp->flags &= ~CSP_FLAG_PIPELINED_REQUEST_WAITING;
1375
1376       request_line = get_header(csp->client_iob);
1377       if ((NULL != request_line) && ('\0' != *request_line))
1378       {
1379          return request_line;
1380       }
1381       else
1382       {
1383          log_error(LOG_LEVEL_CONNECT, "No complete request line "
1384             "received yet. Continuing reading from %d.", csp->cfd);
1385       }
1386    }
1387
1388    do
1389    {
1390       if (
1391 #ifdef FUZZ
1392           0 == (csp->flags & CSP_FLAG_FUZZED_INPUT) &&
1393 #endif
1394           !data_is_available(csp->cfd, csp->config->socket_timeout)
1395           )
1396       {
1397          if (socket_is_still_alive(csp->cfd))
1398          {
1399             log_error(LOG_LEVEL_CONNECT,
1400                "No request line on socket %d received in time. Timeout: %d.",
1401                csp->cfd, csp->config->socket_timeout);
1402             write_socket_delayed(csp->cfd, CLIENT_CONNECTION_TIMEOUT_RESPONSE,
1403                strlen(CLIENT_CONNECTION_TIMEOUT_RESPONSE),
1404                get_write_delay(csp));
1405          }
1406          else
1407          {
1408             log_error(LOG_LEVEL_CONNECT,
1409                "The client side of the connection on socket %d got "
1410                "closed without sending a complete request line.", csp->cfd);
1411          }
1412          return NULL;
1413       }
1414
1415       len = read_socket(csp->cfd, buf, sizeof(buf) - 1);
1416
1417       if (len <= 0) return NULL;
1418
1419       /*
1420        * If there is no memory left for buffering the
1421        * request, there is nothing we can do but hang up
1422        */
1423       if (add_to_iob(csp->client_iob, csp->config->buffer_limit, buf, len))
1424       {
1425          return NULL;
1426       }
1427
1428       request_line = get_header(csp->client_iob);
1429
1430    } while ((NULL != request_line) && ('\0' == *request_line));
1431
1432    return request_line;
1433
1434 }
1435
1436 enum chunk_status
1437 {
1438    CHUNK_STATUS_MISSING_DATA,
1439    CHUNK_STATUS_BODY_COMPLETE,
1440    CHUNK_STATUS_PARSE_ERROR
1441 };
1442
1443
1444 /*********************************************************************
1445  *
1446  * Function    :  chunked_body_is_complete
1447  *
1448  * Description :  Figures out whether or not a chunked body is complete.
1449  *
1450  *                Currently it always starts at the beginning of the
1451  *                buffer which is somewhat wasteful and prevents Privoxy
1452  *                from starting to forward the correctly parsed chunks
1453  *                as soon as theoretically possible.
1454  *
1455  *                Should be modified to work with a common buffer,
1456  *                and allow the caller to skip already parsed chunks.
1457  *
1458  *                This would allow the function to be used for unbuffered
1459  *                response bodies as well.
1460  *
1461  * Parameters  :
1462  *          1  :  iob = Buffer with the body to check.
1463  *          2  :  length = Length of complete body
1464  *
1465  * Returns     :  Enum with the result of the check.
1466  *
1467  *********************************************************************/
1468 static enum chunk_status chunked_body_is_complete(struct iob *iob, size_t *length)
1469 {
1470    unsigned int chunksize;
1471    char *p = iob->cur;
1472
1473    do
1474    {
1475       /*
1476        * We need at least a single digit, followed by "\r\n",
1477        * followed by an unknown amount of data, followed by "\r\n".
1478        */
1479       if (p + 5 > iob->eod)
1480       {
1481          return CHUNK_STATUS_MISSING_DATA;
1482       }
1483       if (sscanf(p, "%x", &chunksize) != 1)
1484       {
1485          return CHUNK_STATUS_PARSE_ERROR;
1486       }
1487
1488       /*
1489        * We want at least a single digit, followed by "\r\n",
1490        * followed by the specified amount of data, followed by "\r\n".
1491        */
1492       if (p + chunksize + 5 > iob->eod)
1493       {
1494          return CHUNK_STATUS_MISSING_DATA;
1495       }
1496
1497       /* Skip chunk-size. */
1498       p = strstr(p, "\r\n");
1499       if (NULL == p)
1500       {
1501          return CHUNK_STATUS_PARSE_ERROR;
1502       }
1503       /* Move beyond the chunkdata. */
1504       p += 2 + chunksize;
1505
1506       /* There should be another "\r\n" to skip */
1507       if (memcmp(p, "\r\n", 2))
1508       {
1509          return CHUNK_STATUS_PARSE_ERROR;
1510       }
1511       p += 2;
1512    } while (chunksize > 0U);
1513
1514    *length = (size_t)(p - iob->cur);
1515    assert(*length <= (size_t)(iob->eod - iob->cur));
1516    assert(p <= iob->eod);
1517
1518    return CHUNK_STATUS_BODY_COMPLETE;
1519
1520 }
1521
1522
1523 /*********************************************************************
1524  *
1525  * Function    : receive_chunked_client_request_body
1526  *
1527  * Description : Read the chunk-encoded client request body.
1528  *               Failures are dealt with.
1529  *
1530  * Parameters  :
1531  *          1  :  csp = Current client state (buffers, headers, etc...)
1532  *
1533  * Returns     :  JB_ERR_OK or JB_ERR_PARSE
1534  *
1535  *********************************************************************/
1536 static jb_err receive_chunked_client_request_body(struct client_state *csp)
1537 {
1538    size_t body_length;
1539    enum chunk_status status;
1540
1541    while (CHUNK_STATUS_MISSING_DATA ==
1542       (status = chunked_body_is_complete(csp->client_iob, &body_length)))
1543    {
1544       char buf[BUFFER_SIZE];
1545       int len;
1546
1547       if (!data_is_available(csp->cfd, csp->config->socket_timeout))
1548       {
1549          log_error(LOG_LEVEL_ERROR,
1550             "Timeout while waiting for the client body.");
1551          break;
1552       }
1553       len = read_socket(csp->cfd, buf, sizeof(buf) - 1);
1554       if (len <= 0)
1555       {
1556          log_error(LOG_LEVEL_ERROR,
1557             "Reading the client body failed: %E");
1558          break;
1559       }
1560       if (add_to_iob(csp->client_iob, csp->config->buffer_limit, buf, len))
1561       {
1562          break;
1563       }
1564    }
1565    if (status != CHUNK_STATUS_BODY_COMPLETE)
1566    {
1567       write_socket_delayed(csp->cfd, CLIENT_BODY_PARSE_ERROR_RESPONSE,
1568          strlen(CLIENT_BODY_PARSE_ERROR_RESPONSE), get_write_delay(csp));
1569       log_error(LOG_LEVEL_CLF,
1570          "%s - - [%T] \"Failed reading chunked client body\" 400 0", csp->ip_addr_str);
1571       return JB_ERR_PARSE;
1572    }
1573    log_error(LOG_LEVEL_CONNECT,
1574       "Chunked client body completely read. Length: %lu", body_length);
1575    csp->expected_client_content_length = body_length;
1576
1577    return JB_ERR_OK;
1578
1579 }
1580
1581
1582 #ifdef FUZZ
1583 /*********************************************************************
1584  *
1585  * Function    :  fuzz_chunked_transfer_encoding
1586  *
1587  * Description :  Treat the fuzzed input as chunked transfer encoding
1588  *                to check and dechunk.
1589  *
1590  * Parameters  :
1591  *          1  :  csp      = Used to store the data.
1592  *          2  :  fuzz_input_file = File to read the input from.
1593  *
1594  * Returns     : Result of dechunking
1595  *
1596  *********************************************************************/
1597 extern int fuzz_chunked_transfer_encoding(struct client_state *csp, char *fuzz_input_file)
1598 {
1599    size_t length;
1600    size_t size = (size_t)(csp->iob->eod - csp->iob->cur);
1601    enum chunk_status status;
1602
1603    status = chunked_body_is_complete(csp->iob, &length);
1604    if (CHUNK_STATUS_BODY_COMPLETE != status)
1605    {
1606       log_error(LOG_LEVEL_INFO, "Chunked body is incomplete or invalid");
1607    }
1608
1609    return (JB_ERR_OK == remove_chunked_transfer_coding(csp->iob->cur, &size));
1610
1611 }
1612
1613
1614 /*********************************************************************
1615  *
1616  * Function    : fuzz_client_request
1617  *
1618  * Description : Try to get a client request from the fuzzed input.
1619  *
1620  * Parameters  :
1621  *          1  :  csp = Current client state (buffers, headers, etc...)
1622  *          2  :  fuzz_input_file = File to read the input from.
1623  *
1624  * Returns     :  Result of fuzzing.
1625  *
1626  *********************************************************************/
1627 extern int fuzz_client_request(struct client_state *csp, char *fuzz_input_file)
1628 {
1629    jb_err err;
1630
1631    csp->cfd = 0;
1632    csp->ip_addr_str = "fuzzer";
1633
1634    if (strcmp(fuzz_input_file, "-") != 0)
1635    {
1636       log_error(LOG_LEVEL_FATAL,
1637          "Fuzzed client requests can currently only be read from stdin (-).");
1638    }
1639    err = receive_client_request(csp);
1640    if (err != JB_ERR_OK)
1641    {
1642       return 1;
1643    }
1644    err = parse_client_request(csp);
1645    if (err != JB_ERR_OK)
1646    {
1647       return 1;
1648    }
1649
1650    return 0;
1651
1652 }
1653 #endif  /* def FUZZ */
1654
1655
1656 #ifdef FEATURE_FORCE_LOAD
1657 /*********************************************************************
1658  *
1659  * Function    :  force_required
1660  *
1661  * Description : Checks a request line to see if it contains
1662  *               the FORCE_PREFIX. If it does, it is removed
1663  *               unless enforcing requests has beend disabled.
1664  *
1665  * Parameters  :
1666  *          1  :  request_line = HTTP request line
1667  *
1668  * Returns     :  TRUE if force is required, FALSE otherwise.
1669  *
1670  *********************************************************************/
1671 static int force_required(const struct client_state *csp, char *request_line)
1672 {
1673    char *p;
1674
1675    p = strstr(request_line, "http://");
1676    if (p != NULL)
1677    {
1678       /* Skip protocol */
1679       p += strlen("http://");
1680    }
1681    else
1682    {
1683       /* Intercepted request usually don't specify the protocol. */
1684       p = request_line;
1685    }
1686
1687    /* Go to the beginning of the path */
1688    p = strstr(p, "/");
1689    if (p == NULL)
1690    {
1691       /*
1692        * If the path is missing the request line is invalid and we
1693        * are done here. The client-visible rejection happens later on.
1694        */
1695       return 0;
1696    }
1697
1698    if (0 == strncmpic(p, FORCE_PREFIX, strlen(FORCE_PREFIX) - 1))
1699    {
1700       if (!(csp->config->feature_flags & RUNTIME_FEATURE_ENFORCE_BLOCKS))
1701       {
1702          /* XXX: Should clean more carefully */
1703          strclean(request_line, FORCE_PREFIX);
1704          log_error(LOG_LEVEL_FORCE,
1705             "Enforcing request: \"%s\".", request_line);
1706
1707          return 1;
1708       }
1709       log_error(LOG_LEVEL_FORCE,
1710          "Ignored force prefix in request: \"%s\".", request_line);
1711    }
1712
1713    return 0;
1714
1715 }
1716 #endif /* def FEATURE_FORCE_LOAD */
1717
1718
1719 /*********************************************************************
1720  *
1721  * Function    :  receive_client_request
1722  *
1723  * Description : Read the client's request (more precisely the
1724  *               client headers) and answer it if necessary.
1725  *
1726  * Parameters  :
1727  *          1  :  csp = Current client state (buffers, headers, etc...)
1728  *
1729  * Returns     :  JB_ERR_OK, JB_ERR_PARSE or JB_ERR_MEMORY
1730  *
1731  *********************************************************************/
1732 static jb_err receive_client_request(struct client_state *csp)
1733 {
1734    char buf[BUFFER_SIZE];
1735    char *p;
1736    char *req = NULL;
1737    struct http_request *http;
1738    int len;
1739    jb_err err;
1740
1741    /* Temporary copy of the client's headers before they get enlisted in csp->headers */
1742    struct list header_list;
1743    struct list *headers = &header_list;
1744
1745    /* We don't care if the arriving data is a valid HTTP request or not. */
1746    csp->requests_received_total++;
1747
1748    http = csp->http;
1749
1750    memset(buf, 0, sizeof(buf));
1751
1752    req = get_request_line(csp);
1753    if (req == NULL)
1754    {
1755       mark_server_socket_tainted(csp);
1756       return JB_ERR_PARSE;
1757    }
1758    assert(*req != '\0');
1759
1760    if (client_protocol_is_unsupported(csp, req))
1761    {
1762       return JB_ERR_PARSE;
1763    }
1764
1765 #ifdef FEATURE_FORCE_LOAD
1766    if (force_required(csp, req))
1767    {
1768       csp->flags |= CSP_FLAG_FORCED;
1769    }
1770 #endif /* def FEATURE_FORCE_LOAD */
1771
1772    err = parse_http_request(req, http);
1773    freez(req);
1774    if (JB_ERR_OK != err)
1775    {
1776       write_socket_delayed(csp->cfd, CHEADER, strlen(CHEADER),
1777          get_write_delay(csp));
1778       /* XXX: Use correct size */
1779       log_error(LOG_LEVEL_CLF, "%s - - [%T] \"Invalid request\" 400 0", csp->ip_addr_str);
1780       log_error(LOG_LEVEL_ERROR,
1781          "Couldn't parse request line received from %s: %s",
1782          csp->ip_addr_str, jb_err_to_string(err));
1783
1784       free_http_request(http);
1785       return JB_ERR_PARSE;
1786    }
1787
1788    /* grab the rest of the client's headers */
1789    init_list(headers);
1790    for (;;)
1791    {
1792       p = get_header(csp->client_iob);
1793
1794       if (p == NULL)
1795       {
1796          /* There are no additional headers to read. */
1797          break;
1798       }
1799
1800       if (*p == '\0')
1801       {
1802          /*
1803           * We didn't receive a complete header
1804           * line yet, get the rest of it.
1805           */
1806          if (!data_is_available(csp->cfd, csp->config->socket_timeout))
1807          {
1808             log_error(LOG_LEVEL_ERROR,
1809                "Stopped grabbing the client headers.");
1810             destroy_list(headers);
1811             return JB_ERR_PARSE;
1812          }
1813
1814          len = read_socket(csp->cfd, buf, sizeof(buf) - 1);
1815          if (len <= 0)
1816          {
1817             log_error(LOG_LEVEL_ERROR, "read from client failed: %E");
1818             destroy_list(headers);
1819             return JB_ERR_PARSE;
1820          }
1821
1822          if (add_to_iob(csp->client_iob, csp->config->buffer_limit, buf, len))
1823          {
1824             /*
1825              * If there is no memory left for buffering the
1826              * request, there is nothing we can do but hang up
1827              */
1828             destroy_list(headers);
1829             return JB_ERR_MEMORY;
1830          }
1831       }
1832       else
1833       {
1834          if (!strncmpic(p, "Transfer-Encoding:", 18))
1835          {
1836             /*
1837              * XXX: should be called through sed()
1838              *      but currently can't.
1839              */
1840             client_transfer_encoding(csp, &p);
1841          }
1842          /*
1843           * We were able to read a complete
1844           * header and can finally enlist it.
1845           */
1846          enlist(headers, p);
1847          freez(p);
1848       }
1849    }
1850
1851    if (http->host == NULL)
1852    {
1853       /*
1854        * If we still don't know the request destination,
1855        * the request is invalid or the client uses
1856        * Privoxy without its knowledge.
1857        */
1858       if (JB_ERR_OK != get_request_destination_elsewhere(csp, headers))
1859       {
1860          /*
1861           * Our attempts to get the request destination
1862           * elsewhere failed or Privoxy is configured
1863           * to only accept proxy requests.
1864           *
1865           * An error response has already been sent
1866           * and we're done here.
1867           */
1868          return JB_ERR_PARSE;
1869       }
1870    }
1871
1872 #ifdef FEATURE_CLIENT_TAGS
1873    /* XXX: If the headers were enlisted sooner, passing csp would do. */
1874    set_client_address(csp, headers);
1875    get_tag_list_for_client(csp->client_tags, csp->client_address);
1876 #endif
1877
1878    /*
1879     * Determine the actions for this URL
1880     */
1881 #ifdef FEATURE_TOGGLE
1882    if (!(csp->flags & CSP_FLAG_TOGGLED_ON))
1883    {
1884       /* Most compatible set of actions (i.e. none) */
1885       init_current_action(csp->action);
1886    }
1887    else
1888 #endif /* ndef FEATURE_TOGGLE */
1889    {
1890       get_url_actions(csp, http);
1891    }
1892
1893    enlist(csp->headers, http->cmd);
1894
1895    /* Append the previously read headers */
1896    err = list_append_list_unique(csp->headers, headers);
1897    destroy_list(headers);
1898
1899    return err;
1900
1901 }
1902
1903
1904 /*********************************************************************
1905  *
1906  * Function    : parse_client_request
1907  *
1908  * Description : Parses the client's request and decides what to do
1909  *               with it.
1910  *
1911  *               Note that since we're not using select() we could get
1912  *               blocked here if a client connected, then didn't say
1913  *               anything!
1914  *
1915  * Parameters  :
1916  *          1  :  csp = Current client state (buffers, headers, etc...)
1917  *
1918  * Returns     :  JB_ERR_OK or JB_ERR_PARSE
1919  *
1920  *********************************************************************/
1921 static jb_err parse_client_request(struct client_state *csp)
1922 {
1923    struct http_request *http = csp->http;
1924    jb_err err;
1925
1926 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
1927    if ((csp->config->feature_flags & RUNTIME_FEATURE_CONNECTION_KEEP_ALIVE)
1928     && (!strcmpic(csp->http->version, "HTTP/1.1"))
1929     && (csp->http->ssl == 0))
1930    {
1931       /* Assume persistence until further notice */
1932       csp->flags |= CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE;
1933    }
1934 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
1935
1936    if (csp->http->ssl == 0)
1937    {
1938       /*
1939        * This whole block belongs to chat() but currently
1940        * has to be executed before sed().
1941        */
1942       if (csp->flags & CSP_FLAG_CHUNKED_CLIENT_BODY)
1943       {
1944          if (receive_chunked_client_request_body(csp) != JB_ERR_OK)
1945          {
1946             return JB_ERR_PARSE;
1947          }
1948       }
1949       else
1950       {
1951          csp->expected_client_content_length = get_expected_content_length(csp->headers);
1952       }
1953       verify_request_length(csp);
1954    }
1955 #ifndef FEATURE_HTTPS_INSPECTION
1956    else
1957    {
1958       csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED;
1959    }
1960 #endif
1961
1962    err = sed(csp, FILTER_CLIENT_HEADERS);
1963    if (JB_ERR_OK != err)
1964    {
1965       log_error(LOG_LEVEL_ERROR, "Failed to parse client request from %s.",
1966          csp->ip_addr_str);
1967       log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 400 0",
1968          csp->ip_addr_str, csp->http->cmd);
1969       write_socket_delayed(csp->cfd, CHEADER, strlen(CHEADER), get_write_delay(csp));
1970       return JB_ERR_PARSE;
1971    }
1972    csp->flags |= CSP_FLAG_CLIENT_HEADER_PARSING_DONE;
1973
1974    /* Check request line for rewrites. */
1975    if ((NULL == csp->headers->first->str)
1976       || (strcmp(http->cmd, csp->headers->first->str) &&
1977          (JB_ERR_OK != change_request_destination(csp))))
1978    {
1979       /*
1980        * A header filter broke the request line - bail out.
1981        */
1982       write_socket_delayed(csp->cfd, MESSED_UP_REQUEST_RESPONSE,
1983          strlen(MESSED_UP_REQUEST_RESPONSE), get_write_delay(csp));
1984       /* XXX: Use correct size */
1985       log_error(LOG_LEVEL_CLF,
1986          "%s - - [%T] \"Invalid request generated\" 400 0", csp->ip_addr_str);
1987       log_error(LOG_LEVEL_ERROR,
1988          "Invalid request line after applying header filters.");
1989       free_http_request(http);
1990
1991       return JB_ERR_PARSE;
1992    }
1993
1994    if (client_has_unsupported_expectations(csp))
1995    {
1996       return JB_ERR_PARSE;
1997    }
1998
1999    return JB_ERR_OK;
2000
2001 }
2002
2003
2004 /*********************************************************************
2005  *
2006  * Function    : read_http_request_body
2007  *
2008  * Description : Reads remaining request body from the client.
2009  *
2010  * Parameters  :
2011  *          1  :  csp = Current client state (buffers, headers, etc...)
2012  *
2013  * Returns     :  0 on success, anything else is an error.
2014  *
2015  *********************************************************************/
2016 static int read_http_request_body(struct client_state *csp)
2017 {
2018    size_t to_read = csp->expected_client_content_length;
2019    int len;
2020
2021    assert(to_read != 0);
2022
2023    /* check if all data has been already read */
2024    if (to_read <= (csp->client_iob->eod - csp->client_iob->cur))
2025    {
2026       return 0;
2027    }
2028
2029    for (to_read -= (size_t)(csp->client_iob->eod - csp->client_iob->cur);
2030         to_read > 0 && data_is_available(csp->cfd, csp->config->socket_timeout);
2031         to_read -= (unsigned)len)
2032    {
2033       char buf[BUFFER_SIZE];
2034       size_t max_bytes_to_read = to_read < sizeof(buf) ? to_read : sizeof(buf);
2035
2036       log_error(LOG_LEVEL_CONNECT,
2037          "Waiting for up to %d bytes of request body from the client.",
2038          max_bytes_to_read);
2039       len = read_socket(csp->cfd, buf, (int)max_bytes_to_read);
2040       if (len <= -1)
2041       {
2042          log_error(LOG_LEVEL_CONNECT, "Failed receiving request body from %s: %E", csp->ip_addr_str);
2043          return 1;
2044       }
2045       if (add_to_iob(csp->client_iob, csp->config->buffer_limit, (char *)buf, len))
2046       {
2047          return 1;
2048       }
2049       assert(to_read >= len);
2050    }
2051
2052    if (to_read != 0)
2053    {
2054       log_error(LOG_LEVEL_CONNECT, "Not enough request body has been read: expected %d more bytes",
2055          csp->expected_client_content_length);
2056       return 1;
2057    }
2058    log_error(LOG_LEVEL_CONNECT, "The last %d bytes of the request body have been read",
2059       csp->expected_client_content_length);
2060    return 0;
2061 }
2062
2063
2064 /*********************************************************************
2065  *
2066  * Function    : update_client_headers
2067  *
2068  * Description : Updates the HTTP headers from the client request.
2069  *
2070  * Parameters  :
2071  *          1  :  csp = Current client state (buffers, headers, etc...)
2072  *          2  :  new_content_length = new content length value to set
2073  *
2074  * Returns     :  0 on success, anything else is an error.
2075  *
2076  *********************************************************************/
2077 static int update_client_headers(struct client_state *csp, size_t new_content_length)
2078 {
2079    static const char content_length[] = "Content-Length:";
2080    int updated = 0;
2081    struct list_entry *p;
2082
2083 #ifndef FEATURE_HTTPS_INSPECTION
2084    for (p = csp->headers->first;
2085 #else
2086    for (p = csp->http->client_ssl ? csp->https_headers->first : csp->headers->first;
2087 #endif
2088         !updated  && (p != NULL); p = p->next)
2089    {
2090       /* Header crunch()ed in previous run? -> ignore */
2091       if (p->str == NULL)
2092       {
2093          continue;
2094       }
2095
2096       /* Does the current parser handle this header? */
2097       if (0 == strncmpic(p->str, content_length, sizeof(content_length) - 1))
2098       {
2099          updated = (JB_ERR_OK == header_adjust_content_length((char **)&(p->str), new_content_length));
2100          if (!updated)
2101          {
2102             return 1;
2103          }
2104       }
2105    }
2106
2107    return !updated;
2108 }
2109
2110
2111 /*********************************************************************
2112  *
2113  * Function    : can_filter_request_body
2114  *
2115  * Description : Checks if the current request body can be stored in
2116  *               the client_iob without hitting buffer limit.
2117  *
2118  * Parameters  :
2119  *          1  : csp = Current client state (buffers, headers, etc...)
2120  *
2121  * Returns     : TRUE if the current request size do not exceed buffer limit
2122  *               FALSE otherwise.
2123  *
2124  *********************************************************************/
2125 static int can_filter_request_body(const struct client_state *csp)
2126 {
2127    if (!can_add_to_iob(csp->client_iob, csp->config->buffer_limit,
2128                        csp->expected_client_content_length))
2129    {
2130       log_error(LOG_LEVEL_INFO,
2131          "Not filtering request body from %s: buffer limit %d will be exceeded "
2132          "(content length %d)", csp->ip_addr_str, csp->config->buffer_limit,
2133          csp->expected_client_content_length);
2134       return FALSE;
2135    }
2136    return TRUE;
2137 }
2138
2139
2140 /*********************************************************************
2141  *
2142  * Function    : send_http_request
2143  *
2144  * Description : Sends the HTTP headers from the client request
2145  *               and all the body data that has already been received.
2146  *
2147  * Parameters  :
2148  *          1  :  csp = Current client state (buffers, headers, etc...)
2149  *
2150  * Returns     :  0 on success, anything else is an error.
2151  *
2152  *********************************************************************/
2153 static int send_http_request(struct client_state *csp)
2154 {
2155    char *hdr;
2156    int write_failure;
2157    const char *to_send;
2158    size_t to_send_len;
2159    int filter_client_body = csp->expected_client_content_length != 0 &&
2160       client_body_filters_enabled(csp->action) && can_filter_request_body(csp);
2161
2162    if (filter_client_body)
2163    {
2164       if (read_http_request_body(csp))
2165       {
2166          return 1;
2167       }
2168       to_send_len = csp->expected_client_content_length;
2169       to_send = execute_client_body_filters(csp, &to_send_len);
2170       if (to_send == NULL)
2171       {
2172          /* just flush client_iob */
2173          filter_client_body = FALSE;
2174       }
2175       else if (to_send_len != csp->expected_client_content_length &&
2176          update_client_headers(csp, to_send_len))
2177       {
2178          log_error(LOG_LEVEL_HEADER, "Error updating client headers");
2179          return 1;
2180       }
2181       csp->expected_client_content_length = 0;
2182    }
2183
2184    hdr = list_to_text(csp->headers);
2185    if (hdr == NULL)
2186    {
2187       /* FIXME Should handle error properly */
2188       log_error(LOG_LEVEL_FATAL, "Out of memory parsing client header");
2189    }
2190    list_remove_all(csp->headers);
2191
2192    /*
2193     * Write the client's (modified) header to the server
2194     * (along with anything else that may be in the buffer)
2195     */
2196    write_failure = 0 != write_socket(csp->server_connection.sfd, hdr, strlen(hdr));
2197    freez(hdr);
2198
2199    if (write_failure)
2200    {
2201       log_error(LOG_LEVEL_CONNECT, "Failed sending request headers to: %s: %E",
2202          csp->http->hostport);
2203       return 1;
2204    }
2205
2206    if (filter_client_body)
2207    {
2208       write_failure = 0 != write_socket(csp->server_connection.sfd, to_send, to_send_len);
2209       freez(to_send);
2210       if (write_failure)
2211       {
2212          log_error(LOG_LEVEL_CONNECT, "Failed sending filtered request body to: %s: %E",
2213             csp->http->hostport);
2214          return 1;
2215       }
2216    }
2217
2218    if (((csp->flags & CSP_FLAG_PIPELINED_REQUEST_WAITING) == 0)
2219       && (flush_iob(csp->server_connection.sfd, csp->client_iob, 0) < 0))
2220    {
2221       log_error(LOG_LEVEL_CONNECT, "Failed sending request body to: %s: %E",
2222          csp->http->hostport);
2223       return 1;
2224    }
2225    return 0;
2226 }
2227
2228
2229 #ifdef FEATURE_HTTPS_INSPECTION
2230 /*********************************************************************
2231  *
2232  * Function    : read_https_request_body
2233  *
2234  * Description : Reads remaining request body from the client.
2235  *
2236  * Parameters  :
2237  *          1  :  csp = Current client state (buffers, headers, etc...)
2238  *
2239  * Returns     :  0 on success, anything else is an error.
2240  *
2241  *********************************************************************/
2242 static int read_https_request_body(struct client_state *csp)
2243 {
2244    size_t to_read = csp->expected_client_content_length;
2245    int len;
2246
2247    assert(to_read != 0);
2248
2249    /* check if all data has been already read */
2250    if (to_read <= (csp->client_iob->eod - csp->client_iob->cur))
2251    {
2252       return 0;
2253    }
2254
2255    for (to_read -= (size_t)(csp->client_iob->eod - csp->client_iob->cur);
2256         to_read > 0 && (is_ssl_pending(&(csp->ssl_client_attr)) ||
2257           data_is_available(csp->cfd, csp->config->socket_timeout));
2258         to_read -= (unsigned)len)
2259    {
2260       unsigned char buf[BUFFER_SIZE];
2261       size_t max_bytes_to_read = to_read < sizeof(buf) ? to_read : sizeof(buf);
2262
2263       log_error(LOG_LEVEL_CONNECT,
2264          "Waiting for up to %d bytes of request body from the client.",
2265          max_bytes_to_read);
2266       len = ssl_recv_data(&(csp->ssl_client_attr), buf,
2267          (unsigned)max_bytes_to_read);
2268       if (len <= 0)
2269       {
2270          log_error(LOG_LEVEL_CONNECT, "Failed receiving request body from %s", csp->ip_addr_str);
2271          return 1;
2272       }
2273       if (add_to_iob(csp->client_iob, csp->config->buffer_limit, (char *)buf, len))
2274       {
2275          return 1;
2276       }
2277       assert(to_read >= len);
2278    }
2279
2280    if (to_read != 0)
2281    {
2282       log_error(LOG_LEVEL_CONNECT, "Not enough request body has been read: expected %d more bytes", to_read);
2283       return 1;
2284    }
2285
2286    log_error(LOG_LEVEL_CONNECT, "The last %d bytes of the request body have been read",
2287       csp->expected_client_content_length);
2288    return 0;
2289 }
2290
2291
2292 /*********************************************************************
2293  *
2294  * Function    : receive_and_send_encrypted_post_data
2295  *
2296  * Description : Reads remaining request body from the client and sends
2297  *               it to the server.
2298  *
2299  * Parameters  :
2300  *          1  :  csp = Current client state (buffers, headers, etc...)
2301  *
2302  * Returns     :  0 on success, anything else is an error.
2303  *
2304  *********************************************************************/
2305 static int receive_and_send_encrypted_post_data(struct client_state *csp)
2306 {
2307    int content_length_known = csp->expected_client_content_length != 0;
2308
2309    while (is_ssl_pending(&(csp->ssl_client_attr))
2310       || (content_length_known && csp->expected_client_content_length != 0))
2311    {
2312       unsigned char buf[BUFFER_SIZE];
2313       int len;
2314       int max_bytes_to_read = sizeof(buf);
2315
2316       if (content_length_known && csp->expected_client_content_length < sizeof(buf))
2317       {
2318          max_bytes_to_read = (int)csp->expected_client_content_length;
2319       }
2320       log_error(LOG_LEVEL_CONNECT,
2321          "Waiting for up to %d bytes of request body from the client.",
2322          max_bytes_to_read);
2323       len = ssl_recv_data(&(csp->ssl_client_attr), buf,
2324          (unsigned)max_bytes_to_read);
2325       if (len == -1)
2326       {
2327          return 1;
2328       }
2329       if (len == 0)
2330       {
2331          /* XXX: Does this actually happen? */
2332          break;
2333       }
2334       log_error(LOG_LEVEL_CONNECT, "Forwarding %d bytes of encrypted request body",
2335          len);
2336       len = ssl_send_data(&(csp->ssl_server_attr), buf, (size_t)len);
2337       if (len == -1)
2338       {
2339          return 1;
2340       }
2341       if (csp->expected_client_content_length != 0)
2342       {
2343          if (csp->expected_client_content_length >= len)
2344          {
2345             csp->expected_client_content_length -= (unsigned)len;
2346          }
2347          if (csp->expected_client_content_length == 0)
2348          {
2349             log_error(LOG_LEVEL_CONNECT, "Forwarded the last %d bytes", len);
2350             break;
2351          }
2352       }
2353    }
2354
2355    log_error(LOG_LEVEL_CONNECT, "Done forwarding encrypted request body");
2356
2357    return 0;
2358
2359 }
2360
2361
2362 /*********************************************************************
2363  *
2364  * Function    : send_https_request
2365  *
2366  * Description : Sends the HTTP headers from the client request
2367  *               and all the body data that has already been received.
2368  *
2369  * Parameters  :
2370  *          1  :  csp = Current client state (buffers, headers, etc...)
2371  *
2372  * Returns     :  0 on success, anything else is an error.
2373  *
2374  *********************************************************************/
2375 static int send_https_request(struct client_state *csp)
2376 {
2377    char *hdr;
2378    int ret;
2379    long flushed = 0;
2380    const char *to_send;
2381    size_t to_send_len;
2382    int filter_client_body = csp->expected_client_content_length != 0 &&
2383       client_body_filters_enabled(csp->action) && can_filter_request_body(csp);
2384
2385    if (filter_client_body)
2386    {
2387       if (read_https_request_body(csp))
2388       {
2389          return 1;
2390       }
2391       to_send_len = csp->expected_client_content_length;
2392       to_send = execute_client_body_filters(csp, &to_send_len);
2393       if (to_send == NULL)
2394       {
2395          /* just flush client_iob */
2396          filter_client_body = FALSE;
2397       }
2398       else if (to_send_len != csp->expected_client_content_length &&
2399          update_client_headers(csp, to_send_len))
2400       {
2401          log_error(LOG_LEVEL_HEADER, "Error updating client headers");
2402          return 1;
2403       }
2404       csp->expected_client_content_length = 0;
2405    }
2406
2407    hdr = list_to_text(csp->https_headers);
2408    if (hdr == NULL)
2409    {
2410       /* FIXME Should handle error properly */
2411       log_error(LOG_LEVEL_FATAL, "Out of memory parsing client header");
2412    }
2413    list_remove_all(csp->https_headers);
2414
2415    /*
2416     * Write the client's (modified) header to the server
2417     * (along with anything else that may be in the buffer)
2418     */
2419    ret = ssl_send_data(&(csp->ssl_server_attr),
2420       (const unsigned char *)hdr, strlen(hdr));
2421    freez(hdr);
2422
2423    if (ret < 0)
2424    {
2425       log_error(LOG_LEVEL_CONNECT,
2426          "Failed sending encrypted request headers to: %s: %E",
2427          csp->http->hostport);
2428       mark_server_socket_tainted(csp);
2429       return 1;
2430    }
2431
2432    if (filter_client_body)
2433    {
2434       ret = ssl_send_data(&(csp->ssl_server_attr), (const unsigned char *)to_send, to_send_len);
2435       freez(to_send);
2436       if (ret < 0)
2437       {
2438          log_error(LOG_LEVEL_CONNECT, "Failed sending filtered request body to: %s",
2439             csp->http->hostport);
2440          return 1;
2441       }
2442    }
2443
2444    if (((csp->flags & CSP_FLAG_PIPELINED_REQUEST_WAITING) == 0)
2445       && ((flushed = ssl_flush_socket(&(csp->ssl_server_attr),
2446             csp->client_iob)) < 0))
2447    {
2448       log_error(LOG_LEVEL_CONNECT, "Failed sending request body to: %s: %E",
2449          csp->http->hostport);
2450       return 1;
2451    }
2452    if (flushed != 0 || csp->expected_client_content_length != 0)
2453    {
2454       if (csp->expected_client_content_length != 0)
2455       {
2456          if (csp->expected_client_content_length < flushed)
2457          {
2458             log_error(LOG_LEVEL_ERROR,
2459                "Flushed %ld bytes of request body while only expecting %llu",
2460                flushed, csp->expected_client_content_length);
2461             csp->expected_client_content_length = 0;
2462          }
2463          else
2464          {
2465             log_error(LOG_LEVEL_CONNECT,
2466                "Flushed %ld bytes of request body while expecting %llu",
2467                flushed, csp->expected_client_content_length);
2468             csp->expected_client_content_length -= (unsigned)flushed;
2469             if (receive_and_send_encrypted_post_data(csp))
2470             {
2471                return 1;
2472             }
2473          }
2474       }
2475       else
2476       {
2477          log_error(LOG_LEVEL_CONNECT,
2478             "Flushed %ld bytes of request body", flushed);
2479       }
2480    }
2481
2482    log_error(LOG_LEVEL_CONNECT, "Encrypted request sent");
2483
2484    return 0;
2485
2486 }
2487
2488
2489 /*********************************************************************
2490  *
2491  * Function    :  receive_encrypted_request
2492  *
2493  * Description :  Receives an encrypted request.
2494  *
2495  * Parameters  :
2496  *          1  :  csp = Current client state (buffers, headers, etc...)
2497  *
2498  * Returns     :  JB_ERR_OK on success,
2499  *                JB_ERR_PARSE or JB_ERR_MEMORY otherwise
2500  *
2501  *********************************************************************/
2502 static jb_err receive_encrypted_request(struct client_state *csp)
2503 {
2504    char buf[BUFFER_SIZE];
2505    int len;
2506    char *p;
2507
2508    do
2509    {
2510       log_error(LOG_LEVEL_HEADER, "Reading encrypted headers");
2511       if (!is_ssl_pending(&(csp->ssl_client_attr)) &&
2512           !data_is_available(csp->cfd, csp->config->socket_timeout))
2513       {
2514          log_error(LOG_LEVEL_CONNECT,
2515             "Socket %d timed out while waiting for client headers", csp->cfd);
2516          return JB_ERR_PARSE;
2517       }
2518       len = ssl_recv_data(&(csp->ssl_client_attr),
2519          (unsigned char *)buf, sizeof(buf));
2520       if (len == 0)
2521       {
2522          log_error(LOG_LEVEL_CONNECT,
2523             "Socket %d closed while waiting for client headers", csp->cfd);
2524          return JB_ERR_PARSE;
2525       }
2526       if (len == -1)
2527       {
2528          return JB_ERR_PARSE;
2529       }
2530       if (add_to_iob(csp->client_iob, csp->config->buffer_limit, buf, len))
2531       {
2532          return JB_ERR_MEMORY;
2533       }
2534       p = strstr(csp->client_iob->cur, "\r\n\r\n");
2535    } while (p == NULL);
2536
2537    log_error(LOG_LEVEL_HEADER, "Encrypted headers received completely");
2538
2539    return JB_ERR_OK;
2540 }
2541
2542
2543 /*********************************************************************
2544  *
2545  * Function    :  change_encrypted_request_destination
2546  *
2547  * Description :  Parse a (rewritten) request line from an encrypted
2548  *                request and regenerate the http request data.
2549  *
2550  * Parameters  :
2551  *          1  :  csp = Current client state (buffers, headers, etc...)
2552  *
2553  * Returns     :  Forwards the parse_http_request() return code.
2554  *                Terminates in case of memory problems.
2555  *
2556  *********************************************************************/
2557 static jb_err change_encrypted_request_destination(struct client_state *csp)
2558 {
2559    jb_err err;
2560    char *original_host = csp->http->host;
2561    int original_port = csp->http->port;
2562
2563    log_error(LOG_LEVEL_REDIRECTS, "Rewrite detected: %s",
2564       csp->https_headers->first->str);
2565    csp->http->host = NULL;
2566    free_http_request(csp->http);
2567    err = parse_http_request(csp->https_headers->first->str, csp->http);
2568    if (JB_ERR_OK != err)
2569    {
2570       log_error(LOG_LEVEL_ERROR, "Couldn't parse rewritten request: %s.",
2571          jb_err_to_string(err));
2572       freez(original_host);
2573       return err;
2574    }
2575
2576    if (csp->http->host == NULL)
2577    {
2578       char port_string[10];
2579       /*
2580        * The rewritten request line did not specify a host
2581        * which means we can use the original host specified
2582        * by the client.
2583        */
2584       csp->http->host = original_host;
2585       csp->http->port = original_port;
2586       log_error(LOG_LEVEL_REDIRECTS, "Keeping the original host: %s",
2587          csp->http->host);
2588       /*
2589        * If the rewritten request line didn't contain a host
2590        * it also didn't contain a port so we can reuse the host
2591        * port.
2592        */
2593       freez(csp->http->hostport);
2594       csp->http->hostport = strdup_or_die(csp->http->host);
2595       snprintf(port_string, sizeof(port_string), ":%d", original_port);
2596       err = string_append(&csp->http->hostport, port_string);
2597       if (err != JB_ERR_OK)
2598       {
2599          log_error(LOG_LEVEL_ERROR, "Failed to rebuild hostport: %s.",
2600             jb_err_to_string(err));
2601          return err;
2602       }
2603
2604       /*
2605        * While the request line didn't mention it,
2606        * we're https-inspecting and want to speak TLS
2607        * with the server.
2608        */
2609       csp->http->server_ssl = 1;
2610       csp->http->ssl = 1;
2611    }
2612    else
2613    {
2614       /* The rewrite filter added a host so we can ditch the original */
2615       freez(original_host);
2616       csp->http->server_ssl = csp->http->ssl;
2617    }
2618
2619    csp->http->client_ssl = 1;
2620
2621    freez(csp->https_headers->first->str);
2622    build_request_line(csp, NULL, &csp->https_headers->first->str);
2623
2624    if (!server_use_ssl(csp))
2625    {
2626       log_error(LOG_LEVEL_REDIRECTS,
2627          "Rewritten request line results in downgrade to http");
2628       /*
2629        * Replace the unencryptd headers received with the
2630        * CONNECT request with the ones we received securely.
2631        */
2632       destroy_list(csp->headers);
2633       csp->headers->first = csp->https_headers->first;
2634       csp->headers->last  = csp->https_headers->last;
2635       csp->https_headers->first = NULL;
2636       csp->https_headers->last = NULL;
2637    }
2638
2639    return JB_ERR_OK;
2640
2641 }
2642
2643
2644 /*********************************************************************
2645  *
2646  * Function    :  process_encrypted_request
2647  *
2648  * Description :  Receives and parses an encrypted request.
2649  *
2650  * Parameters  :
2651  *          1  :  csp = Current client state (buffers, headers, etc...)
2652  *
2653  * Returns     :  JB_ERR_OK on success,
2654  *                JB_ERR_PARSE or JB_ERR_MEMORY otherwise
2655  *
2656  *********************************************************************/
2657 static jb_err process_encrypted_request(struct client_state *csp)
2658 {
2659    char *p;
2660    char *request_line;
2661    jb_err err;
2662    /* Temporary copy of the client's headers before they get enlisted in csp->https_headers */
2663    struct list header_list;
2664    struct list *headers = &header_list;
2665
2666    assert(csp->ssl_with_client_is_opened);
2667
2668 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
2669    if (csp->config->feature_flags & RUNTIME_FEATURE_CONNECTION_KEEP_ALIVE)
2670    {
2671       csp->flags |= CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE;
2672    }
2673 #endif
2674    err = receive_encrypted_request(csp);
2675    if (err != JB_ERR_OK)
2676    {
2677       if (csp->client_iob->cur == NULL ||
2678           csp->client_iob->cur == csp->client_iob->eod)
2679       {
2680          /*
2681           * We did not receive any data, most likely because the
2682           * client is done. Don't log this as a parse failure.
2683           */
2684          return JB_ERR_PARSE;
2685       }
2686       /* XXX: Also used for JB_ERR_MEMORY */
2687       log_error(LOG_LEVEL_ERROR, "Failed to receive encrypted request: %s",
2688          jb_err_to_string(err));
2689       ssl_send_data_delayed(&(csp->ssl_client_attr),
2690          (const unsigned char *)CHEADER, strlen(CHEADER), get_write_delay(csp));
2691       return err;
2692    }
2693
2694    /* We don't need get_request_line() because the whole HTTP head is buffered. */
2695    request_line = get_header(csp->client_iob);
2696    if (request_line == NULL)
2697    {
2698       log_error(LOG_LEVEL_ERROR, "Failed to get the encrypted request line");
2699       ssl_send_data_delayed(&(csp->ssl_client_attr),
2700          (const unsigned char *)CHEADER, strlen(CHEADER), get_write_delay(csp));
2701       return JB_ERR_PARSE;
2702    }
2703    assert(*request_line != '\0');
2704
2705    if (client_protocol_is_unsupported(csp, request_line))
2706    {
2707       /*
2708        * If the protocol is unsupported we're done here.
2709        * client_protocol_is_unsupported() took care of sending
2710        * the error response and logging the error message.
2711        */
2712       return JB_ERR_PARSE;
2713    }
2714
2715 #ifdef FEATURE_FORCE_LOAD
2716    if (force_required(csp, request_line))
2717    {
2718       csp->flags |= CSP_FLAG_FORCED;
2719    }
2720 #endif /* def FEATURE_FORCE_LOAD */
2721
2722    free_http_request(csp->http);
2723
2724    err = parse_http_request(request_line, csp->http);
2725    /* XXX: Restore ssl setting. This is ugly */
2726    csp->http->client_ssl = 1;
2727    csp->http->server_ssl = 1;
2728
2729    freez(request_line);
2730    if (JB_ERR_OK != err)
2731    {
2732       ssl_send_data_delayed(&(csp->ssl_client_attr),
2733          (const unsigned char *)CHEADER, strlen(CHEADER), get_write_delay(csp));
2734       /* XXX: Use correct size */
2735       log_error(LOG_LEVEL_CLF, "%s - - [%T] \"Invalid request\" 400 0", csp->ip_addr_str);
2736       log_error(LOG_LEVEL_ERROR,
2737          "Couldn't parse request line received from %s: %s",
2738          csp->ip_addr_str, jb_err_to_string(err));
2739
2740       free_http_request(csp->http);
2741       return JB_ERR_PARSE;
2742    }
2743
2744    /* Parse the rest of the client's headers. */
2745    init_list(headers);
2746    for (;;)
2747    {
2748       p = get_header(csp->client_iob);
2749
2750       if (p == NULL)
2751       {
2752          /* There are no additional headers to read. */
2753          break;
2754       }
2755       enlist(headers, p);
2756       freez(p);
2757    }
2758
2759    if (JB_ERR_OK != get_destination_from_https_headers(headers, csp->http))
2760    {
2761       /*
2762        * Our attempts to get the request destination
2763        * elsewhere failed.
2764        */
2765       log_error(LOG_LEVEL_ERROR,
2766          "Failed to get the encrypted request destination");
2767       ssl_send_data_delayed(&(csp->ssl_client_attr),
2768          (const unsigned char *)CHEADER, strlen(CHEADER), get_write_delay(csp));
2769       return JB_ERR_PARSE;
2770    }
2771
2772    /* Split the domain we just got for pattern matching */
2773    init_domain_components(csp->http);
2774
2775 #ifdef FEATURE_CLIENT_TAGS
2776    /* XXX: If the headers were enlisted sooner, passing csp would do. */
2777    if (csp->client_address == NULL)
2778    {
2779       set_client_address(csp, headers);
2780       get_tag_list_for_client(csp->client_tags, csp->client_address);
2781    }
2782 #endif
2783
2784 #ifdef FEATURE_TOGGLE
2785    if ((csp->flags & CSP_FLAG_TOGGLED_ON) != 0)
2786 #endif
2787    {
2788       /*
2789        * Determine the actions for this request after
2790        * clearing the ones from the previous one.
2791        */
2792       free_current_action(csp->action);
2793       get_url_actions(csp, csp->http);
2794    }
2795
2796    enlist(csp->https_headers, csp->http->cmd);
2797
2798    /* Append the previously read headers */
2799    err = list_append_list_unique(csp->https_headers, headers);
2800    destroy_list(headers);
2801    if (JB_ERR_OK != err)
2802    {
2803       /* XXX: Send error message */
2804       return err;
2805    }
2806
2807    /* XXX: Work around crash */
2808    csp->error_message = NULL;
2809
2810    /* XXX: Why do this here? */
2811    csp->http->ssl = 1;
2812
2813    err = sed_https(csp);
2814    if (JB_ERR_OK != err)
2815    {
2816       ssl_send_data_delayed(&(csp->ssl_client_attr),
2817          (const unsigned char *)CHEADER, strlen(CHEADER), get_write_delay(csp));
2818       log_error(LOG_LEVEL_ERROR, "Failed to parse client request from %s.",
2819          csp->ip_addr_str);
2820       log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 400 0",
2821          csp->ip_addr_str, csp->http->cmd);
2822       return JB_ERR_PARSE;
2823    }
2824
2825    if ((NULL == csp->https_headers->first->str)
2826       || (strcmp(csp->http->cmd, csp->https_headers->first->str) &&
2827          (JB_ERR_OK != change_encrypted_request_destination(csp))))
2828    {
2829       ssl_send_data_delayed(&(csp->ssl_client_attr),
2830          (const unsigned char *)MESSED_UP_REQUEST_RESPONSE,
2831          strlen(MESSED_UP_REQUEST_RESPONSE), get_write_delay(csp));
2832       log_error(LOG_LEVEL_ERROR,
2833          "Invalid request line after applying header filters.");
2834       /* XXX: Use correct size */
2835       log_error(LOG_LEVEL_CLF,
2836          "%s - - [%T] \"Invalid request generated\" 400 0", csp->ip_addr_str);
2837
2838       return JB_ERR_PARSE;
2839    }
2840
2841    log_error(LOG_LEVEL_HEADER, "Encrypted request processed");
2842    log_applied_actions(csp->action);
2843    log_error(LOG_LEVEL_REQUEST, "https://%s%s", csp->http->hostport,
2844       csp->http->path);
2845
2846    return err;
2847
2848 }
2849
2850 /*********************************************************************
2851  *
2852  * Function    :  cgi_page_requested
2853  *
2854  * Description :  Checks if a request is for an internal CGI page.
2855  *
2856  * Parameters  :
2857  *          1  :  host = The host requested by the client.
2858  *
2859  * Returns     :  1 if a CGI page has been requested, 0 otherwise
2860  *
2861  *********************************************************************/
2862 static int cgi_page_requested(const char *host)
2863 {
2864    if ((0 == strcmpic(host, CGI_SITE_1_HOST))
2865     || (0 == strcmpic(host, CGI_SITE_1_HOST "."))
2866     || (0 == strcmpic(host, CGI_SITE_2_HOST))
2867     || (0 == strcmpic(host, CGI_SITE_2_HOST ".")))
2868    {
2869       return 1;
2870    }
2871
2872    return 0;
2873
2874 }
2875
2876
2877 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
2878 /*********************************************************************
2879  *
2880  * Function    :  continue_https_chat
2881  *
2882  * Description :  Behaves similar to chat() but only deals with
2883  *                https-inspected requests that arrive on an already
2884  *                established connection. The first request is always
2885  *                served by chat() which is a lot more complex as it
2886  *                has to deal with forwarding settings and connection
2887  *                failures etc.
2888  *
2889  *                If a connection to the server has already been
2890  *                opened it is reused unless the request is blocked
2891  *                or the forwarder changed.
2892  *
2893  *                If a connection to the server has not yet been
2894  *                opened (because the previous request was crunched),
2895  *                or the forwarder changed, the connection is dropped
2896  *                so that the client retries on a fresh one.
2897  *
2898  * Parameters  :
2899  *          1  :  csp = Current client state (buffers, headers, etc...)
2900  *
2901  * Returns     :  Nothing.
2902  *
2903  *********************************************************************/
2904 static void continue_https_chat(struct client_state *csp)
2905 {
2906    const struct forward_spec *fwd;
2907
2908    if (JB_ERR_OK != process_encrypted_request(csp))
2909    {
2910       csp->flags &= ~CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE;
2911       return;
2912    }
2913
2914    csp->requests_received_total++;
2915
2916    /*
2917     * We have an encrypted request. Check if one of the crunchers wants it.
2918     */
2919    if (crunch_response_triggered(csp, crunchers_all))
2920    {
2921       /*
2922        * Yes. The client got the crunch response and we're done here.
2923        */
2924       return;
2925    }
2926    if (csp->ssl_with_server_is_opened == 0)
2927    {
2928       log_error(LOG_LEVEL_CONNECT,
2929          "Dropping the client connection on socket %d. "
2930          "The server connection has not been established yet.",
2931          csp->cfd);
2932       csp->flags &= ~CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE;
2933       return;
2934    }
2935    assert(csp->server_connection.sfd != JB_INVALID_SOCKET);
2936
2937    fwd = forward_url(csp, csp->http);
2938    if (!connection_destination_matches(&csp->server_connection, csp->http, fwd))
2939    {
2940       log_error(LOG_LEVEL_CONNECT,
2941          "Dropping the client connection on socket %d with "
2942          "server socket %d connected to %s. The forwarder has changed.",
2943          csp->cfd, csp->server_connection.sfd, csp->server_connection.host);
2944       csp->flags &= ~CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE;
2945       return;
2946    }
2947
2948    log_error(LOG_LEVEL_CONNECT,
2949       "Reusing server socket %d connected to %s. Requests already sent: %u.",
2950       csp->server_connection.sfd, csp->server_connection.host,
2951       csp->server_connection.requests_sent_total);
2952
2953    if (send_https_request(csp))
2954    {
2955       /*
2956        * Most likely the server connection timed out. We can't easily
2957        * create a new one so simply drop the client connection without a
2958        * error response to let the client retry.
2959        */
2960       log_error(LOG_LEVEL_CONNECT,
2961          "Dropping client connection on socket %d. "
2962          "Forwarding the encrypted client request failed.",
2963          csp->cfd);
2964       return;
2965    }
2966    csp->server_connection.requests_sent_total++;
2967    handle_established_connection(csp);
2968    freez(csp->receive_buffer);
2969 }
2970 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
2971 #endif
2972
2973
2974 /*********************************************************************
2975  *
2976  * Function    :  handle_established_connection
2977  *
2978  * Description :  Shuffle data between client and server once the
2979  *                connection has been established.
2980  *
2981  * Parameters  :
2982  *          1  :  csp = Current client state (buffers, headers, etc...)
2983  *
2984  * Returns     :  Nothing.
2985  *
2986  *********************************************************************/
2987 static void handle_established_connection(struct client_state *csp)
2988 {
2989    char *hdr;
2990    char *p;
2991    int n;
2992 #ifdef HAVE_POLL
2993    struct pollfd poll_fds[2];
2994 #else
2995    fd_set rfds;
2996    jb_socket maxfd;
2997    struct timeval timeout;
2998 #endif
2999    int server_body;
3000    int ms_iis5_hack = 0;
3001    unsigned long long byte_count = 0;
3002    struct http_request *http;
3003    long len = 0; /* for buffer sizes (and negative error codes) */
3004    int buffer_and_filter_content = 0;
3005    unsigned int write_delay;
3006 #ifdef FEATURE_HTTPS_INSPECTION
3007    int ret = 0;
3008    int use_ssl_tunnel = 0;
3009    csp->dont_verify_certificate = 0;
3010
3011    if (csp->http->ssl && !(csp->action->flags & ACTION_HTTPS_INSPECTION))
3012    {
3013       /* Pass encrypted content without filtering. */
3014       use_ssl_tunnel = 1;
3015    }
3016 #endif
3017
3018    /* Skeleton for HTTP response, if we should intercept the request */
3019    struct http_response *rsp;
3020 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
3021    int watch_client_socket;
3022 #endif
3023
3024    csp->receive_buffer_size = csp->config->receive_buffer_size;
3025    csp->receive_buffer = zalloc(csp->receive_buffer_size + 1);
3026    if (csp->receive_buffer == NULL)
3027    {
3028       log_error(LOG_LEVEL_ERROR,
3029          "Out of memory. Failed to allocate the receive buffer.");
3030       rsp = cgi_error_memory();
3031       send_crunch_response(csp, rsp);
3032       return;
3033    }
3034
3035    http = csp->http;
3036
3037 #ifndef HAVE_POLL
3038    maxfd = (csp->cfd > csp->server_connection.sfd) ?
3039       csp->cfd : csp->server_connection.sfd;
3040 #endif
3041
3042    /* pass data between the client and server
3043     * until one or the other shuts down the connection.
3044     */
3045
3046    server_body = 0;
3047
3048 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
3049    watch_client_socket = 0 == (csp->flags & CSP_FLAG_PIPELINED_REQUEST_WAITING);
3050 #endif
3051    write_delay = get_write_delay(csp);
3052
3053    for (;;)
3054    {
3055 #ifndef HAVE_POLL
3056       FD_ZERO(&rfds);
3057 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
3058       if (!watch_client_socket)
3059       {
3060          maxfd = csp->server_connection.sfd;
3061       }
3062       else
3063 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
3064       {
3065          FD_SET(csp->cfd, &rfds);
3066       }
3067
3068       FD_SET(csp->server_connection.sfd, &rfds);
3069 #endif /* ndef HAVE_POLL */
3070
3071 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
3072       if ((csp->flags & CSP_FLAG_CHUNKED)
3073          && !(csp->flags & CSP_FLAG_CONTENT_LENGTH_SET)
3074          && ((csp->iob->eod - csp->iob->cur) >= 5)
3075          && !memcmp(csp->iob->eod-5, "0\r\n\r\n", 5))
3076       {
3077          /*
3078           * XXX: This check should be obsolete now,
3079           *      but let's wait a while to be sure.
3080           */
3081          log_error(LOG_LEVEL_CONNECT,
3082             "Looks like we got the last chunk together with "
3083             "the server headers but didn't detect it earlier. "
3084             "We better stop reading.");
3085          byte_count = (unsigned long long)(csp->iob->eod - csp->iob->cur);
3086          csp->expected_content_length = byte_count;
3087          csp->flags |= CSP_FLAG_CONTENT_LENGTH_SET;
3088       }
3089       if (server_body && server_response_is_complete(csp, byte_count))
3090       {
3091          if (csp->expected_content_length == byte_count)
3092          {
3093             log_error(LOG_LEVEL_CONNECT,
3094                "Done reading from server. Content length: %llu as expected. "
3095                "Bytes most recently read: %ld.",
3096                byte_count, len);
3097          }
3098          else
3099          {
3100             log_error(LOG_LEVEL_CONNECT,
3101                "Done reading from server. Expected content length: %llu. "
3102                "Actual content length: %llu. Bytes most recently read: %ld.",
3103                csp->expected_content_length, byte_count, len);
3104          }
3105          len = 0;
3106          /*
3107           * XXX: Should not jump around, handle_established_connection()
3108           * is complicated enough already.
3109           */
3110          goto reading_done;
3111       }
3112 #endif  /* FEATURE_CONNECTION_KEEP_ALIVE */
3113
3114 #ifdef HAVE_POLL
3115       poll_fds[0].fd = csp->cfd;
3116 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
3117       if (!watch_client_socket)
3118       {
3119          /*
3120           * Ignore incoming data, but still watch out
3121           * for disconnects etc. These flags are always
3122           * implied anyway but explicitly setting them
3123           * doesn't hurt.
3124           */
3125          poll_fds[0].events = POLLERR|POLLHUP;
3126       }
3127       else
3128 #endif
3129       {
3130          poll_fds[0].events = POLLIN;
3131       }
3132       poll_fds[1].fd = csp->server_connection.sfd;
3133       poll_fds[1].events = POLLIN;
3134       n = poll(poll_fds, 2, csp->config->socket_timeout * 1000);
3135 #else
3136       timeout.tv_sec = csp->config->socket_timeout;
3137       timeout.tv_usec = 0;
3138       n = select((int)maxfd + 1, &rfds, NULL, NULL, &timeout);
3139 #endif /* def HAVE_POLL */
3140
3141       /*server or client not responding in timeout */
3142       if (n == 0)
3143       {
3144          log_error(LOG_LEVEL_CONNECT, "Socket timeout %d reached: %s",
3145             csp->config->socket_timeout, http->url);
3146          if ((byte_count == 0) && (http->ssl == 0))
3147          {
3148             send_crunch_response(csp, error_response(csp, "connection-timeout"));
3149          }
3150          mark_server_socket_tainted(csp);
3151 #ifdef FEATURE_HTTPS_INSPECTION
3152          close_client_and_server_ssl_connections(csp);
3153 #endif
3154          return;
3155       }
3156       else if (n < 0)
3157       {
3158 #ifdef HAVE_POLL
3159          log_error(LOG_LEVEL_ERROR, "poll() failed!: %E");
3160 #else
3161          log_error(LOG_LEVEL_ERROR, "select() failed!: %E");
3162 #endif
3163          mark_server_socket_tainted(csp);
3164 #ifdef FEATURE_HTTPS_INSPECTION
3165          close_client_and_server_ssl_connections(csp);
3166 #endif
3167          return;
3168       }
3169
3170       /*
3171        * This is the body of the browser's request,
3172        * just read and write it.
3173        *
3174        * Receives data from browser and sends it to server
3175        *
3176        * XXX: Make sure the client doesn't use pipelining
3177        * behind Privoxy's back.
3178        */
3179 #ifdef HAVE_POLL
3180       if ((poll_fds[0].revents & (POLLERR|POLLHUP|POLLNVAL)) != 0)
3181       {
3182          log_error(LOG_LEVEL_CONNECT,
3183             "The client socket %d has become unusable while "
3184             "the server socket %d is still open.",
3185             csp->cfd, csp->server_connection.sfd);
3186          mark_server_socket_tainted(csp);
3187          break;
3188       }
3189
3190       if (poll_fds[0].revents != 0)
3191 #else
3192       if (FD_ISSET(csp->cfd, &rfds))
3193 #endif /* def HAVE_POLL*/
3194       {
3195          int max_bytes_to_read = (int)csp->receive_buffer_size;
3196
3197 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
3198          if ((csp->flags & CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ))
3199          {
3200             if (data_is_available(csp->cfd, 0))
3201             {
3202                /*
3203                 * If the next request is already waiting, we have
3204                 * to stop select()ing the client socket. Otherwise
3205                 * we would always return right away and get nothing
3206                 * else done.
3207                 */
3208                watch_client_socket = 0;
3209                log_error(LOG_LEVEL_CONNECT,
3210                   "Stop watching client socket %d. "
3211                   "There's already another request waiting.",
3212                   csp->cfd);
3213                continue;
3214             }
3215             /*
3216              * If the client socket is set, but there's no data
3217              * available on the socket, the client went fishing
3218              * and continuing talking to the server makes no sense.
3219              */
3220             log_error(LOG_LEVEL_CONNECT,
3221                "The client closed socket %d while "
3222                "the server socket %d is still open.",
3223                csp->cfd, csp->server_connection.sfd);
3224             mark_server_socket_tainted(csp);
3225             break;
3226          }
3227          if (csp->expected_client_content_length != 0)
3228          {
3229             if (csp->expected_client_content_length < csp->receive_buffer_size)
3230             {
3231                max_bytes_to_read = (int)csp->expected_client_content_length;
3232             }
3233             log_error(LOG_LEVEL_CONNECT,
3234                "Waiting for up to %d bytes from the client.",
3235                max_bytes_to_read);
3236          }
3237          assert(max_bytes_to_read <= csp->receive_buffer_size);
3238 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
3239
3240 #ifdef FEATURE_HTTPS_INSPECTION
3241          if (client_use_ssl(csp))
3242          {
3243             if (csp->http->status == 101)
3244             {
3245                len = ssl_recv_data(&(csp->ssl_client_attr),
3246                   (unsigned char *)csp->receive_buffer,
3247                   (size_t)max_bytes_to_read);
3248                if (len == -1)
3249                {
3250                   log_error(LOG_LEVEL_ERROR, "Failed to receive data "
3251                      "on client socket %d for an upgraded connection",
3252                      csp->cfd);
3253                   break;
3254                }
3255                if (len == 0)
3256                {
3257                   log_error(LOG_LEVEL_CONNECT, "Done receiving data "
3258                      "on client socket %d for an upgraded connection",
3259                      csp->cfd);
3260                   break;
3261                }
3262                byte_count += (unsigned long long)len;
3263                len = ssl_send_data(&(csp->ssl_server_attr),
3264                   (unsigned char *)csp->receive_buffer, (size_t)len);
3265                if (len == -1)
3266                {
3267                   log_error(LOG_LEVEL_ERROR, "Failed to send data "
3268                      "on server socket %d for an upgraded connection",
3269                      csp->server_connection.sfd);
3270                   break;
3271                }
3272                continue;
3273             }
3274             log_error(LOG_LEVEL_CONNECT, "Breaking with TLS/SSL.");
3275             break;
3276          }
3277          else
3278 #endif /* def FEATURE_HTTPS_INSPECTION */
3279          {
3280             len = read_socket(csp->cfd, csp->receive_buffer, max_bytes_to_read);
3281
3282             if (len <= 0)
3283             {
3284                /* XXX: not sure if this is necessary. */
3285                mark_server_socket_tainted(csp);
3286                break; /* "game over, man" */
3287             }
3288
3289 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
3290             if (csp->expected_client_content_length != 0)
3291             {
3292                assert(len <= max_bytes_to_read);
3293                csp->expected_client_content_length -= (unsigned)len;
3294                log_error(LOG_LEVEL_CONNECT,
3295                   "Expected client content length set to %llu "
3296                   "after reading %ld bytes.",
3297                   csp->expected_client_content_length, len);
3298                if (csp->expected_client_content_length == 0)
3299                {
3300                   log_error(LOG_LEVEL_CONNECT,
3301                      "Done reading from the client.");
3302                   csp->flags |= CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ;
3303                }
3304             }
3305 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
3306
3307             if (write_socket(csp->server_connection.sfd, csp->receive_buffer, (size_t)len))
3308             {
3309                log_error(LOG_LEVEL_ERROR, "write to: %s failed: %E", http->host);
3310                mark_server_socket_tainted(csp);
3311                return;
3312             }
3313          }
3314          continue;
3315       }
3316
3317       /*
3318        * The server wants to talk. It could be the header or the body.
3319        * If `hdr' is null, then it's the header otherwise it's the body.
3320        * FIXME: Does `hdr' really mean `host'? No.
3321        */
3322 #ifdef HAVE_POLL
3323       if (poll_fds[1].revents != 0)
3324 #else
3325       if (FD_ISSET(csp->server_connection.sfd, &rfds))
3326 #endif /* HAVE_POLL */
3327       {
3328 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
3329          /*
3330           * If we are buffering content, we don't want to eat up to
3331           * buffer-limit bytes if the client no longer cares about them.
3332           * If we aren't buffering, however, a dead client socket will be
3333           * noticed pretty much right away anyway, so we can reduce the
3334           * overhead by skipping the check.
3335           */
3336          if (buffer_and_filter_content && !socket_is_still_alive(csp->cfd))
3337          {
3338 #ifdef _WIN32
3339             log_error(LOG_LEVEL_CONNECT,
3340                "The server still wants to talk, but the client may already have hung up on us.");
3341 #else
3342             log_error(LOG_LEVEL_CONNECT,
3343                "The server still wants to talk, but the client hung up on us.");
3344             mark_server_socket_tainted(csp);
3345 #ifdef FEATURE_HTTPS_INSPECTION
3346             close_client_and_server_ssl_connections(csp);
3347 #endif
3348             return;
3349 #endif /* def _WIN32 */
3350          }
3351 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
3352
3353 #ifdef FEATURE_HTTPS_INSPECTION
3354          /*
3355           * Reading data from standard or secured connection (HTTP/HTTPS)
3356           */
3357          if (server_use_ssl(csp))
3358          {
3359             len = ssl_recv_data(&(csp->ssl_server_attr),
3360                (unsigned char *)csp->receive_buffer, csp->receive_buffer_size);
3361          }
3362          else
3363 #endif
3364          {
3365             len = read_socket(csp->server_connection.sfd, csp->receive_buffer,
3366                (int)csp->receive_buffer_size);
3367          }
3368
3369          if (len < 0)
3370          {
3371             log_error(LOG_LEVEL_ERROR, "read from: %s failed: %E", http->host);
3372
3373             if ((http->ssl && (csp->fwd == NULL))
3374 #ifdef FEATURE_HTTPS_INSPECTION
3375                && use_ssl_tunnel
3376 #endif
3377                 )
3378             {
3379                /*
3380                 * Just hang up. We already confirmed the client's CONNECT
3381                 * request with status code 200 and unencrypted content is
3382                 * no longer welcome.
3383                 */
3384                log_error(LOG_LEVEL_ERROR,
3385                   "CONNECT already confirmed. Unable to tell the client about the problem.");
3386                return;
3387             }
3388             else if (byte_count)
3389             {
3390                /*
3391                 * Just hang up. We already transmitted the original headers
3392                 * and parts of the original content and therefore missed the
3393                 * chance to send an error message (without risking data corruption).
3394                 *
3395                 * XXX: we could retry with a fancy range request here.
3396                 */
3397                log_error(LOG_LEVEL_ERROR, "Already forwarded the original headers. "
3398                   "Unable to tell the client about the problem.");
3399                mark_server_socket_tainted(csp);
3400 #ifdef FEATURE_HTTPS_INSPECTION
3401                close_client_and_server_ssl_connections(csp);
3402 #endif
3403                return;
3404             }
3405             /*
3406              * XXX: Consider handling the cases above the same.
3407              */
3408             mark_server_socket_tainted(csp);
3409             len = 0;
3410          }
3411
3412 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
3413          if (csp->flags & CSP_FLAG_CHUNKED)
3414          {
3415             if ((len >= 5) && !memcmp(csp->receive_buffer+len-5, "0\r\n\r\n", 5))
3416             {
3417                /* XXX: this is a temporary hack */
3418                log_error(LOG_LEVEL_CONNECT,
3419                   "Looks like we reached the end of the last chunk. "
3420                   "We better stop reading.");
3421                csp->expected_content_length = byte_count + (unsigned long long)len;
3422                csp->flags |= CSP_FLAG_CONTENT_LENGTH_SET;
3423             }
3424          }
3425          reading_done:
3426 #endif  /* FEATURE_CONNECTION_KEEP_ALIVE */
3427
3428          /*
3429           * This is guaranteed by allocating with zalloc_or_die()
3430           * and never (intentionally) writing to the last byte.
3431           *
3432           * csp->receive_buffer_size is the size of the part of the
3433           * buffer we intentionally write to, but we actually
3434           * allocated csp->receive_buffer_size+1 bytes so the assertion
3435           * stays within the allocated range.
3436           */
3437          assert(csp->receive_buffer[csp->receive_buffer_size] == '\0');
3438
3439          /*
3440           * Add a trailing zero to let be able to use string operations.
3441           * XXX: do we still need this with filter_popups gone?
3442           */
3443          assert(len <= csp->receive_buffer_size);
3444          csp->receive_buffer[len] = '\0';
3445
3446          /*
3447           * Normally, this would indicate that we've read
3448           * as much as the server has sent us and we can
3449           * close the client connection.  However, Microsoft
3450           * in its wisdom has released IIS/5 with a bug that
3451           * prevents it from sending the trailing \r\n in
3452           * a 302 redirect header (and possibly other headers).
3453           * To work around this if we've haven't parsed
3454           * a full header we'll append a trailing \r\n
3455           * and see if this now generates a valid one.
3456           *
3457           * This hack shouldn't have any impacts.  If we've
3458           * already transmitted the header or if this is a
3459           * SSL connection, then we won't bother with this
3460           * hack.  So we only work on partially received
3461           * headers.  If we append a \r\n and this still
3462           * doesn't generate a valid header, then we won't
3463           * transmit anything to the client.
3464           */
3465          if (len == 0)
3466          {
3467
3468             if (server_body || (http->ssl
3469 #ifdef FEATURE_HTTPS_INSPECTION
3470                   && use_ssl_tunnel
3471 #endif
3472                ))
3473             {
3474                /*
3475                 * If we have been buffering up the document,
3476                 * now is the time to apply content modification
3477                 * and send the result to the client.
3478                 */
3479                if (buffer_and_filter_content)
3480                {
3481                   p = execute_content_filters(csp);
3482                   /*
3483                    * If content filtering fails, use the original
3484                    * buffer and length.
3485                    * (see p != NULL ? p : csp->iob->cur below)
3486                    */
3487                   if (NULL == p)
3488                   {
3489                      csp->content_length = (size_t)(csp->iob->eod - csp->iob->cur);
3490                   }
3491 #ifdef FEATURE_COMPRESSION
3492                   else if ((csp->flags & CSP_FLAG_CLIENT_SUPPORTS_DEFLATE)
3493                      && (csp->content_length > LOWER_LENGTH_LIMIT_FOR_COMPRESSION))
3494                   {
3495                      char *compressed_content = compress_buffer(p,
3496                         (size_t *)&csp->content_length, csp->config->compression_level);
3497                      if (compressed_content != NULL)
3498                      {
3499                         freez(p);
3500                         p = compressed_content;
3501                         csp->flags |= CSP_FLAG_BUFFERED_CONTENT_DEFLATED;
3502                      }
3503                   }
3504 #endif
3505
3506                   if (JB_ERR_OK != update_server_headers(csp))
3507                   {
3508                      log_error(LOG_LEVEL_FATAL,
3509                         "Failed to update server headers. after filtering.");
3510                   }
3511
3512                   hdr = list_to_text(csp->headers);
3513                   if (hdr == NULL)
3514                   {
3515                      /* FIXME Should handle error properly */
3516                      log_error(LOG_LEVEL_FATAL, "Out of memory parsing server header");
3517                   }
3518
3519 #ifdef FEATURE_HTTPS_INSPECTION
3520                   /*
3521                    * Sending data with standard or secured connection (HTTP/HTTPS)
3522                    */
3523                   if (client_use_ssl(csp))
3524                   {
3525                      if ((ssl_send_data_delayed(&(csp->ssl_client_attr),
3526                               (const unsigned char *)hdr, strlen(hdr),
3527                               get_write_delay(csp)) < 0)
3528                         || (ssl_send_data_delayed(&(csp->ssl_client_attr),
3529                               (const unsigned char *) ((p != NULL) ? p : csp->iob->cur),
3530                               csp->content_length, get_write_delay(csp)) < 0))
3531                      {
3532                         log_error(LOG_LEVEL_ERROR, "write modified content to "
3533                            "client over TLS/SSL failed");
3534                         freez(hdr);
3535                         freez(p);
3536                         mark_server_socket_tainted(csp);
3537                         close_client_and_server_ssl_connections(csp);
3538                         return;
3539                      }
3540                   }
3541                   else
3542 #endif /* def FEATURE_HTTPS_INSPECTION */
3543                   {
3544                      if (write_socket_delayed(csp->cfd, hdr, strlen(hdr), write_delay)
3545                       || write_socket_delayed(csp->cfd, ((p != NULL) ? p : csp->iob->cur),
3546                          (size_t)csp->content_length, write_delay))
3547                      {
3548                         log_error(LOG_LEVEL_ERROR, "write modified content to client failed: %E");
3549                         freez(hdr);
3550                         freez(p);
3551                         mark_server_socket_tainted(csp);
3552                         return;
3553                      }
3554                   }
3555
3556                   freez(hdr);
3557                   freez(p);
3558                }
3559
3560                break; /* "game over, man" */
3561             }
3562
3563             /*
3564              * This is NOT the body, so
3565              * Let's pretend the server just sent us a blank line.
3566              */
3567             snprintf(csp->receive_buffer, csp->receive_buffer_size, "\r\n");
3568             len = (int)strlen(csp->receive_buffer);
3569
3570             /*
3571              * Now, let the normal header parsing algorithm below do its
3572              * job.  If it fails, we'll exit instead of continuing.
3573              */
3574
3575             ms_iis5_hack = 1;
3576          }
3577
3578          /*
3579           * If we're in the body of the server document, just write it to
3580           * the client, unless we need to buffer the body for later
3581           * content-filtering.
3582           */
3583          if (server_body || (http->ssl
3584 #ifdef FEATURE_HTTPS_INSPECTION
3585                && use_ssl_tunnel
3586 #endif
3587             ))
3588          {
3589             if (buffer_and_filter_content)
3590             {
3591                /*
3592                 * If there is no memory left for buffering the content, or the buffer limit
3593                 * has been reached, switch to non-filtering mode, i.e. make & write the
3594                 * header, flush the iob and buf, and get out of the way.
3595                 */
3596                if (add_to_iob(csp->iob, csp->config->buffer_limit, csp->receive_buffer, len))
3597                {
3598                   size_t hdrlen;
3599                   long flushed;
3600
3601                   log_error(LOG_LEVEL_INFO,
3602                      "Flushing header and buffers. Stepping back from filtering.");
3603
3604                   hdr = list_to_text(csp->headers);
3605                   if (hdr == NULL)
3606                   {
3607                      /*
3608                       * Memory is too tight to even generate the header.
3609                       * Send our static "Out-of-memory" page.
3610                       */
3611                      log_error(LOG_LEVEL_ERROR, "Out of memory while trying to flush.");
3612                      rsp = cgi_error_memory();
3613                      send_crunch_response(csp, rsp);
3614                      mark_server_socket_tainted(csp);
3615 #ifdef FEATURE_HTTPS_INSPECTION
3616                      close_client_and_server_ssl_connections(csp);
3617 #endif
3618                      return;
3619                   }
3620                   hdrlen = strlen(hdr);
3621
3622 #ifdef FEATURE_HTTPS_INSPECTION
3623                   /*
3624                    * Sending data with standard or secured connection (HTTP/HTTPS)
3625                    */
3626                   if (client_use_ssl(csp))
3627                   {
3628                      if ((ssl_send_data_delayed(&(csp->ssl_client_attr),
3629                              (const unsigned char *)hdr, hdrlen, get_write_delay(csp)) < 0)
3630                         || ((flushed = ssl_flush_socket(&(csp->ssl_client_attr),
3631                                 csp->iob)) < 0)
3632                         || (ssl_send_data_delayed(&(csp->ssl_client_attr),
3633                               (const unsigned char *)csp->receive_buffer, (size_t)len,
3634                               get_write_delay(csp)) < 0))
3635                      {
3636                         log_error(LOG_LEVEL_CONNECT,
3637                            "Flush header and buffers to client failed");
3638                         freez(hdr);
3639                         mark_server_socket_tainted(csp);
3640                         close_client_and_server_ssl_connections(csp);
3641                         return;
3642                      }
3643                   }
3644                   else
3645 #endif /* def FEATURE_HTTPS_INSPECTION */
3646                   {
3647                      if (write_socket_delayed(csp->cfd, hdr, hdrlen, write_delay)
3648                       || ((flushed = flush_iob(csp->cfd, csp->iob, write_delay)) < 0)
3649                       || write_socket_delayed(csp->cfd, csp->receive_buffer, (size_t)len,
3650                             write_delay))
3651                      {
3652                         log_error(LOG_LEVEL_CONNECT,
3653                            "Flush header and buffers to client failed: %E");
3654                         freez(hdr);
3655                         mark_server_socket_tainted(csp);
3656                         return;
3657                      }
3658                   }
3659
3660                   /*
3661                    * Reset the byte_count to the amount of bytes
3662                    * we just flushed. len will be added a few lines below,
3663                    * hdrlen doesn't matter for LOG_LEVEL_CLF.
3664                    */
3665                   byte_count = (unsigned long long)flushed;
3666                   freez(hdr);
3667                   buffer_and_filter_content = 0;
3668                   server_body = 1;
3669                }
3670             }
3671             else
3672             {
3673 #ifdef FEATURE_HTTPS_INSPECTION
3674                /*
3675                 * Sending data with standard or secured connection (HTTP/HTTPS)
3676                 */
3677                if (client_use_ssl(csp))
3678                {
3679                   ret = ssl_send_data_delayed(&(csp->ssl_client_attr),
3680                      (const unsigned char *)csp->receive_buffer, (size_t)len,
3681                      get_write_delay(csp));
3682                   if (ret < 0)
3683                   {
3684                      log_error(LOG_LEVEL_ERROR,
3685                         "Sending data to client failed");
3686                      mark_server_socket_tainted(csp);
3687                      close_client_and_server_ssl_connections(csp);
3688                      return;
3689                   }
3690                }
3691                else
3692 #endif /* def FEATURE_HTTPS_INSPECTION */
3693                {
3694                   if (write_socket_delayed(csp->cfd, csp->receive_buffer,
3695                         (size_t)len, write_delay))
3696                   {
3697                      log_error(LOG_LEVEL_ERROR, "write to client failed: %E");
3698                      mark_server_socket_tainted(csp);
3699                      return;
3700                   }
3701                }
3702             }
3703             byte_count += (unsigned long long)len;
3704             continue;
3705          }
3706          else
3707          {
3708             /*
3709              * We're still looking for the end of the server's header.
3710              * Buffer up the data we just read.  If that fails, there's
3711              * little we can do but send our static out-of-memory page.
3712              */
3713             if (add_to_iob(csp->iob, csp->config->buffer_limit, csp->receive_buffer, len))
3714             {
3715                log_error(LOG_LEVEL_ERROR, "Out of memory while looking for end of server headers.");
3716                rsp = cgi_error_memory();
3717                send_crunch_response(csp, rsp);
3718                mark_server_socket_tainted(csp);
3719 #ifdef FEATURE_HTTPS_INSPECTION
3720                close_client_and_server_ssl_connections(csp);
3721 #endif
3722                return;
3723             }
3724
3725             /* Convert iob into something sed() can digest */
3726             if (JB_ERR_PARSE == get_server_headers(csp))
3727             {
3728                if (ms_iis5_hack)
3729                {
3730                   /*
3731                    * Well, we tried our MS IIS/5 hack and it didn't work.
3732                    * The header is incomplete and there isn't anything
3733                    * we can do about it.
3734                    */
3735                   log_error(LOG_LEVEL_ERROR, "Invalid server headers. "
3736                      "Applying the MS IIS5 hack didn't help.");
3737                   log_error(LOG_LEVEL_CLF,
3738                      "%s - - [%T] \"%s\" 502 0", csp->ip_addr_str, http->cmd);
3739 #ifdef FEATURE_HTTPS_INSPECTION
3740                   /*
3741                    * Sending data with standard or secured connection (HTTP/HTTPS)
3742                    */
3743                   if (client_use_ssl(csp))
3744                   {
3745                      ssl_send_data_delayed(&(csp->ssl_client_attr),
3746                         (const unsigned char *)INVALID_SERVER_HEADERS_RESPONSE,
3747                         strlen(INVALID_SERVER_HEADERS_RESPONSE), get_write_delay(csp));
3748                   }
3749                   else
3750 #endif /* def FEATURE_HTTPS_INSPECTION */
3751                   {
3752                      write_socket_delayed(csp->cfd,
3753                         INVALID_SERVER_HEADERS_RESPONSE,
3754                         strlen(INVALID_SERVER_HEADERS_RESPONSE), write_delay);
3755                   }
3756                   mark_server_socket_tainted(csp);
3757 #ifdef FEATURE_HTTPS_INSPECTION
3758                   close_client_and_server_ssl_connections(csp);
3759 #endif
3760                   return;
3761                }
3762                else
3763                {
3764                   /*
3765                    * Since we have to wait for more from the server before
3766                    * we can parse the headers we just continue here.
3767                    */
3768                   log_error(LOG_LEVEL_CONNECT,
3769                      "Continuing buffering server headers from socket %d. "
3770                      "Bytes most recently read: %ld.", csp->cfd, len);
3771                   continue;
3772                }
3773             }
3774             else
3775             {
3776                /*
3777                 * Account for the content bytes we
3778                 * might have gotten with the headers.
3779                 */
3780                assert(csp->iob->eod >= csp->iob->cur);
3781                byte_count = (unsigned long long)(csp->iob->eod - csp->iob->cur);
3782             }
3783
3784             /* Did we actually get anything? */
3785             if (NULL == csp->headers->first)
3786             {
3787                if ((csp->flags & CSP_FLAG_REUSED_CLIENT_CONNECTION))
3788                {
3789                   log_error(LOG_LEVEL_ERROR,
3790                      "No server or forwarder response received on socket %d. "
3791                      "Closing client socket %d without sending data.",
3792                      csp->server_connection.sfd, csp->cfd);
3793                   log_error(LOG_LEVEL_CLF,
3794                      "%s - - [%T] \"%s\" 502 0", csp->ip_addr_str, http->cmd);
3795                }
3796                else
3797                {
3798                   log_error(LOG_LEVEL_ERROR,
3799                      "No server or forwarder response received on socket %d.",
3800                      csp->server_connection.sfd);
3801                   send_crunch_response(csp, error_response(csp, "no-server-data"));
3802                }
3803                free_http_request(http);
3804                mark_server_socket_tainted(csp);
3805 #ifdef FEATURE_HTTPS_INSPECTION
3806                close_client_and_server_ssl_connections(csp);
3807 #endif
3808                return;
3809             }
3810
3811             if (!csp->headers->first->str)
3812             {
3813                log_error(LOG_LEVEL_ERROR, "header search: csp->headers->first->str == NULL, assert will be called");
3814             }
3815             assert(csp->headers->first->str);
3816
3817             if (strncmpic(csp->headers->first->str, "HTTP", 4) &&
3818                 strncmpic(csp->headers->first->str, "ICY", 3))
3819             {
3820                /*
3821                 * It doesn't look like a HTTP (or Shoutcast) response:
3822                 * tell the client and log the problem.
3823                 */
3824                if (strlen(csp->headers->first->str) > 30)
3825                {
3826                   csp->headers->first->str[30] = '\0';
3827                }
3828                log_error(LOG_LEVEL_ERROR,
3829                   "Invalid server or forwarder response. Starts with: %s",
3830                   csp->headers->first->str);
3831                log_error(LOG_LEVEL_CLF,
3832                   "%s - - [%T] \"%s\" 502 0", csp->ip_addr_str, http->cmd);
3833 #ifdef FEATURE_HTTPS_INSPECTION
3834                /*
3835                 * Sending data with standard or secured connection (HTTP/HTTPS)
3836                 */
3837                if (client_use_ssl(csp))
3838                {
3839                   ssl_send_data_delayed(&(csp->ssl_client_attr),
3840                      (const unsigned char *)INVALID_SERVER_HEADERS_RESPONSE,
3841                      strlen(INVALID_SERVER_HEADERS_RESPONSE),
3842                      get_write_delay(csp));
3843                }
3844                else
3845 #endif /* def FEATURE_HTTPS_INSPECTION */
3846                {
3847                   write_socket_delayed(csp->cfd, INVALID_SERVER_HEADERS_RESPONSE,
3848                      strlen(INVALID_SERVER_HEADERS_RESPONSE), write_delay);
3849                }
3850                free_http_request(http);
3851                mark_server_socket_tainted(csp);
3852 #ifdef FEATURE_HTTPS_INSPECTION
3853                close_client_and_server_ssl_connections(csp);
3854 #endif
3855                return;
3856             }
3857
3858             /*
3859              * We have now received the entire server header,
3860              * filter it and send the result to the client
3861              */
3862             if (JB_ERR_OK != sed(csp, FILTER_SERVER_HEADERS))
3863             {
3864                log_error(LOG_LEVEL_CLF,
3865                   "%s - - [%T] \"%s\" 502 0", csp->ip_addr_str, http->cmd);
3866 #ifdef FEATURE_HTTPS_INSPECTION
3867                /*
3868                 * Sending data with standard or secured connection (HTTP/HTTPS)
3869                 */
3870                if (client_use_ssl(csp))
3871                {
3872                   ssl_send_data_delayed(&(csp->ssl_client_attr),
3873                      (const unsigned char *)INVALID_SERVER_HEADERS_RESPONSE,
3874                      strlen(INVALID_SERVER_HEADERS_RESPONSE),
3875                      get_write_delay(csp));
3876                }
3877                else
3878 #endif
3879                {
3880                   write_socket_delayed(csp->cfd, INVALID_SERVER_HEADERS_RESPONSE,
3881                      strlen(INVALID_SERVER_HEADERS_RESPONSE), write_delay);
3882                }
3883                free_http_request(http);
3884                mark_server_socket_tainted(csp);
3885 #ifdef FEATURE_HTTPS_INSPECTION
3886                close_client_and_server_ssl_connections(csp);
3887 #endif
3888                return;
3889             }
3890             hdr = list_to_text(csp->headers);
3891             if (hdr == NULL)
3892             {
3893                /* FIXME Should handle error properly */
3894                log_error(LOG_LEVEL_FATAL, "Out of memory parsing server header");
3895             }
3896
3897             if ((csp->flags & CSP_FLAG_CHUNKED)
3898                && !(csp->flags & CSP_FLAG_CONTENT_LENGTH_SET)
3899                && ((csp->iob->eod - csp->iob->cur) >= 5)
3900                && !memcmp(csp->iob->eod-5, "0\r\n\r\n", 5))
3901             {
3902                log_error(LOG_LEVEL_CONNECT,
3903                   "Looks like we got the last chunk together with "
3904                   "the server headers. We better stop reading.");
3905                byte_count = (unsigned long long)(csp->iob->eod - csp->iob->cur);
3906                csp->expected_content_length = byte_count;
3907                csp->flags |= CSP_FLAG_CONTENT_LENGTH_SET;
3908             }
3909
3910             csp->server_connection.response_received = time(NULL);
3911
3912             if (crunch_response_triggered(csp, crunchers_light))
3913             {
3914                /*
3915                 * One of the tags created by a server-header
3916                 * tagger triggered a crunch. We already
3917                 * delivered the crunch response to the client
3918                 * and are done here after cleaning up.
3919                 */
3920                freez(hdr);
3921                mark_server_socket_tainted(csp);
3922 #ifdef FEATURE_HTTPS_INSPECTION
3923                close_client_and_server_ssl_connections(csp);
3924 #endif
3925                return;
3926             }
3927
3928             /* Buffer and pcrs filter this if appropriate. */
3929             buffer_and_filter_content = content_requires_filtering(csp);
3930
3931             if (!buffer_and_filter_content)
3932             {
3933                /*
3934                 * Write the server's (modified) header to
3935                 * the client (along with anything else that
3936                 * may be in the buffer). Use standard or secured
3937                 * connection.
3938                 */
3939 #ifdef FEATURE_HTTPS_INSPECTION
3940                if (client_use_ssl(csp))
3941                {
3942                   if ((ssl_send_data_delayed(&(csp->ssl_client_attr),
3943                           (const unsigned char *)hdr, strlen(hdr),
3944                           get_write_delay(csp)) < 0)
3945                      || ((len = ssl_flush_socket(&(csp->ssl_client_attr),
3946                             csp->iob)) < 0))
3947                   {
3948                      log_error(LOG_LEVEL_CONNECT, "Write header to client failed");
3949
3950                      /*
3951                       * The write failed, so don't bother mentioning it
3952                       * to the client... it probably can't hear us anyway.
3953                       */
3954                      freez(hdr);
3955                      mark_server_socket_tainted(csp);
3956 #ifdef FEATURE_HTTPS_INSPECTION
3957                      close_client_and_server_ssl_connections(csp);
3958 #endif
3959                      return;
3960                   }
3961                }
3962                else
3963 #endif /* def FEATURE_HTTPS_INSPECTION */
3964                {
3965                   if (write_socket_delayed(csp->cfd, hdr, strlen(hdr), write_delay)
3966                      || ((len = flush_iob(csp->cfd, csp->iob, write_delay)) < 0))
3967                   {
3968                      log_error(LOG_LEVEL_ERROR,
3969                         "write header to client failed");
3970                      /*
3971                       * The write failed, so don't bother mentioning it
3972                       * to the client... it probably can't hear us anyway.
3973                       */
3974                      freez(hdr);
3975                      mark_server_socket_tainted(csp);
3976                      return;
3977                   }
3978                }
3979                                 }
3980
3981             /* we're finished with the server's header */
3982
3983             freez(hdr);
3984             server_body = 1;
3985
3986             /*
3987              * If this was a MS IIS/5 hack then it means the server
3988              * has already closed the connection. Nothing more to read.
3989              * Time to bail.
3990              */
3991             if (ms_iis5_hack)
3992             {
3993                log_error(LOG_LEVEL_ERROR,
3994                   "Closed server connection detected. "
3995                   "Applying the MS IIS5 hack didn't help.");
3996                log_error(LOG_LEVEL_CLF,
3997                   "%s - - [%T] \"%s\" 502 0", csp->ip_addr_str, http->cmd);
3998 #ifdef FEATURE_HTTPS_INSPECTION
3999                /*
4000                 * Sending data with standard or secured connection (HTTP/HTTPS)
4001                 */
4002                if (client_use_ssl(csp))
4003                {
4004                   ssl_send_data_delayed(&(csp->ssl_client_attr),
4005                      (const unsigned char *)INVALID_SERVER_HEADERS_RESPONSE,
4006                      strlen(INVALID_SERVER_HEADERS_RESPONSE),
4007                      get_write_delay(csp));
4008                }
4009                else
4010 #endif /* def FEATURE_HTTPS_INSPECTION */
4011                {
4012                   write_socket_delayed(csp->cfd, INVALID_SERVER_HEADERS_RESPONSE,
4013                      strlen(INVALID_SERVER_HEADERS_RESPONSE), write_delay);
4014                }
4015                mark_server_socket_tainted(csp);
4016 #ifdef FEATURE_HTTPS_INSPECTION
4017                close_client_and_server_ssl_connections(csp);
4018 #endif
4019                return;
4020             }
4021          }
4022          continue;
4023       }
4024       mark_server_socket_tainted(csp);
4025 #ifdef FEATURE_HTTPS_INSPECTION
4026       close_client_and_server_ssl_connections(csp);
4027 #endif
4028       return; /* huh? we should never get here */
4029    }
4030
4031    if (csp->content_length == 0)
4032    {
4033       /*
4034        * If Privoxy didn't recalculate the Content-Length,
4035        * byte_count is still correct.
4036        */
4037       csp->content_length = byte_count;
4038    }
4039
4040 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
4041    if ((csp->flags & CSP_FLAG_CONTENT_LENGTH_SET)
4042       && (csp->expected_content_length != byte_count))
4043    {
4044       log_error(LOG_LEVEL_CONNECT,
4045          "Received %llu bytes while expecting %llu.",
4046          byte_count, csp->expected_content_length);
4047       mark_server_socket_tainted(csp);
4048    }
4049 #endif
4050
4051 #ifdef FEATURE_HTTPS_INSPECTION
4052    if (client_use_ssl(csp))
4053    {
4054       log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s https://%s%s %s\" 200 %llu",
4055          csp->ip_addr_str, http->gpc, http->hostport, http->path,
4056          http->version, csp->content_length);
4057    }
4058    else
4059 #endif
4060    {
4061       log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 200 %llu",
4062          csp->ip_addr_str, http->ocmd, csp->content_length);
4063    }
4064    csp->server_connection.timestamp = time(NULL);
4065 }
4066
4067
4068 /*********************************************************************
4069  *
4070  * Function    :  chat
4071  *
4072  * Description :  Once a connection from the client has been accepted,
4073  *                this function is called (via serve()) to handle the
4074  *                main business of the communication.  This function
4075  *                returns after dealing with a single request. It can
4076  *                be called multiple times with the same client socket
4077  *                if the client is keeping the connection alive.
4078  *
4079  *                The decision whether or not a client connection will
4080  *                be kept alive is up to the caller which also must
4081  *                close the client socket when done.
4082  *
4083  *                FIXME: chat is nearly thousand lines long.
4084  *                Ridiculous.
4085  *
4086  * Parameters  :
4087  *          1  :  csp = Current client state (buffers, headers, etc...)
4088  *
4089  * Returns     :  Nothing.
4090  *
4091  *********************************************************************/
4092 static void chat(struct client_state *csp)
4093 {
4094    const struct forward_spec *fwd;
4095    struct http_request *http;
4096    /* Skeleton for HTTP response, if we should intercept the request */
4097    struct http_response *rsp;
4098 #ifdef FEATURE_HTTPS_INSPECTION
4099    int use_ssl_tunnel = 0;
4100 #endif
4101
4102    http = csp->http;
4103
4104    if (receive_client_request(csp) != JB_ERR_OK)
4105    {
4106       return;
4107    }
4108    if (parse_client_request(csp) != JB_ERR_OK)
4109    {
4110       return;
4111    }
4112
4113    /* decide how to route the HTTP request */
4114    fwd = forward_url(csp, http);
4115
4116 #ifdef FEATURE_HTTPS_INSPECTION
4117    /*
4118     * Setting flags to use old solution with SSL tunnel and to disable
4119     * certificate verification.
4120     */
4121    if (csp->http->ssl && !(csp->action->flags & ACTION_HTTPS_INSPECTION)
4122       && !cgi_page_requested(csp->http->host))
4123    {
4124       use_ssl_tunnel = 1;
4125    }
4126
4127    if (http->ssl && (csp->action->flags & ACTION_IGNORE_CERTIFICATE_ERRORS))
4128    {
4129       csp->dont_verify_certificate = 1;
4130    }
4131 #endif
4132
4133    /*
4134     * build the http request to send to the server
4135     * we have to do one of the following:
4136     *
4137     * create =    use the original HTTP request to create a new
4138     *             HTTP request that has either the path component
4139     *             without the http://domainspec (w/path) or the
4140     *             full orininal URL (w/url)
4141     *             Note that the path and/or the HTTP version may
4142     *             have been altered by now.
4143     *
4144     * SSL proxy = Open a socket to the host:port of the server
4145     *             and create TLS/SSL connection with server and
4146     *             with client. Then behave like mediator between
4147     *             client and server over TLS/SSL.
4148     *
4149     * SSL proxy = Pass the request unchanged if forwarding a CONNECT
4150     *    with     request to a parent proxy. Note that we'll be sending
4151     * forwarding  the CFAIL message ourselves if connecting to the parent
4152     *             fails, but we won't send a CSUCCEED message if it works,
4153     *             since that would result in a double message (ours and the
4154     *             parent's). After sending the request to the parent, we
4155     *             must parse answer and send it to client. If connection
4156     *             with server is established, we do TLS/SSL proxy. Otherwise
4157     *             we send parent response to client and close connections.
4158     *
4159     * here's the matrix:
4160     *                        SSL
4161     *                    0        1
4162     *                +--------+--------+
4163     *                |        |        |
4164     *             0  | create |   SSL  |
4165     *                | w/path |  proxy |
4166     *  Forwarding    +--------+--------+
4167     *                |        |   SSL  |
4168     *             1  | create |  proxy |
4169     *                | w/url  |+forward|
4170     *                +--------+--------+
4171     *
4172     */
4173
4174 #ifdef FEATURE_HTTPS_INSPECTION
4175    /*
4176     * Presetting SSL client and server flags
4177     */
4178    if (http->ssl && !use_ssl_tunnel)
4179    {
4180       http->client_ssl = 1;
4181       http->server_ssl = 1;
4182    }
4183    else
4184    {
4185       http->client_ssl = 0;
4186       http->server_ssl = 0;
4187    }
4188 #endif
4189
4190 #ifdef FEATURE_HTTPS_INSPECTION
4191    /*
4192     * Log the request unless we're https inspecting
4193     * in which case we don't have the path yet and
4194     * will log the request later.
4195     */
4196    if (!client_use_ssl(csp))
4197 #endif
4198    {
4199       log_error(LOG_LEVEL_REQUEST, "%s%s", http->hostport, http->path);
4200    }
4201
4202    if (http->ssl && connect_port_is_forbidden(csp))
4203    {
4204       const char *acceptable_connect_ports =
4205          csp->action->string[ACTION_STRING_LIMIT_CONNECT];
4206       assert(NULL != acceptable_connect_ports);
4207       log_error(LOG_LEVEL_INFO, "Request from %s marked for blocking. "
4208          "limit-connect{%s} doesn't allow CONNECT requests to %s",
4209          csp->ip_addr_str, acceptable_connect_ports, csp->http->hostport);
4210       csp->action->flags |= ACTION_BLOCK;
4211       http->ssl = 0;
4212 #ifdef FEATURE_HTTPS_INSPECTION
4213       http->client_ssl = 0;
4214       http->server_ssl = 0;
4215 #endif
4216    }
4217
4218
4219    freez(csp->headers->first->str);
4220    build_request_line(csp, fwd, &csp->headers->first->str);
4221
4222    /*
4223     * We have a request. Check if one of the crunchers wants it
4224     * unless the client wants to use TLS/SSL in which case we
4225     * haven't setup the TLS context yet and will send the crunch
4226     * response later.
4227     */
4228    if (
4229 #ifdef FEATURE_HTTPS_INSPECTION
4230        !client_use_ssl(csp) &&
4231 #endif
4232        crunch_response_triggered(csp, crunchers_all))
4233    {
4234       /*
4235        * Yes. The client got the crunch response and we're done here.
4236        */
4237       return;
4238    }
4239
4240    log_applied_actions(csp->action);
4241    if (fwd->forward_host)
4242    {
4243       log_error(LOG_LEVEL_CONNECT, "via [%s]:%d to: %s",
4244          fwd->forward_host, fwd->forward_port, http->hostport);
4245    }
4246    else
4247    {
4248       log_error(LOG_LEVEL_CONNECT, "to %s", http->hostport);
4249    }
4250
4251    /* here we connect to the server, gateway, or the forwarder */
4252
4253 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
4254    if ((csp->server_connection.sfd != JB_INVALID_SOCKET)
4255       && socket_is_still_alive(csp->server_connection.sfd)
4256       && connection_destination_matches(&csp->server_connection, http, fwd))
4257    {
4258       log_error(LOG_LEVEL_CONNECT,
4259          "Reusing server socket %d connected to %s. Total requests: %u.",
4260          csp->server_connection.sfd, csp->server_connection.host,
4261          csp->server_connection.requests_sent_total);
4262    }
4263    else
4264    {
4265       if (csp->server_connection.sfd != JB_INVALID_SOCKET)
4266       {
4267 #ifdef FEATURE_CONNECTION_SHARING
4268          if (csp->config->feature_flags & RUNTIME_FEATURE_CONNECTION_SHARING
4269 #ifdef FEATURE_HTTPS_INSPECTION
4270             && !server_use_ssl(csp)
4271 #endif
4272             )
4273          {
4274             remember_connection(&csp->server_connection);
4275          }
4276          else
4277 #endif /* def FEATURE_CONNECTION_SHARING */
4278          {
4279             log_error(LOG_LEVEL_CONNECT,
4280                "Closing server socket %d connected to %s. Total requests: %u.",
4281                csp->server_connection.sfd, csp->server_connection.host,
4282                csp->server_connection.requests_sent_total);
4283             close_socket(csp->server_connection.sfd);
4284          }
4285          mark_connection_closed(&csp->server_connection);
4286       }
4287 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
4288 #ifdef FEATURE_HTTPS_INSPECTION
4289       if (client_use_ssl(csp) && !use_ssl_tunnel)
4290       {
4291          int ret;
4292          /*
4293           * Creating a SSL proxy.
4294           *
4295           * By sending the CSUCCEED message we're lying to the client as
4296           * the connection hasn't actually been established yet. We don't
4297           * establish the connection until we have seen and parsed the
4298           * encrypted client headers.
4299           */
4300          if (write_socket_delayed(csp->cfd, CSUCCEED,
4301                strlen(CSUCCEED), get_write_delay(csp)) != 0)
4302          {
4303             log_error(LOG_LEVEL_ERROR, "Sending SUCCEED to client failed");
4304             return;
4305          }
4306
4307          ret = create_client_ssl_connection(csp);
4308          if (ret != 0)
4309          {
4310             log_error(LOG_LEVEL_ERROR,
4311                "Failed to open a secure connection with the client");
4312             return;
4313          }
4314          if (JB_ERR_OK != process_encrypted_request(csp))
4315          {
4316             close_client_ssl_connection(csp);
4317             return;
4318          }
4319          /*
4320           * We have an encrypted request. Check if one of the crunchers now
4321           * wants it (for example because the previously invisible path was
4322           * required to match).
4323           */
4324          if (crunch_response_triggered(csp, crunchers_all))
4325          {
4326             /*
4327              * Yes. The client got the crunch response and we're done here.
4328              */
4329             return;
4330          }
4331       }
4332 #endif
4333       /*
4334        * Connecting to destination server
4335        */
4336       csp->server_connection.sfd = forwarded_connect(fwd, http, csp);
4337
4338       if (csp->server_connection.sfd == JB_INVALID_SOCKET)
4339       {
4340          if (fwd->type != SOCKS_NONE)
4341          {
4342             /* Socks error. */
4343             rsp = error_response(csp, "forwarding-failed");
4344          }
4345          else if (errno == EINVAL)
4346          {
4347             rsp = error_response(csp, "no-such-domain");
4348          }
4349          else
4350          {
4351             rsp = error_response(csp, "connect-failed");
4352          }
4353
4354          /* Write the answer to the client */
4355          if (rsp != NULL)
4356          {
4357             send_crunch_response(csp, rsp);
4358          }
4359
4360          /*
4361           * Temporary workaround to prevent already-read client
4362           * bodies from being parsed as new requests. For now we
4363           * err on the safe side and throw all the following
4364           * requests under the bus, even if no client body has been
4365           * buffered. A compliant client will repeat the dropped
4366           * requests on an untainted connection.
4367           *
4368           * The proper fix is to discard the no longer needed
4369           * client body in the buffer (if there is one) and to
4370           * continue parsing the bytes that follow.
4371           */
4372 #ifdef FEATURE_HTTPS_INSPECTION
4373          close_client_ssl_connection(csp);
4374 #endif
4375          drain_and_close_socket(csp->cfd);
4376          csp->cfd = JB_INVALID_SOCKET;
4377
4378          return;
4379       }
4380
4381 #ifdef FEATURE_HTTPS_INSPECTION
4382       /*
4383        * Creating TLS/SSL connections with destination server or parent
4384        * proxy. If forwarding is enabled, we must send client request to
4385        * parent proxy and receive, parse and resend parent proxy answer.
4386        */
4387       if (http->ssl && !use_ssl_tunnel)
4388       {
4389          if (fwd->forward_host != NULL)
4390          {
4391             char server_response[BUFFER_SIZE];
4392             int ret = 0;
4393             int len = 0;
4394             char *hdr = list_to_text(csp->headers);
4395             memset(server_response, 0, sizeof(server_response));
4396
4397             if (hdr == NULL)
4398             {
4399                log_error(LOG_LEVEL_FATAL,
4400                   "Out of memory parsing client header");
4401             }
4402             list_remove_all(csp->headers);
4403
4404             /*
4405              * Sending client's CONNECT request to the parent proxy
4406              */
4407             ret = write_socket(csp->server_connection.sfd, hdr, strlen(hdr));
4408
4409             freez(hdr);
4410
4411             if (ret != 0)
4412             {
4413                log_error(LOG_LEVEL_CONNECT,
4414                   "Sending request headers to: %s failed", http->hostport);
4415                mark_server_socket_tainted(csp);
4416                close_client_ssl_connection(csp);
4417                return;
4418             }
4419
4420             /* Waiting for parent proxy server response */
4421             len = read_socket(csp->server_connection.sfd, server_response,
4422                sizeof(server_response)-1);
4423
4424             if (len <= 0)
4425             {
4426                log_error(LOG_LEVEL_ERROR, "No response from parent proxy "
4427                   "server on socket %d.", csp->server_connection.sfd);
4428
4429                rsp = error_response(csp, "no-server-data");
4430                if (rsp)
4431                {
4432                   send_crunch_response(csp, rsp);
4433                }
4434                mark_server_socket_tainted(csp);
4435                close_client_ssl_connection(csp);
4436                return;
4437             }
4438
4439             /*
4440              * Test if the connection to the destination server was
4441              * established successfully by the parent proxy.
4442              */
4443             if (!tunnel_established_successfully(server_response, (unsigned int)len))
4444             {
4445                log_error(LOG_LEVEL_ERROR,
4446                   "The forwarder %s failed to establish a connection with %s",
4447                   fwd->forward_host, http->host);
4448                rsp = error_response(csp, "connect-failed");
4449                if (rsp)
4450                {
4451                   send_crunch_response(csp, rsp);
4452                }
4453                mark_server_socket_tainted(csp);
4454                close_client_ssl_connection(csp);
4455                return;
4456             }
4457          } /* -END- if (fwd->forward_host != NULL) */
4458
4459          /*
4460           * We can now create the TLS/SSL connection with the destination server.
4461           */
4462          int ret = create_server_ssl_connection(csp);
4463          if (ret != 0)
4464          {
4465             if (csp->server_cert_verification_result != SSL_CERT_VALID &&
4466                 csp->server_cert_verification_result != SSL_CERT_NOT_VERIFIED)
4467             {
4468                /*
4469                 * If the server certificate is invalid, we must inform
4470                 * the client and then close connection to the client.
4471                 */
4472                ssl_send_certificate_error(csp);
4473                close_client_and_server_ssl_connections(csp);
4474                return;
4475             }
4476             if (csp->server_cert_verification_result == SSL_CERT_NOT_VERIFIED
4477              || csp->server_cert_verification_result == SSL_CERT_VALID)
4478             {
4479                /*
4480                 * The TLS/SSL connection wasn't created but an invalid
4481                 * certificate wasn't detected. Report it as connection
4482                 * failure.
4483                 */
4484                rsp = error_response(csp, "connect-failed");
4485                if (rsp)
4486                {
4487                   send_crunch_response(csp, rsp);
4488                }
4489                close_client_and_server_ssl_connections(csp);
4490                return;
4491             }
4492          }
4493       }/* -END- if (http->ssl) */
4494 #endif /* def FEATURE_HTTPS_INSPECTION */
4495
4496 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
4497       save_connection_destination(csp->server_connection.sfd,
4498          http, fwd, &csp->server_connection);
4499       csp->server_connection.keep_alive_timeout =
4500          (unsigned)csp->config->keep_alive_timeout;
4501    }
4502 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
4503
4504    csp->server_connection.requests_sent_total++;
4505
4506    if ((fwd->type == SOCKS_5T) && (NULL == csp->headers->first))
4507    {
4508       /* Client headers have been sent optimistically */
4509       assert(csp->headers->last == NULL);
4510    }
4511    else if (http->ssl == 0 || (fwd->forward_host
4512 #ifdef FEATURE_HTTPS_INSPECTION
4513          && use_ssl_tunnel
4514 #endif
4515            ))
4516    {
4517       if (send_http_request(csp))
4518       {
4519          rsp = error_response(csp, "connect-failed");
4520          if (rsp)
4521          {
4522             send_crunch_response(csp, rsp);
4523          }
4524          return;
4525       }
4526    }
4527    else
4528    {
4529       /*
4530        * Using old solution with SSL tunnel or new solution with SSL proxy
4531        */
4532       list_remove_all(csp->headers);
4533 #ifdef FEATURE_HTTPS_INSPECTION
4534       if (use_ssl_tunnel)
4535 #endif
4536       {
4537          /*
4538          * We're running an SSL tunnel and we're not forwarding,
4539          * so just ditch the client headers, send the "connect succeeded"
4540          * message to the client, flush the rest, and get out of the way.
4541          */
4542          if (write_socket_delayed(csp->cfd, CSUCCEED,
4543                strlen(CSUCCEED), get_write_delay(csp)))
4544          {
4545             return;
4546          }
4547       }
4548 #ifdef FEATURE_HTTPS_INSPECTION
4549       else
4550       {
4551          /*
4552           * If server certificate has been verified and is invalid,
4553           * we must inform the client and then close the connection
4554           * with client and server.
4555           */
4556          if (csp->server_cert_verification_result != SSL_CERT_VALID &&
4557              csp->server_cert_verification_result != SSL_CERT_NOT_VERIFIED)
4558          {
4559             ssl_send_certificate_error(csp);
4560             close_client_and_server_ssl_connections(csp);
4561             return;
4562          }
4563          if (send_https_request(csp))
4564          {
4565             rsp = error_response(csp, "connect-failed");
4566             if (rsp)
4567             {
4568                send_crunch_response(csp, rsp);
4569             }
4570             close_client_and_server_ssl_connections(csp);
4571             return;
4572          }
4573       }
4574 #endif /* def FEATURE_HTTPS_INSPECTION */
4575       clear_iob(csp->client_iob);
4576    }/* -END- else ... if (http->ssl == 1) */
4577
4578    log_error(LOG_LEVEL_CONNECT, "to %s successful", http->hostport);
4579
4580    /* XXX: should the time start earlier for optimistically sent data? */
4581    csp->server_connection.request_sent = time(NULL);
4582
4583    handle_established_connection(csp);
4584    freez(csp->receive_buffer);
4585 }
4586
4587
4588 #ifdef FUZZ
4589 /*********************************************************************
4590  *
4591  * Function    :  fuzz_server_response
4592  *
4593  * Description :  Treat the input as a whole server response.
4594  *
4595  * Parameters  :
4596  *          1  :  csp = Current client state (buffers, headers, etc...)
4597  *          2  :  fuzz_input_file = File to read the input from.
4598  *
4599  * Returns     :  0
4600  *
4601  *********************************************************************/
4602 extern int fuzz_server_response(struct client_state *csp, char *fuzz_input_file)
4603 {
4604    static struct forward_spec fwd; /* Zero'd due to being static */
4605    csp->cfd = 0;
4606
4607    if (strcmp(fuzz_input_file, "-") == 0)
4608    {
4609       /* XXX: Doesn't work yet. */
4610       csp->server_connection.sfd = 0;
4611    }
4612    else
4613    {
4614       csp->server_connection.sfd = open(fuzz_input_file, O_RDONLY);
4615       if (csp->server_connection.sfd == -1)
4616       {
4617          log_error(LOG_LEVEL_FATAL, "Failed to open %s: %E",
4618             fuzz_input_file);
4619       }
4620    }
4621    csp->fwd = &fwd;
4622    csp->content_type |= CT_GIF;
4623    csp->action->flags |= ACTION_DEANIMATE;
4624    csp->action->string[ACTION_STRING_DEANIMATE] = "last";
4625
4626    csp->http->path = strdup_or_die("/");
4627    csp->http->host = strdup_or_die("fuzz.example.org");
4628    csp->http->hostport = strdup_or_die("fuzz.example.org:80");
4629    /* Prevent client socket monitoring */
4630    csp->flags |= CSP_FLAG_PIPELINED_REQUEST_WAITING;
4631    csp->flags |= CSP_FLAG_CHUNKED;
4632
4633    csp->config->feature_flags |= RUNTIME_FEATURE_CONNECTION_KEEP_ALIVE;
4634    csp->flags |= CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE;
4635
4636    csp->content_type |= CT_DECLARED|CT_GIF;
4637
4638    csp->config->socket_timeout = 0;
4639
4640    cgi_init_error_messages();
4641
4642    handle_established_connection(csp);
4643    freez(csp->receive_buffer);
4644
4645    return 0;
4646 }
4647 #endif
4648
4649
4650 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
4651 /*********************************************************************
4652  *
4653  * Function    :  prepare_csp_for_next_request
4654  *
4655  * Description :  Put the csp in a mostly vergin state.
4656  *
4657  * Parameters  :
4658  *          1  :  csp = Current client state (buffers, headers, etc...)
4659  *
4660  * Returns     :  N/A
4661  *
4662  *********************************************************************/
4663 static void prepare_csp_for_next_request(struct client_state *csp)
4664 {
4665    csp->content_type = 0;
4666    csp->content_length = 0;
4667    csp->expected_content_length = 0;
4668    csp->expected_client_content_length = 0;
4669    list_remove_all(csp->headers);
4670    clear_iob(csp->iob);
4671    freez(csp->error_message);
4672    free_http_request(csp->http);
4673    destroy_list(csp->headers);
4674 #ifdef FEATURE_HTTPS_INSPECTION
4675    destroy_list(csp->https_headers);
4676 #endif
4677    destroy_list(csp->tags);
4678 #ifdef FEATURE_CLIENT_TAGS
4679    destroy_list(csp->client_tags);
4680    freez(csp->client_address);
4681 #endif
4682    free_current_action(csp->action);
4683    if (NULL != csp->fwd)
4684    {
4685       unload_forward_spec(csp->fwd);
4686       csp->fwd = NULL;
4687    }
4688    /* XXX: Store per-connection flags someplace else. */
4689    csp->flags = (CSP_FLAG_ACTIVE | CSP_FLAG_REUSED_CLIENT_CONNECTION);
4690 #ifdef FEATURE_TOGGLE
4691    if (global_toggle_state)
4692 #endif /* def FEATURE_TOGGLE */
4693    {
4694       csp->flags |= CSP_FLAG_TOGGLED_ON;
4695    }
4696
4697    if (csp->client_iob->eod > csp->client_iob->cur)
4698    {
4699       long bytes_to_shift = csp->client_iob->cur - csp->client_iob->buf;
4700       size_t data_length  = (size_t)(csp->client_iob->eod - csp->client_iob->cur);
4701
4702       assert(bytes_to_shift > 0);
4703       assert(data_length > 0);
4704
4705       log_error(LOG_LEVEL_CONNECT, "Shifting %lu pipelined bytes by %ld bytes",
4706          data_length, bytes_to_shift);
4707       memmove(csp->client_iob->buf, csp->client_iob->cur, data_length);
4708       csp->client_iob->cur = csp->client_iob->buf;
4709       assert(csp->client_iob->eod == csp->client_iob->buf + bytes_to_shift + data_length);
4710       csp->client_iob->eod = csp->client_iob->buf + data_length;
4711       memset(csp->client_iob->eod, '\0', (size_t)bytes_to_shift);
4712
4713       csp->flags |= CSP_FLAG_PIPELINED_REQUEST_WAITING;
4714    }
4715    else
4716    {
4717       /*
4718        * We mainly care about resetting client_iob->cur so we don't
4719        * waste buffer space at the beginning and don't mess up the
4720        * request restoration done by cgi_show_request().
4721        *
4722        * Freeing the buffer itself isn't technically necessary,
4723        * but makes debugging more convenient.
4724        */
4725       clear_iob(csp->client_iob);
4726    }
4727 }
4728 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
4729
4730
4731 /*********************************************************************
4732  *
4733  * Function    :  serve
4734  *
4735  * Description :  This is little more than chat.  We only "serve" to
4736  *                to close (or remember) any socket that chat may have
4737  *                opened.
4738  *
4739  * Parameters  :
4740  *          1  :  csp = Current client state (buffers, headers, etc...)
4741  *
4742  * Returns     :  N/A
4743  *
4744  *********************************************************************/
4745 static void serve(struct client_state *csp)
4746 {
4747    int config_file_change_detected = 0; /* Only used for debugging */
4748 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
4749 #ifdef FEATURE_CONNECTION_SHARING
4750    static int monitor_thread_running = 0;
4751 #endif /* def FEATURE_CONNECTION_SHARING */
4752    int continue_chatting = 0;
4753
4754    log_error(LOG_LEVEL_CONNECT, "Accepted connection from %s on socket %d",
4755       csp->ip_addr_str, csp->cfd);
4756
4757    do
4758    {
4759       unsigned int latency;
4760
4761 #ifdef FEATURE_HTTPS_INSPECTION
4762       if (continue_chatting && client_use_ssl(csp))
4763       {
4764          continue_https_chat(csp);
4765       }
4766       else
4767 #endif
4768       {
4769          chat(csp);
4770       }
4771
4772       /*
4773        * If the request has been crunched,
4774        * the calculated latency is zero.
4775        */
4776       latency = (unsigned)(csp->server_connection.response_received -
4777          csp->server_connection.request_sent) / 2;
4778
4779       if ((csp->flags & CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE)
4780          && (csp->flags & CSP_FLAG_CRUNCHED)
4781          && (csp->expected_client_content_length != 0))
4782       {
4783          csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED;
4784          log_error(LOG_LEVEL_CONNECT,
4785             "Tainting client socket %d due to unread data.", csp->cfd);
4786       }
4787
4788       continue_chatting = (csp->config->feature_flags
4789          & RUNTIME_FEATURE_CONNECTION_KEEP_ALIVE)
4790          && !(csp->flags & CSP_FLAG_SERVER_SOCKET_TAINTED)
4791          && (csp->cfd != JB_INVALID_SOCKET)
4792          && (csp->flags & CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE)
4793          && ((csp->flags & CSP_FLAG_SERVER_CONTENT_LENGTH_SET)
4794             || (csp->flags & CSP_FLAG_CHUNKED));
4795
4796       if (!(csp->flags & CSP_FLAG_CRUNCHED)
4797          && (csp->server_connection.sfd != JB_INVALID_SOCKET))
4798       {
4799          if (!(csp->flags & CSP_FLAG_SERVER_KEEP_ALIVE_TIMEOUT_SET))
4800          {
4801             csp->server_connection.keep_alive_timeout = csp->config->default_server_timeout;
4802          }
4803          if (!(csp->flags & CSP_FLAG_SERVER_CONNECTION_KEEP_ALIVE)
4804             || (csp->flags & CSP_FLAG_SERVER_SOCKET_TAINTED)
4805             || !socket_is_still_alive(csp->server_connection.sfd)
4806             || !(latency < csp->server_connection.keep_alive_timeout))
4807          {
4808             log_error(LOG_LEVEL_CONNECT,
4809                "Closing server socket %d connected to %s. "
4810                "Keep-alive: %u. Tainted: %u. Socket alive: %u. Timeout: %u.",
4811                csp->server_connection.sfd, csp->server_connection.host,
4812                0 != (csp->flags & CSP_FLAG_SERVER_CONNECTION_KEEP_ALIVE),
4813                0 != (csp->flags & CSP_FLAG_SERVER_SOCKET_TAINTED),
4814                socket_is_still_alive(csp->server_connection.sfd),
4815                csp->server_connection.keep_alive_timeout);
4816 #ifdef FEATURE_CONNECTION_SHARING
4817             if (csp->config->feature_flags & RUNTIME_FEATURE_CONNECTION_SHARING)
4818             {
4819                forget_connection(csp->server_connection.sfd);
4820             }
4821 #endif /* def FEATURE_CONNECTION_SHARING */
4822 #ifdef FEATURE_HTTPS_INSPECTION
4823             close_server_ssl_connection(csp);
4824 #endif
4825             close_socket(csp->server_connection.sfd);
4826             mark_connection_closed(&csp->server_connection);
4827          }
4828       }
4829
4830       if (continue_chatting && any_loaded_file_changed(csp))
4831       {
4832          continue_chatting = 0;
4833          config_file_change_detected = 1;
4834       }
4835 #ifdef FEATURE_HTTPS_INSPECTION
4836       if (continue_chatting && client_use_ssl(csp) &&
4837          csp->ssl_with_client_is_opened == 0)
4838       {
4839          continue_chatting = 0;
4840          log_error(LOG_LEVEL_CONNECT, "Client socket %d is no longer usable. "
4841             "The TLS session has been terminated.", csp->cfd);
4842       }
4843 #endif
4844
4845       if (continue_chatting)
4846       {
4847          if (((csp->flags & CSP_FLAG_PIPELINED_REQUEST_WAITING) != 0)
4848             && socket_is_still_alive(csp->cfd))
4849          {
4850             log_error(LOG_LEVEL_CONNECT, "Client request %d has been "
4851                "pipelined on socket %d and the socket is still alive.",
4852                csp->requests_received_total+1, csp->cfd);
4853             prepare_csp_for_next_request(csp);
4854             continue;
4855          }
4856
4857          if (0 != (csp->flags & CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE))
4858          {
4859             if (csp->server_connection.sfd != JB_INVALID_SOCKET)
4860             {
4861                log_error(LOG_LEVEL_CONNECT,
4862                   "Waiting for the next client request on socket %d. "
4863                   "Keeping the server socket %d to %s open.",
4864                   csp->cfd, csp->server_connection.sfd, csp->server_connection.host);
4865             }
4866             else
4867             {
4868                log_error(LOG_LEVEL_CONNECT,
4869                   "Waiting for the next client request on socket %d. "
4870                   "No server socket to keep open.", csp->cfd);
4871             }
4872          }
4873
4874          if ((csp->flags & CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE)
4875             && data_is_available(csp->cfd, (int)csp->config->keep_alive_timeout)
4876             && socket_is_still_alive(csp->cfd))
4877          {
4878             log_error(LOG_LEVEL_CONNECT,
4879                "Data arrived in time on client socket %d. Requests so far: %u",
4880                csp->cfd, csp->requests_received_total);
4881             prepare_csp_for_next_request(csp);
4882          }
4883          else
4884          {
4885 #ifdef FEATURE_CONNECTION_SHARING
4886             if ((csp->config->feature_flags & RUNTIME_FEATURE_CONNECTION_SHARING)
4887                && (csp->server_connection.sfd != JB_INVALID_SOCKET)
4888                && (socket_is_still_alive(csp->server_connection.sfd))
4889 #ifdef FEATURE_HTTPS_INSPECTION
4890                && !server_use_ssl(csp)
4891 #endif
4892                 )
4893             {
4894                time_t time_open = time(NULL) - csp->server_connection.timestamp;
4895
4896                if (csp->server_connection.keep_alive_timeout < time_open - (time_t)latency)
4897                {
4898                   break;
4899                }
4900
4901                remember_connection(&csp->server_connection);
4902                csp->server_connection.sfd = JB_INVALID_SOCKET;
4903                drain_and_close_socket(csp->cfd);
4904                csp->cfd = JB_INVALID_SOCKET;
4905                privoxy_mutex_lock(&connection_reuse_mutex);
4906                if (!monitor_thread_running)
4907                {
4908                   monitor_thread_running = 1;
4909                   privoxy_mutex_unlock(&connection_reuse_mutex);
4910                   wait_for_alive_connections();
4911                   privoxy_mutex_lock(&connection_reuse_mutex);
4912                   monitor_thread_running = 0;
4913                }
4914                privoxy_mutex_unlock(&connection_reuse_mutex);
4915             }
4916 #endif /* def FEATURE_CONNECTION_SHARING */
4917             break;
4918          }
4919       }
4920       else if (csp->server_connection.sfd != JB_INVALID_SOCKET)
4921       {
4922          log_error(LOG_LEVEL_CONNECT,
4923             "Closing server socket %d connected to %s. Keep-alive: %u. "
4924             "Tainted: %u. Socket alive: %u. Timeout: %u. "
4925             "Configuration file change detected: %u",
4926             csp->server_connection.sfd, csp->server_connection.host,
4927             0 != (csp->flags & CSP_FLAG_SERVER_CONNECTION_KEEP_ALIVE),
4928             0 != (csp->flags & CSP_FLAG_SERVER_SOCKET_TAINTED),
4929             socket_is_still_alive(csp->server_connection.sfd),
4930             csp->server_connection.keep_alive_timeout,
4931             config_file_change_detected);
4932       }
4933    } while (continue_chatting);
4934
4935 #else
4936    chat(csp);
4937 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
4938
4939    if (csp->cfd != JB_INVALID_SOCKET)
4940    {
4941       log_error(LOG_LEVEL_CONNECT, "Closing client socket %d. "
4942          "Keep-alive: %u. Socket alive: %u. Data available: %u. "
4943          "Configuration file change detected: %u. Requests received: %u.",
4944          csp->cfd, 0 != (csp->flags & CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE),
4945          socket_is_still_alive(csp->cfd), data_is_available(csp->cfd, 0),
4946          config_file_change_detected, csp->requests_received_total);
4947 #ifdef FEATURE_HTTPS_INSPECTION
4948       close_client_ssl_connection(csp);
4949 #endif
4950       drain_and_close_socket(csp->cfd);
4951    }
4952
4953    if (csp->server_connection.sfd != JB_INVALID_SOCKET)
4954    {
4955 #ifdef FEATURE_CONNECTION_SHARING
4956       if (csp->config->feature_flags & RUNTIME_FEATURE_CONNECTION_SHARING)
4957       {
4958          forget_connection(csp->server_connection.sfd);
4959       }
4960 #endif /* def FEATURE_CONNECTION_SHARING */
4961
4962 #ifdef FEATURE_HTTPS_INSPECTION
4963       close_server_ssl_connection(csp);
4964 #endif /* def FEATURE_HTTPS_INSPECTION */
4965
4966       close_socket(csp->server_connection.sfd);
4967    }
4968
4969 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
4970    mark_connection_closed(&csp->server_connection);
4971 #endif
4972
4973    free_csp_resources(csp);
4974
4975    csp->flags &= ~CSP_FLAG_ACTIVE;
4976
4977 }
4978
4979
4980 #ifdef __BEOS__
4981 /*********************************************************************
4982  *
4983  * Function    :  server_thread
4984  *
4985  * Description :  We only exist to call `serve' in a threaded environment.
4986  *
4987  * Parameters  :
4988  *          1  :  data = Current client state (buffers, headers, etc...)
4989  *
4990  * Returns     :  Always 0.
4991  *
4992  *********************************************************************/
4993 static int32 server_thread(void *data)
4994 {
4995    serve((struct client_state *) data);
4996    return 0;
4997
4998 }
4999 #endif
5000
5001
5002 #if !defined(_WIN32) || defined(_WIN_CONSOLE)
5003 /*********************************************************************
5004  *
5005  * Function    :  usage
5006  *
5007  * Description :  Print usage info & exit.
5008  *
5009  * Parameters  :  Pointer to argv[0] for identifying ourselves
5010  *
5011  * Returns     :  No. ,-)
5012  *
5013  *********************************************************************/
5014 static void usage(const char *name)
5015 {
5016    printf("Privoxy version " VERSION " (" HOME_PAGE_URL ")\n"
5017           "Usage: %s [--config-test] "
5018 #if defined(unix)
5019           "[--chroot] "
5020 #endif /* defined(unix) */
5021           "[--help] "
5022 #if defined(unix)
5023           "[--no-daemon] [--pidfile pidfile] [--pre-chroot-nslookup hostname] [--user user[.group]] "
5024 #endif /* defined(unix) */
5025          "[--version] [configfile]\n",
5026           name);
5027
5028 #ifdef FUZZ
5029    show_fuzz_usage(name);
5030 #endif
5031
5032    printf("Aborting\n");
5033
5034    exit(2);
5035
5036 }
5037 #endif /* #if !defined(_WIN32) || defined(_WIN_CONSOLE) */
5038
5039
5040 #ifdef MUTEX_LOCKS_AVAILABLE
5041 /*********************************************************************
5042  *
5043  * Function    :  privoxy_mutex_lock
5044  *
5045  * Description :  Locks a mutex.
5046  *
5047  * Parameters  :
5048  *          1  :  mutex = The mutex to lock.
5049  *
5050  * Returns     :  Void. May exit in case of errors.
5051  *
5052  *********************************************************************/
5053 void privoxy_mutex_lock(privoxy_mutex_t *mutex)
5054 {
5055 #ifdef FEATURE_PTHREAD
5056    int err = pthread_mutex_lock(mutex);
5057    if (err)
5058    {
5059       if (mutex != &log_mutex)
5060       {
5061          log_error(LOG_LEVEL_FATAL,
5062             "Mutex locking failed: %s.\n", strerror(err));
5063       }
5064       exit(1);
5065    }
5066 #else
5067    EnterCriticalSection(mutex);
5068 #endif /* def FEATURE_PTHREAD */
5069 }
5070
5071
5072 /*********************************************************************
5073  *
5074  * Function    :  privoxy_mutex_unlock
5075  *
5076  * Description :  Unlocks a mutex.
5077  *
5078  * Parameters  :
5079  *          1  :  mutex = The mutex to unlock.
5080  *
5081  * Returns     :  Void. May exit in case of errors.
5082  *
5083  *********************************************************************/
5084 void privoxy_mutex_unlock(privoxy_mutex_t *mutex)
5085 {
5086 #ifdef FEATURE_PTHREAD
5087    int err = pthread_mutex_unlock(mutex);
5088    if (err)
5089    {
5090       if (mutex != &log_mutex)
5091       {
5092          log_error(LOG_LEVEL_FATAL,
5093             "Mutex unlocking failed: %s.\n", strerror(err));
5094       }
5095       exit(1);
5096    }
5097 #else
5098    LeaveCriticalSection(mutex);
5099 #endif /* def FEATURE_PTHREAD */
5100 }
5101
5102
5103 /*********************************************************************
5104  *
5105  * Function    :  privoxy_mutex_init
5106  *
5107  * Description :  Prepares a mutex.
5108  *
5109  * Parameters  :
5110  *          1  :  mutex = The mutex to initialize.
5111  *
5112  * Returns     :  Void. May exit in case of errors.
5113  *
5114  *********************************************************************/
5115 static void privoxy_mutex_init(privoxy_mutex_t *mutex)
5116 {
5117 #ifdef FEATURE_PTHREAD
5118    int err = pthread_mutex_init(mutex, 0);
5119    if (err)
5120    {
5121       printf("Fatal error. Mutex initialization failed: %s.\n",
5122          strerror(err));
5123       exit(1);
5124    }
5125 #else
5126    InitializeCriticalSection(mutex);
5127 #endif /* def FEATURE_PTHREAD */
5128 }
5129 #endif /* def MUTEX_LOCKS_AVAILABLE */
5130
5131 /*********************************************************************
5132  *
5133  * Function    :  initialize_mutexes
5134  *
5135  * Description :  Prepares mutexes if mutex support is available.
5136  *
5137  * Parameters  :  None
5138  *
5139  * Returns     :  Void, exits in case of errors.
5140  *
5141  *********************************************************************/
5142 static void initialize_mutexes(void)
5143 {
5144 #ifdef MUTEX_LOCKS_AVAILABLE
5145    /*
5146     * Prepare global mutex semaphores
5147     */
5148
5149 #ifdef FEATURE_HTTPS_INSPECTION
5150    privoxy_mutex_init(&certificate_mutex);
5151    privoxy_mutex_init(&ssl_init_mutex);
5152 #endif
5153
5154    privoxy_mutex_init(&log_mutex);
5155    privoxy_mutex_init(&log_init_mutex);
5156    privoxy_mutex_init(&connection_reuse_mutex);
5157 #ifdef FEATURE_EXTERNAL_FILTERS
5158    privoxy_mutex_init(&external_filter_mutex);
5159 #endif
5160 #ifdef FEATURE_CLIENT_TAGS
5161    privoxy_mutex_init(&client_tags_mutex);
5162 #endif
5163 #ifdef FEATURE_EXTENDED_STATISTICS
5164    privoxy_mutex_init(&filter_statistics_mutex);
5165    privoxy_mutex_init(&block_statistics_mutex);
5166 #endif
5167
5168    /*
5169     * XXX: The assumptions below are a bit naive
5170     * and can cause locks that aren't necessary.
5171     *
5172     * For example older FreeBSD versions (< 6.x?)
5173     * have no gethostbyname_r, but gethostbyname is
5174     * thread safe.
5175     */
5176 #if !defined(HAVE_GETHOSTBYADDR_R) || !defined(HAVE_GETHOSTBYNAME_R)
5177    privoxy_mutex_init(&resolver_mutex);
5178 #endif /* !defined(HAVE_GETHOSTBYADDR_R) || !defined(HAVE_GETHOSTBYNAME_R) */
5179    /*
5180     * XXX: should we use a single mutex for
5181     * localtime() and gmtime() as well?
5182     */
5183 #ifndef HAVE_GMTIME_R
5184    privoxy_mutex_init(&gmtime_mutex);
5185 #endif /* ndef HAVE_GMTIME_R */
5186
5187 #ifndef HAVE_LOCALTIME_R
5188    privoxy_mutex_init(&localtime_mutex);
5189 #endif /* ndef HAVE_GMTIME_R */
5190
5191 #if !defined(HAVE_ARC4RANDOM) && !defined(HAVE_RANDOM)
5192    privoxy_mutex_init(&rand_mutex);
5193 #endif /* !defined(HAVE_ARC4RANDOM) && !defined(HAVE_RANDOM) */
5194
5195 #endif /* def MUTEX_LOCKS_AVAILABLE */
5196 }
5197
5198 /*********************************************************************
5199  *
5200  * Function    :  main
5201  *
5202  * Description :  Load the config file and start the listen loop.
5203  *                This function is a lot more *sane* with the `load_config'
5204  *                and `listen_loop' functions; although it stills does
5205  *                a *little* too much for my taste.
5206  *
5207  * Parameters  :
5208  *          1  :  argc = Number of parameters (including $0).
5209  *          2  :  argv = Array of (char *)'s to the parameters.
5210  *
5211  * Returns     :  1 if : can't open config file, unrecognized directive,
5212  *                stats requested in multi-thread mode, can't open the
5213  *                log file, can't open the jar file, listen port is invalid,
5214  *                any load fails, and can't bind port.
5215  *
5216  *                Else main never returns, the process must be signaled
5217  *                to terminate execution.  Or, on Windows, use the
5218  *                "File", "Exit" menu option.
5219  *
5220  *********************************************************************/
5221 #ifdef __MINGW32__
5222 int real_main(int argc, char **argv)
5223 #else
5224 int main(int argc, char **argv)
5225 #endif
5226 {
5227    int argc_pos = 0;
5228    int do_config_test = 0;
5229 #ifndef HAVE_ARC4RANDOM
5230    unsigned int random_seed;
5231 #endif
5232 #ifdef unix
5233    struct passwd *pw = NULL;
5234    struct group *grp = NULL;
5235    int do_chroot = 0;
5236    char *pre_chroot_nslookup_to_load_resolver = NULL;
5237 #endif
5238 #ifdef FUZZ
5239    char *fuzz_input_type = NULL;
5240    char *fuzz_input_file = NULL;
5241 #endif
5242
5243    Argc = argc;
5244    Argv = argv;
5245
5246    configfile =
5247 #if !defined(_WIN32)
5248    "config"
5249 #else
5250    "config.txt"
5251 #endif
5252       ;
5253
5254    /* Prepare mutexes if supported and necessary. */
5255    initialize_mutexes();
5256
5257    /* Enable logging until further notice. */
5258    init_log_module();
5259
5260    /*
5261     * Parse the command line arguments
5262     *
5263     * XXX: simply printing usage information in case of
5264     * invalid arguments isn't particularly user friendly.
5265     */
5266    while (++argc_pos < argc)
5267    {
5268 #ifdef _WIN32
5269       /* Check to see if the service must be installed or uninstalled */
5270       if (strncmp(argv[argc_pos], "--install", 9) == 0)
5271       {
5272          const char *pName = argv[argc_pos] + 9;
5273          if (*pName == ':')
5274             pName++;
5275          exit((install_service(pName)) ? 0 : 1);
5276       }
5277       else if (strncmp(argv[argc_pos], "--uninstall", 11) == 0)
5278       {
5279          const char *pName = argv[argc_pos] + 11;
5280          if (*pName == ':')
5281             pName++;
5282          exit((uninstall_service(pName)) ? 0 : 1);
5283       }
5284       else if (strcmp(argv[argc_pos], "--service") == 0)
5285       {
5286          bRunAsService = TRUE;
5287          w32_set_service_cwd();
5288          atexit(w32_service_exit_notify);
5289       }
5290       else
5291 #endif /* defined(_WIN32) */
5292
5293
5294 #if !defined(_WIN32) || defined(_WIN_CONSOLE)
5295
5296       if (strcmp(argv[argc_pos], "--help") == 0)
5297       {
5298          usage(argv[0]);
5299       }
5300
5301       else if (strcmp(argv[argc_pos], "--version") == 0)
5302       {
5303          printf("Privoxy version " VERSION " (" HOME_PAGE_URL ")\n");
5304          exit(0);
5305       }
5306
5307 #if defined(unix)
5308
5309       else if (strcmp(argv[argc_pos], "--no-daemon") == 0)
5310       {
5311          set_debug_level(LOG_LEVEL_FATAL | LOG_LEVEL_ERROR | LOG_LEVEL_INFO);
5312          daemon_mode = 0;
5313       }
5314
5315       else if (strcmp(argv[argc_pos], "--pidfile") == 0)
5316       {
5317          if (++argc_pos == argc) usage(argv[0]);
5318          pidfile = strdup_or_die(argv[argc_pos]);
5319       }
5320
5321       else if (strcmp(argv[argc_pos], "--user") == 0)
5322       {
5323          char *user_arg;
5324          char *group_name;
5325
5326          if (++argc_pos == argc) usage(argv[argc_pos]);
5327
5328          user_arg = strdup_or_die(argv[argc_pos]);
5329          group_name = strchr(user_arg, '.');
5330          if (NULL != group_name)
5331          {
5332             /* Nul-terminate the user name */
5333             *group_name = '\0';
5334
5335             /* Skip the former delimiter to actually reach the group name */
5336             group_name++;
5337
5338             grp = getgrnam(group_name);
5339             if (NULL == grp)
5340             {
5341                log_error(LOG_LEVEL_FATAL, "Group '%s' not found.", group_name);
5342             }
5343          }
5344          pw = getpwnam(user_arg);
5345          if (NULL == pw)
5346          {
5347             log_error(LOG_LEVEL_FATAL, "User '%s' not found.", user_arg);
5348          }
5349
5350          freez(user_arg);
5351       }
5352
5353       else if (strcmp(argv[argc_pos], "--pre-chroot-nslookup") == 0)
5354       {
5355          if (++argc_pos == argc) usage(argv[0]);
5356          pre_chroot_nslookup_to_load_resolver = strdup_or_die(argv[argc_pos]);
5357       }
5358
5359       else if (strcmp(argv[argc_pos], "--chroot") == 0)
5360       {
5361          do_chroot = 1;
5362       }
5363 #endif /* defined(unix) */
5364
5365       else if (strcmp(argv[argc_pos], "--config-test") == 0)
5366       {
5367          do_config_test = 1;
5368       }
5369 #ifdef FUZZ
5370       else if (strcmp(argv[argc_pos], "--fuzz") == 0)
5371       {
5372          argc_pos++;
5373          if (argc < argc_pos + 2) usage(argv[0]);
5374          fuzz_input_type = argv[argc_pos];
5375          argc_pos++;
5376          fuzz_input_file = argv[argc_pos];
5377       }
5378       else if (strcmp(argv[argc_pos], "--stfu") == 0)
5379       {
5380          set_debug_level(LOG_LEVEL_STFU);
5381       }
5382 #endif
5383       else if (argc_pos + 1 != argc)
5384       {
5385          /*
5386           * This is neither the last command line
5387           * option, nor was it recognized before,
5388           * therefore it must be invalid.
5389           */
5390          usage(argv[0]);
5391       }
5392       else
5393
5394 #endif /* defined(_WIN32) && !defined(_WIN_CONSOLE) */
5395       {
5396          configfile = argv[argc_pos];
5397       }
5398
5399    } /* -END- while (more arguments) */
5400
5401    show_version(Argv[0]);
5402
5403 #if defined(unix)
5404    if (*configfile != '/')
5405    {
5406       char cwd[BUFFER_SIZE];
5407       char *abs_file;
5408       size_t abs_file_size;
5409
5410       /* make config-filename absolute here */
5411       if (NULL == getcwd(cwd, sizeof(cwd)))
5412       {
5413          perror("failed to get current working directory");
5414          exit(1);
5415       }
5416
5417       basedir = strdup_or_die(cwd);
5418       /* XXX: why + 5? */
5419       abs_file_size = strlen(cwd) + strlen(configfile) + 5;
5420       abs_file = malloc_or_die(abs_file_size);
5421       strlcpy(abs_file, basedir, abs_file_size);
5422       strlcat(abs_file, "/", abs_file_size);
5423       strlcat(abs_file, configfile, abs_file_size);
5424       configfile = abs_file;
5425    }
5426 #endif /* defined unix */
5427
5428
5429    files->next = NULL;
5430    clients->next = NULL;
5431
5432    /* XXX: factor out initialising after the next stable release. */
5433 #ifdef _WIN32
5434    InitWin32();
5435 #endif
5436
5437 #ifndef HAVE_ARC4RANDOM
5438    random_seed = (unsigned int)time(NULL);
5439 #ifdef HAVE_RANDOM
5440    srandom(random_seed);
5441 #else
5442    srand(random_seed);
5443 #endif /* ifdef HAVE_RANDOM */
5444 #endif /* ifndef HAVE_ARC4RANDOM */
5445
5446    /*
5447     * Unix signal handling
5448     *
5449     * Catch the abort, interrupt and terminate signals for a graceful exit
5450     * Catch the hangup signal so the errlog can be reopened.
5451     *
5452     * Ignore the broken pipe signal as connection failures
5453     * are handled when and where they occur without relying
5454     * on a signal.
5455     */
5456 #if !defined(_WIN32)
5457 {
5458    int idx;
5459    const int catched_signals[] = { SIGTERM, SIGINT, SIGHUP };
5460
5461    for (idx = 0; idx < SZ(catched_signals); idx++)
5462    {
5463 #ifdef sun /* FIXME: Is it safe to check for HAVE_SIGSET instead? */
5464       if (sigset(catched_signals[idx], sig_handler) == SIG_ERR)
5465 #else
5466       if (signal(catched_signals[idx], sig_handler) == SIG_ERR)
5467 #endif /* ifdef sun */
5468       {
5469          log_error(LOG_LEVEL_FATAL, "Can't set signal-handler for signal %d: %E", catched_signals[idx]);
5470       }
5471    }
5472
5473    if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
5474    {
5475       log_error(LOG_LEVEL_FATAL, "Can't set ignore-handler for SIGPIPE: %E");
5476    }
5477
5478 }
5479 #else /* ifdef _WIN32 */
5480 # ifdef _WIN_CONSOLE
5481    /*
5482     * We *are* in a windows console app.
5483     * Print a verbose messages about FAQ's and such
5484     */
5485    printf("%s", win32_blurb);
5486 # endif /* def _WIN_CONSOLE */
5487 #endif /* def _WIN32 */
5488
5489 #ifdef FUZZ
5490    if (fuzz_input_type != NULL)
5491    {
5492       exit(process_fuzzed_input(fuzz_input_type, fuzz_input_file));
5493    }
5494    log_error(LOG_LEVEL_FATAL,
5495       "When compiled with fuzzing support, Privoxy should only be used for fuzzing. "
5496       "Various data structures are static which is unsafe when using threads.");
5497 #endif
5498
5499    if (do_config_test)
5500    {
5501       exit(NULL == load_config());
5502    }
5503
5504    /* Initialize the CGI subsystem */
5505    cgi_init_error_messages();
5506
5507    /*
5508     * If running on unix and without the --no-daemon
5509     * option, become a daemon. I.e. fork, detach
5510     * from tty and get process group leadership
5511     */
5512 #if defined(unix)
5513 {
5514    if (daemon_mode)
5515    {
5516       int fd;
5517       pid_t pid = fork();
5518
5519       if (pid < 0) /* error */
5520       {
5521          perror("fork");
5522          exit(3);
5523       }
5524       else if (pid != 0) /* parent */
5525       {
5526          int status;
5527          pid_t wpid;
5528          /*
5529           * must check for errors
5530           * child died due to missing files aso
5531           */
5532          sleep(1);
5533          wpid = waitpid(pid, &status, WNOHANG);
5534          if (wpid != 0)
5535          {
5536             exit(1);
5537          }
5538          exit(0);
5539       }
5540       /* child */
5541
5542       setsid();
5543
5544       /*
5545        * stderr (fd 2) will be closed later on,
5546        * when the config file has been parsed.
5547        */
5548       close(0);
5549       close(1);
5550
5551       /*
5552        * Reserve fd 0 and 1 to prevent abort() and friends
5553        * from sending stuff to the clients or servers.
5554        */
5555       fd = open("/dev/null", O_RDONLY);
5556       if (fd == -1)
5557       {
5558          log_error(LOG_LEVEL_FATAL, "Failed to open /dev/null: %E");
5559       }
5560       else if (fd != 0)
5561       {
5562          if (dup2(fd, 0) == -1)
5563          {
5564             log_error(LOG_LEVEL_FATAL, "Failed to reserve fd 0: %E");
5565          }
5566          close(fd);
5567       }
5568       fd = open("/dev/null", O_WRONLY);
5569       if (fd == -1)
5570       {
5571          log_error(LOG_LEVEL_FATAL, "Failed to open /dev/null: %E");
5572       }
5573       else if (fd != 1)
5574       {
5575          if (dup2(fd, 1) == -1)
5576          {
5577             log_error(LOG_LEVEL_FATAL, "Failed to reserve fd 1: %E");
5578          }
5579          close(fd);
5580       }
5581
5582 #ifdef FEATURE_EXTERNAL_FILTERS
5583       for (fd = 0; fd < 3; fd++)
5584       {
5585          mark_socket_for_close_on_execute(fd);
5586       }
5587 #endif
5588
5589       if (chdir("/") != 0)
5590       {
5591          log_error(LOG_LEVEL_FATAL, "Failed to cd into '/': %E");
5592       }
5593
5594    } /* -END- if (daemon_mode) */
5595
5596    /*
5597     * As soon as we have written the PID file, we can switch
5598     * to the user and group ID indicated by the --user option
5599     */
5600    if (pidfile != NULL)
5601    {
5602       write_pid_file(pidfile);
5603    }
5604    if (NULL != pw)
5605    {
5606       if (setgid((NULL != grp) ? grp->gr_gid : pw->pw_gid))
5607       {
5608          log_error(LOG_LEVEL_FATAL, "Cannot setgid(): Insufficient permissions.");
5609       }
5610       if (NULL != grp)
5611       {
5612          if (setgroups(1, &grp->gr_gid))
5613          {
5614             log_error(LOG_LEVEL_FATAL, "setgroups() failed: %E");
5615          }
5616       }
5617       else if (initgroups(pw->pw_name, pw->pw_gid))
5618       {
5619          log_error(LOG_LEVEL_FATAL, "initgroups() failed: %E");
5620       }
5621       if (do_chroot)
5622       {
5623          if (!pw->pw_dir)
5624          {
5625             log_error(LOG_LEVEL_FATAL, "Home directory for %s undefined", pw->pw_name);
5626          }
5627          /* Read the time zone file from /etc before doing chroot. */
5628          tzset();
5629          if (NULL != pre_chroot_nslookup_to_load_resolver
5630              && '\0' != pre_chroot_nslookup_to_load_resolver[0])
5631          {
5632             /* Initialize resolver library. */
5633             (void) resolve_hostname_to_ip(pre_chroot_nslookup_to_load_resolver);
5634          }
5635          if (chroot(pw->pw_dir) < 0)
5636          {
5637             log_error(LOG_LEVEL_FATAL, "Cannot chroot to %s", pw->pw_dir);
5638          }
5639          if (chdir ("/"))
5640          {
5641             log_error(LOG_LEVEL_FATAL, "Cannot chdir /");
5642          }
5643       }
5644       if (setuid(pw->pw_uid))
5645       {
5646          log_error(LOG_LEVEL_FATAL, "Cannot setuid(): Insufficient permissions.");
5647       }
5648       if (do_chroot)
5649       {
5650          char putenv_dummy[64];
5651
5652          strlcpy(putenv_dummy, "HOME=/", sizeof(putenv_dummy));
5653          if (putenv(putenv_dummy) != 0)
5654          {
5655             log_error(LOG_LEVEL_FATAL, "Cannot putenv(): HOME");
5656          }
5657
5658          snprintf(putenv_dummy, sizeof(putenv_dummy), "USER=%s", pw->pw_name);
5659          if (putenv(putenv_dummy) != 0)
5660          {
5661             log_error(LOG_LEVEL_FATAL, "Cannot putenv(): USER");
5662          }
5663       }
5664    }
5665    else if (do_chroot)
5666    {
5667       log_error(LOG_LEVEL_FATAL, "Cannot chroot without --user argument.");
5668    }
5669 }
5670 #endif /* defined unix */
5671
5672 #ifdef _WIN32
5673    /* This will be FALSE unless the command line specified --service
5674     */
5675    if (bRunAsService)
5676    {
5677       /* Yup, so now we must attempt to establish a connection
5678        * with the service dispatcher. This will only work if this
5679        * process was launched by the service control manager to
5680        * actually run as a service. If this isn't the case, i've
5681        * known it take around 30 seconds or so for the call to return.
5682        */
5683
5684       /* The StartServiceCtrlDispatcher won't return until the service is stopping */
5685       if (w32_start_service_ctrl_dispatcher(w32ServiceDispatchTable))
5686       {
5687          /* Service has run, and at this point is now being stopped, so just return */
5688          return 0;
5689       }
5690
5691 #ifdef _WIN_CONSOLE
5692       printf("Warning: Failed to connect to Service Control Dispatcher\nwhen starting as a service!\n");
5693 #endif
5694       /* An error occurred. Usually it's because --service was wrongly specified
5695        * and we were unable to connect to the Service Control Dispatcher because
5696        * it wasn't expecting us and is therefore not listening.
5697        *
5698        * For now, just continue below to call the listen_loop function.
5699        */
5700    }
5701 #endif /* def _WIN32 */
5702
5703    listen_loop();
5704
5705    /* NOTREACHED */
5706    return(-1);
5707
5708 }
5709
5710
5711 /*********************************************************************
5712  *
5713  * Function    :  bind_port_helper
5714  *
5715  * Description :  Bind the listen port.  Handles logging, and aborts
5716  *                on failure.
5717  *
5718  * Parameters  :
5719  *          1  :  haddr = Host address to bind to. Use NULL to bind to
5720  *                        INADDR_ANY.
5721  *          2  :  hport = Specifies port to bind to.
5722  *          3  :  backlog = Listen backlog.
5723  *
5724  * Returns     :  Port that was opened.
5725  *
5726  *********************************************************************/
5727 static jb_socket bind_port_helper(const char *haddr, int hport, int backlog)
5728 {
5729    int result;
5730    jb_socket bfd;
5731
5732    result = bind_port(haddr, hport, backlog, &bfd);
5733
5734    if (result < 0)
5735    {
5736       const char *bind_address = (NULL != haddr) ? haddr : "INADDR_ANY";
5737       switch(result)
5738       {
5739          case -3:
5740             log_error(LOG_LEVEL_FATAL,
5741                "can't bind to %s:%d: There may be another Privoxy "
5742                "or some other proxy running on port %d",
5743                bind_address, hport, hport);
5744
5745          case -2:
5746             log_error(LOG_LEVEL_FATAL,
5747                "can't bind to %s:%d: The hostname is not resolvable",
5748                bind_address, hport);
5749
5750          default:
5751             log_error(LOG_LEVEL_FATAL, "can't bind to %s:%d: %E",
5752                bind_address, hport);
5753       }
5754
5755       /* shouldn't get here */
5756       return JB_INVALID_SOCKET;
5757    }
5758
5759 #ifndef HAVE_POLL
5760 #ifndef _WIN32
5761    if (bfd >= FD_SETSIZE)
5762    {
5763       log_error(LOG_LEVEL_FATAL,
5764          "Bind socket number too high to use select(): %d >= %d",
5765          bfd, FD_SETSIZE);
5766    }
5767 #endif
5768 #endif
5769
5770    if (haddr == NULL)
5771    {
5772       log_error(LOG_LEVEL_INFO, "Listening on port %d on all IP addresses",
5773          hport);
5774    }
5775    else
5776    {
5777       log_error(LOG_LEVEL_INFO, "Listening on port %d on IP address %s",
5778          hport, haddr);
5779    }
5780
5781    return bfd;
5782 }
5783
5784
5785 /*********************************************************************
5786  *
5787  * Function    :  bind_ports_helper
5788  *
5789  * Description :  Bind the listen ports.  Handles logging, and aborts
5790  *                on failure.
5791  *
5792  * Parameters  :
5793  *          1  :  config = Privoxy configuration.  Specifies ports
5794  *                         to bind to.
5795  *          2  :  sockets = Preallocated array of opened sockets
5796  *                          corresponding to specification in config.
5797  *                          All non-opened sockets will be set to
5798  *                          JB_INVALID_SOCKET.
5799  *
5800  * Returns     :  Nothing. Inspect sockets argument.
5801  *
5802  *********************************************************************/
5803 static void bind_ports_helper(struct configuration_spec * config,
5804                               jb_socket sockets[])
5805 {
5806    int i;
5807
5808    for (i = 0; i < MAX_LISTENING_SOCKETS; i++)
5809    {
5810       if (config->hport[i])
5811       {
5812          sockets[i] = bind_port_helper(config->haddr[i],
5813             config->hport[i], config->listen_backlog);
5814 #if defined(FEATURE_ACCEPT_FILTER) && defined(SO_ACCEPTFILTER)
5815          if (config->enable_accept_filter && sockets[i] != JB_INVALID_SOCKET)
5816          {
5817             struct accept_filter_arg af_options;
5818             bzero(&af_options, sizeof(af_options));
5819             strlcpy(af_options.af_name, "httpready", sizeof(af_options.af_name));
5820             if (setsockopt(sockets[i], SOL_SOCKET, SO_ACCEPTFILTER, &af_options,
5821                   sizeof(af_options)))
5822             {
5823                log_error(LOG_LEVEL_ERROR,
5824                   "Enabling accept filter for socket %d failed: %E", sockets[i]);
5825             }
5826          }
5827 #endif
5828       }
5829       else
5830       {
5831          sockets[i] = JB_INVALID_SOCKET;
5832       }
5833    }
5834    config->need_bind = 0;
5835 }
5836
5837
5838 /*********************************************************************
5839  *
5840  * Function    :  close_ports_helper
5841  *
5842  * Description :  Close listenings ports.
5843  *
5844  * Parameters  :
5845  *          1  :  sockets = Array of opened and non-opened sockets to
5846  *                          close. All sockets will be set to
5847  *                          JB_INVALID_SOCKET.
5848  *
5849  * Returns     :  Nothing.
5850  *
5851  *********************************************************************/
5852 static void close_ports_helper(jb_socket sockets[])
5853 {
5854    int i;
5855
5856    for (i = 0; i < MAX_LISTENING_SOCKETS; i++)
5857    {
5858       if (JB_INVALID_SOCKET != sockets[i])
5859       {
5860          close_socket(sockets[i]);
5861       }
5862       sockets[i] = JB_INVALID_SOCKET;
5863    }
5864 }
5865
5866
5867 #ifdef _WIN32
5868 /* Without this simple workaround we get this compiler warning from _beginthread
5869  *     warning C4028: formal parameter 1 different from declaration
5870  */
5871 void w32_service_listen_loop(void *p)
5872 {
5873    listen_loop();
5874 }
5875 #endif /* def _WIN32 */
5876
5877
5878 /*********************************************************************
5879  *
5880  * Function    :  listen_loop
5881  *
5882  * Description :  bind the listen port and enter a "FOREVER" listening loop.
5883  *
5884  * Parameters  :  N/A
5885  *
5886  * Returns     :  Never.
5887  *
5888  *********************************************************************/
5889 static void listen_loop(void)
5890 {
5891    struct client_states *csp_list = NULL;
5892    struct client_state *csp = NULL;
5893    jb_socket bfds[MAX_LISTENING_SOCKETS];
5894    struct configuration_spec *config;
5895    unsigned int active_threads = 0;
5896 #if defined(FEATURE_PTHREAD)
5897    pthread_attr_t attrs;
5898
5899    pthread_attr_init(&attrs);
5900    pthread_attr_setdetachstate(&attrs, PTHREAD_CREATE_DETACHED);
5901 #endif
5902
5903    config = load_config();
5904
5905 #ifdef FEATURE_CONNECTION_SHARING
5906    /*
5907     * XXX: Should be relocated once it no
5908     * longer needs to emit log messages.
5909     */
5910    initialize_reusable_connections();
5911 #endif /* def FEATURE_CONNECTION_SHARING */
5912
5913    bind_ports_helper(config, bfds);
5914
5915 #ifdef FEATURE_GRACEFUL_TERMINATION
5916    while (!g_terminate)
5917 #else
5918    for (;;)
5919 #endif
5920    {
5921 #if !defined(FEATURE_PTHREAD) && !defined(_WIN32) && !defined(__BEOS__)
5922       while (waitpid(-1, NULL, WNOHANG) > 0)
5923       {
5924          /* zombie children */
5925       }
5926 #endif /* !defined(FEATURE_PTHREAD) && !defined(_WIN32) && !defined(__BEOS__) */
5927
5928       /*
5929        * Free data that was used by died threads
5930        */
5931       active_threads = sweep();
5932
5933 #if defined(unix)
5934       /*
5935        * Re-open the errlog after HUP signal
5936        */
5937       if (received_hup_signal)
5938       {
5939          if (NULL != config->logfile)
5940          {
5941             init_error_log(Argv[0], config->logfile);
5942          }
5943          received_hup_signal = 0;
5944       }
5945 #endif
5946
5947       csp_list = zalloc_or_die(sizeof(*csp_list));
5948       csp = &csp_list->csp;
5949
5950       log_error(LOG_LEVEL_CONNECT,
5951          "Waiting for the next client connection. Currently active threads: %u",
5952          active_threads);
5953
5954       /*
5955        * This config may be outdated, but for accept_connection()
5956        * it's fresh enough.
5957        */
5958       csp->config = config;
5959
5960       if (!accept_connection(csp, bfds))
5961       {
5962          log_error(LOG_LEVEL_CONNECT, "accept failed: %E");
5963          freez(csp_list);
5964          continue;
5965       }
5966
5967       csp->flags |= CSP_FLAG_ACTIVE;
5968       csp->server_connection.sfd = JB_INVALID_SOCKET;
5969
5970       csp->config = config = load_config();
5971
5972       if (config->need_bind)
5973       {
5974          /*
5975           * Since we were listening to the "old port", we will not see
5976           * a "listen" param change until the next request.  So, at
5977           * least 1 more request must be made for us to find the new
5978           * setting.  I am simply closing the old socket and binding the
5979           * new one.
5980           *
5981           * Which-ever is correct, we will serve 1 more page via the
5982           * old settings.  This should probably be a "show-status"
5983           * request.  This should not be a so common of an operation
5984           * that this will hurt people's feelings.
5985           */
5986
5987          close_ports_helper(bfds);
5988
5989          bind_ports_helper(config, bfds);
5990       }
5991
5992 #ifdef FEATURE_TOGGLE
5993       if (global_toggle_state)
5994 #endif /* def FEATURE_TOGGLE */
5995       {
5996          csp->flags |= CSP_FLAG_TOGGLED_ON;
5997       }
5998
5999       if (run_loader(csp))
6000       {
6001          log_error(LOG_LEVEL_FATAL, "a loader failed - must exit");
6002          /* Never get here - LOG_LEVEL_FATAL causes program exit */
6003       }
6004
6005 #ifdef FEATURE_ACL
6006       if (block_acl(NULL,csp))
6007       {
6008          log_error(LOG_LEVEL_CONNECT,
6009             "Connection from %s on %s (socket %d) dropped due to ACL",
6010             csp->ip_addr_str, csp->listen_addr_str, csp->cfd);
6011          close_socket(csp->cfd);
6012          freez(csp->ip_addr_str);
6013          freez(csp->listen_addr_str);
6014          freez(csp_list);
6015          continue;
6016       }
6017 #endif /* def FEATURE_ACL */
6018
6019       if ((0 != config->max_client_connections)
6020          && (active_threads >= config->max_client_connections))
6021       {
6022          log_error(LOG_LEVEL_CONNECT,
6023             "Rejecting connection from %s. Maximum number of connections reached.",
6024             csp->ip_addr_str);
6025          write_socket_delayed(csp->cfd, TOO_MANY_CONNECTIONS_RESPONSE,
6026             strlen(TOO_MANY_CONNECTIONS_RESPONSE), get_write_delay(csp));
6027          close_socket(csp->cfd);
6028          freez(csp->ip_addr_str);
6029          freez(csp->listen_addr_str);
6030          freez(csp_list);
6031          continue;
6032       }
6033
6034       /* add it to the list of clients */
6035       csp_list->next = clients->next;
6036       clients->next = csp_list;
6037
6038       if (config->multi_threaded)
6039       {
6040          int child_id;
6041
6042 /* this is a switch () statement in the C preprocessor - ugh */
6043 #undef SELECTED_ONE_OPTION
6044
6045 /* Use Pthreads in preference to native code */
6046 #if defined(FEATURE_PTHREAD) && !defined(SELECTED_ONE_OPTION)
6047 #define SELECTED_ONE_OPTION
6048          {
6049             pthread_t the_thread;
6050             int ret;
6051
6052             ret = pthread_create(&the_thread, &attrs,
6053                (void * (*)(void *))serve, csp);
6054             child_id = ret ? -1 : 0;
6055          }
6056 #endif
6057
6058 #if defined(_WIN32) && !defined(_CYGWIN) && !defined(SELECTED_ONE_OPTION)
6059 #define SELECTED_ONE_OPTION
6060          child_id = _beginthread(
6061             (void (*)(void *))serve,
6062             64 * 1024,
6063             csp);
6064 #endif
6065
6066 #if defined(__BEOS__) && !defined(SELECTED_ONE_OPTION)
6067 #define SELECTED_ONE_OPTION
6068          {
6069             thread_id tid = spawn_thread
6070                (server_thread, "server", B_NORMAL_PRIORITY, csp);
6071
6072             if ((tid >= 0) && (resume_thread(tid) == B_OK))
6073             {
6074                child_id = (int) tid;
6075             }
6076             else
6077             {
6078                child_id = -1;
6079             }
6080          }
6081 #endif
6082
6083 #if !defined(SELECTED_ONE_OPTION)
6084          child_id = fork();
6085
6086          /* This block is only needed when using fork().
6087           * When using threads, the server thread was
6088           * created and run by the call to _beginthread().
6089           */
6090          if (child_id == 0)   /* child */
6091          {
6092             int rc = 0;
6093 #ifdef FEATURE_TOGGLE
6094             int inherited_toggle_state = global_toggle_state;
6095 #endif /* def FEATURE_TOGGLE */
6096
6097             serve(csp);
6098
6099             /*
6100              * If we've been toggled or we've blocked the request, tell Mom
6101              */
6102
6103 #ifdef FEATURE_TOGGLE
6104             if (inherited_toggle_state != global_toggle_state)
6105             {
6106                rc |= RC_FLAG_TOGGLED;
6107             }
6108 #endif /* def FEATURE_TOGGLE */
6109
6110 #ifdef FEATURE_STATISTICS
6111             if (csp->flags & CSP_FLAG_REJECTED)
6112             {
6113                rc |= RC_FLAG_BLOCKED;
6114             }
6115 #endif /* ndef FEATURE_STATISTICS */
6116
6117             _exit(rc);
6118          }
6119          else if (child_id > 0) /* parent */
6120          {
6121             /* in a fork()'d environment, the parent's
6122              * copy of the client socket and the CSP
6123              * are not used.
6124              */
6125             int child_status;
6126 #if !defined(_WIN32) && !defined(__CYGWIN__)
6127
6128             wait(&child_status);
6129
6130             /*
6131              * Evaluate child's return code: If the child has
6132              *  - been toggled, toggle ourselves
6133              *  - blocked its request, bump up the stats counter
6134              */
6135
6136 #ifdef FEATURE_TOGGLE
6137             if (WIFEXITED(child_status) && (WEXITSTATUS(child_status) & RC_FLAG_TOGGLED))
6138             {
6139                global_toggle_state = !global_toggle_state;
6140             }
6141 #endif /* def FEATURE_TOGGLE */
6142
6143 #ifdef FEATURE_STATISTICS
6144             urls_read++;
6145             if (WIFEXITED(child_status) && (WEXITSTATUS(child_status) & RC_FLAG_BLOCKED))
6146             {
6147                urls_rejected++;
6148             }
6149 #endif /* def FEATURE_STATISTICS */
6150
6151 #endif /* !defined(_WIN32) && defined(__CYGWIN__) */
6152             close_socket(csp->cfd);
6153             csp->flags &= ~CSP_FLAG_ACTIVE;
6154          }
6155 #endif
6156
6157 #undef SELECTED_ONE_OPTION
6158 /* end of cpp switch () */
6159
6160          if (child_id < 0)
6161          {
6162             /*
6163              * Spawning the child failed, assume it's because
6164              * there are too many children running already.
6165              * XXX: If you assume ...
6166              */
6167             log_error(LOG_LEVEL_ERROR,
6168                "Unable to take any additional connections: %E. Active threads: %u",
6169                active_threads);
6170             write_socket_delayed(csp->cfd, TOO_MANY_CONNECTIONS_RESPONSE,
6171                strlen(TOO_MANY_CONNECTIONS_RESPONSE), get_write_delay(csp));
6172             close_socket(csp->cfd);
6173             csp->flags &= ~CSP_FLAG_ACTIVE;
6174          }
6175       }
6176       else
6177       {
6178          serve(csp);
6179       }
6180    }
6181
6182 #if defined(FEATURE_PTHREAD)
6183    pthread_attr_destroy(&attrs);
6184 #endif
6185
6186    /* NOTREACHED unless FEATURE_GRACEFUL_TERMINATION is defined */
6187
6188 #ifdef FEATURE_GRACEFUL_TERMINATION
6189
6190    log_error(LOG_LEVEL_INFO, "Graceful termination requested.");
6191
6192    unload_current_config_file();
6193    unload_current_actions_file();
6194    unload_current_re_filterfile();
6195 #ifdef FEATURE_TRUST
6196    unload_current_trust_file();
6197 #endif
6198
6199    if (config->multi_threaded)
6200    {
6201       int i = 60;
6202       do
6203       {
6204          sleep(1);
6205          sweep();
6206       } while ((clients->next != NULL) && (--i > 0));
6207
6208       if (i <= 0)
6209       {
6210          log_error(LOG_LEVEL_ERROR, "Graceful termination failed "
6211             "- still some live clients after 1 minute wait.");
6212       }
6213    }
6214    sweep();
6215    sweep();
6216
6217 #if defined(unix)
6218    freez(basedir);
6219 #endif
6220
6221 #ifdef FEATURE_HTTPS_INSPECTION
6222    /*
6223     * Only release TLS backed resources if there
6224     * are no active connections left.
6225     */
6226    if (clients->next == NULL)
6227    {
6228       ssl_release();
6229    }
6230 #endif
6231
6232    log_error(LOG_LEVEL_INFO, "Exiting gracefully.");
6233
6234 #if defined(_WIN32) && !defined(_WIN_CONSOLE)
6235    /* Cleanup - remove taskbar icon etc. */
6236    TermLogWindow();
6237 #endif
6238
6239    exit(0);
6240 #endif /* FEATURE_GRACEFUL_TERMINATION */
6241
6242 }
6243
6244
6245 /*
6246   Local Variables:
6247   tab-width: 3
6248   end:
6249 */