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