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