|
|
| version 1.32, 2001/07/03 09:37:04 | 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://paf.design.ru) | |
| Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf) | $Id$ |
| */ | */ |
| static const char *RCSId="$Id$"; | |
| #include "classes.h" | #include "classes.h" |
| #include "pa_config_includes.h" | #include "pa_config_includes.h" |
| Line 22 static const char *RCSId="$Id$"; | Line 22 static const char *RCSId="$Id$"; |
| #define OP_CLASS_NAME "OP" | #define OP_CLASS_NAME "OP" |
| #define MAIN_SQL_NAME "SQL" | |
| #define MAIN_SQL_DRIVERS_NAME "drivers" | |
| #define SWITCH_DATA_NAME "SWITCH-DATA" | |
| #define DEFAULT_VALUE "DEFAULT" | |
| // local variable | |
| static Methoded *OP; | |
| // class | // class |
| class MOP : public Methoded { | class MOP : public Methoded { |
| Line 40 public: | Line 30 public: |
| public: // Methoded | public: // Methoded |
| bool used_directly() { return true; } | bool used_directly() { return true; } |
| void configure_user(Request& r); | void configure_user(Request& r); |
| private: | private: |
| String main_sql_name; | String main_sql_name; |
| String main_sql_drivers_name; | String main_sql_drivers_name; |
| Line 48 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->get_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->get_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->get_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 81 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"); |
| } | } |
| { | { |
| Value& vbody=params->get_no_junction(params->size()-1, "body must not be code"); | Value& vbody=params->as_no_junction(params->size()-1, "body must not be code"); |
| String result(r.pool()); | String result(r.pool()); |
| result.append( | result.append( |
| Line 116 static void _process(Request& r, const S | Line 109 static void _process(Request& r, const S |
| origin.file, 1+origin.line, | origin.file, 1+origin.line, |
| method_name.cstr()); | method_name.cstr()); |
| #else | #else |
| strncpy(place, MAX_STRING, method_name.cstr()); | strncpy(place, method_name.cstr(), MAX_STRING-1); place[MAX_STRING-1]=0; |
| #endif | #endif |
| VStateless_class& self_class=*r.self->get_class(); | VStateless_class& self_class=*r.self->get_class(); |
| Line 128 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 143 static void _process(Request& r, const S | Line 136 static void _process(Request& r, const S |
| } | } |
| static void _rem(Request& r, const String&, MethodParams *params) { | static void _rem(Request& r, const String&, MethodParams *params) { |
| params->get_junction(0, "body must be code"); | params->as_junction(0, "body must be code"); |
| } | } |
| static void _while(Request& r, const String& method_name, MethodParams *params) { | static void _while(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& vcondition=params->get_junction(0, "condition must be expression"); | Value& vcondition=params->as_junction(0, "condition must be expression"); |
| Value& body=params->get_junction(1, "body must be code"); | Value& body=params->as_junction(1, "body must be code"); |
| // while... | // while... |
| 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 174 static void _while(Request& r, const Str | Line 167 static void _while(Request& r, const Str |
| } | } |
| static void _use(Request& r, const String& method_name, MethodParams *params) { | static void _use(Request& r, const String& method_name, MethodParams *params) { |
| Value& vfile=params->get_no_junction(0, "file name must not be code"); | Value& vfile=params->as_no_junction(0, "file name must not be code"); |
| r.use_file(r.absolute(vfile.as_string())); | r.use_file(vfile.as_string()); |
| } | } |
| 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->get_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.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 209 static void _for(Request& r, const Strin | Line 202 static void _for(Request& r, const Strin |
| } | } |
| static void _eval(Request& r, const String& method_name, MethodParams *params) { | static void _eval(Request& r, const String& method_name, MethodParams *params) { |
| Value& expr=params->get_junction(0, "need expression"); | Value& expr=params->as_junction(0, "need expression"); |
| // evaluate expresion | // evaluate expresion |
| 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->get_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(); |
| String& string=*new(pool) String(pool); | String& string=*new(pool) String(pool); |
| Line 226 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) { | |
| typedef double (*math1_func_ptr)(double); | |
| static double round(double param) { return floor(param+0.5); } | |
| static double sign(double param) { return param > 0 ? 1 : ( param < 0 ? -1 : 0 ); } | |
| static void math1(Request& r, | |
| const String& method_name, MethodParams *params, | |
| math1_func_ptr func) { | |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& param=params->get_junction(0, "parameter must be expression"); | |
| Value& result=*new(pool) VDouble(pool, (*func)(r.process(param).as_double())); | const String& serror=params->as_string(0, "message must be string"); |
| result.set_name(method_name); | throw Exception(0, 0, |
| r.write_no_lang(result); | &method_name, |
| "%s", serror.cstr()); | |
| } | } |
| #define MATH1(name) \ | |
| static void _##name(Request& r, const String& method_name, MethodParams *params) {\ | |
| math1(r, method_name, params, &name);\ | |
| } | |
| #define MATH1P(name_parser, name_c) \ | |
| static void _##name_parser(Request& r, const String& method_name, MethodParams *params) {\ | |
| math1(r, method_name, params, &name_c);\ | |
| } | |
| MATH1(round); MATH1(floor); MATH1P(ceiling, ceil); | |
| MATH1P(abs, fabs); MATH1(sign); | |
| MATH1(exp); MATH1(log); | |
| MATH1(sin); MATH1(asin); | |
| MATH1(cos); MATH1(acos); | |
| MATH1(tan); MATH1(atan); | |
| MATH1(sqrt); | |
| /// @todo rewrite ugly code with try/try to autoobject TempConnection | |
| typedef double (*math2_func_ptr)(double, double); | static void _connect(Request& r, const String& method_name, MethodParams *params) { |
| static void math2(Request& r, | |
| const String& method_name, MethodParams *params, | |
| math2_func_ptr func) { | |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& a=params->get_junction(0, "parameter must be expression"); | |
| Value& b=params->get_junction(1, "parameter must be expression"); | |
| Value& result=*new(pool) VDouble(pool, (*func)( | |
| r.process(a).as_double(), | |
| r.process(b).as_double())); | |
| result.set_name(method_name); | |
| r.write_no_lang(result); | |
| } | |
| #define MATH2(name) \ | |
| static void _##name(Request& r, const String& method_name, MethodParams *params) {\ | |
| math2(r, method_name, params, &name);\ | |
| } | |
| MATH2(pow); | |
| Value& url=params->as_no_junction(0, "url must not be code"); | |
| static void _connect(Request& r, const String&, MethodParams *params) { | Value& body_code=params->as_junction(1, "body must be code"); |
| Pool& pool=r.pool(); | |
| Value& url=params->get_no_junction(0, "url must not be code"); | |
| Value& body_code=params->get_junction(1, "body must be code"); | |
| Table *protocol2driver_and_client= | Table *protocol2driver_and_client= |
| static_cast<Table *>(r.classes_conf.get(OP->name())); | static_cast<Table *>(r.classes_conf.get(r.OP.name())); |
| // connect | // connect |
| SQL_Connection& connection=SQL_driver_manager->get_connection( | SQL_Connection& connection=SQL_driver_manager->get_connection( |
| url.as_string(), 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()); | |
| } | } |
| static String *switch_data_name; | #ifndef DOXYGEN |
| static String *default_value; | |
| struct Switch_data { | struct Switch_data { |
| Value *searching; | Value *searching; |
| Value *found; | Value *found; |
| Value *_default; | Value *_default; |
| }; | }; |
| #endif | |
| static void _switch(Request& r, const String&, MethodParams *params) { | static void _switch(Request& r, const String&, MethodParams *params) { |
| void *backup=r.classes_conf.get(*switch_data_name); | void *backup=r.classes_conf.get(*switch_data_name); |
| Switch_data data={&r.process(params->get(0))}; | Switch_data data={&r.process(params->get(0))}; |
| r.classes_conf.put(*switch_data_name, &data); | r.classes_conf.put(*switch_data_name, &data); |
| r.process(params->get_junction(1, "switch cases must be code")); // and ignore result | r.process(params->as_junction(1, "switch cases must be code")); // and ignore result |
| r.classes_conf.put(*switch_data_name, backup); | r.classes_conf.put(*switch_data_name, backup); |
| Line 352 static void _switch(Request& r, const St | Line 294 static void _switch(Request& r, const St |
| r.write_pass_lang(r.process(*code)); | r.write_pass_lang(r.process(*code)); |
| } | } |
| static void _case(Request& r, const String&, MethodParams *params) { | static void _case(Request& r, const String& method_name, MethodParams *params) { |
| Switch_data& data=*static_cast<Switch_data *>(r.classes_conf.get(*switch_data_name)); | Pool& pool=r.pool(); |
| Switch_data *data=static_cast<Switch_data *>(r.classes_conf.get(*switch_data_name)); | |
| if(!data) | |
| throw Exception(0, 0, | |
| &method_name, | |
| "without switch"); | |
| int count=params->size(); | int count=params->size(); |
| Value *code=¶ms->get_junction(--count, "case result must be code"); | Value *code=¶ms->as_junction(--count, "case result must be code"); |
| for(int i=0; i<count; i++) { | for(int i=0; i<count; i++) { |
| Value& value=r.process(params->get(i)); | Value& value=r.process(params->get(i)); |
| if(value.as_string() == *default_value) { | if(value.as_string() == *case_default_value) { |
| data._default=code; | data->_default=code; |
| break; | break; |
| } | } |
| bool matches; | bool matches; |
| if(data.searching->is_string()) | if(data->searching->is_string()) |
| matches=data.searching->as_string() == value.as_string(); | matches=data->searching->as_string() == value.as_string(); |
| else | else |
| matches=data.searching->as_double() == value.as_double(); | matches=data->searching->as_double() == value.as_double(); |
| if(matches) { | if(matches) { |
| data.found=code; | data->found=code; |
| break; | break; |
| } | } |
| } | } |
| Line 386 MOP::MOP(Pool& apool) : Methoded(apool), | Line 334 MOP::MOP(Pool& apool) : Methoded(apool), |
| { | { |
| set_name(*NEW String(pool(), OP_CLASS_NAME)); | set_name(*NEW String(pool(), OP_CLASS_NAME)); |
| switch_data_name=NEW String(pool(), SWITCH_DATA_NAME); | |
| default_value=NEW String(pool(), DEFAULT_VALUE); | |
| // ^if(condition){code-when-true} | // ^if(condition){code-when-true} |
| // ^if(condition){code-when-true}{code-when-false} | // ^if(condition){code-when-true}{code-when-false} |
| 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 403 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 411 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] | |
| // math functions | add_native_method("error", Method::CT_ANY, _error, 1, 1); |
| // ^FUNC(expr) | |
| #define ADD_MATH1(name) \ | |
| add_native_method(#name, Method::CT_ANY, _##name, 1, 1) | |
| ADD_MATH1(round); ADD_MATH1(floor); ADD_MATH1(ceiling); | |
| ADD_MATH1(abs); ADD_MATH1(sign); | |
| ADD_MATH1(exp); ADD_MATH1(log); | |
| ADD_MATH1(sin); ADD_MATH1(asin); | |
| ADD_MATH1(cos); ADD_MATH1(acos); | |
| ADD_MATH1(tan); ADD_MATH1(atan); | |
| ADD_MATH1(sqrt); | |
| #define ADD_MATH2(name) \ | |
| add_native_method(#name, Method::CT_ANY, _##name, 2, 2) | |
| // ^pow(x;y) | |
| ADD_MATH2(pow); | |
| // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} | // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} |
| Line 449 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 |
| Methoded *MOP_create(Pool& pool) { | Methoded *MOP_create(Pool& pool) { |
| return OP=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(); |