X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=recv.c;h=bc727de0cb65a102c8f1861d75d31c713b748a19;hp=ae8a1f663b725cbed40f763e0105fd04d2d4dcc9;hb=b01605d7062e4d1f005d5aaaaed158d8efe06d79;hpb=be9f7c9b442f7b723427a79f85774d5fa7718425

diff --git a/recv.c b/recv.c
index ae8a1f66..bc727de0 100644
--- a/recv.c
+++ b/recv.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2005-2011 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 2005-2013 Andre Noll <maan@systemlinux.org>
  *
  * Licensed under the GPL v2. For licencing details see COPYING.
  */
@@ -8,22 +8,25 @@
 
 #include <regex.h>
 #include <sys/types.h>
-#include <dirent.h>
-#include <stdbool.h>
 
 #include "para.h"
 #include "list.h"
 #include "sched.h"
 #include "ggo.h"
+#include "buffer_tree.h"
 #include "recv.h"
 #include "recv.cmdline.h"
 #include "fd.h"
 #include "string.h"
 #include "error.h"
 #include "stdout.h"
-#include "buffer_tree.h"
 #include "version.h"
 
+extern void afh_recv_init(struct receiver *r);
+#undef AFH_RECEIVER
+#define AFH_RECEIVER {.name = "afh", .init = afh_recv_init},
+DEFINE_RECEIVER_ARRAY;
+
 /** The gengetopt args info struct. */
 static struct recv_args_info conf;
 
@@ -49,17 +52,6 @@ __noreturn static void print_help_and_die(void)
 	exit(0);
 }
 
-static void *parse_config(int argc, char *argv[], int *receiver_num)
-{
-	if (recv_cmdline_parser(argc, argv, &conf))
-		return NULL;
-	HANDLE_VERSION_FLAG("recv", conf);
-	if (conf.help_given || conf.detailed_help_given)
-		print_help_and_die();
-	loglevel = get_loglevel_by_name(conf.loglevel_arg);
-	return check_receiver_arg(conf.receiver_arg, receiver_num);
-}
-
 /**
  * The main function of para_recv.
  *
@@ -79,16 +71,18 @@ int main(int argc, char *argv[])
 	struct stdout_task sot;
 	static struct sched s;
 
-	s.default_timeout.tv_sec = 1;
-	s.default_timeout.tv_usec = 0;
+	recv_cmdline_parser(argc, argv, &conf);
+	loglevel = get_loglevel_by_name(conf.loglevel_arg);
+	HANDLE_VERSION_FLAG("recv", conf);
+	recv_init();
+	if (conf.help_given || conf.detailed_help_given)
+		print_help_and_die();
 
-	memset(&sot, 0, sizeof(struct stdout_task));
 	memset(&rn, 0, sizeof(struct receiver_node));
-	recv_init();
-	ret = -E_RECV_SYNTAX;
-	rn.conf = parse_config(argc, argv, &receiver_num);
+	rn.conf = check_receiver_arg(conf.receiver_arg, &receiver_num);
 	if (!rn.conf) {
-		PARA_EMERG_LOG("parse failed\n");
+		PARA_EMERG_LOG("invalid receiver specifier\n");
+		ret = -E_RECV_SYNTAX;
 		goto out;
 	}
 	r = &receivers[receiver_num];
@@ -100,24 +94,28 @@ int main(int argc, char *argv[])
 		goto out;
 	r_opened = 1;
 
+	memset(&sot, 0, sizeof(struct stdout_task));
 	sot.btrn = btr_new_node(&(struct btr_node_description)
 		EMBRACE(.parent = rn.btrn, .name = "stdout"));
 	stdout_set_defaults(&sot);
-	register_task(&sot.task);
+	register_task(&s, &sot.task);
 
 	rn.task.pre_select = r->pre_select;
 	rn.task.post_select = r->post_select;
 	sprintf(rn.task.status, "%s", r->name);
-	register_task(&rn.task);
+	register_task(&s, &rn.task);
 
+	s.default_timeout.tv_sec = 1;
+	s.default_timeout.tv_usec = 0;
 	ret = schedule(&s);
 out:
 	if (r_opened)
 		r->close(&rn);
-	btr_free_node(rn.btrn);
-	btr_free_node(sot.btrn);
+	btr_remove_node(&rn.btrn);
+	btr_remove_node(&sot.btrn);
 	if (rn.conf)
 		r->free_config(rn.conf);
+
 	if (ret < 0)
 		PARA_ERROR_LOG("%s\n", para_strerror(-ret));
 	return ret < 0? EXIT_FAILURE : EXIT_SUCCESS;