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