125 lines
2.9 KiB
C++
125 lines
2.9 KiB
C++
#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;
|
|
}
|