X-Git-Url: http://git.tuebingen.mpg.de/?a=blobdiff_plain;f=recv.c;h=f872fec8b2f5c3ffdb250ad0b06ace34a2a5a904;hb=40cbbb95e256e63e1e6958e9034311d4904604af;hp=fd82bc0fa6d998e51072284aefe1dc5d79dd3663;hpb=adbf9f09c9bf2a680c61b1ef3541c04f1cde97cf;p=paraslash.git

diff --git a/recv.c b/recv.c
index fd82bc0f..f872fec8 100644
--- a/recv.c
+++ b/recv.c
@@ -17,9 +17,13 @@
  */
 #include "para.h"
 
+#include "list.h"
+#include "sched.h"
 #include "recv.h"
 #include "recv.cmdline.h"
+#include "fd.h"
 #include "error.h"
+#include "stdout.h"
 
 struct gengetopt_args_info conf;
 
@@ -54,6 +58,7 @@ static void *parse_config(int argc, char *argv[], int *receiver_num)
 	return check_receiver_arg(conf.receiver_arg, receiver_num);
 }
 
+#if 0
 int main(int argc, char *argv[])
 {
 	int ret, eof = 0, max, r_opened = 0, receiver_num;
@@ -81,16 +86,14 @@ recv:
 	FD_ZERO(&rfds);
 	FD_ZERO(&wfds);
 	timeout.tv_sec = 0;
-	timeout.tv_usec = 1000 * 1000;
+	timeout.tv_usec = 999 * 1000;
 	max = -1;
 	ret = r->pre_select(&rn, &rfds, &wfds, &timeout);
-	max = MAX(max, ret);
+	max = PARA_MAX(max, ret);
 
-	PARA_DEBUG_LOG("timeout: %lums\n", tv2ms(&timeout));
-	ret = select(max + 1, &rfds, &wfds, NULL, &timeout);
+	PARA_DEBUG_LOG("timeout: %lums, max: %d\n", tv2ms(&timeout), max);
+	ret = para_select(max + 1, &rfds, &wfds, &timeout);
 	if (ret < 0) {
-		if (errno == EINTR || errno == EAGAIN)
-			goto recv;
 		ret = -E_RECV_SELECT;
 		goto out;
 	}
@@ -126,3 +129,64 @@ out:
 		PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
 	return ret;
 }
+#endif
+
+void rn_event_handler(struct task *t)
+{
+	PARA_NOTICE_LOG("%s\n", PARA_STRERROR(-t->ret));
+	unregister_task(t);
+}
+
+int main(int argc, char *argv[])
+{
+	int ret, r_opened = 0, receiver_num;
+	struct	receiver *r = NULL;
+	struct receiver_node rn;
+	struct stdout_task sot;
+	struct sched s;
+
+	init_sched();
+	s.default_timeout.tv_sec = 1;
+	s.default_timeout.tv_usec = 0;
+
+	memset(&rn, 0, sizeof(struct receiver_node));
+	for (ret = 0; receivers[ret].name; ret++)
+		receivers[ret].init(&receivers[ret]);
+	ret = -E_RECV_SYNTAX;
+	rn.conf = parse_config(argc, argv, &receiver_num);
+	if (!rn.conf) {
+		PARA_EMERG_LOG("%s", "parse failed\n");
+		goto out;
+	}
+	r = &receivers[receiver_num];
+	rn.receiver = r;
+	ret = r->open(&rn);
+	if (ret < 0)
+		goto out;
+	r_opened = 1;
+
+	stdout_set_defaults(&sot);
+	sot.buf = rn.buf;
+	sot.loaded = &rn.loaded;
+	sot.eof = &rn.eof;
+	register_task(&sot.task);
+
+	rn.task.private_data = &rn;
+	rn.task.pre_select = r->pre_select;
+	rn.task.post_select = r->post_select;
+	rn.task.event_handler = rn_event_handler;
+	rn.task.flags = 0;
+	sprintf(rn.task.status, "receiver node");
+	register_task(&rn.task);
+
+
+	ret = sched(&s);
+out:
+	if (r_opened)
+		r->close(&rn);
+	if (r)
+		r->shutdown();
+	if (ret < 0)
+		PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
+	return ret;
+}