task_register() conversion: server command task

[paraslash.git] / udp_recv.c

diff --git a/udp_recv.c b/udp_recv.c
index ca0cfc333e3233f61e7dadce080e8865755b3518..4d4b67f9b772d9d178ddbd92bc9d10fbf9f347ec 100644 (file)
--- a/udp_recv.c
+++ b/udp_recv.c
@@ -5,9 +5,14 @@
  */
 /** \file udp_recv.c Paraslash's udp receiver */
 
+#include <netinet/in.h>
 #include <regex.h>
 #include <sys/socket.h>
 #include <net/if.h>
+#include <sys/types.h>
+#include <arpa/inet.h>
+#include <sys/un.h>
+#include <netdb.h>
 
 #include "para.h"
 #include "error.h"
@@ -24,7 +29,7 @@
 
 static void udp_recv_pre_select(struct sched *s, struct task *t)
 {
-       struct receiver_node *rn = container_of(t, struct receiver_node, task);
+       struct receiver_node *rn = task_context(t);
 
        if (generic_recv_pre_select(s, t) <= 0)
                return;
@@ -51,7 +56,7 @@ static int udp_check_eof(size_t sz, struct iovec iov[2])
 
 static int udp_recv_post_select(__a_unused struct sched *s, struct task *t)
 {
-       struct receiver_node *rn = container_of(t, struct receiver_node, task);
+       struct receiver_node *rn = task_context(t);
        struct btr_node *btrn = rn->btrn;
        size_t num_bytes;
        struct iovec iov[2];