--- parser3/src/classes/op.C 2001/04/15 14:40:53 1.7 +++ parser3/src/classes/op.C 2001/07/26 16:16:57 1.35 @@ -1,22 +1,48 @@ /** @file - Parser: @b ROOT parser class. + Parser: parser @b operators. Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) Author: Alexander Petrosyan (http://design.ru/paf) - - $Id: op.C,v 1.7 2001/04/15 14:40:53 paf Exp $ */ +static const char *RCSId="$Id: op.C,v 1.35 2001/07/26 16:16:57 parser Exp $"; +#include "classes.h" #include "pa_config_includes.h" -#include - #include "pa_common.h" #include "pa_request.h" -#include "_op.h" #include "pa_vint.h" #include "pa_sql_connection.h" +// limits + +#define MAX_LOOPS 10000 + +// defines + +#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" + +// class + +class MOP : public Methoded { +public: + MOP(Pool& pool); +public: // Methoded + bool used_directly() { return true; } + void configure_user(Request& r); +private: + String main_sql_name; + String main_sql_drivers_name; +}; + +// methods + static void _if(Request& r, const String&, MethodParams *params) { Value& condition_code=params->get(0); @@ -24,9 +50,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->as_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->as_junction(2, "'else' parameter must be code"))); } static void _untaint(Request& r, const String& method_name, MethodParams *params) { @@ -41,7 +67,7 @@ static void _untaint(Request& r, const S "invalid untaint language"); { - Value& vbody=params->get_junction(1, "body must be code"); + Value& vbody=params->as_junction(1, "body must be code"); Temp_lang temp_lang(r, lang); // set temporarily specified ^untaint[language; r.write_pass_lang(r.process(vbody)); // process marking tainted with that lang @@ -66,7 +92,7 @@ static void _taint(Request& r, const Str } { - 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()); result.append( @@ -113,19 +139,19 @@ static void _process(Request& r, const S } 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) { Pool& pool=r.pool(); - Value& vcondition=params->get_junction(0, "condition must be expression"); - Value& body=params->get_junction(1, "body must be code"); + Value& vcondition=params->as_junction(0, "condition must be expression"); + Value& body=params->as_junction(1, "body must be code"); // 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"); @@ -144,24 +170,23 @@ static void _while(Request& r, const Str } 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())); } -/// ^for[i;from-number;to-number-inclusive]{code}[delim] 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(); - Value& body_code=params->get_junction(3, "body must be code"); + int from=r.process(params->get(1)).as_int(); + int to=r.process(params->get(2)).as_int(); + Value& body_code=params->as_junction(3, "body must be code"); Value *delim_code=params->size()==3+1+1?¶ms->get(3+1):0; bool need_delim=false; 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"); @@ -179,69 +204,37 @@ static void _for(Request& r, const Strin } } -static void _eval(Request& r, const String&, MethodParams *params) { - Value& expr=params->get_junction(0, "need expression"); +static void _eval(Request& r, const String& method_name, MethodParams *params) { + Value& expr=params->as_junction(0, "need expression"); // evaluate expresion Value *result=r.process(expr, - 0/*no name*/, + 0/*no name YET*/, true/*don't intercept string*/).as_expr_result(); if(params->size()==2) { - 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(); String& string=*new(pool) String(pool); string.APPEND_CONST(format(pool, result->as_double(), fmt.as_string().cstr())); result=new(pool) VString(string); } + result->set_name(method_name); r.write_no_lang(*result); } -typedef double (*math_one_double_op_func_ptr)(double); -static double round(double op) { return floor(op+0.5); } -static double sign(double op) { return op > 0 ? 1 : ( op < 0 ? -1 : 0 ); } - -static void double_one_op( - Request& r, - const String& method_name, MethodParams *params, - math_one_double_op_func_ptr func) { - 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())); - r.write_no_lang(result); -} - -static void _round(Request& r, const String& method_name, MethodParams *params) { - double_one_op(r, method_name, params, &round); -} - -static void _floor(Request& r, const String& method_name, MethodParams *params) { - double_one_op(r, method_name, params, &floor); -} - -static void _ceiling(Request& r, const String& method_name, MethodParams *params) { - double_one_op(r, method_name, params, &ceil); -} - -static void _abs(Request& r, const String& method_name, MethodParams *params) { - double_one_op(r, method_name, params, &fabs); -} - -static void _sign(Request& r, const String& method_name, MethodParams *params) { - double_one_op(r, method_name, params, &sign); -} - -/// ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} 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"); + Value& url=params->as_no_junction(0, "url must not be code"); + Value& body_code=params->as_junction(1, "body must be code"); + + Table *protocol2driver_and_client= + static_cast(r.classes_conf.get(r.OP.name())); // connect SQL_Connection& connection=SQL_driver_manager->get_connection( - url.as_string(), r.protocol2library); + url.as_string(), protocol2driver_and_client); Exception rethrow_me; // remember/set current connection @@ -279,60 +272,120 @@ static void _connect(Request& r, const S rethrow_me.comment()); } -// initialize +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->as_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->as_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), + main_sql_name(apool, MAIN_SQL_NAME), + main_sql_drivers_name(apool, MAIN_SQL_DRIVERS_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); -void initialize_op_class(Pool& pool, VStateless_class& vclass) { // ^if(condition){code-when-true} // ^if(condition){code-when-true}{code-when-false} - vclass.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} - vclass.add_native_method("untaint", Method::CT_ANY, _untaint, 2, 2); + add_native_method("untaint", Method::CT_ANY, _untaint, 2, 2); // ^taint[as-is|uri|sql|js|html|html-typo]{code} - vclass.add_native_method("taint", Method::CT_ANY, _taint, 1, 2); + add_native_method("taint", Method::CT_ANY, _taint, 1, 2); // ^process[code] - vclass.add_native_method("process", Method::CT_ANY, _process, 1, 1); + add_native_method("process", Method::CT_ANY, _process, 1, 1); // ^rem{code} - vclass.add_native_method("rem", Method::CT_ANY, _rem, 1, 1); + add_native_method("rem", Method::CT_ANY, _rem, 1, 1000); // ^while(condition){code} - vclass.add_native_method("while", Method::CT_ANY, _while, 2, 2); + add_native_method("while", Method::CT_ANY, _while, 2, 2); // ^use[file] - vclass.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] - vclass.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)[format] - vclass.add_native_method("eval", Method::CT_ANY, _eval, 1, 2); + add_native_method("eval", Method::CT_ANY, _eval, 1, 2); - // math functions + // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} + add_native_method("connect", Method::CT_ANY, _connect, 2, 2); - // ^round(expr) - vclass.add_native_method("round", Method::CT_ANY, _round, 1, 1); + // switch - // ^floor(expr) - vclass.add_native_method("floor", Method::CT_ANY, _floor, 1, 1); + // ^switch[value]{cases} + add_native_method("switch", Method::CT_ANY, _switch, 2, 2); - // ^ceiling(expr) - vclass.add_native_method("ceiling", Method::CT_ANY, _ceiling, 1, 1); + // ^case[value]{code} + add_native_method("case", Method::CT_ANY, _case, 2, 1000); +} - // ^abs(expr) - vclass.add_native_method("abs", Method::CT_ANY, _abs, 1, 1); +// constructor & configurator - // ^sign(expr) - vclass.add_native_method("sign", Method::CT_ANY, _sign, 1, 1); +Methoded *MOP_create(Pool& pool) { + return new(pool) MOP(pool); +} - - // connect - // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} - vclass.add_native_method("connect", Method::CT_ANY, _connect, 2, 2); +void MOP::configure_user(Request& r) { + Pool& pool=r.pool(); + // $MAIN:SQL.drivers + if(Value *sql=r.main_class->get_element(main_sql_name)) + if(Value *element=sql->get_element(main_sql_drivers_name)) + if(Table *protocol2library=element->get_table()) + r.classes_conf.put(name(), protocol2library); }