projects / catagits/fcgi2.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | history | HEAD
Remove the printf() and #include of stdio.h from examples/echo2.c.
[catagits/fcgi2.git] / libfcgi / fcgiapp.c
CommitLineData
0198fd3c 1/*
2 * fcgiapp.c --
3 *
4 * FastCGI application library: request-at-a-time
5 *
6 *
7 * Copyright (c) 1996 Open Market, Inc.
8 *
9 * See the file "LICENSE.TERMS" for information on usage and redistribution
10 * of this file, and for a DISCLAIMER OF ALL WARRANTIES.
11 *
12 */
13
14#ifndef lint
19d98b91 15static const char rcsid[] = "$Id: fcgiapp.c,v 1.3 1999/02/12 00:46:41 roberts Exp $";
0198fd3c 16#endif /* not lint */
17
18#ifdef _WIN32
19#define DLLAPI __declspec(dllexport)
20#endif
21
22#include <stdio.h>
23#include <sys/types.h>
24#ifdef HAVE_SYS_TIME_H
25#include <sys/time.h>
26#endif
27
28#include "fcgi_config.h"
29#ifdef HAVE_UNISTD_H
30#include <unistd.h>
31#endif
32
33#include <assert.h>
34#include <stdlib.h>
35#include <string.h>
36#include <memory.h> /* for memchr() */
37#include <errno.h>
38#include <stdarg.h>
39#include <math.h>
40#ifdef HAVE_SYS_SOCKET_H
41#include <sys/socket.h> /* for getpeername */
42#endif
43#include <fcntl.h> /* for fcntl */
44
45#include "fcgimisc.h"
46#include "fcgiapp.h"
47#include "fcgiappmisc.h"
48#include "fastcgi.h"
49#include "fcgios.h"
50
51/*
52 * This is a workaround for one version of the HP C compiler
53 * (c89 on HP-UX 9.04, also Stratus FTX), which will dump core
54 * if given 'long double' for varargs.
55 */
56#ifdef HAVE_VA_ARG_LONG_DOUBLE_BUG
57#define LONG_DOUBLE double
58#else
59#define LONG_DOUBLE long double
60#endif
61
62static int osLibInitialized = 0;
63
64static void *Malloc(size_t size)
65{
66 void *result = malloc(size);
67 ASSERT(size == 0 || result != NULL);
68 return result;
69}
70
71static char *StringCopy(char *str)
72{
73 int strLen = strlen(str);
74 char *newString = Malloc(strLen + 1);
75 memcpy(newString, str, strLen);
76 newString[strLen] = '\000';
77 return newString;
78}
79
80\f
81/*
82 *----------------------------------------------------------------------
83 *
84 * FCGX_GetChar --
85 *
86 * Reads a byte from the input stream and returns it.
87 *
88 * Results:
89 * The byte, or EOF (-1) if the end of input has been reached.
90 *
91 *----------------------------------------------------------------------
92 */
93int FCGX_GetChar(FCGX_Stream *stream)
94{
95 if(stream->rdNext != stream->stop)
96 return *stream->rdNext++;
97 if(stream->isClosed || !stream->isReader)
98 return EOF;
99 stream->fillBuffProc(stream);
100 stream->stopUnget = stream->rdNext;
101 if(stream->rdNext != stream->stop)
102 return *stream->rdNext++;
103 ASSERT(stream->isClosed); /* bug in fillBufProc if not */
104 return EOF;
105}
106\f
107/*
108 *----------------------------------------------------------------------
109 *
110 * FCGX_GetStr --
111 *
112 * Reads up to n consecutive bytes from the input stream
113 * into the character array str. Performs no interpretation
114 * of the input bytes.
115 *
116 * Results:
117 * Number of bytes read. If result is smaller than n,
118 * the end of input has been reached.
119 *
120 *----------------------------------------------------------------------
121 */
122int FCGX_GetStr(char *str, int n, FCGX_Stream *stream)
123{
124 int m, bytesMoved;
125
126 if(n <= 0) {
127 return 0;
128 }
129 /*
130 * Fast path: n bytes are already available
131 */
132 if(n <= (stream->stop - stream->rdNext)) {
133 memcpy(str, stream->rdNext, n);
134 stream->rdNext += n;
135 return n;
136 }
137 /*
138 * General case: stream is closed or buffer fill procedure
139 * needs to be called
140 */
141 bytesMoved = 0;
142 for (;;) {
143 if(stream->rdNext != stream->stop) {
144 m = min(n - bytesMoved, stream->stop - stream->rdNext);
145 memcpy(str, stream->rdNext, m);
146 bytesMoved += m;
147 stream->rdNext += m;
148 if(bytesMoved == n)
149 return bytesMoved;
150 str += m;
151 }
152 if(stream->isClosed || !stream->isReader)
153 return bytesMoved;
154 stream->fillBuffProc(stream);
155 stream->stopUnget = stream->rdNext;
156 }
157}
158\f
159/*
160 *----------------------------------------------------------------------
161 *
162 * FCGX_GetLine --
163 *
164 * Reads up to n-1 consecutive bytes from the input stream
165 * into the character array str. Stops before n-1 bytes
166 * have been read if '\n' or EOF is read. The terminating '\n'
167 * is copied to str. After copying the last byte into str,
168 * stores a '\0' terminator.
169 *
170 * Results:
171 * NULL if EOF is the first thing read from the input stream,
172 * str otherwise.
173 *
174 *----------------------------------------------------------------------
175 */
176char *FCGX_GetLine(char *str, int n, FCGX_Stream *stream)
177{
178 int c;
179 char *p = str;
180 n--;
181 while (n > 0) {
182 c = FCGX_GetChar(stream);
183 if(c == EOF) {
184 if(p == str)
185 return NULL;
186 else
187 break;
188 }
189 *p++ = c;
190 n--;
191 if(c == '\n')
192 break;
193 }
194 *p = '\0';
195 return str;
196}
197\f
198/*
199 *----------------------------------------------------------------------
200 *
201 * FCGX_UnGetChar --
202 *
203 * Pushes back the character c onto the input stream. One
204 * character of pushback is guaranteed once a character
205 * has been read. No pushback is possible for EOF.
206 *
207 * Results:
208 * Returns c if the pushback succeeded, EOF if not.
209 *
210 *----------------------------------------------------------------------
211 */
212int FCGX_UnGetChar(int c, FCGX_Stream *stream) {
213 if(c == EOF
214 || stream->isClosed
215 || !stream->isReader
216 || stream->rdNext == stream->stopUnget)
217 return EOF;
218 --(stream->rdNext);
219 *stream->rdNext = c;
220 return c;
221}
222
223/*
224 *----------------------------------------------------------------------
225 *
226 * FCGX_HasSeenEOF --
227 *
228 * Returns EOF if end-of-file has been detected while reading
229 * from stream; otherwise returns 0.
230 *
231 * Note that FCGX_HasSeenEOF(s) may return 0, yet an immediately
232 * following FCGX_GetChar(s) may return EOF. This function, like
233 * the standard C stdio function feof, does not provide the
234 * ability to peek ahead.
235 *
236 * Results:
237 * EOF if end-of-file has been detected, 0 if not.
238 *
239 *----------------------------------------------------------------------
240 */
241int FCGX_HasSeenEOF(FCGX_Stream *stream) {
242 return (stream->isClosed) ? EOF : 0;
243}
244\f
245/*
246 *----------------------------------------------------------------------
247 *
248 * FCGX_PutChar --
249 *
250 * Writes a byte to the output stream.
251 *
252 * Results:
253 * The byte, or EOF (-1) if an error occurred.
254 *
255 *----------------------------------------------------------------------
256 */
257int FCGX_PutChar(int c, FCGX_Stream *stream)
258{
259 if(stream->wrNext != stream->stop)
260 return (*stream->wrNext++ = c);
261 if(stream->isClosed || stream->isReader)
262 return EOF;
263 stream->emptyBuffProc(stream, FALSE);
264 if(stream->wrNext != stream->stop)
265 return (*stream->wrNext++ = c);
266 ASSERT(stream->isClosed); /* bug in emptyBuffProc if not */
267 return EOF;
268}
269\f
270/*
271 *----------------------------------------------------------------------
272 *
273 * FCGX_PutStr --
274 *
275 * Writes n consecutive bytes from the character array str
276 * into the output stream. Performs no interpretation
277 * of the output bytes.
278 *
279 * Results:
280 * Number of bytes written (n) for normal return,
281 * EOF (-1) if an error occurred.
282 *
283 *----------------------------------------------------------------------
284 */
285int FCGX_PutStr(const char *str, int n, FCGX_Stream *stream)
286{
287 int m, bytesMoved;
288
289 /*
290 * Fast path: room for n bytes in the buffer
291 */
292 if(n <= (stream->stop - stream->wrNext)) {
293 memcpy(stream->wrNext, str, n);
294 stream->wrNext += n;
295 return n;
296 }
297 /*
298 * General case: stream is closed or buffer empty procedure
299 * needs to be called
300 */
301 bytesMoved = 0;
302 for (;;) {
303 if(stream->wrNext != stream->stop) {
304 m = min(n - bytesMoved, stream->stop - stream->wrNext);
305 memcpy(stream->wrNext, str, m);
306 bytesMoved += m;
307 stream->wrNext += m;
308 if(bytesMoved == n)
309 return bytesMoved;
310 str += m;
311 }
312 if(stream->isClosed || stream->isReader)
313 return -1;
314 stream->emptyBuffProc(stream, FALSE);
315 }
316}
317\f
318/*
319 *----------------------------------------------------------------------
320 *
321 * FCGX_PutS --
322 *
323 * Writes a character string to the output stream.
324 *
325 * Results:
326 * number of bytes written for normal return,
327 * EOF (-1) if an error occurred.
328 *
329 *----------------------------------------------------------------------
330 */
331int FCGX_PutS(const char *str, FCGX_Stream *stream)
332{
333 return FCGX_PutStr(str, strlen(str), stream);
334}
335\f
336/*
337 *----------------------------------------------------------------------
338 *
339 * FCGX_FPrintF --
340 *
341 * Performs output formatting and writes the results
342 * to the output stream.
343 *
344 * Results:
345 * number of bytes written for normal return,
346 * EOF (-1) if an error occurred.
347 *
348 *----------------------------------------------------------------------
349 */
350int FCGX_FPrintF(FCGX_Stream *stream, const char *format, ...)
351{
352 int result;
353 va_list ap;
354 va_start(ap, format);
355 result = FCGX_VFPrintF(stream, format, ap);
356 va_end(ap);
357 return result;
358}
359
360/*
361 *----------------------------------------------------------------------
362 *
363 * FCGX_VFPrintF --
364 *
365 * Performs output formatting and writes the results
366 * to the output stream.
367 *
368 * Results:
369 * number of bytes written for normal return,
370 * EOF (-1) if an error occurred.
371 *
372 *----------------------------------------------------------------------
373 */
374
375#define PRINTF_BUFFLEN 100
376 /*
377 * More than sufficient space for all unmodified conversions
378 * except %s and %f.
379 */
380#define FMT_BUFFLEN 25
381 /*
382 * Max size of a format specifier is 1 + 5 + 7 + 7 + 2 + 1 + slop
383 */
384static void CopyAndAdvance(char **destPtr, char **srcPtr, int n);
385
386int FCGX_VFPrintF(FCGX_Stream *stream, const char *format, va_list arg)
387{
388 char *f, *fStop, *percentPtr, *p, *fmtBuffPtr, *buffPtr;
389 int op, performedOp, sizeModifier, buffCount, buffLen, specifierLength;
390 int fastPath, n, auxBuffLen, buffReqd, minWidth, precision, exp;
391 char *auxBuffPtr = NULL;
392 int streamCount = 0;
393 char fmtBuff[FMT_BUFFLEN];
394 char buff[PRINTF_BUFFLEN];
395
396 int intArg;
397 short shortArg;
398 long longArg;
399 unsigned unsignedArg;
400 unsigned long uLongArg;
401 unsigned short uShortArg;
402 char *charPtrArg;
403 void *voidPtrArg;
404 int *intPtrArg;
405 long *longPtrArg;
406 short *shortPtrArg;
407 double doubleArg;
408 LONG_DOUBLE lDoubleArg;
409
410 fmtBuff[0] = '%';
411 f = (char *) format;
412 fStop = f + strlen(f);
413 while (f != fStop) {
414 percentPtr = memchr(f, '%', fStop - f);
415 if(percentPtr == NULL) percentPtr = fStop;
416 if(percentPtr != f) {
417 if(FCGX_PutStr(f, percentPtr - f, stream) < 0) goto ErrorReturn;
418 streamCount += percentPtr - f;
419 f = percentPtr;
420 if(f == fStop) break;
421 }
422 fastPath = TRUE;
423 /*
424 * The following loop always executes either once or twice.
425 */
426 for (;;) {
427 if(fastPath) {
428 /*
429 * Fast path: Scan optimistically, hoping that no flags,
430 * minimum field width, or precision are specified.
431 * Use the preallocated buffer, which is large enough
432 * for all fast path cases. If the conversion specifier
433 * is really more complex, run the loop a second time
434 * using the slow path.
435 * Note that fast path execution of %s bypasses the buffer
436 * and %f is not attempted on the fast path due to
437 * its large buffering requirements.
438 */
439 op = *(percentPtr + 1);
440 switch(op) {
441 case 'l':
442 case 'L':
443 case 'h':
444 sizeModifier = op;
445 op = *(percentPtr + 2);
446 fmtBuff[1] = sizeModifier;
447 fmtBuff[2] = op;
448 fmtBuff[3] = '\0';
449 specifierLength = 3;
450 break;
451 default:
452 sizeModifier = ' ';
453 fmtBuff[1] = op;
454 fmtBuff[2] = '\0';
455 specifierLength = 2;
456 break;
457 }
458 buffPtr = buff;
459 buffLen = PRINTF_BUFFLEN;
460 } else {
461 /*
462 * Slow path: Scan the conversion specifier and construct
463 * a new format string, compute an upper bound on the
464 * amount of buffering that sprintf will require,
465 * and allocate a larger buffer if necessary.
466 */
467 p = percentPtr + 1;
468 fmtBuffPtr = &fmtBuff[1];
469 /*
470 * Scan flags
471 */
472 n = strspn(p, "-0+ #");
473 if(n > 5) goto ErrorReturn;
474 CopyAndAdvance(&fmtBuffPtr, &p, n);
475 /*
476 * Scan minimum field width
477 */
478 n = strspn(p, "0123456789");
479 if(n == 0) {
480 if(*p == '*') {
481 minWidth = va_arg(arg, int);
482 if(abs(minWidth) > 999999) goto ErrorReturn;
483 /*
484 * The following use of strlen rather than the
485 * value returned from sprintf is because SUNOS4
486 * returns a char * instead of an int count.
487 */
488 sprintf(fmtBuffPtr, "%d", minWidth);
489 fmtBuffPtr += strlen(fmtBuffPtr);
490 p++;
491 } else {
492 minWidth = 0;
493 }
494 } else if(n <= 6) {
495 minWidth = strtol(p, NULL, 10);
496 CopyAndAdvance(&fmtBuffPtr, &p, n);
497 } else {
498 goto ErrorReturn;
499 }
500 /*
501 * Scan precision
502 */
503 if(*p == '.') {
504 CopyAndAdvance(&fmtBuffPtr, &p, 1);
505 n = strspn(p, "0123456789");
506 if(n == 0) {
507 if(*p == '*') {
508 precision = va_arg(arg, int);
509 if(precision < 0) precision = 0;
510 if(precision > 999999) goto ErrorReturn;
511 /*
512 * The following use of strlen rather than the
513 * value returned from sprintf is because SUNOS4
514 * returns a char * instead of an int count.
515 */
516 sprintf(fmtBuffPtr, "%d", precision);
517 fmtBuffPtr += strlen(fmtBuffPtr);
518 p++;
519 } else {
520 precision = 0;
521 }
522 } else if(n <= 6) {
523 precision = strtol(p, NULL, 10);
524 CopyAndAdvance(&fmtBuffPtr, &p, n);
525 } else {
526 goto ErrorReturn;
527 }
528 } else {
529 precision = -1;
530 }
531 /*
532 * Scan size modifier and conversion operation
533 */
534 switch(*p) {
535 case 'l':
536 case 'L':
537 case 'h':
538 sizeModifier = *p;
539 CopyAndAdvance(&fmtBuffPtr, &p, 1);
540 break;
541 default:
542 sizeModifier = ' ';
543 break;
544 }
545 op = *p;
546 CopyAndAdvance(&fmtBuffPtr, &p, 1);
547 ASSERT(fmtBuffPtr - fmtBuff < FMT_BUFFLEN);
548 *fmtBuffPtr = '\0';
549 specifierLength = p - percentPtr;
550 /*
551 * Bound the required buffer size. For s and f
552 * conversions this requires examining the argument.
553 */
554 switch(op) {
555 case 'd':
556 case 'i':
557 case 'u':
558 case 'o':
559 case 'x':
560 case 'X':
561 case 'c':
562 case 'p':
563 buffReqd = max(precision, 46);
564 break;
565 case 's':
566 charPtrArg = va_arg(arg, char *);
567 if(precision == -1) {
568 buffReqd = strlen(charPtrArg);
569 } else {
570 p = memchr(charPtrArg, '\0', precision);
571 buffReqd =
572 (p == NULL) ? precision : p - charPtrArg;
573 }
574 break;
575 case 'f':
576 switch(sizeModifier) {
577 case ' ':
578 doubleArg = va_arg(arg, double);
579 frexp(doubleArg, &exp);
580 break;
581 case 'L':
582 lDoubleArg = va_arg(arg, LONG_DOUBLE);
583 frexp(lDoubleArg, &exp);
584 break;
585 default:
586 goto ErrorReturn;
587 }
588 if(precision == -1) precision = 6;
589 buffReqd = precision + 3 + ((exp > 0) ? exp/3 : 0);
590 break;
591 case 'e':
592 case 'E':
593 case 'g':
594 case 'G':
595 if(precision == -1) precision = 6;
596 buffReqd = precision + 8;
597 break;
598 case 'n':
599 case '%':
600 default:
601 goto ErrorReturn;
602 break;
603 }
604 buffReqd = max(buffReqd + 10, minWidth);
605 /*
606 * Allocate the buffer
607 */
608 if(buffReqd <= PRINTF_BUFFLEN) {
609 buffPtr = buff;
610 buffLen = PRINTF_BUFFLEN;
611 } else {
612 if(auxBuffPtr == NULL || buffReqd > auxBuffLen) {
613 if(auxBuffPtr != NULL) free(auxBuffPtr);
614 auxBuffPtr = Malloc(buffReqd);
615 auxBuffLen = buffReqd;
616 if(auxBuffPtr == NULL) goto ErrorReturn;
617 }
618 buffPtr = auxBuffPtr;
619 buffLen = auxBuffLen;
620 }
621 }
622 /*
623 * This giant switch statement requires the following variables
624 * to be set up: op, sizeModifier, arg, buffPtr, fmtBuff.
625 * When fastPath == FALSE and op == 's' or 'f', the argument
626 * has been read into charPtrArg, doubleArg, or lDoubleArg.
627 * The statement produces the boolean performedOp, TRUE iff
628 * the op/sizeModifier were executed and argument consumed;
629 * if performedOp, the characters written into buffPtr[]
630 * and the character count buffCount (== EOF meaning error).
631 *
632 * The switch cases are arranged in the same order as in the
633 * description of fprintf in section 15.11 of Harbison and Steele.
634 */
635 performedOp = TRUE;
636 switch(op) {
637 case 'd':
638 case 'i':
639 switch(sizeModifier) {
640 case ' ':
641 intArg = va_arg(arg, int);
642 sprintf(buffPtr, fmtBuff, intArg);
643 buffCount = strlen(buffPtr);
644 break;
645 case 'l':
646 longArg = va_arg(arg, long);
647 sprintf(buffPtr, fmtBuff, longArg);
648 buffCount = strlen(buffPtr);
649 break;
650 case 'h':
651 shortArg = va_arg(arg, short);
652 sprintf(buffPtr, fmtBuff, shortArg);
653 buffCount = strlen(buffPtr);
654 break;
655 default:
656 goto ErrorReturn;
657 }
658 break;
659 case 'u':
660 case 'o':
661 case 'x':
662 case 'X':
663 switch(sizeModifier) {
664 case ' ':
665 unsignedArg = va_arg(arg, unsigned);
666 sprintf(buffPtr, fmtBuff, unsignedArg);
667 buffCount = strlen(buffPtr);
668 break;
669 case 'l':
670 uLongArg = va_arg(arg, unsigned long);
671 sprintf(buffPtr, fmtBuff, uLongArg);
672 buffCount = strlen(buffPtr);
673 break;
674 case 'h':
675 uShortArg = va_arg(arg, unsigned short);
676 sprintf(buffPtr, fmtBuff, uShortArg);
677 buffCount = strlen(buffPtr);
678 break;
679 default:
680 goto ErrorReturn;
681 }
682 break;
683 case 'c':
684 switch(sizeModifier) {
685 case ' ':
686 intArg = va_arg(arg, int);
687 sprintf(buffPtr, fmtBuff, intArg);
688 buffCount = strlen(buffPtr);
689 break;
690 case 'l':
691 /*
692 * XXX: Allowed by ISO C Amendment 1, but
693 * many platforms don't yet support wint_t
694 */
695 goto ErrorReturn;
696 default:
697 goto ErrorReturn;
698 }
699 break;
700 case 's':
701 switch(sizeModifier) {
702 case ' ':
703 if(fastPath) {
704 buffPtr = va_arg(arg, char *);
705 buffCount = strlen(buffPtr);
706 buffLen = buffCount + 1;
707 } else {
708 sprintf(buffPtr, fmtBuff, charPtrArg);
709 buffCount = strlen(buffPtr);
710 }
711 break;
712 case 'l':
713 /*
714 * XXX: Don't know how to convert a sequence
715 * of wide characters into a byte stream, or
716 * even how to predict the buffering required.
717 */
718 goto ErrorReturn;
719 default:
720 goto ErrorReturn;
721 }
722 break;
723 case 'p':
724 if(sizeModifier != ' ') goto ErrorReturn;
725 voidPtrArg = va_arg(arg, void *);
726 sprintf(buffPtr, fmtBuff, voidPtrArg);
727 buffCount = strlen(buffPtr);
728 break;
729 case 'n':
730 switch(sizeModifier) {
731 case ' ':
732 intPtrArg = va_arg(arg, int *);
733 *intPtrArg = streamCount;
734 break;
735 case 'l':
736 longPtrArg = va_arg(arg, long *);
737 *longPtrArg = streamCount;
738 break;
739 case 'h':
740 shortPtrArg = va_arg(arg, short *);
741 *shortPtrArg = streamCount;
742 break;
743 default:
744 goto ErrorReturn;
745 }
746 buffCount = 0;
747 break;
748 case 'f':
749 if(fastPath) {
750 performedOp = FALSE;
751 break;
752 }
753 switch(sizeModifier) {
754 case ' ':
755 sprintf(buffPtr, fmtBuff, doubleArg);
756 buffCount = strlen(buffPtr);
757 break;
758 case 'L':
759 sprintf(buffPtr, fmtBuff, lDoubleArg);
760 buffCount = strlen(buffPtr);
761 break;
762 default:
763 goto ErrorReturn;
764 }
765 break;
766 case 'e':
767 case 'E':
768 case 'g':
769 case 'G':
770 switch(sizeModifier) {
771 case ' ':
772 doubleArg = va_arg(arg, double);
773 sprintf(buffPtr, fmtBuff, doubleArg);
774 buffCount = strlen(buffPtr);
775 break;
776 case 'L':
777 lDoubleArg = va_arg(arg, LONG_DOUBLE);
778 sprintf(buffPtr, fmtBuff, lDoubleArg);
779 buffCount = strlen(buffPtr);
780 break;
781 default:
782 goto ErrorReturn;
783 }
784 break;
785 case '%':
786 if(sizeModifier != ' ')
787 goto ErrorReturn;
788 buff[0] = '%';
789 buffCount = 1;
790 break;
791 case '\0':
792 goto ErrorReturn;
793 default:
794 performedOp = FALSE;
795 break;
796 } /* switch(op) */
797 if(performedOp) break;
798 if(!fastPath) goto ErrorReturn;
799 fastPath = FALSE;
800 } /* for (;;) */
801 ASSERT(buffCount < buffLen);
802 if(buffCount > 0) {
803 if(FCGX_PutStr(buffPtr, buffCount, stream) < 0)
804 goto ErrorReturn;
805 streamCount += buffCount;
806 } else if(buffCount < 0) {
807 goto ErrorReturn;
808 }
809 f += specifierLength;
810 } /* while(f != fStop) */
811 goto NormalReturn;
812 ErrorReturn:
813 streamCount = -1;
814 NormalReturn:
815 if(auxBuffPtr != NULL) free(auxBuffPtr);
816 return streamCount;
817}
818
819/*
820 * Copy n characters from *srcPtr to *destPtr, then increment
821 * both *srcPtr and *destPtr by n.
822 */
823static void CopyAndAdvance(char **destPtr, char **srcPtr, int n)
824{
825 char *dest = *destPtr;
826 char *src = *srcPtr;
827 int i;
828 for (i = 0; i < n; i++)
829 *dest++ = *src++;
830 *destPtr = dest;
831 *srcPtr = src;
832}
833\f
834/*
835 *----------------------------------------------------------------------
836 *
837 * FCGX_FFlush --
838 *
839 * Flushes any buffered output.
840 *
841 * Server-push is a legitimate application of FCGX_FFlush.
842 * Otherwise, FCGX_FFlush is not very useful, since FCGX_Accept
843 * does it implicitly. FCGX_FFlush may reduce performance
844 * by increasing the total number of operating system calls
845 * the application makes.
846 *
847 * Results:
848 * EOF (-1) if an error occurred.
849 *
850 *----------------------------------------------------------------------
851 */
852int FCGX_FFlush(FCGX_Stream *stream)
853{
854 if(stream->isClosed || stream->isReader)
855 return 0;
856 stream->emptyBuffProc(stream, FALSE);
857 return (stream->isClosed) ? -1 : 0;
858}
859\f
860/*
861 *----------------------------------------------------------------------
862 *
863 * FCGX_FClose --
864 *
865 * Performs FCGX_FFlush and closes the stream.
866 *
867 * This is not a very useful operation, since FCGX_Accept
868 * does it implicitly. Closing the out stream before the
869 * err stream results in an extra write if there's nothing
870 * in the err stream, and therefore reduces performance.
871 *
872 * Results:
873 * EOF (-1) if an error occurred.
874 *
875 *----------------------------------------------------------------------
876 */
877int FCGX_FClose(FCGX_Stream *stream)
878{
879 if(!stream->wasFCloseCalled) {
880 if(!stream->isReader) {
881 stream->emptyBuffProc(stream, TRUE);
882 }
883 stream->wasFCloseCalled = TRUE;
884 stream->isClosed = TRUE;
885 if(stream->isReader) {
886 stream->wrNext = stream->stop = stream->rdNext;
887 } else {
888 stream->rdNext = stream->stop = stream->wrNext;
889 }
890 }
891 return (stream->FCGI_errno == 0) ? 0 : EOF;
892}
893\f
894/*
895 *----------------------------------------------------------------------
896 *
897 * SetError --
898 *
899 * An error has occurred; save the error code in the stream
900 * for diagnostic purposes and set the stream state so that
901 * reads return EOF and writes have no effect.
902 *
903 *----------------------------------------------------------------------
904 */
905static void SetError(FCGX_Stream *stream, int FCGI_errno)
906{
907 /*
908 * Preserve only the first error.
909 */
910 if(stream->FCGI_errno == 0) {
911 stream->FCGI_errno = FCGI_errno;
912 stream->isClosed = TRUE;
913 }
914}
915
916/*
917 *----------------------------------------------------------------------
918 *
919 * FCGX_GetError --
920 *
921 * Return the stream error code. 0 means no error, > 0
922 * is an errno(2) error, < 0 is an FCGX_errno error.
923 *
924 *----------------------------------------------------------------------
925 */
926int FCGX_GetError(FCGX_Stream *stream) {
927 return stream->FCGI_errno;
928}
929
930/*
931 *----------------------------------------------------------------------
932 *
933 * FCGX_ClearError --
934 *
935 * Clear the stream error code and end-of-file indication.
936 *
937 *----------------------------------------------------------------------
938 */
939void FCGX_ClearError(FCGX_Stream *stream) {
940 stream->FCGI_errno = 0;
941 /*
942 * stream->isClosed = FALSE;
943 * XXX: should clear isClosed but work is needed to make it safe
944 * to do so. For example, if an application calls FClose, gets
945 * an I/O error on the write, calls ClearError and retries
946 * the FClose, FClose (really EmptyBuffProc) will write a second
947 * EOF record. If an application calls PutChar instead of FClose
948 * after the ClearError, the application will write more data.
949 * The stream's state must discriminate between various states
950 * of the stream that are now all lumped under isClosed.
951 */
952}
953\f
954/*
955 *======================================================================
956 * Parameters
957 *======================================================================
958 */
959
960/*
961 * A vector of pointers representing the parameters received
962 * by a FastCGI application server, with the vector's length
963 * and last valid element so adding new parameters is efficient.
964 */
965
966typedef struct Params {
967 FCGX_ParamArray vec; /* vector of strings */
968 int length; /* number of string vec can hold */
969 char **cur; /* current item in vec; *cur == NULL */
970} Params;
971typedef Params *ParamsPtr;
972\f
973/*
974 *----------------------------------------------------------------------
975 *
976 * NewParams --
977 *
978 * Creates a new Params structure.
979 *
980 * Results:
981 * Pointer to the new structure.
982 *
983 *----------------------------------------------------------------------
984 */
985static ParamsPtr NewParams(int length)
986{
987 ParamsPtr result;
988 result = Malloc(sizeof(Params));
989 result->vec = (char **) Malloc(length * sizeof(char *));
990 result->length = length;
991 result->cur = result->vec;
992 *result->cur = NULL;
993 return result;
994}
995\f
996/*
997 *----------------------------------------------------------------------
998 *
999 * FreeParams --
1000 *
1001 * Frees a Params structure and all the parameters it contains.
1002 *
1003 * Side effects:
1004 * paramsPtr becomes invalid.
1005 *
1006 *----------------------------------------------------------------------
1007 */
1008static void FreeParams(ParamsPtr *paramsPtrPtr)
1009{
1010 ParamsPtr paramsPtr = *paramsPtrPtr;
1011 char **p;
1012 if(paramsPtr == NULL) {
1013 return;
1014 }
1015 for (p = paramsPtr->vec; p < paramsPtr->cur; p++) {
1016 free(*p);
1017 }
1018 free(paramsPtr->vec);
1019 free(paramsPtr);
1020 *paramsPtrPtr = NULL;
1021}
1022\f
1023/*
1024 *----------------------------------------------------------------------
1025 *
1026 * PutParam --
1027 *
1028 * Add a name/value pair to a Params structure.
1029 *
1030 * Results:
1031 * None.
1032 *
1033 * Side effects:
1034 * Parameters structure updated.
1035 *
1036 *----------------------------------------------------------------------
1037 */
1038static void PutParam(ParamsPtr paramsPtr, char *nameValue)
1039{
1040 int size;
1041
1042 *paramsPtr->cur++ = nameValue;
1043 size = paramsPtr->cur - paramsPtr->vec;
1044 if(size >= paramsPtr->length) {
1045 paramsPtr->length *= 2;
1046 paramsPtr->vec =
1047 realloc(paramsPtr->vec, paramsPtr->length * sizeof(char *));
1048 paramsPtr->cur = paramsPtr->vec + size;
1049 }
1050 *paramsPtr->cur = NULL;
1051}
1052\f
1053/*
1054 *----------------------------------------------------------------------
1055 *
1056 * FCGX_GetParam -- obtain value of FCGI parameter in environment
1057 *
1058 *
1059 * Results:
1060 * Value bound to name, NULL if name not present in the
1061 * environment envp. Caller must not mutate the result
1062 * or retain it past the end of this request.
1063 *
1064 *----------------------------------------------------------------------
1065 */
1066char *FCGX_GetParam(const char *name, FCGX_ParamArray envp)
1067{
1068 int len;
1069 char **p;
1070 len = strlen(name);
1071 if(len == 0) return NULL;
1072 for (p = envp; *p != NULL; p++) {
1073 if((strncmp(name, *p, len) == 0) && ((*p)[len] == '=')) {
1074 return *p+len+1;
1075 }
1076 }
1077 return NULL;
1078}
1079\f
1080/*
1081 *----------------------------------------------------------------------
1082 *
1083 * Start of FastCGI-specific code
1084 *
1085 *----------------------------------------------------------------------
1086 */
1087\f
1088/*
1089 *----------------------------------------------------------------------
1090 *
1091 * ReadParams --
1092 *
1093 * Reads FastCGI name-value pairs from stream until EOF. Converts
1094 * each pair to name=value format and adds it to Params structure.
1095 *
1096 *----------------------------------------------------------------------
1097 */
1098static int ReadParams(Params *paramsPtr, FCGX_Stream *stream)
1099{
1100 int nameLen, valueLen;
1101 unsigned char lenBuff[3];
1102 char *nameValue;
1103
1104 while((nameLen = FCGX_GetChar(stream)) != EOF) {
1105 /*
1106 * Read name length (one or four bytes) and value length
1107 * (one or four bytes) from stream.
1108 */
1109 if((nameLen & 0x80) != 0) {
1110 if(FCGX_GetStr((char *) &lenBuff[0], 3, stream) != 3) {
1111 SetError(stream, FCGX_PARAMS_ERROR);
1112 return -1;
1113 }
1114 nameLen = ((nameLen & 0x7f) << 24) + (lenBuff[0] << 16)
1115 + (lenBuff[1] << 8) + lenBuff[2];
1116 }
1117 if((valueLen = FCGX_GetChar(stream)) == EOF) {
1118 SetError(stream, FCGX_PARAMS_ERROR);
1119 return -1;
1120 }
1121 if((valueLen & 0x80) != 0) {
1122 if(FCGX_GetStr((char *) &lenBuff[0], 3, stream) != 3) {
1123 SetError(stream, FCGX_PARAMS_ERROR);
1124 return -1;
1125 }
1126 valueLen = ((valueLen & 0x7f) << 24) + (lenBuff[0] << 16)
1127 + (lenBuff[1] << 8) + lenBuff[2];
1128 }
1129 /*
1130 * nameLen and valueLen are now valid; read the name and value
1131 * from stream and construct a standard environment entry.
1132 */
1133 nameValue = Malloc(nameLen + valueLen + 2);
1134 if(FCGX_GetStr(nameValue, nameLen, stream) != nameLen) {
1135 SetError(stream, FCGX_PARAMS_ERROR);
1136 free(nameValue);
1137 return -1;
1138 }
1139 *(nameValue + nameLen) = '=';
1140 if(FCGX_GetStr(nameValue + nameLen + 1, valueLen, stream)
1141 != valueLen) {
1142 SetError(stream, FCGX_PARAMS_ERROR);
1143 free(nameValue);
1144 return -1;
1145 }
1146 *(nameValue + nameLen + valueLen + 1) = '\0';
1147 PutParam(paramsPtr, nameValue);
1148 }
1149 return 0;
1150}
1151\f
1152/*
1153 *----------------------------------------------------------------------
1154 *
1155 * MakeHeader --
1156 *
1157 * Constructs an FCGI_Header struct.
1158 *
1159 *----------------------------------------------------------------------
1160 */
1161static FCGI_Header MakeHeader(
1162 int type,
1163 int requestId,
1164 int contentLength,
1165 int paddingLength)
1166{
1167 FCGI_Header header;
1168 ASSERT(contentLength >= 0 && contentLength <= FCGI_MAX_LENGTH);
1169 ASSERT(paddingLength >= 0 && paddingLength <= 0xff);
1170 header.version = FCGI_VERSION_1;
1171 header.type = type;
1172 header.requestIdB1 = (requestId >> 8) & 0xff;
1173 header.requestIdB0 = (requestId ) & 0xff;
1174 header.contentLengthB1 = (contentLength >> 8) & 0xff;
1175 header.contentLengthB0 = (contentLength ) & 0xff;
1176 header.paddingLength = paddingLength;
1177 header.reserved = 0;
1178 return header;
1179}
1180\f
1181/*
1182 *----------------------------------------------------------------------
1183 *
1184 * MakeEndRequestBody --
1185 *
1186 * Constructs an FCGI_EndRequestBody struct.
1187 *
1188 *----------------------------------------------------------------------
1189 */
1190static FCGI_EndRequestBody MakeEndRequestBody(
1191 int appStatus,
1192 int protocolStatus)
1193{
1194 FCGI_EndRequestBody body;
1195 body.appStatusB3 = (appStatus >> 24) & 0xff;
1196 body.appStatusB2 = (appStatus >> 16) & 0xff;
1197 body.appStatusB1 = (appStatus >> 8) & 0xff;
1198 body.appStatusB0 = (appStatus ) & 0xff;
1199 body.protocolStatus = protocolStatus;
1200 memset(body.reserved, 0, sizeof(body.reserved));
1201 return body;
1202}
1203\f
1204/*
1205 *----------------------------------------------------------------------
1206 *
1207 * MakeUnknownTypeBody --
1208 *
1209 * Constructs an FCGI_MakeUnknownTypeBody struct.
1210 *
1211 *----------------------------------------------------------------------
1212 */
1213static FCGI_UnknownTypeBody MakeUnknownTypeBody(
1214 int type)
1215{
1216 FCGI_UnknownTypeBody body;
1217 body.type = type;
1218 memset(body.reserved, 0, sizeof(body.reserved));
1219 return body;
1220}
1221\f
1222/*
1223 *----------------------------------------------------------------------
1224 *
1225 * AlignInt8 --
1226 *
1227 * Returns the smallest integer greater than or equal to n
1228 * that's a multiple of 8.
1229 *
1230 *----------------------------------------------------------------------
1231 */
1232static int AlignInt8(unsigned n) {
1233 return (n + 7) & (UINT_MAX - 7);
1234}
1235
1236/*
1237 *----------------------------------------------------------------------
1238 *
1239 * AlignPtr8 --
1240 *
1241 * Returns the smallest pointer greater than or equal to p
1242 * that's a multiple of 8.
1243 *
1244 *----------------------------------------------------------------------
1245 */
1246static unsigned char *AlignPtr8(unsigned char *p) {
1247 unsigned long u = (unsigned long) p;
1248 u = ((u + 7) & (ULONG_MAX - 7)) - u;
1249 return p + u;
1250}
1251\f
1252/*
1253 * State associated with a request
1254 */
1255typedef struct ReqData {
1256 int ipcFd; /* < 0 means no connection */
1257 int isBeginProcessed; /* FCGI_BEGIN_REQUEST seen */
1258 int requestId; /* valid if isBeginProcessed */
1259 int keepConnection; /* don't close ipcFd at end of request */
1260 int role;
1261 int appStatus;
1262 int nWriters; /* number of open writers (0..2) */
1263 FCGX_Stream *inStream;
1264 FCGX_Stream *outStream;
1265 FCGX_Stream *errStream;
1266 ParamsPtr paramsPtr;
1267} ReqData;
1268
1269
1270/*
1271 * State associated with a stream
1272 */
1273typedef struct FCGX_Stream_Data {
1274 unsigned char *buff; /* buffer after alignment */
1275 int bufflen; /* number of bytes buff can store */
1276 unsigned char *mBuff; /* buffer as returned by Malloc */
1277 unsigned char *buffStop; /* reader: last valid byte + 1 of entire buffer.
1278 * stop generally differs from buffStop for
1279 * readers because of record structure.
1280 * writer: buff + bufflen */
1281 int type; /* reader: FCGI_PARAMS or FCGI_STDIN
1282 * writer: FCGI_STDOUT or FCGI_STDERR */
1283 int eorStop; /* reader: stop stream at end-of-record */
1284 int skip; /* reader: don't deliver content bytes */
1285 int contentLen; /* reader: bytes of unread content */
1286 int paddingLen; /* reader: bytes of unread padding */
1287 int isAnythingWritten; /* writer: data has been written to ipcFd */
1288 int rawWrite; /* writer: write data without stream headers */
1289 ReqData *reqDataPtr; /* request data not specific to one stream */
1290} FCGX_Stream_Data;
1291\f
1292/*
1293 *----------------------------------------------------------------------
1294 *
1295 * WriteCloseRecords --
1296 *
1297 * Writes an EOF record for the stream content if necessary.
1298 * If this is the last writer to close, writes an FCGI_END_REQUEST
1299 * record.
1300 *
1301 *----------------------------------------------------------------------
1302 */
1303static void WriteCloseRecords(struct FCGX_Stream *stream)
1304{
1305 FCGX_Stream_Data *data = stream->data;
1306 /*
1307 * Enter rawWrite mode so final records won't be encapsulated as
1308 * stream data.
1309 */
1310 data->rawWrite = TRUE;
1311 /*
1312 * Generate EOF for stream content if needed.
1313 */
1314 if(!(data->type == FCGI_STDERR
1315 && stream->wrNext == data->buff
1316 && !data->isAnythingWritten)) {
1317 FCGI_Header header;
1318 header = MakeHeader(data->type, data->reqDataPtr->requestId, 0, 0);
1319 FCGX_PutStr((char *) &header, sizeof(header), stream);
1320 };
1321 /*
1322 * Generate FCGI_END_REQUEST record if needed.
1323 */
1324 if(data->reqDataPtr->nWriters == 1) {
1325 FCGI_EndRequestRecord endRequestRecord;
1326 endRequestRecord.header = MakeHeader(FCGI_END_REQUEST,
1327 data->reqDataPtr->requestId,
1328 sizeof(endRequestRecord.body), 0);
1329 endRequestRecord.body = MakeEndRequestBody(
1330 data->reqDataPtr->appStatus, FCGI_REQUEST_COMPLETE);
1331 FCGX_PutStr((char *) &endRequestRecord,
1332 sizeof(endRequestRecord), stream);
1333 }
1334 data->reqDataPtr->nWriters--;
1335}
1336\f
19d98b91 1337
1338
1339static int write_it_all(int fd, char *buf, int len)
1340{
1341 int wrote;
1342
1343 while (len) {
1344 wrote = OS_Write(fd, buf, len);
1345 if (wrote < 0)
1346 return wrote;
1347 len -= wrote;
1348 buf += wrote;
1349 }
1350 return len;
1351}
1352
0198fd3c 1353/*
1354 *----------------------------------------------------------------------
1355 *
1356 * EmptyBuffProc --
1357 *
1358 * Encapsulates any buffered stream content in a FastCGI
1359 * record. Writes the data, making the buffer empty.
1360 *
1361 *----------------------------------------------------------------------
1362 */
1363static void EmptyBuffProc(struct FCGX_Stream *stream, int doClose)
1364{
1365 FCGX_Stream_Data *data = stream->data;
1366 int cLen, eLen;
1367 /*
1368 * If the buffer contains stream data, fill in the header.
1369 * Pad the record to a multiple of 8 bytes in length. Padding
1370 * can't overflow the buffer because the buffer is a multiple
1371 * of 8 bytes in length. If the buffer contains no stream
1372 * data, reclaim the space reserved for the header.
1373 */
1374 if(!data->rawWrite) {
1375 cLen = stream->wrNext - data->buff - sizeof(FCGI_Header);
1376 if(cLen > 0) {
1377 eLen = AlignInt8(cLen);
1378 /*
1379 * Giving the padding a well-defined value keeps Purify happy.
1380 */
1381 memset(stream->wrNext, 0, eLen - cLen);
1382 stream->wrNext += eLen - cLen;
1383 *((FCGI_Header *) data->buff)
1384 = MakeHeader(data->type,
1385 data->reqDataPtr->requestId, cLen, eLen - cLen);
1386 } else {
1387 stream->wrNext = data->buff;
1388 }
1389 }
1390 if(doClose) {
1391 WriteCloseRecords(stream);
1392 };
19d98b91 1393 if (stream->wrNext != data->buff) {
0198fd3c 1394 data->isAnythingWritten = TRUE;
19d98b91 1395 if (write_it_all(data->reqDataPtr->ipcFd, (char *)data->buff, stream->wrNext - data->buff) < 0) {
0198fd3c 1396 SetError(stream, OS_Errno);
1397 return;
1398 }
1399 stream->wrNext = data->buff;
1400 }
1401 /*
1402 * The buffer is empty.
1403 */
1404 if(!data->rawWrite) {
1405 stream->wrNext += sizeof(FCGI_Header);
1406 }
1407}
1408\f
1409/*
1410 * Return codes for Process* functions
1411 */
1412#define STREAM_RECORD 0
1413#define SKIP 1
1414#define BEGIN_RECORD 2
1415#define MGMT_RECORD 3
1416
1417/*
1418 *----------------------------------------------------------------------
1419 *
1420 * ProcessManagementRecord --
1421 *
1422 * Reads and responds to a management record. The only type of
1423 * management record this library understands is FCGI_GET_VALUES.
1424 * The only variables that this library's FCGI_GET_VALUES
1425 * understands are FCGI_MAX_CONNS, FCGI_MAX_REQS, and FCGI_MPXS_CONNS.
1426 * Ignore other FCGI_GET_VALUES variables; respond to other
1427 * management records with a FCGI_UNKNOWN_TYPE record.
1428 *
1429 *----------------------------------------------------------------------
1430 */
1431static int ProcessManagementRecord(int type, FCGX_Stream *stream)
1432{
1433 FCGX_Stream_Data *data = stream->data;
1434 ParamsPtr paramsPtr = NewParams(3);
1435 char **pPtr;
1436 char response[64]; /* 64 = 8 + 3*(1+1+14+1)* + padding */
1437 char *responseP = &response[FCGI_HEADER_LEN];
1438 char *name, value;
1439 int len, paddedLen;
1440 if(type == FCGI_GET_VALUES) {
1441 ReadParams(paramsPtr, stream);
1442 if((FCGX_GetError(stream) != 0) || (data->contentLen != 0)) {
1443 FreeParams(&paramsPtr);
1444 return FCGX_PROTOCOL_ERROR;
1445 }
1446 for (pPtr = paramsPtr->vec; pPtr < paramsPtr->cur; pPtr++) {
1447 name = *pPtr;
1448 *(strchr(name, '=')) = '\0';
1449 if(strcmp(name, FCGI_MAX_CONNS) == 0) {
1450 value = '1';
1451 } else if(strcmp(name, FCGI_MAX_REQS) == 0) {
1452 value = '1';
1453 } else if(strcmp(name, FCGI_MPXS_CONNS) == 0) {
1454 value = '0';
1455 } else {
1456 name = NULL;
1457 }
1458 if(name != NULL) {
1459 len = strlen(name);
1460 sprintf(responseP, "%c%c%s%c", len, 1, name, value);
1461 responseP += len + 3;
1462 }
1463 }
1464 len = responseP - &response[FCGI_HEADER_LEN];
1465 paddedLen = AlignInt8(len);
1466 *((FCGI_Header *) response)
1467 = MakeHeader(FCGI_GET_VALUES_RESULT, FCGI_NULL_REQUEST_ID,
1468 len, paddedLen - len);
1469 FreeParams(&paramsPtr);
1470 } else {
1471 paddedLen = len = sizeof(FCGI_UnknownTypeBody);
1472 ((FCGI_UnknownTypeRecord *) response)->header
1473 = MakeHeader(FCGI_UNKNOWN_TYPE, FCGI_NULL_REQUEST_ID,
1474 len, 0);
1475 ((FCGI_UnknownTypeRecord *) response)->body
1476 = MakeUnknownTypeBody(type);
1477 }
19d98b91 1478 if (write_it_all(data->reqDataPtr->ipcFd, response, FCGI_HEADER_LEN + paddedLen) < 0) {
0198fd3c 1479 SetError(stream, OS_Errno);
1480 return -1;
1481 }
19d98b91 1482
0198fd3c 1483 return MGMT_RECORD;
1484}
1485\f
1486/*
1487 *----------------------------------------------------------------------
1488 *
1489 * ProcessBeginRecord --
1490 *
1491 * Reads an FCGI_BEGIN_REQUEST record.
1492 *
1493 * Results:
1494 * BEGIN_RECORD for normal return. FCGX_PROTOCOL_ERROR for
1495 * protocol error. SKIP for attempt to multiplex
1496 * connection. -1 for error from write (errno in stream).
1497 *
1498 * Side effects:
1499 * In case of BEGIN_RECORD return, stores requestId, role,
1500 * keepConnection values, and sets isBeginProcessed = TRUE.
1501 *
1502 *----------------------------------------------------------------------
1503 */
1504static int ProcessBeginRecord(int requestId, FCGX_Stream *stream)
1505{
1506 FCGX_Stream_Data *data = stream->data;
1507 FCGI_BeginRequestBody body;
1508 if(requestId == 0 || data->contentLen != sizeof(body)) {
1509 return FCGX_PROTOCOL_ERROR;
1510 }
1511 if(data->reqDataPtr->isBeginProcessed) {
1512 /*
1513 * The Web server is multiplexing the connection. This library
1514 * doesn't know how to handle multiplexing, so respond with
1515 * FCGI_END_REQUEST{protocolStatus = FCGI_CANT_MPX_CONN}
1516 */
1517 FCGI_EndRequestRecord endRequestRecord;
1518 endRequestRecord.header = MakeHeader(FCGI_END_REQUEST,
1519 requestId, sizeof(endRequestRecord.body), 0);
1520 endRequestRecord.body
1521 = MakeEndRequestBody(0, FCGI_CANT_MPX_CONN);
19d98b91 1522 if (write_it_all(data->reqDataPtr->ipcFd, (char *)&endRequestRecord, sizeof(endRequestRecord)) < 0) {
0198fd3c 1523 SetError(stream, OS_Errno);
1524 return -1;
1525 }
19d98b91 1526
0198fd3c 1527 return SKIP;
1528 }
1529 /*
1530 * Accept this new request. Read the record body.
1531 */
1532 data->reqDataPtr->requestId = requestId;
1533 if(FCGX_GetStr((char *) &body, sizeof(body), stream)
1534 != sizeof(body)) {
1535 return FCGX_PROTOCOL_ERROR;
1536 }
1537 data->reqDataPtr->keepConnection = (body.flags & FCGI_KEEP_CONN);
1538 data->reqDataPtr->role = (body.roleB1 << 8) + body.roleB0;
1539 data->reqDataPtr->isBeginProcessed = TRUE;
1540 return BEGIN_RECORD;
1541}
1542\f
1543/*
1544 *----------------------------------------------------------------------
1545 *
1546 * ProcessHeader --
1547 *
1548 * Interprets FCGI_Header. Processes FCGI_BEGIN_REQUEST and
1549 * management records here; extracts information from stream
1550 * records (FCGI_PARAMS, FCGI_STDIN) into stream.
1551 *
1552 * Results:
1553 * >= 0 for a normal return, < 0 for error
1554 *
1555 * Side effects:
1556 * XXX: Many (more than there used to be).
1557 * If !stream->isRequestIdSet, ProcessHeader initializes
1558 * stream->requestId from header and sets stream->isRequestIdSet
1559 * to TRUE. ProcessHeader also sets stream->contentLen to header's
1560 * contentLength, and sets stream->paddingLen to the header's
1561 * paddingLength.
1562 *
1563 *----------------------------------------------------------------------
1564 */
1565static int ProcessHeader(FCGI_Header header, FCGX_Stream *stream)
1566{
1567 FCGX_Stream_Data *data = stream->data;
1568 int requestId;
1569 if(header.version != FCGI_VERSION_1) {
1570 return FCGX_UNSUPPORTED_VERSION;
1571 }
1572 requestId = (header.requestIdB1 << 8)
1573 + header.requestIdB0;
1574 data->contentLen = (header.contentLengthB1 << 8)
1575 + header.contentLengthB0;
1576 data->paddingLen = header.paddingLength;
1577 if(header.type == FCGI_BEGIN_REQUEST) {
1578 return ProcessBeginRecord(requestId, stream);
1579 }
1580 if(requestId == FCGI_NULL_REQUEST_ID) {
1581 return ProcessManagementRecord(header.type, stream);
1582 }
1583 if(requestId != data->reqDataPtr->requestId) {
1584 return SKIP;
1585 }
1586 if(header.type != data->type) {
1587 return FCGX_PROTOCOL_ERROR;
1588 }
1589 return STREAM_RECORD;
1590}
1591\f
1592/*
1593 *----------------------------------------------------------------------
1594 *
1595 * FillBuffProc --
1596 *
1597 * Reads bytes from the ipcFd, supplies bytes to a stream client.
1598 *
1599 *----------------------------------------------------------------------
1600 */
1601static void FillBuffProc(FCGX_Stream *stream)
1602{
1603 FCGX_Stream_Data *data = stream->data;
1604 FCGI_Header header;
1605 int headerLen = 0;
1606 int status, count;
1607
1608 for (;;) {
1609 /*
1610 * If data->buff is empty, do a read.
1611 */
1612 if(stream->rdNext == data->buffStop) {
1613 count = OS_Read(data->reqDataPtr->ipcFd, (char *)data->buff,
1614 data->bufflen);
1615 if(count <= 0) {
1616 SetError(stream, (count == 0 ? FCGX_PROTOCOL_ERROR : OS_Errno));
1617 return;
1618 }
1619 stream->rdNext = data->buff;
1620 data->buffStop = data->buff + count;
1621 }
1622 /*
1623 * Now data->buff is not empty. If the current record contains
1624 * more content bytes, deliver all that are present in data->buff.
1625 */
1626 if(data->contentLen > 0) {
1627 count = min(data->contentLen, data->buffStop - stream->rdNext);
1628 data->contentLen -= count;
1629 if(!data->skip) {
1630 stream->wrNext = stream->stop = stream->rdNext + count;
1631 return;
1632 } else {
1633 stream->rdNext += count;
1634 if(data->contentLen > 0) {
1635 continue;
1636 } else {
1637 data->skip = FALSE;
1638 }
1639 }
1640 }
1641 /*
1642 * If the current record (whose content has been fully consumed by
1643 * the client) was padded, skip over the padding bytes.
1644 */
1645 if(data->paddingLen > 0) {
1646 count = min(data->paddingLen, data->buffStop - stream->rdNext);
1647 data->paddingLen -= count;
1648 stream->rdNext += count;
1649 if(data->paddingLen > 0) {
1650 continue;
1651 }
1652 }
1653 /*
1654 * All done with the current record, including the padding.
1655 * If we're in a recursive call from ProcessHeader, deliver EOF.
1656 */
1657 if(data->eorStop) {
1658 stream->stop = stream->rdNext;
1659 stream->isClosed = TRUE;
1660 return;
1661 }
1662 /*
1663 * Fill header with bytes from the input buffer.
1664 */
1665 count = min((int)sizeof(header) - headerLen,
1666 data->buffStop - stream->rdNext);
1667 memcpy(((char *)(&header)) + headerLen, stream->rdNext, count);
1668 headerLen += count;
1669 stream->rdNext += count;
1670 if(headerLen < sizeof(header)) {
1671 continue;
1672 };
1673 headerLen = 0;
1674 /*
1675 * Interpret header. eorStop prevents ProcessHeader from reading
1676 * past the end-of-record when using stream to read content.
1677 */
1678 data->eorStop = TRUE;
1679 stream->stop = stream->rdNext;
1680 status = ProcessHeader(header, stream);
1681 data->eorStop = FALSE;
1682 stream->isClosed = FALSE;
1683 switch(status) {
1684 case STREAM_RECORD:
1685 /*
1686 * If this stream record header marked the end of stream
1687 * data deliver EOF to the stream client, otherwise loop
1688 * and deliver data.
1689 *
1690 * XXX: If this is final stream and
1691 * stream->rdNext != data->buffStop, buffered
1692 * data is next request (server pipelining)?
1693 */
1694 if(data->contentLen == 0) {
1695 stream->wrNext = stream->stop = stream->rdNext;
1696 stream->isClosed = TRUE;
1697 return;
1698 }
1699 break;
1700 case SKIP:
1701 data->skip = TRUE;
1702 break;
1703 case BEGIN_RECORD:
1704 /*
1705 * If this header marked the beginning of a new
1706 * request, return role information to caller.
1707 */
1708 return;
1709 break;
1710 case MGMT_RECORD:
1711 break;
1712 default:
1713 ASSERT(status < 0);
1714 SetError(stream, status);
1715 return;
1716 break;
1717 }
1718 }
1719}
1720\f
1721/*
1722 *----------------------------------------------------------------------
1723 *
1724 * NewStream --
1725 *
1726 * Creates a stream to read or write from an open ipcFd.
1727 * The stream performs reads/writes of up to bufflen bytes.
1728 *
1729 *----------------------------------------------------------------------
1730 */
1731static FCGX_Stream *NewStream(
1732 ReqData *reqDataPtr, int bufflen, int isReader, int streamType)
1733{
1734 /*
1735 * XXX: It would be a lot cleaner to have a NewStream that only
1736 * knows about the type FCGX_Stream, with all other
1737 * necessary data passed in. It appears that not just
1738 * data and the two procs are needed for initializing stream,
1739 * but also data->buff and data->buffStop. This has implications
1740 * for procs that want to swap buffers, too.
1741 */
1742 FCGX_Stream *stream = Malloc(sizeof(FCGX_Stream));
1743 FCGX_Stream_Data *data = Malloc(sizeof(FCGX_Stream_Data));
1744 data->reqDataPtr = reqDataPtr;
1745 bufflen = AlignInt8(min(max(bufflen, 32), FCGI_MAX_LENGTH + 1));
1746 data->bufflen = bufflen;
1747 data->mBuff = Malloc(bufflen);
1748 data->buff = AlignPtr8(data->mBuff);
1749 if(data->buff != data->mBuff) {
1750 data->bufflen -= 8;
1751 }
1752 if(isReader) {
1753 data->buffStop = data->buff;
1754 } else {
1755 data->buffStop = data->buff + data->bufflen;
1756 }
1757 data->type = streamType;
1758 data->eorStop = FALSE;
1759 data->skip = FALSE;
1760 data->contentLen = 0;
1761 data->paddingLen = 0;
1762 data->isAnythingWritten = FALSE;
1763 data->rawWrite = FALSE;
1764
1765 stream->data = data;
1766 stream->isReader = isReader;
1767 stream->isClosed = FALSE;
1768 stream->wasFCloseCalled = FALSE;
1769 stream->FCGI_errno = 0;
1770 if(isReader) {
1771 stream->fillBuffProc = FillBuffProc;
1772 stream->emptyBuffProc = NULL;
1773 stream->rdNext = data->buff;
1774 stream->stop = stream->rdNext;
1775 stream->stopUnget = data->buff;
1776 stream->wrNext = stream->stop;
1777 } else {
1778 stream->fillBuffProc = NULL;
1779 stream->emptyBuffProc = EmptyBuffProc;
1780 stream->wrNext = data->buff + sizeof(FCGI_Header);
1781 stream->stop = data->buffStop;
1782 stream->stopUnget = NULL;
1783 stream->rdNext = stream->stop;
1784 }
1785 return stream;
1786}
1787\f
1788/*
1789 *----------------------------------------------------------------------
1790 *
1791 * FreeStream --
1792 *
1793 * Frees all storage allocated when *streamPtr was created,
1794 * and nulls out *streamPtr.
1795 *
1796 *----------------------------------------------------------------------
1797 */
1798void FreeStream(FCGX_Stream **streamPtr)
1799{
1800 FCGX_Stream *stream = *streamPtr;
1801 FCGX_Stream_Data *data;
1802 if(stream == NULL) {
1803 return;
1804 }
1805 data = stream->data;
1806 data->reqDataPtr = NULL;
1807 free(data->mBuff);
1808 free(data);
1809 free(stream);
1810 *streamPtr = NULL;
1811}
1812\f
1813/*
1814 *----------------------------------------------------------------------
1815 *
1816 * SetReaderType --
1817 *
1818 * Re-initializes the stream to read data of the specified type.
1819 *
1820 *----------------------------------------------------------------------
1821 */
1822static FCGX_Stream *SetReaderType(FCGX_Stream *stream, int streamType)
1823{
1824 FCGX_Stream_Data *data = stream->data;
1825 ASSERT(stream->isReader);
1826 data->type = streamType;
1827 data->eorStop = FALSE;
1828 data->skip = FALSE;
1829 data->contentLen = 0;
1830 data->paddingLen = 0;
1831 stream->wrNext = stream->stop = stream->rdNext;
1832 stream->isClosed = FALSE;
1833 return stream;
1834}
1835\f
1836/*
1837 *----------------------------------------------------------------------
1838 *
1839 * NewReader --
1840 *
1841 * Creates a stream to read streamType records for the given
1842 * request. The stream performs OS reads of up to bufflen bytes.
1843 *
1844 *----------------------------------------------------------------------
1845 */
1846static FCGX_Stream *NewReader(ReqData *reqDataPtr, int bufflen, int streamType)
1847{
1848 return NewStream(reqDataPtr, bufflen, TRUE, streamType);
1849}
1850
1851
1852/*
1853 *----------------------------------------------------------------------
1854 *
1855 * NewWriter --
1856 *
1857 * Creates a stream to write streamType FastCGI records, using
1858 * the ipcFd and RequestId contained in *reqDataPtr.
1859 * The stream performs OS writes of up to bufflen bytes.
1860 *
1861 *----------------------------------------------------------------------
1862 */
1863static FCGX_Stream *NewWriter(ReqData *reqDataPtr, int bufflen, int streamType)
1864{
1865 return NewStream(reqDataPtr, bufflen, FALSE, streamType);
1866}
1867
1868
1869/*
1870 *----------------------------------------------------------------------
1871 *
1872 * CreateWriter --
1873 *
1874 * Creates a stream to write streamType FastCGI records, using
1875 * the given ipcFd and request Id. This function is provided
1876 * for use by cgi-fcgi. In order to be defensive against misuse,
1877 * this function leaks a little storage; cgi-fcgi doesn't care.
1878 *
1879 *----------------------------------------------------------------------
1880 */
1881FCGX_Stream *CreateWriter(
1882 int ipcFd,
1883 int requestId,
1884 int bufflen,
1885 int streamType)
1886{
1887 ReqData *reqDataPtr = Malloc(sizeof(ReqData));
1888 reqDataPtr->ipcFd = ipcFd;
1889 reqDataPtr->requestId = requestId;
1890 /*
1891 * Suppress writing an FCGI_END_REQUEST record.
1892 */
1893 reqDataPtr->nWriters = 2;
1894 return NewWriter(reqDataPtr, bufflen, streamType);
1895}
1896\f
1897/*
1898 *======================================================================
1899 * Control
1900 *======================================================================
1901 */
1902
1903static int isCGI = -1;
1904static int isFastCGI = -1;
1905
1906/*
1907 *----------------------------------------------------------------------
1908 *
1909 * FCGX_IsCGI --
1910 *
1911 * This routine determines if the process is running as a CGI or
1912 * FastCGI process. The distinction is made by determining whether
1913 * FCGI_LISTENSOCK_FILENO is a listener ipcFd or the end of a
1914 * pipe (ie. standard in).
1915 *
1916 * Results:
1917 * TRUE if the process is a CGI process, FALSE if FastCGI.
1918 *
1919 * Side effects:
1920 * If this is a FastCGI process there's a chance that a connection
1921 * will be accepted while performing the test. If this occurs,
1922 * the connection is saved and used later by the FCGX_Accept logic.
1923 *
1924 *----------------------------------------------------------------------
1925 */
1926int FCGX_IsCGI(void)
1927{
1928 /*
1929 * Already been here, no need to test again.
1930 */
1931 if(isCGI != -1) {
1932 return isCGI;
1933 }
1934
1935 if(!osLibInitialized) {
1936 if(OS_LibInit(NULL) == -1) {
1937 exit(OS_Errno);
1938 }
1939 osLibInitialized = 1;
1940 }
1941
1942 isFastCGI = OS_IsFcgi();
1943 isCGI = !isFastCGI;
1944 return isCGI;
1945}
1946\f
1947/*
1948 *----------------------------------------------------------------------
1949 *
1950 * FCGX_Finish --
1951 *
1952 * Finishes the current request from the HTTP server.
1953 *
1954 * Side effects:
1955 *
1956 * Finishes the request accepted by (and frees any
1957 * storage allocated by) the previous call to FCGX_Accept.
1958 *
1959 * DO NOT retain pointers to the envp array or any strings
1960 * contained in it (e.g. to the result of calling FCGX_GetParam),
1961 * since these will be freed by the next call to FCGX_Finish
1962 * or FCGX_Accept.
1963 *
1964 *----------------------------------------------------------------------
1965 */
1966static ReqData *reqDataPtr = NULL;
1967
1968void FCGX_Finish(void)
1969{
1970 if(reqDataPtr != NULL && reqDataPtr->inStream != NULL) {
1971 /*
1972 * Complete the previous request.
1973 */
1974 int errStatus = FCGX_FClose(reqDataPtr->errStream);
1975 int outStatus = FCGX_FClose(reqDataPtr->outStream);
1976 int prevRequestFailed = (errStatus != 0)
1977 || (outStatus != 0)
1978 || (FCGX_GetError(reqDataPtr->inStream) != 0);
1979 ASSERT(reqDataPtr->nWriters == 0);
1980 FreeStream(&reqDataPtr->inStream);
1981 FreeStream(&reqDataPtr->outStream);
1982 FreeStream(&reqDataPtr->errStream);
1983 FreeParams(&reqDataPtr->paramsPtr);
1984 if(prevRequestFailed || !reqDataPtr->keepConnection) {
1985 OS_IpcClose(reqDataPtr->ipcFd);
1986 reqDataPtr->ipcFd = -1;
1987 }
1988 }
1989}
1990\f
1991/*
1992 *----------------------------------------------------------------------
1993 *
1994 * FCGX_Accept --
1995 *
1996 * Accepts a new request from the HTTP server.
1997 *
1998 * Results:
1999 * 0 for successful call, -1 for error.
2000 *
2001 * Side effects:
2002 *
2003 * Finishes the request accepted by (and frees any
2004 * storage allocated by) the previous call to FCGX_Accept.
2005 * Creates input, output, and error streams and
2006 * assigns them to *in, *out, and *err respectively.
2007 * Creates a parameters data structure to be accessed
2008 * via getenv(3) (if assigned to environ) or by FCGX_GetParam
2009 * and assigns it to *envp.
2010 *
2011 * DO NOT retain pointers to the envp array or any strings
2012 * contained in it (e.g. to the result of calling FCGX_GetParam),
2013 * since these will be freed by the next call to FCGX_Finish
2014 * or FCGX_Accept.
2015 *
2016 *----------------------------------------------------------------------
2017 */
2018static ReqData reqData;
2019static char *webServerAddressList = NULL;
2020
2021int FCGX_Accept(
2022 FCGX_Stream **in,
2023 FCGX_Stream **out,
2024 FCGX_Stream **err,
2025 FCGX_ParamArray *envp)
2026{
2027 /*
2028 * If our compiler doesn't play by the ISO rules for struct
2029 * layout, halt.
2030 */
2031 ASSERT(sizeof(FCGI_Header) == FCGI_HEADER_LEN);
2032
2033 if(!osLibInitialized) {
2034 if(OS_LibInit(NULL) == -1) {
2035 exit(OS_Errno);
2036 }
2037 osLibInitialized = 1;
2038 }
2039
2040 /*
2041 * If our compiler doesn't play by the ISO rules for struct
2042 * layout, halt.
2043 */
2044 ASSERT(sizeof(FCGI_Header) == FCGI_HEADER_LEN);
2045
2046 if(reqDataPtr == NULL) {
2047 /*
2048 * Very first call, so capture FCGI_WEB_SERVER_ADDRS from
2049 * the initial environment, and initialize reqDataPtr
2050 * and parts of reqData.
2051 */
2052 char *p = getenv("FCGI_WEB_SERVER_ADDRS");
2053 if (p != NULL) {
2054 webServerAddressList = StringCopy(p);
2055 }
2056 reqDataPtr = &reqData;
2057 reqDataPtr->ipcFd = -1;
2058 reqDataPtr->inStream = NULL;
2059 reqDataPtr->outStream = NULL;
2060 reqDataPtr->errStream = NULL;
2061 } else {
2062 /*
2063 * Not the first call. Finish the current request, if any.
2064 */
2065 FCGX_Finish();
2066 }
2067 for(;;) {
2068 /*
2069 * If a connection isn't open, accept a new connection (blocking).
2070 * If an OS error occurs in accepting the connection,
2071 * return -1 to the caller, who should exit.
2072 */
2073 if(reqDataPtr->ipcFd < 0) {
2074 reqDataPtr->ipcFd = OS_FcgiIpcAccept(webServerAddressList);
2075 if(reqDataPtr->ipcFd < 0) {
2076 reqDataPtr = NULL;
a3c43615 2077 return (errno > 0) ? (0 - errno) : -9999;
0198fd3c 2078 }
2079 }
2080 /*
2081 * A connection is open. Read from the connection in order to
2082 * get the request's role and environment. If protocol or other
2083 * errors occur, close the connection and try again.
2084 */
2085 reqDataPtr->isBeginProcessed = FALSE;
2086 reqDataPtr->inStream = NewReader(reqDataPtr, 8192, 0);
2087 FillBuffProc(reqDataPtr->inStream);
2088 if(!reqDataPtr->isBeginProcessed) {
2089 goto TryAgain;
2090 }
2091 {
2092 char *roleStr;
2093 switch(reqDataPtr->role) {
2094 case FCGI_RESPONDER:
2095 roleStr = "FCGI_ROLE=RESPONDER";
2096 break;
2097 case FCGI_AUTHORIZER:
2098 roleStr = "FCGI_ROLE=AUTHORIZER";
2099 break;
2100 case FCGI_FILTER:
2101 roleStr = "FCGI_ROLE=FILTER";
2102 break;
2103 default:
2104 goto TryAgain;
2105 }
2106 reqDataPtr->paramsPtr = NewParams(30);
2107 PutParam(reqDataPtr->paramsPtr, StringCopy(roleStr));
2108 }
2109 SetReaderType(reqDataPtr->inStream, FCGI_PARAMS);
2110 if(ReadParams(reqDataPtr->paramsPtr, reqDataPtr->inStream) >= 0) {
2111 /*
2112 * Finished reading the environment. No errors occurred, so
2113 * leave the connection-retry loop.
2114 */
2115 break;
2116 }
2117 /*
2118 * Close the connection and try again.
2119 */
2120 TryAgain:
2121 FreeParams(&reqDataPtr->paramsPtr);
2122 FreeStream(&reqDataPtr->inStream);
2123 OS_Close(reqDataPtr->ipcFd);
2124 reqDataPtr->ipcFd = -1;
2125 } /* for (;;) */
2126 /*
2127 * Build the remaining data structures representing the new
2128 * request and return successfully to the caller.
2129 */
2130 SetReaderType(reqDataPtr->inStream, FCGI_STDIN);
2131 reqDataPtr->outStream = NewWriter(reqDataPtr, 8192, FCGI_STDOUT);
2132 reqDataPtr->errStream = NewWriter(reqDataPtr, 512, FCGI_STDERR);
2133 reqDataPtr->nWriters = 2;
2134 *in = reqDataPtr->inStream;
2135 *out = reqDataPtr->outStream;
2136 *err = reqDataPtr->errStream;
2137 *envp = reqDataPtr->paramsPtr->vec;
2138 return 0;
2139}
2140\f
2141/*
2142 *----------------------------------------------------------------------
2143 *
2144 * FCGX_StartFilterData --
2145 *
2146 * stream is an input stream for a FCGI_FILTER request.
2147 * stream is positioned at EOF on FCGI_STDIN.
2148 * Repositions stream to the start of FCGI_DATA.
2149 * If the preconditions are not met (e.g. FCGI_STDIN has not
2150 * been read to EOF) sets the stream error code to
2151 * FCGX_CALL_SEQ_ERROR.
2152 *
2153 * Results:
2154 * 0 for a normal return, < 0 for error
2155 *
2156 *----------------------------------------------------------------------
2157 */
2158
2159int FCGX_StartFilterData(FCGX_Stream *stream)
2160{
2161 FCGX_Stream_Data *data = stream->data;
2162 if(data->reqDataPtr->role != FCGI_FILTER
2163 || !stream->isReader
2164 || !stream->isClosed
2165 || data->type != FCGI_STDIN) {
2166 SetError(stream, FCGX_CALL_SEQ_ERROR);
2167 return -1;
2168 }
2169 SetReaderType(reqDataPtr->inStream, FCGI_DATA);
2170 return 0;
2171}
2172\f
2173/*
2174 *----------------------------------------------------------------------
2175 *
2176 * FCGX_SetExitStatus --
2177 *
2178 * Sets the exit status for stream's request. The exit status
2179 * is the status code the request would have exited with, had
2180 * the request been run as a CGI program. You can call
2181 * SetExitStatus several times during a request; the last call
2182 * before the request ends determines the value.
2183 *
2184 *----------------------------------------------------------------------
2185 */
2186
2187void FCGX_SetExitStatus(int status, FCGX_Stream *stream)
2188{
2189 FCGX_Stream_Data *data = stream->data;
2190 data->reqDataPtr->appStatus = status;
2191}
Unnamed repository; edit this file to name it for gitweb.