5329852db53533bb652d0c997b7fdc55b76932b1
2 * Copyright (C) 2006-2007 Andre Noll <maan@systemlinux.org>
4 * Licensed under the GPL v2. For licencing details see COPYING.
7 /** \file dccp_recv.c paraslash's dccp receiver */
10 * based on client.c of dccp-cs-0.01.tar.bz2,
11 * (C) 2005 Ian McDonald <imcdnzl@gmail.com>
14 #include <sys/types.h>
27 #include "dccp_recv.cmdline.h"
29 /** the size of the output buffer */
30 #define DCCP_BUFSIZE 40960
33 * data specific to the dccp receiver
35 * \sa receiver receiver_node
37 struct private_dccp_recv_data
{
38 /** the file descriptor for the dccp socket */
43 static void dccp_recv_close(struct receiver_node
*rn
)
46 struct private_dccp_recv_data
*pdd
= rn
->private_data
;
48 if (pdd
&& pdd
->fd
> 0)
52 free(rn
->private_data
);
53 rn
->private_data
= NULL
;
57 static int dccp_recv_open(struct receiver_node
*rn
)
59 struct private_dccp_recv_data
*pdd
;
60 struct dccp_recv_args_info
*conf
= rn
->conf
;
61 int ret
= makesock(AF_UNSPEC
, IPPROTO_DCCP
, 0, conf
->host_arg
, conf
->port_arg
);
66 rn
->buf
= para_calloc(DCCP_BUFSIZE
);
67 rn
->private_data
= pdd
= para_calloc(sizeof(struct private_dccp_recv_data
));
70 mark_fd_nonblocking(pdd
->fd
);
74 static void dccp_shutdown(void)
79 static void *dccp_recv_parse_config(int argc
, char **argv
)
81 struct dccp_recv_args_info
*tmp
= para_calloc(sizeof(struct dccp_recv_args_info
));
83 if (!dccp_recv_cmdline_parser(argc
, argv
, tmp
))
89 static void dccp_recv_pre_select(struct sched
*s
, struct task
*t
)
91 struct receiver_node
*rn
= t
->private_data
;
92 struct private_dccp_recv_data
*pdd
= rn
->private_data
;
95 para_fd_set(pdd
->fd
, &s
->rfds
, &s
->max_fileno
);
98 static void dccp_recv_post_select(struct sched
*s
, struct task
*t
)
100 struct receiver_node
*rn
= t
->private_data
;
101 struct private_dccp_recv_data
*pdd
= rn
->private_data
;
103 t
->ret
= -E_DCCP_RECV_EOF
;
104 if (rn
->output_eof
&& *rn
->output_eof
)
107 if (!s
->select_ret
|| !FD_ISSET(pdd
->fd
, &s
->rfds
))
108 goto out
; /* nothing to do */
109 t
->ret
= -E_DCCP_OVERRUN
;
110 if (rn
->loaded
>= DCCP_BUFSIZE
)
112 t
->ret
= recv_bin_buffer(pdd
->fd
, rn
->buf
+ rn
->loaded
,
113 DCCP_BUFSIZE
- rn
->loaded
);
116 t
->ret
= -E_DCCP_RECV_EOF
;
119 rn
->loaded
+= t
->ret
;
127 * the init function of the dccp receiver
129 * \param r pointer to the receiver struct to initialize
131 * Initialize all function pointers of \a r
133 void dccp_recv_init(struct receiver
*r
)
135 r
->shutdown
= dccp_shutdown
;
136 r
->open
= dccp_recv_open
;
137 r
->close
= dccp_recv_close
;
138 r
->pre_select
= dccp_recv_pre_select
;
139 r
->post_select
= dccp_recv_post_select
;
140 r
->parse_config
= dccp_recv_parse_config
;