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