--- parser3/src/classes/Attic/root.C	2001/03/30 05:51:12	1.53
+++ 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 <paf@design.ru> (http://design.ru/paf)
 
-	$Id: root.C,v 1.53 2001/03/30 05:51:12 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<Value *>(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<Value *>(params->get(0))).as_string();
 	String::Untaint_lang lang=static_cast<String::Untaint_lang>(
 		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<Value *>(params->get(0))).as_string();
-	String::Untaint_lang lang=static_cast<String::Untaint_lang>(
-		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<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(1));
-		// forcing ^untaint[]{this param type}
-		r.fail_if_junction_(true, *vbody, 
-			method_name, "body must not be junction");
+		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_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<Value *>(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,6 +276,61 @@ 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<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");
+
+	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) {
@@ -274,7 +342,7 @@ void initialize_root_class(Pool& pool, V
 	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, 2, 2);
+	vclass.add_native_method("taint", Method::CT_ANY, _taint, 1, 2);
 
 	// ^process[code]
 	vclass.add_native_method("process", Method::CT_ANY, _process, 1, 1);
@@ -312,4 +380,11 @@ void initialize_root_class(Pool& pool, V
 
 	// ^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);
+
 }