--- parser3/src/classes/Attic/root.C 2001/03/28 09:01:20 1.52 +++ parser3/src/classes/Attic/root.C 2001/04/04 11:47:27 1.58 @@ -1,9 +1,11 @@ -/* - 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.52 2001/03/28 09:01:20 paf Exp $ + $Id: root.C,v 1.58 2001/04/04 11:47:27 paf Exp $ */ #include "pa_config_includes.h" @@ -13,6 +15,7 @@ #include "pa_request.h" #include "_root.h" #include "pa_vint.h" +#include "pa_sql_connection.h" static void _if(Request& r, const String& method_name, Array *params) { Value& condition_code=*static_cast(params->get(0)); @@ -39,11 +42,13 @@ static void _if(Request& r, const String } 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) - RTHROW(0, 0, + PTHROW(0, 0, &lang_name, "invalid untaint language"); @@ -59,22 +64,28 @@ static void _untaint(Request& r, const S } static void _taint(Request& r, const String& method_name, Array *params) { - 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) - RTHROW(0, 0, + 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(1)); - // forcing ^untaint[]{this param type} - r.fail_if_junction_(true, *vbody, - method_name, "body must not be junction"); + 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"); // set temporarily as-is language - Temp_lang temp_lang(r, String::UL_AS_IS); + Temp_lang temp_lang(r, String::UL_PASS_APPENDED); String result(r.pool()); result.append( r.process(*vbody).as_string(), // process marking tainted with that lang @@ -126,6 +137,8 @@ static void _rem(Request& r, const Strin } 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, @@ -140,7 +153,7 @@ static void _while(Request& r, const Str int endless_loop_count=0; while(true) { if(++endless_loop_count>=1973) // endless loop? - RTHROW(0, 0, + PTHROW(0, 0, &method_name, "endless loop detected"); @@ -184,7 +197,7 @@ static void _for(Request& r, const Strin int endless_loop_count=0; for(int i=from; i<=to; i++) { if(++endless_loop_count>=2001) // endless loop? - RTHROW(0, 0, + PTHROW(0, 0, &method_name, "endless loop detected"); vint->set_int(i); @@ -263,53 +276,115 @@ static void _sign(Request& r, const Stri 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"); + + 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|uri|sql|js|html|html-typo]{code} - vclass.add_native_method("untaint", _untaint, 2, 2); + 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", _taint, 2, 2); + 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", _while, 2, 2); + vclass.add_native_method("while", Method::CT_ANY, _while, 2, 2); // ^use[file] - vclass.add_native_method("use", _use, 1, 1); + 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", _for, 3+1, 3+1+1); + vclass.add_native_method("for", Method::CT_ANY, _for, 3+1, 3+1+1); // ^eval(expr) // ^eval(expr)[format] - vclass.add_native_method("eval", _eval, 1, 2); + vclass.add_native_method("eval", Method::CT_ANY, _eval, 1, 2); // math functions // ^round(expr) - vclass.add_native_method("round", _round, 1, 1); + vclass.add_native_method("round", Method::CT_ANY, _round, 1, 1); // ^floor(expr) - vclass.add_native_method("floor", _floor, 1, 1); + vclass.add_native_method("floor", Method::CT_ANY, _floor, 1, 1); // ^ceiling(expr) - vclass.add_native_method("ceiling", _ceiling, 1, 1); + vclass.add_native_method("ceiling", Method::CT_ANY, _ceiling, 1, 1); // ^abs(expr) - vclass.add_native_method("abs", _abs, 1, 1); + vclass.add_native_method("abs", Method::CT_ANY, _abs, 1, 1); // ^sign(expr) - vclass.add_native_method("sign", _sign, 1, 1); + 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); + }