Annotation of parser3/src/types/pa_vfile.C, revision 1.58

1.34      paf         1: 
1.5       paf         2: /** @file
1.14      paf         3:        Parser: @b file parser type.
1.3       paf         4: 
1.51      moko        5:        Copyright (c) 2001-2012 Art. Lebedev Studio (http://www.artlebedev.com)
1.26      paf         6:        Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.30      paf         7: */
1.3       paf         8: 
1.37      paf         9: #include "classes.h"
1.1       paf        10: #include "pa_vfile.h"
                     11: #include "pa_vstring.h"
1.3       paf        12: #include "pa_vint.h"
1.50      misha      13: #include "pa_request.h"
1.1       paf        14: 
1.58    ! moko       15: volatile const char * IDENT_PA_VFILE_C="$Id: pa_vfile.C,v 1.57 2013/03/11 23:37:43 misha Exp $" IDENT_PA_VFILE_H;
1.51      moko       16: 
1.37      paf        17: // externs
                     18: 
                     19: extern Methoded* file_class;
                     20: 
                     21: // defines for statics
                     22: 
                     23: #define SIZE_NAME "size"
                     24: #define TEXT_NAME "text"
                     25: 
1.50      misha      26: #define MODE_VALUE_TEXT "text"
                     27: #define MODE_VALUE_BINARY "binary"
                     28: 
1.58    ! moko       29: #define CONTENT_TYPE_TEXT "text/plain"
        !            30: #define CONTENT_TYPE_BINARY "application/octet-stream"
        !            31: 
1.37      paf        32: // statics
                     33: 
                     34: static const String size_name(SIZE_NAME);
                     35: static const String text_name(TEXT_NAME);
1.50      misha      36: 
                     37: static const String mode_value_text(MODE_VALUE_TEXT);
                     38: static const String mode_value_binary(MODE_VALUE_BINARY);
1.37      paf        39: 
1.58    ! moko       40: static const String content_type_text(CONTENT_TYPE_TEXT);
        !            41: static const String content_type_binary(CONTENT_TYPE_BINARY);
        !            42: 
1.37      paf        43: // methods
                     44: 
                     45: VStateless_class *VFile::get_class() { return file_class; }
                     46: 
1.58    ! moko       47: void VFile::set_all(bool atainted, bool ais_text_mode, const char* avalue_ptr, size_t avalue_size, const String* afile_name, Value* acontent_type, Request* r) {
1.8       paf        48:        fvalue_ptr=avalue_ptr;
                     49:        fvalue_size=avalue_size;
1.53      misha      50: 
1.43      misha      51:        ftext_tainted=atainted;
1.8       paf        52: 
1.10      paf        53:        ffields.clear();
1.43      misha      54: 
1.58    ! moko       55:        // $.mode
        !            56:        set_mode(ais_text_mode);
        !            57: 
1.53      misha      58:        // $.size
1.50      misha      59:        ffields.put(size_name, new VInt(fvalue_size));
                     60: 
1.53      misha      61:        // $.name
1.50      misha      62:        set_name(afile_name);
                     63: 
                     64:        // $mime-type
                     65:        set_content_type(acontent_type, afile_name, r);
1.54      moko       66: }
1.53      misha      67: 
1.54      moko       68: void VFile::set(bool atainted, bool ais_text_mode, char* avalue_ptr, size_t avalue_size, const String* afile_name, Value* acontent_type, Request* r) {
                     69:        if(ais_text_mode && avalue_ptr && avalue_size) {
                     70:                fix_line_breaks(avalue_ptr, avalue_size);
                     71:        }
1.58    ! moko       72:        set_all(atainted, ais_text_mode, avalue_ptr, avalue_size, afile_name, acontent_type, r);
1.54      moko       73:        fis_text_content=ais_text_mode;
1.50      misha      74: }
                     75: 
1.54      moko       76: void VFile::set_binary(bool atainted, const char* avalue_ptr, size_t avalue_size, const String* afile_name, Value* acontent_type, Request* r) {
1.58    ! moko       77:        set_all(atainted, false, avalue_ptr, avalue_size, afile_name, acontent_type, r);
1.54      moko       78:        fis_text_content=false;
                     79: }
1.53      misha      80: 
1.58    ! moko       81: void VFile::set(VFile& avfile, bool aset_text_mode, bool ais_text_mode, const String* afile_name, Value* acontent_type, Request* r) {
        !            82:        fvalue_ptr=avfile.fvalue_ptr;
        !            83:        fvalue_size=avfile.fvalue_size;
        !            84:        ftext_tainted=avfile.ftext_tainted;
        !            85:        fis_text_content=avfile.fis_text_content;
1.53      misha      86: 
                     87:        ffields.clear();
1.50      misha      88:        for(HashStringValue::Iterator i(avfile.ffields); i; i.next())
1.53      misha      89:                if(i.key() != text_name) // do not copy cached .text value
1.58    ! moko       90:                        ffields.put(*new String(i.key(), String::L_TAINTED), i.value());
1.54      moko       91: 
1.58    ! moko       92:        if(aset_text_mode)
        !            93:                set_mode(ais_text_mode);
        !            94: 
        !            95:        if(afile_name)
        !            96:                set_name(afile_name);
        !            97: 
        !            98:        if(acontent_type || afile_name)
        !            99:                set_content_type(acontent_type, afile_name, r);
1.53      misha     100: }
                    101: 
                    102: const char* VFile::text_cstr() {
                    103:        const char* p=value_ptr();
1.54      moko      104:        if(fis_text_content)
1.53      misha     105:                return p;
                    106: 
                    107:        size_t size=fvalue_size;
                    108: 
                    109:        if(const char *premature_zero_pos=(const char *)memchr(p, 0, size))
                    110:                size=premature_zero_pos-p;
                    111: 
                    112:        char *copy_ptr=size?strdup(p, size):0;
1.54      moko      113:        // text mode but binary content
1.53      misha     114:        if(fis_text_mode && size)
                    115:                fix_line_breaks(copy_ptr, size);
                    116:        return copy_ptr;
1.50      misha     117: }
                    118: 
1.53      misha     119: void VFile::set_mode(bool ais_text_mode){
                    120:        fis_text_mode=ais_text_mode;
1.56      moko      121:        if(fvalue_ptr)
                    122:                ffields.put(mode_name, new VString(ais_text_mode? mode_value_text : mode_value_binary ));
1.50      misha     123: }
                    124: 
                    125: void VFile::set_name(const String* afile_name){
1.14      paf       126:        char *lfile_name;
1.57      misha     127:        if(afile_name && !afile_name->is_empty()) {
1.50      misha     128:                lfile_name=strdup(afile_name->taint_cstr(String::L_FILE_SPEC));
1.10      paf       129:                if(char *after_slash=rsplit(lfile_name, '\\'))
                    130:                        lfile_name=after_slash;
                    131:                if(char *after_slash=rsplit(lfile_name, '/'))
                    132:                        lfile_name=after_slash;
1.14      paf       133:        } else
1.54      moko      134:                lfile_name=(char *)NONAME_DAT;
1.50      misha     135: 
1.53      misha     136:        ffields.put(name_name, new VString(*new String(lfile_name, String::L_FILE_SPEC)));
1.50      misha     137: }
1.43      misha     138: 
1.50      misha     139: void VFile::set_content_type(Value* acontent_type, const String* afile_name, Request* r){
                    140:        if(!acontent_type && afile_name && r)
                    141:                acontent_type=new VString(r->mime_type_of(afile_name));
1.43      misha     142: 
1.58    ! moko      143:        if(!acontent_type)
        !           144:                acontent_type=new VString(fis_text_mode ? content_type_text : content_type_binary);
        !           145: 
        !           146:        ffields.put(content_type_name, acontent_type);
1.43      misha     147: }
                    148: 
1.46      misha     149: void VFile::save(Request_charsets& charsets, const String& file_spec, bool is_text, Charset* asked_charset) {
                    150:        if(fvalue_ptr)
                    151:                file_write(charsets, file_spec, fvalue_ptr, fvalue_size, is_text, false/*do_append*/, asked_charset);
                    152:        else
1.58    ! moko      153:                throw Exception(PARSER_RUNTIME, &file_spec, "saving stat-ed file");
1.46      misha     154: }
                    155: 
1.50      misha     156: bool VFile::is_text_mode(const String& mode) {
                    157:        if(mode==mode_value_text)
                    158:                return true;
                    159:        if(mode==mode_value_binary)
                    160:                return false;
1.58    ! moko      161:        throw Exception(PARSER_RUNTIME, &mode, "is invalid mode, must be either '"MODE_VALUE_TEXT"' or '"MODE_VALUE_BINARY"'");
1.50      misha     162: }
1.58    ! moko      163: 
1.50      misha     164: bool VFile::is_valid_mode (const String& mode) {
                    165:        return (mode==mode_value_text || mode==mode_value_binary);
                    166: }
                    167: 
1.45      misha     168: Value* VFile::get_element(const String& aname) {
1.43      misha     169:        Value* result;
                    170: 
                    171:        // $method
1.45      misha     172:        if(result=VStateless_object::get_element(aname))
1.43      misha     173:                return result;
                    174: 
                    175:        // $field
                    176:        if(result=ffields.get(aname))
                    177:                return result;
                    178: 
                    179:        // $text - if not cached
1.53      misha     180:        if(aname == text_name && fvalue_ptr && fvalue_size) {
1.43      misha     181:                // assigned file have ptr and we really have some bytes
1.42      misha     182: 
1.53      misha     183:                result=new VString(*new String(text_cstr(), ftext_tainted ? String::L_TAINTED : String::L_AS_IS));
1.42      misha     184: 
1.53      misha     185:                // cache it
1.43      misha     186:                ffields.put(text_name, result);
1.35      paf       187: 
                    188:                return result;
1.43      misha     189:        }
1.35      paf       190: 
1.43      misha     191:        return 0;
1.3       paf       192: }
1.48      misha     193: 
1.52      moko      194: const String* VFile::get_json_string(Json_options& options){
1.49      moko      195:        String& result=*new String("{\n", String::L_AS_IS);
                    196:        
                    197:        String * indent=NULL;
                    198: 
1.52      moko      199:        if (options.indent){
                    200:                indent = new String(",\n\t", String::L_AS_IS); *indent << options.indent << "\"";
                    201:                result << "\t" << options.indent;
1.49      moko      202:        }
                    203: 
                    204:        result << "\"class\":\"file\"";
1.48      misha     205: 
                    206:        for(HashStringValue::Iterator i(ffields); i; i.next() ){
                    207:                String::Body key=i.key();
                    208:                if(key != text_name){
1.49      moko      209:                        indent ? result << *indent : result << ",\n\"";
                    210:                        result << String(key, String::L_JSON) << "\":" << *i.value()->get_json_string(options);
1.48      misha     211:                }
                    212:        }
                    213: 
                    214:        if(fvalue_ptr){
1.52      moko      215:                switch(options.file){
1.48      misha     216:                        case Json_options::F_BASE64:
                    217:                                {
1.49      moko      218:                                        indent ? result << *indent : result << ",\n\"";
                    219:                                        result << "base64\":\"";
1.48      misha     220:                                        const char* encoded=pa_base64_encode(fvalue_ptr, fvalue_size);
1.55      moko      221:                                        result.append_help_length(encoded, 0, String::L_JSON);
1.49      moko      222:                                        result << "\"";
1.48      misha     223:                                        break;
                    224:                                }
                    225:                        case Json_options::F_TEXT:
                    226:                                {
1.49      moko      227:                                        indent ? result << *indent : result << ",\n\"";
                    228:                                        result << "text\":\"";
1.55      moko      229:                                        result.append_help_length(text_cstr(), 0, String::L_JSON);
1.49      moko      230:                                        result << "\"";
1.48      misha     231:                                        break;
                    232:                                }
                    233:                }
                    234:        }
                    235: 
1.52      moko      236:        result << "\n" << options.indent << "}";
1.49      moko      237:        return &result;
1.48      misha     238: }

E-mail: