Repaired other cases of auth mode failure. ICS calendar sync works better now.
[citadel.git] / webcit / context_loop.c
1 /*
2  * This is the other half of the webserver.  It handles the task of hooking
3  * up HTTP requests with the sessions they belong to, using HTTP cookies to
4  * keep track of things.  If the HTTP request doesn't belong to any currently
5  * active session, a new session is started.
6  *
7  * Copyright (c) 1996-2011 by the citadel.org team
8  *
9  * This program is open source software.  You can redistribute it and/or
10  * modify it under the terms of the GNU General Public License as
11  * published by the Free Software Foundation; either version 3 of the
12  * License, or (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22  */
23
24 #include "webcit.h"
25 #include "webserver.h"
26 #include "modules_init.h"
27
28 /* Only one thread may manipulate SessionList at a time... */
29 pthread_mutex_t SessionListMutex;
30
31 wcsession *SessionList = NULL;  /* Linked list of all webcit sessions */
32
33 pthread_key_t MyConKey;         /* TSD key for MySession() */
34 HashList *HttpReqTypes = NULL;
35 HashList *HttpHeaderHandler = NULL;
36 extern HashList *HandlerHash;
37
38 /* the following two values start at 1 because the initial parent thread counts as one. */
39 int num_threads_existing = 1;           /* Number of worker threads which exist. */
40 int num_threads_executing = 1;          /* Number of worker threads currently executing. */
41
42 extern void session_loop(void);
43 void spawn_another_worker_thread(void);
44
45
46 void DestroyHttpHeaderHandler(void *V)
47 {
48         OneHttpHeader *pHdr;
49         pHdr = (OneHttpHeader*) V;
50         FreeStrBuf(&pHdr->Val);
51         free(pHdr);
52 }
53
54 void shutdown_sessions(void)
55 {
56         wcsession *sptr;
57         
58         for (sptr = SessionList; sptr != NULL; sptr = sptr->next) {
59                 sptr->killthis = 1;
60         }
61 }
62
63 void do_housekeeping(void)
64 {
65         wcsession *sptr, *ss;
66         wcsession *sessions_to_kill = NULL;
67
68         /*
69          * Lock the session list, moving any candidates for euthanasia into
70          * a separate list.
71          */
72         CtdlLogResult(pthread_mutex_lock(&SessionListMutex));
73         for (sptr = SessionList; sptr != NULL; sptr = sptr->next) {
74
75                 /* Kill idle sessions */
76                 if ((time(NULL) - (sptr->lastreq)) > (time_t) WEBCIT_TIMEOUT) {
77                         syslog(3, "Timeout session %d", sptr->wc_session);
78                         sptr->killthis = 1;
79                 }
80
81                 /* Remove sessions flagged for kill */
82                 if (sptr->killthis) {
83
84                         /* remove session from linked list */
85                         if (sptr == SessionList) {
86                                 SessionList = SessionList->next;
87                         }
88                         else for (ss=SessionList;ss!=NULL;ss=ss->next) {
89                                 if (ss->next == sptr) {
90                                         ss->next = ss->next->next;
91                                 }
92                         }
93
94                         sptr->next = sessions_to_kill;
95                         sessions_to_kill = sptr;
96                 }
97         }
98         CtdlLogResult(pthread_mutex_unlock(&SessionListMutex));
99
100         /*
101          * Now free up and destroy the culled sessions.
102          */
103         while (sessions_to_kill != NULL) {
104                 syslog(3, "Destroying session %d", sessions_to_kill->wc_session);
105                 sptr = sessions_to_kill->next;
106                 session_destroy_modules(&sessions_to_kill);
107                 sessions_to_kill = sptr;
108         }
109 }
110
111 /*
112  * Check the size of our thread pool.  If all threads are executing, spawn another.
113  */
114 void check_thread_pool_size(void)
115 {
116         if (time_to_die) return;                /* don't expand the thread pool during shutdown */
117
118         begin_critical_section(S_SPAWNER);      /* only one of these should run at a time */
119         if (
120                 (num_threads_executing >= num_threads_existing)
121                 && (num_threads_existing < MAX_WORKER_THREADS)
122         ) {
123                 syslog(3, "%d of %d threads are executing.  Adding another worker thread.",
124                         num_threads_executing,
125                         num_threads_existing
126                 );
127                 spawn_another_worker_thread();
128         }
129         end_critical_section(S_SPAWNER);
130 }
131
132
133 /*
134  * Wake up occasionally and clean house
135  */
136 void housekeeping_loop(void)
137 {
138         while (1) {
139                 sleeeeeeeeeep(HOUSEKEEPING);
140                 do_housekeeping();
141         }
142 }
143
144
145 /*
146  * Create a Session id
147  * Generate a unique WebCit session ID (which is not the same thing as the
148  * Citadel session ID).
149  */
150 int GenerateSessionID(void)
151 {
152         static int seq = (-1);
153
154         if (seq < 0) {
155                 seq = (int) time(NULL);
156         }
157                 
158         return ++seq;
159 }
160
161 wcsession *FindSession(wcsession **wclist, ParsedHttpHdrs *Hdr, pthread_mutex_t *ListMutex)
162 {
163         wcsession *sptr = NULL;
164         wcsession *TheSession = NULL;   
165         
166         if (Hdr->HR.got_auth == AUTH_BASIC) {
167                 GetAuthBasic(Hdr);
168         }
169
170         CtdlLogResult(pthread_mutex_lock(ListMutex));
171         for (sptr = *wclist; ((sptr != NULL) && (TheSession == NULL)); sptr = sptr->next) {
172                 
173                 /* If HTTP-AUTH, look for a session with matching credentials */
174                 switch (Hdr->HR.got_auth)
175                 {
176                 case AUTH_BASIC:
177                         if (    (!strcasecmp(ChrPtr(Hdr->c_username), ChrPtr(sptr->wc_username)))
178                                 && (!strcasecmp(ChrPtr(Hdr->c_password), ChrPtr(sptr->wc_password)))
179                                 && (sptr->killthis == 0)
180                         ) {
181                                 syslog(LOG_DEBUG, "Matched a session with the same http-auth");
182                                 TheSession = sptr;
183                         }
184                         break;
185                 case AUTH_COOKIE:
186                         /* If cookie-session, look for a session with matching session ID */
187                         if (    (Hdr->HR.desired_session != 0)
188                                 && (sptr->wc_session == Hdr->HR.desired_session)
189                         ) {
190                                 syslog(LOG_DEBUG, "Matched a session with the same cookie");
191                                 TheSession = sptr;
192                         }
193                         break;                       
194                 case NO_AUTH:
195                         /* Any unbound session is a candidate */
196                         if ( (sptr->wc_session == 0) && (sptr->inuse == 0) ) {
197                                 syslog(LOG_DEBUG, "Reusing an unbound session");
198                                 TheSession = sptr;
199                         }
200                         break;
201                 }
202         }
203         CtdlLogResult(pthread_mutex_unlock(ListMutex));
204         if (TheSession == NULL) {
205                 syslog(LOG_DEBUG, "No existing session was matched");
206         }
207         return TheSession;
208 }
209
210 wcsession *CreateSession(int Lockable, int Static, wcsession **wclist, ParsedHttpHdrs *Hdr, pthread_mutex_t *ListMutex)
211 {
212         wcsession *TheSession;
213         TheSession = (wcsession *) malloc(sizeof(wcsession));
214         memset(TheSession, 0, sizeof(wcsession));
215         TheSession->Hdr = Hdr;
216         TheSession->serv_sock = (-1);
217
218         pthread_setspecific(MyConKey, (void *)TheSession);
219         
220         /* If we're recreating a session that expired, it's best to give it the same
221          * session number that it had before.  The client browser ought to pick up
222          * the new session number and start using it, but in some rare situations it
223          * doesn't, and that's a Bad Thing because it causes lots of spurious sessions
224          * to get created.
225          */     
226         if (Hdr->HR.desired_session == 0) {
227                 TheSession->wc_session = GenerateSessionID();
228                 syslog(3, "Created new session %d", TheSession->wc_session);
229         }
230         else {
231                 TheSession->wc_session = Hdr->HR.desired_session;
232                 syslog(3, "Re-created session %d", TheSession->wc_session);
233         }
234         Hdr->HR.Static = Static;
235         session_new_modules(TheSession);
236
237         if (Lockable) {
238                 pthread_mutex_init(&TheSession->SessionMutex, NULL);
239
240                 if (ListMutex != NULL)
241                         CtdlLogResult(pthread_mutex_lock(ListMutex));
242
243                 if (wclist != NULL) {
244                         TheSession->nonce = rand();
245                         TheSession->next = *wclist;
246                         *wclist = TheSession;
247                 }
248                 if (ListMutex != NULL)
249                         CtdlLogResult(pthread_mutex_unlock(ListMutex));
250         }
251         return TheSession;
252 }
253
254
255 /* If it's a "force 404" situation then display the error and bail. */
256 void do_404(void)
257 {
258         hprintf("HTTP/1.1 404 Not found\r\n");
259         hprintf("Content-Type: text/plain\r\n");
260         wc_printf("Not found\r\n");
261         end_burst();
262 }
263
264 int ReadHttpSubject(ParsedHttpHdrs *Hdr, StrBuf *Line, StrBuf *Buf)
265 {
266         const char *Args;
267         void *vLine, *vHandler;
268         const char *Pos = NULL;
269
270
271         Hdr->HR.ReqLine = Line;
272         /* The requesttype... GET, POST... */
273         StrBufExtract_token(Buf, Hdr->HR.ReqLine, 0, ' ');
274         if (GetHash(HttpReqTypes, SKEY(Buf), &vLine) &&
275             (vLine != NULL))
276         {
277                 Hdr->HR.eReqType = *(long*)vLine;
278         }
279         else {
280                 Hdr->HR.eReqType = eGET;
281                 return 1;
282         }
283         StrBufCutLeft(Hdr->HR.ReqLine, StrLength(Buf) + 1);
284
285         /* the HTTP Version... */
286         StrBufExtract_token(Buf, Hdr->HR.ReqLine, 1, ' ');
287         StrBufCutRight(Hdr->HR.ReqLine, StrLength(Buf) + 1);
288         
289         if (StrLength(Buf) == 0) {
290                 Hdr->HR.eReqType = eGET;
291                 return 1;
292         }
293
294         StrBufAppendBuf(Hdr->this_page, Hdr->HR.ReqLine, 0);
295
296         /* chop Filename / query arguments */
297         Args = strchr(ChrPtr(Hdr->HR.ReqLine), '?');
298         if (Args == NULL) /* whe're not that picky about params... TODO: this will spoil '&' in filenames.*/
299                 Args = strchr(ChrPtr(Hdr->HR.ReqLine), '&');
300         if (Args != NULL) {
301                 Args ++; /* skip the ? */
302                 StrBufPlain(Hdr->PlainArgs, 
303                             Args, 
304                             StrLength(Hdr->HR.ReqLine) -
305                             (Args - ChrPtr(Hdr->HR.ReqLine)));
306                 StrBufCutAt(Hdr->HR.ReqLine, 0, Args - 1);
307         } /* don't parse them yet, maybe we don't even care... */
308         
309         /* now lookup what we are going to do with this... */
310         /* skip first slash */
311         StrBufExtract_NextToken(Buf, Hdr->HR.ReqLine, &Pos, '/');
312         do {
313                 StrBufExtract_NextToken(Buf, Hdr->HR.ReqLine, &Pos, '/');
314
315                 GetHash(HandlerHash, SKEY(Buf), &vHandler),
316                 Hdr->HR.Handler = (WebcitHandler*) vHandler;
317                 if (Hdr->HR.Handler == NULL)
318                         break;
319                 /*
320                  * If the request is prefixed by "/webcit" then chop that off.  This
321                  * allows a front end web server to forward all /webcit requests to us
322                  * while still using the same web server port for other things.
323                  */
324                 if ((Hdr->HR.Handler->Flags & URLNAMESPACE) != 0)
325                         continue;
326                 break;
327         } while (1);
328         /* remove the handlername from the URL */
329         if ((Pos != NULL) && (Pos != StrBufNOTNULL)){
330                 StrBufCutLeft(Hdr->HR.ReqLine, 
331                               Pos - ChrPtr(Hdr->HR.ReqLine));
332         }
333
334         if (Hdr->HR.Handler != NULL) {
335                 if ((Hdr->HR.Handler->Flags & BOGUS) != 0)
336                         return 1;
337                 Hdr->HR.DontNeedAuth = (
338                         ((Hdr->HR.Handler->Flags & ISSTATIC) != 0) ||
339                         ((Hdr->HR.Handler->Flags & ANONYMOUS) != 0)
340                         );
341         }
342         else {
343                 Hdr->HR.DontNeedAuth = 1; /* Flat request? show him the login screen... */
344         }
345
346         return 0;
347 }
348
349 int AnalyseHeaders(ParsedHttpHdrs *Hdr)
350 {
351         OneHttpHeader *pHdr;
352         void *vHdr;
353         long HKLen;
354         const char *HashKey;
355         HashPos *at = GetNewHashPos(Hdr->HTTPHeaders, 0);
356         
357         while (GetNextHashPos(Hdr->HTTPHeaders, at, &HKLen, &HashKey, &vHdr) && 
358                (vHdr != NULL)) {
359                 pHdr = (OneHttpHeader *)vHdr;
360                 if (pHdr->HaveEvaluator)
361                         pHdr->H(pHdr->Val, Hdr);
362
363         }
364         DeleteHashPos(&at);
365         return 0;
366 }
367
368 /*const char *nix(void *vptr) {return ChrPtr( (StrBuf*)vptr);}*/
369
370 /*
371  * Read in the request
372  */
373 int ReadHTTPRequest (ParsedHttpHdrs *Hdr)
374 {
375         const char *pch, *pchs, *pche;
376         OneHttpHeader *pHdr;
377         StrBuf *Line, *LastLine, *HeaderName;
378         int nLine = 0;
379         void *vF;
380         int isbogus = 0;
381
382         HeaderName = NewStrBuf();
383         LastLine = NULL;
384         do {
385                 nLine ++;
386                 Line = NewStrBufPlain(NULL, SIZ / 4);
387
388                 if (ClientGetLine(Hdr, Line) < 0) return 1;
389
390                 if (StrLength(Line) == 0) {
391                         FreeStrBuf(&Line);
392                         continue;
393                 }
394                 if (nLine == 1) {
395                         Hdr->HTTPHeaders = NewHash(1, NULL);
396                         pHdr = (OneHttpHeader*) malloc(sizeof(OneHttpHeader));
397                         memset(pHdr, 0, sizeof(OneHttpHeader));
398                         pHdr->Val = Line;
399                         Put(Hdr->HTTPHeaders, HKEY("GET /"), pHdr, DestroyHttpHeaderHandler);
400                         syslog(9, "%s", ChrPtr(Line));
401                         isbogus = ReadHttpSubject(Hdr, Line, HeaderName);
402                         if (isbogus) break;
403                         continue;
404                 }
405
406                 /* Do we need to Unfold? */
407                 if ((LastLine != NULL) && 
408                     (isspace(*ChrPtr(Line)))) {
409                         pch = pchs = ChrPtr(Line);
410                         pche = pchs + StrLength(Line);
411                         while (isspace(*pch) && (pch < pche))
412                                 pch ++;
413                         StrBufCutLeft(Line, pch - pchs);
414                         StrBufAppendBuf(LastLine, Line, 0);
415
416                         FreeStrBuf(&Line);
417                         continue;
418                 }
419
420                 StrBufSanitizeAscii(Line, '§');
421                 StrBufExtract_token(HeaderName, Line, 0, ':');
422
423                 pchs = ChrPtr(Line);
424                 pche = pchs + StrLength(Line);
425                 pch = pchs + StrLength(HeaderName) + 1;
426                 pche = pchs + StrLength(Line);
427                 while ((pch < pche) && isspace(*pch))
428                         pch ++;
429                 StrBufCutLeft(Line, pch - pchs);
430
431                 StrBufUpCase(HeaderName);
432
433                 pHdr = (OneHttpHeader*) malloc(sizeof(OneHttpHeader));
434                 memset(pHdr, 0, sizeof(OneHttpHeader));
435                 pHdr->Val = Line;
436
437                 if (GetHash(HttpHeaderHandler, SKEY(HeaderName), &vF) &&
438                     (vF != NULL))
439                 {
440                         OneHttpHeader *FHdr = (OneHttpHeader*) vF;
441                         pHdr->H = FHdr->H;
442                         pHdr->HaveEvaluator = 1;
443                 }
444                 Put(Hdr->HTTPHeaders, SKEY(HeaderName), pHdr, DestroyHttpHeaderHandler);
445                 LastLine = Line;
446         } while (Line != NULL);
447
448         FreeStrBuf(&HeaderName);
449
450         return isbogus;
451 }
452
453 void OverrideRequest(ParsedHttpHdrs *Hdr, const char *Line, long len)
454 {
455         StrBuf *Buf = NewStrBuf();
456
457         if (Hdr->HR.ReqLine != NULL) {
458                 FlushStrBuf(Hdr->HR.ReqLine);
459                 StrBufPlain(Hdr->HR.ReqLine, Line, len);
460         }
461         else {
462                 Hdr->HR.ReqLine = NewStrBufPlain(Line, len);
463         }
464         ReadHttpSubject(Hdr, Hdr->HR.ReqLine, Buf);
465
466         FreeStrBuf(&Buf);
467 }
468
469 /*
470  * handle one request
471  *
472  * This loop gets called once for every HTTP connection made to WebCit.  At
473  * this entry point we have an HTTP socket with a browser allegedly on the
474  * other end, but we have not yet bound to a WebCit session.
475  *
476  * The job of this function is to locate the correct session and bind to it,
477  * or create a session if necessary and bind to it, then run the WebCit
478  * transaction loop.  Afterwards, we unbind from the session.  When this
479  * function returns, the worker thread is then free to handle another
480  * transaction.
481  */
482 void context_loop(ParsedHttpHdrs *Hdr)
483 {
484         int isbogus = 0;
485         wcsession *TheSession;
486         struct timeval tx_start;
487         struct timeval tx_finish;
488         int session_may_be_reused = 1;
489         
490         gettimeofday(&tx_start, NULL);          /* start a stopwatch for performance timing */
491
492         /*
493          * Find out what it is that the web browser is asking for
494          */
495         isbogus = ReadHTTPRequest(Hdr);
496
497         Hdr->HR.dav_depth = 32767; /* TODO: find a general way to have non-0 defaults */
498
499         if (!isbogus) {
500                 isbogus = AnalyseHeaders(Hdr);
501         }
502
503         if (    (isbogus)
504                 || ((Hdr->HR.Handler != NULL)
505                 && ((Hdr->HR.Handler->Flags & BOGUS) != 0))
506         ) {
507                 wcsession *Bogus;
508                 Bogus = CreateSession(0, 1, NULL, Hdr, NULL);
509                 do_404();
510                 syslog(9, "HTTP: 404 [%ld.%06ld] %s %s",
511                         ((tx_finish.tv_sec*1000000 + tx_finish.tv_usec) - (tx_start.tv_sec*1000000 + tx_start.tv_usec)) / 1000000,
512                         ((tx_finish.tv_sec*1000000 + tx_finish.tv_usec) - (tx_start.tv_sec*1000000 + tx_start.tv_usec)) % 1000000,
513                         ReqStrs[Hdr->HR.eReqType],
514                         ChrPtr(Hdr->this_page)
515                         );
516                 session_detach_modules(Bogus);
517                 session_destroy_modules(&Bogus);
518                 return;
519         }
520
521         if ((Hdr->HR.Handler != NULL) && ((Hdr->HR.Handler->Flags & ISSTATIC) != 0))
522         {
523                 wcsession *Static;
524                 Static = CreateSession(0, 1, NULL, Hdr, NULL);
525                 
526                 Hdr->HR.Handler->F();
527
528                 /* How long did this transaction take? */
529                 gettimeofday(&tx_finish, NULL);
530                 
531                 syslog(9, "HTTP: 200 [%ld.%06ld] %s %s",
532                         ((tx_finish.tv_sec*1000000 + tx_finish.tv_usec) - (tx_start.tv_sec*1000000 + tx_start.tv_usec)) / 1000000,
533                         ((tx_finish.tv_sec*1000000 + tx_finish.tv_usec) - (tx_start.tv_sec*1000000 + tx_start.tv_usec)) % 1000000,
534                         ReqStrs[Hdr->HR.eReqType],
535                         ChrPtr(Hdr->this_page)
536                 );
537                 session_detach_modules(Static);
538                 session_destroy_modules(&Static);
539                 return;
540         }
541
542         if (Hdr->HR.got_auth == AUTH_BASIC) {
543                 CheckAuthBasic(Hdr);
544         }
545
546         if (Hdr->HR.got_auth) {
547                 session_may_be_reused = 0;
548         }
549
550         /*
551          * See if there's an existing session open with any of:
552          * - The desired Session ID
553          * - A matching http-auth username and password
554          * - An unbound session flagged as reusable
555          */
556         TheSession = FindSession(&SessionList, Hdr, &SessionListMutex);
557
558         /*
559          * If there were no qualifying sessions, then create a new one.
560          */
561         if (TheSession == NULL) {
562                 TheSession = CreateSession(1, 0, &SessionList, Hdr, &SessionListMutex);
563         }
564
565         /*
566          * Reject transactions which require http-auth, if http-auth was not provided
567          */
568         if (    (StrLength(Hdr->c_username) == 0)
569                 && (!Hdr->HR.DontNeedAuth)
570                 && (Hdr->HR.Handler != NULL)
571                 && ((XHTTP_COMMANDS & Hdr->HR.Handler->Flags) == XHTTP_COMMANDS)
572         ) {
573                 syslog(LOG_DEBUG, "http-auth required but not provided");
574                 OverrideRequest(Hdr, HKEY("GET /401 HTTP/1.0"));
575                 Hdr->HR.prohibit_caching = 1;                           
576         }
577
578         /*
579          * A future improvement might be to check the session integrity
580          * at this point before continuing.
581          */
582
583         /*
584          * Bind to the session and perform the transaction
585          */
586         CtdlLogResult(pthread_mutex_lock(&TheSession->SessionMutex));
587         pthread_setspecific(MyConKey, (void *)TheSession);
588         
589         TheSession->inuse = 1;                                  /* mark the session as bound */
590         TheSession->lastreq = time(NULL);                       /* log */
591         TheSession->Hdr = Hdr;
592
593         /*
594          * If a language was requested via a cookie, select that language now.
595          */
596         if (StrLength(Hdr->c_language) > 0) {
597                 syslog(LOG_DEBUG, "Session cookie requests language '%s'", ChrPtr(Hdr->c_language));
598                 set_selected_language(ChrPtr(Hdr->c_language));
599                 go_selected_language();
600         }
601
602         /*
603          * do the transaction
604          */
605         session_attach_modules(TheSession);
606         session_loop();
607
608         /* How long did this transaction take? */
609         gettimeofday(&tx_finish, NULL);
610
611         syslog(9, "HTTP: 200 [%ld.%06ld] %s %s",
612                 ((tx_finish.tv_sec*1000000 + tx_finish.tv_usec) - (tx_start.tv_sec*1000000 + tx_start.tv_usec)) / 1000000,
613                 ((tx_finish.tv_sec*1000000 + tx_finish.tv_usec) - (tx_start.tv_sec*1000000 + tx_start.tv_usec)) % 1000000,
614                 ReqStrs[Hdr->HR.eReqType],
615                 ChrPtr(Hdr->this_page)
616         );
617
618         session_detach_modules(TheSession);
619
620         /* If *this* very transaction did not explicitly specify a session cookie,
621          * and it did not log in, we want to flag the session as a candidate for
622          * re-use by the next unbound client that comes along.  This keeps our session
623          * table from getting bombarded with new sessions when, for example, a web
624          * spider crawls the site without using cookies.
625          */
626         if ((session_may_be_reused) && (!WC->logged_in)) {
627                 WC->wc_session = 0;                     /* flag as available for re-use */
628                 TheSession->selected_language = 0;      /* clear any non-default language setting */
629         }
630
631         TheSession->Hdr = NULL;
632         TheSession->inuse = 0;                                  /* mark the session as unbound */
633         CtdlLogResult(pthread_mutex_unlock(&TheSession->SessionMutex));
634 }
635
636 void tmplput_nonce(StrBuf *Target, WCTemplputParams *TP)
637 {
638         wcsession *WCC = WC;
639         StrBufAppendPrintf(Target, "%ld",
640                            (WCC != NULL)? WCC->nonce:0);                   
641 }
642
643 void tmplput_current_user(StrBuf *Target, WCTemplputParams *TP)
644 {
645         StrBufAppendTemplate(Target, TP, WC->wc_fullname, 0);
646 }
647
648 void Header_HandleContentLength(StrBuf *Line, ParsedHttpHdrs *hdr)
649 {
650         hdr->HR.ContentLength = StrToi(Line);
651 }
652
653 void Header_HandleContentType(StrBuf *Line, ParsedHttpHdrs *hdr)
654 {
655         hdr->HR.ContentType = Line;
656 }
657
658
659 void Header_HandleHost(StrBuf *Line, ParsedHttpHdrs *hdr)
660 {
661         if (hdr->HostHeader != NULL) {
662                 FreeStrBuf(&hdr->HostHeader);
663         }
664         hdr->HostHeader = NewStrBuf();
665         StrBufAppendPrintf(hdr->HostHeader, "%s://", (is_https ? "https" : "http") );
666         StrBufAppendBuf(hdr->HostHeader, Line, 0);
667 }
668
669 void Header_HandleXFFHost(StrBuf *Line, ParsedHttpHdrs *hdr)
670 {
671         if (!follow_xff) return;
672
673         if (hdr->HostHeader != NULL) {
674                 FreeStrBuf(&hdr->HostHeader);
675         }
676
677         hdr->HostHeader = NewStrBuf();
678         StrBufAppendPrintf(hdr->HostHeader, "http://"); /* this is naive; do something about it */
679         StrBufAppendBuf(hdr->HostHeader, Line, 0);
680 }
681
682
683 void Header_HandleXFF(StrBuf *Line, ParsedHttpHdrs *hdr)
684 {
685         hdr->HR.browser_host = Line;
686
687         while (StrBufNum_tokens(hdr->HR.browser_host, ',') > 1) {
688                 StrBufRemove_token(hdr->HR.browser_host, 0, ',');
689         }
690         StrBufTrim(hdr->HR.browser_host);
691 }
692
693 void Header_HandleIfModSince(StrBuf *Line, ParsedHttpHdrs *hdr)
694 {
695         hdr->HR.if_modified_since = httpdate_to_timestamp(Line);
696 }
697
698 void Header_HandleAcceptEncoding(StrBuf *Line, ParsedHttpHdrs *hdr)
699 {
700         /*
701          * Can we compress?
702          */
703         if (strstr(&ChrPtr(Line)[16], "gzip")) {
704                 hdr->HR.gzip_ok = 1;
705         }
706 }
707 const char *ReqStrs[eNONE] = {
708         "GET",
709         "POST",
710         "OPTIONS",
711         "PROPFIND",
712         "PUT",
713         "DELETE",
714         "HEAD",
715         "MOVE",
716         "COPY"
717 };
718
719 void
720 ServerStartModule_CONTEXT
721 (void)
722 {
723         long *v;
724         HttpReqTypes = NewHash(1, NULL);
725         HttpHeaderHandler = NewHash(1, NULL);
726
727         v = malloc(sizeof(long));
728         *v = eGET;
729         Put(HttpReqTypes, HKEY("GET"), v, NULL);
730
731         v = malloc(sizeof(long));
732         *v = ePOST;
733         Put(HttpReqTypes, HKEY("POST"), v, NULL);
734
735         v = malloc(sizeof(long));
736         *v = eOPTIONS;
737         Put(HttpReqTypes, HKEY("OPTIONS"), v, NULL);
738
739         v = malloc(sizeof(long));
740         *v = ePROPFIND;
741         Put(HttpReqTypes, HKEY("PROPFIND"), v, NULL);
742
743         v = malloc(sizeof(long));
744         *v = ePUT;
745         Put(HttpReqTypes, HKEY("PUT"), v, NULL);
746
747         v = malloc(sizeof(long));
748         *v = eDELETE;
749         Put(HttpReqTypes, HKEY("DELETE"), v, NULL);
750
751         v = malloc(sizeof(long));
752         *v = eHEAD;
753         Put(HttpReqTypes, HKEY("HEAD"), v, NULL);
754
755         v = malloc(sizeof(long));
756         *v = eMOVE;
757         Put(HttpReqTypes, HKEY("MOVE"), v, NULL);
758
759         v = malloc(sizeof(long));
760         *v = eCOPY;
761         Put(HttpReqTypes, HKEY("COPY"), v, NULL);
762 }
763
764 void 
765 ServerShutdownModule_CONTEXT
766 (void)
767 {
768         DeleteHash(&HttpReqTypes);
769         DeleteHash(&HttpHeaderHandler);
770 }
771
772 void RegisterHeaderHandler(const char *Name, long Len, Header_Evaluator F)
773 {
774         OneHttpHeader *pHdr;
775         pHdr = (OneHttpHeader*) malloc(sizeof(OneHttpHeader));
776         memset(pHdr, 0, sizeof(OneHttpHeader));
777         pHdr->H = F;
778         Put(HttpHeaderHandler, Name, Len, pHdr, DestroyHttpHeaderHandler);
779 }
780
781
782 void 
783 InitModule_CONTEXT
784 (void)
785 {
786         RegisterHeaderHandler(HKEY("CONTENT-LENGTH"), Header_HandleContentLength);
787         RegisterHeaderHandler(HKEY("CONTENT-TYPE"), Header_HandleContentType);
788         RegisterHeaderHandler(HKEY("X-FORWARDED-HOST"), Header_HandleXFFHost); /* Apache way... */
789         RegisterHeaderHandler(HKEY("X-REAL-IP"), Header_HandleXFFHost);        /* NGinX way... */
790         RegisterHeaderHandler(HKEY("HOST"), Header_HandleHost);
791         RegisterHeaderHandler(HKEY("X-FORWARDED-FOR"), Header_HandleXFF);
792         RegisterHeaderHandler(HKEY("ACCEPT-ENCODING"), Header_HandleAcceptEncoding);
793         RegisterHeaderHandler(HKEY("IF-MODIFIED-SINCE"), Header_HandleIfModSince);
794
795         RegisterNamespace("CURRENT_USER", 0, 1, tmplput_current_user, NULL, CTX_NONE);
796         RegisterNamespace("NONCE", 0, 0, tmplput_nonce, NULL, 0);
797
798         WebcitAddUrlHandler(HKEY("404"), "", 0, do_404, ANONYMOUS|COOKIEUNNEEDED);
799 /*
800  * Look for commonly-found probes of malware such as worms, viruses, trojans, and Microsoft Office.
801  * Short-circuit these requests so we don't have to send them through the full processing loop.
802  */
803         WebcitAddUrlHandler(HKEY("scripts"), "", 0, do_404, ANONYMOUS|BOGUS);           /* /root.exe - Worms and trojans and viruses, oh my! */
804         WebcitAddUrlHandler(HKEY("c"), "", 0, do_404, ANONYMOUS|BOGUS);         /* /winnt */
805         WebcitAddUrlHandler(HKEY("MSADC"), "", 0, do_404, ANONYMOUS|BOGUS);
806         WebcitAddUrlHandler(HKEY("_vti"), "", 0, do_404, ANONYMOUS|BOGUS);              /* Broken Microsoft DAV implementation */
807         WebcitAddUrlHandler(HKEY("MSOffice"), "", 0, do_404, ANONYMOUS|BOGUS);          /* Stoopid MSOffice thinks everyone is IIS */
808         WebcitAddUrlHandler(HKEY("nonexistenshit"), "", 0, do_404, ANONYMOUS|BOGUS);    /* Exploit found in the wild January 2009 */
809 }
810         
811
812 void 
813 HttpNewModule_CONTEXT
814 (ParsedHttpHdrs *httpreq)
815 {
816         httpreq->PlainArgs = NewStrBufPlain(NULL, SIZ);
817         httpreq->this_page = NewStrBufPlain(NULL, SIZ);
818 }
819
820 void 
821 HttpDetachModule_CONTEXT
822 (ParsedHttpHdrs *httpreq)
823 {
824         FlushStrBuf(httpreq->PlainArgs);
825         FlushStrBuf(httpreq->HostHeader);
826         FlushStrBuf(httpreq->this_page);
827         FlushStrBuf(httpreq->PlainArgs);
828         DeleteHash(&httpreq->HTTPHeaders);
829         memset(&httpreq->HR, 0, sizeof(HdrRefs));
830 }
831
832 void 
833 HttpDestroyModule_CONTEXT
834 (ParsedHttpHdrs *httpreq)
835 {
836         FreeStrBuf(&httpreq->this_page);
837         FreeStrBuf(&httpreq->PlainArgs);
838         FreeStrBuf(&httpreq->this_page);
839         FreeStrBuf(&httpreq->PlainArgs);
840         FreeStrBuf(&httpreq->HostHeader);
841         DeleteHash(&httpreq->HTTPHeaders);
842
843 }