|
|
| version 1.45, 2001/09/26 10:32:25 | version 1.62, 2001/11/22 12:40:48 |
|---|---|
| Line 2 | Line 2 |
| Parser: parser @b operators. | Parser: parser @b operators. |
| Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) |
| Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf) | Author: Alexander Petrosyan <paf@design.ru> (http://paf.design.ru) |
| $Id$ | $Id$ |
| */ | */ |
| Line 39 private: | Line 39 private: |
| // methods | // methods |
| static void _if(Request& r, const String&, MethodParams *params) { | static void _if(Request& r, const String&, MethodParams *params) { |
| Value& condition_code=params->get(0); | Value& condition_code=params->as_junction(0, "condition must be expression"); |
| bool condition=r.process(condition_code, | bool condition=r.process(condition_code, |
| 0/*no name*/, | 0/*no name*/, |
| false/*don't intercept string*/).as_bool(); | false/*don't intercept string*/).as_bool(); |
| if(condition) | if(condition) |
| r.write_pass_lang(r.process(params->as_junction(1, "'then' parameter must be code"))); | r.write_pass_lang(r.process(params->as_junction(1, "'then' parameter must be code"))); |
| else if(params->size()==3) | else if(params->size()>2) |
| r.write_pass_lang(r.process(params->as_junction(2, "'else' parameter must be code"))); | r.write_pass_lang(r.process(params->as_junction(2, "'else' parameter must be code"))); |
| } | } |
| static void _untaint(Request& r, const String& method_name, MethodParams *params) { | static void _untaint(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| const String& lang_name=r.process(params->get(0)).as_string(); | uchar lang; |
| String::Untaint_lang lang=static_cast<String::Untaint_lang>( | if(params->size()==1) |
| untaint_lang_name2enum->get_int(lang_name)); | lang=String::UL_AS_IS; // mark as simply 'tainted'. useful in html from sql |
| if(!lang) | else { |
| PTHROW(0, 0, | const String& lang_name=params->as_string(0, "lang must be string"); |
| &lang_name, | lang=untaint_lang_name2enum->get_int(lang_name); |
| "invalid untaint language"); | if(!lang) |
| throw Exception(0, 0, | |
| &lang_name, | |
| "invalid taint language"); | |
| } | |
| { | { |
| Value& vbody=params->as_junction(1, "body must be code"); | Value& vbody=params->as_junction(params->size()-1, "body must be code"); |
| Temp_lang temp_lang(r, lang); // set temporarily specified ^untaint[language; | Temp_lang temp_lang(r, lang); // set temporarily specified ^untaint[language; |
| r.write_pass_lang(r.process(vbody)); // process marking tainted with that lang | r.write_pass_lang(r.process(vbody)); // process marking tainted with that lang |
| Line 72 static void _untaint(Request& r, const S | Line 76 static void _untaint(Request& r, const S |
| static void _taint(Request& r, const String&, MethodParams *params) { | static void _taint(Request& r, const String&, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| String::Untaint_lang lang; | uchar lang; |
| if(params->size()==1) | if(params->size()==1) |
| lang=String::UL_TAINTED; // mark as simply 'tainted'. useful in table:set | lang=String::UL_TAINTED; // mark as simply 'tainted'. useful in table:set |
| else { | else { |
| const String& lang_name= | const String& lang_name=params->as_string(0, "lang must be string"); |
| r.process(params->get(0)).as_string(); | lang=untaint_lang_name2enum->get_int(lang_name); |
| lang=static_cast<String::Untaint_lang>( | |
| untaint_lang_name2enum->get_int(lang_name)); | |
| if(!lang) | if(!lang) |
| PTHROW(0, 0, | throw Exception(0, 0, |
| &lang_name, | &lang_name, |
| "invalid taint language"); | "invalid taint language"); |
| } | } |
| Line 119 static void _process(Request& r, const S | Line 121 static void _process(Request& r, const S |
| // evaluate source to process | // evaluate source to process |
| const String& source= | const String& source= |
| r.process(params->get(0)).as_string(); | r.process(params->as_junction(0, "body must be code")).as_string(); |
| // process source code, append processed methods to 'self' class | // process source code, append processed methods to 'self' class |
| // maybe-define new @main | // maybe-define new @main |
| r.use_buf(source.cstr(), place, &self_class); | r.use_buf(source.cstr(String::UL_UNSPECIFIED, r.connection), place, &self_class); |
| // maybe-execute @main[] | // maybe-execute @main[] |
| if(const Method *method=self_class.get_method(*main_method_name)) { | if(const Method *method=self_class.get_method(*main_method_name)) { |
| Line 147 static void _while(Request& r, const Str | Line 149 static void _while(Request& r, const Str |
| int endless_loop_count=0; | int endless_loop_count=0; |
| while(true) { | while(true) { |
| if(++endless_loop_count>=MAX_LOOPS) // endless loop? | if(++endless_loop_count>=MAX_LOOPS) // endless loop? |
| PTHROW(0, 0, | throw Exception(0, 0, |
| &method_name, | &method_name, |
| "endless loop detected"); | "endless loop detected"); |
| Line 171 static void _use(Request& r, const Strin | Line 173 static void _use(Request& r, const Strin |
| static void _for(Request& r, const String& method_name, MethodParams *params) { | static void _for(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| const String& var_name=r.process(params->get(0)).as_string(); | const String& var_name=params->as_string(0, "var name must be string"); |
| int from=r.process(params->get(1)).as_int(); | int from=params->as_int(1, "from must be int", r); |
| int to=r.process(params->get(2)).as_int(); | int to=params->as_int(2, "to must be int", r); |
| Value& body_code=params->as_junction(3, "body must be code"); | Value& body_code=params->as_junction(3, "body must be code"); |
| Value *delim_code=params->size()==3+1+1?¶ms->get(3+1):0; | Value *delim_maybe_code=params->size()>4?¶ms->get(4):0; |
| if(to-from>=MAX_LOOPS) // too long loop? | |
| throw Exception(0, 0, | |
| &method_name, | |
| "endless loop detected"); | |
| bool need_delim=false; | bool need_delim=false; |
| VInt *vint=new(pool) VInt(pool, 0); | VInt *vint=new(pool) VInt(pool, 0); |
| int endless_loop_count=0; | |
| for(int i=from; i<=to; i++) { | for(int i=from; i<=to; i++) { |
| if(++endless_loop_count>=MAX_LOOPS) // endless loop? | |
| PTHROW(0, 0, | |
| &method_name, | |
| "endless loop detected"); | |
| vint->set_int(i); | vint->set_int(i); |
| r.self/*root*/->put_element(var_name, vint); | r.root->put_element(var_name, vint); |
| Value& processed_body=r.process(body_code); | Value& processed_body=r.process(body_code); |
| if(delim_code) { // delimiter set? | if(delim_maybe_code) { // delimiter set? |
| const String *string=processed_body.get_string(); | const String *string=processed_body.get_string(); |
| if(need_delim && string && string->size()) // need delim & iteration produced string? | if(need_delim && string && string->size()) // need delim & iteration produced string? |
| r.write_pass_lang(r.process(*delim_code)); | r.write_pass_lang(r.process(*delim_maybe_code)); |
| need_delim=true; | need_delim=true; |
| } | } |
| r.write_pass_lang(processed_body); | r.write_pass_lang(processed_body); |
| Line 205 static void _eval(Request& r, const Stri | Line 207 static void _eval(Request& r, const Stri |
| Value *result=r.process(expr, | Value *result=r.process(expr, |
| 0/*no name YET*/, | 0/*no name YET*/, |
| true/*don't intercept string*/).as_expr_result(); | true/*don't intercept string*/).as_expr_result(); |
| if(params->size()==2) { | if(params->size()>1) { |
| Value& fmt=params->as_no_junction(1, "fmt must not be code"); | Value& fmt=params->as_no_junction(1, "fmt must not be code"); |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Line 217 static void _eval(Request& r, const Stri | Line 219 static void _eval(Request& r, const Stri |
| r.write_no_lang(*result); | r.write_no_lang(*result); |
| } | } |
| static void _error(Request& r, const String& method_name, MethodParams *params) { | |
| Pool& pool=r.pool(); | |
| const String& serror=params->as_string(0, "message must be string"); | |
| throw Exception(0, 0, | |
| &method_name, | |
| "%s", serror.cstr()); | |
| } | |
| /// @todo rewrite ugly code with try/try to autoobject TempConnection | |
| static void _connect(Request& r, const String& method_name, MethodParams *params) { | static void _connect(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Line 231 static void _connect(Request& r, const S | Line 243 static void _connect(Request& r, const S |
| SQL_Connection& connection=SQL_driver_manager->get_connection( | SQL_Connection& connection=SQL_driver_manager->get_connection( |
| url.as_string(), method_name, protocol2driver_and_client); | url.as_string(), method_name, protocol2driver_and_client); |
| Exception rethrow_me; | |
| // remember/set current connection | // remember/set current connection |
| SQL_Connection *saved_connection=r.connection; | SQL_Connection *saved_connection=r.connection; |
| r.connection=&connection; | r.connection=&connection; |
| // execute body | // execute body |
| bool body_failed=false; | try { |
| PTRY | try { |
| r.write_assign_lang(r.process(body_code)); | r.write_assign_lang(r.process(body_code)); |
| PCATCH(e) { // connect/process problem | |
| rethrow_me=e; body_failed=true; | |
| } | |
| PEND_CATCH | |
| bool finalizer_failed=false; | |
| PTRY | |
| // FINALLY | |
| if(body_failed) | |
| connection.rollback(); | |
| else | |
| connection.commit(); | connection.commit(); |
| PCATCH(e) { // commit/rollback problem | } catch(...) { // process/commit problem |
| rethrow_me=e; finalizer_failed=true; | connection.rollback(); |
| /*re*/throw; | |
| } | |
| } catch(...) { | |
| // close connection [cache it] | |
| connection.close(); | |
| // recall current connection from remembered | |
| r.connection=saved_connection; | |
| /*re*/throw; | |
| } | } |
| PEND_CATCH | |
| // and anyway | |
| // close connection [cache it] | // close connection [cache it] |
| connection.close(); | connection.close(); |
| // recall current connection from remembered | // recall current connection from remembered |
| r.connection=saved_connection; | r.connection=saved_connection; |
| if(body_failed || finalizer_failed) // were there an exception for us to rethrow? | |
| PTHROW(rethrow_me.type(), rethrow_me.code(), | |
| rethrow_me.problem_source(), | |
| rethrow_me.comment()); | |
| } | } |
| #ifndef DOXYGEN | #ifndef DOXYGEN |
| Line 292 static void _case(Request& r, const Stri | Line 299 static void _case(Request& r, const Stri |
| Switch_data *data=static_cast<Switch_data *>(r.classes_conf.get(*switch_data_name)); | Switch_data *data=static_cast<Switch_data *>(r.classes_conf.get(*switch_data_name)); |
| if(!data) | if(!data) |
| PTHROW(0, 0, | throw Exception(0, 0, |
| &method_name, | &method_name, |
| "without switch"); | "without switch"); |
| Line 332 MOP::MOP(Pool& apool) : Methoded(apool), | Line 339 MOP::MOP(Pool& apool) : Methoded(apool), |
| add_native_method("if", Method::CT_ANY, _if, 2, 3); | add_native_method("if", Method::CT_ANY, _if, 2, 3); |
| // ^untaint[as-is|uri|sql|js|html|html-typo]{code} | // ^untaint[as-is|uri|sql|js|html|html-typo]{code} |
| add_native_method("untaint", Method::CT_ANY, _untaint, 2, 2); | add_native_method("untaint", Method::CT_ANY, _untaint, 1, 2); |
| // ^taint[as-is|uri|sql|js|html|html-typo]{code} | // ^taint[as-is|uri|sql|js|html|html-typo]{code} |
| add_native_method("taint", Method::CT_ANY, _taint, 1, 2); | add_native_method("taint", Method::CT_ANY, _taint, 1, 2); |
| Line 341 MOP::MOP(Pool& apool) : Methoded(apool), | Line 348 MOP::MOP(Pool& apool) : Methoded(apool), |
| add_native_method("process", Method::CT_ANY, _process, 1, 1); | add_native_method("process", Method::CT_ANY, _process, 1, 1); |
| // ^rem{code} | // ^rem{code} |
| add_native_method("rem", Method::CT_ANY, _rem, 1, 1000); | add_native_method("rem", Method::CT_ANY, _rem, 1, 10000); |
| // ^while(condition){code} | // ^while(condition){code} |
| add_native_method("while", Method::CT_ANY, _while, 2, 2); | add_native_method("while", Method::CT_ANY, _while, 2, 2); |
| Line 349 MOP::MOP(Pool& apool) : Methoded(apool), | Line 356 MOP::MOP(Pool& apool) : Methoded(apool), |
| // ^use[file] | // ^use[file] |
| add_native_method("use", Method::CT_ANY, _use, 1, 1); | add_native_method("use", Method::CT_ANY, _use, 1, 1); |
| // ^for[i;from-number;to-number-inclusive]{code}[delim] | // ^for[i](from-number;to-number-inclusive){code}[delim] |
| add_native_method("for", Method::CT_ANY, _for, 3+1, 3+1+1); | add_native_method("for", Method::CT_ANY, _for, 3+1, 3+1+1); |
| // ^eval(expr) | // ^eval(expr) |
| // ^eval(expr)[format] | // ^eval(expr)[format] |
| add_native_method("eval", Method::CT_ANY, _eval, 1, 2); | add_native_method("eval", Method::CT_ANY, _eval, 1, 2); |
| // ^error[msg] | |
| add_native_method("error", Method::CT_ANY, _error, 1, 1); | |
| // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} | // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} |
| add_native_method("connect", Method::CT_ANY, _connect, 2, 2); | add_native_method("connect", Method::CT_ANY, _connect, 2, 2); |
| Line 366 MOP::MOP(Pool& apool) : Methoded(apool), | Line 376 MOP::MOP(Pool& apool) : Methoded(apool), |
| add_native_method("switch", Method::CT_ANY, _switch, 2, 2); | add_native_method("switch", Method::CT_ANY, _switch, 2, 2); |
| // ^case[value]{code} | // ^case[value]{code} |
| add_native_method("case", Method::CT_ANY, _case, 2, 1000); | add_native_method("case", Method::CT_ANY, _case, 2, 10000); |
| } | } |
| // constructor & configurator | // constructor & configurator |
| Line 375 Methoded *MOP_create(Pool& pool) { | Line 385 Methoded *MOP_create(Pool& pool) { |
| return new(pool) MOP(pool); | return new(pool) MOP(pool); |
| } | } |
| void MOP::configure_user(Request& r) { | void MOP::configure_user(Request& r) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |