c5f30c2ad7c0e4122e9ffec14b26195cf0d29e0e
[citadel.git] / webcit / webserver.c
1 /*
2  * $Id$
3  *
4  * This contains a simple multithreaded TCP server manager.  It sits around
5  * waiting on the specified port for incoming HTTP connections.  When a
6  * connection is established, it calls context_loop() from context_loop.c.
7  *
8  */
9
10 #include "webcit.h"
11 #include "webserver.h"
12
13 #if HAVE_BACKTRACE
14 #include <execinfo.h>
15 #endif
16
17 #ifndef HAVE_SNPRINTF
18 int vsnprintf(char *buf, size_t max, const char *fmt, va_list argp);
19 #endif
20
21 int verbosity = 9;              /* Logging level */
22 int msock;                      /* master listening socket */
23 int is_https = 0;               /* Nonzero if I am an HTTPS service */
24 int follow_xff = 0;             /* Follow X-Forwarded-For: header */
25 int home_specified = 0;         /* did the user specify a homedir? */
26 int time_to_die = 0;            /* Nonzero if server is shutting down */
27 extern void *context_loop(int);
28 extern void *housekeeping_loop(void);
29 extern pthread_mutex_t SessionListMutex;
30 extern pthread_key_t MyConKey;
31
32
33 char ctdl_key_dir[PATH_MAX]=SSL_DIR;
34 char file_crpt_file_key[PATH_MAX]="";
35 char file_crpt_file_csr[PATH_MAX]="";
36 char file_crpt_file_cer[PATH_MAX]="";
37
38 char socket_dir[PATH_MAX];                      /* where to talk to our citadel server */
39 static const char editor_absolut_dir[PATH_MAX]=EDITORDIR;       /* nailed to what configure gives us. */
40 static char static_dir[PATH_MAX];               /* calculated on startup */
41 static char static_local_dir[PATH_MAX];         /* calculated on startup */
42 char  *static_dirs[]={                          /* needs same sort order as the web mapping */
43         (char*)static_dir,                      /* our templates on disk */
44         (char*)static_local_dir,                /* user provided templates disk */
45         (char*)editor_absolut_dir               /* the editor on disk */
46 };
47
48 /*
49  * Subdirectories from which the client may request static content
50  *
51  * (If you add more, remember to increment 'ndirs' below)
52  */
53 char *static_content_dirs[] = {
54         "static",                     /** static templates */
55         "static.local",               /** site local static templates */
56         "tiny_mce"                    /** rich text editor */
57 };
58
59 int ndirs=3;
60
61
62 char *server_cookie = NULL;     /* our Cookie connection to the client */
63 int http_port = PORT_NUM;       /* Port to listen on */
64 char *ctdlhost = DEFAULT_HOST;  /* our name */
65 char *ctdlport = DEFAULT_PORT;  /* our Port */
66 int setup_wizard = 0;           /* should we run the setup wizard? \todo */
67 char wizard_filename[PATH_MAX]; /* where's the setup wizard? */
68 int running_as_daemon = 0;      /* should we deamonize on startup? */
69
70
71 /* 
72  * This is a generic function to set up a master socket for listening on
73  * a TCP port.  The server shuts down if the bind fails.
74  *
75  * ip_addr      IP address to bind
76  * port_number  port number to bind
77  * queue_len    number of incoming connections to allow in the queue
78  */
79 int ig_tcp_server(char *ip_addr, int port_number, int queue_len)
80 {
81         struct sockaddr_in sin;
82         int s, i;
83
84         memset(&sin, 0, sizeof(sin));
85         sin.sin_family = AF_INET;
86         if (ip_addr == NULL) {
87                 sin.sin_addr.s_addr = INADDR_ANY;
88         } else {
89                 sin.sin_addr.s_addr = inet_addr(ip_addr);
90         }
91
92         if (sin.sin_addr.s_addr == INADDR_NONE) {
93                 sin.sin_addr.s_addr = INADDR_ANY;
94         }
95
96         if (port_number == 0) {
97                 lprintf(1, "Cannot start: no port number specified.\n");
98                 exit(WC_EXIT_BIND);
99         }
100         sin.sin_port = htons((u_short) port_number);
101
102         s = socket(PF_INET, SOCK_STREAM, (getprotobyname("tcp")->p_proto));
103         if (s < 0) {
104                 lprintf(1, "Can't create a socket: %s\n", strerror(errno));
105                 exit(WC_EXIT_BIND);
106         }
107         /* Set some socket options that make sense. */
108         i = 1;
109         setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &i, sizeof(i));
110
111         fcntl(s, F_SETFL, O_NONBLOCK); /* maide: this statement is incorrect
112                                           there should be a preceding F_GETFL
113                                           and a bitwise OR with the previous
114                                           fd flags */
115         
116         if (bind(s, (struct sockaddr *) &sin, sizeof(sin)) < 0) {
117                 lprintf(1, "Can't bind: %s\n", strerror(errno));
118                 exit(WC_EXIT_BIND);
119         }
120         if (listen(s, queue_len) < 0) {
121                 lprintf(1, "Can't listen: %s\n", strerror(errno));
122                 exit(WC_EXIT_BIND);
123         }
124         return (s);
125 }
126
127
128
129 /*
130  * \brief Create a Unix domain socket and listen on it
131  * \param sockpath file name of the unix domain socket
132  * \param queue_len Number of incoming connections to allow in the queue
133  */
134 int ig_uds_server(char *sockpath, int queue_len)
135 {
136         struct sockaddr_un addr;
137         int s;
138         int i;
139         int actual_queue_len;
140
141         actual_queue_len = queue_len;
142         if (actual_queue_len < 5) actual_queue_len = 5;
143
144         i = unlink(sockpath);
145         if (i != 0) if (errno != ENOENT) {
146                 lprintf(1, "webserver: can't unlink %s: %s\n",
147                         sockpath, strerror(errno));
148                 exit(WC_EXIT_BIND);
149         }
150
151         memset(&addr, 0, sizeof(addr));
152         addr.sun_family = AF_UNIX;
153         safestrncpy(addr.sun_path, sockpath, sizeof addr.sun_path);
154
155         s = socket(AF_UNIX, SOCK_STREAM, 0);
156         if (s < 0) {
157                 lprintf(1, "webserver: Can't create a socket: %s\n",
158                         strerror(errno));
159                 exit(WC_EXIT_BIND);
160         }
161
162         if (bind(s, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
163                 lprintf(1, "webserver: Can't bind: %s\n",
164                         strerror(errno));
165                 exit(WC_EXIT_BIND);
166         }
167
168         if (listen(s, actual_queue_len) < 0) {
169                 lprintf(1, "webserver: Can't listen: %s\n",
170                         strerror(errno));
171                 exit(WC_EXIT_BIND);
172         }
173
174         chmod(sockpath, 0777);
175         return(s);
176 }
177
178
179
180
181 /*
182  * \brief Read data from the client socket.
183  * \param sock socket fd to read from
184  * \param buf buffer to read into 
185  * \param bytes number of bytes to read
186  * \param timeout Number of seconds to wait before timing out
187  * \return values are\
188  *      1       Requested number of bytes has been read.\
189  *      0       Request timed out.\
190  *         -1           Connection is broken, or other error.
191  */
192 int client_read_to(int sock, char *buf, int bytes, int timeout)
193 {
194         int len, rlen;
195         fd_set rfds;
196         struct timeval tv;
197         int retval;
198
199
200 #ifdef HAVE_OPENSSL
201         if (is_https) {
202                 return (client_read_ssl(buf, bytes, timeout));
203         }
204 #endif
205
206         len = 0;
207         while (len < bytes) {
208                 FD_ZERO(&rfds);
209                 FD_SET(sock, &rfds);
210                 tv.tv_sec = timeout;
211                 tv.tv_usec = 0;
212
213                 retval = select((sock) + 1, &rfds, NULL, NULL, &tv);
214                 if (FD_ISSET(sock, &rfds) == 0) {
215                         return (0);
216                 }
217
218                 rlen = read(sock, &buf[len], bytes - len);
219
220                 if (rlen < 1) {
221                         lprintf(2, "client_read() failed: %s\n",
222                                 strerror(errno));
223                         return (-1);
224                 }
225                 len = len + rlen;
226         }
227
228 #ifdef HTTP_TRACING
229         write(2, "\033[32m", 5);
230         write(2, buf, bytes);
231         write(2, "\033[30m", 5);
232 #endif
233         return (1);
234 }
235
236 /*
237  * \brief write data to the client
238  * \param buf data to write to the client
239  * \param count size of buffer
240  */
241 ssize_t client_write(const void *buf, size_t count)
242 {
243         char *newptr;
244         size_t newalloc;
245         size_t bytesWritten = 0;
246         ssize_t res;
247         fd_set wset;
248         int fdflags;
249
250         if (WC->burst != NULL) {
251                 if ((WC->burst_len + count) >= WC->burst_alloc) {
252                         newalloc = (WC->burst_alloc * 2);
253                         if ((WC->burst_len + count) >= newalloc) {
254                                 newalloc += count;
255                         }
256                         newptr = realloc(WC->burst, newalloc);
257                         if (newptr != NULL) {
258                                 WC->burst = newptr;
259                                 WC->burst_alloc = newalloc;
260                         }
261                 }
262                 if ((WC->burst_len + count) < WC->burst_alloc) {
263                         memcpy(&WC->burst[WC->burst_len], buf, count);
264                         WC->burst_len += count;
265                         return (count);
266                 }
267                 else {
268                         return(-1);
269                 }
270         }
271 #ifdef HAVE_OPENSSL
272         if (is_https) {
273                 client_write_ssl((char *) buf, count);
274                 return (count);
275         }
276 #endif
277 #ifdef HTTP_TRACING
278         write(2, "\033[34m", 5);
279         write(2, buf, count);
280         write(2, "\033[30m", 5);
281 #endif
282         fdflags = fcntl(WC->http_sock, F_GETFL);
283
284         while (bytesWritten < count) {
285                 if ((fdflags & O_NONBLOCK) == O_NONBLOCK) {
286                         FD_ZERO(&wset);
287                         FD_SET(WC->http_sock, &wset);
288                         if (select(1, NULL, &wset, NULL, NULL) == -1) {
289                                 lprintf(2, "client_write: Socket select failed (%s)\n", strerror(errno));
290                                 return -1;
291                         }
292                 }
293
294                 if ((res = write(WC->http_sock, (char*)buf + bytesWritten,
295                   count - bytesWritten)) == -1) {
296                         lprintf(2, "client_write: Socket write failed (%s)\n", strerror(errno));
297                         return res;
298                 }
299                 bytesWritten += res;
300         }
301
302         return bytesWritten;
303 }
304
305 /*
306  * \brief Begin buffering HTTP output so we can transmit it all in one write operation later.
307  */
308 void begin_burst(void)
309 {
310         if (WC->burst != NULL) {
311                 free(WC->burst);
312                 WC->burst = NULL;
313         }
314         WC->burst_len = 0;
315         WC->burst_alloc = 32768;
316         WC->burst = malloc(WC->burst_alloc);
317 }
318
319
320 /*
321  * \brief uses the same calling syntax as compress2(), but it
322  * creates a stream compatible with HTTP "Content-encoding: gzip"
323  */
324 #ifdef HAVE_ZLIB
325 #define DEF_MEM_LEVEL 8 /*< memlevel??? */
326 #define OS_CODE 0x03    /*< unix */
327 int ZEXPORT compress_gzip(Bytef * dest,         /*< compressed buffer*/
328                           size_t * destLen,     /*< length of the compresed data */
329                           const Bytef * source, /*< source to encode */
330                           uLong sourceLen,      /*< length of source to encode */
331                           int level)            /*< compression level */
332 {
333         const int gz_magic[2] = { 0x1f, 0x8b }; /* gzip magic header */
334
335         /* write gzip header */
336         snprintf((char *) dest, *destLen, 
337                  "%c%c%c%c%c%c%c%c%c%c",
338                  gz_magic[0], gz_magic[1], Z_DEFLATED,
339                  0 /*flags */ , 0, 0, 0, 0 /*time */ , 0 /* xflags */ ,
340                  OS_CODE);
341
342         /* normal deflate */
343         z_stream stream;
344         int err;
345         stream.next_in = (Bytef *) source;
346         stream.avail_in = (uInt) sourceLen;
347         stream.next_out = dest + 10L;   // after header
348         stream.avail_out = (uInt) * destLen;
349         if ((uLong) stream.avail_out != *destLen)
350                 return Z_BUF_ERROR;
351
352         stream.zalloc = (alloc_func) 0;
353         stream.zfree = (free_func) 0;
354         stream.opaque = (voidpf) 0;
355
356         err = deflateInit2(&stream, level, Z_DEFLATED, -MAX_WBITS,
357                            DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY);
358         if (err != Z_OK)
359                 return err;
360
361         err = deflate(&stream, Z_FINISH);
362         if (err != Z_STREAM_END) {
363                 deflateEnd(&stream);
364                 return err == Z_OK ? Z_BUF_ERROR : err;
365         }
366         *destLen = stream.total_out + 10L;
367
368         /* write CRC and Length */
369         uLong crc = crc32(0L, source, sourceLen);
370         int n;
371         for (n = 0; n < 4; ++n, ++*destLen) {
372                 dest[*destLen] = (int) (crc & 0xff);
373                 crc >>= 8;
374         }
375         uLong len = stream.total_in;
376         for (n = 0; n < 4; ++n, ++*destLen) {
377                 dest[*destLen] = (int) (len & 0xff);
378                 len >>= 8;
379         }
380         err = deflateEnd(&stream);
381         return err;
382 }
383 #endif
384
385 /*
386  * \brief Finish buffering HTTP output.  [Compress using zlib and] output with a Content-Length: header.
387  */
388 void end_burst(void)
389 {
390         size_t the_len;
391         char *the_data;
392
393         if (WC->burst == NULL)
394                 return;
395
396         the_len = WC->burst_len;
397         the_data = WC->burst;
398
399         WC->burst_len = 0;
400         WC->burst_alloc = 0;
401         WC->burst = NULL;
402
403 #ifdef HAVE_ZLIB
404         /* Perform gzip compression, if enabled and supported by client */
405         if (WC->gzip_ok) {
406                 char *compressed_data = NULL;
407                 size_t compressed_len;
408
409                 compressed_len = ((the_len * 101) / 100) + 100;
410                 compressed_data = malloc(compressed_len);
411
412                 if (compress_gzip((Bytef *) compressed_data,
413                                   &compressed_len,
414                                   (Bytef *) the_data,
415                                   (uLongf) the_len, Z_BEST_SPEED) == Z_OK) {
416                         wprintf("Content-encoding: gzip\r\n");
417                         free(the_data);
418                         the_data = compressed_data;
419                         the_len = compressed_len;
420                 } else {
421                         free(compressed_data);
422                 }
423         }
424 #endif  /* HAVE_ZLIB */
425
426         wprintf("Content-length: %d\r\n\r\n", the_len);
427         client_write(the_data, the_len);
428         free(the_data);
429         return;
430 }
431
432
433
434 /*
435  * \brief Read data from the client socket with default timeout.
436  * (This is implemented in terms of client_read_to() and could be
437  * justifiably moved out of sysdep.c)
438  * \param sock the socket fd to read from
439  * \param buf the buffer to write to
440  * \param bytes Number of bytes to read
441  */
442 int client_read(int sock, char *buf, int bytes)
443 {
444         return (client_read_to(sock, buf, bytes, SLEEPING));
445 }
446
447
448 /*
449  * \brief Get a LF-terminated line of text from the client.
450  * (This is implemented in terms of client_read() and could be
451  * justifiably moved out of sysdep.c)
452  * \param sock socket fd to get client line from
453  * \param buf buffer to write read data to
454  * \param bufsiz how many bytes to read
455  * \return  number of bytes read???
456  */
457 int client_getln(int sock, char *buf, int bufsiz)
458 {
459         int i, retval;
460
461         /* Read one character at a time.*/
462         for (i = 0;; i++) {
463                 retval = client_read(sock, &buf[i], 1);
464                 if (retval != 1 || buf[i] == '\n' || i == (bufsiz-1))
465                         break;
466                 if ( (!isspace(buf[i])) && (!isprint(buf[i])) ) {
467                         /* Non printable character recieved from client */
468                         return(-1);
469                 }
470         }
471
472         /* If we got a long line, discard characters until the newline. */
473         if (i == (bufsiz-1))
474                 while (buf[i] != '\n' && retval == 1)
475                         retval = client_read(sock, &buf[i], 1);
476
477         /*
478          * Strip any trailing non-printable characters.
479          */
480         buf[i] = 0;
481         while ((i > 0) && (!isprint(buf[i - 1]))) {
482                 buf[--i] = 0;
483         }
484         return (retval);
485 }
486
487 /*
488  * \brief shut us down the regular way.
489  * param signum the signal we want to forward
490  */
491 pid_t current_child;
492 void graceful_shutdown_watcher(int signum) {
493         lprintf (1, "bye; shutting down watcher.");
494         kill(current_child, signum);
495         if (signum != SIGHUP)
496                 exit(0);
497 }
498
499 /*
500  * \brief shut us down the regular way.
501  * param signum the signal we want to forward
502  */
503 pid_t current_child;
504 void graceful_shutdown(int signum) {
505 //      kill(current_child, signum);
506         char wd[SIZ];
507         FILE *FD;
508         int fd;
509         getcwd(wd, SIZ);
510         lprintf (1, "bye going down gracefull.[%d][%s]\n", signum, wd);
511         fd = msock;
512         msock = -1;
513         time_to_die = 1;
514         FD=fdopen(fd, "a+");
515         fflush (FD);
516         fclose (FD);
517         close(fd);
518 }
519
520
521 /*
522  * \brief       Start running as a daemon.  
523  *
524  * param        do_close_stdio          Only close stdio if set.
525  */
526
527 /*
528  * Start running as a daemon.
529  */
530 void start_daemon(char *pid_file) 
531 {
532         int status = 0;
533         pid_t child = 0;
534         FILE *fp;
535         int do_restart = 0;
536
537         current_child = 0;
538
539         /* Close stdin/stdout/stderr and replace them with /dev/null.
540          * We don't just call close() because we don't want these fd's
541          * to be reused for other files.
542          */
543         chdir("/");
544
545         signal(SIGHUP, SIG_IGN);
546         signal(SIGINT, SIG_IGN);
547         signal(SIGQUIT, SIG_IGN);
548
549         child = fork();
550         if (child != 0) {
551                 exit(0);
552         }
553
554         setsid();
555         umask(0);
556         freopen("/dev/null", "r", stdin);
557         freopen("/dev/null", "w", stdout);
558         freopen("/dev/null", "w", stderr);
559         signal(SIGTERM, graceful_shutdown_watcher);
560         signal(SIGHUP, graceful_shutdown_watcher);
561
562         do {
563                 current_child = fork();
564
565         
566                 if (current_child < 0) {
567                         perror("fork");
568                         exit(errno);
569                 }
570         
571                 else if (current_child == 0) {  // child process
572 //                      signal(SIGTERM, graceful_shutdown);
573                         signal(SIGHUP, graceful_shutdown);
574
575                         return; /* continue starting webcit. */
576                 }
577         
578                 else { // watcher process
579 //                      signal(SIGTERM, SIG_IGN);
580 //                      signal(SIGHUP, SIG_IGN);
581                         if (pid_file) {
582                                 fp = fopen(pid_file, "w");
583                                 if (fp != NULL) {
584                                         fprintf(fp, "%d\n", getpid());
585                                         fclose(fp);
586                                 }
587                         }
588                         waitpid(current_child, &status, 0);
589                 }
590
591                 do_restart = 0;
592
593                 /* Did the main process exit with an actual exit code? */
594                 if (WIFEXITED(status)) {
595
596                         /* Exit code 0 means the watcher should exit */
597                         if (WEXITSTATUS(status) == 0) {
598                                 do_restart = 0;
599                         }
600
601                         /* Exit code 101-109 means the watcher should exit */
602                         else if ( (WEXITSTATUS(status) >= 101) && (WEXITSTATUS(status) <= 109) ) {
603                                 do_restart = 0;
604                         }
605
606                         /* Any other exit code means we should restart. */
607                         else {
608                                 do_restart = 1;
609                         }
610                 }
611
612                 /* Any other type of termination (signals, etc.) should also restart. */
613                 else {
614                         do_restart = 1;
615                 }
616
617         } while (do_restart);
618
619         if (pid_file) {
620                 unlink(pid_file);
621         }
622         exit(WEXITSTATUS(status));
623 }
624
625 /*
626  * \brief       Spawn an additional worker thread into the pool.
627  */
628 void spawn_another_worker_thread()
629 {
630         pthread_t SessThread;   /*< Thread descriptor */
631         pthread_attr_t attr;    /*< Thread attributes */
632         int ret;
633
634         lprintf(3, "Creating a new thread\n");
635
636         /* set attributes for the new thread */
637         pthread_attr_init(&attr);
638         pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
639
640         /*
641          * Our per-thread stacks need to be bigger than the default size, otherwise
642          * the MIME parser crashes on FreeBSD, and the IMAP service crashes on
643          * 64-bit Linux.
644          */
645         if ((ret = pthread_attr_setstacksize(&attr, 1024 * 1024))) {
646                 lprintf(1, "pthread_attr_setstacksize: %s\n",
647                         strerror(ret));
648                 pthread_attr_destroy(&attr);
649         }
650
651         /* now create the thread */
652         if (pthread_create(&SessThread, &attr,
653                            (void *(*)(void *)) worker_entry, NULL)
654             != 0) {
655                 lprintf(1, "Can't create thread: %s\n", strerror(errno));
656         }
657
658         /* free up the attributes */
659         pthread_attr_destroy(&attr);
660 }
661
662 /*
663  * \brief Here's where it all begins.
664  * \param argc number of commandline args
665  * \param argv the commandline arguments
666  */
667 int main(int argc, char **argv)
668 {
669         pthread_t SessThread;   /*< Thread descriptor */
670         pthread_attr_t attr;    /*< Thread attributes */
671         int a, i;                       /*< General-purpose variables */
672         char tracefile[PATH_MAX];
673         char ip_addr[256]="0.0.0.0";
674         char dirbuffer[PATH_MAX]="";
675         int relh=0;
676         int home=0;
677         int home_specified=0;
678         char relhome[PATH_MAX]="";
679         char webcitdir[PATH_MAX] = DATADIR;
680         char *pidfile = NULL;
681         char *hdir;
682         const char *basedir;
683 #ifdef ENABLE_NLS
684         char *locale = NULL;
685         char *mo = NULL;
686 #endif /* ENABLE_NLS */
687         char uds_listen_path[PATH_MAX]; /*< listen on a unix domain socket? */
688
689         /* Ensure that we are linked to the correct version of libcitadel */
690         if (libcitadel_version_number() < LIBCITADEL_VERSION_NUMBER) {
691                 fprintf(stderr, " You are running libcitadel version %d.%02d\n",
692                         (libcitadel_version_number() / 100), (libcitadel_version_number() % 100));
693                 fprintf(stderr, "WebCit was compiled against version %d.%02d\n",
694                         (LIBCITADEL_VERSION_NUMBER / 100), (LIBCITADEL_VERSION_NUMBER % 100));
695                 return(1);
696         }
697
698         strcpy(uds_listen_path, "");
699
700         /* Parse command line */
701 #ifdef HAVE_OPENSSL
702         while ((a = getopt(argc, argv, "h:i:p:t:x:dD:cfs")) != EOF)
703 #else
704         while ((a = getopt(argc, argv, "h:i:p:t:x:dD:cf")) != EOF)
705 #endif
706                 switch (a) {
707                 case 'h':
708                         hdir = strdup(optarg);
709                         relh=hdir[0]!='/';
710                         if (!relh) safestrncpy(webcitdir, hdir,
711                                                                    sizeof webcitdir);
712                         else
713                                 safestrncpy(relhome, relhome,
714                                                         sizeof relhome);
715                         /* free(hdir); TODO: SHOULD WE DO THIS? */
716                         home_specified = 1;
717                         home=1;
718                         break;
719                 case 'd':
720                         running_as_daemon = 1;
721                         break;
722                 case 'D':
723                         pidfile = strdup(optarg);
724                         running_as_daemon = 1;
725                         break;
726                 case 'i':
727                         safestrncpy(ip_addr, optarg, sizeof ip_addr);
728                         break;
729                 case 'p':
730                         http_port = atoi(optarg);
731                         if (http_port == 0) {
732                                 safestrncpy(uds_listen_path, optarg, sizeof uds_listen_path);
733                         }
734                         break;
735                 case 't':
736                         safestrncpy(tracefile, optarg, sizeof tracefile);
737                         freopen(tracefile, "w", stdout);
738                         freopen(tracefile, "w", stderr);
739                         freopen(tracefile, "r", stdin);
740                         break;
741                 case 'x':
742                         verbosity = atoi(optarg);
743                         break;
744                 case 'f':
745                         follow_xff = 1;
746                         break;
747                 case 'c':
748                         server_cookie = malloc(256);
749                         if (server_cookie != NULL) {
750                                 safestrncpy(server_cookie,
751                                        "Set-cookie: wcserver=",
752                                         256);
753                                 if (gethostname
754                                     (&server_cookie[strlen(server_cookie)],
755                                      200) != 0) {
756                                         lprintf(2, "gethostname: %s\n",
757                                                 strerror(errno));
758                                         free(server_cookie);
759                                 }
760                         }
761                         break;
762                 case 's':
763                         is_https = 1;
764                         break;
765                 default:
766                         fprintf(stderr, "usage: webserver "
767                                 "[-i ip_addr] [-p http_port] "
768                                 "[-t tracefile] [-c] [-f] "
769                                 "[-d] "
770 #ifdef HAVE_OPENSSL
771                                 "[-s] "
772 #endif
773                                 "[remotehost [remoteport]]\n");
774                         return 1;
775                 }
776
777         if (optind < argc) {
778                 ctdlhost = argv[optind];
779                 if (++optind < argc)
780                         ctdlport = argv[optind];
781         }
782
783         /* daemonize, if we were asked to */
784         if (running_as_daemon) {
785                 start_daemon(pidfile);
786         }
787         else {
788 ///             signal(SIGTERM, graceful_shutdown);
789                 signal(SIGHUP, graceful_shutdown);
790         }
791
792         /* Tell 'em who's in da house */
793         lprintf(1, PACKAGE_STRING "\n");
794         lprintf(1, "Copyright (C) 1996-2008 by the Citadel development team.\n"
795                 "This software is distributed under the terms of the "
796                 "GNU General Public License.\n\n"
797         );
798
799
800         /* initialize the International Bright Young Thing */
801 #ifdef ENABLE_NLS
802         initialize_locales();
803         locale = setlocale(LC_ALL, "");
804         mo = malloc(strlen(webcitdir) + 20);
805         lprintf(9, "Message catalog directory: %s\n", bindtextdomain("webcit", LOCALEDIR"/locale"));
806         free(mo);
807         lprintf(9, "Text domain: %s\n", textdomain("webcit"));
808         lprintf(9, "Text domain Charset: %s\n", bind_textdomain_codeset("webcit","UTF8"));
809         preset_locale();
810 #endif
811
812
813         /* calculate all our path on a central place */
814     /* where to keep our config */
815         
816 #define COMPUTE_DIRECTORY(SUBDIR) memcpy(dirbuffer,SUBDIR, sizeof dirbuffer);\
817         snprintf(SUBDIR,sizeof SUBDIR,  "%s%s%s%s%s%s%s", \
818                          (home&!relh)?webcitdir:basedir, \
819              ((basedir!=webcitdir)&(home&!relh))?basedir:"/", \
820              ((basedir!=webcitdir)&(home&!relh))?"/":"", \
821                          relhome, \
822              (relhome[0]!='\0')?"/":"",\
823                          dirbuffer,\
824                          (dirbuffer[0]!='\0')?"/":"");
825         basedir=RUNDIR;
826         COMPUTE_DIRECTORY(socket_dir);
827         basedir=WWWDIR "/static";
828         COMPUTE_DIRECTORY(static_dir);
829         basedir=WWWDIR "/static.local";
830         COMPUTE_DIRECTORY(static_local_dir);
831
832         snprintf(file_crpt_file_key,
833                  sizeof file_crpt_file_key, 
834                  "%s/citadel.key",
835                  ctdl_key_dir);
836         snprintf(file_crpt_file_csr,
837                  sizeof file_crpt_file_csr, 
838                  "%s/citadel.csr",
839                  ctdl_key_dir);
840         snprintf(file_crpt_file_cer,
841                  sizeof file_crpt_file_cer, 
842                  "%s/citadel.cer",
843                  ctdl_key_dir);
844
845         /* we should go somewhere we can leave our coredump, if enabled... */
846         lprintf(9, "Changing directory to %s\n", socket_dir);
847         if (chdir(webcitdir) != 0) {
848                 perror("chdir");
849         }       
850         initialize_viewdefs();
851         initialize_axdefs();
852
853         /*
854          * Set up a place to put thread-specific data.
855          * We only need a single pointer per thread - it points to the
856          * wcsession struct to which the thread is currently bound.
857          */
858         if (pthread_key_create(&MyConKey, NULL) != 0) {
859                 lprintf(1, "Can't create TSD key: %s\n", strerror(errno));
860         }
861         InitialiseSemaphores ();
862
863         /*
864          * Set up a place to put thread-specific SSL data.
865          * We don't stick this in the wcsession struct because SSL starts
866          * up before the session is bound, and it gets torn down between
867          * transactions.
868          */
869 #ifdef HAVE_OPENSSL
870         if (pthread_key_create(&ThreadSSL, NULL) != 0) {
871                 lprintf(1, "Can't create TSD key: %s\n", strerror(errno));
872         }
873 #endif
874
875         /*
876          * Bind the server to our favorite port.
877          * There is no need to check for errors, because ig_tcp_server()
878          * exits if it doesn't succeed.
879          */
880
881         if (!IsEmptyStr(uds_listen_path)) {
882                 lprintf(2, "Attempting to create listener socket at %s...\n", uds_listen_path);
883                 msock = ig_uds_server(uds_listen_path, LISTEN_QUEUE_LENGTH);
884         }
885         else {
886                 lprintf(2, "Attempting to bind to port %d...\n", http_port);
887                 msock = ig_tcp_server(ip_addr, http_port, LISTEN_QUEUE_LENGTH);
888         }
889
890         lprintf(2, "Listening on socket %d\n", msock);
891         signal(SIGPIPE, SIG_IGN);
892
893         pthread_mutex_init(&SessionListMutex, NULL);
894
895         /*
896          * Start up the housekeeping thread
897          */
898         pthread_attr_init(&attr);
899         pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
900         pthread_create(&SessThread, &attr,
901                        (void *(*)(void *)) housekeeping_loop, NULL);
902
903
904         /*
905          * If this is an HTTPS server, fire up SSL
906          */
907 #ifdef HAVE_OPENSSL
908         if (is_https) {
909                 init_ssl();
910         }
911 #endif
912
913         /* Start a few initial worker threads */
914         for (i = 0; i < (MIN_WORKER_THREADS); ++i) {
915                 spawn_another_worker_thread();
916         }
917
918         /* now the original thread becomes another worker */
919         worker_entry();
920         return 0;
921 }
922
923
924 /*
925  * Entry point for worker threads
926  */
927 void worker_entry(void)
928 {
929         int ssock;
930         int i = 0;
931         int fail_this_transaction = 0;
932         int ret;
933         struct timeval tv;
934         fd_set readset, tempset;
935
936         tv.tv_sec = 0;
937         tv.tv_usec = 10000;
938         FD_ZERO(&readset);
939         FD_SET(msock, &readset);
940
941         do {
942                 /* Only one thread can accept at a time */
943                 fail_this_transaction = 0;
944                 ssock = -1; 
945                 errno = EAGAIN;
946                 do {
947                         ret = -1; /* just one at once should select... */
948                         begin_critical_section(S_SELECT);
949
950                         FD_ZERO(&tempset);
951                         if (msock > 0) FD_SET(msock, &tempset);
952                         tv.tv_sec = 0;
953                         tv.tv_usec = 10000;
954                         if (msock > 0)  ret = select(msock+1, &tempset, NULL, NULL,  &tv);
955                         end_critical_section(S_SELECT);
956                         if ((ret < 0) && (errno != EINTR) && (errno != EAGAIN))
957                         {// EINTR and EAGAIN are thrown but not of interest.
958                                 lprintf(2, "accept() failed:%d %s\n",
959                                         errno, strerror(errno));
960                         }
961                         else if ((ret > 0) && (msock > 0) && FD_ISSET(msock, &tempset))
962                         {// Successfully selected, and still not shutting down? Accept!
963                                 ssock = accept(msock, NULL, 0);
964                         }
965                         
966                 } while ((msock > 0) && (ssock < 0)  && (time_to_die == 0));
967
968                 if ((msock == -1)||(time_to_die))
969                 {// ok, we're going down.
970                         int shutdown = 0;
971
972                         /* the first to come here will have to do the cleanup.
973                          * make shure its realy just one.
974                          */
975                         begin_critical_section(S_SHUTDOWN);
976                         if (msock == -1)
977                         {
978                                 msock = -2;
979                                 shutdown = 1;
980                         }
981                         end_critical_section(S_SHUTDOWN);
982                         if (shutdown == 1)
983                         {// we're the one to cleanup the mess.
984                                 lprintf(2, "I'm master shutdown: tagging sessions to be killed.\n");
985                                 shutdown_sessions();
986                                 lprintf(2, "master shutdown: waiting for others\n");
987                                 sleeeeeeeeeep(1); // wait so some others might finish...
988                                 lprintf(2, "master shutdown: cleaning up sessions\n");
989                                 do_housekeeping();
990 #ifdef WEBCIT_WITH_CALENDAR_SERVICE
991                                 lprintf(2, "master shutdown: cleaning up libical\n");
992                                 free_zone_directory ();
993                                 icaltimezone_release_zone_tab ();
994                                 icalmemory_free_ring ();
995 #endif
996                                 lprintf(2, "master shutdown exiting!.\n");                              
997                                 exit(0);
998                         }
999                         break;
1000                 }
1001                 if (ssock < 0 ) continue;
1002
1003                 if (msock < 0) {
1004                         if (ssock > 0) close (ssock);
1005                         lprintf(2, "inbetween.");
1006                         pthread_exit(NULL);
1007                 } else { // Got it? do some real work!
1008                         /* Set the SO_REUSEADDR socket option */
1009                         i = 1;
1010                         setsockopt(ssock, SOL_SOCKET, SO_REUSEADDR,
1011                                    &i, sizeof(i));
1012
1013                         /* If we are an HTTPS server, go crypto now. */
1014 #ifdef HAVE_OPENSSL
1015                         if (is_https) {
1016                                 if (starttls(ssock) != 0) {
1017                                         fail_this_transaction = 1;
1018                                         close(ssock);
1019                                 }
1020                         }
1021 #endif
1022
1023                         if (fail_this_transaction == 0) {
1024
1025                                 /* Perform an HTTP transaction... */
1026                                 context_loop(ssock);
1027
1028                                 /* Shut down SSL/TLS if required... */
1029 #ifdef HAVE_OPENSSL
1030                                 if (is_https) {
1031                                         endtls();
1032                                 }
1033 #endif
1034
1035                                 /* ...and close the socket. */
1036                                 lingering_close(ssock);
1037                         }
1038
1039                 }
1040
1041         } while (!time_to_die);
1042
1043         lprintf (1, "bye\n");
1044         pthread_exit(NULL);
1045 }
1046
1047 /*
1048  * \brief logprintf. log messages 
1049  * logs to stderr if loglevel is lower than the verbosity set at startup
1050  * \param loglevel level of the message
1051  * \param format the printf like format string
1052  * \param ... the strings to put into format
1053  */
1054 int lprintf(int loglevel, const char *format, ...)
1055 {
1056         va_list ap;
1057
1058         if (loglevel <= verbosity) {
1059                 va_start(ap, format);
1060                 vfprintf(stderr, format, ap);
1061                 va_end(ap);
1062                 fflush(stderr);
1063         }
1064         return 1;
1065 }
1066
1067
1068 /*
1069  * \brief print the actual stack frame.
1070  */
1071 void wc_backtrace(void)
1072 {
1073 #ifdef HAVE_BACKTRACE
1074         void *stack_frames[50];
1075         size_t size, i;
1076         char **strings;
1077
1078
1079         size = backtrace(stack_frames, sizeof(stack_frames) / sizeof(void*));
1080         strings = backtrace_symbols(stack_frames, size);
1081         for (i = 0; i < size; i++) {
1082                 if (strings != NULL)
1083                         lprintf(1, "%s\n", strings[i]);
1084                 else
1085                         lprintf(1, "%p\n", stack_frames[i]);
1086         }
1087         free(strings);
1088 #endif
1089 }
1090
1091 /*@}*/