Annotation of parser3/src/classes/op.C, revision 1.38
1.1 paf 1: /** @file
1.9 paf 2: Parser: parser @b operators.
1.1 paf 3:
4: Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com)
5:
6: Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf)
7: */
1.38 ! parser 8: static const char *RCSId="$Id: op.C,v 1.37 2001/08/06 16:18:26 parser Exp $";
1.1 paf 9:
1.13 paf 10: #include "classes.h"
1.1 paf 11: #include "pa_config_includes.h"
12: #include "pa_common.h"
13: #include "pa_request.h"
14: #include "pa_vint.h"
15: #include "pa_sql_connection.h"
16:
1.18 parser 17: // limits
18:
19: #define MAX_LOOPS 10000
20:
1.10 paf 21: // defines
22:
23: #define OP_CLASS_NAME "OP"
24:
1.11 paf 25: #define MAIN_SQL_NAME "SQL"
26: #define MAIN_SQL_DRIVERS_NAME "drivers"
27:
1.10 paf 28: // class
29:
30: class MOP : public Methoded {
31: public:
32: MOP(Pool& pool);
1.11 paf 33: public: // Methoded
1.10 paf 34: bool used_directly() { return true; }
1.11 paf 35: void configure_user(Request& r);
1.36 parser 36:
1.11 paf 37: private:
38: String main_sql_name;
39: String main_sql_drivers_name;
1.10 paf 40: };
41:
42: // methods
43:
1.5 paf 44: static void _if(Request& r, const String&, MethodParams *params) {
45: Value& condition_code=params->get(0);
1.1 paf 46:
47: bool condition=r.process(condition_code,
48: 0/*no name*/,
49: false/*don't intercept string*/).as_bool();
1.6 paf 50: if(condition)
1.34 parser 51: r.write_pass_lang(r.process(params->as_junction(1, "'then' parameter must be code")));
1.6 paf 52: else if(params->size()==3)
1.34 parser 53: r.write_pass_lang(r.process(params->as_junction(2, "'else' parameter must be code")));
1.1 paf 54: }
55:
1.5 paf 56: static void _untaint(Request& r, const String& method_name, MethodParams *params) {
1.1 paf 57: Pool& pool=r.pool();
58:
1.5 paf 59: const String& lang_name=r.process(params->get(0)).as_string();
1.1 paf 60: String::Untaint_lang lang=static_cast<String::Untaint_lang>(
61: untaint_lang_name2enum->get_int(lang_name));
62: if(!lang)
63: PTHROW(0, 0,
64: &lang_name,
65: "invalid untaint language");
66:
67: {
1.34 parser 68: Value& vbody=params->as_junction(1, "body must be code");
1.1 paf 69:
70: Temp_lang temp_lang(r, lang); // set temporarily specified ^untaint[language;
1.5 paf 71: r.write_pass_lang(r.process(vbody)); // process marking tainted with that lang
1.1 paf 72: }
73: }
74:
1.5 paf 75: static void _taint(Request& r, const String&, MethodParams *params) {
1.1 paf 76: Pool& pool=r.pool();
77:
78: String::Untaint_lang lang;
79: if(params->size()==1)
80: lang=String::UL_TAINTED; // mark as simply 'tainted'. useful in table:set
81: else {
1.3 paf 82: const String& lang_name=
1.5 paf 83: r.process(params->get(0)).as_string();
1.1 paf 84: lang=static_cast<String::Untaint_lang>(
85: untaint_lang_name2enum->get_int(lang_name));
86: if(!lang)
87: PTHROW(0, 0,
1.3 paf 88: &lang_name,
89: "invalid taint language");
1.1 paf 90: }
91:
92: {
1.34 parser 93: Value& vbody=params->as_no_junction(params->size()-1, "body must not be code");
1.1 paf 94:
95: String result(r.pool());
96: result.append(
97: vbody.as_string(), // process marking tainted with that lang
98: lang, true); // force result language to specified
99: r.write_pass_lang(result);
100: }
101: }
102:
1.5 paf 103: static void _process(Request& r, const String& method_name, MethodParams *params) {
1.1 paf 104: // calculate pseudo file name of processed chars
105: // would be something like "/some/file(4) process"
106: char place[MAX_STRING];
107: #ifndef NO_STRING_ORIGIN
108: const Origin& origin=method_name.origin();
109: snprintf(place, MAX_STRING, "%s(%d) %s",
110: origin.file, 1+origin.line,
111: method_name.cstr());
112: #else
113: strncpy(place, MAX_STRING, method_name.cstr());
114: #endif
115:
116: VStateless_class& self_class=*r.self->get_class();
117: {
118: // temporary zero @main so to maybe-replace it in processed code
119: Temp_method temp_method_main(self_class, *main_method_name, 0);
120: // temporary zero @auto so it wouldn't be auto-called in Request::use_buf
121: Temp_method temp_method_auto(self_class, *auto_method_name, 0);
122:
123: // evaluate source to process
124: const String& source=
1.5 paf 125: r.process(params->get(0)).as_string();
1.1 paf 126:
127: // process source code, append processed methods to 'self' class
128: // maybe-define new @main
129: r.use_buf(source.cstr(), place, &self_class);
130:
131: // maybe-execute @main[]
132: if(const Method *method=self_class.get_method(*main_method_name)) {
133: // execute!
134: r.execute(*method->parser_code);
135: }
136: }
137: }
138:
1.5 paf 139: static void _rem(Request& r, const String&, MethodParams *params) {
1.34 parser 140: params->as_junction(0, "body must be code");
1.1 paf 141: }
142:
1.5 paf 143: static void _while(Request& r, const String& method_name, MethodParams *params) {
1.1 paf 144: Pool& pool=r.pool();
145:
1.34 parser 146: Value& vcondition=params->as_junction(0, "condition must be expression");
147: Value& body=params->as_junction(1, "body must be code");
1.1 paf 148:
149: // while...
150: int endless_loop_count=0;
151: while(true) {
1.18 parser 152: if(++endless_loop_count>=MAX_LOOPS) // endless loop?
1.1 paf 153: PTHROW(0, 0,
154: &method_name,
155: "endless loop detected");
156:
157: bool condition=
158: r.process(
159: vcondition,
160: 0/*no name*/,
161: false/*don't intercept string*/).as_bool();
162: if(!condition) // ...condition is true
163: break;
164:
165: // write processed body
166: r.write_pass_lang(r.process(body));
167: }
168: }
169:
1.5 paf 170: static void _use(Request& r, const String& method_name, MethodParams *params) {
1.34 parser 171: Value& vfile=params->as_no_junction(0, "file name must not be code");
1.37 parser 172: r.use_file(vfile.as_string());
1.1 paf 173: }
174:
1.5 paf 175: static void _for(Request& r, const String& method_name, MethodParams *params) {
1.1 paf 176: Pool& pool=r.pool();
1.5 paf 177: const String& var_name=r.process(params->get(0)).as_string();
1.18 parser 178: int from=r.process(params->get(1)).as_int();
179: int to=r.process(params->get(2)).as_int();
1.34 parser 180: Value& body_code=params->as_junction(3, "body must be code");
1.5 paf 181: Value *delim_code=params->size()==3+1+1?¶ms->get(3+1):0;
1.1 paf 182:
183: bool need_delim=false;
184: VInt *vint=new(pool) VInt(pool, 0);
185: int endless_loop_count=0;
186: for(int i=from; i<=to; i++) {
1.18 parser 187: if(++endless_loop_count>=MAX_LOOPS) // endless loop?
1.1 paf 188: PTHROW(0, 0,
189: &method_name,
190: "endless loop detected");
191: vint->set_int(i);
1.4 paf 192: r.root->put_element(var_name, vint);
1.1 paf 193:
194: Value& processed_body=r.process(body_code);
195: if(delim_code) { // delimiter set?
196: const String *string=processed_body.get_string();
197: if(need_delim && string && string->size()) // need delim & iteration produced string?
198: r.write_pass_lang(r.process(*delim_code));
199: need_delim=true;
200: }
201: r.write_pass_lang(processed_body);
202: }
203: }
204:
1.15 paf 205: static void _eval(Request& r, const String& method_name, MethodParams *params) {
1.34 parser 206: Value& expr=params->as_junction(0, "need expression");
1.1 paf 207: // evaluate expresion
208: Value *result=r.process(expr,
1.16 paf 209: 0/*no name YET*/,
1.1 paf 210: true/*don't intercept string*/).as_expr_result();
211: if(params->size()==2) {
1.34 parser 212: Value& fmt=params->as_no_junction(1, "fmt must not be code");
1.1 paf 213:
214: Pool& pool=r.pool();
215: String& string=*new(pool) String(pool);
216: string.APPEND_CONST(format(pool, result->as_double(), fmt.as_string().cstr()));
217: result=new(pool) VString(string);
218: }
1.16 paf 219: result->set_name(method_name);
1.1 paf 220: r.write_no_lang(*result);
221: }
222:
223:
1.5 paf 224: static void _connect(Request& r, const String&, MethodParams *params) {
1.1 paf 225: Pool& pool=r.pool();
226:
1.34 parser 227: Value& url=params->as_no_junction(0, "url must not be code");
228: Value& body_code=params->as_junction(1, "body must be code");
1.1 paf 229:
1.19 parser 230: Table *protocol2driver_and_client=
1.35 parser 231: static_cast<Table *>(r.classes_conf.get(r.OP.name()));
1.11 paf 232:
1.1 paf 233: // connect
234: SQL_Connection& connection=SQL_driver_manager->get_connection(
1.19 parser 235: url.as_string(), protocol2driver_and_client);
1.1 paf 236:
237: Exception rethrow_me;
238: // remember/set current connection
239: SQL_Connection *saved_connection=r.connection;
240: r.connection=&connection;
241: // execute body
242: bool body_failed=false;
243: PTRY
244: r.write_assign_lang(r.process(body_code));
245: PCATCH(e) { // connect/process problem
246: rethrow_me=e; body_failed=true;
247: }
248: PEND_CATCH
249:
250: bool finalizer_failed=false;
251: PTRY
252: // FINALLY
253: if(body_failed)
254: connection.rollback();
255: else
256: connection.commit();
257: PCATCH(e) { // commit/rollback problem
258: rethrow_me=e; finalizer_failed=true;
259: }
260: PEND_CATCH
261:
262: // close connection [cache it]
1.21 parser 263: connection.close();
1.1 paf 264: // recall current connection from remembered
265: r.connection=saved_connection;
266:
267: if(body_failed || finalizer_failed) // were there an exception for us to rethrow?
268: PTHROW(rethrow_me.type(), rethrow_me.code(),
269: rethrow_me.problem_source(),
270: rethrow_me.comment());
271: }
272:
1.28 parser 273: struct Switch_data {
274: Value *searching;
275: Value *found;
276: Value *_default;
277: };
278:
279: static void _switch(Request& r, const String&, MethodParams *params) {
280: void *backup=r.classes_conf.get(*switch_data_name);
281: Switch_data data={&r.process(params->get(0))};
282: r.classes_conf.put(*switch_data_name, &data);
283:
1.34 parser 284: r.process(params->as_junction(1, "switch cases must be code")); // and ignore result
1.28 parser 285:
286: r.classes_conf.put(*switch_data_name, backup);
287:
288: if(Value *code=data.found ? data.found : data._default)
289: r.write_pass_lang(r.process(*code));
290: }
291:
1.38 ! parser 292: static void _case(Request& r, const String& method_name, MethodParams *params) {
! 293: Pool& pool=r.pool();
! 294:
! 295: Switch_data *data=static_cast<Switch_data *>(r.classes_conf.get(*switch_data_name));
! 296: if(!data)
! 297: PTHROW(0, 0,
! 298: &method_name,
! 299: "without switch");
1.28 parser 300:
301: int count=params->size();
1.34 parser 302: Value *code=¶ms->as_junction(--count, "case result must be code");
1.28 parser 303: for(int i=0; i<count; i++) {
304: Value& value=r.process(params->get(i));
305:
1.36 parser 306: if(value.as_string() == *case_default_value) {
1.38 ! parser 307: data->_default=code;
1.28 parser 308: break;
309: }
310:
311: bool matches;
1.38 ! parser 312: if(data->searching->is_string())
! 313: matches=data->searching->as_string() == value.as_string();
1.28 parser 314: else
1.38 ! parser 315: matches=data->searching->as_double() == value.as_double();
1.28 parser 316:
317: if(matches) {
1.38 ! parser 318: data->found=code;
1.28 parser 319: break;
320: }
321: }
322: }
323:
1.10 paf 324: // constructor
325:
1.11 paf 326: MOP::MOP(Pool& apool) : Methoded(apool),
327: main_sql_name(apool, MAIN_SQL_NAME),
328: main_sql_drivers_name(apool, MAIN_SQL_DRIVERS_NAME)
329: {
1.10 paf 330: set_name(*NEW String(pool(), OP_CLASS_NAME));
1.28 parser 331:
1.1 paf 332: // ^if(condition){code-when-true}
333: // ^if(condition){code-when-true}{code-when-false}
1.10 paf 334: add_native_method("if", Method::CT_ANY, _if, 2, 3);
1.1 paf 335:
336: // ^untaint[as-is|uri|sql|js|html|html-typo]{code}
1.10 paf 337: add_native_method("untaint", Method::CT_ANY, _untaint, 2, 2);
1.1 paf 338:
339: // ^taint[as-is|uri|sql|js|html|html-typo]{code}
1.10 paf 340: add_native_method("taint", Method::CT_ANY, _taint, 1, 2);
1.1 paf 341:
342: // ^process[code]
1.10 paf 343: add_native_method("process", Method::CT_ANY, _process, 1, 1);
1.1 paf 344:
345: // ^rem{code}
1.26 parser 346: add_native_method("rem", Method::CT_ANY, _rem, 1, 1000);
1.1 paf 347:
348: // ^while(condition){code}
1.10 paf 349: add_native_method("while", Method::CT_ANY, _while, 2, 2);
1.1 paf 350:
351: // ^use[file]
1.10 paf 352: add_native_method("use", Method::CT_ANY, _use, 1, 1);
1.1 paf 353:
354: // ^for[i;from-number;to-number-inclusive]{code}[delim]
1.10 paf 355: add_native_method("for", Method::CT_ANY, _for, 3+1, 3+1+1);
1.1 paf 356:
357: // ^eval(expr)
358: // ^eval(expr)[format]
1.10 paf 359: add_native_method("eval", Method::CT_ANY, _eval, 1, 2);
1.32 parser 360:
1.1 paf 361:
362: // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s}
1.10 paf 363: add_native_method("connect", Method::CT_ANY, _connect, 2, 2);
364:
1.28 parser 365: // switch
366:
367: // ^switch[value]{cases}
368: add_native_method("switch", Method::CT_ANY, _switch, 2, 2);
369:
370: // ^case[value]{code}
371: add_native_method("case", Method::CT_ANY, _case, 2, 1000);
1.10 paf 372: }
1.11 paf 373:
374: // constructor & configurator
1.1 paf 375:
1.10 paf 376: Methoded *MOP_create(Pool& pool) {
1.35 parser 377: return new(pool) MOP(pool);
1.11 paf 378: }
379:
380:
381: void MOP::configure_user(Request& r) {
382: Pool& pool=r.pool();
383:
384: // $MAIN:SQL.drivers
385: if(Value *sql=r.main_class->get_element(main_sql_name))
386: if(Value *element=sql->get_element(main_sql_drivers_name))
387: if(Table *protocol2library=element->get_table())
388: r.classes_conf.put(name(), protocol2library);
1.1 paf 389: }
E-mail:
![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to op.C CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [parser3project] / parser3 / src / classes