--- parser3/src/classes/Attic/root.C 2001/03/12 17:16:48 1.26 +++ parser3/src/classes/Attic/root.C 2001/04/05 13:19:40 1.60 @@ -1,51 +1,96 @@ -/* - Parser +/** @file + Parser: @b ROOT parser class. + Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) + Author: Alexander Petrosyan (http://design.ru/paf) - $Id: root.C,v 1.26 2001/03/12 17:16:48 paf Exp $ + $Id: root.C,v 1.60 2001/04/05 13:19:40 paf Exp $ */ -#include +#include "pa_config_includes.h" +#include +#include "pa_common.h" #include "pa_request.h" #include "_root.h" +#include "pa_vint.h" +#include "pa_sql_connection.h" -static void _if(Request& r, const String&, Array *params) { - bool condition= - r.process( - *static_cast(params->get(0)), - 0/*no name*/, - false/*don't intercept string*/).get_bool(); +static void _if(Request& r, const String& method_name, Array *params) { + Value& condition_code=*static_cast(params->get(0)); + // forcing ^if(this param type) + r.fail_if_junction_(false, condition_code, + method_name, "condition must be junction"); + + bool condition=r.process(condition_code, + 0/*no name*/, + false/*don't intercept string*/).as_bool(); if(condition) { - Value& value=r.process(*static_cast(params->get(1))); - r.write_pass_lang(value); + Value& then_code=*static_cast(params->get(1)); + // forcing ^if{this param type} + r.fail_if_junction_(false, then_code, + method_name, "then-parameter must be junction"); + r.write_pass_lang(r.process(then_code)); } else if(params->size()==3) { - Value& value=r.process(*static_cast(params->get(2))); - r.write_pass_lang(value); + Value& else_code=*static_cast(params->get(2)); + // forcing ^if{this param type} + r.fail_if_junction_(false, else_code, + method_name, "else-parameter must be junction"); + r.write_pass_lang(r.process(else_code)); } } static void _untaint(Request& r, const String& method_name, Array *params) { + Pool& pool=r.pool(); + const String& lang_name=r.process(*static_cast(params->get(0))).as_string(); String::Untaint_lang lang=static_cast( untaint_lang_name2enum->get_int(lang_name)); if(!lang) - R_THROW(0, 0, + PTHROW(0, 0, &lang_name, "invalid untaint language"); { - Temp_lang temp_lang(r, lang); Value *vbody=static_cast(params->get(1)); // forcing ^untaint[]{this param type} r.fail_if_junction_(false, *vbody, method_name, "body must be junction"); - r.write_pass_lang(r.process(*vbody)); + Temp_lang temp_lang(r, lang); // set temporarily specified ^untaint[language; + r.write_pass_lang(r.process(*vbody)); // process marking tainted with that lang + } +} + +static void _taint(Request& r, const String& method_name, Array *params) { + Pool& pool=r.pool(); + + String::Untaint_lang lang; + if(params->size()==1) + lang=String::UL_TAINTED; // mark as simply 'tainted'. useful in table:set + else { + const String& lang_name=r.process(*static_cast(params->get(0))).as_string(); + lang=static_cast( + untaint_lang_name2enum->get_int(lang_name)); + if(!lang) + PTHROW(0, 0, + &lang_name, + "invalid taint language"); + } + + { + Value& vbody=*static_cast(params->get(params->size()-1)); + // forcing [this param type] + r.fail_if_junction_(true, vbody, method_name, "body must not be junction"); + + String result(r.pool()); + result.append( + vbody.as_string(), // process marking tainted with that lang + lang, true); // force result language to specified + r.write_pass_lang(result); } } - static void _process(Request& r, const String& method_name, Array *params) { // calculate pseudo file name of processed chars @@ -60,10 +105,12 @@ static void _process(Request& r, const S strncpy(place, MAX_STRING, method_name.cstr()); #endif - VClass& self_class=*r.self->get_class(); + VStateless_class& self_class=*r.self->get_class(); { // temporary zero @main so to maybe-replace it in processed code - Temp_method temp_method(self_class, *main_method_name, 0); + Temp_method temp_method_main(self_class, *main_method_name, 0); + // temporary zero @auto so it wouldn't be auto-called in Request::use_buf + Temp_method temp_method_auto(self_class, *auto_method_name, 0); // evaluate source to process const String& source= @@ -82,21 +129,261 @@ static void _process(Request& r, const S } static void _rem(Request& r, const String& method_name, Array *params) { + // forcing ^rem{this param type} r.fail_if_junction_(false, *static_cast(params->get(0)), method_name, "body must be junction"); } -void initialize_root_class(Pool& pool, VClass& vclass) { +static void _while(Request& r, const String& method_name, Array *params) { + Pool& pool=r.pool(); + + Value& vcondition=*static_cast(params->get(0)); + // forcing ^while(this param type){} + r.fail_if_junction_(false, vcondition, + method_name, "condition must be junction"); + + Value& body=*static_cast(params->get(1)); + // forcing ^while(){this param type} + r.fail_if_junction_(false, body, + method_name, "body must be junction"); + + // while... + int endless_loop_count=0; + while(true) { + if(++endless_loop_count>=1973) // endless loop? + PTHROW(0, 0, + &method_name, + "endless loop detected"); + + bool condition= + r.process( + vcondition, + 0/*no name*/, + false/*don't intercept string*/).as_bool(); + if(!condition) // ...condition is true + break; + + // write processed body + r.write_pass_lang(r.process(body)); + } +} + +static void _use(Request& r, const String& method_name, Array *params) { + Value& vfile=*static_cast(params->get(0)); + // forcing ^rem{this param type} + r.fail_if_junction_(true, vfile, + method_name, "file name must not be junction"); + + r.use_file(r.absolute(vfile.as_string())); +} + +static void _for(Request& r, const String& method_name, Array *params) { + // ^for[i;from-number;to-number-inclusive]{code}[delim] + + Pool& pool=r.pool(); + const String& var_name=r.process(*static_cast(params->get(0))).as_string(); + int from=(int)r.process(*static_cast(params->get(1))).as_double(); + int to=(int)r.process(*static_cast(params->get(2))).as_double(); + Value& body_code=*static_cast(params->get(3)); + // forcing ^menu{this param type} + r.fail_if_junction_(false, body_code, + method_name, "body must be junction"); + Value *delim_code=params->size()==3+1+1?static_cast(params->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? + PTHROW(0, 0, + &method_name, + "endless loop detected"); + vint->set_int(i); + r.wcontext->put_element(var_name, vint); + + Value& processed_body=r.process(body_code); + if(delim_code) { // delimiter set? + const String *string=processed_body.get_string(); + if(need_delim && string && string->size()) // need delim & iteration produced string? + r.write_pass_lang(r.process(*delim_code)); + need_delim=true; + } + r.write_pass_lang(processed_body); + } +} + +static void _eval(Request& r, const String& method_name, Array *params) { + Value& expr=*static_cast(params->get(0)); + r.fail_if_junction_(false, expr, + method_name, "need expression"); + // evaluate expresion + Value *result=r.process(expr, + 0/*no name*/, + true/*don't intercept string*/).as_expr_result(); + if(params->size()==2) { + Value& fmt=*static_cast(params->get(1)); + // forcing ^format[this param type] + r.fail_if_junction_(true, fmt, + method_name, "fmt must not be junction"); + + 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); + } + 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, Array *params, + math_one_double_op_func_ptr func) { + Pool& pool=r.pool(); + Value& param=*static_cast(params->get(0)); + + // forcing ^round(this param type) + r.fail_if_junction_(false, param, + method_name, "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, Array *params) { + double_one_op(r, method_name, params, &round); +} + +static void _floor(Request& r, const String& method_name, Array *params) { + double_one_op(r, method_name, params, &floor); +} + +static void _ceiling(Request& r, const String& method_name, Array *params) { + double_one_op(r, method_name, params, &ceil); +} + +static void _abs(Request& r, const String& method_name, Array *params) { + double_one_op(r, method_name, params, &fabs); +} + +static void _sign(Request& r, const String& method_name, Array *params) { + double_one_op(r, method_name, params, &sign); +} + +/// ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} +/** + @test make params not Array but something with useful method for extracting, + with typecast and junction/not test +*/ +static void _connect(Request& r, const String& method_name, Array *params) { + Pool& pool=r.pool(); + + Value& url=*static_cast(params->get(0)); + r.fail_if_junction_(true, url, + method_name, "url must not be junction"); + + Value& body_code=*static_cast(params->get(1)); + r.fail_if_junction_(false, body_code, + method_name, "body must be junction"); + + // connect + SQL_Connection& connection=SQL_driver_manager->get_connection( + url.as_string(), r.protocol2library); + + Exception rethrow_me; + // remember/set current connection + SQL_Connection *saved_connection=r.connection; + r.connection=&connection; + // execute body + bool body_failed=false; + PTRY + 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(); + PCATCH(e) { // commit/rollback problem + rethrow_me=e; finalizer_failed=true; + } + PEND_CATCH + + // close connection [cache it] + connection.close(); + // recall current connection from remembered + 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()); +} + +// initialize + +void initialize_root_class(Pool& pool, VStateless_class& vclass) { // ^if(condition){code-when-true} // ^if(condition){code-when-true}{code-when-false} - vclass.add_native_method("if", _if, 2, 3); + vclass.add_native_method("if", Method::CT_ANY, _if, 2, 3); - // ^untaint[as-is|sql|js|html|html-typo]{code} - vclass.add_native_method("untaint", _untaint, 2, 2); + // ^untaint[as-is|uri|sql|js|html|html-typo]{code} + vclass.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); // ^process[code] - vclass.add_native_method("process", _process, 1, 1); + vclass.add_native_method("process", Method::CT_ANY, _process, 1, 1); // ^rem{code} - vclass.add_native_method("rem", _rem, 1, 1); + vclass.add_native_method("rem", Method::CT_ANY, _rem, 1, 1); + + // ^while(condition){code} + vclass.add_native_method("while", Method::CT_ANY, _while, 2, 2); + + // ^use[file] + vclass.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); + + // ^eval(expr) + // ^eval(expr)[format] + vclass.add_native_method("eval", Method::CT_ANY, _eval, 1, 2); + + + // math functions + + // ^round(expr) + vclass.add_native_method("round", Method::CT_ANY, _round, 1, 1); + + // ^floor(expr) + vclass.add_native_method("floor", Method::CT_ANY, _floor, 1, 1); + + // ^ceiling(expr) + vclass.add_native_method("ceiling", Method::CT_ANY, _ceiling, 1, 1); + + // ^abs(expr) + vclass.add_native_method("abs", Method::CT_ANY, _abs, 1, 1); + + // ^sign(expr) + vclass.add_native_method("sign", Method::CT_ANY, _sign, 1, 1); + + + // connect + + // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} + vclass.add_native_method("connect", Method::CT_ANY, _connect, 2, 2); + }