--- parser3/src/classes/op.C 2001/05/08 06:13:00 1.17 +++ parser3/src/classes/op.C 2001/06/28 07:44:17 1.30 @@ -4,9 +4,8 @@ Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) Author: Alexander Petrosyan (http://design.ru/paf) - - $Id: op.C,v 1.17 2001/05/08 06:13:00 paf Exp $ */ +static const char *RCSId="$Id: op.C,v 1.30 2001/06/28 07:44:17 parser Exp $"; #include "classes.h" #include "pa_config_includes.h" @@ -15,6 +14,10 @@ #include "pa_vint.h" #include "pa_sql_connection.h" +// limits + +#define MAX_LOOPS 10000 + // defines #define OP_CLASS_NAME "OP" @@ -22,6 +25,9 @@ #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; @@ -48,9 +54,9 @@ static void _if(Request& r, const String 0/*no name*/, false/*don't intercept string*/).as_bool(); if(condition) - r.write_pass_lang(r.process(params->get(1))); + r.write_pass_lang(r.process(params->get_junction(1, "'then' parameter must be code"))); else if(params->size()==3) - r.write_pass_lang(r.process(params->get(2))); + r.write_pass_lang(r.process(params->get_junction(2, "'else' parameter must be code"))); } static void _untaint(Request& r, const String& method_name, MethodParams *params) { @@ -149,7 +155,7 @@ static void _while(Request& r, const Str // while... int endless_loop_count=0; while(true) { - if(++endless_loop_count>=1973) // endless loop? + if(++endless_loop_count>=MAX_LOOPS) // endless loop? PTHROW(0, 0, &method_name, "endless loop detected"); @@ -175,8 +181,8 @@ static void _use(Request& r, const Strin static void _for(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); const String& var_name=r.process(params->get(0)).as_string(); - int from=(int)r.process(params->get(1)).as_double(); - int to=(int)r.process(params->get(2)).as_double(); + int from=r.process(params->get(1)).as_int(); + int to=r.process(params->get(2)).as_int(); Value& body_code=params->get_junction(3, "body must be code"); Value *delim_code=params->size()==3+1+1?¶ms->get(3+1):0; @@ -184,7 +190,7 @@ static void _for(Request& r, const Strin VInt *vint=new(pool) VInt(pool, 0); int endless_loop_count=0; for(int i=from; i<=to; i++) { - if(++endless_loop_count>=2001) // endless loop? + if(++endless_loop_count>=MAX_LOOPS) // endless loop? PTHROW(0, 0, &method_name, "endless loop detected"); @@ -256,17 +262,26 @@ static void _sign(Request& r, const Stri double_one_op(r, method_name, params, &sign); } +static void _log(Request& r, const String& method_name, MethodParams *params) { + double_one_op(r, method_name, params, &log); +} + +static void _exp(Request& r, const String& method_name, MethodParams *params) { + double_one_op(r, method_name, params, &exp); +} + static void _connect(Request& r, const String&, MethodParams *params) { 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 *protocol2library=static_cast(r.classes_conf.get(OP->name())); + Table *protocol2driver_and_client= + static_cast
(r.classes_conf.get(OP->name())); // connect SQL_Connection& connection=SQL_driver_manager->get_connection( - url.as_string(), protocol2library); + url.as_string(), protocol2driver_and_client); Exception rethrow_me; // remember/set current connection @@ -304,6 +319,54 @@ static void _connect(Request& r, const S rethrow_me.comment()); } +static String *switch_data_name; +static String *default_value; + +struct Switch_data { + Value *searching; + Value *found; + Value *_default; +}; + +static void _switch(Request& r, const String&, MethodParams *params) { + void *backup=r.classes_conf.get(*switch_data_name); + Switch_data data={&r.process(params->get(0))}; + r.classes_conf.put(*switch_data_name, &data); + + r.process(params->get_junction(1, "switch cases must be code")); // and ignore result + + r.classes_conf.put(*switch_data_name, backup); + + if(Value *code=data.found ? data.found : data._default) + r.write_pass_lang(r.process(*code)); +} + +static void _case(Request& r, const String&, MethodParams *params) { + Switch_data& data=*static_cast(r.classes_conf.get(*switch_data_name)); + + int count=params->size(); + Value *code=¶ms->get_junction(--count, "case result must be code"); + for(int i=0; iget(i)); + + if(value.as_string() == *default_value) { + data._default=code; + break; + } + + bool matches; + if(data.searching->is_string()) + matches=data.searching->as_string() == value.as_string(); + else + matches=data.searching->as_double() == value.as_double(); + + if(matches) { + data.found=code; + break; + } + } +} + // constructor MOP::MOP(Pool& apool) : Methoded(apool), @@ -312,6 +375,9 @@ MOP::MOP(Pool& apool) : Methoded(apool), { 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}{code-when-false} add_native_method("if", Method::CT_ANY, _if, 2, 3); @@ -326,7 +392,7 @@ MOP::MOP(Pool& apool) : Methoded(apool), add_native_method("process", Method::CT_ANY, _process, 1, 1); // ^rem{code} - add_native_method("rem", Method::CT_ANY, _rem, 1, 1); + add_native_method("rem", Method::CT_ANY, _rem, 1, 1000); // ^while(condition){code} add_native_method("while", Method::CT_ANY, _while, 2, 2); @@ -359,12 +425,25 @@ MOP::MOP(Pool& apool) : Methoded(apool), // ^sign(expr) add_native_method("sign", Method::CT_ANY, _sign, 1, 1); + // ^log(expr) + add_native_method("log", Method::CT_ANY, _log, 1, 1); + + // ^exp(expr) + add_native_method("exp", Method::CT_ANY, _exp, 1, 1); + // connect // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} add_native_method("connect", Method::CT_ANY, _connect, 2, 2); + // switch + + // ^switch[value]{cases} + add_native_method("switch", Method::CT_ANY, _switch, 2, 2); + + // ^case[value]{code} + add_native_method("case", Method::CT_ANY, _case, 2, 1000); } // constructor & configurator