|
|
| version 1.19, 2001/03/12 09:31:26 | version 1.59, 2001/04/05 08:09:17 |
|---|---|
| Line 1 | Line 1 |
| /* | /** @file |
| Parser | Parser: @b ROOT parser class. |
| 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://design.ru/paf) | Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf) |
| $Id$ | $Id$ |
| */ | */ |
| #include <string.h> | #include "pa_config_includes.h" |
| #include <math.h> | |
| #include "pa_common.h" | |
| #include "pa_request.h" | #include "pa_request.h" |
| #include "_root.h" | #include "_root.h" |
| #include "pa_vint.h" | |
| #include "pa_sql_connection.h" | |
| static void _if(Request& r, const String&, Array *params) { | static void _if(Request& r, const String& method_name, Array *params) { |
| bool condition= | Value& condition_code=*static_cast<Value *>(params->get(0)); |
| r.process( | // forcing ^if(this param type) |
| *static_cast<Value *>(params->get(0)), | r.fail_if_junction_(false, condition_code, |
| 0/*no name*/, | method_name, "condition must be junction"); |
| false/*don't intercept string*/).get_bool(); | |
| bool condition=r.process(condition_code, | |
| 0/*no name*/, | |
| false/*don't intercept string*/).as_bool(); | |
| if(condition) { | if(condition) { |
| Value& value=r.process(*static_cast<Value *>(params->get(1))); | Value& then_code=*static_cast<Value *>(params->get(1)); |
| r.wcontext->write(value, String::Untaint_lang::PASS_APPENDED); | // 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) { | } else if(params->size()==3) { |
| Value& value=r.process(*static_cast<Value *>(params->get(2))); | Value& else_code=*static_cast<Value *>(params->get(2)); |
| r.wcontext->write(value, String::Untaint_lang::PASS_APPENDED); | // 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& name, Array *params) { | static void _untaint(Request& r, const String& method_name, Array *params) { |
| Pool& pool=r.pool(); | |
| const String& lang_name=r.process(*static_cast<Value *>(params->get(0))).as_string(); | const String& lang_name=r.process(*static_cast<Value *>(params->get(0))).as_string(); |
| String::Untaint_lang lang=static_cast<String::Untaint_lang>( | String::Untaint_lang lang=static_cast<String::Untaint_lang>( |
| untaint_lang_name_to_enum->get_int(lang_name)); | untaint_lang_name2enum->get_int(lang_name)); |
| if(!lang) | if(!lang) |
| R_THROW(0, 0, | PTHROW(0, 0, |
| &lang_name, | &lang_name, |
| "invalid untaint language"); | "invalid untaint language"); |
| Temp_lang temp_lang(r, lang); | { |
| Value *value=static_cast<Value *>(params->get(1)); | Value *vbody=static_cast<Value *>(params->get(1)); |
| // forcing ^untaint[]{this param type} | // forcing ^untaint[]{this param type} |
| if(!value->get_junction()) | r.fail_if_junction_(false, *vbody, |
| R_THROW(0, 0, | method_name, "body must be junction"); |
| &name, | |
| "body must be junction"); | Temp_lang temp_lang(r, lang); // set temporarily specified ^untaint[language; |
| r.write_pass_lang(r.process(*vbody)); // process marking tainted with that lang | |
| value=&r.process(*value); | } |
| r.wcontext->write(*value, String::Untaint_lang::PASS_APPENDED); | |
| } | } |
| static void _process(Request& r, const String& name, Array *params) { | static void _taint(Request& r, const String& method_name, Array *params) { |
| Value *vsource=static_cast<Value *>(params->get(0)); | Pool& pool=r.pool(); |
| // evaluate source to process | |
| const String& source=r.process(*vsource).as_string(); | 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<Value *>(params->get(0))).as_string(); | |
| lang=static_cast<String::Untaint_lang>( | |
| untaint_lang_name2enum->get_int(lang_name)); | |
| if(!lang) | |
| PTHROW(0, 0, | |
| &lang_name, | |
| "invalid taint language"); | |
| } | |
| { | |
| Value *vbody=static_cast<Value *>(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_PASS_APPENDED); | |
| String result(r.pool()); | |
| result.append( | |
| r.process(*vbody).as_string(), // process marking tainted with that lang | |
| lang, true); // force result language to specified | |
| r.write_pass_lang(result); | |
| } | |
| } | |
| // process source code, append processed methods to 'self' class | static void _process(Request& r, const String& method_name, Array *params) { |
| // calculate pseudo file name of processed chars | |
| // would be something like "/some/file(4) process" | |
| char place[MAX_STRING]; | char place[MAX_STRING]; |
| #ifndef NO_STRING_ORIGIN | #ifndef NO_STRING_ORIGIN |
| const Origin& origin=source.origin(); | const Origin& origin=method_name.origin(); |
| snprintf(place, MAX_STRING, "%s(%d) %s", | snprintf(place, MAX_STRING, "%s(%d) %s", |
| origin.file, 1+origin.line, | origin.file, 1+origin.line, |
| name.cstr()); | method_name.cstr()); |
| #else | #else |
| strncpy(place, MAX_STRING, name.cstr()); | strncpy(place, MAX_STRING, method_name.cstr()); |
| #endif | #endif |
| VClass &self_class=*r.self->get_class(); | VStateless_class& self_class=*r.self->get_class(); |
| // temporarily zeroing @main so to maybe-replace it in processed code | { |
| Temp_method temp_method(self_class, *main_method_name, 0); | // temporary zero @main so to maybe-replace it in processed code |
| Temp_method temp_method_main(self_class, *main_method_name, 0); | |
| // maybe-define new @main | // temporary zero @auto so it wouldn't be auto-called in Request::use_buf |
| r.use_buf(source.cstr(), place, &self_class); | Temp_method temp_method_auto(self_class, *auto_method_name, 0); |
| // maybe-execute @main[] | // evaluate source to process |
| if(Value *value=r.self->get_element(*main_method_name)) { // found some 'main' element | const String& source= |
| if(Junction *junction=value->get_junction()) // it even has junction! | r.process(*static_cast<Value *>(params->get(0))).as_string(); |
| if(const Method *method=junction->method) { // and junction is method-junction! call it | |
| // execute! | // process source code, append processed methods to 'self' class |
| r.execute(*method->parser_code); | // maybe-define new @main |
| } | r.use_buf(source.cstr(), place, &self_class); |
| // maybe-execute @main[] | |
| if(const Method *method=self_class.get_method(*main_method_name)) { | |
| // execute! | |
| r.execute(*method->parser_code); | |
| } | |
| } | } |
| } | } |
| static void _rem(Request& r, const String& method_name, Array *params) { | |
| // forcing ^rem{this param type} | |
| r.fail_if_junction_(false, *static_cast<Value *>(params->get(0)), | |
| method_name, "body must be junction"); | |
| } | |
| static void _while(Request& r, const String& method_name, Array *params) { | |
| Pool& pool=r.pool(); | |
| Value& vcondition=*static_cast<Value *>(params->get(0)); | |
| // forcing ^while(this param type){} | |
| r.fail_if_junction_(false, vcondition, | |
| method_name, "condition must be junction"); | |
| Value& body=*static_cast<Value *>(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<Value *>(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<Value *>(params->get(0))).as_string(); | |
| int from=(int)r.process(*static_cast<Value *>(params->get(1))).as_double(); | |
| int to=(int)r.process(*static_cast<Value *>(params->get(2))).as_double(); | |
| Value& body_code=*static_cast<Value *>(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<Value *>(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<Value *>(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<Value *>(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<Value *>(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); | |
| } | |
| void initialize_root_class(Pool& pool, VClass& vclass) { | 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<Value *>(params->get(0)); | |
| r.fail_if_junction_(true, url, | |
| method_name, "url must not be junction"); | |
| Value& body_code=*static_cast<Value *>(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} |
| // ^if(condition){code-when-true}{code-when-false} | // ^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} | // ^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", Method::CT_ANY, _taint, 1, 2); | |
| // ^process[code] | // ^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", 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); | |
| } | } |