4 files changed, 71 insertions, 39 deletions

diff --git a/lib/CopyJob.cc b/lib/CopyJob.cc
index f1d3310..e46e3da 100644
--- a/lib/CopyJob.cc
+++ b/lib/CopyJob.cc
@@ -5,27 +5,29 @@ CopyJob::CopyJob(Handle * as, Handle * ad, ssize_t asiz) : s(as), d(ad), siz(asi
 
 CopyJob::~CopyJob() { }
 
-int CopyJob::Do() {
+int CopyJob::Do() throw (GeneralException) {
     int r, tr;
     
     while (!s->IsClosed() || (siz != cursiz)) {
-	if (!current) {
+	switch (current) {
+	case 0:
 	    tr = siz >= 0 ? siz - cursiz : COPY_BUFSIZ;
 	    try {
 		r = s->read(buffer, MIN(COPY_BUFSIZ, tr));
 	    }
 	    catch (IOAgain e) {
-		return TASK_ON_HOLD;
+		throw TaskSwitch();
+	    }
+	case 1:
+	    try {
+		d->write(buffer, r);
+	    }
+	    catch (IOAgain e) {
+		current = 1;
+		throw TaskSwitch();
 	    }
 	    current = 0;
 	}
-	try {
-	    d->write(buffer, r);
-	}
-	catch (IOAgain e) {
-	    current = 1;
-	    return TASK_ON_HOLD;
-	}
 	cursiz += r;
     }
 }
diff --git a/lib/ReadJob.cc b/lib/ReadJob.cc
index a63eeca..2c244ad 100644
--- a/lib/ReadJob.cc
+++ b/lib/ReadJob.cc
@@ -5,37 +5,36 @@ ReadJob::ReadJob(Handle * as, Handle * ad) : s(as), d(ad), current(0) { }
 
 ReadJob::~ReadJob() { }
 
-int ReadJob::Do() {
-    int r;
+int ReadJob::Do() throw (GeneralException) {
     String buff;
     
     cerr << "ReadJob running...\n";
     
     while (!s->IsClosed()) {
-	if (!current) {
-	    r = 0;
+	switch (current) {
+	case 0:
 	    try {
 		cerr << "Trying to read...\n";
 		*s >> buff;
 	    }
 	    catch (IOAgain e) {
 		cerr << "Suspending ReadJob to wait for reading...\n";
-		r = 1;
+		throw TaskSwitch();
 	    }
-	    if (r) Suspend();
 	    cerr << "Read some bytes...\n";
+	case 1:
+	    try {
+		*d << buff << endnl;
+	    }
+	    catch (IOAgain e) {
+		cerr << "Suspending ReadJob to wait for writing...\n";
+		current = 1;
+		throw TaskSwitch();
+	    }
+	    current = 0;
+	    cerr << "Wrote some bytes...\n";
+	    if (buff == "") return TASK_DONE;
 	}
-	r = 0;
-	try {
-	    *d << buff << endnl;
-	}
-	catch (IOAgain e) {
-	    cerr << "Suspending ReadJob to wait for writing...\n";
-	    r = 1;
-	}
-	if (r) Suspend();
-	cerr << "Wrote some bytes...\n";
-	if (buff == "") return TASK_DONE;
     }
     
     return TASK_DONE;
diff --git a/lib/Task.cc b/lib/Task.cc
index 2db8ce5..338eaba 100644
--- a/lib/Task.cc
+++ b/lib/Task.cc
@@ -5,7 +5,7 @@
 Task::Task() : state(TASK_ON_HOLD), suspended(false) {}
 Task::~Task() {}
 
-int Task::Do() {
+int Task::Do() throw (GeneralException) {
     return TASK_ON_HOLD;
 }
 
@@ -43,18 +43,18 @@ void Task::Suspend() throw (GeneralException) {
     throw TaskSwitch();
 }
 
-void WaitFor(Handle * h) {
+void Task::WaitFor(Handle * h) {
     w4ha.push_back(h);
 }
 
-void WaitFor(Task * t) {
+void Task::WaitFor(Task * t) {
     w4ta.push_back(t);
 }
 
-void WaitFor(pid_t p) {
+void Task::WaitFor(pid_t p) {
     w4pr.push_back(p);
 }
 
-void WaitFor(struct timeval t) {
+void Task::WaitFor(struct timeval t) {
     w4to.push_back(t);
 }
diff --git a/lib/TaskMan.cc b/lib/TaskMan.cc
index 9abffb1..6a38491 100644
--- a/lib/TaskMan.cc
+++ b/lib/TaskMan.cc
@@ -1,13 +1,19 @@
 #include <signal.h>
+#include <wait.h>
+#include <vector.h>
 #include "TaskMan.h"
 
 bool TaskMan::inited = false;
 
-static got_sigchild = 0;
+static int got_sigchild = 0;
+static vector<pid_t> process;
+static int nbprocess = 0;
 
 void taskman_sigchild(int sig) {
     got_sigchild = 1;
-    signal(SIGCHILD, taskman_sigchild);
+    process.push_back(wait(NULL));
+    signal(SIGCHLD, taskman_sigchild);
+    nbprocess++;
 }
 
 TaskMan::TaskMan() throw (GeneralException) {
@@ -19,7 +25,7 @@ void TaskMan::Init() throw (GeneralException) {
 	throw GeneralException("Task Manager already initialised.");
     }
     
-    signal(SIGCHILD, taskman_sigchild);
+    signal(SIGCHLD, taskman_sigchild);
     
     inited = true;
     number = 0;
@@ -37,7 +43,7 @@ int TaskMan::RemoveTask(Task * t) {
     
     for (i = 0; i < number; i++) {
 	if (TaskList[i] == t) {
-	    TaskList.erase(i);
+	    TaskList.erase(&TaskList[i]);
 	    number--;
 	    return 0;
 	}
@@ -46,14 +52,39 @@ int TaskMan::RemoveTask(Task * t) {
 }
 
 void TaskMan::MainLoop() throw (GeneralException) {
+    Task ** p, * t;
+
     while (1) {
 	if (number == 0) {
 	    throw GeneralException("TaskMan: No more task to manage.");
 	}
-
-#ifdef HAVE_POLL
 	
+	p = TaskList.begin();
+	
+	while (1) {
+	    t = *p;
+
+#ifdef HAVE_POLL	
 #else
-#endif	
+#endif
+
+	    try {
+		t->Do();
+	    }
+	    catch (TaskSwitch) {
+		continue;
+	    }
+	    if (t->GetState() == TASK_DONE) {
+		TaskList.erase(p);
+	    }
+	    
+	    if (p == TaskList.end()) {
+		break;
+	    }
+	    
+	    p++;
+	}
+
+	
     }
 }