Annotation of libwww/Library/src/HTBound.c, revision 2.1

2.1     ! frystyk     1: /*                                                                  HTBound.c
        !             2: **     MIME MULTIPART PARSER STREAM
        !             3: **
        !             4: **     (c) COPYRIGHT MIT 1995.
        !             5: **     Please first read the full copyright statement in the file COPYRIGH.
        !             6: **
        !             7: **     This stream parses a MIME multipart stream and builds a set of new
        !             8: **     streams via the stream stack each time we encounter a boundary start.
        !             9: **     We get the boundary from the normal MIME parser via the Request object
        !            10: **
        !            11: ** Authors
        !            12: **     HF      Henrik Frystyk <frystyk@w3.org>
        !            13: **
        !            14: ** History:
        !            15: **     Nov 95  Written from scratch
        !            16: **
        !            17: */
        !            18: 
        !            19: /* Library include files */
        !            20: #include "tcp.h"
        !            21: #include "HTUtils.h"
        !            22: #include "HTString.h"
        !            23: #include "HTParse.h"
        !            24: #include "HTSocket.h"
        !            25: #include "HTStream.h"
        !            26: #include "HTFWrite.h"
        !            27: #include "HTReqMan.h"
        !            28: #include "HTBound.h"                                    /* Implemented here */
        !            29: 
        !            30: #define PUTBLOCK(b, l) (*me->target->isa->put_block)(me->target, b, l)
        !            31: #define PUTDEBUG(b, l) (*me->debug->isa->put_block)(me->debug, b, l)
        !            32: #define FREE_TARGET    (*me->target->isa->_free)(me->target)
        !            33: 
        !            34: struct _HTStream {
        !            35:     CONST HTStreamClass *      isa;
        !            36:     HTStream *                 target;
        !            37:     HTStream *                 orig_target;
        !            38:     HTFormat                   format;
        !            39:     HTStream *                 debug;            /* For preamble and epilog */
        !            40:     HTRequest *                        request;
        !            41:     BOOL                       body;             /* Body or preamble|epilog */
        !            42:     HTSocketEOL                        state;
        !            43:     int                                dash;                    /* Number of dashes */
        !            44:     char *                     boundary;
        !            45:     char *                     bpos;
        !            46: };
        !            47: 
        !            48: /* ------------------------------------------------------------------------- */
        !            49: 
        !            50: /*
        !            51: **     Searches for FTP line until buffer fills up or a CRLF or LF is found
        !            52: */
        !            53: PRIVATE int HTBoundary_put_block (HTStream * me, CONST char * b, int l)
        !            54: {
        !            55:     CONST char *start = b;
        !            56:     CONST char *end = b;
        !            57:     while (l-- > 0) {
        !            58:        if (me->state == EOL_FCR) {
        !            59:            me->state = (*b == LF) ? EOL_FLF : EOL_BEGIN;
        !            60:        } else if (me->state == EOL_FLF) {
        !            61:            if (me->dash == 2) {
        !            62:                while (l>0 && *me->bpos && *me->bpos==*b) l--, me->bpos++, b++;
        !            63:                if (!*me->bpos) {
        !            64:                    if (STREAM_TRACE && !*me->bpos)
        !            65:                        TTYPrint(TDEST, "Boundary.... `%s\' found\n", me->boundary);
        !            66:                    me->bpos = me->boundary;
        !            67:                    me->body = YES;
        !            68:                    me->state = EOL_DOT;
        !            69:                } else if (l>0) {
        !            70:                    me->dash = 0;
        !            71:                    me->bpos = me->boundary;
        !            72:                    me->state = EOL_BEGIN;
        !            73:                }
        !            74:            }
        !            75:            if (*b == '-') {
        !            76:                me->dash++;
        !            77:            } else if (*b != LF) {
        !            78:                me->dash = 0;
        !            79:                me->state = EOL_BEGIN;
        !            80:            }
        !            81:        } else if (me->state == EOL_SLF) {          /* Look for closing '--' */
        !            82:            if (me->dash == 4) {
        !            83:                if (end > start) {
        !            84:                    int status = PUTBLOCK(start, end-start);
        !            85:                    if (status != HT_OK) return status;
        !            86:                }
        !            87:                if (STREAM_TRACE) TTYPrint(TDEST, "Boundary.... Ending\n");
        !            88:                start = b;
        !            89:                me->dash = 0;
        !            90:                me->state = EOL_BEGIN;
        !            91:            }
        !            92:            if (*b == '-') {
        !            93:                me->dash++;
        !            94:            } else if (*b != LF) {
        !            95:                me->dash = 0;
        !            96:                me->state = EOL_BEGIN;
        !            97:            }
        !            98:            me->body = NO;
        !            99:        } else if (me->state == EOL_DOT) {
        !           100:            int status;
        !           101:            if (me->body) {
        !           102:                if (me->target) FREE_TARGET;
        !           103:                me->target = HTStreamStack(WWW_MIME,me->format,me->orig_target,
        !           104:                                           me->request, YES);
        !           105:                if (end > start) {
        !           106:                    if ((status = PUTBLOCK(start, end-start)) != HT_OK)
        !           107:                        return status;
        !           108:                }
        !           109:            } else {
        !           110:                if (me->debug)
        !           111:                    if ((status = PUTDEBUG(start, end-start)) != HT_OK)
        !           112:                        return status;
        !           113:            }
        !           114:            start = b;
        !           115:            if (*b == '-') me->dash++;
        !           116:            me->state = EOL_SLF;
        !           117:        } else if (*b == CR) {
        !           118:            me->state = EOL_FCR;
        !           119:            end = b;
        !           120:        } else if (*b == LF) {
        !           121:            if (me->state != EOL_FCR) end = b;
        !           122:            me->state = EOL_FLF;
        !           123:        }
        !           124:        b++;
        !           125:     }
        !           126:     return (start<b && me->body) ? PUTBLOCK(start, b-start) : HT_OK;
        !           127: }
        !           128: 
        !           129: PRIVATE int HTBoundary_put_string (HTStream * me, CONST char * s)
        !           130: {
        !           131:     return HTBoundary_put_block(me, s, (int) strlen(s));
        !           132: }
        !           133: 
        !           134: PRIVATE int HTBoundary_put_character (HTStream * me, char c)
        !           135: {
        !           136:     return HTBoundary_put_block(me, &c, 1);
        !           137: }
        !           138: 
        !           139: PRIVATE int HTBoundary_flush (HTStream * me)
        !           140: {
        !           141:     return (*me->target->isa->flush)(me->target);
        !           142: }
        !           143: 
        !           144: PRIVATE int HTBoundary_free (HTStream * me)
        !           145: {
        !           146:     int status = HT_OK;
        !           147:     if (me->target) {
        !           148:        if ((status = (*me->target->isa->_free)(me->target)) == HT_WOULD_BLOCK)
        !           149:            return HT_WOULD_BLOCK;
        !           150:     }
        !           151:     if (PROT_TRACE) TTYPrint(TDEST, "Boundary.... FREEING....\n");
        !           152:     FREE(me->boundary);
        !           153:     free(me);
        !           154:     return status;
        !           155: }
        !           156: 
        !           157: PRIVATE int HTBoundary_abort (HTStream * me, HTList * e)
        !           158: {
        !           159:     int status = HT_ERROR;
        !           160:     if (me->target) status = (*me->target->isa->abort)(me->target, e);
        !           161:     if (PROT_TRACE) TTYPrint(TDEST, "Boundary.... ABORTING...\n");
        !           162:     FREE(me->boundary);
        !           163:     free(me);
        !           164:     return status;
        !           165: }
        !           166: 
        !           167: PRIVATE CONST HTStreamClass HTBoundaryClass =
        !           168: {              
        !           169:     "HTBoundary",
        !           170:     HTBoundary_flush,
        !           171:     HTBoundary_free,
        !           172:     HTBoundary_abort,
        !           173:     HTBoundary_put_character,
        !           174:     HTBoundary_put_string,
        !           175:     HTBoundary_put_block
        !           176: };
        !           177: 
        !           178: PUBLIC HTStream * HTBoundary   (HTRequest *    request,
        !           179:                                void *          param,
        !           180:                                HTFormat        input_format,
        !           181:                                HTFormat        output_format,
        !           182:                                HTStream *      output_stream)
        !           183: {
        !           184:     HTStream * me = (HTStream *) calloc(1, sizeof(HTStream));
        !           185:     if (!me) outofmem(__FILE__, "HTBoundary");
        !           186:     me->isa = &HTBoundaryClass;
        !           187:     me->request = request;
        !           188:     me->format = output_format;
        !           189:     me->orig_target = output_stream;
        !           190:     me->debug = HTRequest_debugStream(request);
        !           191:     me->state = EOL_FLF;
        !           192:     if (request->boundary) {
        !           193:        StrAllocCopy(me->boundary, request->boundary);         /* Local copy */
        !           194:        me->bpos = me->boundary;
        !           195:        if (STREAM_TRACE)
        !           196:            TTYPrint(TDEST,"Boundary.... Stream created with boundary '%s\'\n", me->boundary);
        !           197:        return me;
        !           198:     } else {
        !           199:        if (STREAM_TRACE) TTYPrint(TDEST, "Boundary.... <UNKNOWN>\n");
        !           200:        return HTBlackHole();
        !           201:     }
        !           202: }

Webmaster