f113968346e19c136fad2d764e8a1a1b31422c7d
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>
24 #include "dccp_recv.cmdline.h"
26 /* needed by getaddrinfo */
27 #include <sys/types.h>
28 #include <sys/socket.h>
31 /** the size of the output buffer */
32 #define DCCP_BUFSIZE 40960
35 * data specific to the dccp receiver
37 * \sa receiver receiver_node
39 struct private_dccp_recv_data
{
40 /** the file descriptor for the dccp socket */
45 static void dccp_recv_close(struct receiver_node
*rn
)
48 struct private_dccp_recv_data
*pdd
= rn
->private_data
;
50 if (pdd
&& pdd
->fd
> 0)
54 free(rn
->private_data
);
55 rn
->private_data
= NULL
;
59 static int dccp_recv_open(struct receiver_node
*rn
)
61 struct private_dccp_recv_data
*pdd
;
62 struct dccp_recv_args_info
*conf
= rn
->conf
;
67 rn
->buf
= para_calloc(DCCP_BUFSIZE
);
68 rn
->private_data
= para_calloc(sizeof(struct private_dccp_recv_data
));
69 pdd
= rn
->private_data
;
70 ret
= dccp_get_socket();
75 tmp
= make_message("%d", conf
->port_arg
);
76 ret
= getaddrinfo(conf
->host_arg
, tmp
, NULL
, &ai
);
82 ret
= dccp_set_socket(pdd
->fd
);
85 PARA_NOTICE_LOG("connecting to %s:%d\n", conf
->host_arg
, conf
->port_arg
);
86 ret
= connect(pdd
->fd
, ai
->ai_addr
, ai
->ai_addrlen
);
89 ret
= -E_DCCP_CONNECT
;
92 mark_fd_nonblock(pdd
->fd
);
99 static void dccp_shutdown(void)
101 ; /* nothing to do */
104 static void *dccp_recv_parse_config(int argc
, char **argv
)
106 struct dccp_recv_args_info
*tmp
= para_calloc(sizeof(struct dccp_recv_args_info
));
108 if (!dccp_recv_cmdline_parser(argc
, argv
, tmp
))
114 static void dccp_recv_pre_select(struct sched
*s
, struct task
*t
)
116 struct receiver_node
*rn
= t
->private_data
;
117 struct private_dccp_recv_data
*pdd
= rn
->private_data
;
120 para_fd_set(pdd
->fd
, &s
->rfds
, &s
->max_fileno
);
123 static void dccp_recv_post_select(struct sched
*s
, struct task
*t
)
125 struct receiver_node
*rn
= t
->private_data
;
126 struct private_dccp_recv_data
*pdd
= rn
->private_data
;
128 t
->ret
= -E_DCCP_RECV_EOF
;
129 if (rn
->output_eof
&& *rn
->output_eof
)
132 if (!s
->select_ret
|| !FD_ISSET(pdd
->fd
, &s
->rfds
))
133 goto out
; /* nothing to do */
134 t
->ret
= -E_DCCP_OVERRUN
;
135 if (rn
->loaded
>= DCCP_BUFSIZE
)
137 t
->ret
= recv_bin_buffer(pdd
->fd
, rn
->buf
+ rn
->loaded
,
138 DCCP_BUFSIZE
- rn
->loaded
);
141 t
->ret
= -E_DCCP_RECV_EOF
;
144 rn
->loaded
+= t
->ret
;
152 * the init function of the dccp receiver
154 * \param r pointer to the receiver struct to initialize
156 * Initialize all function pointers of \a r
158 void dccp_recv_init(struct receiver
*r
)
160 r
->shutdown
= dccp_shutdown
;
161 r
->open
= dccp_recv_open
;
162 r
->close
= dccp_recv_close
;
163 r
->pre_select
= dccp_recv_pre_select
;
164 r
->post_select
= dccp_recv_post_select
;
165 r
->parse_config
= dccp_recv_parse_config
;