--- parser3/src/classes/Attic/root.C	2001/03/08 13:42:30	1.2
+++ parser3/src/classes/Attic/root.C	2001/04/05 13:19:40	1.60
@@ -1,31 +1,389 @@
-/*
-$Id: root.C,v 1.2 2001/03/08 13:42:30 paf Exp $
+/** @file
+	Parser: @b ROOT parser class.
+
+	Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com)
+
+	Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf)
+
+	$Id: root.C,v 1.60 2001/04/05 13:19:40 paf Exp $
 */
 
+#include "pa_config_includes.h"
+#include <math.h>
+
+#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& method_name, Array *params) {
+	Value& condition_code=*static_cast<Value *>(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& then_code=*static_cast<Value *>(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& else_code=*static_cast<Value *>(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<Value *>(params->get(0))).as_string();
+	String::Untaint_lang lang=static_cast<String::Untaint_lang>(
+		untaint_lang_name2enum->get_int(lang_name));
+	if(!lang)
+		PTHROW(0, 0,
+			&lang_name,
+			"invalid untaint language");
+
+	{
+		Value *vbody=static_cast<Value *>(params->get(1));
+		// forcing ^untaint[]{this param type}
+		r.fail_if_junction_(false, *vbody, 
+			method_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
+	}
+}
+
+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<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");
+		
+		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
+	// would be something like "/some/file(4) process"
+	char place[MAX_STRING];
+#ifndef NO_STRING_ORIGIN
+	const Origin& origin=method_name.origin();
+	snprintf(place, MAX_STRING, "%s(%d) %s", 
+		origin.file, 1+origin.line,
+		method_name.cstr());
+#else
+	strncpy(place, MAX_STRING, method_name.cstr());
+#endif	
+
+	VStateless_class& self_class=*r.self->get_class();
+	{
+		// temporary zero @main so to maybe-replace it in processed code
+		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=
+			r.process(*static_cast<Value *>(params->get(0))).as_string();
+
+		// process source code, append processed methods to 'self' class
+		// 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);
+}
+
+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}{code-when-false}
+	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", 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", 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
 
-static void _if(Request& r, Array& params) {
-	bool condition=
-		r.autocalc(
-			*static_cast<Value *>(params.get(0)), 
-			false/*don't make it string*/).get_bool();
-	Value& value=r.autocalc(
-		*static_cast<Value *>(params.get(condition?1:2)), 
-		true/*make it string*/);
-	r.write(value);
-}
-
-void construct_root_class(Request& request) {
-	Pool& pool=request.pool();
-	String& IF_NAME=*new(pool) String(pool);
-	IF_NAME.APPEND_CONST("if");
-
-	Method& IF_METHOD=*new(pool) Method(pool,
-		IF_NAME,
-		3/*numbered_params_count*/,
-		0/*params_names*/, 0/*locals_names*/,
-		0/*parser_code*/, _if
-	);
+	// ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s}
+	vclass.add_native_method("connect", Method::CT_ANY, _connect, 2, 2);
 
-	request.root_class.add_method(IF_NAME, IF_METHOD);
 }