summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/Task.cc16
-rw-r--r--lib/TaskMan.cc111
-rw-r--r--lib/Variables.cc6
3 files changed, 119 insertions, 14 deletions
diff --git a/lib/Task.cc b/lib/Task.cc
index 839c003..3cac327 100644
--- a/lib/Task.cc
+++ b/lib/Task.cc
@@ -33,6 +33,22 @@ int Task::Run() {
return state;
}
+int Task::DryRun() {
+ while (state != TASK_DONE) {
+ try {
+ state = Do();
+ }
+ catch (TaskSwitch) {
+ }
+ catch (GeneralException e) {
+ cerr << _("Task ") << GetName() << _(" caused an unexpected exception during dry-run: \"") << e.GetMsg() << _("\". Terminating.\n");
+ state = TASK_DONE;
+ }
+ }
+
+ return state;
+}
+
int Task::GetState() {
return state;
}
diff --git a/lib/TaskMan.cc b/lib/TaskMan.cc
index 3169f88..4d58be9 100644
--- a/lib/TaskMan.cc
+++ b/lib/TaskMan.cc
@@ -22,6 +22,12 @@ bool TaskMan::stopped = false;
int TaskMan::number = 0;
bool TaskMan::inited = false;
+int TaskMan::event, TaskMan::eprocess, TaskMan::estatus;
+Task * TaskMan::etask;
+Handle * TaskMan::ehandle;
+
+sigset_t TaskMan::sigchildset;
+
static int got_sigchild = 0;
void taskman_sigchild(int sig) {
@@ -30,13 +36,17 @@ void taskman_sigchild(int sig) {
pid = wait(&status);
if (TaskMan::GotChild(pid, status)) {
- got_sigchild = 1;
+ got_sigchild++;
+ } else {
+ TaskMan::WaitFor(pid, 0, status);
}
+
+ cerr << "Got SIGCHILD, pid = " << pid << " and status = " << status << endl;
signal(SIGCHLD, taskman_sigchild);
}
-void taskman_sigpipe(int sig) {
- signal(SIGPIPE, taskman_sigpipe);
+void taskman_sighole(int sig) {
+ signal(sig, taskman_sighole);
}
int TaskMan::GotChild(pid_t pid, int status) {
@@ -59,7 +69,12 @@ void TaskMan::Init() throw (GeneralException) {
}
signal(SIGCHLD, taskman_sigchild);
- signal(SIGPIPE, taskman_sigpipe);
+ signal(SIGPIPE, taskman_sighole);
+ signal(SIGHUP, taskman_sighole);
+
+ sigemptyset(&sigchildset);
+ sigaddset(&sigchildset, SIGCHLD);
+ sigprocmask(SIG_BLOCK, &sigchildset, 0);
inited = true;
number = 0;
@@ -69,6 +84,26 @@ void TaskMan::Stop() {
stopped = true;
}
+int TaskMan::Event() {
+ return event;
+}
+
+Task * TaskMan::Etask() {
+ return etask;
+}
+
+Handle * TaskMan::Ehandle() {
+ return ehandle;
+}
+
+int TaskMan::Eprocess() {
+ return eprocess;
+}
+
+int TaskMan::Estatus() {
+ return estatus;
+}
+
void TaskMan::AddTask(Task * t) {
if (!inited) {
Init();
@@ -132,8 +167,19 @@ void TaskMan::WaitFor(Handle * h, Task * t, int flags) {
w4ha.push_back(w4ha_t(h, flags, t));
}
-void TaskMan::WaitFor(pid_t p, Task * t) {
- w4pr.push_back(w4pr_t(p, t));
+void TaskMan::WaitFor(pid_t pid, Task * t, int status) {
+ if (status == -1) {
+ for (vector<w4pr_t>::iterator p = w4pr.begin(); p && (p != w4pr.end()); p++) {
+ if (p->pr == pid) {
+ p->T = t;
+ p->flag = true;
+ got_sigchild++;
+ return;
+ }
+ }
+ }
+ w4pr.push_back(w4pr_t(pid, t));
+ w4pr[w4pr.size() - 1].status = status;
}
void TaskMan::WaitFor(timeval t, Task * T, int flags) {
@@ -163,19 +209,24 @@ void TaskMan::MainLoop() throw (GeneralException) {
while (!no_burst) {
no_burst = 1;
/* First, we will check for any burning task and run 'em */
+ event = E_BURST;
for (TaskList_t::iterator p = TaskList.begin(); p && (p != TaskList.end()); p++) {
Task * t = *p;
-// cerr << "-=- TaskMan: task " << t->GetName() << endl;
-
+ cerr << "-=- TaskMan: task " << t->GetName() << endl;
+
+ if (t->IsStopped()) {
+ continue;
+ }
+
if (t->GetState() == TASK_BURST) {
- cerr << "-=- TaskMan: running burning task " << t->GetName() << endl;
+// cerr << "-=- TaskMan: running burning task " << t->GetName() << endl;
t->Run();
/* if the task added some new tasks, we have to rerun the loop */
no_burst = 0;
break;
}
-
+
if (t->GetState() == TASK_DONE) {
TaskList.erase(p);
number--;
@@ -185,6 +236,10 @@ void TaskMan::MainLoop() throw (GeneralException) {
}
}
+ /* Now is time to check all the handle and enters into a wait state. */
+
+ event = E_HANDLE;
+
nfds = w4ha.size();
no_burst = 1;
@@ -223,7 +278,9 @@ void TaskMan::MainLoop() throw (GeneralException) {
}
}
- r = poll(ufsd, nfds, (no_burst) && !(Zombies.size()) ? -1: 0);
+ sigprocmask(SIG_UNBLOCK, &sigchildset, 0);
+ r = poll(ufsd, nfds, (no_burst) && !(Zombies.size()) && !(got_sigchild) ? -1: 0);
+ sigprocmask(SIG_BLOCK, &sigchildset, 0);
#else
FD_ZERO(readfds);
FD_ZERO(writefds);
@@ -241,14 +298,16 @@ void TaskMan::MainLoop() throw (GeneralException) {
FD_SET(p->ha->GetHandle(), exceptfds);
}
+ sigprocmask(SIG_UNBLOCK, &sigchildset, 0);
r = select(highest + 1, &readfds, &writefds, &exceptfds, NULL);
+ sigprocmask(SIG_BLOCK, &sigchildset, 0);
#endif
if (r < 0) {
if (errno != EINTR) {
throw GeneralException(String(_("Error during poll: ")) + strerror(errno));
}
} else if (r == 0) {
- // timeout...
+ // timeout or child.
// **FIXME**
#warning FIXME
} else {
@@ -294,6 +353,7 @@ void TaskMan::MainLoop() throw (GeneralException) {
touched = true;
+ ehandle = p->ha;
w4.T->Run();
if (w4.T->GetState() == TASK_DONE) {
@@ -316,9 +376,13 @@ void TaskMan::MainLoop() throw (GeneralException) {
#endif
}
+ /* And finally, let's clean-up all the zombies around here. */
+
int no_zombies;
no_zombies = 0;
+ event = E_TASK;
+
while (!no_zombies) {
no_zombies = 1;
while (Zombies.size()) {
@@ -331,6 +395,7 @@ void TaskMan::MainLoop() throw (GeneralException) {
if ((o = t->WaitedBy())) {
cerr << "-=- TaskMan: running task " << o->GetName() << " for task " << t->GetName() << endl;
+ etask = t;
o->Run();
if (o->GetState() == TASK_DONE) {
@@ -350,9 +415,27 @@ void TaskMan::MainLoop() throw (GeneralException) {
}
}
+ /* To end up the loop, let's recall task waiting for processes */
+
+ event = E_PROCESS;
+
if (got_sigchild) {
- // **FIXME**
-#warning FIXME
+ for (vector<w4pr_t>::iterator p = w4pr.begin(); p && (p != w4pr.end()); p++) {
+ if (p->flag) {
+ Task * t;
+ if (p->T->IsStopped()) {
+ continue;
+ }
+ eprocess = p->pr;
+ estatus = p->status;
+ cerr << "-=- TaskMan: running task " << p->T->GetName() << " for process " << p->pr << " (" << p->status << ")\n";
+ t = p->T;
+ w4pr.erase(p);
+ got_sigchild--;
+ t->Run();
+ break;
+ }
+ }
}
}
}
diff --git a/lib/Variables.cc b/lib/Variables.cc
index 274a4ed..d96cad8 100644
--- a/lib/Variables.cc
+++ b/lib/Variables.cc
@@ -5,6 +5,12 @@
Variables::Variables(int nb) : Vars(nb), nbvars(nb) { }
+Variables::Variables(const Variables & v) : nbvars(v.nbvars) {
+ for (int i = 0; i < nbvars; i++) {
+ Vars.push_back(v.Vars[i]);
+ }
+}
+
Variables::~Variables() { }
void Variables::SetTo(int i, const String & s) {
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-02 20:21:17 +0000