29.07.2011 - fundamental new

2011-07-29 08:17:24 +04:00
parent b21a0496cd
commit 29190ea465
49 changed files with 4704 additions and 1052 deletions
--- a/piprocess.cpp
+++ b/piprocess.cpp
@@ -0,0 +1,124 @@
+#include "piprocess.h"
+
+
+PIProcess::PIProcess(): PIThread() {
+	exit_code = -1;
+	pid = 0;
+	is_exec = false;
+	g_in = g_out = g_err = false;
+	t_in = t_out = t_err = false;
+	env = PIProcess::currentEnvironment();
+}
+
+
+PIProcess::~PIProcess() {
+	if (t_in)  f_in.remove();
+	if (t_out) f_out.remove();
+	if (t_err) f_err.remove();
+}
+
+
+void PIProcess::exec_() {
+	is_exec = false;
+	startOnce();
+	//cout << "exec wait" << endl;
+	while (!is_exec)
+		msleep(1);
+	//cout << "exec end" << endl;
+}
+
+
+void PIProcess::run() {
+	//cout << "run" << endl;
+	string str;
+	/// arguments convertion
+	char * a[args.size_s() + 1];
+	for (int i = 0; i < args.size_s(); ++i) {
+		str = args[i].stdString();
+		a[i] = new char[str.size() + 1];
+		memcpy(a[i], str.c_str(), str.size());
+		a[i][str.size()] = 0;
+		//cout << a[i] << endl;
+	}
+	a[args.size_s()] = 0;
+	/// environment convertion
+	char * e[env.size_s() + 1];
+	for (int i = 0; i < env.size_s(); ++i) {
+		str = env[i].stdString();
+		e[i] = new char[str.size() + 1];
+		memcpy(e[i], str.c_str(), str.size());
+		e[i][str.size()] = 0;
+		//cout << e[i] << endl;
+	}
+	e[env.size_s()] = 0;
+	/// files for stdin/out/err
+	t_in = t_out = t_err = false;
+	if (f_in.path().isEmpty()) {
+		f_in = PIFile::openTemporary(PIFile::New | PIFile::Read);
+		t_in = true;
+	}
+	f_in.open(PIFile::New | PIFile::Read); f_in.close();
+	if (f_out.path().isEmpty()) {
+		f_out = PIFile::openTemporary(PIFile::New | PIFile::Write);
+		t_out = true;
+	}
+	f_out.open(PIFile::New | PIFile::Write); f_out.close();
+	if (f_err.path().isEmpty()) {
+		f_err = PIFile::openTemporary(PIFile::New | PIFile::Write);
+		t_err = true;
+	}
+	f_err.open(PIFile::New | PIFile::Write); f_err.close();
+	
+	str = args.front().stdString();
+	is_exec = true;
+	pid = fork();
+	if (pid == 0) {
+		FILE * tf;
+		//cout << "exec" << endl;
+		//cout << f_out.path() << endl;
+		if (g_in)  tf = freopen(f_in.path().data(), "r", stdin);
+		if (g_out) tf = freopen(f_out.path().data(), "w", stdout);
+		if (g_err) tf = freopen(f_err.path().data(), "w", stderr);
+		if (!wd.isEmpty()) system(("cd " + wd).data());
+		if (execvpe(str.c_str(), a, e) < 0)
+			cout << "[PIProcess] \"execvpe\" error, " << errorString() << endl;
+	} else {
+		msleep(1);
+		//cout << "wait" << endl;
+		wait(&exit_code);
+		pid = 0;
+		//cout << "wait done" << endl;
+	}
+	is_exec = false;
+	for (int i = 0; i < env.size_s(); ++i)
+		delete e[i];
+	for (int i = 0; i < args.size_s(); ++i)
+		delete a[i];
+	//cout << "end" << endl;
+}
+
+
+void PIProcess::removeEnvironmentVariable(const PIString & variable) {
+	PIString s;
+	for (int i = 0; i < env.size_s(); ++i) {
+		s = env[i];
+		if (s.left(s.find("=")).trimmed() == variable) {
+			env.remove(i);
+			--i;
+		}
+	}
+}
+
+
+void PIProcess::setEnvironmentVariable(const PIString & variable, const PIString & value) {
+	PIString s, v;
+	for (int i = 0; i < env.size_s(); ++i) {
+		s = env[i];
+		v = s.left(s.find("=")).trimmed();
+		if (v == variable) {
+			env[i] = v + "=" + value;
+			return;
+		}
+	}
+	env << variable + "=" + value;
+}