1 const char jcc_rcs[] = "$Id: jcc.c,v 1.403 2012/10/21 13:00:06 fabiankeil Exp $";
2 /*********************************************************************
4 * File : $Source: /cvsroot/ijbswa/current/jcc.c,v $
6 * Purpose : Main file. Contains main() method, main loop, and
7 * the main connection-handling function.
9 * Copyright : Written by and Copyright (C) 2001-2012 the
10 * Privoxy team. http://www.privoxy.org/
12 * Based on the Internet Junkbuster originally written
13 * by and Copyright (C) 1997 Anonymous Coders and
14 * Junkbusters Corporation. http://www.junkbusters.com
16 * This program is free software; you can redistribute it
17 * and/or modify it under the terms of the GNU General
18 * Public License as published by the Free Software
19 * Foundation; either version 2 of the License, or (at
20 * your option) any later version.
22 * This program is distributed in the hope that it will
23 * be useful, but WITHOUT ANY WARRANTY; without even the
24 * implied warranty of MERCHANTABILITY or FITNESS FOR A
25 * PARTICULAR PURPOSE. See the GNU General Public
26 * License for more details.
28 * The GNU General Public License should be included with
29 * this file. If not, you can view it at
30 * http://www.gnu.org/copyleft/gpl.html
31 * or write to the Free Software Foundation, Inc., 59
32 * Temple Place - Suite 330, Boston, MA 02111-1307, USA.
34 *********************************************************************/
40 #include <sys/types.h>
49 # ifndef FEATURE_PTHREAD
55 # endif /* ndef FEATURE_PTHREAD */
60 # endif /* ndef _WIN_CONSOLE */
61 # include "w32svrapi.h"
63 #else /* ifndef _WIN32 */
65 # if !defined (__OS2__)
67 # include <sys/wait.h>
68 # endif /* ndef __OS2__ */
69 # include <sys/time.h>
70 # include <sys/stat.h>
71 # include <sys/ioctl.h>
74 #include <sys/termios.h>
85 # include <socket.h> /* BeOS has select() for sockets only. */
86 # include <OS.h> /* declarations for threads and stuff. */
89 # if defined(__EMX__) || defined(__OS2__)
90 # include <sys/select.h> /* OS/2/EMX needs a little help with select */
95 #define bzero(B,N) memset(B,0x00,n)
110 #include "miscutil.h"
112 #include "jbsockets.h"
117 #include "urlmatch.h"
119 const char jcc_h_rcs[] = JCC_H_VERSION;
120 const char project_h_rcs[] = PROJECT_H_VERSION;
123 struct client_states clients[1];
124 struct file_list files[1];
126 #ifdef FEATURE_STATISTICS
127 int urls_read = 0; /* total nr of urls read inc rejected */
128 int urls_rejected = 0; /* total nr of urls rejected */
129 #endif /* def FEATURE_STATISTICS */
131 #ifdef FEATURE_GRACEFUL_TERMINATION
135 #if !defined(_WIN32) && !defined(__OS2__) && !defined(AMIGA)
136 static void sig_handler(int the_signal);
138 static int client_protocol_is_unsupported(const struct client_state *csp, char *req);
139 static jb_err get_request_destination_elsewhere(struct client_state *csp, struct list *headers);
140 static jb_err get_server_headers(struct client_state *csp);
141 static const char *crunch_reason(const struct http_response *rsp);
142 static void send_crunch_response(const struct client_state *csp, struct http_response *rsp);
143 static char *get_request_line(struct client_state *csp);
144 static jb_err receive_client_request(struct client_state *csp);
145 static jb_err parse_client_request(struct client_state *csp);
146 static void build_request_line(struct client_state *csp, const struct forward_spec *fwd, char **request_line);
147 static jb_err change_request_destination(struct client_state *csp);
148 static void chat(struct client_state *csp);
149 static void serve(struct client_state *csp);
150 #if !defined(_WIN32) || defined(_WIN_CONSOLE)
151 static void usage(const char *myname);
153 static void initialize_mutexes(void);
154 static jb_socket bind_port_helper(const char *haddr, int hport);
155 static void bind_ports_helper(struct configuration_spec *config, jb_socket sockets[]);
156 static void close_ports_helper(jb_socket sockets[]);
157 static void listen_loop(void);
160 void serve(struct client_state *csp);
161 #else /* ifndef AMIGA */
162 static void serve(struct client_state *csp);
163 #endif /* def AMIGA */
166 static int32 server_thread(void *data);
167 #endif /* def __BEOS__ */
170 #define sleep(N) Sleep(((N) * 1000))
174 #define sleep(N) DosSleep(((N) * 100))
177 #ifdef MUTEX_LOCKS_AVAILABLE
179 * XXX: Does the locking stuff really belong in this file?
181 privoxy_mutex_t log_mutex;
182 privoxy_mutex_t log_init_mutex;
183 privoxy_mutex_t connection_reuse_mutex;
185 #if !defined(HAVE_GETHOSTBYADDR_R) || !defined(HAVE_GETHOSTBYNAME_R)
186 privoxy_mutex_t resolver_mutex;
187 #endif /* !defined(HAVE_GETHOSTBYADDR_R) || !defined(HAVE_GETHOSTBYNAME_R) */
189 #ifndef HAVE_GMTIME_R
190 privoxy_mutex_t gmtime_mutex;
191 #endif /* ndef HAVE_GMTIME_R */
193 #ifndef HAVE_LOCALTIME_R
194 privoxy_mutex_t localtime_mutex;
195 #endif /* ndef HAVE_GMTIME_R */
198 privoxy_mutex_t rand_mutex;
199 #endif /* ndef HAVE_RANDOM */
201 #endif /* def MUTEX_LOCKS_AVAILABLE */
204 const char *basedir = NULL;
205 const char *pidfile = NULL;
206 static int received_hup_signal = 0;
207 #endif /* defined unix */
209 /* HTTP snipplets. */
210 static const char CSUCCEED[] =
211 "HTTP/1.1 200 Connection established\r\n"
212 "Proxy-Agent: Privoxy/" VERSION "\r\n\r\n";
214 static const char CHEADER[] =
215 "HTTP/1.1 400 Invalid header received from client\r\n"
216 "Proxy-Agent: Privoxy " VERSION "\r\n"
217 "Content-Type: text/plain\r\n"
218 "Connection: close\r\n\r\n"
219 "Invalid header received from client.\r\n";
221 static const char FTP_RESPONSE[] =
222 "HTTP/1.1 400 Invalid request received from client\r\n"
223 "Content-Type: text/plain\r\n"
224 "Connection: close\r\n\r\n"
225 "Invalid request. Privoxy doesn't support FTP.\r\n";
227 static const char GOPHER_RESPONSE[] =
228 "HTTP/1.1 400 Invalid request received from client\r\n"
229 "Content-Type: text/plain\r\n"
230 "Connection: close\r\n\r\n"
231 "Invalid request. Privoxy doesn't support gopher.\r\n";
233 /* XXX: should be a template */
234 static const char MISSING_DESTINATION_RESPONSE[] =
235 "HTTP/1.1 400 Bad request received from client\r\n"
236 "Proxy-Agent: Privoxy " VERSION "\r\n"
237 "Content-Type: text/plain\r\n"
238 "Connection: close\r\n\r\n"
239 "Bad request. Privoxy was unable to extract the destination.\r\n";
241 /* XXX: should be a template */
242 static const char INVALID_SERVER_HEADERS_RESPONSE[] =
243 "HTTP/1.1 502 Server or forwarder response invalid\r\n"
244 "Proxy-Agent: Privoxy " VERSION "\r\n"
245 "Content-Type: text/plain\r\n"
246 "Connection: close\r\n\r\n"
247 "Bad response. The server or forwarder response doesn't look like HTTP.\r\n";
249 /* XXX: should be a template */
250 static const char MESSED_UP_REQUEST_RESPONSE[] =
251 "HTTP/1.1 400 Malformed request after rewriting\r\n"
252 "Proxy-Agent: Privoxy " VERSION "\r\n"
253 "Content-Type: text/plain\r\n"
254 "Connection: close\r\n\r\n"
255 "Bad request. Messed up with header filters.\r\n";
257 static const char TOO_MANY_CONNECTIONS_RESPONSE[] =
258 "HTTP/1.1 503 Too many open connections\r\n"
259 "Proxy-Agent: Privoxy " VERSION "\r\n"
260 "Content-Type: text/plain\r\n"
261 "Connection: close\r\n\r\n"
262 "Maximum number of open connections reached.\r\n";
264 static const char CLIENT_CONNECTION_TIMEOUT_RESPONSE[] =
265 "HTTP/1.1 504 Connection timeout\r\n"
266 "Proxy-Agent: Privoxy " VERSION "\r\n"
267 "Content-Type: text/plain\r\n"
268 "Connection: close\r\n\r\n"
269 "The connection timed out because the client request didn't arrive in time.\r\n";
271 /* A function to crunch a response */
272 typedef struct http_response *(*crunch_func_ptr)(struct client_state *);
274 /* Crunch function flags */
275 #define CF_NO_FLAGS 0
276 /* Cruncher applies to forced requests as well */
277 #define CF_IGNORE_FORCE 1
278 /* Crunched requests are counted for the block statistics */
279 #define CF_COUNT_AS_REJECT 2
281 /* A crunch function and its flags */
284 const crunch_func_ptr cruncher;
288 static int crunch_response_triggered(struct client_state *csp, const struct cruncher crunchers[]);
290 /* Complete list of cruncher functions */
291 static const struct cruncher crunchers_all[] = {
292 { direct_response, CF_COUNT_AS_REJECT|CF_IGNORE_FORCE},
293 { block_url, CF_COUNT_AS_REJECT },
295 { trust_url, CF_COUNT_AS_REJECT },
296 #endif /* def FEATURE_TRUST */
297 { redirect_url, CF_NO_FLAGS },
298 { dispatch_cgi, CF_IGNORE_FORCE},
302 /* Light version, used after tags are applied */
303 static const struct cruncher crunchers_light[] = {
304 { block_url, CF_COUNT_AS_REJECT },
305 { redirect_url, CF_NO_FLAGS },
311 * XXX: Don't we really mean
317 #if !defined(_WIN32) && !defined(__OS2__) && !defined(AMIGA)
318 /*********************************************************************
320 * Function : sig_handler
322 * Description : Signal handler for different signals.
323 * Exit gracefully on TERM and INT
324 * or set a flag that will cause the errlog
325 * to be reopened by the main thread on HUP.
328 * 1 : the_signal = the signal cause this function to call
332 *********************************************************************/
333 static void sig_handler(int the_signal)
339 log_error(LOG_LEVEL_INFO, "exiting by signal %d .. bye", the_signal);
351 received_hup_signal = 1;
357 * We shouldn't be here, unless we catch signals
358 * in main() that we can't handle here!
360 log_error(LOG_LEVEL_FATAL, "sig_handler: exiting on unexpected signal %d", the_signal);
368 /*********************************************************************
370 * Function : client_protocol_is_unsupported
372 * Description : Checks if the client used a known unsupported
373 * protocol and deals with it by sending an error
377 * 1 : csp = Current client state (buffers, headers, etc...)
378 * 2 : req = the first request line send by the client
380 * Returns : TRUE if an error response has been generated, or
381 * FALSE if the request doesn't look invalid.
383 *********************************************************************/
384 static int client_protocol_is_unsupported(const struct client_state *csp, char *req)
387 * If it's a FTP or gopher request, we don't support it.
389 * These checks are better than nothing, but they might
390 * not work in all configurations and some clients might
391 * have problems digesting the answer.
393 * They should, however, never cause more problems than
394 * Privoxy's old behaviour (returning the misleading HTML
397 * "Could not resolve http://(ftp|gopher)://example.org").
399 if (!strncmpic(req, "GET ftp://", 10) || !strncmpic(req, "GET gopher://", 13))
401 const char *response = NULL;
402 const char *protocol = NULL;
404 if (!strncmpic(req, "GET ftp://", 10))
406 response = FTP_RESPONSE;
411 response = GOPHER_RESPONSE;
414 log_error(LOG_LEVEL_ERROR,
415 "%s tried to use Privoxy as %s proxy: %s",
416 csp->ip_addr_str, protocol, req);
417 log_error(LOG_LEVEL_CLF,
418 "%s - - [%T] \"%s\" 400 0", csp->ip_addr_str, req);
420 write_socket(csp->cfd, response, strlen(response));
429 /*********************************************************************
431 * Function : get_request_destination_elsewhere
433 * Description : If the client's request was redirected into
434 * Privoxy without the client's knowledge,
435 * the request line lacks the destination host.
437 * This function tries to get it elsewhere,
438 * provided accept-intercepted-requests is enabled.
440 * "Elsewhere" currently only means "Host: header",
441 * but in the future we may ask the redirecting
442 * packet filter to look the destination up.
444 * If the destination stays unknown, an error
445 * response is send to the client and headers
446 * are freed so that chat() can return directly.
449 * 1 : csp = Current client state (buffers, headers, etc...)
450 * 2 : headers = a header list
452 * Returns : JB_ERR_OK if the destination is now known, or
453 * JB_ERR_PARSE if it isn't.
455 *********************************************************************/
456 static jb_err get_request_destination_elsewhere(struct client_state *csp, struct list *headers)
460 if (!(csp->config->feature_flags & RUNTIME_FEATURE_ACCEPT_INTERCEPTED_REQUESTS))
462 log_error(LOG_LEVEL_ERROR, "%s's request: \'%s\' is invalid."
463 " Privoxy isn't configured to accept intercepted requests.",
464 csp->ip_addr_str, csp->http->cmd);
465 /* XXX: Use correct size */
466 log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 400 0",
467 csp->ip_addr_str, csp->http->cmd);
469 write_socket(csp->cfd, CHEADER, strlen(CHEADER));
470 destroy_list(headers);
474 else if (JB_ERR_OK == get_destination_from_headers(headers, csp->http))
476 #ifndef FEATURE_EXTENDED_HOST_PATTERNS
477 /* Split the domain we just got for pattern matching */
478 init_domain_components(csp->http);
485 /* We can't work without destination. Go spread the news.*/
487 req = list_to_text(headers);
489 /* XXX: Use correct size */
490 log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 400 0",
491 csp->ip_addr_str, csp->http->cmd);
492 log_error(LOG_LEVEL_ERROR,
493 "Privoxy was unable to get the destination for %s's request:\n%s\n%s",
494 csp->ip_addr_str, csp->http->cmd, req);
497 write_socket(csp->cfd, MISSING_DESTINATION_RESPONSE, strlen(MISSING_DESTINATION_RESPONSE));
498 destroy_list(headers);
503 * TODO: If available, use PF's ioctl DIOCNATLOOK as last resort
504 * to get the destination IP address, use it as host directly
505 * or do a reverse DNS lookup first.
510 /*********************************************************************
512 * Function : get_server_headers
514 * Description : Parses server headers in iob and fills them
515 * into csp->headers so that they can later be
519 * 1 : csp = Current client state (buffers, headers, etc...)
521 * Returns : JB_ERR_OK if everything went fine, or
522 * JB_ERR_PARSE if the headers were incomplete.
524 *********************************************************************/
525 static jb_err get_server_headers(struct client_state *csp)
527 int continue_hack_in_da_house = 0;
530 while (((header = get_header(csp->iob)) != NULL) || continue_hack_in_da_house)
535 * continue hack in da house. Ignore the ending of
536 * this head and continue enlisting header lines.
537 * The reason is described below.
539 enlist(csp->headers, "");
540 continue_hack_in_da_house = 0;
543 else if (0 == strncmpic(header, "HTTP/1.1 100", 12))
546 * It's a bodyless continue response, don't
547 * stop header parsing after reaching its end.
549 * As a result Privoxy will concatenate the
550 * next response's head and parse and deliver
551 * the headers as if they belonged to one request.
553 * The client will separate them because of the
554 * empty line between them.
556 * XXX: What we're doing here is clearly against
557 * the intended purpose of the continue header,
558 * and under some conditions (HTTP/1.0 client request)
559 * it's a standard violation.
561 * Anyway, "sort of against the spec" is preferable
562 * to "always getting confused by Continue responses"
563 * (Privoxy's behaviour before this hack was added)
565 log_error(LOG_LEVEL_HEADER, "Continue hack in da house.");
566 continue_hack_in_da_house = 1;
568 else if (*header == '\0')
571 * If the header is empty, but the Continue hack
572 * isn't active, we can assume that we reached the
573 * end of the buffer before we hit the end of the
576 * Inform the caller an let it decide how to handle it.
581 if (JB_ERR_MEMORY == enlist(csp->headers, header))
584 * XXX: Should we quit the request and return a
585 * out of memory error page instead?
587 log_error(LOG_LEVEL_ERROR,
588 "Out of memory while enlisting server headers. %s lost.",
598 /*********************************************************************
600 * Function : crunch_reason
602 * Description : Translates the crunch reason code into a string.
605 * 1 : rsp = a http_response
607 * Returns : A string with the crunch reason or an error description.
609 *********************************************************************/
610 static const char *crunch_reason(const struct http_response *rsp)
612 char * reason = NULL;
617 return "Internal error while searching for crunch reason";
620 switch (rsp->crunch_reason)
623 reason = "Unsupported HTTP feature";
629 reason = "Untrusted";
632 reason = "Redirected";
638 reason = "DNS failure";
640 case FORWARDING_FAILED:
641 reason = "Forwarding failed";
644 reason = "Connection failure";
647 reason = "Out of memory (may mask other reasons)";
649 case CONNECTION_TIMEOUT:
650 reason = "Connection timeout";
653 reason = "No server data received";
656 reason = "No reason recorded";
664 /*********************************************************************
666 * Function : log_applied_actions
668 * Description : Logs the applied actions if LOG_LEVEL_ACTIONS is
672 * 1 : actions = Current action spec to log
676 *********************************************************************/
677 static void log_applied_actions(const struct current_action_spec *actions)
680 * The conversion to text requires lots of memory allocations so
681 * we only do the conversion if the user is actually interested.
683 if (debug_level_is_enabled(LOG_LEVEL_ACTIONS))
685 char *actions_as_text = actions_to_line_of_text(actions);
686 log_error(LOG_LEVEL_ACTIONS, "%s", actions_as_text);
687 freez(actions_as_text);
692 /*********************************************************************
694 * Function : send_crunch_response
696 * Description : Delivers already prepared response for
697 * intercepted requests, logs the interception
698 * and frees the response.
701 * 1 : csp = Current client state (buffers, headers, etc...)
702 * 1 : rsp = Fully prepared response. Will be freed on exit.
706 *********************************************************************/
707 static void send_crunch_response(const struct client_state *csp, struct http_response *rsp)
709 const struct http_request *http = csp->http;
713 assert(rsp->head != NULL);
717 log_error(LOG_LEVEL_FATAL, "NULL response in send_crunch_response.");
721 * Extract the status code from the actual head
722 * that will be send to the client. It is the only
723 * way to get it right for all requests, including
724 * the fixed ones for out-of-memory problems.
726 * A head starts like this: 'HTTP/1.1 200...'
730 status_code[0] = rsp->head[9];
731 status_code[1] = rsp->head[10];
732 status_code[2] = rsp->head[11];
733 status_code[3] = '\0';
735 /* Log that the request was crunched and why. */
736 log_applied_actions(csp->action);
737 log_error(LOG_LEVEL_CRUNCH, "%s: %s", crunch_reason(rsp), http->url);
738 log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" %s %u",
739 csp->ip_addr_str, http->ocmd, status_code, rsp->content_length);
741 /* Write the answer to the client */
742 if (write_socket(csp->cfd, rsp->head, rsp->head_length)
743 || write_socket(csp->cfd, rsp->body, rsp->content_length))
745 /* There is nothing we can do about it. */
746 log_error(LOG_LEVEL_ERROR,
747 "Couldn't deliver the error message through client socket %d: %E",
751 /* Clean up and return */
752 if (cgi_error_memory() != rsp)
754 free_http_response(rsp);
760 /*********************************************************************
762 * Function : crunch_response_triggered
764 * Description : Checks if the request has to be crunched,
765 * and delivers the crunch response if necessary.
768 * 1 : csp = Current client state (buffers, headers, etc...)
769 * 2 : crunchers = list of cruncher functions to run
771 * Returns : TRUE if the request was answered with a crunch response
774 *********************************************************************/
775 static int crunch_response_triggered(struct client_state *csp, const struct cruncher crunchers[])
777 struct http_response *rsp = NULL;
778 const struct cruncher *c;
781 * If CGI request crunching is disabled,
782 * check the CGI dispatcher out of order to
783 * prevent unintentional blocks or redirects.
785 if (!(csp->config->feature_flags & RUNTIME_FEATURE_CGI_CRUNCHING)
786 && (NULL != (rsp = dispatch_cgi(csp))))
788 /* Deliver, log and free the interception response. */
789 send_crunch_response(csp, rsp);
790 csp->flags |= CSP_FLAG_CRUNCHED;
794 for (c = crunchers; c->cruncher != NULL; c++)
797 * Check the cruncher if either Privoxy is toggled
798 * on and the request isn't forced, or if the cruncher
799 * applies to forced requests as well.
801 if (((csp->flags & CSP_FLAG_TOGGLED_ON) &&
802 !(csp->flags & CSP_FLAG_FORCED)) ||
803 (c->flags & CF_IGNORE_FORCE))
805 rsp = c->cruncher(csp);
808 /* Deliver, log and free the interception response. */
809 send_crunch_response(csp, rsp);
810 csp->flags |= CSP_FLAG_CRUNCHED;
811 #ifdef FEATURE_STATISTICS
812 if (c->flags & CF_COUNT_AS_REJECT)
814 csp->flags |= CSP_FLAG_REJECTED;
816 #endif /* def FEATURE_STATISTICS */
827 /*********************************************************************
829 * Function : build_request_line
831 * Description : Builds the HTTP request line.
833 * If a HTTP forwarder is used it expects the whole URL,
834 * web servers only get the path.
837 * 1 : csp = Current client state (buffers, headers, etc...)
838 * 2 : fwd = The forwarding spec used for the request
839 * XXX: Should use http->fwd instead.
840 * 3 : request_line = The old request line which will be replaced.
842 * Returns : Nothing. Terminates in case of memory problems.
844 *********************************************************************/
845 static void build_request_line(struct client_state *csp, const struct forward_spec *fwd, char **request_line)
847 struct http_request *http = csp->http;
849 assert(http->ssl == 0);
852 * Downgrade http version from 1.1 to 1.0
853 * if +downgrade action applies.
855 if ((csp->action->flags & ACTION_DOWNGRADE)
856 && (!strcmpic(http->ver, "HTTP/1.1")))
859 http->ver = strdup_or_die("HTTP/1.0");
863 * Rebuild the request line.
865 freez(*request_line);
866 *request_line = strdup(http->gpc);
867 string_append(request_line, " ");
869 if (fwd->forward_host)
871 string_append(request_line, http->url);
875 string_append(request_line, http->path);
877 string_append(request_line, " ");
878 string_append(request_line, http->ver);
880 if (*request_line == NULL)
882 log_error(LOG_LEVEL_FATAL, "Out of memory writing HTTP command");
884 log_error(LOG_LEVEL_HEADER, "New HTTP Request-Line: %s", *request_line);
888 /*********************************************************************
890 * Function : change_request_destination
892 * Description : Parse a (rewritten) request line and regenerate
893 * the http request data.
896 * 1 : csp = Current client state (buffers, headers, etc...)
898 * Returns : Forwards the parse_http_request() return code.
899 * Terminates in case of memory problems.
901 *********************************************************************/
902 static jb_err change_request_destination(struct client_state *csp)
904 struct http_request *http = csp->http;
907 log_error(LOG_LEVEL_INFO, "Rewrite detected: %s", csp->headers->first->str);
908 free_http_request(http);
909 err = parse_http_request(csp->headers->first->str, http);
910 if (JB_ERR_OK != err)
912 log_error(LOG_LEVEL_ERROR, "Couldn't parse rewritten request: %s.",
913 jb_err_to_string(err));
917 /* XXX: ocmd is a misleading name */
918 http->ocmd = strdup_or_die(http->cmd);
925 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
926 /*********************************************************************
928 * Function : server_response_is_complete
930 * Description : Determines whether we should stop reading
931 * from the server socket.
934 * 1 : csp = Current client state (buffers, headers, etc...)
935 * 2 : content_length = Length of content received so far.
937 * Returns : TRUE if the response is complete,
940 *********************************************************************/
941 static int server_response_is_complete(struct client_state *csp,
942 unsigned long long content_length)
944 int content_length_known = !!(csp->flags & CSP_FLAG_CONTENT_LENGTH_SET);
946 if (!strcmpic(csp->http->gpc, "HEAD"))
949 * "HEAD" implies no body, we are thus expecting
950 * no content. XXX: incomplete "list" of methods?
952 csp->expected_content_length = 0;
953 content_length_known = TRUE;
956 if (csp->http->status == 204 || csp->http->status == 304)
959 * Expect no body. XXX: incomplete "list" of status codes?
961 csp->expected_content_length = 0;
962 content_length_known = TRUE;
965 return (content_length_known && ((0 == csp->expected_content_length)
966 || (csp->expected_content_length <= content_length)));
970 #ifdef FEATURE_CONNECTION_SHARING
971 /*********************************************************************
973 * Function : wait_for_alive_connections
975 * Description : Waits for alive connections to timeout.
981 *********************************************************************/
982 static void wait_for_alive_connections(void)
984 int connections_alive = close_unusable_connections();
986 while (0 < connections_alive)
988 log_error(LOG_LEVEL_CONNECT,
989 "Waiting for %d connections to timeout.",
992 connections_alive = close_unusable_connections();
995 log_error(LOG_LEVEL_CONNECT, "No connections to wait for left.");
998 #endif /* def FEATURE_CONNECTION_SHARING */
1001 /*********************************************************************
1003 * Function : save_connection_destination
1005 * Description : Remembers a connection for reuse later on.
1008 * 1 : sfd = Open socket to remember.
1009 * 2 : http = The destination for the connection.
1010 * 3 : fwd = The forwarder settings used.
1011 * 3 : server_connection = storage.
1015 *********************************************************************/
1016 void save_connection_destination(jb_socket sfd,
1017 const struct http_request *http,
1018 const struct forward_spec *fwd,
1019 struct reusable_connection *server_connection)
1021 assert(sfd != JB_INVALID_SOCKET);
1022 assert(NULL != http->host);
1024 server_connection->sfd = sfd;
1025 server_connection->host = strdup_or_die(http->host);
1026 server_connection->port = http->port;
1028 assert(NULL != fwd);
1029 assert(server_connection->gateway_host == NULL);
1030 assert(server_connection->gateway_port == 0);
1031 assert(server_connection->forwarder_type == 0);
1032 assert(server_connection->forward_host == NULL);
1033 assert(server_connection->forward_port == 0);
1035 server_connection->forwarder_type = fwd->type;
1036 if (NULL != fwd->gateway_host)
1038 server_connection->gateway_host = strdup_or_die(fwd->gateway_host);
1042 server_connection->gateway_host = NULL;
1044 server_connection->gateway_port = fwd->gateway_port;
1046 if (NULL != fwd->forward_host)
1048 server_connection->forward_host = strdup_or_die(fwd->forward_host);
1052 server_connection->forward_host = NULL;
1054 server_connection->forward_port = fwd->forward_port;
1058 /*********************************************************************
1060 * Function : verify_request_length
1062 * Description : Checks if we already got the whole client requests
1063 * and sets CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ if
1066 * Data that doesn't belong to the current request is
1067 * thrown away to let the client retry on a clean socket.
1069 * XXX: This is a hack until we can deal with multiple
1070 * pipelined requests at the same time.
1074 * 1 : csp = Current client state (buffers, headers, etc...)
1078 *********************************************************************/
1079 static void verify_request_length(struct client_state *csp)
1081 unsigned long long buffered_request_bytes =
1082 (unsigned long long)(csp->client_iob->eod - csp->client_iob->cur);
1084 if ((csp->expected_client_content_length != 0)
1085 && (buffered_request_bytes != 0))
1087 if (csp->expected_client_content_length >= buffered_request_bytes)
1089 csp->expected_client_content_length -= buffered_request_bytes;
1090 log_error(LOG_LEVEL_CONNECT, "Reduced expected bytes to %llu "
1091 "to account for the %llu ones we already got.",
1092 csp->expected_client_content_length, buffered_request_bytes);
1096 assert(csp->client_iob->eod > csp->client_iob->cur + csp->expected_client_content_length);
1097 csp->client_iob->eod = csp->client_iob->cur + csp->expected_client_content_length;
1098 log_error(LOG_LEVEL_CONNECT, "Reducing expected bytes to 0. "
1099 "Marking the server socket tainted after throwing %llu bytes away.",
1100 buffered_request_bytes - csp->expected_client_content_length);
1101 csp->expected_client_content_length = 0;
1102 csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED;
1105 if (csp->expected_client_content_length == 0)
1107 csp->flags |= CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ;
1111 if (!(csp->flags & CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ)
1112 && ((csp->client_iob->cur[0] != '\0') || (csp->expected_client_content_length != 0)))
1114 if (strcmpic(csp->http->gpc, "GET")
1115 && strcmpic(csp->http->gpc, "HEAD")
1116 && strcmpic(csp->http->gpc, "TRACE")
1117 && strcmpic(csp->http->gpc, "OPTIONS")
1118 && strcmpic(csp->http->gpc, "DELETE"))
1120 /* XXX: this is an incomplete hack */
1121 csp->flags &= ~CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ;
1122 csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED;
1123 log_error(LOG_LEVEL_CONNECT,
1124 "There might be a request body. The connection will not be kept alive.");
1128 csp->flags |= CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ;
1130 if ((csp->config->feature_flags & RUNTIME_FEATURE_TOLERATE_PIPELINING) == 0)
1132 csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED;
1133 log_error(LOG_LEVEL_CONNECT,
1134 "Possible pipeline attempt detected. The connection will not "
1135 "be kept alive and we will only serve the first request.");
1136 /* Nuke the pipelined requests from orbit, just to be sure. */
1137 clear_iob(csp->client_iob);
1142 * Keep the pipelined data around for now, we'll deal with
1143 * it once we're done serving the current request.
1145 csp->flags |= CSP_FLAG_PIPELINED_REQUEST_WAITING;
1146 assert(csp->client_iob->eod > csp->client_iob->cur);
1147 log_error(LOG_LEVEL_CONNECT, "Complete client request followed by "
1148 "%d bytes of pipelined data received.",
1149 (int)(csp->client_iob->eod - csp->client_iob->cur));
1155 csp->flags |= CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ;
1156 log_error(LOG_LEVEL_CONNECT, "Complete client request received.");
1159 #endif /* FEATURE_CONNECTION_KEEP_ALIVE */
1162 /*********************************************************************
1164 * Function : mark_server_socket_tainted
1166 * Description : Makes sure we don't reuse a server socket
1167 * (if we didn't read everything the server sent
1168 * us reusing the socket would lead to garbage).
1171 * 1 : csp = Current client state (buffers, headers, etc...)
1175 *********************************************************************/
1176 static void mark_server_socket_tainted(struct client_state *csp)
1179 * For consistency we always mark the server socket
1180 * tainted, however, to reduce the log noise we only
1181 * emit a log message if the server socket could have
1182 * actually been reused.
1184 if ((csp->flags & CSP_FLAG_SERVER_CONNECTION_KEEP_ALIVE)
1185 && !(csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED))
1187 log_error(LOG_LEVEL_CONNECT,
1188 "Marking the server socket %d tainted.",
1189 csp->server_connection.sfd);
1191 csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED;
1194 /*********************************************************************
1196 * Function : get_request_line
1198 * Description : Read the client request line.
1201 * 1 : csp = Current client state (buffers, headers, etc...)
1203 * Returns : Pointer to request line or NULL in case of errors.
1205 *********************************************************************/
1206 static char *get_request_line(struct client_state *csp)
1208 char buf[BUFFER_SIZE];
1209 char *request_line = NULL;
1212 memset(buf, 0, sizeof(buf));
1214 if ((csp->flags & CSP_FLAG_PIPELINED_REQUEST_WAITING) != 0)
1217 * If there are multiple pipelined requests waiting,
1218 * the flag will be set again once the next request
1221 csp->flags &= ~CSP_FLAG_PIPELINED_REQUEST_WAITING;
1223 request_line = get_header(csp->client_iob);
1224 if ((NULL != request_line) && ('\0' != *request_line))
1226 return request_line;
1230 log_error(LOG_LEVEL_CONNECT, "No complete request line "
1231 "received yet. Continuing reading from %d.", csp->cfd);
1237 if (!data_is_available(csp->cfd, csp->config->socket_timeout))
1239 if (socket_is_still_alive(csp->cfd))
1241 log_error(LOG_LEVEL_CONNECT,
1242 "No request line on socket %d received in time. Timeout: %d.",
1243 csp->cfd, csp->config->socket_timeout);
1244 write_socket(csp->cfd, CLIENT_CONNECTION_TIMEOUT_RESPONSE,
1245 strlen(CLIENT_CONNECTION_TIMEOUT_RESPONSE));
1249 log_error(LOG_LEVEL_CONNECT,
1250 "The client side of the connection on socket %d got "
1251 "closed without sending a complete request line.", csp->cfd);
1256 len = read_socket(csp->cfd, buf, sizeof(buf) - 1);
1258 if (len <= 0) return NULL;
1261 * If there is no memory left for buffering the
1262 * request, there is nothing we can do but hang up
1264 if (add_to_iob(csp->client_iob, csp->config->buffer_limit, buf, len))
1269 request_line = get_header(csp->client_iob);
1271 } while ((NULL != request_line) && ('\0' == *request_line));
1273 return request_line;
1278 /*********************************************************************
1280 * Function : receive_client_request
1282 * Description : Read the client's request (more precisely the
1283 * client headers) and answer it if necessary.
1286 * 1 : csp = Current client state (buffers, headers, etc...)
1288 * Returns : JB_ERR_OK, JB_ERR_PARSE or JB_ERR_MEMORY
1290 *********************************************************************/
1291 static jb_err receive_client_request(struct client_state *csp)
1293 char buf[BUFFER_SIZE];
1296 struct http_request *http;
1300 /* Temporary copy of the client's headers before they get enlisted in csp->headers */
1301 struct list header_list;
1302 struct list *headers = &header_list;
1304 /* We don't care if the arriving data is a valid HTTP request or not. */
1305 csp->requests_received_total++;
1309 memset(buf, 0, sizeof(buf));
1311 req = get_request_line(csp);
1314 mark_server_socket_tainted(csp);
1315 return JB_ERR_PARSE;
1317 assert(*req != '\0');
1319 if (client_protocol_is_unsupported(csp, req))
1321 return JB_ERR_PARSE;
1324 #ifdef FEATURE_FORCE_LOAD
1326 * If this request contains the FORCE_PREFIX and blocks
1327 * aren't enforced, get rid of it and set the force flag.
1329 if (strstr(req, FORCE_PREFIX))
1331 if (csp->config->feature_flags & RUNTIME_FEATURE_ENFORCE_BLOCKS)
1333 log_error(LOG_LEVEL_FORCE,
1334 "Ignored force prefix in request: \"%s\".", req);
1338 strclean(req, FORCE_PREFIX);
1339 log_error(LOG_LEVEL_FORCE, "Enforcing request: \"%s\".", req);
1340 csp->flags |= CSP_FLAG_FORCED;
1343 #endif /* def FEATURE_FORCE_LOAD */
1345 err = parse_http_request(req, http);
1347 if (JB_ERR_OK != err)
1349 write_socket(csp->cfd, CHEADER, strlen(CHEADER));
1350 /* XXX: Use correct size */
1351 log_error(LOG_LEVEL_CLF, "%s - - [%T] \"Invalid request\" 400 0", csp->ip_addr_str);
1352 log_error(LOG_LEVEL_ERROR,
1353 "Couldn't parse request line received from %s: %s",
1354 csp->ip_addr_str, jb_err_to_string(err));
1356 free_http_request(http);
1357 return JB_ERR_PARSE;
1360 /* grab the rest of the client's headers */
1364 p = get_header(csp->client_iob);
1368 /* There are no additional headers to read. */
1375 * We didn't receive a complete header
1376 * line yet, get the rest of it.
1378 if (!data_is_available(csp->cfd, csp->config->socket_timeout))
1380 log_error(LOG_LEVEL_ERROR,
1381 "Stopped grabbing the client headers.");
1382 destroy_list(headers);
1383 return JB_ERR_PARSE;
1386 len = read_socket(csp->cfd, buf, sizeof(buf) - 1);
1389 log_error(LOG_LEVEL_ERROR, "read from client failed: %E");
1390 destroy_list(headers);
1391 return JB_ERR_PARSE;
1394 if (add_to_iob(csp->client_iob, csp->config->buffer_limit, buf, len))
1397 * If there is no memory left for buffering the
1398 * request, there is nothing we can do but hang up
1400 destroy_list(headers);
1401 return JB_ERR_MEMORY;
1407 * We were able to read a complete
1408 * header and can finally enlist it.
1415 if (http->host == NULL)
1418 * If we still don't know the request destination,
1419 * the request is invalid or the client uses
1420 * Privoxy without its knowledge.
1422 if (JB_ERR_OK != get_request_destination_elsewhere(csp, headers))
1425 * Our attempts to get the request destination
1426 * elsewhere failed or Privoxy is configured
1427 * to only accept proxy requests.
1429 * An error response has already been send
1430 * and we're done here.
1432 return JB_ERR_PARSE;
1437 * Determine the actions for this URL
1439 #ifdef FEATURE_TOGGLE
1440 if (!(csp->flags & CSP_FLAG_TOGGLED_ON))
1442 /* Most compatible set of actions (i.e. none) */
1443 init_current_action(csp->action);
1446 #endif /* ndef FEATURE_TOGGLE */
1448 get_url_actions(csp, http);
1452 * Save a copy of the original request for logging
1454 http->ocmd = strdup_or_die(http->cmd);
1455 enlist(csp->headers, http->cmd);
1457 /* Append the previously read headers */
1458 list_append_list_unique(csp->headers, headers);
1459 destroy_list(headers);
1466 /*********************************************************************
1468 * Function : parse_client_request
1470 * Description : Parses the client's request and decides what to do
1473 * Note that since we're not using select() we could get
1474 * blocked here if a client connected, then didn't say
1478 * 1 : csp = Current client state (buffers, headers, etc...)
1480 * Returns : JB_ERR_OK or JB_ERR_PARSE
1482 *********************************************************************/
1483 static jb_err parse_client_request(struct client_state *csp)
1485 struct http_request *http = csp->http;
1488 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
1489 if ((csp->config->feature_flags & RUNTIME_FEATURE_CONNECTION_KEEP_ALIVE)
1490 && (!strcmpic(csp->http->ver, "HTTP/1.1"))
1491 && (csp->http->ssl == 0))
1493 /* Assume persistence until further notice */
1494 csp->flags |= CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE;
1497 if (csp->http->ssl == 0)
1499 csp->expected_client_content_length = get_expected_content_length(csp->headers);
1500 verify_request_length(csp);
1502 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
1504 err = sed(csp, FILTER_CLIENT_HEADERS);
1505 if (JB_ERR_OK != err)
1507 /* XXX: Should be handled in sed(). */
1508 assert(err == JB_ERR_PARSE);
1509 log_error(LOG_LEVEL_FATAL, "Failed to parse client headers.");
1511 csp->flags |= CSP_FLAG_CLIENT_HEADER_PARSING_DONE;
1513 /* Check request line for rewrites. */
1514 if ((NULL == csp->headers->first->str)
1515 || (strcmp(http->cmd, csp->headers->first->str) &&
1516 (JB_ERR_OK != change_request_destination(csp))))
1519 * A header filter broke the request line - bail out.
1521 write_socket(csp->cfd, MESSED_UP_REQUEST_RESPONSE, strlen(MESSED_UP_REQUEST_RESPONSE));
1522 /* XXX: Use correct size */
1523 log_error(LOG_LEVEL_CLF,
1524 "%s - - [%T] \"Invalid request generated\" 500 0", csp->ip_addr_str);
1525 log_error(LOG_LEVEL_ERROR,
1526 "Invalid request line after applying header filters.");
1527 free_http_request(http);
1529 return JB_ERR_PARSE;
1537 /*********************************************************************
1541 * Description : Once a connection from the client has been accepted,
1542 * this function is called (via serve()) to handle the
1543 * main business of the communication. This function
1544 * returns after dealing with a single request. It can
1545 * be called multiple times with the same client socket
1546 * if the client is keeping the connection alive.
1548 * The decision whether or not a client connection will
1549 * be kept alive is up to the caller which also must
1550 * close the client socket when done.
1552 * FIXME: chat is nearly thousand lines long.
1556 * 1 : csp = Current client state (buffers, headers, etc...)
1558 * Returns : Nothing.
1560 *********************************************************************/
1561 static void chat(struct client_state *csp)
1563 char buf[BUFFER_SIZE];
1570 int ms_iis5_hack = 0;
1571 unsigned long long byte_count = 0;
1572 const struct forward_spec *fwd;
1573 struct http_request *http;
1574 long len = 0; /* for buffer sizes (and negative error codes) */
1575 int buffer_and_filter_content = 0;
1577 /* Skeleton for HTTP response, if we should intercept the request */
1578 struct http_response *rsp;
1579 struct timeval timeout;
1580 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
1581 int watch_client_socket;
1584 memset(buf, 0, sizeof(buf));
1588 if (receive_client_request(csp) != JB_ERR_OK)
1592 if (parse_client_request(csp) != JB_ERR_OK)
1597 /* decide how to route the HTTP request */
1598 fwd = forward_url(csp, http);
1601 log_error(LOG_LEVEL_FATAL, "gateway spec is NULL!?!? This can't happen!");
1602 /* Never get here - LOG_LEVEL_FATAL causes program exit */
1607 * build the http request to send to the server
1608 * we have to do one of the following:
1610 * create = use the original HTTP request to create a new
1611 * HTTP request that has either the path component
1612 * without the http://domainspec (w/path) or the
1613 * full orininal URL (w/url)
1614 * Note that the path and/or the HTTP version may
1615 * have been altered by now.
1617 * connect = Open a socket to the host:port of the server
1618 * and short-circuit server and client socket.
1620 * pass = Pass the request unchanged if forwarding a CONNECT
1621 * request to a parent proxy. Note that we'll be sending
1622 * the CFAIL message ourselves if connecting to the parent
1623 * fails, but we won't send a CSUCCEED message if it works,
1624 * since that would result in a double message (ours and the
1625 * parent's). After sending the request to the parent, we simply
1628 * here's the matrix:
1631 * +--------+--------+
1633 * 0 | create | connect|
1635 * Forwarding +--------+--------+
1637 * 1 | create | pass |
1639 * +--------+--------+
1643 if (http->ssl && connect_port_is_forbidden(csp))
1645 const char *acceptable_connect_ports =
1646 csp->action->string[ACTION_STRING_LIMIT_CONNECT];
1647 assert(NULL != acceptable_connect_ports);
1648 log_error(LOG_LEVEL_INFO, "Request from %s marked for blocking. "
1649 "limit-connect{%s} doesn't allow CONNECT requests to %s",
1650 csp->ip_addr_str, acceptable_connect_ports, csp->http->hostport);
1651 csp->action->flags |= ACTION_BLOCK;
1657 freez(csp->headers->first->str);
1658 build_request_line(csp, fwd, &csp->headers->first->str);
1662 * We have a request. Check if one of the crunchers wants it.
1664 if (crunch_response_triggered(csp, crunchers_all))
1667 * Yes. The client got the crunch response and we're done here.
1672 log_applied_actions(csp->action);
1673 log_error(LOG_LEVEL_GPC, "%s%s", http->hostport, http->path);
1675 if (fwd->forward_host)
1677 log_error(LOG_LEVEL_CONNECT, "via [%s]:%d to: %s",
1678 fwd->forward_host, fwd->forward_port, http->hostport);
1682 log_error(LOG_LEVEL_CONNECT, "to %s", http->hostport);
1685 /* here we connect to the server, gateway, or the forwarder */
1687 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
1688 if ((csp->server_connection.sfd != JB_INVALID_SOCKET)
1689 && socket_is_still_alive(csp->server_connection.sfd)
1690 && connection_destination_matches(&csp->server_connection, http, fwd))
1692 log_error(LOG_LEVEL_CONNECT,
1693 "Reusing server socket %d connected to %s. Total requests: %u.",
1694 csp->server_connection.sfd, csp->server_connection.host,
1695 csp->server_connection.requests_sent_total);
1699 if (csp->server_connection.sfd != JB_INVALID_SOCKET)
1701 log_error(LOG_LEVEL_CONNECT,
1702 "Closing server socket %u. Opened for %s.",
1703 csp->server_connection.sfd, csp->server_connection.host);
1704 close_socket(csp->server_connection.sfd);
1705 mark_connection_closed(&csp->server_connection);
1707 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
1709 csp->server_connection.sfd = forwarded_connect(fwd, http, csp);
1711 if (csp->server_connection.sfd == JB_INVALID_SOCKET)
1713 if (fwd->type != SOCKS_NONE)
1716 rsp = error_response(csp, "forwarding-failed");
1718 else if (errno == EINVAL)
1720 rsp = error_response(csp, "no-such-domain");
1724 rsp = error_response(csp, "connect-failed");
1727 /* Write the answer to the client */
1730 send_crunch_response(csp, rsp);
1735 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
1736 save_connection_destination(csp->server_connection.sfd,
1737 http, fwd, &csp->server_connection);
1738 csp->server_connection.keep_alive_timeout =
1739 (unsigned)csp->config->keep_alive_timeout;
1741 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
1743 csp->server_connection.requests_sent_total++;
1745 if (fwd->forward_host || (http->ssl == 0))
1748 hdr = list_to_text(csp->headers);
1751 /* FIXME Should handle error properly */
1752 log_error(LOG_LEVEL_FATAL, "Out of memory parsing client header");
1754 list_remove_all(csp->headers);
1757 * Write the client's (modified) header to the server
1758 * (along with anything else that may be in the buffer)
1760 write_failure = 0 != write_socket(csp->server_connection.sfd, hdr, strlen(hdr));
1765 log_error(LOG_LEVEL_CONNECT,
1766 "Failed sending request headers to: %s: %E", http->hostport);
1768 else if (((csp->flags & CSP_FLAG_PIPELINED_REQUEST_WAITING) == 0)
1769 && (flush_socket(csp->server_connection.sfd, csp->client_iob) < 0))
1772 log_error(LOG_LEVEL_CONNECT,
1773 "Failed sending request body to: %s: %E", http->hostport);
1778 rsp = error_response(csp, "connect-failed");
1781 send_crunch_response(csp, rsp);
1789 * We're running an SSL tunnel and we're not forwarding,
1790 * so just ditch the client headers, send the "connect succeeded"
1791 * message to the client, flush the rest, and get out of the way.
1793 list_remove_all(csp->headers);
1794 if (write_socket(csp->cfd, CSUCCEED, strlen(CSUCCEED)))
1798 clear_iob(csp->client_iob);
1801 log_error(LOG_LEVEL_CONNECT, "to %s successful", http->hostport);
1803 csp->server_connection.request_sent = time(NULL);
1805 maxfd = (csp->cfd > csp->server_connection.sfd) ?
1806 csp->cfd : csp->server_connection.sfd;
1808 /* pass data between the client and server
1809 * until one or the other shuts down the connection.
1814 watch_client_socket = 0 == (csp->flags & CSP_FLAG_PIPELINED_REQUEST_WAITING);
1820 * FD_ZERO here seems to point to an errant macro which crashes.
1821 * So do this by hand for now...
1823 memset(&rfds,0x00,sizeof(fd_set));
1827 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
1828 if (!watch_client_socket)
1830 maxfd = csp->server_connection.sfd;
1833 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
1835 FD_SET(csp->cfd, &rfds);
1838 FD_SET(csp->server_connection.sfd, &rfds);
1840 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
1841 if ((csp->flags & CSP_FLAG_CHUNKED)
1842 && !(csp->flags & CSP_FLAG_CONTENT_LENGTH_SET)
1843 && ((csp->iob->eod - csp->iob->cur) >= 5)
1844 && !memcmp(csp->iob->eod-5, "0\r\n\r\n", 5))
1847 * XXX: This check should be obsolete now,
1848 * but let's wait a while to be sure.
1850 log_error(LOG_LEVEL_CONNECT,
1851 "Looks like we got the last chunk together with "
1852 "the server headers but didn't detect it earlier. "
1853 "We better stop reading.");
1854 byte_count = (unsigned long long)(csp->iob->eod - csp->iob->cur);
1855 csp->expected_content_length = byte_count;
1856 csp->flags |= CSP_FLAG_CONTENT_LENGTH_SET;
1858 if (server_body && server_response_is_complete(csp, byte_count))
1860 if (csp->expected_content_length == byte_count)
1862 log_error(LOG_LEVEL_CONNECT,
1863 "Done reading from server. Content length: %llu as expected. "
1864 "Bytes most recently read: %d.",
1869 log_error(LOG_LEVEL_CONNECT,
1870 "Done reading from server. Expected content length: %llu. "
1871 "Actual content length: %llu. Bytes most recently read: %d.",
1872 csp->expected_content_length, byte_count, len);
1876 * XXX: should not jump around,
1877 * chat() is complicated enough already.
1881 #endif /* FEATURE_CONNECTION_KEEP_ALIVE */
1883 timeout.tv_sec = csp->config->socket_timeout;
1884 timeout.tv_usec = 0;
1885 n = select((int)maxfd+1, &rfds, NULL, NULL, &timeout);
1889 log_error(LOG_LEVEL_ERROR,
1890 "Didn't receive data in time: %s", http->url);
1891 if ((byte_count == 0) && (http->ssl == 0))
1893 send_crunch_response(csp, error_response(csp, "connection-timeout"));
1895 mark_server_socket_tainted(csp);
1900 log_error(LOG_LEVEL_ERROR, "select() failed!: %E");
1901 mark_server_socket_tainted(csp);
1906 * This is the body of the browser's request,
1907 * just read and write it.
1909 * XXX: Make sure the client doesn't use pipelining
1910 * behind Privoxy's back.
1912 if (FD_ISSET(csp->cfd, &rfds))
1914 int max_bytes_to_read = sizeof(buf) - 1;
1916 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
1917 if ((csp->flags & CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ))
1919 if (data_is_available(csp->cfd, 0))
1922 * If the next request is already waiting, we have
1923 * to stop select()ing the client socket. Otherwise
1924 * we would always return right away and get nothing
1927 watch_client_socket = 0;
1928 log_error(LOG_LEVEL_CONNECT,
1929 "Stopping to watch the client socket %d. "
1930 "There's already another request waiting.",
1935 * If the client socket is set, but there's no data
1936 * available on the socket, the client went fishing
1937 * and continuing talking to the server makes no sense.
1939 log_error(LOG_LEVEL_CONNECT,
1940 "The client closed socket %d while "
1941 "the server socket %d is still open.",
1942 csp->cfd, csp->server_connection.sfd);
1943 mark_server_socket_tainted(csp);
1946 if (csp->expected_client_content_length != 0)
1948 if (csp->expected_client_content_length < (sizeof(buf) - 1))
1950 max_bytes_to_read = (int)csp->expected_client_content_length;
1952 log_error(LOG_LEVEL_CONNECT,
1953 "Waiting for up to %d bytes from the client.",
1956 assert(max_bytes_to_read < sizeof(buf));
1957 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
1959 len = read_socket(csp->cfd, buf, max_bytes_to_read);
1963 /* XXX: not sure if this is necessary. */
1964 mark_server_socket_tainted(csp);
1965 break; /* "game over, man" */
1968 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
1969 if (csp->expected_client_content_length != 0)
1971 assert(len <= max_bytes_to_read);
1972 csp->expected_client_content_length -= (unsigned)len;
1973 log_error(LOG_LEVEL_CONNECT,
1974 "Expected client content length set to %llu "
1975 "after reading %d bytes.",
1976 csp->expected_client_content_length, len);
1977 if (csp->expected_client_content_length == 0)
1979 log_error(LOG_LEVEL_CONNECT,
1980 "Done reading from the client.");
1981 csp->flags |= CSP_FLAG_CLIENT_REQUEST_COMPLETELY_READ;
1984 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
1986 if (write_socket(csp->server_connection.sfd, buf, (size_t)len))
1988 log_error(LOG_LEVEL_ERROR, "write to: %s failed: %E", http->host);
1989 mark_server_socket_tainted(csp);
1996 * The server wants to talk. It could be the header or the body.
1997 * If `hdr' is null, then it's the header otherwise it's the body.
1998 * FIXME: Does `hdr' really mean `host'? No.
2000 if (FD_ISSET(csp->server_connection.sfd, &rfds))
2002 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
2004 * If we are buffering content, we don't want to eat up to
2005 * buffer-limit bytes if the client no longer cares about them.
2006 * If we aren't buffering, however, a dead client socket will be
2007 * noticed pretty much right away anyway, so we can reduce the
2008 * overhead by skipping the check.
2010 if (buffer_and_filter_content && !socket_is_still_alive(csp->cfd))
2013 log_error(LOG_LEVEL_CONNECT,
2014 "The server still wants to talk, but the client may already have hung up on us.");
2016 log_error(LOG_LEVEL_CONNECT,
2017 "The server still wants to talk, but the client hung up on us.");
2018 mark_server_socket_tainted(csp);
2020 #endif /* def _WIN32 */
2022 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
2024 len = read_socket(csp->server_connection.sfd, buf, sizeof(buf) - 1);
2028 log_error(LOG_LEVEL_ERROR, "read from: %s failed: %E", http->host);
2030 if (http->ssl && (fwd->forward_host == NULL))
2033 * Just hang up. We already confirmed the client's CONNECT
2034 * request with status code 200 and unencrypted content is
2035 * no longer welcome.
2037 log_error(LOG_LEVEL_ERROR,
2038 "CONNECT already confirmed. Unable to tell the client about the problem.");
2041 else if (byte_count)
2044 * Just hang up. We already transmitted the original headers
2045 * and parts of the original content and therefore missed the
2046 * chance to send an error message (without risking data corruption).
2048 * XXX: we could retry with a fancy range request here.
2050 log_error(LOG_LEVEL_ERROR, "Already forwarded the original headers. "
2051 "Unable to tell the client about the problem.");
2052 mark_server_socket_tainted(csp);
2056 * XXX: Consider handling the cases above the same.
2058 mark_server_socket_tainted(csp);
2062 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
2063 if (csp->flags & CSP_FLAG_CHUNKED)
2065 if ((len >= 5) && !memcmp(buf+len-5, "0\r\n\r\n", 5))
2067 /* XXX: this is a temporary hack */
2068 log_error(LOG_LEVEL_CONNECT,
2069 "Looks like we reached the end of the last chunk. "
2070 "We better stop reading.");
2071 csp->expected_content_length = byte_count + (unsigned long long)len;
2072 csp->flags |= CSP_FLAG_CONTENT_LENGTH_SET;
2076 #endif /* FEATURE_CONNECTION_KEEP_ALIVE */
2079 * Add a trailing zero to let be able to use string operations.
2080 * XXX: do we still need this with filter_popups gone?
2085 * Normally, this would indicate that we've read
2086 * as much as the server has sent us and we can
2087 * close the client connection. However, Microsoft
2088 * in its wisdom has released IIS/5 with a bug that
2089 * prevents it from sending the trailing \r\n in
2090 * a 302 redirect header (and possibly other headers).
2091 * To work around this if we've haven't parsed
2092 * a full header we'll append a trailing \r\n
2093 * and see if this now generates a valid one.
2095 * This hack shouldn't have any impacts. If we've
2096 * already transmitted the header or if this is a
2097 * SSL connection, then we won't bother with this
2098 * hack. So we only work on partially received
2099 * headers. If we append a \r\n and this still
2100 * doesn't generate a valid header, then we won't
2101 * transmit anything to the client.
2106 if (server_body || http->ssl)
2109 * If we have been buffering up the document,
2110 * now is the time to apply content modification
2111 * and send the result to the client.
2113 if (buffer_and_filter_content)
2115 p = execute_content_filters(csp);
2117 * If content filtering fails, use the original
2118 * buffer and length.
2119 * (see p != NULL ? p : csp->iob->cur below)
2123 csp->content_length = (size_t)(csp->iob->eod - csp->iob->cur);
2125 #ifdef FEATURE_COMPRESSION
2126 else if ((csp->flags & CSP_FLAG_CLIENT_SUPPORTS_DEFLATE)
2127 && (csp->content_length > LOWER_LENGTH_LIMIT_FOR_COMPRESSION))
2129 char *compressed_content = compress_buffer(p,
2130 (size_t *)&csp->content_length, csp->config->compression_level);
2131 if (compressed_content != NULL)
2134 p = compressed_content;
2135 csp->flags |= CSP_FLAG_BUFFERED_CONTENT_DEFLATED;
2140 if (JB_ERR_OK != update_server_headers(csp))
2142 log_error(LOG_LEVEL_FATAL,
2143 "Failed to update server headers. after filtering.");
2146 hdr = list_to_text(csp->headers);
2149 /* FIXME Should handle error properly */
2150 log_error(LOG_LEVEL_FATAL, "Out of memory parsing server header");
2153 if (write_socket(csp->cfd, hdr, strlen(hdr))
2154 || write_socket(csp->cfd,
2155 ((p != NULL) ? p : csp->iob->cur), (size_t)csp->content_length))
2157 log_error(LOG_LEVEL_ERROR, "write modified content to client failed: %E");
2160 mark_server_socket_tainted(csp);
2168 break; /* "game over, man" */
2172 * This is NOT the body, so
2173 * Let's pretend the server just sent us a blank line.
2175 snprintf(buf, sizeof(buf), "\r\n");
2176 len = (int)strlen(buf);
2179 * Now, let the normal header parsing algorithm below do its
2180 * job. If it fails, we'll exit instead of continuing.
2187 * If this is an SSL connection or we're in the body
2188 * of the server document, just write it to the client,
2189 * unless we need to buffer the body for later content-filtering
2191 if (server_body || http->ssl)
2193 if (buffer_and_filter_content)
2196 * If there is no memory left for buffering the content, or the buffer limit
2197 * has been reached, switch to non-filtering mode, i.e. make & write the
2198 * header, flush the iob and buf, and get out of the way.
2200 if (add_to_iob(csp->iob, csp->config->buffer_limit, buf, len))
2205 log_error(LOG_LEVEL_INFO,
2206 "Flushing header and buffers. Stepping back from filtering.");
2208 hdr = list_to_text(csp->headers);
2212 * Memory is too tight to even generate the header.
2213 * Send our static "Out-of-memory" page.
2215 log_error(LOG_LEVEL_ERROR, "Out of memory while trying to flush.");
2216 rsp = cgi_error_memory();
2217 send_crunch_response(csp, rsp);
2218 mark_server_socket_tainted(csp);
2221 hdrlen = strlen(hdr);
2223 if (write_socket(csp->cfd, hdr, hdrlen)
2224 || ((flushed = flush_socket(csp->cfd, csp->iob)) < 0)
2225 || (write_socket(csp->cfd, buf, (size_t)len)))
2227 log_error(LOG_LEVEL_CONNECT,
2228 "Flush header and buffers to client failed: %E");
2230 mark_server_socket_tainted(csp);
2235 * Reset the byte_count to the amount of bytes
2236 * we just flushed. len will be added a few lines below,
2237 * hdrlen doesn't matter for LOG_LEVEL_CLF.
2239 byte_count = (unsigned long long)flushed;
2241 buffer_and_filter_content = 0;
2247 if (write_socket(csp->cfd, buf, (size_t)len))
2249 log_error(LOG_LEVEL_ERROR, "write to client failed: %E");
2250 mark_server_socket_tainted(csp);
2254 byte_count += (unsigned long long)len;
2260 * We're still looking for the end of the server's header.
2261 * Buffer up the data we just read. If that fails, there's
2262 * little we can do but send our static out-of-memory page.
2264 if (add_to_iob(csp->iob, csp->config->buffer_limit, buf, len))
2266 log_error(LOG_LEVEL_ERROR, "Out of memory while looking for end of server headers.");
2267 rsp = cgi_error_memory();
2268 send_crunch_response(csp, rsp);
2269 mark_server_socket_tainted(csp);
2273 /* Convert iob into something sed() can digest */
2274 if (JB_ERR_PARSE == get_server_headers(csp))
2279 * Well, we tried our MS IIS/5 hack and it didn't work.
2280 * The header is incomplete and there isn't anything
2281 * we can do about it.
2283 log_error(LOG_LEVEL_ERROR, "Invalid server headers. "
2284 "Applying the MS IIS5 hack didn't help.");
2285 log_error(LOG_LEVEL_CLF,
2286 "%s - - [%T] \"%s\" 502 0", csp->ip_addr_str, http->cmd);
2287 write_socket(csp->cfd, INVALID_SERVER_HEADERS_RESPONSE,
2288 strlen(INVALID_SERVER_HEADERS_RESPONSE));
2289 mark_server_socket_tainted(csp);
2295 * Since we have to wait for more from the server before
2296 * we can parse the headers we just continue here.
2298 log_error(LOG_LEVEL_CONNECT,
2299 "Continuing buffering server headers from socket %d. "
2300 "Bytes most recently read: %d.", csp->cfd, len);
2307 * Account for the content bytes we
2308 * might have gotten with the headers.
2310 assert(csp->iob->eod >= csp->iob->cur);
2311 byte_count = (unsigned long long)(csp->iob->eod - csp->iob->cur);
2314 /* Did we actually get anything? */
2315 if (NULL == csp->headers->first)
2317 if ((csp->flags & CSP_FLAG_REUSED_CLIENT_CONNECTION))
2319 log_error(LOG_LEVEL_ERROR,
2320 "No server or forwarder response received on socket %d. "
2321 "Closing client socket %d without sending data.",
2322 csp->server_connection.sfd, csp->cfd);
2323 log_error(LOG_LEVEL_CLF,
2324 "%s - - [%T] \"%s\" 502 0", csp->ip_addr_str, http->cmd);
2328 log_error(LOG_LEVEL_ERROR,
2329 "No server or forwarder response received on socket %d.",
2330 csp->server_connection.sfd);
2331 send_crunch_response(csp, error_response(csp, "no-server-data"));
2333 free_http_request(http);
2334 mark_server_socket_tainted(csp);
2338 assert(csp->headers->first->str);
2340 if (strncmpic(csp->headers->first->str, "HTTP", 4) &&
2341 strncmpic(csp->headers->first->str, "ICY", 3))
2344 * It doesn't look like a HTTP (or Shoutcast) response:
2345 * tell the client and log the problem.
2347 if (strlen(csp->headers->first->str) > 30)
2349 csp->headers->first->str[30] = '\0';
2351 log_error(LOG_LEVEL_ERROR,
2352 "Invalid server or forwarder response. Starts with: %s",
2353 csp->headers->first->str);
2354 log_error(LOG_LEVEL_CLF,
2355 "%s - - [%T] \"%s\" 502 0", csp->ip_addr_str, http->cmd);
2356 write_socket(csp->cfd, INVALID_SERVER_HEADERS_RESPONSE,
2357 strlen(INVALID_SERVER_HEADERS_RESPONSE));
2358 free_http_request(http);
2359 mark_server_socket_tainted(csp);
2364 * We have now received the entire server header,
2365 * filter it and send the result to the client
2367 if (JB_ERR_OK != sed(csp, FILTER_SERVER_HEADERS))
2369 log_error(LOG_LEVEL_FATAL, "Failed to parse server headers.");
2371 hdr = list_to_text(csp->headers);
2374 /* FIXME Should handle error properly */
2375 log_error(LOG_LEVEL_FATAL, "Out of memory parsing server header");
2378 if ((csp->flags & CSP_FLAG_CHUNKED)
2379 && !(csp->flags & CSP_FLAG_CONTENT_LENGTH_SET)
2380 && ((csp->iob->eod - csp->iob->cur) >= 5)
2381 && !memcmp(csp->iob->eod-5, "0\r\n\r\n", 5))
2383 log_error(LOG_LEVEL_CONNECT,
2384 "Looks like we got the last chunk together with "
2385 "the server headers. We better stop reading.");
2386 byte_count = (unsigned long long)(csp->iob->eod - csp->iob->cur);
2387 csp->expected_content_length = byte_count;
2388 csp->flags |= CSP_FLAG_CONTENT_LENGTH_SET;
2391 csp->server_connection.response_received = time(NULL);
2393 if (crunch_response_triggered(csp, crunchers_light))
2396 * One of the tags created by a server-header
2397 * tagger triggered a crunch. We already
2398 * delivered the crunch response to the client
2399 * and are done here after cleaning up.
2402 mark_server_socket_tainted(csp);
2405 /* Buffer and pcrs filter this if appropriate. */
2407 if (!http->ssl) /* We talk plaintext */
2409 buffer_and_filter_content = content_requires_filtering(csp);
2412 * Only write if we're not buffering for content modification
2414 if (!buffer_and_filter_content)
2417 * Write the server's (modified) header to
2418 * the client (along with anything else that
2419 * may be in the buffer)
2422 if (write_socket(csp->cfd, hdr, strlen(hdr))
2423 || ((len = flush_socket(csp->cfd, csp->iob)) < 0))
2425 log_error(LOG_LEVEL_CONNECT, "write header to client failed: %E");
2428 * The write failed, so don't bother mentioning it
2429 * to the client... it probably can't hear us anyway.
2432 mark_server_socket_tainted(csp);
2437 /* we're finished with the server's header */
2443 * If this was a MS IIS/5 hack then it means the server
2444 * has already closed the connection. Nothing more to read.
2449 log_error(LOG_LEVEL_ERROR,
2450 "Closed server connection detected. "
2451 "Applying the MS IIS5 hack didn't help.");
2452 log_error(LOG_LEVEL_CLF,
2453 "%s - - [%T] \"%s\" 502 0", csp->ip_addr_str, http->cmd);
2454 write_socket(csp->cfd, INVALID_SERVER_HEADERS_RESPONSE,
2455 strlen(INVALID_SERVER_HEADERS_RESPONSE));
2456 mark_server_socket_tainted(csp);
2462 mark_server_socket_tainted(csp);
2463 return; /* huh? we should never get here */
2466 if (csp->content_length == 0)
2469 * If Privoxy didn't recalculate the Content-Length,
2470 * byte_count is still correct.
2472 csp->content_length = byte_count;
2475 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
2476 if ((csp->flags & CSP_FLAG_CONTENT_LENGTH_SET)
2477 && (csp->expected_content_length != byte_count))
2479 log_error(LOG_LEVEL_CONNECT,
2480 "Received %llu bytes while expecting %llu.",
2481 byte_count, csp->expected_content_length);
2482 mark_server_socket_tainted(csp);
2486 log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 200 %llu",
2487 csp->ip_addr_str, http->ocmd, csp->content_length);
2489 csp->server_connection.timestamp = time(NULL);
2493 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
2494 /*********************************************************************
2496 * Function : prepare_csp_for_next_request
2498 * Description : Put the csp in a mostly vergin state.
2501 * 1 : csp = Current client state (buffers, headers, etc...)
2505 *********************************************************************/
2506 static void prepare_csp_for_next_request(struct client_state *csp)
2508 unsigned int toggled_on_flag_set = (0 != (csp->flags & CSP_FLAG_TOGGLED_ON));
2510 csp->content_type = 0;
2511 csp->content_length = 0;
2512 csp->expected_content_length = 0;
2513 csp->expected_client_content_length = 0;
2514 list_remove_all(csp->headers);
2515 clear_iob(csp->iob);
2516 freez(csp->error_message);
2517 free_http_request(csp->http);
2518 destroy_list(csp->headers);
2519 destroy_list(csp->tags);
2520 free_current_action(csp->action);
2521 if (NULL != csp->fwd)
2523 unload_forward_spec(csp->fwd);
2526 /* XXX: Store per-connection flags someplace else. */
2527 csp->flags = (CSP_FLAG_ACTIVE | CSP_FLAG_REUSED_CLIENT_CONNECTION);
2528 if (toggled_on_flag_set)
2530 csp->flags |= CSP_FLAG_TOGGLED_ON;
2533 if (csp->client_iob->eod > csp->client_iob->cur)
2535 long bytes_to_shift = csp->client_iob->cur - csp->client_iob->buf;
2536 size_t data_length = (size_t)(csp->client_iob->eod - csp->client_iob->cur);
2538 assert(bytes_to_shift > 0);
2539 assert(data_length > 0);
2541 log_error(LOG_LEVEL_CONNECT, "Shifting %d pipelined bytes by %d bytes",
2542 data_length, bytes_to_shift);
2543 memmove(csp->client_iob->buf, csp->client_iob->cur, data_length);
2544 csp->client_iob->cur = csp->client_iob->buf;
2545 assert(csp->client_iob->eod == csp->client_iob->buf + bytes_to_shift + data_length);
2546 csp->client_iob->eod = csp->client_iob->buf + data_length;
2547 memset(csp->client_iob->eod, '\0', (size_t)bytes_to_shift);
2549 csp->flags |= CSP_FLAG_PIPELINED_REQUEST_WAITING;
2554 * We mainly care about resetting client_iob->cur so we don't
2555 * waste buffer space at the beginning and don't mess up the
2556 * request restoration done by cgi_show_request().
2558 * Freeing the buffer itself isn't technically necessary,
2559 * but makes debugging more convenient.
2561 clear_iob(csp->client_iob);
2564 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
2567 /*********************************************************************
2571 * Description : This is little more than chat. We only "serve" to
2572 * to close (or remember) any socket that chat may have
2576 * 1 : csp = Current client state (buffers, headers, etc...)
2580 *********************************************************************/
2582 void serve(struct client_state *csp)
2583 #else /* ifndef AMIGA */
2584 static void serve(struct client_state *csp)
2585 #endif /* def AMIGA */
2587 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
2588 #ifdef FEATURE_CONNECTION_SHARING
2589 static int monitor_thread_running = 0;
2590 #endif /* def FEATURE_CONNECTION_SHARING */
2591 int continue_chatting = 0;
2592 int config_file_change_detected = 0; /* Only used for debugging */
2594 log_error(LOG_LEVEL_CONNECT, "Accepted connection from %s on socket %d",
2595 csp->ip_addr_str, csp->cfd);
2599 unsigned int latency;
2604 * If the request has been crunched,
2605 * the calculated latency is zero.
2607 latency = (unsigned)(csp->server_connection.response_received -
2608 csp->server_connection.request_sent) / 2;
2610 if ((csp->flags & CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE)
2611 && (csp->flags & CSP_FLAG_CRUNCHED)
2612 && (csp->expected_client_content_length != 0))
2614 csp->flags |= CSP_FLAG_SERVER_SOCKET_TAINTED;
2615 log_error(LOG_LEVEL_CONNECT,
2616 "Tainting client socket %d due to unread data.", csp->cfd);
2619 continue_chatting = (csp->config->feature_flags
2620 & RUNTIME_FEATURE_CONNECTION_KEEP_ALIVE)
2621 && !(csp->flags & CSP_FLAG_SERVER_SOCKET_TAINTED)
2622 && (csp->cfd != JB_INVALID_SOCKET)
2623 && ((csp->flags & CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE)
2624 || (csp->config->feature_flags &
2625 RUNTIME_FEATURE_CONNECTION_SHARING));
2627 if (!(csp->flags & CSP_FLAG_CRUNCHED)
2628 && (csp->server_connection.sfd != JB_INVALID_SOCKET))
2630 if (!(csp->flags & CSP_FLAG_SERVER_KEEP_ALIVE_TIMEOUT_SET))
2632 csp->server_connection.keep_alive_timeout = csp->config->default_server_timeout;
2634 if (!(csp->flags & CSP_FLAG_SERVER_CONNECTION_KEEP_ALIVE)
2635 || (csp->flags & CSP_FLAG_SERVER_SOCKET_TAINTED)
2636 || !socket_is_still_alive(csp->server_connection.sfd)
2637 || !(latency < csp->server_connection.keep_alive_timeout))
2639 log_error(LOG_LEVEL_CONNECT,
2640 "Closing server socket %d connected to %s: "
2641 "Keep-alive %u. Tainted: %u. Socket alive %u. Timeout: %u.",
2642 csp->server_connection.sfd, csp->server_connection.host,
2643 0 != (csp->flags & CSP_FLAG_SERVER_CONNECTION_KEEP_ALIVE),
2644 0 != (csp->flags & CSP_FLAG_SERVER_SOCKET_TAINTED),
2645 socket_is_still_alive(csp->server_connection.sfd),
2646 csp->server_connection.keep_alive_timeout);
2647 #ifdef FEATURE_CONNECTION_SHARING
2648 if (csp->config->feature_flags & RUNTIME_FEATURE_CONNECTION_SHARING)
2650 forget_connection(csp->server_connection.sfd);
2652 #endif /* def FEATURE_CONNECTION_SHARING */
2653 close_socket(csp->server_connection.sfd);
2654 mark_connection_closed(&csp->server_connection);
2658 if (continue_chatting && any_loaded_file_changed(csp->config->config_file_list))
2660 continue_chatting = 0;
2661 config_file_change_detected = 1;
2664 if (continue_chatting)
2666 if (((csp->flags & CSP_FLAG_PIPELINED_REQUEST_WAITING) != 0)
2667 && socket_is_still_alive(csp->cfd))
2669 log_error(LOG_LEVEL_CONNECT, "Client request %d has been "
2670 "pipelined on socket %d and the socket is still alive.",
2671 csp->requests_received_total+1, csp->cfd);
2672 prepare_csp_for_next_request(csp);
2676 if (0 != (csp->flags & CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE))
2678 if (csp->server_connection.sfd != JB_INVALID_SOCKET)
2680 log_error(LOG_LEVEL_CONNECT,
2681 "Waiting for the next client request on socket %d. "
2682 "Keeping the server socket %d to %s open.",
2683 csp->cfd, csp->server_connection.sfd, csp->server_connection.host);
2687 log_error(LOG_LEVEL_CONNECT,
2688 "Waiting for the next client request on socket %d. "
2689 "No server socket to keep open.", csp->cfd);
2693 if ((csp->flags & CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE)
2694 && data_is_available(csp->cfd, (int)csp->config->keep_alive_timeout)
2695 && socket_is_still_alive(csp->cfd))
2697 log_error(LOG_LEVEL_CONNECT,
2698 "Client request %u arrived in time on socket %d.",
2699 csp->requests_received_total+1, csp->cfd);
2700 prepare_csp_for_next_request(csp);
2704 #ifdef FEATURE_CONNECTION_SHARING
2705 if ((csp->config->feature_flags & RUNTIME_FEATURE_CONNECTION_SHARING)
2706 && (csp->server_connection.sfd != JB_INVALID_SOCKET)
2707 && (socket_is_still_alive(csp->server_connection.sfd)))
2709 time_t time_open = time(NULL) - csp->server_connection.timestamp;
2711 if (csp->server_connection.keep_alive_timeout < time_open - (time_t)latency)
2716 remember_connection(&csp->server_connection);
2717 csp->server_connection.sfd = JB_INVALID_SOCKET;
2718 drain_and_close_socket(csp->cfd);
2719 csp->cfd = JB_INVALID_SOCKET;
2720 privoxy_mutex_lock(&connection_reuse_mutex);
2721 if (!monitor_thread_running)
2723 monitor_thread_running = 1;
2724 privoxy_mutex_unlock(&connection_reuse_mutex);
2725 wait_for_alive_connections();
2726 privoxy_mutex_lock(&connection_reuse_mutex);
2727 monitor_thread_running = 0;
2729 privoxy_mutex_unlock(&connection_reuse_mutex);
2731 #endif /* def FEATURE_CONNECTION_SHARING */
2735 else if (csp->server_connection.sfd != JB_INVALID_SOCKET)
2737 log_error(LOG_LEVEL_CONNECT,
2738 "Closing server socket %d connected to %s. Keep-alive: %u. "
2739 "Tainted: %u. Socket alive: %u. Timeout: %u. "
2740 "Configuration file change detected: %u",
2741 csp->server_connection.sfd, csp->server_connection.host,
2742 0 != (csp->flags & CSP_FLAG_SERVER_CONNECTION_KEEP_ALIVE),
2743 0 != (csp->flags & CSP_FLAG_SERVER_SOCKET_TAINTED),
2744 socket_is_still_alive(csp->server_connection.sfd),
2745 csp->server_connection.keep_alive_timeout,
2746 config_file_change_detected);
2748 } while (continue_chatting);
2752 #endif /* def FEATURE_CONNECTION_KEEP_ALIVE */
2754 if (csp->server_connection.sfd != JB_INVALID_SOCKET)
2756 #ifdef FEATURE_CONNECTION_SHARING
2757 if (csp->config->feature_flags & RUNTIME_FEATURE_CONNECTION_SHARING)
2759 forget_connection(csp->server_connection.sfd);
2761 #endif /* def FEATURE_CONNECTION_SHARING */
2762 close_socket(csp->server_connection.sfd);
2765 #ifdef FEATURE_CONNECTION_KEEP_ALIVE
2766 mark_connection_closed(&csp->server_connection);
2769 if (csp->cfd != JB_INVALID_SOCKET)
2771 log_error(LOG_LEVEL_CONNECT, "Closing client socket %d. "
2772 "Keep-alive: %u. Socket alive: %u. Data available: %u. "
2773 "Configuration file change detected: %u. Requests received: %u.",
2774 csp->cfd, 0 != (csp->flags & CSP_FLAG_CLIENT_CONNECTION_KEEP_ALIVE),
2775 socket_is_still_alive(csp->cfd), data_is_available(csp->cfd, 0),
2776 config_file_change_detected, csp->requests_received_total);
2777 drain_and_close_socket(csp->cfd);
2780 csp->flags &= ~CSP_FLAG_ACTIVE;
2786 /*********************************************************************
2788 * Function : server_thread
2790 * Description : We only exist to call `serve' in a threaded environment.
2793 * 1 : data = Current client state (buffers, headers, etc...)
2795 * Returns : Always 0.
2797 *********************************************************************/
2798 static int32 server_thread(void *data)
2800 serve((struct client_state *) data);
2807 #if !defined(_WIN32) || defined(_WIN_CONSOLE)
2808 /*********************************************************************
2812 * Description : Print usage info & exit.
2814 * Parameters : Pointer to argv[0] for identifying ourselves
2818 *********************************************************************/
2819 static void usage(const char *myname)
2821 printf("Privoxy version " VERSION " (" HOME_PAGE_URL ")\n"
2825 #endif /* defined(unix) */
2828 "[--no-daemon] [--pidfile pidfile] [--pre-chroot-nslookup hostname] [--user user[.group]] "
2829 #endif /* defined(unix) */
2830 "[--version] [configfile]\n"
2831 "Aborting\n", myname);
2836 #endif /* #if !defined(_WIN32) || defined(_WIN_CONSOLE) */
2839 #ifdef MUTEX_LOCKS_AVAILABLE
2840 /*********************************************************************
2842 * Function : privoxy_mutex_lock
2844 * Description : Locks a mutex.
2847 * 1 : mutex = The mutex to lock.
2849 * Returns : Void. May exit in case of errors.
2851 *********************************************************************/
2852 void privoxy_mutex_lock(privoxy_mutex_t *mutex)
2854 #ifdef FEATURE_PTHREAD
2855 int err = pthread_mutex_lock(mutex);
2858 if (mutex != &log_mutex)
2860 log_error(LOG_LEVEL_FATAL,
2861 "Mutex locking failed: %s.\n", strerror(err));
2866 EnterCriticalSection(mutex);
2867 #endif /* def FEATURE_PTHREAD */
2871 /*********************************************************************
2873 * Function : privoxy_mutex_unlock
2875 * Description : Unlocks a mutex.
2878 * 1 : mutex = The mutex to unlock.
2880 * Returns : Void. May exit in case of errors.
2882 *********************************************************************/
2883 void privoxy_mutex_unlock(privoxy_mutex_t *mutex)
2885 #ifdef FEATURE_PTHREAD
2886 int err = pthread_mutex_unlock(mutex);
2889 if (mutex != &log_mutex)
2891 log_error(LOG_LEVEL_FATAL,
2892 "Mutex unlocking failed: %s.\n", strerror(err));
2897 LeaveCriticalSection(mutex);
2898 #endif /* def FEATURE_PTHREAD */
2902 /*********************************************************************
2904 * Function : privoxy_mutex_init
2906 * Description : Prepares a mutex.
2909 * 1 : mutex = The mutex to initialize.
2911 * Returns : Void. May exit in case of errors.
2913 *********************************************************************/
2914 static void privoxy_mutex_init(privoxy_mutex_t *mutex)
2916 #ifdef FEATURE_PTHREAD
2917 int err = pthread_mutex_init(mutex, 0);
2920 printf("Fatal error. Mutex initialization failed: %s.\n",
2925 InitializeCriticalSection(mutex);
2926 #endif /* def FEATURE_PTHREAD */
2928 #endif /* def MUTEX_LOCKS_AVAILABLE */
2930 /*********************************************************************
2932 * Function : initialize_mutexes
2934 * Description : Prepares mutexes if mutex support is available.
2938 * Returns : Void, exits in case of errors.
2940 *********************************************************************/
2941 static void initialize_mutexes(void)
2943 #ifdef MUTEX_LOCKS_AVAILABLE
2945 * Prepare global mutex semaphores
2947 privoxy_mutex_init(&log_mutex);
2948 privoxy_mutex_init(&log_init_mutex);
2949 privoxy_mutex_init(&connection_reuse_mutex);
2952 * XXX: The assumptions below are a bit naive
2953 * and can cause locks that aren't necessary.
2955 * For example older FreeBSD versions (< 6.x?)
2956 * have no gethostbyname_r, but gethostbyname is
2959 #if !defined(HAVE_GETHOSTBYADDR_R) || !defined(HAVE_GETHOSTBYNAME_R)
2960 privoxy_mutex_init(&resolver_mutex);
2961 #endif /* !defined(HAVE_GETHOSTBYADDR_R) || !defined(HAVE_GETHOSTBYNAME_R) */
2963 * XXX: should we use a single mutex for
2964 * localtime() and gmtime() as well?
2966 #ifndef HAVE_GMTIME_R
2967 privoxy_mutex_init(&gmtime_mutex);
2968 #endif /* ndef HAVE_GMTIME_R */
2970 #ifndef HAVE_LOCALTIME_R
2971 privoxy_mutex_init(&localtime_mutex);
2972 #endif /* ndef HAVE_GMTIME_R */
2975 privoxy_mutex_init(&rand_mutex);
2976 #endif /* ndef HAVE_RANDOM */
2978 #endif /* def MUTEX_LOCKS_AVAILABLE */
2982 /*********************************************************************
2986 * Description : Load the config file and start the listen loop.
2987 * This function is a lot more *sane* with the `load_config'
2988 * and `listen_loop' functions; although it stills does
2989 * a *little* too much for my taste.
2992 * 1 : argc = Number of parameters (including $0).
2993 * 2 : argv = Array of (char *)'s to the parameters.
2995 * Returns : 1 if : can't open config file, unrecognized directive,
2996 * stats requested in multi-thread mode, can't open the
2997 * log file, can't open the jar file, listen port is invalid,
2998 * any load fails, and can't bind port.
3000 * Else main never returns, the process must be signaled
3001 * to terminate execution. Or, on Windows, use the
3002 * "File", "Exit" menu option.
3004 *********************************************************************/
3006 int real_main(int argc, char **argv)
3008 int main(int argc, char **argv)
3012 unsigned int random_seed;
3014 struct passwd *pw = NULL;
3015 struct group *grp = NULL;
3017 char *pre_chroot_nslookup_to_load_resolver = NULL;
3024 #if !defined(_WIN32)
3031 /* Prepare mutexes if supported and necessary. */
3032 initialize_mutexes();
3034 /* Enable logging until further notice. */
3038 * Parse the command line arguments
3040 * XXX: simply printing usage information in case of
3041 * invalid arguments isn't particularly user friendly.
3043 while (++argc_pos < argc)
3046 /* Check to see if the service must be installed or uninstalled */
3047 if (strncmp(argv[argc_pos], "--install", 9) == 0)
3049 const char *pName = argv[argc_pos] + 9;
3052 exit((install_service(pName)) ? 0 : 1);
3054 else if (strncmp(argv[argc_pos], "--uninstall", 11) == 0)
3056 const char *pName = argv[argc_pos] + 11;
3059 exit((uninstall_service(pName)) ? 0 : 1);
3061 else if (strcmp(argv[argc_pos], "--service") == 0)
3063 bRunAsService = TRUE;
3064 w32_set_service_cwd();
3065 atexit(w32_service_exit_notify);
3068 #endif /* defined(_WIN32) */
3071 #if !defined(_WIN32) || defined(_WIN_CONSOLE)
3073 if (strcmp(argv[argc_pos], "--help") == 0)
3078 else if (strcmp(argv[argc_pos], "--version") == 0)
3080 printf("Privoxy version " VERSION " (" HOME_PAGE_URL ")\n");
3086 else if (strcmp(argv[argc_pos], "--no-daemon") == 0)
3088 set_debug_level(LOG_LEVEL_FATAL | LOG_LEVEL_ERROR | LOG_LEVEL_INFO);
3092 else if (strcmp(argv[argc_pos], "--pidfile") == 0)
3094 if (++argc_pos == argc) usage(argv[0]);
3095 pidfile = strdup_or_die(argv[argc_pos]);
3098 else if (strcmp(argv[argc_pos], "--user") == 0)
3103 if (++argc_pos == argc) usage(argv[argc_pos]);
3105 user_arg = strdup_or_die(argv[argc_pos]);
3106 group_name = strchr(user_arg, '.');
3107 if (NULL != group_name)
3109 /* Nul-terminate the user name */
3112 /* Skip the former delimiter to actually reach the group name */
3115 grp = getgrnam(group_name);
3118 log_error(LOG_LEVEL_FATAL, "Group '%s' not found.", group_name);
3121 pw = getpwnam(user_arg);
3124 log_error(LOG_LEVEL_FATAL, "User '%s' not found.", user_arg);
3130 else if (strcmp(argv[argc_pos], "--pre-chroot-nslookup") == 0)
3132 if (++argc_pos == argc) usage(argv[0]);
3133 pre_chroot_nslookup_to_load_resolver = strdup_or_die(argv[argc_pos]);
3136 else if (strcmp(argv[argc_pos], "--chroot") == 0)
3140 #endif /* defined(unix) */
3142 else if (argc_pos + 1 != argc)
3145 * This is neither the last command line
3146 * option, nor was it recognized before,
3147 * therefore it must be invalid.
3153 #endif /* defined(_WIN32) && !defined(_WIN_CONSOLE) */
3155 configfile = argv[argc_pos];
3158 } /* -END- while (more arguments) */
3160 show_version(Argv[0]);
3163 if (*configfile != '/')
3165 char cwd[BUFFER_SIZE];
3167 size_t abs_file_size;
3169 /* make config-filename absolute here */
3170 if (NULL == getcwd(cwd, sizeof(cwd)))
3172 perror("failed to get current working directory");
3176 basedir = strdup_or_die(cwd);
3178 abs_file_size = strlen(cwd) + strlen(configfile) + 5;
3179 abs_file = malloc_or_die(abs_file_size);
3180 strlcpy(abs_file, basedir, abs_file_size);
3181 strlcat(abs_file, "/", abs_file_size);
3182 strlcat(abs_file, configfile, abs_file_size);
3183 configfile = abs_file;
3185 #endif /* defined unix */
3189 clients->next = NULL;
3191 /* XXX: factor out initialising after the next stable release. */
3194 #elif defined(_WIN32)
3198 random_seed = (unsigned int)time(NULL);
3200 srandom(random_seed);
3203 #endif /* ifdef HAVE_RANDOM */
3206 * Unix signal handling
3208 * Catch the abort, interrupt and terminate signals for a graceful exit
3209 * Catch the hangup signal so the errlog can be reopened.
3210 * Ignore the broken pipe signals (FIXME: Why?)
3212 #if !defined(_WIN32) && !defined(__OS2__) && !defined(AMIGA)
3215 const int catched_signals[] = { SIGTERM, SIGINT, SIGHUP };
3217 for (idx = 0; idx < SZ(catched_signals); idx++)
3219 #ifdef sun /* FIXME: Is it safe to check for HAVE_SIGSET instead? */
3220 if (sigset(catched_signals[idx], sig_handler) == SIG_ERR)
3222 if (signal(catched_signals[idx], sig_handler) == SIG_ERR)
3223 #endif /* ifdef sun */
3225 log_error(LOG_LEVEL_FATAL, "Can't set signal-handler for signal %d: %E", catched_signals[idx]);
3229 if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
3231 log_error(LOG_LEVEL_FATAL, "Can't set ignore-handler for SIGPIPE: %E");
3235 #else /* ifdef _WIN32 */
3236 # ifdef _WIN_CONSOLE
3238 * We *are* in a windows console app.
3239 * Print a verbose messages about FAQ's and such
3241 printf("%s", win32_blurb);
3242 # endif /* def _WIN_CONSOLE */
3243 #endif /* def _WIN32 */
3246 /* Initialize the CGI subsystem */
3247 cgi_init_error_messages();
3250 * If runnig on unix and without the --nodaemon
3251 * option, become a daemon. I.e. fork, detach
3252 * from tty and get process group leadership
3261 if (pid < 0) /* error */
3266 else if (pid != 0) /* parent */
3271 * must check for errors
3272 * child died due to missing files aso
3275 wpid = waitpid(pid, &status, WNOHANG);
3287 * stderr (fd 2) will be closed later on,
3288 * when the config file has been parsed.
3294 * Reserve fd 0 and 1 to prevent abort() and friends
3295 * from sending stuff to the clients or servers.
3297 fd = open("/dev/null", O_RDONLY);
3300 log_error(LOG_LEVEL_FATAL, "Failed to open /dev/null: %E");
3304 if (dup2(fd, 0) == -1)
3306 log_error(LOG_LEVEL_FATAL, "Failed to reserve fd 0: %E");
3310 fd = open("/dev/null", O_WRONLY);
3313 log_error(LOG_LEVEL_FATAL, "Failed to open /dev/null: %E");
3317 if (dup2(fd, 1) == -1)
3319 log_error(LOG_LEVEL_FATAL, "Failed to reserve fd 1: %E");
3326 } /* -END- if (daemon_mode) */
3329 * As soon as we have written the PID file, we can switch
3330 * to the user and group ID indicated by the --user option
3336 if (setgid((NULL != grp) ? grp->gr_gid : pw->pw_gid))
3338 log_error(LOG_LEVEL_FATAL, "Cannot setgid(): Insufficient permissions.");
3342 if (setgroups(1, &grp->gr_gid))
3344 log_error(LOG_LEVEL_FATAL, "setgroups() failed: %E");
3347 else if (initgroups(pw->pw_name, pw->pw_gid))
3349 log_error(LOG_LEVEL_FATAL, "initgroups() failed: %E");
3355 log_error(LOG_LEVEL_FATAL, "Home directory for %s undefined", pw->pw_name);
3357 /* Read the time zone file from /etc before doing chroot. */
3359 if (NULL != pre_chroot_nslookup_to_load_resolver
3360 && '\0' != pre_chroot_nslookup_to_load_resolver[0])
3362 /* Initialize resolver library. */
3363 (void) resolve_hostname_to_ip(pre_chroot_nslookup_to_load_resolver);
3365 if (chroot(pw->pw_dir) < 0)
3367 log_error(LOG_LEVEL_FATAL, "Cannot chroot to %s", pw->pw_dir);
3371 log_error(LOG_LEVEL_FATAL, "Cannot chdir /");
3374 if (setuid(pw->pw_uid))
3376 log_error(LOG_LEVEL_FATAL, "Cannot setuid(): Insufficient permissions.");
3380 char putenv_dummy[64];
3382 strlcpy(putenv_dummy, "HOME=/", sizeof(putenv_dummy));
3383 if (putenv(putenv_dummy) != 0)
3385 log_error(LOG_LEVEL_FATAL, "Cannot putenv(): HOME");
3388 snprintf(putenv_dummy, sizeof(putenv_dummy), "USER=%s", pw->pw_name);
3389 if (putenv(putenv_dummy) != 0)
3391 log_error(LOG_LEVEL_FATAL, "Cannot putenv(): USER");
3397 log_error(LOG_LEVEL_FATAL, "Cannot chroot without --user argument.");
3400 #endif /* defined unix */
3403 /* This will be FALSE unless the command line specified --service
3407 /* Yup, so now we must attempt to establish a connection
3408 * with the service dispatcher. This will only work if this
3409 * process was launched by the service control manager to
3410 * actually run as a service. If this isn't the case, i've
3411 * known it take around 30 seconds or so for the call to return.
3414 /* The StartServiceCtrlDispatcher won't return until the service is stopping */
3415 if (w32_start_service_ctrl_dispatcher(w32ServiceDispatchTable))
3417 /* Service has run, and at this point is now being stopped, so just return */
3422 printf("Warning: Failed to connect to Service Control Dispatcher\nwhen starting as a service!\n");
3424 /* An error occurred. Usually it's because --service was wrongly specified
3425 * and we were unable to connect to the Service Control Dispatcher because
3426 * it wasn't expecting us and is therefore not listening.
3428 * For now, just continue below to call the listen_loop function.
3431 #endif /* def _WIN32 */
3441 /*********************************************************************
3443 * Function : bind_port_helper
3445 * Description : Bind the listen port. Handles logging, and aborts
3449 * 1 : haddr = Host addres to bind to. Use NULL to bind to
3451 * 2 : hport = Specifies port to bind to.
3453 * Returns : Port that was opened.
3455 *********************************************************************/
3456 static jb_socket bind_port_helper(const char *haddr, int hport)
3461 result = bind_port(haddr, hport, &bfd);
3465 const char *bind_address = (NULL != haddr) ? haddr : "INADDR_ANY";
3469 log_error(LOG_LEVEL_FATAL,
3470 "can't bind to %s:%d: There may be another Privoxy "
3471 "or some other proxy running on port %d",
3472 bind_address, hport, hport);
3475 log_error(LOG_LEVEL_FATAL,
3476 "can't bind to %s:%d: The hostname is not resolvable",
3477 bind_address, hport);
3480 log_error(LOG_LEVEL_FATAL, "can't bind to %s:%d: %E",
3481 bind_address, hport);
3484 /* shouldn't get here */
3485 return JB_INVALID_SOCKET;
3490 log_error(LOG_LEVEL_INFO, "Listening on port %d on all IP addresses",
3495 log_error(LOG_LEVEL_INFO, "Listening on port %d on IP address %s",
3503 /*********************************************************************
3505 * Function : bind_ports_helper
3507 * Description : Bind the listen ports. Handles logging, and aborts
3511 * 1 : config = Privoxy configuration. Specifies ports
3513 * 2 : sockets = Preallocated array of opened sockets
3514 * corresponding to specification in config.
3515 * All non-opened sockets will be set to
3516 * JB_INVALID_SOCKET.
3518 * Returns : Nothing. Inspect sockets argument.
3520 *********************************************************************/
3521 static void bind_ports_helper(struct configuration_spec * config,
3522 jb_socket sockets[])
3526 for (i = 0; i < MAX_LISTENING_SOCKETS; i++)
3528 if (config->hport[i])
3530 sockets[i] = bind_port_helper(config->haddr[i], config->hport[i]);
3534 sockets[i] = JB_INVALID_SOCKET;
3537 config->need_bind = 0;
3541 /*********************************************************************
3543 * Function : close_ports_helper
3545 * Description : Close listenings ports.
3548 * 1 : sockets = Array of opened and non-opened sockets to
3549 * close. All sockets will be set to
3550 * JB_INVALID_SOCKET.
3552 * Returns : Nothing.
3554 *********************************************************************/
3555 static void close_ports_helper(jb_socket sockets[])
3559 for (i = 0; i < MAX_LISTENING_SOCKETS; i++)
3561 if (JB_INVALID_SOCKET != sockets[i])
3563 close_socket(sockets[i]);
3565 sockets[i] = JB_INVALID_SOCKET;
3571 /* Without this simple workaround we get this compiler warning from _beginthread
3572 * warning C4028: formal parameter 1 different from declaration
3574 void w32_service_listen_loop(void *p)
3578 #endif /* def _WIN32 */
3581 /*********************************************************************
3583 * Function : listen_loop
3585 * Description : bind the listen port and enter a "FOREVER" listening loop.
3591 *********************************************************************/
3592 static void listen_loop(void)
3594 struct client_states *csp_list = NULL;
3595 struct client_state *csp = NULL;
3596 jb_socket bfds[MAX_LISTENING_SOCKETS];
3597 struct configuration_spec *config;
3598 unsigned int active_threads = 0;
3600 config = load_config();
3602 #ifdef FEATURE_CONNECTION_SHARING
3604 * XXX: Should be relocated once it no
3605 * longer needs to emit log messages.
3607 initialize_reusable_connections();
3608 #endif /* def FEATURE_CONNECTION_SHARING */
3610 bind_ports_helper(config, bfds);
3612 #ifdef FEATURE_GRACEFUL_TERMINATION
3613 while (!g_terminate)
3618 #if !defined(FEATURE_PTHREAD) && !defined(_WIN32) && !defined(__BEOS__) && !defined(AMIGA) && !defined(__OS2__)
3619 while (waitpid(-1, NULL, WNOHANG) > 0)
3621 /* zombie children */
3623 #endif /* !defined(FEATURE_PTHREAD) && !defined(_WIN32) && !defined(__BEOS__) && !defined(AMIGA) */
3626 * Free data that was used by died threads
3628 active_threads = sweep();
3632 * Re-open the errlog after HUP signal
3634 if (received_hup_signal)
3636 if (NULL != config->logfile)
3638 init_error_log(Argv[0], config->logfile);
3640 received_hup_signal = 0;
3644 csp_list = (struct client_states *)zalloc(sizeof(*csp_list));
3645 if (NULL == csp_list)
3647 log_error(LOG_LEVEL_FATAL,
3648 "malloc(%d) for csp_list failed: %E", sizeof(*csp_list));
3651 csp = &csp_list->csp;
3653 log_error(LOG_LEVEL_CONNECT, "Listening for new connections ... ");
3655 if (!accept_connection(csp, bfds))
3657 log_error(LOG_LEVEL_CONNECT, "accept failed: %E");
3669 csp->flags |= CSP_FLAG_ACTIVE;
3670 csp->server_connection.sfd = JB_INVALID_SOCKET;
3672 csp->config = config = load_config();
3674 if (config->need_bind)
3677 * Since we were listening to the "old port", we will not see
3678 * a "listen" param change until the next request. So, at
3679 * least 1 more request must be made for us to find the new
3680 * setting. I am simply closing the old socket and binding the
3683 * Which-ever is correct, we will serve 1 more page via the
3684 * old settings. This should probably be a "show-proxy-args"
3685 * request. This should not be a so common of an operation
3686 * that this will hurt people's feelings.
3689 close_ports_helper(bfds);
3691 bind_ports_helper(config, bfds);
3694 #ifdef FEATURE_TOGGLE
3695 if (global_toggle_state)
3696 #endif /* def FEATURE_TOGGLE */
3698 csp->flags |= CSP_FLAG_TOGGLED_ON;
3701 if (run_loader(csp))
3703 log_error(LOG_LEVEL_FATAL, "a loader failed - must exit");
3704 /* Never get here - LOG_LEVEL_FATAL causes program exit */
3708 if (block_acl(NULL,csp))
3710 log_error(LOG_LEVEL_CONNECT,
3711 "Connection from %s on socket %d dropped due to ACL", csp->ip_addr_str, csp->cfd);
3712 close_socket(csp->cfd);
3713 freez(csp->ip_addr_str);
3717 #endif /* def FEATURE_ACL */
3719 if ((0 != config->max_client_connections)
3720 && (active_threads >= config->max_client_connections))
3722 log_error(LOG_LEVEL_CONNECT,
3723 "Rejecting connection from %s. Maximum number of connections reached.",
3725 write_socket(csp->cfd, TOO_MANY_CONNECTIONS_RESPONSE,
3726 strlen(TOO_MANY_CONNECTIONS_RESPONSE));
3727 close_socket(csp->cfd);
3728 freez(csp->ip_addr_str);
3733 /* add it to the list of clients */
3734 csp_list->next = clients->next;
3735 clients->next = csp_list;
3737 if (config->multi_threaded)
3741 /* this is a switch () statement in the C preprocessor - ugh */
3742 #undef SELECTED_ONE_OPTION
3744 /* Use Pthreads in preference to native code */
3745 #if defined(FEATURE_PTHREAD) && !defined(SELECTED_ONE_OPTION)
3746 #define SELECTED_ONE_OPTION
3748 pthread_t the_thread;
3749 pthread_attr_t attrs;
3751 pthread_attr_init(&attrs);
3752 pthread_attr_setdetachstate(&attrs, PTHREAD_CREATE_DETACHED);
3753 errno = pthread_create(&the_thread, &attrs,
3754 (void * (*)(void *))serve, csp);
3755 child_id = errno ? -1 : 0;
3756 pthread_attr_destroy(&attrs);
3760 #if defined(_WIN32) && !defined(_CYGWIN) && !defined(SELECTED_ONE_OPTION)
3761 #define SELECTED_ONE_OPTION
3762 child_id = _beginthread(
3763 (void (*)(void *))serve,
3768 #if defined(__OS2__) && !defined(SELECTED_ONE_OPTION)
3769 #define SELECTED_ONE_OPTION
3770 child_id = _beginthread(
3771 (void(* _Optlink)(void*))serve,
3777 #if defined(__BEOS__) && !defined(SELECTED_ONE_OPTION)
3778 #define SELECTED_ONE_OPTION
3780 thread_id tid = spawn_thread
3781 (server_thread, "server", B_NORMAL_PRIORITY, csp);
3783 if ((tid >= 0) && (resume_thread(tid) == B_OK))
3785 child_id = (int) tid;
3794 #if defined(AMIGA) && !defined(SELECTED_ONE_OPTION)
3795 #define SELECTED_ONE_OPTION
3796 csp->cfd = ReleaseSocket(csp->cfd, -1);
3799 child_id = (int)CreateNewProcTags(NP_Entry, (ULONG)server_thread,
3800 NP_Output, Output(),
3801 NP_CloseOutput, FALSE,
3802 NP_Name, (ULONG)"privoxy child",
3806 child_id = (int)CreateNewProcTags(NP_Entry, (ULONG)server_thread,
3807 NP_Output, Output(),
3808 NP_CloseOutput, FALSE,
3809 NP_Name, (ULONG)"privoxy child",
3810 NP_StackSize, 200*1024,
3816 ((struct Task *)child_id)->tc_UserData = csp;
3817 Signal((struct Task *)child_id, SIGF_SINGLE);
3822 #if !defined(SELECTED_ONE_OPTION)
3825 /* This block is only needed when using fork().
3826 * When using threads, the server thread was
3827 * created and run by the call to _beginthread().
3829 if (child_id == 0) /* child */
3832 #ifdef FEATURE_TOGGLE
3833 int inherited_toggle_state = global_toggle_state;
3834 #endif /* def FEATURE_TOGGLE */
3839 * If we've been toggled or we've blocked the request, tell Mom
3842 #ifdef FEATURE_TOGGLE
3843 if (inherited_toggle_state != global_toggle_state)
3845 rc |= RC_FLAG_TOGGLED;
3847 #endif /* def FEATURE_TOGGLE */
3849 #ifdef FEATURE_STATISTICS
3850 if (csp->flags & CSP_FLAG_REJECTED)
3852 rc |= RC_FLAG_BLOCKED;
3854 #endif /* ndef FEATURE_STATISTICS */
3858 else if (child_id > 0) /* parent */
3860 /* in a fork()'d environment, the parent's
3861 * copy of the client socket and the CSP
3865 #if !defined(_WIN32) && !defined(__CYGWIN__)
3867 wait(&child_status);
3870 * Evaluate child's return code: If the child has
3871 * - been toggled, toggle ourselves
3872 * - blocked its request, bump up the stats counter
3875 #ifdef FEATURE_TOGGLE
3876 if (WIFEXITED(child_status) && (WEXITSTATUS(child_status) & RC_FLAG_TOGGLED))
3878 global_toggle_state = !global_toggle_state;
3880 #endif /* def FEATURE_TOGGLE */
3882 #ifdef FEATURE_STATISTICS
3884 if (WIFEXITED(child_status) && (WEXITSTATUS(child_status) & RC_FLAG_BLOCKED))
3888 #endif /* def FEATURE_STATISTICS */
3890 #endif /* !defined(_WIN32) && defined(__CYGWIN__) */
3891 close_socket(csp->cfd);
3892 csp->flags &= ~CSP_FLAG_ACTIVE;
3896 #undef SELECTED_ONE_OPTION
3897 /* end of cpp switch () */
3902 * Spawning the child failed, assume it's because
3903 * there are too many children running already.
3904 * XXX: If you assume ...
3906 log_error(LOG_LEVEL_ERROR,
3907 "Unable to take any additional connections: %E");
3908 write_socket(csp->cfd, TOO_MANY_CONNECTIONS_RESPONSE,
3909 strlen(TOO_MANY_CONNECTIONS_RESPONSE));
3910 close_socket(csp->cfd);
3911 csp->flags &= ~CSP_FLAG_ACTIVE;
3920 /* NOTREACHED unless FEATURE_GRACEFUL_TERMINATION is defined */
3922 /* Clean up. Aim: free all memory (no leaks) */
3923 #ifdef FEATURE_GRACEFUL_TERMINATION
3925 log_error(LOG_LEVEL_ERROR, "Graceful termination requested");
3927 unload_current_config_file();
3928 unload_current_actions_file();
3929 unload_current_re_filterfile();
3930 #ifdef FEATURE_TRUST
3931 unload_current_trust_file();
3934 if (config->multi_threaded)
3941 } while ((clients->next != NULL) && (--i > 0));
3945 log_error(LOG_LEVEL_ERROR, "Graceful termination failed - still some live clients after 1 minute wait.");
3955 #if defined(_WIN32) && !defined(_WIN_CONSOLE)
3956 /* Cleanup - remove taskbar icon etc. */
3961 #endif /* FEATURE_GRACEFUL_TERMINATION */