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

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.59    ! moko       15: volatile const char * IDENT_PA_VFILE_C="$Id: pa_vfile.C,v 1.58 2013/06/24 22:00:18 moko 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.59    ! moko       43: inline bool content_type_is_default(Value *content_type){
        !            44:        if(content_type){
        !            45:                const String *ct=content_type->get_string();
        !            46:                return ct == &content_type_text || ct == &content_type_binary;
        !            47:        }
        !            48:        return true;
        !            49: }
        !            50: 
1.37      paf        51: // methods
                     52: 
                     53: VStateless_class *VFile::get_class() { return file_class; }
                     54: 
1.58      moko       55: 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        56:        fvalue_ptr=avalue_ptr;
                     57:        fvalue_size=avalue_size;
1.53      misha      58: 
1.43      misha      59:        ftext_tainted=atainted;
1.8       paf        60: 
1.10      paf        61:        ffields.clear();
1.43      misha      62: 
1.58      moko       63:        // $.mode
                     64:        set_mode(ais_text_mode);
                     65: 
1.53      misha      66:        // $.size
1.50      misha      67:        ffields.put(size_name, new VInt(fvalue_size));
                     68: 
1.53      misha      69:        // $.name
1.50      misha      70:        set_name(afile_name);
                     71: 
                     72:        // $mime-type
                     73:        set_content_type(acontent_type, afile_name, r);
1.54      moko       74: }
1.53      misha      75: 
1.54      moko       76: 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) {
                     77:        if(ais_text_mode && avalue_ptr && avalue_size) {
                     78:                fix_line_breaks(avalue_ptr, avalue_size);
                     79:        }
1.58      moko       80:        set_all(atainted, ais_text_mode, avalue_ptr, avalue_size, afile_name, acontent_type, r);
1.54      moko       81:        fis_text_content=ais_text_mode;
1.50      misha      82: }
                     83: 
1.54      moko       84: 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       85:        set_all(atainted, false, avalue_ptr, avalue_size, afile_name, acontent_type, r);
1.54      moko       86:        fis_text_content=false;
                     87: }
1.53      misha      88: 
1.58      moko       89: void VFile::set(VFile& avfile, bool aset_text_mode, bool ais_text_mode, const String* afile_name, Value* acontent_type, Request* r) {
                     90:        fvalue_ptr=avfile.fvalue_ptr;
                     91:        fvalue_size=avfile.fvalue_size;
                     92:        ftext_tainted=avfile.ftext_tainted;
                     93:        fis_text_content=avfile.fis_text_content;
1.53      misha      94: 
                     95:        ffields.clear();
1.50      misha      96:        for(HashStringValue::Iterator i(avfile.ffields); i; i.next())
1.53      misha      97:                if(i.key() != text_name) // do not copy cached .text value
1.58      moko       98:                        ffields.put(*new String(i.key(), String::L_TAINTED), i.value());
1.54      moko       99: 
1.58      moko      100:        if(aset_text_mode)
                    101:                set_mode(ais_text_mode);
                    102: 
                    103:        if(afile_name)
                    104:                set_name(afile_name);
                    105: 
1.59    ! moko      106:        if(acontent_type || afile_name || aset_text_mode && content_type_is_default(ffields.get(content_type_name)))
1.58      moko      107:                set_content_type(acontent_type, afile_name, r);
1.53      misha     108: }
                    109: 
                    110: const char* VFile::text_cstr() {
                    111:        const char* p=value_ptr();
1.54      moko      112:        if(fis_text_content)
1.53      misha     113:                return p;
                    114: 
                    115:        size_t size=fvalue_size;
                    116: 
                    117:        if(const char *premature_zero_pos=(const char *)memchr(p, 0, size))
                    118:                size=premature_zero_pos-p;
                    119: 
                    120:        char *copy_ptr=size?strdup(p, size):0;
1.54      moko      121:        // text mode but binary content
1.53      misha     122:        if(fis_text_mode && size)
                    123:                fix_line_breaks(copy_ptr, size);
                    124:        return copy_ptr;
1.50      misha     125: }
                    126: 
1.53      misha     127: void VFile::set_mode(bool ais_text_mode){
                    128:        fis_text_mode=ais_text_mode;
1.56      moko      129:        if(fvalue_ptr)
                    130:                ffields.put(mode_name, new VString(ais_text_mode? mode_value_text : mode_value_binary ));
1.50      misha     131: }
                    132: 
                    133: void VFile::set_name(const String* afile_name){
1.14      paf       134:        char *lfile_name;
1.57      misha     135:        if(afile_name && !afile_name->is_empty()) {
1.50      misha     136:                lfile_name=strdup(afile_name->taint_cstr(String::L_FILE_SPEC));
1.10      paf       137:                if(char *after_slash=rsplit(lfile_name, '\\'))
                    138:                        lfile_name=after_slash;
                    139:                if(char *after_slash=rsplit(lfile_name, '/'))
                    140:                        lfile_name=after_slash;
1.14      paf       141:        } else
1.54      moko      142:                lfile_name=(char *)NONAME_DAT;
1.50      misha     143: 
1.53      misha     144:        ffields.put(name_name, new VString(*new String(lfile_name, String::L_FILE_SPEC)));
1.50      misha     145: }
1.43      misha     146: 
1.50      misha     147: void VFile::set_content_type(Value* acontent_type, const String* afile_name, Request* r){
                    148:        if(!acontent_type && afile_name && r)
                    149:                acontent_type=new VString(r->mime_type_of(afile_name));
1.43      misha     150: 
1.58      moko      151:        if(!acontent_type)
                    152:                acontent_type=new VString(fis_text_mode ? content_type_text : content_type_binary);
                    153: 
                    154:        ffields.put(content_type_name, acontent_type);
1.43      misha     155: }
                    156: 
1.46      misha     157: void VFile::save(Request_charsets& charsets, const String& file_spec, bool is_text, Charset* asked_charset) {
                    158:        if(fvalue_ptr)
                    159:                file_write(charsets, file_spec, fvalue_ptr, fvalue_size, is_text, false/*do_append*/, asked_charset);
                    160:        else
1.58      moko      161:                throw Exception(PARSER_RUNTIME, &file_spec, "saving stat-ed file");
1.46      misha     162: }
                    163: 
1.50      misha     164: bool VFile::is_text_mode(const String& mode) {
                    165:        if(mode==mode_value_text)
                    166:                return true;
                    167:        if(mode==mode_value_binary)
                    168:                return false;
1.58      moko      169:        throw Exception(PARSER_RUNTIME, &mode, "is invalid mode, must be either '"MODE_VALUE_TEXT"' or '"MODE_VALUE_BINARY"'");
1.50      misha     170: }
1.58      moko      171: 
1.50      misha     172: bool VFile::is_valid_mode (const String& mode) {
                    173:        return (mode==mode_value_text || mode==mode_value_binary);
                    174: }
                    175: 
1.45      misha     176: Value* VFile::get_element(const String& aname) {
1.43      misha     177:        Value* result;
                    178: 
                    179:        // $method
1.45      misha     180:        if(result=VStateless_object::get_element(aname))
1.43      misha     181:                return result;
                    182: 
                    183:        // $field
                    184:        if(result=ffields.get(aname))
                    185:                return result;
                    186: 
                    187:        // $text - if not cached
1.53      misha     188:        if(aname == text_name && fvalue_ptr && fvalue_size) {
1.43      misha     189:                // assigned file have ptr and we really have some bytes
1.42      misha     190: 
1.53      misha     191:                result=new VString(*new String(text_cstr(), ftext_tainted ? String::L_TAINTED : String::L_AS_IS));
1.42      misha     192: 
1.53      misha     193:                // cache it
1.43      misha     194:                ffields.put(text_name, result);
1.35      paf       195: 
                    196:                return result;
1.43      misha     197:        }
1.35      paf       198: 
1.43      misha     199:        return 0;
1.3       paf       200: }
1.48      misha     201: 
1.52      moko      202: const String* VFile::get_json_string(Json_options& options){
1.49      moko      203:        String& result=*new String("{\n", String::L_AS_IS);
                    204:        
                    205:        String * indent=NULL;
                    206: 
1.52      moko      207:        if (options.indent){
                    208:                indent = new String(",\n\t", String::L_AS_IS); *indent << options.indent << "\"";
                    209:                result << "\t" << options.indent;
1.49      moko      210:        }
                    211: 
                    212:        result << "\"class\":\"file\"";
1.48      misha     213: 
                    214:        for(HashStringValue::Iterator i(ffields); i; i.next() ){
                    215:                String::Body key=i.key();
                    216:                if(key != text_name){
1.49      moko      217:                        indent ? result << *indent : result << ",\n\"";
                    218:                        result << String(key, String::L_JSON) << "\":" << *i.value()->get_json_string(options);
1.48      misha     219:                }
                    220:        }
                    221: 
                    222:        if(fvalue_ptr){
1.52      moko      223:                switch(options.file){
1.48      misha     224:                        case Json_options::F_BASE64:
                    225:                                {
1.49      moko      226:                                        indent ? result << *indent : result << ",\n\"";
                    227:                                        result << "base64\":\"";
1.48      misha     228:                                        const char* encoded=pa_base64_encode(fvalue_ptr, fvalue_size);
1.55      moko      229:                                        result.append_help_length(encoded, 0, String::L_JSON);
1.49      moko      230:                                        result << "\"";
1.48      misha     231:                                        break;
                    232:                                }
                    233:                        case Json_options::F_TEXT:
                    234:                                {
1.49      moko      235:                                        indent ? result << *indent : result << ",\n\"";
                    236:                                        result << "text\":\"";
1.55      moko      237:                                        result.append_help_length(text_cstr(), 0, String::L_JSON);
1.49      moko      238:                                        result << "\"";
1.48      misha     239:                                        break;
                    240:                                }
                    241:                }
                    242:        }
                    243: 
1.52      moko      244:        result << "\n" << options.indent << "}";
1.49      moko      245:        return &result;
1.48      misha     246: }

E-mail: