Annotation of parser3/src/classes/op.C, revision 1.206
1.1 paf 1: /** @file
1.9 paf 2: Parser: parser @b operators.
1.1 paf 3:
1.184 misha 4: Copyright (c) 2001-2009 ArtLebedev Group (http://www.artlebedev.com)
1.71 paf 5: Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.98 paf 6: */
1.1 paf 7:
1.206 ! moko 8: static const char * const IDENT_OP_C="$Date: 2010-11-15 23:31:08 $";
1.1 paf 9:
1.13 paf 10: #include "classes.h"
1.129 paf 11: #include "pa_vmethod_frame.h"
12:
1.1 paf 13: #include "pa_common.h"
1.134 paf 14: #include "pa_os.h"
1.1 paf 15: #include "pa_request.h"
16: #include "pa_vint.h"
17: #include "pa_sql_connection.h"
1.79 paf 18: #include "pa_vdate.h"
1.105 paf 19: #include "pa_vmethod_frame.h"
1.129 paf 20: #include "pa_vclass.h"
1.144 paf 21: #include "pa_charset.h"
1.1 paf 22:
1.18 parser 23: // limits
24:
1.162 paf 25: #define MAX_LOOPS 20000
1.18 parser 26:
1.10 paf 27: // defines
28:
1.82 paf 29: #define CASE_DEFAULT_VALUE "DEFAULT"
1.146 paf 30: #define PROCESS_MAIN_OPTION_NAME "main"
31: #define PROCESS_FILE_OPTION_NAME "file"
32: #define PROCESS_LINENO_OPTION_NAME "lineno"
1.11 paf 33:
1.10 paf 34: // class
35:
1.113 paf 36: class VClassMAIN: public VClass {
1.10 paf 37: public:
1.129 paf 38: VClassMAIN();
1.10 paf 39: };
40:
1.129 paf 41: // defines for statics
42:
43: #define SWITCH_DATA_NAME "SWITCH-DATA"
44: #define CACHE_DATA_NAME "CACHE-DATA"
1.164 paf 45:
1.129 paf 46: #define EXCEPTION_VAR_NAME "exception"
47:
48: // statics
49:
50: //^switch ^case
51: static const String switch_data_name(SWITCH_DATA_NAME);
52: //^cache
53: static const String cache_data_name(CACHE_DATA_NAME);
54:
55: static const String exception_var_name(EXCEPTION_VAR_NAME);
56:
1.164 paf 57:
1.136 paf 58: // local defines
59:
60: #define CACHE_EXCEPTION_HANDLED_CACHE_NAME "cache"
61:
1.129 paf 62: // helpers
63:
1.187 misha 64: class Untaint_lang_name2enum: public HashString<String::Language> {
1.129 paf 65: public:
66: Untaint_lang_name2enum() {
67: #define ULN(name, LANG) \
1.130 paf 68: put(String::Body(name), (value_type)(String::L_##LANG));
1.198 misha 69: ULN("clean", CLEAN);
1.129 paf 70: ULN("as-is", AS_IS);
71: ULN("optimized-as-is", AS_IS|String::L_OPTIMIZE_BIT);
72: ULN("file-spec", FILE_SPEC);
73: ULN("http-header", HTTP_HEADER);
74: ULN("mail-header", MAIL_HEADER);
75: ULN("uri", URI);
76: ULN("sql", SQL);
77: ULN("js", JS);
78: ULN("xml", XML);
79: ULN("optimized-xml", XML|String::L_OPTIMIZE_BIT);
80: ULN("html", HTML);
81: ULN("optimized-html", HTML|String::L_OPTIMIZE_BIT);
1.159 paf 82: ULN("regex", REGEX);
1.193 misha 83: ULN("parser-code", PARSER_CODE);
1.204 misha 84: ULN("json", JSON);
1.129 paf 85: #undef ULN
86: }
87: } untaint_lang_name2enum;
88:
1.10 paf 89: // methods
90:
1.129 paf 91: static void _if(Request& r, MethodParams& params) {
1.200 moko 92: size_t max_param=params.count()-1;
93: size_t i=0;
94: do {
95: bool condition=params.as_bool(i, "condition must be expression", r);
96: if(condition) {
97: r.process_write(*params.get(i+1));
98: return;
99: }
100: i+=2;
101: } while (i < max_param);
102:
103: if(i == max_param)
104: r.process_write(*params.get(i));
1.1 paf 105: }
106:
1.175 misha 107: static String::Language get_untaint_lang(MethodParams& params, int index){
108: const String& lang_name=params.as_string(index, "lang must be string");
109: String::Language lang=untaint_lang_name2enum.get(lang_name);
110: if(!lang)
1.183 misha 111: throw Exception(PARSER_RUNTIME,
1.175 misha 112: &lang_name,
113: "invalid taint language");
114: return lang;
115: }
116:
1.129 paf 117: static void _untaint(Request& r, MethodParams& params) {
118: String::Language lang;
119: if(params.count()==1)
1.175 misha 120: lang=String::L_AS_IS; // mark as simply 'as-is'. useful in html from sql
121: else
122: lang=get_untaint_lang(params, 0);
1.1 paf 123:
124: {
1.129 paf 125: Value& vbody=params.as_junction(params.count()-1, "body must be code");
1.1 paf 126:
1.191 misha 127: Temp_lang temp_lang(r, lang); // set temporarily specified ^untaint[language;
128: StringOrValue result=r.process(vbody); // process marking tainted with that lang
1.188 misha 129: r.write_assign_lang(result);
1.1 paf 130: }
131: }
132:
1.129 paf 133: static void _taint(Request& r, MethodParams& params) {
134: String::Language lang;
135: if(params.count()==1)
1.175 misha 136: lang=String::L_TAINTED; // mark as simply 'tainted'. useful in table:create
137: else
138: lang=get_untaint_lang(params, 0);
1.1 paf 139:
140: {
1.129 paf 141: Value& vbody=params.as_no_junction(params.count()-1, "body must not be code");
1.1 paf 142:
1.191 misha 143: String result(vbody.as_string(), lang); // force result language to specified
1.188 misha 144: r.write_assign_lang(result);
1.1 paf 145: }
146: }
147:
1.205 moko 148: static void _apply_taint(Request& r, MethodParams& params) {
149: String::Language lang=params.count()==1 ? String::L_AS_IS : get_untaint_lang(params, 0);
150: const String &sbody=params.as_string(params.count()-1, "body must be string");
151: String::Body result_body=sbody.cstr_to_string_body_untaint(lang, 0, &r.charsets);
152: r.write_pass_lang(*new String(result_body, String::L_AS_IS));
153: }
154:
1.129 paf 155: static void _process(Request& r, MethodParams& params) {
156: Method* main_method;
157:
158: size_t index=0;
159: Value* target_self;
160: Value& maybe_target_self=params[index];
161: if(maybe_target_self.get_string() || maybe_target_self.get_junction())
162: target_self=&r.get_method_frame()->caller()->self();
163: else {
164: target_self=&maybe_target_self; index++;
165: }
166:
1.1 paf 167: {
1.197 misha 168: VStateless_class *target_class=target_self->get_class();
1.116 paf 169: if(!target_class)
1.166 misha 170: throw Exception(PARSER_RUNTIME,
1.129 paf 171: 0,
1.116 paf 172: "no target class");
1.114 paf 173:
1.73 paf 174: // temporary remove language change
1.192 misha 175: Temp_lang temp_lang(r, String::L_PARSER_CODE);
1.1 paf 176: // temporary zero @main so to maybe-replace it in processed code
1.129 paf 177: Temp_method temp_method_main(*target_class, main_method_name, 0);
1.1 paf 178: // temporary zero @auto so it wouldn't be auto-called in Request::use_buf
1.129 paf 179: Temp_method temp_method_auto(*target_class, auto_method_name, 0);
180:
1.146 paf 181: size_t options_index=index+1;
182: HashStringValue* options=0;
183: if(options_index<params.count()) {
1.195 misha 184: Value& voptions=params.as_no_junction(options_index, OPTIONS_MUST_NOT_BE_CODE);
1.146 paf 185: options=voptions.get_hash();
186: if(!options)
1.201 misha 187: throw Exception(PARSER_RUNTIME, 0, OPTIONS_MUST_BE_HASH);
1.146 paf 188: }
189:
1.129 paf 190: const String* main_alias=0;
1.146 paf 191: const String* file_alias=0;
192: int line_no_alias_offset=0;
193: if(options) {
194: int valid_options=0;
195: if(Value* vmain_alias=options->get(PROCESS_MAIN_OPTION_NAME)) {
196: valid_options++;
197: main_alias=&vmain_alias->as_string();
198: }
199: if(Value* vfile_alias=options->get(PROCESS_FILE_OPTION_NAME)) {
200: valid_options++;
201: file_alias=&vfile_alias->as_string();
202: }
203: if(Value* vline_no_alias_offset=options->get(PROCESS_LINENO_OPTION_NAME)) {
204: valid_options++;
205: line_no_alias_offset=vline_no_alias_offset->as_int();
206: }
207:
208: if(valid_options!=options->count())
1.201 misha 209: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
1.146 paf 210: }
1.129 paf 211:
1.146 paf 212: uint processe_file_no=file_alias?
213: r.register_file(r.absolute(*file_alias))
214: : pseudo_file_no__process;
215: // process...{string}
216: Value& vjunction=params.as_junction(index, "body must be code");
217: // evaluate source to process
218: const String& source=r.process_to_string(vjunction);
219: r.use_buf(*target_class,
1.190 misha 220: source.untaint_cstr(String::L_AS_IS, r.connection(false)),
1.146 paf 221: main_alias,
222: processe_file_no,
223: line_no_alias_offset);
1.1 paf 224:
1.73 paf 225: // main_method
1.129 paf 226: main_method=target_class->get_method(main_method_name);
1.73 paf 227: }
228: // after restoring current-request-lang
229: // maybe-execute @main[]
230: if(main_method) {
1.202 moko 231: VMethodFrame frame(*main_method, r.get_method_frame()->caller(), *target_self);
232: frame.empty_params();
1.203 moko 233: r.op_call(frame);
234: r.write_pass_lang(frame.result());
1.1 paf 235: }
236: }
237:
1.138 paf 238: static void _rem(Request&, MethodParams& params) {
1.129 paf 239: params.as_junction(0, "body must be code");
1.1 paf 240: }
241:
1.129 paf 242: static void _while(Request& r, MethodParams& params) {
1.196 misha 243: InCycle temp(r);
1.164 paf 244:
1.170 misha 245: Value& vcondition=params.as_expression(0, "condition must be number, bool or expression");
246:
1.162 paf 247: Value& body_code=params.as_junction(1, "body must be code");
248: Value* delim_maybe_code=params.count()>2?¶ms[2]:0;
1.1 paf 249:
250: // while...
251: int endless_loop_count=0;
1.186 misha 252: if(delim_maybe_code){ // delimiter set
1.185 misha 253: bool need_delim=false;
254: while(true) {
255: if(++endless_loop_count>=MAX_LOOPS) // endless loop?
256: throw Exception(PARSER_RUNTIME,
257: 0,
258: "endless loop detected");
259:
260: if(!r.process_to_value(vcondition, false/*don't intercept string*/).as_bool())
261: break;
1.1 paf 262:
1.185 misha 263: StringOrValue sv_processed=r.process(body_code);
264: Request::Skip lskip=r.get_skip(); r.set_skip(Request::SKIP_NOTHING);
1.186 misha 265:
1.185 misha 266: const String* s_processed=sv_processed.get_string();
1.186 misha 267: if(s_processed && !s_processed->is_empty()) { // we have body
1.185 misha 268: if(need_delim) // need delim & iteration produced string?
269: r.write_pass_lang(r.process(*delim_maybe_code));
270: else
271: need_delim=true;
272: }
273: r.write_pass_lang(sv_processed);
1.1 paf 274:
1.185 misha 275: if(lskip==Request::SKIP_BREAK)
276: break;
1.162 paf 277: }
1.185 misha 278: } else {
279: while(true) {
280: if(++endless_loop_count>=MAX_LOOPS) // endless loop?
281: throw Exception(PARSER_RUNTIME,
282: 0,
283: "endless loop detected");
284:
285: if(!r.process_to_value(vcondition, false/*don't intercept string*/).as_bool())
286: break;
1.164 paf 287:
1.185 misha 288: r.process_write(body_code);
289: Request::Skip lskip=r.get_skip(); r.set_skip(Request::SKIP_NOTHING);
290:
291: if(lskip==Request::SKIP_BREAK)
292: break;
293: }
1.1 paf 294: }
295: }
296:
1.129 paf 297: static void _use(Request& r, MethodParams& params) {
1.171 misha 298: Value& vfile=params.as_no_junction(0, FILE_NAME_MUST_NOT_BE_CODE);
1.199 misha 299:
300: // _use could be called from the parser3 method only, so caller is always defined
1.202 moko 301: r.use_file(r.main_class, vfile.as_string(), r.get_method_filename(&r.get_method_frame()->caller()->method));
1.1 paf 302: }
303:
1.164 paf 304: static void set_skip(Request& r, Request::Skip askip) {
1.196 misha 305: if(!r.get_in_cycle())
1.166 misha 306: throw Exception(PARSER_RUNTIME,
1.164 paf 307: 0,
308: "without cycle");
309:
310: r.set_skip(askip);
311: }
312:
313: static void _break(Request& r, MethodParams&) {
314: set_skip(r, Request::SKIP_BREAK);
315: }
316:
317: static void _continue(Request& r, MethodParams&) {
318: set_skip(r, Request::SKIP_CONTINUE);
319: }
320:
1.129 paf 321: static void _for(Request& r, MethodParams& params) {
1.196 misha 322: InCycle temp(r);
1.164 paf 323:
1.129 paf 324: const String& var_name=params.as_string(0, "var name must be string");
325: int from=params.as_int(1, "from must be int", r);
326: int to=params.as_int(2, "to must be int", r);
1.186 misha 327: Value& body_code=params.as_junction(3, "body must be code");
1.129 paf 328: Value* delim_maybe_code=params.count()>4?¶ms[4]:0;
1.1 paf 329:
1.57 paf 330: if(to-from>=MAX_LOOPS) // too long loop?
1.166 misha 331: throw Exception(PARSER_RUNTIME,
1.129 paf 332: 0,
1.57 paf 333: "endless loop detected");
334:
1.129 paf 335: VInt* vint=new VInt(0);
1.154 paf 336:
337: VMethodFrame& caller=*r.get_method_frame()->caller();
1.197 misha 338: caller.put_element(var_name, vint, false);
1.186 misha 339: if(delim_maybe_code){ // delimiter set
1.185 misha 340: bool need_delim=false;
341:
342: for(int i=from; i<=to; i++) {
343: vint->set_int(i);
344:
345: StringOrValue sv_processed=r.process(body_code);
346: Request::Skip lskip=r.get_skip(); r.set_skip(Request::SKIP_NOTHING);
1.186 misha 347:
1.185 misha 348: const String* s_processed=sv_processed.get_string();
1.186 misha 349: if(s_processed && !s_processed->is_empty()) { // we have body
1.185 misha 350: if(need_delim) // need delim & iteration produced string?
351: r.write_pass_lang(r.process(*delim_maybe_code));
352: else
353: need_delim=true;
354: }
355: r.write_pass_lang(sv_processed);
1.1 paf 356:
1.185 misha 357: if(lskip==Request::SKIP_BREAK)
358: break;
1.1 paf 359: }
1.185 misha 360: } else {
361: for(int i=from; i<=to; i++) {
362: vint->set_int(i);
363:
364: r.process_write(body_code);
365: Request::Skip lskip=r.get_skip(); r.set_skip(Request::SKIP_NOTHING);
1.164 paf 366:
1.185 misha 367: if(lskip==Request::SKIP_BREAK)
368: break;
369: }
1.1 paf 370: }
371: }
372:
1.129 paf 373: static void _eval(Request& r, MethodParams& params) {
374: Value& expr=params.as_junction(0, "need expression");
1.1 paf 375: // evaluate expresion
1.129 paf 376: Value& value_result=r.process_to_value(expr,
1.165 misha 377: false/*don't intercept string*/).as_expr_result();
1.129 paf 378: if(params.count()>1) {
1.177 misha 379: const String& fmt=params.as_string(1, "fmt must be string").trim();
380: if(fmt.is_empty()){
381: r.write_no_lang(value_result);
382: } else {
383: r.write_no_lang(String(format(value_result.as_double(), fmt.cstrm())));
384: }
1.91 paf 385: } else
1.129 paf 386: r.write_no_lang(value_result);
1.1 paf 387: }
388:
1.129 paf 389: static void _connect(Request& r, MethodParams& params) {
1.63 paf 390: #ifdef RESOURCES_DEBUG
391: struct timeval mt[2];
392: #endif
1.129 paf 393: Value& url=params.as_no_junction(0, "url must not be code");
394: Value& body_code=params.as_junction(1, "body must be code");
1.1 paf 395:
1.129 paf 396: Table* protocol2driver_and_client=0;
1.197 misha 397: if(Value* sql=r.main_class.get_element(String(MAIN_SQL_NAME))) {
398: if(Value* element=sql->get_element(String(MAIN_SQL_DRIVERS_NAME))) {
1.113 paf 399: protocol2driver_and_client=element->get_table();
400: }
401: }
1.11 paf 402:
1.63 paf 403: #ifdef RESOURCES_DEBUG
404: //measure:before
405: gettimeofday(&mt[0],NULL);
406: #endif
1.1 paf 407: // connect
1.142 paf 408: SQL_Connection* connection=SQL_driver_manager->get_connection(url.as_string(),
1.144 paf 409: protocol2driver_and_client,
1.180 misha 410: r.charsets.source().NAME().cstr(),
411: r.request_info.document_root);
1.1 paf 412:
1.63 paf 413: #ifdef RESOURCES_DEBUG
414: //measure:after connect
415: gettimeofday(&mt[1],NULL);
416:
417: double t[2];
418: for(int i=0;i<2;i++)
419: t[i]=mt[i].tv_sec+mt[i].tv_usec/1000000.0;
420:
421: r.sql_connect_time+=t[1]-t[0];
422: #endif
1.129 paf 423: Temp_connection temp_connection(r, connection);
1.1 paf 424: // execute body
1.53 parser 425: try {
1.189 misha 426: r.process_write(body_code);
1.129 paf 427: connection->commit();
428: connection->close();
1.75 paf 429: } catch(...) { // process problem
1.129 paf 430: connection->rollback();
431: connection->close();
432: rethrow;
1.1 paf 433: }
434: }
435:
1.41 parser 436: #ifndef DOXYGEN
1.129 paf 437: class Switch_data: public PA_Object {
438: public:
439: Request& r;
1.181 misha 440: const String* searching_string;
441: double searching_double;
1.129 paf 442: Value* found;
443: Value* _default;
444: public:
445: Switch_data(Request& ar, Value& asearching):
1.181 misha 446: r(ar)
447: {
448: if(asearching.is_string() || asearching.is_void()){
449: searching_string=&asearching.as_string();
450: searching_double=0;
451: } else {
452: searching_string=0;
453: searching_double=asearching.as_double();
454: }
455: }
1.28 parser 456: };
1.41 parser 457: #endif
1.129 paf 458: static void _switch(Request& r, MethodParams& params) {
459: Switch_data* data=new Switch_data(r, r.process_to_value(params[0]));
1.135 paf 460: Temp_hash_value<const String::Body, void*>
1.129 paf 461: switch_data_setter(r.classes_conf, switch_data_name, data);
1.28 parser 462:
1.129 paf 463: Value& cases_code=params.as_junction(1, "switch cases must be code");
1.82 paf 464: // execution of found ^case[...]{code} must be in context of ^switch[...]{code}
465: // because of stacked WWrapper used there as wcontext
1.84 paf 466: r.process(cases_code, true/*intercept_string*/);
1.147 paf 467: if(Value* selected_code=data->found? data->found: data->_default)
1.107 paf 468: r.write_pass_lang(r.process(*selected_code));
1.28 parser 469: }
470:
1.129 paf 471: static void _case(Request& r, MethodParams& params) {
472: Switch_data* data=static_cast<Switch_data*>(r.classes_conf.get(switch_data_name));
1.38 parser 473: if(!data)
1.166 misha 474: throw Exception(PARSER_RUNTIME,
1.129 paf 475: 0,
1.38 parser 476: "without switch");
1.28 parser 477:
1.181 misha 478: if(data->found) // matches already was found
479: return;
480:
1.129 paf 481: int count=params.count();
1.184 misha 482: Value* code=¶ms.as_expression(--count, "case result must be code");
483:
484: #ifdef USE_DESTRUCTORS
485: Junction *j=code->get_junction();
486: if (j){
487: code=new VJunction(j->self,j->method,j->method_frame,j->rcontext,j->wcontext,j->code);
488: if (j->wcontext) j->wcontext->attach_junction((VJunction *)code);
489: }
490: #endif
1.83 paf 491:
1.181 misha 492: for(int i=0; i<count; i++){
1.129 paf 493: Value& value=r.process_to_value(params[i]);
1.28 parser 494:
1.181 misha 495: if(value.is_string() && value.as_string() == CASE_DEFAULT_VALUE){
1.184 misha 496: data->_default=code;
1.181 misha 497: continue;
1.28 parser 498: }
499:
500: bool matches;
1.181 misha 501: if(data->searching_string)
502: matches=(*data->searching_string) == value.as_string();
1.28 parser 503: else
1.181 misha 504: matches=data->searching_double == value.as_double();
1.82 paf 505:
1.181 misha 506: if(matches){
1.184 misha 507: data->found=code;
1.28 parser 508: break;
509: }
510: }
511: }
1.136 paf 512: #ifndef DOXYGEN
513: struct Try_catch_result {
514: StringOrValue processed_code;
515: const String* exception_should_be_handled;
516:
517: Try_catch_result(): exception_should_be_handled(0) {}
518: };
519: #endif
520:
521: /// used by ^try and ^cache, @returns $exception.handled[string] if any
522: template<class I>
523: static Try_catch_result try_catch(Request& r,
524: StringOrValue body_code(Request&, I), I info,
1.179 misha 525: Value* catch_code,
526: bool could_be_handled_by_caller=false)
1.136 paf 527: {
528: Try_catch_result result;
1.179 misha 529:
1.136 paf 530: if(!catch_code) {
531: result.processed_code=body_code(r, info);
532: return result;
533: }
534:
535: // taking snapshot of try-context
536: Request_context_saver try_context(r);
537: try {
538: result.processed_code=body_code(r, info);
539: } catch(const Exception& e) {
540: Request_context_saver throw_context(r); // taking snapshot of throw-context [stack trace contains error]
541: Request::Exception_details details=r.get_details(e);
542: try_context.restore(); // restoring try-context to perform catch-code
543:
544: Junction* junction=catch_code->get_junction();
545: Value* method_frame=junction->method_frame;
1.197 misha 546: Value* saved_exception_var_value=method_frame->get_element(exception_var_name);
1.154 paf 547: VMethodFrame& frame=*junction->method_frame;
1.197 misha 548: frame.put_element(exception_var_name, &details.vhash, false);
1.179 misha 549:
1.136 paf 550: result.processed_code=r.process(*catch_code);
551:
552: // retriving $exception.handled, restoring $exception var
553: Value* vhandled=details.vhash.hash().get(exception_handled_part_name);
1.197 misha 554: frame.put_element(exception_var_name, saved_exception_var_value, false);
1.136 paf 555:
556: bool bhandled=false;
557: if(vhandled) {
558: if(vhandled->is_string()) { // not simple $exception.handled(1/0)?
1.160 paf 559: if(could_be_handled_by_caller) { // and we can possibly handle it
560: result.exception_should_be_handled=vhandled->get_string(); // considering 'recovered' and let the caller recover
561: return result;
562: }
563:
564: bhandled=false;
565: } else
1.179 misha 566: bhandled=vhandled->as_bool();
1.136 paf 567: }
568:
569: if(!bhandled) {
570: throw_context.restore(); // restoring throw-context [exception were not handled]
571: rethrow;
572: }
573: }
1.179 misha 574:
1.136 paf 575: return result;
576: }
1.28 parser 577:
1.63 paf 578: // cache--
579:
580: // consts
581:
1.152 paf 582: const int DATA_STRING_SERIALIZED_VERSION=0x0006;
1.63 paf 583:
584: // helper types
585:
586: #ifndef DOXYGEN
587: struct Data_string_serialized_prolog {
588: int version;
1.79 paf 589: time_t expires;
1.63 paf 590: };
591: #endif
592:
1.68 paf 593: void cache_delete(const String& file_spec) {
1.167 misha 594: file_delete(file_spec, false/*fail_on_problem*/);
1.63 paf 595: }
1.69 paf 596:
597: #ifndef DOXYGEN
1.135 paf 598: struct Cache_scope {
1.129 paf 599: public:
1.79 paf 600: time_t expires;
1.135 paf 601: const String* body_from_disk;
1.79 paf 602: };
1.69 paf 603: struct Locked_process_and_cache_put_action_info {
604: Request *r;
1.136 paf 605: Cache_scope *scope;
1.167 misha 606: Value* body_code;
607: Value* catch_code;
1.136 paf 608: const String* processed_code;
1.69 paf 609: };
610: #endif
1.136 paf 611:
612:
613:
614: static StringOrValue process_cache_body_code(Request& r, Value* body_code) {
1.153 paf 615: return StringOrValue(r.process_to_string(*body_code));
1.136 paf 616: }
617:
1.129 paf 618: /* @todo maybe network order worth spending some effort?
619: don't bothering myself with network byte order,
620: am not planning to be able to move resulting file across platforms
621: */
1.69 paf 622: static void locked_process_and_cache_put_action(int f, void *context) {
623: Locked_process_and_cache_put_action_info& info=
624: *static_cast<Locked_process_and_cache_put_action_info *>(context);
1.129 paf 625:
1.160 paf 626: const String* body_from_disk=info.scope->body_from_disk;
1.69 paf 627: // body->process
1.136 paf 628: Try_catch_result result=try_catch(*info.r,
629: process_cache_body_code, info.body_code,
1.160 paf 630: info.catch_code, body_from_disk!=0 /*we have something old=we can handle=recover later*/);
1.136 paf 631:
632: if(result.exception_should_be_handled) {
633: if(*result.exception_should_be_handled==CACHE_EXCEPTION_HANDLED_CACHE_NAME) {
1.160 paf 634: assert(body_from_disk);
635: info.processed_code=body_from_disk;
1.136 paf 636: } else
1.166 misha 637: throw Exception(PARSER_RUNTIME,
1.136 paf 638: result.exception_should_be_handled,
639: "$"EXCEPTION_VAR_NAME"."EXCEPTION_HANDLED_PART_NAME" value must be "
640: "either boolean or string '"CACHE_EXCEPTION_HANDLED_CACHE_NAME"'");
641: } else
642: info.processed_code=&result.processed_code.as_string();
1.69 paf 643:
1.79 paf 644: // expiration time not spoiled by ^cache(0) or something?
1.136 paf 645: if(info.scope->expires > time(0)) {
1.79 paf 646: // string -serialize> buffer
1.136 paf 647: String::Cm serialized=info.processed_code->serialize(
1.129 paf 648: sizeof(Data_string_serialized_prolog));
1.79 paf 649: Data_string_serialized_prolog& prolog=
1.129 paf 650: *reinterpret_cast<Data_string_serialized_prolog *>(serialized.str);
1.79 paf 651: prolog.version=DATA_STRING_SERIALIZED_VERSION;
1.136 paf 652: prolog.expires=info.scope->expires;
1.79 paf 653:
654: // buffer -write> file
1.129 paf 655: write(f, serialized.str, serialized.length);
1.79 paf 656: } else // expired!
1.136 paf 657: info.scope->expires=0; // flag it so that could be easily checked by caller
1.69 paf 658: }
1.129 paf 659: const String* locked_process_and_cache_put(Request& r,
1.128 paf 660: Value& body_code,
1.136 paf 661: Value* catch_code,
662: Cache_scope& scope,
663: const String& file_spec)
664: {
1.140 paf 665: Locked_process_and_cache_put_action_info info={&r, &scope, &body_code, catch_code, 0};
1.69 paf 666:
1.129 paf 667: const String* result=file_write_action_under_lock(
1.69 paf 668: file_spec,
1.168 misha 669: "cache_put",
670: locked_process_and_cache_put_action,
671: &info,
1.69 paf 672: false/*as_text*/,
673: false/*do_append*/,
1.173 misha 674: false/*block == don't wait till other thread release lock*/,
675: false/*dun throw exception if lock failed*/) ? info.processed_code: 0;
1.168 misha 676:
1.100 paf 677: time_t now=time(0);
1.136 paf 678: if(scope.expires<=now)
1.79 paf 679: cache_delete(file_spec);
680: return result;
1.63 paf 681: }
1.137 paf 682: #ifndef DOXYGEN
1.135 paf 683: struct Cache_get_result {
684: const String* body;
685: bool expired;
1.137 paf 686: };
687: #endif
688: static Cache_get_result cache_get(Request_charsets& charsets, const String& file_spec, time_t now) {
1.140 paf 689: Cache_get_result result={0, false};
1.135 paf 690:
1.129 paf 691: File_read_result file=file_read(charsets, file_spec,
1.63 paf 692: false/*as_text*/,
1.129 paf 693: 0, //no params
694: false/*fail_on_read_problem*/);
695: if(file.success && file.length/* ignore reads which are empty due to
1.72 paf 696: non-unary open+lockEX conflict with lockSH */) {
1.129 paf 697:
1.72 paf 698: Data_string_serialized_prolog& prolog=
1.129 paf 699: *reinterpret_cast<Data_string_serialized_prolog *>(file.str);
1.63 paf 700:
1.135 paf 701: String* body=new String;
1.72 paf 702: if(
1.129 paf 703: file.length>=sizeof(Data_string_serialized_prolog)
1.135 paf 704: && prolog.version==DATA_STRING_SERIALIZED_VERSION) {
705: if(body->deserialize(sizeof(Data_string_serialized_prolog), file.str, file.length)) {
706: result.body=body;
707: result.expired=prolog.expires <= now;
708: }
709: }
1.72 paf 710: }
1.63 paf 711:
1.135 paf 712: return result;
1.63 paf 713: }
1.129 paf 714: static time_t as_expires(Request& r, MethodParams& params,
1.79 paf 715: int index, time_t now) {
716: time_t result;
1.197 misha 717: if(Value* vdate=params[index].as(VDATE_TYPE))
1.129 paf 718: result=static_cast<VDate*>(vdate)->get_time();
1.79 paf 719: else
1.129 paf 720: result=now+(time_t)params.as_double(index, "lifespan must be date or number", r);
1.79 paf 721:
722: return result;
723: }
1.129 paf 724: static const String& as_file_spec(Request& r, MethodParams& params, int index) {
725: return r.absolute(params.as_string(index, "filespec must be string"));
1.79 paf 726: }
1.129 paf 727: static void _cache(Request& r, MethodParams& params) {
1.173 misha 728: if(params.count()==0) {
729: // ^cache[] -- return current expiration time
1.158 paf 730: Cache_scope* scope=static_cast<Cache_scope*>(r.classes_conf.get(cache_data_name));
731: if(!scope)
1.166 misha 732: throw Exception(PARSER_RUNTIME,
1.158 paf 733: 0,
734: "expire-time get without cache");
735: r.write_no_lang(*new VDate(scope->expires));
736: return;
737: }
738:
1.79 paf 739: time_t now=time(0);
740:
1.129 paf 741: if(params.count()==1) {
1.173 misha 742: // ^cache[filename] ^cache(seconds) ^cache[expires date]
1.129 paf 743: if(params[0].is_string()) { // filename?
1.79 paf 744: cache_delete(as_file_spec(r, params, 0));
745: return;
746: }
747:
748: // secods|expires date
1.135 paf 749: Cache_scope* scope=static_cast<Cache_scope*>(r.classes_conf.get(cache_data_name));
750: if(!scope)
1.166 misha 751: throw Exception(PARSER_RUNTIME,
1.129 paf 752: 0,
1.79 paf 753: "expire-time reducing instruction without cache");
754:
1.129 paf 755: time_t expires=as_expires(r, params, 0, now);
1.135 paf 756: if(expires < scope->expires)
757: scope->expires=expires;
1.79 paf 758:
759: return;
1.149 paf 760: } else if(params.count()<3)
1.166 misha 761: throw Exception(PARSER_RUNTIME,
1.149 paf 762: 0,
763: "invalid number of parameters");
1.63 paf 764:
765: // file_spec, expires, body code
1.172 misha 766: const String& file_spec=as_file_spec(r, params, 0);
1.65 paf 767:
1.141 paf 768: Cache_scope scope={as_expires(r, params, 1, now), 0};
1.135 paf 769:
770: Temp_hash_value<const String::Body, void*>
771: cache_scope_setter(r.classes_conf, cache_data_name, &scope);
1.136 paf 772: Value& body_code=params.as_junction(2, "body_code must be code");
773: Value* catch_code=0;
774: if(params.count()>3)
775: catch_code=¶ms.as_junction(3, "catch_code must be code");
1.63 paf 776:
1.168 misha 777: if(scope.expires>now) {
778: Cache_get_result cached=cache_get(r.charsets, file_spec, now);
1.69 paf 779:
1.168 misha 780: if(cached.body) { // have cached copy
1.174 misha 781: if(cached.expired) {
782: scope.body_from_disk=cached.body; // storing for user to retrive it with ^cache[]
783: } else {
784: // and it's not expired yet write it out
1.168 misha 785: r.write_assign_lang(*cached.body);
786: // happy with it
787: return;
788: }
789: }
790:
791: // no cached info or it's already expired
1.173 misha 792:
793: // trying to process it under lock and store result in file
1.174 misha 794: const String* processed_body=locked_process_and_cache_put(r, body_code, catch_code, scope, file_spec);
1.173 misha 795: if(processed_body){
1.174 misha 796: // write it out
797: r.write_assign_lang(*processed_body);
798: // happy with it
799: return;
1.173 misha 800: } else {
801: // we fail while get exclusive lock. nvm, we just execute body_code a bit later
1.174 misha 802: }
1.69 paf 803: } else {
1.79 paf 804: // instructed not to cache; forget cached copy
1.68 paf 805: cache_delete(file_spec);
1.69 paf 806: }
1.168 misha 807:
808: // process without cacheing
809: const String& processed_body=r.process_to_string(body_code);
810: // write it out
811: r.write_assign_lang(processed_body);
1.63 paf 812: }
813:
1.136 paf 814: static StringOrValue process_try_body_code(Request& r, Value* body_code) {
815: return r.process(*body_code);
816: }
1.129 paf 817: static void _try_operator(Request& r, MethodParams& params) {
818: Value& body_code=params.as_junction(0, "body_code must be code");
819: Value& catch_code=params.as_junction(1, "catch_code must be code");
1.179 misha 820: Value* finally_code=(params.count()==3) ? ¶ms.as_junction(2, "finally_code must be code") : 0;
821:
822: Try_catch_result result;
823: StringOrValue finally_result;
824: try{
825: result=try_catch(r,
826: process_try_body_code, &body_code,
827: &catch_code);
828: if(result.exception_should_be_handled)
829: throw Exception(PARSER_RUNTIME,
830: result.exception_should_be_handled,
831: "catch block must set $exception.handled to some boolean value, not string");
832: } catch(...){
833: if(finally_code)
834: finally_result=r.process(*finally_code);
835: rethrow;
836: }
1.74 paf 837:
1.179 misha 838: if(finally_code)
839: finally_result=r.process(*finally_code);
1.123 paf 840:
1.136 paf 841: // write out processed body_code or catch_code
842: r.write_pass_lang(result.processed_code);
1.179 misha 843:
844: // write out processed finally code
845: if(finally_code)
846: r.write_pass_lang(finally_result);
1.74 paf 847: }
848:
1.138 paf 849: static void _throw_operator(Request&, MethodParams& params) {
1.178 misha 850: if(params.count()==1 && !params[0].is_string()) {
1.129 paf 851: if(HashStringValue *hash=params[0].get_hash()) {
852: const char* type=0;
853: if(Value* value=hash->get(exception_type_part_name))
1.78 paf 854: type=value->as_string().cstr();
1.129 paf 855: const String* source=0;
856: if(Value* value=hash->get(exception_source_part_name))
1.74 paf 857: source=&value->as_string();
1.138 paf 858: const char* comment=0;
1.129 paf 859: if(Value* value=hash->get(exception_comment_part_name))
1.74 paf 860: comment=value->as_string().cstr();
861:
1.78 paf 862: throw Exception(type,
1.129 paf 863: source?source:0,
864: "%s", comment?comment:"");
1.74 paf 865: } else
1.166 misha 866: throw Exception(PARSER_RUNTIME,
1.129 paf 867: 0,
1.178 misha 868: "one-param version has hash or string param");
1.74 paf 869: } else {
1.129 paf 870: const char* type=params.as_string(0, "type must be string").cstr();
1.178 misha 871: const String* source=params.count()>1? ¶ms.as_string(1, "source must be string"):0;
872: const char* comment=params.count()>2? params.as_string(2, "comment must be string").cstr():0;
873: throw Exception(type, source, "%s", comment?comment:"");
1.74 paf 874: }
1.132 paf 875: }
1.129 paf 876:
1.150 paf 877: static void _sleep_operator(Request& r, MethodParams& params) {
878: double seconds=params.as_double(0, "seconds must be double", r);
1.206 ! moko 879: if(seconds>0)
! 880: pa_sleep((int)trunc(seconds), (int)trunc((seconds-trunc(seconds))*1000000));
1.150 paf 881: }
882:
1.129 paf 883: #if defined(WIN32) && defined(_DEBUG)
884: # define PA_BPT
1.138 paf 885: static void _bpt(Request&, MethodParams&) {
1.129 paf 886: _asm int 3;
887: }
888: #endif
889:
1.10 paf 890: // constructor
891:
1.129 paf 892: VClassMAIN::VClassMAIN(): VClass() {
893: set_name(*new String(MAIN_CLASS_NAME));
894:
895: #ifdef PA_BPT
896: // ^bpt[]
897: add_native_method("bpt", Method::CT_ANY, _bpt, 0, 0);
898: #endif
1.121 paf 899:
1.1 paf 900: // ^if(condition){code-when-true}
901: // ^if(condition){code-when-true}{code-when-false}
1.200 moko 902: // ^if(condition){code-when-true} (another condition){code-when-true} ... {code-when-false}
903: add_native_method("if", Method::CT_ANY, _if, 2, 10000, Method::CO_WITHOUT_FRAME);
1.1 paf 904:
1.194 misha 905: // ^untaint[as-is|uri|sql|js|html|html-typo|regex|parser-code]{code}
906: add_native_method("untaint", Method::CT_ANY, _untaint, 1, 2, Method::CO_WITHOUT_FRAME);
1.1 paf 907:
1.194 misha 908: // ^taint[as-is|uri|sql|js|html|html-typo|regex|parser-code]{code}
909: add_native_method("taint", Method::CT_ANY, _taint, 1, 2, Method::CO_WITHOUT_FRAME);
1.1 paf 910:
1.205 moko 911: // ^apply-taint[untaint lang][string]
912: add_native_method("apply-taint", Method::CT_ANY, _apply_taint, 1, 2, Method::CO_WITHOUT_FRAME);
913:
1.1 paf 914: // ^process[code]
1.146 paf 915: add_native_method("process", Method::CT_ANY, _process, 1, 3);
1.1 paf 916:
917: // ^rem{code}
1.185 misha 918: add_native_method("rem", Method::CT_ANY, _rem, 1, 10000, Method::CO_WITHOUT_FRAME);
1.1 paf 919:
920: // ^while(condition){code}
1.185 misha 921: add_native_method("while", Method::CT_ANY, _while, 2, 3, Method::CO_WITHOUT_FRAME);
1.1 paf 922:
923: // ^use[file]
1.10 paf 924: add_native_method("use", Method::CT_ANY, _use, 1, 1);
1.1 paf 925:
1.164 paf 926: // ^break[]
1.185 misha 927: add_native_method("break", Method::CT_ANY, _break, 0, 0, Method::CO_WITHOUT_FRAME);
928:
1.164 paf 929: // ^continue[]
1.185 misha 930: add_native_method("continue", Method::CT_ANY, _continue, 0, 0, Method::CO_WITHOUT_FRAME);
931:
1.54 paf 932: // ^for[i](from-number;to-number-inclusive){code}[delim]
1.185 misha 933: add_native_method("for", Method::CT_ANY, _for, 3+1, 3+1+1, Method::CO_WITHOUT_WCONTEXT);
1.1 paf 934:
935: // ^eval(expr)
936: // ^eval(expr)[format]
1.185 misha 937: add_native_method("eval", Method::CT_ANY, _eval, 1, 2, Method::CO_WITHOUT_FRAME);
1.52 parser 938:
1.1 paf 939: // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s}
1.10 paf 940: add_native_method("connect", Method::CT_ANY, _connect, 2, 2);
941:
1.63 paf 942:
1.136 paf 943: // ^cache[file_spec](time){code}[{catch code}] time=0 no cache
1.65 paf 944: // ^cache[file_spec] delete cache
1.161 paf 945: // ^cache[] get current expiration time
946: add_native_method("cache", Method::CT_ANY, _cache, 0, 4);
1.63 paf 947:
1.28 parser 948: // switch
949:
950: // ^switch[value]{cases}
1.185 misha 951: add_native_method("switch", Method::CT_ANY, _switch, 2, 2, Method::CO_WITHOUT_FRAME);
1.28 parser 952:
953: // ^case[value]{code}
1.185 misha 954: add_native_method("case", Method::CT_ANY, _case, 2, 10000, Method::CO_WITHOUT_FRAME);
1.74 paf 955:
956: // try-catch
957:
958: // ^try{code}{catch code}
1.185 misha 959: add_native_method("try", Method::CT_ANY, _try_operator, 2, 3, Method::CO_WITHOUT_FRAME);
1.74 paf 960: // ^throw[$exception hash]
961: // ^throw[type;source;comment]
962: add_native_method("throw", Method::CT_ANY, _throw_operator, 1, 3);
963:
1.150 paf 964: add_native_method("sleep", Method::CT_ANY, _sleep_operator, 1, 1);
1.10 paf 965: }
1.11 paf 966:
967: // constructor & configurator
1.1 paf 968:
1.129 paf 969: VStateless_class& VClassMAIN_create() {
970: return *new VClassMAIN;
1.1 paf 971: }
E-mail: