|
|
| version 1.6, 2001/04/15 13:12:17 | version 1.46, 2001/09/28 15:58:26 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser: @b ROOT parser class. | Parser: parser @b operators. |
| 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 "classes.h" | |
| #include "pa_config_includes.h" | #include "pa_config_includes.h" |
| #include <math.h> | |
| #include "pa_common.h" | #include "pa_common.h" |
| #include "pa_request.h" | #include "pa_request.h" |
| #include "_op.h" | |
| #include "pa_vint.h" | #include "pa_vint.h" |
| #include "pa_sql_connection.h" | #include "pa_sql_connection.h" |
| // limits | |
| #define MAX_LOOPS 10000 | |
| // defines | |
| #define OP_CLASS_NAME "OP" | |
| // 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) { | static void _if(Request& r, const String&, MethodParams *params) { |
| Value& condition_code=params->get(0); | Value& condition_code=params->get(0); |
| Line 24 static void _if(Request& r, const String | Line 45 static void _if(Request& r, const String |
| 0/*no name*/, | 0/*no name*/, |
| false/*don't intercept string*/).as_bool(); | false/*don't intercept string*/).as_bool(); |
| if(condition) | 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) | 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) { | static void _untaint(Request& r, const String& method_name, MethodParams *params) { |
| Line 41 static void _untaint(Request& r, const S | Line 62 static void _untaint(Request& r, const S |
| "invalid untaint language"); | "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; | Temp_lang temp_lang(r, lang); // set temporarily specified ^untaint[language; |
| r.write_pass_lang(r.process(vbody)); // process marking tainted with that lang | r.write_pass_lang(r.process(vbody)); // process marking tainted with that lang |
| Line 66 static void _taint(Request& r, const Str | Line 87 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()); | String result(r.pool()); |
| result.append( | result.append( |
| Line 86 static void _process(Request& r, const S | Line 107 static void _process(Request& r, const S |
| origin.file, 1+origin.line, | origin.file, 1+origin.line, |
| method_name.cstr()); | method_name.cstr()); |
| #else | #else |
| strncpy(place, MAX_STRING, method_name.cstr()); | strncpy(place, method_name.cstr(), MAX_STRING-1); place[MAX_STRING-1]=0; |
| #endif | #endif |
| VStateless_class& self_class=*r.self->get_class(); | VStateless_class& self_class=*r.self->get_class(); |
| Line 113 static void _process(Request& r, const S | Line 134 static void _process(Request& r, const S |
| } | } |
| static void _rem(Request& r, const String&, MethodParams *params) { | 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) { | static void _while(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& vcondition=params->get_junction(0, "condition must be expression"); | Value& vcondition=params->as_junction(0, "condition must be expression"); |
| Value& body=params->get_junction(1, "body must be code"); | Value& body=params->as_junction(1, "body must be code"); |
| // while... | // while... |
| int endless_loop_count=0; | int endless_loop_count=0; |
| while(true) { | while(true) { |
| if(++endless_loop_count>=1973) // endless loop? | if(++endless_loop_count>=MAX_LOOPS) // endless loop? |
| PTHROW(0, 0, | PTHROW(0, 0, |
| &method_name, | &method_name, |
| "endless loop detected"); | "endless loop detected"); |
| Line 144 static void _while(Request& r, const Str | Line 165 static void _while(Request& r, const Str |
| } | } |
| static void _use(Request& r, const String& method_name, MethodParams *params) { | 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())); | r.use_file(vfile.as_string()); |
| } | } |
| /// ^for[i;from-number;to-number-inclusive]{code}[delim] | |
| static void _for(Request& r, const String& method_name, MethodParams *params) { | static void _for(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| const String& var_name=r.process(params->get(0)).as_string(); | const String& var_name=r.process(params->get(0)).as_string(); |
| int from=(int)r.process(params->get(1)).as_double(); | int from=r.process(params->get(1)).as_int(); |
| int to=(int)r.process(params->get(2)).as_double(); | int to=r.process(params->get(2)).as_int(); |
| Value& body_code=params->get_junction(3, "body must be code"); | Value& body_code=params->as_junction(3, "body must be code"); |
| Value *delim_code=params->size()==3+1+1?¶ms->get(3+1):0; | Value *delim_code=params->size()==3+1+1?¶ms->get(3+1):0; |
| bool need_delim=false; | bool need_delim=false; |
| VInt *vint=new(pool) VInt(pool, 0); | VInt *vint=new(pool) VInt(pool, 0); |
| int endless_loop_count=0; | int endless_loop_count=0; |
| for(int i=from; i<=to; i++) { | 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, | PTHROW(0, 0, |
| &method_name, | &method_name, |
| "endless loop detected"); | "endless loop detected"); |
| vint->set_int(i); | vint->set_int(i); |
| r.root->put_element(var_name, vint); | r.self/*root*/->put_element(var_name, vint); |
| Value& processed_body=r.process(body_code); | Value& processed_body=r.process(body_code); |
| if(delim_code) { // delimiter set? | if(delim_code) { // delimiter set? |
| Line 179 static void _for(Request& r, const Strin | Line 199 static void _for(Request& r, const Strin |
| } | } |
| } | } |
| static void _eval(Request& r, const String&, MethodParams *params) { | static void _eval(Request& r, const String& method_name, MethodParams *params) { |
| Value& expr=params->get_junction(0, "need expression"); | Value& expr=params->as_junction(0, "need expression"); |
| // evaluate expresion | // evaluate expresion |
| Value *result=r.process(expr, | Value *result=r.process(expr, |
| 0/*no name*/, | 0/*no name YET*/, |
| true/*don't intercept string*/).as_expr_result(); | true/*don't intercept string*/).as_expr_result(); |
| if(params->size()==2) { | 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(); | Pool& pool=r.pool(); |
| String& string=*new(pool) String(pool); | String& string=*new(pool) String(pool); |
| string.APPEND_CONST(format(pool, result->as_double(), fmt.as_string().cstr())); | string.APPEND_CONST(format(pool, result->as_double(), fmt.as_string().cstr())); |
| result=new(pool) VString(string); | result=new(pool) VString(string); |
| } | } |
| result->set_name(method_name); | |
| r.write_no_lang(*result); | r.write_no_lang(*result); |
| } | } |
| typedef double (*math_one_double_op_func_ptr)(double); | static void _connect(Request& r, const String& method_name, MethodParams *params) { |
| 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(); | 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) { | Value& url=params->as_no_junction(0, "url must not be code"); |
| double_one_op(r, method_name, params, &sign); | Value& body_code=params->as_junction(1, "body must be code"); |
| } | |
| /// ^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&, MethodParams *params) { | |
| Pool& pool=r.pool(); | |
| Value& url=params->get_no_junction(0, "url must not be code"); | Table *protocol2driver_and_client= |
| Value& body_code=params->get_junction(1, "body must be code"); | static_cast<Table *>(r.classes_conf.get(r.OP.name())); |
| // connect | // connect |
| SQL_Connection& connection=SQL_driver_manager->get_connection( | SQL_Connection& connection=SQL_driver_manager->get_connection( |
| url.as_string(), r.protocol2library); | url.as_string(), method_name, protocol2driver_and_client); |
| Exception rethrow_me; | Exception rethrow_me; |
| // remember/set current connection | // remember/set current connection |
| Line 283 static void _connect(Request& r, const S | Line 267 static void _connect(Request& r, const S |
| rethrow_me.comment()); | rethrow_me.comment()); |
| } | } |
| // initialize | #ifndef DOXYGEN |
| struct Switch_data { | |
| Value *searching; | |
| Value *found; | |
| Value *_default; | |
| }; | |
| #endif | |
| 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& method_name, MethodParams *params) { | |
| Pool& pool=r.pool(); | |
| Switch_data *data=static_cast<Switch_data *>(r.classes_conf.get(*switch_data_name)); | |
| if(!data) | |
| PTHROW(0, 0, | |
| &method_name, | |
| "without switch"); | |
| int count=params->size(); | |
| Value *code=¶ms->as_junction(--count, "case result must be code"); | |
| for(int i=0; i<count; i++) { | |
| Value& value=r.process(params->get(i)); | |
| if(value.as_string() == *case_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)); | |
| void initialize_op_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", 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} | // ^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} | // ^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] | // ^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} | // ^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} | // ^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] | // ^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] | // ^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) |
| // ^eval(expr)[format] | // ^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) | // switch |
| vclass.add_native_method("round", Method::CT_ANY, _round, 1, 1); | |
| // ^floor(expr) | // ^switch[value]{cases} |
| vclass.add_native_method("floor", Method::CT_ANY, _floor, 1, 1); | 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 | |
| Methoded *MOP_create(Pool& pool) { | |
| return new(pool) MOP(pool); | |
| } | |
| /* | |
| static void element2ctypes(unsigned char *tables, | |
| Value& ctype, const String& name, | |
| unsigned char bit, | |
| int group_offset=-1, | |
| bool skip_ws=true) { | |
| Value *value=ctype.get_element(name); | |
| if(!value) | |
| return; | |
| unsigned char *ctypes_table=tables+ctypes_offset; | |
| const unsigned char *cstr= | |
| (const unsigned char *)value->as_string().cstr(String::UL_AS_IS); | |
| for(; *cstr; cstr++) { | |
| unsigned char c=*cstr; | |
| if(skip_ws && (c=='\n' || c=='\t' || c==' ')) | |
| continue; | |
| ctypes_table[c]|=bit; | |
| if(group_offset>=0) | |
| tables[cbits_offset+group_offset+c/8] |= 1 << (c%8); | |
| } | |
| } | |
| static void cstr2ctypes(unsigned char *tables, const unsigned char *cstr, | |
| unsigned char bit) { | |
| unsigned char *ctypes_table=tables+ctypes_offset; | |
| ctypes_table[0]=bit; | |
| for(; *cstr; cstr++) { | |
| unsigned char c=*cstr; | |
| ctypes_table[c]|=bit; | |
| } | |
| } | |
| static void prepare_case_tables(unsigned char *tables) { | |
| unsigned char *lcc_table=tables+lcc_offset; | |
| unsigned char *fcc_table=tables+fcc_offset; | |
| for(int i=0; i<0x100; i++) | |
| lcc_table[i]=fcc_table[i]=i; | |
| } | |
| static void element2case(unsigned char *tables, Value& ctype, const String& name) { | |
| Value *value=ctype.get_element(name); | |
| if(!value) | |
| return; | |
| unsigned char *lcc_table=tables+lcc_offset; | |
| unsigned char *fcc_table=tables+fcc_offset; | |
| const unsigned char *cstr= | |
| (const unsigned char *)value->as_string().cstr(String::UL_AS_IS); | |
| unsigned char from=0; | |
| for(; *cstr; cstr++) { | |
| unsigned char c=*cstr; | |
| if(c=='\n' || c=='\t' || c==' ') | |
| continue; | |
| if(from) { | |
| lcc_table[from]=c; | |
| fcc_table[from]=c; fcc_table[c]=from; | |
| from=0; | |
| } else | |
| from=c; | |
| } | |
| } | |
| // ^ceiling(expr) | */ |
| vclass.add_native_method("ceiling", Method::CT_ANY, _ceiling, 1, 1); | |
| // ^abs(expr) | void MOP::configure_user(Request& r) { |
| vclass.add_native_method("abs", Method::CT_ANY, _abs, 1, 1); | Pool& pool=r.pool(); |
| // ^sign(expr) | // $MAIN:SQL.drivers |
| vclass.add_native_method("sign", Method::CT_ANY, _sign, 1, 1); | 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); | |
| // connect | |
| // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s} | |
| vclass.add_native_method("connect", Method::CT_ANY, _connect, 2, 2); | |
| } | } |