Fixing warning on Windows
[privoxy.git] / jcc.c
1 const char jcc_rcs[] = "$Id: jcc.c,v 1.73 2002/03/05 23:57:30 hal9 Exp $";
2 /*********************************************************************
3  *
4  * File        :  $Source: /cvsroot/ijbswa/current/jcc.c,v $
5  *
6  * Purpose     :  Main file.  Contains main() method, main loop, and
7  *                the main connection-handling function.
8  *
9  * Copyright   :  Written by and Copyright (C) 2001 the SourceForge
10  *                IJBSWA team.  http://ijbswa.sourceforge.net
11  *
12  *                Based on the Internet Junkbuster originally written
13  *                by and Copyright (C) 1997 Anonymous Coders and
14  *                Junkbusters Corporation.  http://www.junkbusters.com
15  *
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.
21  *
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.
27  *
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.
33  *
34  * Revisions   :
35  *    $Log: jcc.c,v $
36  *    Revision 1.73  2002/03/05 23:57:30  hal9
37  *    Stray character 's' on line 1618 was breaking build.
38  *
39  *    Revision 1.72  2002/03/05 21:33:45  david__schmidt
40  *    - Re-enable OS/2 building after new parms were added
41  *    - Fix false out of memory report when resolving CGI templates when no IP
42  *      address is available of failed attempt (a la no such domain)
43  *
44  *    Revision 1.71  2002/03/05 18:13:56  oes
45  *    Added --user option
46  *
47  *    Revision 1.70  2002/03/05 04:52:42  oes
48  *    Deleted non-errlog debugging code
49  *
50  *    Revision 1.69  2002/03/04 23:50:00  jongfoster
51  *    Splitting off bind_port() call into bind_port_helper(), with
52  *    improved logging.
53  *
54  *    Revision 1.68  2002/03/04 20:17:32  oes
55  *    Fixed usage info
56  *
57  *    Revision 1.67  2002/03/04 18:18:57  oes
58  *    - Removed _DEBUG mode
59  *    - Cleand up cmdline parsing
60  *    - Introduced --no-daemon, --pidfile options
61  *    - Cleaned up signal handling:
62  *      - Terminate cleanly on INT, TERM and ABRT
63  *      - Schedule logfile for re-opening on HUP
64  *      - Ignore CHLD and PIPE
65  *      - Leave the rest with their default handlers
66  *      - Uniform handler registration
67  *    - Added usage() function
68  *    - Played styleguide police
69  *
70  *    Revision 1.66  2002/03/03 15:06:55  oes
71  *    Re-enabled automatic config reloading
72  *
73  *    Revision 1.65  2002/03/03 14:49:11  oes
74  *    Fixed CLF logging: Now uses client's original HTTP request
75  *
76  *    Revision 1.64  2002/03/03 09:18:03  joergs
77  *    Made jumbjuster work on AmigaOS again.
78  *
79  *    Revision 1.63  2002/03/02 04:14:50  david__schmidt
80  *    Clean up a little CRLF unpleasantness that suddenly appeared
81  *
82  *    Revision 1.62  2002/02/20 23:17:23  jongfoster
83  *    Detecting some out-of memory conditions and exiting with a log message.
84  *
85  *    Revision 1.61  2002/01/17 21:01:52  jongfoster
86  *    Moving all our URL and URL pattern parsing code to urlmatch.c.
87  *
88  *    Revision 1.60  2001/12/30 14:07:32  steudten
89  *    - Add signal handling (unix)
90  *    - Add SIGHUP handler (unix)
91  *    - Add creation of pidfile (unix)
92  *    - Add action 'top' in rc file (RH)
93  *    - Add entry 'SIGNALS' to manpage
94  *    - Add exit message to logfile (unix)
95  *
96  *    Revision 1.59  2001/12/13 14:07:18  oes
97  *    Fixed Bug: 503 error page now sent OK
98  *
99  *    Revision 1.58  2001/11/30 23:37:24  jongfoster
100  *    Renaming the Win32 config file to config.txt - this is almost the
101  *    same as the corresponding UNIX name "config"
102  *
103  *    Revision 1.57  2001/11/16 00:47:43  jongfoster
104  *    Changing the tty-disconnection code to use setsid().
105  *
106  *    Revision 1.56  2001/11/13 20:20:54  jongfoster
107  *    Tabs->spaces, fixing a bug with missing {} around an if()
108  *
109  *    Revision 1.55  2001/11/13 20:14:53  jongfoster
110  *    Patch for FreeBSD setpgrp() as suggested by Alexander Lazic
111  *
112  *    Revision 1.54  2001/11/07 00:03:14  steudten
113  *    Give reliable return value if an error
114  *    occurs not just 0 with new daemon mode.
115  *
116  *    Revision 1.53  2001/11/05 21:41:43  steudten
117  *    Add changes to be a real daemon just for unix os.
118  *    (change cwd to /, detach from controlling tty, set
119  *    process group and session leader to the own process.
120  *    Add DBG() Macro.
121  *    Add some fatal-error log message for failed malloc().
122  *    Add '-d' if compiled with 'configure --with-debug' to
123  *    enable debug output.
124  *
125  *    Revision 1.52  2001/10/26 20:11:20  jongfoster
126  *    Fixing type mismatch
127  *
128  *    Revision 1.51  2001/10/26 17:38:28  oes
129  *    Cosmetics
130  *
131  *    Revision 1.50  2001/10/25 03:40:48  david__schmidt
132  *    Change in porting tactics: OS/2's EMX porting layer doesn't allow multiple
133  *    threads to call select() simultaneously.  So, it's time to do a real, live,
134  *    native OS/2 port.  See defines for __EMX__ (the porting layer) vs. __OS2__
135  *    (native). Both versions will work, but using __OS2__ offers multi-threading.
136  *
137  *    Revision 1.49  2001/10/23 21:41:35  jongfoster
138  *    Added call to initialize the (statically-allocated of course)
139  *    "out of memory" CGI response.
140  *
141  *    Revision 1.48  2001/10/10 19:56:46  jongfoster
142  *    Moving some code that wasn't cookie-related out of an #ifdef
143  *    FEATURE_COOKIE_JAR
144  *
145  *    Revision 1.47  2001/10/10 16:44:36  oes
146  *    Added CONNECT destination port limitation check
147  *
148  *    Revision 1.46  2001/10/08 15:17:41  oes
149  *    Re-enabled SSL forwarding
150  *
151  *    Revision 1.45  2001/10/07 15:42:11  oes
152  *    Replaced 6 boolean members of csp with one bitmap (csp->flags)
153  *
154  *    Moved downgrading of the HTTP version from parse_http_request to
155  *      chat(), since we can't decide if it is necessary before we have
156  *      determined the actions for the URL. The HTTP command is now
157  *      *always* re-built so the repairs need no longer be special-cased.
158  *
159  *    filter_popups now gets a csp pointer so it can raise the new
160  *      CSP_FLAG_MODIFIED flag.
161  *
162  *    Bugfix
163  *
164  *    Added configurable size limit for the IOB. If the IOB grows so
165  *      large that the next read would exceed the limit, the header
166  *      is generated, and the header & unfiltered buffer are flushed
167  *      to the client. Chat then continues in non-buffering,
168  *      non-filtering body mode.
169  *
170  *    Revision 1.44  2001/10/02 18:13:57  oes
171  *    Ooops
172  *
173  *    Revision 1.43  2001/10/02 15:32:13  oes
174  *    Moved generation of hdr
175  *
176  *    Revision 1.42  2001/09/21 23:02:02  david__schmidt
177  *    Cleaning up 2 compiler warnings on OS/2.
178  *
179  *    Revision 1.41  2001/09/16 17:05:14  jongfoster
180  *    Removing unused #include showarg.h
181  *
182  *    Revision 1.40  2001/09/16 15:41:45  jongfoster
183  *    Fixing signed/unsigned comparison warning.
184  *
185  *    Revision 1.39  2001/09/16 13:21:27  jongfoster
186  *    Changes to use new list functions.
187  *
188  *    Revision 1.38  2001/09/16 13:01:46  jongfoster
189  *    Removing redundant function call that zeroed zalloc()'d memory.
190  *
191  *    Revision 1.37  2001/09/10 11:12:24  oes
192  *    Deleted unused variable
193  *
194  *    Revision 1.36  2001/09/10 10:56:15  oes
195  *    Silenced compiler warnings
196  *
197  *    Revision 1.35  2001/07/31 14:44:22  oes
198  *    Deleted unused size parameter from filter_popups()
199  *
200  *    Revision 1.34  2001/07/30 22:08:36  jongfoster
201  *    Tidying up #defines:
202  *    - All feature #defines are now of the form FEATURE_xxx
203  *    - Permanently turned off WIN_GUI_EDIT
204  *    - Permanently turned on WEBDAV and SPLIT_PROXY_ARGS
205  *
206  *    Revision 1.33  2001/07/29 19:32:00  jongfoster
207  *    Renaming _main() [mingw32 only] to real_main(), for ANSI compliance.
208  *
209  *    Revision 1.32  2001/07/29 18:47:05  jongfoster
210  *    Adding missing #include "loadcfg.h"
211  *
212  *    Revision 1.31  2001/07/29 12:17:48  oes
213  *    Applied pthread fix by Paul Lieverse
214  *
215  *    Revision 1.30  2001/07/25 22:57:13  jongfoster
216  *    __BEOS__ no longer overrides FEATURE_PTHREAD.
217  *    This is because FEATURE_PTHREAD will soon be widely used, so I
218  *    want to keep it simple.
219  *
220  *    Revision 1.29  2001/07/24 12:47:06  oes
221  *    Applied BeOS support update by Eugenia
222  *
223  *    Revision 1.28  2001/07/23 13:26:12  oes
224  *    Fixed bug in popup-killing for the first read that caused binary garbage to be sent between headers and body
225  *
226  *    Revision 1.27  2001/07/19 19:09:47  haroon
227  *    - Added code to take care of the situation where while processing the first
228  *      server response (which includes the server header), after finding the end
229  *      of the headers we were not looking past the end of the headers for
230  *      content modification. I enabled it for filter_popups.
231  *      Someone else should look to see if other similar operations should be
232  *      done to the discarded portion of the buffer.
233  *
234  *      Note 2001/07/20: No, the other content modification mechanisms will process
235  *                       the whole iob later anyway. --oes
236  *
237  *    Revision 1.26  2001/07/18 12:31:36  oes
238  *    cosmetics
239  *
240  *    Revision 1.25  2001/07/15 19:43:49  jongfoster
241  *    Supports POSIX threads.
242  *    Also removed some unused #includes.
243  *
244  *    Revision 1.24  2001/07/13 14:00:40  oes
245  *     - Generic content modification scheme:
246  *       Each feature has its own applicability flag that is set
247  *       from csp->action->flags.
248  *       Replaced the "filtering" int flag , by a function pointer
249  *       "content_filter" to the function that will do the content
250  *       modification. If it is != NULL, the document will be buffered
251  *       and processed through *content_filter, which must set
252  *       csp->content_length and return a modified copy of the body
253  *       or return NULL (on failiure).
254  *     - Changed csp->is_text to the more generic bitmap csp->content_type
255  *       which can currently take the valued CT_TEXT or CT_GIF
256  *     - Reformatting etc
257  *     - Removed all #ifdef PCRS
258  *
259  *    Revision 1.23  2001/07/02 02:28:25  iwanttokeepanon
260  *    Added "#ifdef ACL_FILES" conditional compilation to line 1291 to exclude
261  *    the `block_acl' call.  This prevents a compilation error when the user
262  *    does not wish to use the "ACL" feature.
263  *
264  *    Revision 1.22  2001/06/29 21:45:41  oes
265  *    Indentation, CRLF->LF, Tab-> Space
266  *
267  *    Revision 1.21  2001/06/29 13:29:36  oes
268  *    - Cleaned up, improved comments
269  *    - Unified all possible interceptors (CGI,
270  *      block, trust, fast_redirect) in one
271  *      place, with one (CGI) answer generation
272  *      mechansim. Much clearer now.
273  *    - Removed the GIF image generation, which
274  *      is now done in filters.c:block_url()
275  *    - Made error conditions like domain lookup
276  *      failiure or (various) problems while talking
277  *      to the server use cgi.c:error_response()
278  *      instead of generating HTML/HTTP in chat() (yuck!)
279  *    - Removed logentry from cancelled commit
280  *
281  *    Revision 1.20  2001/06/09 10:55:28  jongfoster
282  *    Changing BUFSIZ ==> BUFFER_SIZE
283  *
284  *    Revision 1.19  2001/06/07 23:12:52  jongfoster
285  *    Replacing function pointer in struct gateway with a directly
286  *    called function forwarded_connect().
287  *    Replacing struct gateway with struct forward_spec
288  *
289  *    Revision 1.18  2001/06/03 19:12:16  oes
290  *    introduced new cgi handling
291  *
292  *    Revision 1.17  2001/06/01 20:07:23  jongfoster
293  *    Now uses action +image-blocker{} rather than config->tinygif
294  *
295  *    Revision 1.16  2001/06/01 18:49:17  jongfoster
296  *    Replaced "list_share" with "list" - the tiny memory gain was not
297  *    worth the extra complexity.
298  *
299  *    Revision 1.15  2001/05/31 21:24:47  jongfoster
300  *    Changed "permission" to "action" throughout.
301  *    Removed DEFAULT_USER_AGENT - it must now be specified manually.
302  *    Moved vanilla wafer check into chat(), since we must now
303  *    decide whether or not to add it based on the URL.
304  *
305  *    Revision 1.14  2001/05/29 20:14:01  joergs
306  *    AmigaOS bugfix: PCRS needs a lot of stack, stacksize for child threads
307  *    increased.
308  *
309  *    Revision 1.13  2001/05/29 09:50:24  jongfoster
310  *    Unified blocklist/imagelist/permissionslist.
311  *    File format is still under discussion, but the internal changes
312  *    are (mostly) done.
313  *
314  *    Also modified interceptor behaviour:
315  *    - We now intercept all URLs beginning with one of the following
316  *      prefixes (and *only* these prefixes):
317  *        * http://i.j.b/
318  *        * http://ijbswa.sf.net/config/
319  *        * http://ijbswa.sourceforge.net/config/
320  *    - New interceptors "home page" - go to http://i.j.b/ to see it.
321  *    - Internal changes so that intercepted and fast redirect pages
322  *      are not replaced with an image.
323  *    - Interceptors now have the option to send a binary page direct
324  *      to the client. (i.e. ijb-send-banner uses this)
325  *    - Implemented show-url-info interceptor.  (Which is why I needed
326  *      the above interceptors changes - a typical URL is
327  *      "http://i.j.b/show-url-info?url=www.somesite.com/banner.gif".
328  *      The previous mechanism would not have intercepted that, and
329  *      if it had been intercepted then it then it would have replaced
330  *      it with an image.)
331  *
332  *    Revision 1.12  2001/05/27 22:17:04  oes
333  *
334  *    - re_process_buffer no longer writes the modified buffer
335  *      to the client, which was very ugly. It now returns the
336  *      buffer, which it is then written by chat.
337  *
338  *    - content_length now adjusts the Content-Length: header
339  *      for modified documents rather than crunch()ing it.
340  *      (Length info in csp->content_length, which is 0 for
341  *      unmodified documents)
342  *
343  *    - For this to work, sed() is called twice when filtering.
344  *
345  *    Revision 1.11  2001/05/26 17:27:53  jongfoster
346  *    Added support for CLF and fixed LOG_LEVEL_LOG.
347  *    Also did CRLF->LF fix of my previous patch.
348  *
349  *    Revision 1.10  2001/05/26 15:26:15  jongfoster
350  *    ACL feature now provides more security by immediately dropping
351  *    connections from untrusted hosts.
352  *
353  *    Revision 1.9  2001/05/26 00:28:36  jongfoster
354  *    Automatic reloading of config file.
355  *    Removed obsolete SIGHUP support (Unix) and Reload menu option (Win32).
356  *    Most of the global variables have been moved to a new
357  *    struct configuration_spec, accessed through csp->config->globalname
358  *    Most of the globals remaining are used by the Win32 GUI.
359  *
360  *    Revision 1.8  2001/05/25 22:43:18  jongfoster
361  *    Fixing minor memory leak and buffer overflow.
362  *
363  *    Revision 1.7  2001/05/25 22:34:30  jongfoster
364  *    Hard tabs->Spaces
365  *
366  *    Revision 1.6  2001/05/23 00:13:58  joergs
367  *    AmigaOS support fixed.
368  *
369  *    Revision 1.5  2001/05/22 18:46:04  oes
370  *
371  *    - Enabled filtering banners by size rather than URL
372  *      by adding patterns that replace all standard banner
373  *      sizes with the "Junkbuster" gif to the re_filterfile
374  *
375  *    - Enabled filtering WebBugs by providing a pattern
376  *      which kills all 1x1 images
377  *
378  *    - Added support for PCRE_UNGREEDY behaviour to pcrs,
379  *      which is selected by the (nonstandard and therefore
380  *      capital) letter 'U' in the option string.
381  *      It causes the quantifiers to be ungreedy by default.
382  *      Appending a ? turns back to greedy (!).
383  *
384  *    - Added a new interceptor ijb-send-banner, which
385  *      sends back the "Junkbuster" gif. Without imagelist or
386  *      MSIE detection support, or if tinygif = 1, or the
387  *      URL isn't recognized as an imageurl, a lame HTML
388  *      explanation is sent instead.
389  *
390  *    - Added new feature, which permits blocking remote
391  *      script redirects and firing back a local redirect
392  *      to the browser.
393  *      The feature is conditionally compiled, i.e. it
394  *      can be disabled with --disable-fast-redirects,
395  *      plus it must be activated by a "fast-redirects"
396  *      line in the config file, has its own log level
397  *      and of course wants to be displayed by show-proxy-args
398  *      Note: Boy, all the #ifdefs in 1001 locations and
399  *      all the fumbling with configure.in and acconfig.h
400  *      were *way* more work than the feature itself :-(
401  *
402  *    - Because a generic redirect template was needed for
403  *      this, tinygif = 3 now uses the same.
404  *
405  *    - Moved GIFs, and other static HTTP response templates
406  *      to project.h
407  *
408  *    - Some minor fixes
409  *
410  *    - Removed some >400 CRs again (Jon, you really worked
411  *      a lot! ;-)
412  *
413  *    Revision 1.4  2001/05/21 19:34:01  jongfoster
414  *    Made failure to bind() a fatal error.
415  *
416  *    Revision 1.3  2001/05/20 01:21:20  jongfoster
417  *    Version 2.9.4 checkin.
418  *    - Merged popupfile and cookiefile, and added control over PCRS
419  *      filtering, in new "permissionsfile".
420  *    - Implemented LOG_LEVEL_FATAL, so that if there is a configuration
421  *      file error you now get a message box (in the Win32 GUI) rather
422  *      than the program exiting with no explanation.
423  *    - Made killpopup use the PCRS MIME-type checking and HTTP-header
424  *      skipping.
425  *    - Removed tabs from "config"
426  *    - Moved duplicated url parsing code in "loaders.c" to a new funcition.
427  *    - Bumped up version number.
428  *
429  *    Revision 1.2  2001/05/17 22:34:44  oes
430  *     - Added hint on GIF char array generation to jcc.c
431  *     - Cleaned CRLF's from the sources and related files
432  *     - Repaired logging for REF and FRC
433  *
434  *    Revision 1.1.1.1  2001/05/15 13:58:56  oes
435  *    Initial import of version 2.9.3 source tree
436  *
437  *
438  *********************************************************************/
439 \f
440
441 #include "config.h"
442
443 #include <stdio.h>
444 #include <sys/types.h>
445 #include <stdlib.h>
446 #include <string.h>
447 #include <signal.h>
448 #include <fcntl.h>
449 #include <errno.h>
450
451 #ifdef FEATURE_PTHREAD
452 #include <pthread.h>
453 #endif /* def FEATURE_PTHREAD */
454
455 #ifdef _WIN32
456 # ifndef FEATURE_PTHREAD
457 #  include <windows.h>
458 #  include <process.h>
459 # endif /* ndef FEATURE_PTHREAD */
460
461 # include "win32.h"
462 # ifndef _WIN_CONSOLE
463 #  include "w32log.h"
464 # endif /* ndef _WIN_CONSOLE */
465
466 #else /* ifndef _WIN32 */
467
468 # if !defined (__OS2__)
469 # include <unistd.h>
470 # include <sys/wait.h>
471 # endif /* ndef __OS2__ */
472 # include <sys/time.h>
473 # include <sys/stat.h>
474 # include <sys/ioctl.h>
475
476 #ifdef sun
477 #include <sys/termios.h>
478 #endif /* sun */
479
480 #ifdef unix
481 #include <pwd.h>
482 #endif
483
484 # include <signal.h>
485
486 # ifdef __BEOS__
487 #  include <socket.h>  /* BeOS has select() for sockets only. */
488 #  include <OS.h>      /* declarations for threads and stuff. */
489 # endif
490
491 # if defined(__EMX__) || defined(__OS2__)
492 #  include <sys/select.h>  /* OS/2/EMX needs a little help with select */
493 # endif
494 # ifdef __OS2__
495 #define INCL_DOS
496 # include <os2.h>
497 #define bzero(B,N) memset(B,0x00,n)
498 # endif
499
500 # ifndef FD_ZERO
501 #  include <select.h>
502 # endif
503
504 #endif
505
506 #include "project.h"
507 #include "list.h"
508 #include "jcc.h"
509 #include "filters.h"
510 #include "loaders.h"
511 #include "parsers.h"
512 #include "killpopup.h"
513 #include "miscutil.h"
514 #include "errlog.h"
515 #include "jbsockets.h"
516 #include "gateway.h"
517 #include "actions.h"
518 #include "cgi.h"
519 #include "loadcfg.h"
520 #include "urlmatch.h"
521
522 const char jcc_h_rcs[] = JCC_H_VERSION;
523 const char project_h_rcs[] = PROJECT_H_VERSION;
524
525 int no_daemon = 0;
526 struct client_state  clients[1];
527 struct file_list     files[1];
528
529 #ifdef FEATURE_STATISTICS
530 int urls_read     = 0;     /* total nr of urls read inc rejected */
531 int urls_rejected = 0;     /* total nr of urls rejected */
532 #endif /* def FEATURE_STATISTICS */
533
534
535 static void listen_loop(void);
536 static void chat(struct client_state *csp);
537 #ifdef AMIGA
538 void serve(struct client_state *csp);
539 #else /* ifndef AMIGA */
540 static void serve(struct client_state *csp);
541 #endif /* def AMIGA */
542
543 #ifdef __BEOS__
544 static int32 server_thread(void *data);
545 #endif /* def __BEOS__ */
546
547 #ifdef _WIN32
548 #define sleep(N)  Sleep(((N) * 1000))
549 #endif
550
551 #ifdef __OS2__
552 #define sleep(N)  DosSleep(((N) * 100))
553 #endif
554
555 #if defined(unix)
556 const char *basedir;
557 const char *pidfile = NULL;
558 int received_hup_signal = 0;
559 #endif /* defined unix */
560
561 /* The vanilla wafer. */
562 static const char VANILLA_WAFER[] =
563    "NOTICE=TO_WHOM_IT_MAY_CONCERN_"
564    "Do_not_send_me_any_copyrighted_information_other_than_the_"
565    "document_that_I_am_requesting_or_any_of_its_necessary_components._"
566    "In_particular_do_not_send_me_any_cookies_that_"
567    "are_subject_to_a_claim_of_copyright_by_anybody._"
568    "Take_notice_that_I_refuse_to_be_bound_by_any_license_condition_"
569    "(copyright_or_otherwise)_applying_to_any_cookie._";
570
571
572 #if !defined(_WIN32) && !defined(__OS2__) && !defined(AMIGA)
573 /*********************************************************************
574  *
575  * Function    :  sig_handler 
576  *
577  * Description :  Signal handler for different signals.
578  *                Exit gracefully on ABRT, TERM and  INT
579  *                or set a flag that will cause the errlog
580  *                to be reopened by the main thread on HUP.
581  *
582  * Parameters  :
583  *          1  :  the_signal - the signal cause this function to call 
584  *
585  * Returns     :  - 
586  *
587  *********************************************************************/
588 static void sig_handler(int the_signal)
589 {
590    switch(the_signal)
591    {
592       case SIGABRT:
593       case SIGTERM:
594       case SIGINT:
595          log_error(LOG_LEVEL_INFO, "exiting by signal %d .. bye", the_signal);
596 #if defined(unix)
597          unlink(pidfile);
598 #endif /* unix */
599          exit(the_signal);
600          break;
601
602       case SIGHUP:
603          received_hup_signal = 1;
604          break;         
605
606       default:
607          /* 
608           * We shouldn't be here, unless we catch signals
609           * in main() that we can't handle here!
610           */
611          log_error(LOG_LEVEL_FATAL, "sig_handler: exiting on unexpected signal %d", the_signal);
612    }
613    return;
614
615 }
616 #endif
617
618
619 /*********************************************************************
620  *
621  * Function    :  chat
622  *
623  * Description :  Once a connection to the client has been accepted,
624  *                this function is called (via serve()) to handle the
625  *                main business of the communication.  When this
626  *                function returns, the caller must close the client
627  *                socket handle.
628  *
629  * Parameters  :
630  *          1  :  csp = Current client state (buffers, headers, etc...)
631  *
632  * Returns     :  On success, the number of bytes written are returned (zero
633  *                indicates nothing was written).  On error, -1 is returned,
634  *                and errno is set appropriately.  If count is zero and the
635  *                file descriptor refers to a regular file, 0 will be
636  *                returned without causing any other effect.  For a special
637  *                file, the results are not portable.
638  *
639  *********************************************************************/
640 static void chat(struct client_state *csp)
641 {
642 /*
643  * This next lines are a little ugly, but they simplifies the if statements
644  * below.  Basically if TOGGLE, then we want the if to test if the
645  * CSP_FLAG_TOGGLED_ON flag ist set, else we don't.  And if FEATURE_FORCE_LOAD,
646  * then we want the if to test for CSP_FLAG_FORCED , else we don't
647  */
648 #ifdef FEATURE_TOGGLE
649 #   define IS_TOGGLED_ON_AND (csp->flags & CSP_FLAG_TOGGLED_ON) &&
650 #else /* ifndef FEATURE_TOGGLE */
651 #   define IS_TOGGLED_ON_AND
652 #endif /* ndef FEATURE_TOGGLE */
653 #ifdef FEATURE_FORCE_LOAD
654 #   define IS_NOT_FORCED_AND !(csp->flags & CSP_FLAG_FORCED) &&
655 #else /* ifndef FEATURE_FORCE_LOAD */
656 #   define IS_NOT_FORCED_AND
657 #endif /* def FEATURE_FORCE_LOAD */
658
659 #define IS_ENABLED_AND   IS_TOGGLED_ON_AND IS_NOT_FORCED_AND
660
661    char buf[BUFFER_SIZE];
662    char *hdr, *p, *req;
663    fd_set rfds;
664    int n, maxfd, server_body;
665    int ms_iis5_hack = 0;
666    int byte_count = 0;
667    const struct forward_spec * fwd;
668    struct http_request *http;
669 #ifdef FEATURE_KILL_POPUPS
670    int block_popups;         /* bool, 1==will block popups */
671    int block_popups_now = 0; /* bool, 1==currently blocking popups */
672 #endif /* def FEATURE_KILL_POPUPS */
673
674    int pcrs_filter;        /* bool, 1==will filter through pcrs */
675    int gif_deanimate;      /* bool, 1==will deanimate gifs */
676
677    /* Function that does the content filtering for the current request */
678    char *(*content_filter)() = NULL;
679
680    /* Skeleton for HTTP response, if we should intercept the request */
681    struct http_response *rsp;
682
683    http = csp->http;
684
685    /*
686     * Read the client's request.  Note that since we're not using select() we
687     * could get blocked here if a client connected, then didn't say anything!
688     */
689
690    while (FOREVER)
691    {
692       n = read_socket(csp->cfd, buf, sizeof(buf));
693
694       if (n <= 0) break;      /* error! */
695
696       add_to_iob(csp, buf, n);
697
698       req = get_header(csp);
699
700       if (req == NULL)
701       {
702          break;    /* no HTTP request! */
703       }
704
705       if (*req == '\0')
706       {
707          continue;   /* more to come! */
708       }
709
710 #ifdef FEATURE_FORCE_LOAD
711       /* If this request contains the FORCE_PREFIX,
712        * better get rid of it now and set the force flag --oes
713        */
714
715       if (strstr(req, FORCE_PREFIX))
716       {
717          strclean(req, FORCE_PREFIX);
718          log_error(LOG_LEVEL_FORCE, "Enforcing request \"%s\".\n", req);
719          csp->flags |= CSP_FLAG_FORCED;
720       }
721
722 #endif /* def FEATURE_FORCE_LOAD */
723
724       parse_http_request(req, http, csp);
725       freez(req);
726       break;
727    }
728
729    if (http->cmd == NULL)
730    {
731       strcpy(buf, CHEADER);
732       write_socket(csp->cfd, buf, strlen(buf));
733
734       log_error(LOG_LEVEL_CLF, "%s - - [%T] \" \" 400 0", csp->ip_addr_str);
735
736       return;
737    }
738
739    /* decide how to route the HTTP request */
740
741    if ((fwd = forward_url(http, csp)) == NULL)
742    {
743       log_error(LOG_LEVEL_FATAL, "gateway spec is NULL!?!?  This can't happen!");
744       /* Never get here - LOG_LEVEL_FATAL causes program exit */
745    }
746
747    /* build the http request to send to the server
748     * we have to do one of the following:
749     *
750     * create = use the original HTTP request to create a new
751     *          HTTP request that has either the path component
752     *          without the http://domainspec (w/path) or the
753     *          full orininal URL (w/url)
754     *          Note that the path and/or the HTTP version may
755     *          have been altered by now.
756     *
757     * connect = Open a socket to the host:port of the server
758     *           and short-circuit server and client socket.
759     *
760     * pass =  Pass the request unchanged if forwarding a CONNECT
761     *         request to a parent proxy. Note that we'll be sending
762     *         the CFAIL message ourselves if connecting to the parent
763     *         fails, but we won't send a CSUCCEED message if it works,
764     *         since that would result in a double message (ours and the
765     *         parent's). After sending the request to the parent, we simply
766     *         tunnel.
767     *
768     * here's the matrix:
769     *                        SSL
770     *                    0        1
771     *                +--------+--------+
772     *                |        |        |
773     *             0  | create | connect|
774     *                | w/path |        |
775     *  Forwarding    +--------+--------+
776     *                |        |        |
777     *             1  | create | pass   |
778     *                | w/url  |        |
779     *                +--------+--------+
780     *
781     */
782
783    /*
784     * Determine the actions for this URL
785     */
786 #ifdef FEATURE_TOGGLE
787    if (!(csp->flags & CSP_FLAG_TOGGLED_ON))
788    {
789       /* Most compatible set of actions (i.e. none) */
790       init_current_action(csp->action);
791    }
792    else
793 #endif /* ndef FEATURE_TOGGLE */
794    {
795       url_actions(http, csp);
796    }
797
798
799    /*
800     * Check if a CONNECT request is allowable:
801     * In the absence of a +limit-connect action, allow only port 443.
802     * If there is an action, allow whatever matches the specificaton.
803     */
804    if(http->ssl)
805    {
806       if(  ( !(csp->action->flags & ACTION_LIMIT_CONNECT) && csp->http->port != 443)
807            || (csp->action->flags & ACTION_LIMIT_CONNECT
808               && !match_portlist(csp->action->string[ACTION_STRING_LIMIT_CONNECT], csp->http->port)) )
809       {
810          strcpy(buf, CFORBIDDEN);
811          write_socket(csp->cfd, buf, strlen(buf));
812
813          log_error(LOG_LEVEL_CONNECT, "Denying suspicious CONNECT request from %s", csp->ip_addr_str);
814          log_error(LOG_LEVEL_CLF, "%s - - [%T] \" \" 403 0", csp->ip_addr_str);
815
816          return;
817       }
818    }
819
820
821    /*
822     * Downgrade http version from 1.1 to 1.0 if +downgrade
823     * action applies
824     */
825    if ( (http->ssl == 0)
826      && (!strcmpic(http->ver, "HTTP/1.1"))
827      && (csp->action->flags & ACTION_DOWNGRADE))
828    {
829       freez(http->ver);
830       http->ver = strdup("HTTP/1.0");
831
832       if (http->ver == NULL)
833       {
834          log_error(LOG_LEVEL_FATAL, "Out of memory downgrading HTTP version");
835       }
836    }
837
838    /* 
839     * Save a copy of the original request for logging
840     */
841    http->ocmd = strdup(http->cmd);
842
843    if (http->ocmd == NULL)
844    {
845       log_error(LOG_LEVEL_FATAL, "Out of memory copying HTTP request line");
846    }
847
848    /*
849     * (Re)build the HTTP request for non-SSL requests.
850     * If forwarding, use the whole URL, else, use only the path.
851     */
852    if (http->ssl == 0)
853    {
854       freez(http->cmd);
855
856       http->cmd = strdup(http->gpc);
857       string_append(&http->cmd, " ");
858
859       if (fwd->forward_host)
860       {
861          string_append(&http->cmd, http->url);
862       }
863       else
864       {
865          string_append(&http->cmd, http->path);
866       }
867
868       string_append(&http->cmd, " ");
869       string_append(&http->cmd, http->ver);
870
871       if (http->cmd == NULL)
872       {
873          log_error(LOG_LEVEL_FATAL, "Out of memory rewiting SSL command");
874       }
875    }
876    enlist(csp->headers, http->cmd);
877
878
879 #ifdef FEATURE_COOKIE_JAR
880    /*
881     * If we're logging cookies in a cookie jar, and the user has not
882     * supplied any wafers, and the user has not told us to suppress the
883     * vanilla wafer, then send the vanilla wafer.
884     */
885    if ((csp->config->jarfile != NULL)
886        && list_is_empty(csp->action->multi[ACTION_MULTI_WAFER])
887        && ((csp->action->flags & ACTION_VANILLA_WAFER) != 0))
888    {
889       enlist(csp->action->multi[ACTION_MULTI_WAFER], VANILLA_WAFER);
890    }
891 #endif /* def FEATURE_COOKIE_JAR */
892
893
894 #ifdef FEATURE_KILL_POPUPS
895    block_popups               = ((csp->action->flags & ACTION_NO_POPUPS) != 0);
896 #endif /* def FEATURE_KILL_POPUPS */
897
898    pcrs_filter                = (csp->rlist != NULL) &&  /* There are expressions to be used */
899                                 ((csp->action->flags & ACTION_FILTER) != 0);
900
901    gif_deanimate              = ((csp->action->flags & ACTION_DEANIMATE) != 0);
902
903    /* grab the rest of the client's headers */
904
905    while (FOREVER)
906    {
907       if ( ( p = get_header(csp) ) && ( *p == '\0' ) )
908       {
909          n = read_socket(csp->cfd, buf, sizeof(buf));
910          if (n <= 0)
911          {
912             log_error(LOG_LEVEL_ERROR, "read from client failed: %E");
913             return;
914          }
915          add_to_iob(csp, buf, n);
916          continue;
917       }
918
919       if (p == NULL) break;
920
921       enlist(csp->headers, p);
922       freez(p);
923    }
924    /*
925     * We have a request. Now, check to see if we need to
926     * intercept it, i.e. If ..
927     */
928
929    if (
930        /* a CGI call was detected and answered */
931        (NULL != (rsp = dispatch_cgi(csp)))
932
933        /* or we are enabled and... */
934        || (IS_ENABLED_AND (
935
936             /* ..the request was blocked */
937           ( NULL != (rsp = block_url(csp)))
938
939           /* ..or untrusted */
940 #ifdef FEATURE_TRUST
941           || ( NULL != (rsp = trust_url(csp)))
942 #endif /* def FEATURE_TRUST */
943
944           /* ..or a fast redirect kicked in */
945 #ifdef FEATURE_FAST_REDIRECTS
946           || (((csp->action->flags & ACTION_FAST_REDIRECTS) != 0) &&
947                 (NULL != (rsp = redirect_url(csp))))
948 #endif /* def FEATURE_FAST_REDIRECTS */
949           ))
950       )
951    {
952       /* Write the answer to the client */
953       if ((write_socket(csp->cfd, rsp->head, rsp->head_length) != rsp->head_length)
954            || (write_socket(csp->cfd, rsp->body, rsp->content_length) != rsp->content_length))
955       {
956          log_error(LOG_LEVEL_ERROR, "write to: %s failed: %E", http->host);
957       }
958
959 #ifdef FEATURE_STATISTICS
960       /* Count as a rejected request */
961       csp->flags |= CSP_FLAG_REJECTED;
962 #endif /* def FEATURE_STATISTICS */
963
964       /* Log (FIXME: All intercept reasons apprear as "crunch" with Status 200) */
965       log_error(LOG_LEVEL_GPC, "%s%s crunch!", http->hostport, http->path);
966       log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 200 3", csp->ip_addr_str, http->ocmd);
967
968       /* Clean up and return */
969       free_http_response(rsp);
970       return;
971    }
972
973    log_error(LOG_LEVEL_GPC, "%s%s", http->hostport, http->path);
974
975    if (fwd->forward_host)
976    {
977       log_error(LOG_LEVEL_CONNECT, "via %s:%d to: %s",
978                fwd->forward_host, fwd->forward_port, http->hostport);
979    }
980    else
981    {
982       log_error(LOG_LEVEL_CONNECT, "to %s", http->hostport);
983    }
984
985    /* here we connect to the server, gateway, or the forwarder */
986
987    csp->sfd = forwarded_connect(fwd, http, csp);
988
989    if (csp->sfd < 0)
990    {
991       log_error(LOG_LEVEL_CONNECT, "connect to: %s failed: %E",
992                 http->hostport);
993
994       if (errno == EINVAL)
995       {
996          rsp = error_response(csp, "no-such-domain", errno);
997
998          log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 404 0",
999                    csp->ip_addr_str, http->ocmd);
1000       }
1001       else
1002       {
1003          rsp = error_response(csp, "connect-failed", errno);
1004
1005          log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 503 0",
1006                    csp->ip_addr_str, http->ocmd);
1007       }
1008
1009
1010       /* Write the answer to the client */
1011       if(rsp)
1012       {
1013          if ((write_socket(csp->cfd, rsp->head, rsp->head_length) != rsp->head_length)
1014           || (write_socket(csp->cfd, rsp->body, rsp->content_length) != rsp->content_length))
1015          {
1016             log_error(LOG_LEVEL_ERROR, "write to: %s failed: %E", http->host);
1017          }
1018       }
1019
1020       free_http_response(rsp);
1021       return;
1022    }
1023
1024    log_error(LOG_LEVEL_CONNECT, "OK");
1025
1026    hdr = sed(client_patterns, add_client_headers, csp);
1027    if (hdr == NULL)
1028    {
1029       /* FIXME Should handle error properly */
1030       log_error(LOG_LEVEL_FATAL, "Out of memory parsing client header");
1031    }
1032
1033    list_remove_all(csp->headers);
1034
1035    if (fwd->forward_host || (http->ssl == 0))
1036    {
1037       /* write the client's (modified) header to the server
1038        * (along with anything else that may be in the buffer)
1039        */
1040
1041       n = strlen(hdr);
1042
1043       if ((write_socket(csp->sfd, hdr, n) != n)
1044           || (flush_socket(csp->sfd, csp   ) <  0))
1045       {
1046          log_error(LOG_LEVEL_CONNECT, "write header to: %s failed: %E",
1047                     http->hostport);
1048
1049          log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 503 0",
1050                    csp->ip_addr_str, http->ocmd);
1051
1052          rsp = error_response(csp, "connect-failed", errno);
1053
1054          if(rsp)
1055          {
1056             if ((write_socket(csp->cfd, rsp->head, rsp->head_length) != rsp->head_length)
1057              || (write_socket(csp->cfd, rsp->body, rsp->content_length) != rsp->content_length))
1058             {
1059                log_error(LOG_LEVEL_ERROR, "write to: %s failed: %E", http->host);
1060             }
1061          }
1062
1063          free_http_response(rsp);
1064          freez(hdr);
1065          return;
1066       }
1067    }
1068    else
1069    {
1070       /*
1071        * We're running an SSL tunnel and we're not forwarding,
1072        * so just send the "connect succeeded" message to the
1073        * client, flush the rest, and get out of the way.
1074        */
1075       log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 200 2\n",
1076                 csp->ip_addr_str, http->ocmd);
1077
1078       if (write_socket(csp->cfd, CSUCCEED, sizeof(CSUCCEED)-1) < 0)
1079       {
1080          freez(hdr);
1081          return;
1082       }
1083       IOB_RESET(csp);
1084    }
1085
1086    /* we're finished with the client's header */
1087    freez(hdr);
1088
1089    maxfd = ( csp->cfd > csp->sfd ) ? csp->cfd : csp->sfd;
1090
1091    /* pass data between the client and server
1092     * until one or the other shuts down the connection.
1093     */
1094
1095    server_body = 0;
1096
1097    while (FOREVER)
1098    {
1099       FD_ZERO(&rfds);
1100
1101       FD_SET(csp->cfd, &rfds);
1102       FD_SET(csp->sfd, &rfds);
1103
1104       n = select(maxfd+1, &rfds, NULL, NULL, NULL);
1105
1106       if (n < 0)
1107       {
1108          log_error(LOG_LEVEL_ERROR, "select() failed!: %E");
1109          return;
1110       }
1111
1112       /* this is the body of the browser's request
1113        * just read it and write it.
1114        */
1115
1116       if (FD_ISSET(csp->cfd, &rfds))
1117       {
1118          n = read_socket(csp->cfd, buf, sizeof(buf));
1119
1120          if (n <= 0)
1121          {
1122             break; /* "game over, man" */
1123          }
1124
1125          if (write_socket(csp->sfd, buf, n) != n)
1126          {
1127             log_error(LOG_LEVEL_ERROR, "write to: %s failed: %E", http->host);
1128             return;
1129          }
1130          continue;
1131       }
1132
1133       /*
1134        * The server wants to talk.  It could be the header or the body.
1135        * If `hdr' is null, then it's the header otherwise it's the body.
1136        * FIXME: Does `hdr' really mean `host'? No.
1137        */
1138
1139
1140       if (FD_ISSET(csp->sfd, &rfds))
1141       {
1142          fflush( 0 );
1143          n = read_socket(csp->sfd, buf, sizeof(buf) - 1);
1144
1145          if (n < 0)
1146          {
1147             log_error(LOG_LEVEL_ERROR, "read from: %s failed: %E", http->host);
1148
1149             log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 503 0",
1150                       csp->ip_addr_str, http->ocmd);
1151
1152             rsp = error_response(csp, "connect-failed", errno);
1153
1154             if(rsp)
1155             {
1156                if ((write_socket(csp->cfd, rsp->head, rsp->head_length) != rsp->head_length)
1157                 || (write_socket(csp->cfd, rsp->body, rsp->content_length) != rsp->content_length))
1158                {
1159                   log_error(LOG_LEVEL_ERROR, "write to: %s failed: %E", http->host);
1160                }
1161             }
1162
1163             free_http_response(rsp);
1164             return;
1165          }
1166
1167          /* Add a trailing zero.  This lets filter_popups
1168           * use string operations.
1169           */
1170          buf[n] = '\0';
1171
1172 #ifdef FEATURE_KILL_POPUPS
1173          /* Filter the popups on this read. */
1174          if (block_popups_now)
1175          {
1176             filter_popups(buf, csp);
1177          }
1178 #endif /* def FEATURE_KILL_POPUPS */
1179
1180          /* Normally, this would indicate that we've read
1181           * as much as the server has sent us and we can
1182           * close the client connection.  However, Microsoft
1183           * in its wisdom has released IIS/5 with a bug that
1184           * prevents it from sending the trailing \r\n in
1185           * a 302 redirect header (and possibly other headers).
1186           * To work around this if we've haven't parsed
1187           * a full header we'll append a trailing \r\n
1188           * and see if this now generates a valid one.
1189           *
1190           * This hack shouldn't have any impacts.  If we've
1191           * already transmitted the header or if this is a
1192           * SSL connection, then we won't bother with this
1193           * hack.  So we only work on partially received
1194           * headers.  If we append a \r\n and this still
1195           * doesn't generate a valid header, then we won't
1196           * transmit anything to the client.
1197           */
1198          if (n == 0)
1199          {
1200
1201             if (server_body || http->ssl)
1202             {
1203                /*
1204                 * If we have been buffering up the document,
1205                 * now is the time to apply content modification
1206                 * and send the result to the client.
1207                 */
1208                if (content_filter)
1209                {
1210                   /*
1211                    * If the content filter fails, use the original
1212                    * buffer and length.
1213                    * (see p != NULL ? p : csp->iob->cur below)
1214                    */
1215                   if (NULL == (p = (*content_filter)(csp)))
1216                   {
1217                      csp->content_length = csp->iob->eod - csp->iob->cur;
1218                   }
1219
1220                   hdr = sed(server_patterns, add_server_headers, csp);
1221                   if (hdr == NULL)
1222                   {
1223                      /* FIXME Should handle error properly */
1224                      log_error(LOG_LEVEL_FATAL, "Out of memory parsing server header");
1225                   }
1226
1227                   n = strlen(hdr);
1228
1229                   if ((write_socket(csp->cfd, hdr, n) != n)
1230                       || (write_socket(csp->cfd, p != NULL ? p : csp->iob->cur, csp->content_length) != (int)csp->content_length))
1231                   {
1232                      log_error(LOG_LEVEL_ERROR, "write modified content to client failed: %E");
1233                      return;
1234                   }
1235
1236                   freez(hdr);
1237                   if (NULL != p) {
1238                      freez(p);
1239                   }
1240                }
1241
1242                break; /* "game over, man" */
1243             }
1244
1245             /*
1246              * This is NOT the body, so
1247              * Let's pretend the server just sent us a blank line.
1248              */
1249             n = sprintf(buf, "\r\n");
1250
1251             /*
1252              * Now, let the normal header parsing algorithm below do its
1253              * job.  If it fails, we'll exit instead of continuing.
1254              */
1255
1256             ms_iis5_hack = 1;
1257          }
1258
1259          /*
1260           * If this is an SSL connection or we're in the body
1261           * of the server document, just write it to the client,
1262           * unless we need to buffer the body for later content-filtering
1263           */
1264
1265          if (server_body || http->ssl)
1266          {
1267             if (content_filter)
1268             {
1269                add_to_iob(csp, buf, n);
1270
1271                /*
1272                 * If the buffer limit will be reached on the next read,
1273                 * switch to non-filtering mode, i.e. make & write the
1274                 * header, flush the socket and get out of the way.
1275                 */
1276                if (((size_t)(csp->iob->eod - csp->iob->buf)) + (size_t)BUFFER_SIZE > csp->config->buffer_limit)
1277                {
1278                   log_error(LOG_LEVEL_ERROR, "Buffer size limit reached! Flushing and stepping back.");
1279
1280                   hdr = sed(server_patterns, add_server_headers, csp);
1281                   if (hdr == NULL)
1282                   {
1283                      /* FIXME Should handle error properly */
1284                      log_error(LOG_LEVEL_FATAL, "Out of memory parsing server header");
1285                   }
1286
1287                   n   = strlen(hdr);
1288                   byte_count += n;
1289
1290                   if (((write_socket(csp->cfd, hdr, n) != n)
1291                        || (n = flush_socket(csp->cfd, csp) < 0)))
1292                   {
1293                      log_error(LOG_LEVEL_CONNECT, "write header to client failed: %E");
1294
1295                      freez(hdr);
1296                      return;
1297                   }
1298
1299                   freez(hdr);
1300                   byte_count += n;
1301
1302                   content_filter = NULL;
1303                   server_body = 1;
1304
1305                }
1306             }
1307             else
1308             {
1309                if (write_socket(csp->cfd, buf, n) != n)
1310                {
1311                   log_error(LOG_LEVEL_ERROR, "write to client failed: %E");
1312                   return;
1313                }
1314             }
1315             byte_count += n;
1316             continue;
1317          }
1318          else
1319          {
1320             /* we're still looking for the end of the
1321              * server's header ... (does that make header
1322              * parsing an "out of body experience" ?
1323              */
1324
1325             /* buffer up the data we just read */
1326             add_to_iob(csp, buf, n);
1327
1328             /* get header lines from the iob */
1329
1330             while ((p = get_header(csp)))
1331             {
1332                if (*p == '\0')
1333                {
1334                   /* see following note */
1335                   break;
1336                }
1337                enlist(csp->headers, p);
1338                freez(p);
1339             }
1340
1341             /* NOTE: there are no "empty" headers so
1342              * if the pointer `p' is not NULL we must
1343              * assume that we reached the end of the
1344              * buffer before we hit the end of the header.
1345              */
1346
1347             if (p)
1348             {
1349                if (ms_iis5_hack)
1350                {
1351                   /* Well, we tried our MS IIS/5
1352                    * hack and it didn't work.
1353                    * The header is incomplete
1354                    * and there isn't anything
1355                    * we can do about it.
1356                    */
1357                   break;
1358                }
1359                else
1360                {
1361                   /* Since we have to wait for
1362                    * more from the server before
1363                    * we can parse the headers
1364                    * we just continue here.
1365                    */
1366                   continue;
1367                }
1368             }
1369
1370             /* we have now received the entire header.
1371              * filter it and send the result to the client
1372              */
1373
1374             hdr = sed(server_patterns, add_server_headers, csp);
1375             if (hdr == NULL)
1376             {
1377                /* FIXME Should handle error properly */
1378                log_error(LOG_LEVEL_FATAL, "Out of memory parsing server header");
1379             }
1380
1381             n   = strlen(hdr);
1382
1383             /* write the server's (modified) header to
1384              * the client (along with anything else that
1385              * may be in the buffer)
1386              */
1387
1388 #ifdef FEATURE_KILL_POPUPS
1389             /* Start blocking popups if appropriate. */
1390
1391             if ((csp->content_type & CT_TEXT) &&  /* It's a text / * MIME-Type */
1392                 !http->ssl    &&                  /* We talk plaintext */
1393                 block_popups)                     /* Policy allows */
1394             {
1395                block_popups_now = 1;
1396                /*
1397                 * Filter the part of the body that came in the same read
1398                 * as the last headers:
1399                 */
1400                filter_popups(csp->iob->cur, csp);
1401             }
1402
1403 #endif /* def FEATURE_KILL_POPUPS */
1404
1405             /* Buffer and pcrs filter this if appropriate. */
1406
1407             if ((csp->content_type & CT_TEXT) &&  /* It's a text / * MIME-Type */
1408                 !http->ssl    &&                  /* We talk plaintext */
1409                 pcrs_filter)                      /* Policy allows */
1410             {
1411                content_filter = pcrs_filter_response;
1412             }
1413
1414             /* Buffer and gif_deanimate this if appropriate. */
1415
1416             if ((csp->content_type & CT_GIF)  &&  /* It's a image/gif MIME-Type */
1417                 !http->ssl    &&                  /* We talk plaintext */
1418                 gif_deanimate)                    /* Policy allows */
1419             {
1420                content_filter = gif_deanimate_response;
1421             }
1422
1423
1424             /*
1425              * Only write if we're not buffering for content modification
1426              */
1427             if (!content_filter && ((write_socket(csp->cfd, hdr, n) != n)
1428                 || (n = flush_socket(csp->cfd, csp) < 0)))
1429             {
1430                log_error(LOG_LEVEL_CONNECT, "write header to client failed: %E");
1431
1432                /* the write failed, so don't bother
1433                 * mentioning it to the client...
1434                 * it probably can't hear us anyway.
1435                 */
1436                freez(hdr);
1437                return;
1438             }
1439
1440             if(!content_filter) byte_count += n;
1441
1442             /* we're finished with the server's header */
1443
1444             freez(hdr);
1445             server_body = 1;
1446
1447             /* If this was a MS IIS/5 hack then it means
1448              * the server has already closed the
1449              * connection.  Nothing more to read.  Time
1450              * to bail.
1451              */
1452             if (ms_iis5_hack)
1453             {
1454                break;
1455             }
1456          }
1457          continue;
1458       }
1459
1460       return; /* huh? we should never get here */
1461    }
1462
1463    log_error(LOG_LEVEL_CLF, "%s - - [%T] \"%s\" 200 %d",
1464              csp->ip_addr_str, http->ocmd, byte_count);
1465 }
1466
1467
1468 /*********************************************************************
1469  *
1470  * Function    :  serve
1471  *
1472  * Description :  This is little more than chat.  We only "serve" to
1473  *                to close any socket that chat may have opened.
1474  *
1475  * Parameters  :
1476  *          1  :  csp = Current client state (buffers, headers, etc...)
1477  *
1478  * Returns     :  N/A
1479  *
1480  *********************************************************************/
1481 #ifdef AMIGA
1482 void serve(struct client_state *csp)
1483 #else /* ifndef AMIGA */
1484 static void serve(struct client_state *csp)
1485 #endif /* def AMIGA */
1486 {
1487    chat(csp);
1488    close_socket(csp->cfd);
1489
1490    if (csp->sfd >= 0)
1491    {
1492       close_socket(csp->sfd);
1493    }
1494
1495    csp->flags &= ~CSP_FLAG_ACTIVE;
1496
1497 }
1498
1499
1500 #ifdef __BEOS__
1501 /*********************************************************************
1502  *
1503  * Function    :  server_thread
1504  *
1505  * Description :  We only exist to call `serve' in a threaded environment.
1506  *
1507  * Parameters  :
1508  *          1  :  data = Current client state (buffers, headers, etc...)
1509  *
1510  * Returns     :  Always 0.
1511  *
1512  *********************************************************************/
1513 static int32 server_thread(void *data)
1514 {
1515    serve((struct client_state *) data);
1516    return 0;
1517
1518 }
1519 #endif
1520
1521
1522 /*********************************************************************
1523  *
1524  * Function    :  usage
1525  *
1526  * Description :  Print usage info & exit.
1527  *
1528  * Parameters  :  Pointer to argv[0] for identifying ourselves
1529  *
1530  * Returns     :  No. ,-)
1531  *
1532  *********************************************************************/
1533 void usage(const char *myname)
1534 {
1535    printf("JunkBuster proxy version " VERSION " (" HOME_PAGE_URL ")\n"
1536            "Usage: %s [--help] [--version] [--no-daemon] [--pidfile pidfile] [--user user] [configfile]\n"
1537            "Aborting.\n", myname);
1538  
1539    exit(2);
1540
1541 }
1542
1543
1544 /*********************************************************************
1545  *
1546  * Function    :  main
1547  *
1548  * Description :  Load the config file and start the listen loop.
1549  *                This function is a lot more *sane* with the `load_config'
1550  *                and `listen_loop' functions; although it stills does
1551  *                a *little* too much for my taste.
1552  *
1553  * Parameters  :
1554  *          1  :  argc = Number of parameters (including $0).
1555  *          2  :  argv = Array of (char *)'s to the parameters.
1556  *
1557  * Returns     :  1 if : can't open config file, unrecognized directive,
1558  *                stats requested in multi-thread mode, can't open the
1559  *                log file, can't open the jar file, listen port is invalid,
1560  *                any load fails, and can't bind port.
1561  *
1562  *                Else main never returns, the process must be signaled
1563  *                to terminate execution.  Or, on Windows, use the
1564  *                "File", "Exit" menu option.
1565  *
1566  *********************************************************************/
1567 #ifdef __MINGW32__
1568 int real_main(int argc, const char *argv[])
1569 #else
1570 int main(int argc, const char *argv[])
1571 #endif
1572 {
1573    int argc_pos = 0;
1574 #ifdef unix
1575    struct passwd *pw;
1576 #endif
1577
1578    Argc = argc;
1579    Argv = argv;
1580
1581    configfile =
1582 #if !defined(_WIN32)
1583    "config"
1584 #else
1585    "config.txt"
1586 #endif
1587       ;
1588
1589    /*
1590     * Parse the command line arguments
1591     */
1592    while (++argc_pos < argc)
1593    {
1594 #if !defined(_WIN32) || defined(_WIN_CONSOLE)
1595
1596       if (strcmp(argv[argc_pos], "--help") == 0)
1597       {
1598          usage(argv[0]);
1599       }
1600
1601       else if(strcmp(argv[argc_pos], "--version") == 0)
1602       {
1603          printf("Junkbuster version " VERSION " (" HOME_PAGE_URL ")\n");
1604          exit(0);
1605       }
1606
1607       else if (strcmp(argv[argc_pos], "--no-daemon" ) == 0)
1608       {
1609          no_daemon = 1;
1610       }
1611 #if defined(unix)
1612       else if (strcmp(argv[argc_pos], "--pidfile" ) == 0)
1613       {
1614          if (++argc_pos == argc) usage(argv[0]);
1615          pidfile = strdup(argv[argc_pos]);
1616       }
1617
1618       else if (strcmp(argv[argc_pos], "--user" ) == 0)
1619       {
1620          if (++argc_pos == argc) usage(argv[0]);
1621          pw = getpwnam(argv[argc_pos]);
1622                    
1623          if (pw == NULL)
1624          {
1625             log_error(LOG_LEVEL_FATAL, "User %s not found.", argv[argc_pos]);
1626          }
1627       }
1628 #endif /* defined(unix) */
1629       else
1630 #endif /* defined(_WIN32) && !defined(_WIN_CONSOLE) */
1631       {
1632          configfile = argv[argc_pos];
1633       }
1634
1635    } /* -END- while (more arguments) */
1636
1637 #if defined(unix)
1638    if ( *configfile != '/' )
1639    {
1640       char *abs_file;
1641
1642       /* make config-filename absolute here */
1643       if ( !(basedir = getcwd( NULL, 1024 )))
1644       {
1645          perror("get working dir failed");
1646          exit( 1 );
1647       }
1648
1649       if ( !(abs_file = malloc( strlen( basedir ) + strlen( configfile ) + 5 )))
1650       {
1651          perror("malloc failed");
1652          exit( 1 );
1653       }
1654       strcpy( abs_file, basedir );
1655       strcat( abs_file, "/" );
1656       strcat( abs_file, configfile );
1657       configfile = abs_file;
1658    }
1659 #endif /* defined unix */
1660
1661
1662    files->next = NULL;
1663
1664 #ifdef AMIGA
1665    InitAmiga();
1666 #elif defined(_WIN32)
1667    InitWin32();
1668 #endif
1669
1670    /*
1671     * Unix signal handling
1672     *
1673     * Catch the abort, interrupt and terminate signals for a graceful exit
1674     * Catch the hangup signal so the errlog can be reopened.
1675     * Ignore the broken pipe and child signals
1676     *  FIXME: Isn't ignoring the default for SIGCHLD anyway and why ignore SIGPIPE? 
1677     */
1678 #if !defined(_WIN32) && !defined(__OS2__) && !defined(AMIGA)
1679 {
1680    int idx;
1681    const int catched_signals[] = { SIGABRT, SIGTERM, SIGINT, SIGHUP, 0 };
1682    const int ignored_signals[] = { SIGPIPE, SIGCHLD, 0 };
1683
1684    for (idx = 0; catched_signals[idx] != 0; idx++)
1685    {
1686       if (signal(catched_signals[idx], sig_handler) == SIG_ERR)
1687       {
1688          log_error(LOG_LEVEL_FATAL, "Can't set signal-handler for signal %d: %E", catched_signals[idx]);
1689       }
1690    }
1691
1692    for (idx = 0; ignored_signals[idx] != 0; idx++)
1693    {
1694       if (signal(ignored_signals[idx], SIG_IGN) == SIG_ERR)
1695       {
1696          log_error(LOG_LEVEL_FATAL, "Can't set ignore-handler for signal %d: %E", ignored_signals[idx]);
1697       }
1698    }
1699
1700 }
1701 #else /* ifdef _WIN32 */
1702 # ifdef _WIN_CONSOLE
1703    /*
1704     * We *are* in a windows console app.
1705     * Print a verbose messages about FAQ's and such
1706     */
1707    printf(win32_blurb);
1708 # endif /* def _WIN_CONSOLE */
1709 #endif /* def _WIN32 */
1710
1711
1712    /* Initialize the CGI subsystem */
1713    cgi_init_error_messages();
1714
1715    /*
1716     * If runnig on unix and without the --nodaemon
1717     * option, become a daemon. I.e. fork, detach
1718     * from tty and get process group leadership
1719     */
1720 #if defined(unix)
1721 {
1722    pid_t pid = 0;
1723 #if 0
1724    int   fd;
1725 #endif
1726
1727    if (!no_daemon)
1728    {
1729       pid  = fork();
1730
1731       if ( pid < 0 ) /* error */
1732       {
1733          perror("fork");
1734          exit( 3 );
1735       }
1736       else if ( pid != 0 ) /* parent */
1737       {
1738          int status;
1739          pid_t wpid;
1740          /*
1741           * must check for errors
1742           * child died due to missing files aso
1743           */
1744          sleep( 1 );
1745          wpid = waitpid( pid, &status, WNOHANG );
1746          if ( wpid != 0 )
1747          {
1748             exit( 1 );
1749          }
1750          exit( 0 );
1751       }
1752       /* child */
1753 #if 1
1754       /* Should be more portable, but not as well tested */
1755       setsid();
1756 #else /* !1 */
1757 #ifdef __FreeBSD__
1758       setpgrp(0,0);
1759 #else /* ndef __FreeBSD__ */
1760       setpgrp();
1761 #endif /* ndef __FreeBSD__ */
1762       fd = open("/dev/tty", O_RDONLY);
1763       if ( fd )
1764       {
1765          /* no error check here */
1766          ioctl( fd, TIOCNOTTY,0 );
1767          close ( fd );
1768       }
1769 #endif /* 1 */
1770       /* FIXME: should close stderr (fd 2) here too, but the test
1771        * for existence
1772        * and load config file is done in listen_loop() and puts
1773        * some messages on stderr there.
1774        */
1775
1776       close( 0 );
1777       close( 1 );
1778       chdir("/");
1779
1780    } /* -END- if (!no_daemon) */
1781
1782    /*
1783     * As soon as we have written the PID file, we can switch
1784     * to the user ID indicated by the --user option
1785     */
1786    write_pid_file();
1787    
1788    if (setuid(pw->pw_uid))
1789    {
1790       log_error(LOG_LEVEL_FATAL, "Cannot setuid(): Insufficient permissions.");
1791    }
1792
1793 }
1794 #endif /* defined unix */
1795
1796    listen_loop();
1797
1798    /* NOTREACHED */
1799    return(-1);
1800
1801 }
1802
1803
1804 /*********************************************************************
1805  *
1806  * Function    :  bind_port_helper
1807  *
1808  * Description :  Bind the listen port.  Handles logging, and aborts
1809  *                on failure.
1810  *
1811  * Parameters  :
1812  *          1  :  config = Junkbuster configuration.  Specifies port
1813  *                         to bind to.
1814  *
1815  * Returns     :  Port that was opened.
1816  *
1817  *********************************************************************/
1818 static int bind_port_helper(struct configuration_spec * config)
1819 {
1820    int bfd;
1821
1822    if ( (config->haddr != NULL)
1823      && (config->haddr[0] == '1')
1824      && (config->haddr[1] == '2')
1825      && (config->haddr[2] == '7')
1826      && (config->haddr[3] == '.') )
1827    {
1828       log_error(LOG_LEVEL_INFO, "Listening on port %d for local connections only",
1829                 config->hport);
1830    }
1831    else if (config->haddr == NULL)
1832    {
1833       log_error(LOG_LEVEL_INFO, "Listening on port %d on all IP addresses",
1834                 config->hport);
1835    }
1836    else
1837    {
1838       log_error(LOG_LEVEL_INFO, "Listening on port %d on IP address %s",
1839                 config->hport, config->haddr);
1840    }
1841
1842    bfd = bind_port(config->haddr, config->hport);
1843
1844    if (bfd < 0)
1845    {
1846       log_error(LOG_LEVEL_FATAL, "can't bind %s:%d: %E "
1847          "- There may be another junkbuster or some other "
1848          "proxy running on port %d",
1849          (NULL != config->haddr) ? config->haddr : "INADDR_ANY",
1850          config->hport, config->hport
1851       );
1852       /* shouldn't get here */
1853       return -1;
1854    }
1855
1856    config->need_bind = 0;
1857
1858    return bfd;
1859 }
1860
1861
1862 /*********************************************************************
1863  *
1864  * Function    :  listen_loop
1865  *
1866  * Description :  bind the listen port and enter a "FOREVER" listening loop.
1867  *
1868  * Parameters  :  N/A
1869  *
1870  * Returns     :  Never.
1871  *
1872  *********************************************************************/
1873 static void listen_loop(void)
1874 {
1875    struct client_state *csp = NULL;
1876    int bfd;
1877    struct configuration_spec * config;
1878
1879    config = load_config();
1880
1881    bfd = bind_port_helper(config);
1882
1883    while (FOREVER)
1884    {
1885 #if !defined(FEATURE_PTHREAD) && !defined(_WIN32) && !defined(__BEOS__) && !defined(AMIGA) && !defined(__OS2__)
1886       while (waitpid(-1, NULL, WNOHANG) > 0)
1887       {
1888          /* zombie children */
1889       }
1890 #endif /* !defined(FEATURE_PTHREAD) && !defined(_WIN32) && !defined(__BEOS__) && !defined(AMIGA) */
1891
1892       /*
1893        * Free data that was used by died threads
1894        */
1895       sweep();
1896
1897 #if defined(unix)
1898       /*
1899        * Re-open the errlog after HUP signal
1900        */
1901       if (received_hup_signal)
1902       {
1903          init_error_log(Argv[0], config->logfile, config->debug);
1904          received_hup_signal = 0;
1905       }
1906 #endif
1907
1908       if ( NULL == (csp = (struct client_state *) zalloc(sizeof(*csp))) )
1909       {
1910          log_error(LOG_LEVEL_FATAL, "malloc(%d) for csp failed: %E", sizeof(*csp));
1911          continue;
1912       }
1913
1914       csp->flags |= CSP_FLAG_ACTIVE;
1915       csp->sfd    = -1;
1916
1917       csp->config = config = load_config();
1918
1919       if ( config->need_bind )
1920       {
1921          /*
1922           * Since we were listening to the "old port", we will not see
1923           * a "listen" param change until the next IJB request.  So, at
1924           * least 1 more request must be made for us to find the new
1925           * setting.  I am simply closing the old socket and binding the
1926           * new one.
1927           *
1928           * Which-ever is correct, we will serve 1 more page via the
1929           * old settings.  This should probably be a "show-proxy-args"
1930           * request.  This should not be a so common of an operation
1931           * that this will hurt people's feelings.
1932           */
1933
1934          close_socket(bfd);
1935
1936          bfd = bind_port_helper(config);
1937       }
1938
1939       log_error(LOG_LEVEL_CONNECT, "accept connection ... ");
1940
1941       if (!accept_connection(csp, bfd))
1942       {
1943          log_error(LOG_LEVEL_CONNECT, "accept failed: %E");
1944
1945 #ifdef AMIGA
1946          if(!childs)
1947          {
1948             exit(1);
1949          }
1950 #endif
1951          freez(csp);
1952          continue;
1953       }
1954       else
1955       {
1956          log_error(LOG_LEVEL_CONNECT, "OK");
1957       }
1958
1959 #ifdef FEATURE_TOGGLE
1960       if (g_bToggleIJB)
1961       {
1962          csp->flags |= CSP_FLAG_TOGGLED_ON;
1963       }
1964 #endif /* def FEATURE_TOGGLE */
1965
1966       if (run_loader(csp))
1967       {
1968          log_error(LOG_LEVEL_FATAL, "a loader failed - must exit");
1969          /* Never get here - LOG_LEVEL_FATAL causes program exit */
1970       }
1971
1972 #ifdef FEATURE_ACL
1973       if (block_acl(NULL,csp))
1974       {
1975          log_error(LOG_LEVEL_CONNECT, "Connection dropped due to ACL");
1976          close_socket(csp->cfd);
1977          freez(csp);
1978          continue;
1979       }
1980 #endif /* def FEATURE_ACL */
1981
1982       /* add it to the list of clients */
1983       csp->next = clients->next;
1984       clients->next = csp;
1985
1986       if (config->multi_threaded)
1987       {
1988          int child_id;
1989
1990 /* this is a switch () statment in the C preprocessor - ugh */
1991 #undef SELECTED_ONE_OPTION
1992
1993 /* Use Pthreads in preference to native code */
1994 #if defined(FEATURE_PTHREAD) && !defined(SELECTED_ONE_OPTION)
1995 #define SELECTED_ONE_OPTION
1996          {
1997             pthread_t the_thread;
1998             pthread_attr_t attrs;
1999
2000             pthread_attr_init(&attrs);
2001             pthread_attr_setdetachstate(&attrs, PTHREAD_CREATE_DETACHED);
2002             child_id = (pthread_create(&the_thread, &attrs,
2003                (void*)serve, csp) ? -1 : 0);
2004             pthread_attr_destroy(&attrs);
2005          }
2006 #endif
2007
2008 #if defined(_WIN32) && !defined(_CYGWIN) && !defined(SELECTED_ONE_OPTION)
2009 #define SELECTED_ONE_OPTION
2010          child_id = _beginthread(
2011             (void*)serve,
2012             64 * 1024,
2013             csp);
2014 #endif
2015
2016 #if defined(__OS2__) && !defined(SELECTED_ONE_OPTION)
2017 #define SELECTED_ONE_OPTION
2018          child_id = _beginthread(
2019             serve,
2020             NULL,
2021             64 * 1024,
2022             csp);
2023 #endif
2024
2025 #if defined(__BEOS__) && !defined(SELECTED_ONE_OPTION)
2026 #define SELECTED_ONE_OPTION
2027          {
2028             thread_id tid = spawn_thread
2029                (server_thread, "server", B_NORMAL_PRIORITY, csp);
2030
2031             if ((tid >= 0) && (resume_thread(tid) == B_OK))
2032             {
2033                child_id = (int) tid;
2034             }
2035             else
2036             {
2037                child_id = -1;
2038             }
2039          }
2040 #endif
2041
2042 #if defined(AMIGA) && !defined(SELECTED_ONE_OPTION)
2043 #define SELECTED_ONE_OPTION
2044          csp->cfd = ReleaseSocket(csp->cfd, -1);
2045          if((child_id = (int)CreateNewProcTags(
2046             NP_Entry, (ULONG)server_thread,
2047             NP_Output, Output(),
2048             NP_CloseOutput, FALSE,
2049             NP_Name, (ULONG)"junkbuster child",
2050             NP_StackSize, 200*1024,
2051             TAG_DONE)))
2052          {
2053             childs++;
2054             ((struct Task *)child_id)->tc_UserData = csp;
2055             Signal((struct Task *)child_id, SIGF_SINGLE);
2056             Wait(SIGF_SINGLE);
2057          }
2058 #endif
2059
2060 #if !defined(SELECTED_ONE_OPTION)
2061          child_id = fork();
2062
2063          /* This block is only needed when using fork().
2064           * When using threads, the server thread was
2065           * created and run by the call to _beginthread().
2066           */
2067          if (child_id == 0)   /* child */
2068          {
2069             serve(csp);
2070             _exit(0);
2071
2072          }
2073          else if (child_id > 0) /* parent */
2074          {
2075             /* in a fork()'d environment, the parent's
2076              * copy of the client socket and the CSP
2077              * are not used.
2078              */
2079
2080 #if !defined(_WIN32) && defined(__CYGWIN__)
2081             wait( NULL );
2082 #endif /* !defined(_WIN32) && defined(__CYGWIN__) */
2083             close_socket(csp->cfd);
2084             csp->flags &= ~CSP_FLAG_ACTIVE;
2085          }
2086 #endif
2087
2088 #undef SELECTED_ONE_OPTION
2089 /* end of cpp switch () */
2090
2091          if (child_id < 0) /* failed */
2092          {
2093             char buf[BUFFER_SIZE];
2094
2095             log_error(LOG_LEVEL_ERROR, "can't fork: %E");
2096
2097             sprintf(buf , "JunkBuster: can't fork: errno = %d", errno);
2098
2099             write_socket(csp->cfd, buf, strlen(buf));
2100             close_socket(csp->cfd);
2101             csp->flags &= ~CSP_FLAG_ACTIVE;
2102             sleep(5);
2103             continue;
2104          }
2105       }
2106       else
2107       {
2108          serve(csp);
2109       }
2110    }
2111    /* NOTREACHED */
2112
2113 }
2114
2115
2116 /*
2117   Local Variables:
2118   tab-width: 3
2119   end:
2120 */