Annotation of parser3/src/types/pa_vform.C, revision 1.85

1.27      paf         1: /**    @file
                      2:        Parser: @b form class.
1.10      paf         3: 
1.84      paf         4:        Copyright(c) 2001-2005 ArtLebedev Group (http://www.artlebedev.com)
1.52      paf         5:        Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.39      parser      6: 
1.38      parser      7:        based on The CGI_C library, by Thomas Boutell.
1.3       paf         8: */
1.59      paf         9: 
1.85    ! misha      10: static const char * const IDENT_VFORM_C="$Date: 2005/08/09 08:14:54 $";
1.3       paf        11: 
1.14      paf        12: #include "pa_sapi.h"
1.1       paf        13: #include "pa_vform.h"
                     14: #include "pa_vstring.h"
1.3       paf        15: #include "pa_globals.h"
                     16: #include "pa_request.h"
1.9       paf        17: #include "pa_vfile.h"
1.21      paf        18: #include "pa_common.h"
1.30      parser     19: #include "pa_vtable.h"
1.50      paf        20: #include "pa_charset.h"
1.3       paf        21: 
1.62      paf        22: // defines
                     23: 
                     24: //#define DEBUG_POST
                     25: 
1.3       paf        26: // parse helper funcs
                     27: 
1.75      paf        28: static size_t getHeader(const char* data, size_t len){
1.25      paf        29:     size_t i;
                     30:     int enter=-1;
1.3       paf        31:     if (data)
1.62      paf        32:                for (i=0;i<len;i++)
                     33:                        if (data[i]=='\n'){
                     34:                                if (enter>=0) enter++;
                     35:                                if (enter>1) return i;
                     36:                        } else if (data[i]!='\r') enter=0;
                     37:                return 0;
1.3       paf        38: }
                     39: 
1.75      paf        40: static const char* searchAttribute(const char* data, 
                     41:                                                         const char* attr,  //< expected to be lowercased
1.62      paf        42:                                                         size_t len){
1.16      paf        43:     size_t i;
1.3       paf        44:     if (data)
1.62      paf        45:                for (i=0;i<len;i++)
1.83      paf        46:                        if (tolower((unsigned char)data[i])==*attr){
1.62      paf        47:                                size_t j;
                     48:                                for (j=i+1;j<=len;j++)
                     49:                                        if (!attr[j-i]) return &data[j];
                     50:                                        else {
                     51:                                                if (j==len) break;
1.83      paf        52:                                                if (attr[j-i]!=tolower((unsigned char)data[j])) break;
1.62      paf        53:                                        }
                     54:                        }
                     55: 
                     56:        return NULL;
1.3       paf        57: }
                     58: 
                     59: // VForm
                     60: 
1.75      paf        61: extern Methoded* form_base_class;
1.57      paf        62: 
1.77      paf        63: VForm::VForm(Request_charsets& acharsets, Request_info& arequest_info): VStateless_class(0, form_base_class),
                     64:        fcharsets(acharsets),
                     65:        frequest_info(arequest_info),
                     66:        filled_source(0),
                     67:        filled_client(0) {
1.50      paf        68: }
                     69: 
1.75      paf        70: char *VForm::strpart(const char* str, size_t len) {
                     71:     char *result=new(PointerFreeGC) char[len+1];
1.18      paf        72:     memcpy(result, str, len);
1.3       paf        73:     result[len]=0;
                     74:     return result;
                     75: }
                     76: 
1.75      paf        77: char *VForm::getAttributeValue(const char* data, char *attr, size_t len) {
                     78:     const char* value=searchAttribute(data, attr, len);
1.3       paf        79:     if (value){
1.18      paf        80:                size_t i;
                     81:                if (!(len-=value-data)) return NULL;
                     82:                if (*value=='"') {
                     83:                        for (i=1;i<len;i++)     if (value[i]=='"') break;
                     84:                        return strpart(&value[1], i-1);
                     85:                } else {
                     86:                        for (i=0;i<len;i++)     if (strchr(" ;\"\n\r", value[i])) break;
                     87:                        return strpart(value, i);
                     88:                }
1.3       paf        89:     }
                     90:     return NULL;
                     91: }
                     92: 
1.75      paf        93: String::C VForm::transcode(const char* client, size_t client_size) {
                     94:        return Charset::transcode(
                     95:                String::C(strdup(client, client_size), client_size),
1.77      paf        96:                fcharsets.client(),
                     97:                fcharsets.source());
1.49      paf        98: }
                     99: 
1.75      paf       100: void VForm::ParseGetFormInput(const char* query_string, size_t length) {
1.42      parser    101:        ParseFormInput(query_string, length);
1.3       paf       102: }
                    103: 
1.69      paf       104: 
1.75      paf       105: static int atoi(const char* data, size_t alength) {
1.69      paf       106:        char buf[MAX_STRING];
1.82      paf       107:        size_t length=min(alength, sizeof(buf)-1);
                    108:        memcpy(buf, data, length); buf[length]=0;
1.69      paf       109:        return atoi(buf);
                    110: }
1.75      paf       111: void VForm::ParseFormInput(const char* data, size_t length) {
1.68      paf       112:        // cut out ?image_map_tail
                    113:        {
1.67      paf       114:                for(size_t pos=0; pos<length; pos++) {
                    115:                        if(data[pos]=='?') {
1.69      paf       116:                                size_t start=pos+1;
                    117:                                size_t aftercomma=start;
                    118:                                size_t lookingcomma=start;
                    119:                                for(; lookingcomma<length; lookingcomma++) {
                    120:                                        if(data[lookingcomma]==',') {
                    121:                                                aftercomma=++lookingcomma;
                    122:                                                break;
                    123:                                        }
                    124:                                }                               
                    125:                                
                    126:                                if(aftercomma>start) { // ?x,y
                    127:                                        int x=atoi(data+start, aftercomma-1-start);
                    128:                                        int y=atoi(data+aftercomma, length-aftercomma);
1.75      paf       129:                                        imap.put(String("x"), new VInt(x));
                    130:                                        imap.put(String("y"), new VInt(y));
1.69      paf       131:                                } else { // ?qtail
                    132:                                        AppendFormEntry("qtail", data+start, length-start);
                    133:                                }
1.67      paf       134:                                // cut tail
                    135:                                length=pos;
                    136:                                break;
                    137:                        }
                    138:                }
                    139:        }
1.68      paf       140:        // Scan for pairs, unescaping and storing them as they are found
                    141:        for(size_t pos=0; pos<length; ) {
1.16      paf       142:                size_t start=pos;
1.68      paf       143:                size_t finish=length;
                    144:                for(; pos<length; pos++) {
                    145:                        if(data[pos]=='&') {
                    146:                                finish=pos++;
1.3       paf       147:                                break;
                    148:                        }
                    149:                }
1.68      paf       150: 
                    151:                size_t aftereq=start;
                    152:                size_t lookingeq=start;
                    153:                for(; lookingeq<finish; lookingeq++) {
                    154:                        if(data[lookingeq]=='=') {
                    155:                                aftereq=++lookingeq;
1.3       paf       156:                                break;
                    157:                        }
                    158:                }
1.45      paf       159: 
1.75      paf       160:                const char* attr=aftereq>start?unescape_chars(data+start, aftereq-1-start):"nameless";
                    161:                char *value=unescape_chars(data+aftereq, finish-aftereq);
1.68      paf       162:                AppendFormEntry(attr, value, strlen(value));
1.3       paf       163:        }
                    164: }
                    165: 
1.62      paf       166: static char *pa_tolower(char *s) {
                    167:        for(char *p=s; *p; p++)
1.83      paf       168:                *p=(char)tolower((unsigned char)*p);
1.62      paf       169:        return s;
                    170: }
1.49      paf       171: void VForm::ParseMimeInput(
                    172:                                                   char *content_type, 
1.75      paf       173:                                                   const char* data, size_t length) {
1.3       paf       174: /* Scan for mime-presented pairs, storing them as they are found. */
                    175:        const char 
1.62      paf       176:                *boundary=pa_tolower(getAttributeValue(content_type, "boundary=", strlen(content_type))), 
1.3       paf       177:            *lastData=&data[length];
                    178:        if(!boundary) 
1.54      paf       179:                throw Exception(0, 
1.18      paf       180:                        0, 
1.3       paf       181:                        "VForm::ParseMimeInput no boundary attribute of Content-Type");
                    182: 
                    183:        while(true) {
1.66      paf       184:                const char 
1.18      paf       185:                        *dataStart=searchAttribute(data, boundary, lastData-data), 
                    186:                        *dataEnd=searchAttribute(dataStart, boundary, lastData-dataStart);
                    187:                size_t headerSize=getHeader(dataStart, lastData-dataStart);
1.3       paf       188: 
                    189:                if(!dataStart|!dataEnd|!headerSize) break;
1.47      paf       190:                if(searchAttribute(dataStart, "content-disposition: form-data", headerSize)) {
1.16      paf       191:                        size_t valueSize=(dataEnd-dataStart)-headerSize-5-strlen(boundary);
1.18      paf       192:                        char *attr=getAttributeValue(dataStart, " name=", headerSize), 
                    193:                             *fName=getAttributeValue(dataStart, " filename=", headerSize);
1.3       paf       194: 
1.78      paf       195:                        if(attr) {
1.3       paf       196:                                /* OK, we have a new pair, add it to the list. */
1.80      paf       197:                                // fName checks are because MSIE passes unassigned <input type=file> as filename="" and empty body 
1.85    ! misha     198:                                if( fName && (strlen(fName) || valueSize) ){
        !           199:                                        AppendFormFileEntry(attr, 
1.80      paf       200:                                                valueSize? &dataStart[headerSize+1]: "", 
                    201:                                                valueSize, 
1.85    ! misha     202:                                                fName); 
        !           203:                                } else {
        !           204:                                        AppendFormEntry(attr, 
        !           205:                                                valueSize? &dataStart[headerSize+1]: "", 
        !           206:                                                valueSize); 
        !           207:                                }
1.3       paf       208:                        }
                    209:                }
                    210:                data=(dataEnd-strlen(boundary));
                    211:        }
                    212: }
                    213: 
1.85    ! misha     214: void VForm::AppendFormFileEntry(const char* cname_cstr, 
1.75      paf       215:                            const char* raw_cvalue_ptr, const size_t raw_cvalue_size, 
1.85    ! misha     216:                            const char* file_name_cstr){
        !           217: 
1.75      paf       218:        const String& sname=*new String(transcode(cname_cstr, strlen(cname_cstr)));
1.85    ! misha     219:        VFile* vfile=new VFile;
        !           220:        // maybe transcode text/* files?
        !           221:        const String& sfile_name=*new String(strdup(file_name_cstr));
        !           222:        vfile->set(true/*tainted*/, raw_cvalue_ptr, raw_cvalue_size, sfile_name.cstr());
        !           223: 
        !           224:        fields.put_dont_replace(sname, vfile);
        !           225: 
        !           226:        // files
        !           227:        Value* vhash=files.get(sname);
        !           228:        if(!vhash){
        !           229:                // first appearence
        !           230:                vhash=new VHash;
        !           231:                files.put(sname, vhash);
        !           232:        }
        !           233:        HashStringValue& hash=*vhash->get_hash();
1.75      paf       234: 
1.85    ! misha     235:        hash.put(String::Body::Format(hash.count()), vfile);
        !           236: }
        !           237: 
        !           238: void VForm::AppendFormEntry(const char* cname_cstr, const char* raw_cvalue_ptr, const size_t raw_cvalue_size) {
        !           239:        const String& sname=*new String(transcode(cname_cstr, strlen(cname_cstr)));
1.75      paf       240: 
1.85    ! misha     241:        const char* premature_zero_pos=(const char* )memchr(raw_cvalue_ptr, 0, raw_cvalue_size);
        !           242:        size_t cvalue_size=premature_zero_pos?premature_zero_pos-(const char* )raw_cvalue_ptr
        !           243:                :raw_cvalue_size;
        !           244:        char *cvalue_ptr=strdup(raw_cvalue_ptr, cvalue_size); 
        !           245:        fix_line_breaks(cvalue_ptr, cvalue_size);
        !           246:        String& string=*new String(transcode(cvalue_ptr, cvalue_size), true);
        !           247: 
        !           248:        // tables
        !           249:        {
        !           250:                Value* vtable=tables.get(sname);
        !           251:                if(!vtable) {
        !           252:                        // first appearence
        !           253:                        Table::columns_type columns(new ArrayString(1));
        !           254:                        *columns+=new String("field");
1.75      paf       255: 
1.85    ! misha     256:                        vtable=new VTable(new Table(columns));
        !           257:                        tables.put(sname, vtable);
1.36      parser    258:                }
1.85    ! misha     259:                Table& table=*vtable->get_table();
        !           260: 
        !           261:                // this string becomes next row
        !           262:                Table::element_type row(new ArrayString(1));
        !           263:                *row+=&string;
        !           264:                table+=row;
1.3       paf       265:        }
1.69      paf       266: 
1.85    ! misha     267:        fields.put_dont_replace(sname, new VString(string));
1.69      paf       268: }
                    269: 
1.85    ! misha     270: 
        !           271: void VForm::refill_fields_tables_and_files() {
1.77      paf       272:        fields.clear();
                    273:        tables.clear();
1.85    ! misha     274:        files.clear();
1.77      paf       275:        imap.clear();
1.75      paf       276: 
1.77      paf       277:        //frequest_info.query_string="a=123";
1.3       paf       278:        // parsing QS [GET and ?name=value from uri rewrite)]
1.77      paf       279:        if(frequest_info.query_string) {
                    280:                size_t length=strlen(frequest_info.query_string);
                    281:                char *buf=strdup(frequest_info.query_string, length);
1.42      parser    282:                ParseGetFormInput(buf, length);
                    283:        }
1.62      paf       284: 
                    285: #ifdef DEBUG_POST
1.77      paf       286:        frequest_info.method="POST";
1.62      paf       287:        void *data;
1.80      paf       288:        file_read(*new String("test.stdin"),
1.62      paf       289:                           data, request.post_size, 
                    290:                           false/*as_text*/);   
                    291:        request.post_data=(char*)data;
1.77      paf       292:        frequest_info.content_type="multipart/form-data; boundary=----------mcqY2UDNcdEAoN1mLmne2i";
1.62      paf       293: 
                    294: #endif
1.57      paf       295: 
                    296:        // parsing POST data
1.77      paf       297:        if(frequest_info.method) {
                    298:                if(const char* content_type=frequest_info.content_type)
                    299:                        if(StrEqNc(frequest_info.method, "post", true)) {
1.18      paf       300:                                if(StrEqNc(content_type, "application/x-www-form-urlencoded", true)) 
1.77      paf       301:                                        ParseFormInput(frequest_info.post_data, frequest_info.post_size);
1.75      paf       302:                                else if(StrEqNc(content_type, "multipart/form-data", 0))
                    303:                                        ParseMimeInput(strdup(content_type), 
1.77      paf       304:                                                frequest_info.post_data, frequest_info.post_size);
1.3       paf       305:                        }
1.57      paf       306:        }
1.56      paf       307: 
1.77      paf       308:        filled_source=&fcharsets.source();
                    309:        filled_client=&fcharsets.client();
                    310: }
                    311: 
1.85    ! misha     312: bool VForm::should_refill_fields_tables_and_files() {
1.77      paf       313:        return &fcharsets.source()!=filled_source
                    314:                || &fcharsets.client()!=filled_client;
1.1       paf       315: }
1.33      parser    316: 
1.75      paf       317: Value* VForm::get_element(const String& aname, Value& aself, bool looking_up) {
1.85    ! misha     318:        if(should_refill_fields_tables_and_files())
        !           319:                refill_fields_tables_and_files();
1.56      paf       320: 
1.36      parser    321:        // $fields
                    322:        if(aname==FORM_FIELDS_ELEMENT_NAME)
1.75      paf       323:                return new VHash(fields);
1.36      parser    324: 
                    325:        // $tables
                    326:        if(aname==FORM_TABLES_ELEMENT_NAME)
1.75      paf       327:                return new VHash(tables);
1.69      paf       328: 
1.85    ! misha     329:        // $files
        !           330:        if(aname==FORM_FILES_ELEMENT_NAME)
        !           331:                return new VHash(files);
        !           332: 
1.77      paf       333:        // $imap
1.69      paf       334:        if(aname==FORM_IMAP_ELEMENT_NAME)
1.75      paf       335:                return new VHash(imap);
1.36      parser    336: 
1.63      paf       337:        // $method
1.75      paf       338:        if(Value* result=VStateless_class::get_element(aname, aself, looking_up))
1.33      parser    339:                return result;
                    340: 
                    341:        // $element
1.75      paf       342:        return fields.get(aname);
1.72      paf       343: }

E-mail: