2 * Copyright (C) 2005-2009 Andre Noll <maan@systemlinux.org>
4 * Licensed under the GPL v2. For licencing details see COPYING.
7 /** \file http_recv.c paraslash's http receiver */
10 #include <sys/types.h>
20 #include "http_recv.cmdline.h"
24 #include "buffer_tree.h"
26 /** the output buffer size of the http receiver */
27 #define BUFSIZE (32 * 1024)
30 * the possible states of a http receiver node
34 enum http_recv_status
{HTTP_CONNECTED
, HTTP_SENT_GET_REQUEST
, HTTP_STREAMING
};
37 * data specific to the http receiver
39 * Each running instance of the http receiver reserves space for one such struct.
41 struct private_http_recv_data
{
45 * the current status of the http receiver node
47 * It gets initialized to \p HTTP_CONNECTED by the open function of the
50 * \sa receiver::open, receiver_node.
52 enum http_recv_status status
;
56 * the file descriptor used for receiving the http stream
58 * The pre_select function of the http receiver adds this file descriptor to
59 * the set of file decriptors which are checked for reading/writing (depending
60 * on the current status) by the select loop of the application (para_audiod or
63 * The post_select function of the http receiver uses \a fd, if ready, to
64 * establish the http connection, and updates \a status according to the new
65 * state of the connection. As soon as \a status is \p HTTP_STREAMING, \a fd is
66 * going to be only checked for reading. If data is available, it is read into
67 * the output buffer of the receiver node by post_select.
69 * \sa receiver::pre_select receiver::post_select receiver_node, http_recv_status
74 static void http_shutdown(void)
79 static char *make_request_msg(void)
81 char *ret
, *hn
= para_hostname();
82 ret
= make_message("%s1.0\nHost: %s\nUser-Agent: para_recv/%s\n\n\n",
83 HTTP_GET_MSG
, hn
, PACKAGE_VERSION
);
88 static void http_recv_pre_select(struct sched
*s
, struct task
*t
)
90 struct receiver_node
*rn
= container_of(t
, struct receiver_node
, task
);
91 struct private_http_recv_data
*phd
= rn
->private_data
;
94 if (phd
->status
== HTTP_CONNECTED
)
95 para_fd_set(phd
->fd
, &s
->wfds
, &s
->max_fileno
);
97 para_fd_set(phd
->fd
, &s
->rfds
, &s
->max_fileno
);
100 #define HTTP_RECV_MAX_PENDING (1024 * 1024)
101 #define HTTP_RECV_READ_BUF_SIZE 4096
103 static void http_recv_post_select(struct sched
*s
, struct task
*t
)
105 struct receiver_node
*rn
= container_of(t
, struct receiver_node
, task
);
106 struct private_http_recv_data
*phd
= rn
->private_data
;
107 struct http_recv_args_info
*conf
= rn
->conf
;
109 if (rn
->output_error
&& *rn
->output_error
< 0) {
110 t
->error
= *rn
->output_error
;
113 if (phd
->status
== HTTP_CONNECTED
) {
115 if (!FD_ISSET(phd
->fd
, &s
->wfds
))
117 rq
= make_request_msg();
118 PARA_INFO_LOG("sending http request\n");
119 t
->error
= send_va_buffer(phd
->fd
, "%s", rq
);
122 phd
->status
= HTTP_SENT_GET_REQUEST
;
125 if (!FD_ISSET(phd
->fd
, &s
->rfds
))
127 if (phd
->status
== HTTP_SENT_GET_REQUEST
) {
128 t
->error
= recv_pattern(phd
->fd
, HTTP_OK_MSG
, strlen(HTTP_OK_MSG
));
130 PARA_INFO_LOG("received ok msg, streaming\n");
131 phd
->status
= HTTP_STREAMING
;
135 if (conf
->buffer_tree_given
) {
137 if (btr_bytes_pending(rn
->btrn
) > HTTP_RECV_MAX_PENDING
) {
138 t
->error
= -E_HTTP_RECV_OVERRUN
;
141 buf
= para_malloc(HTTP_RECV_READ_BUF_SIZE
);
142 t
->error
= recv_bin_buffer(phd
->fd
, buf
, HTTP_RECV_READ_BUF_SIZE
);
144 t
->error
= -E_RECV_EOF
;
149 btr_add_output(buf
, t
->error
, rn
->btrn
);
152 t
->error
= -E_HTTP_RECV_OVERRUN
;
153 if (rn
->loaded
>= BUFSIZE
)
155 t
->error
= recv_bin_buffer(phd
->fd
, rn
->buf
+ rn
->loaded
,
156 BUFSIZE
- rn
->loaded
);
158 t
->error
= -E_RECV_EOF
;
161 rn
->loaded
+= t
->error
;
164 static void http_recv_close(struct receiver_node
*rn
)
166 struct private_http_recv_data
*phd
= rn
->private_data
;
169 free(rn
->private_data
);
172 static void *http_recv_parse_config(int argc
, char **argv
)
174 struct http_recv_args_info
*tmp
= para_calloc(sizeof(struct http_recv_args_info
));
176 if (!http_recv_cmdline_parser(argc
, argv
, tmp
))
182 static int http_recv_open(struct receiver_node
*rn
)
184 struct private_http_recv_data
*phd
;
185 struct http_recv_args_info
*conf
= rn
->conf
;
186 int fd
, ret
= makesock(AF_UNSPEC
, IPPROTO_TCP
, 0, conf
->host_arg
,
192 ret
= mark_fd_nonblocking(fd
);
197 rn
->buf
= para_calloc(BUFSIZE
);
198 rn
->private_data
= phd
= para_calloc(sizeof(struct private_http_recv_data
));
200 phd
->status
= HTTP_CONNECTED
;
205 * the init function of the http receiver
207 * \param r pointer to the receiver struct to initialize
209 * Just initialize all function pointers of \a r.
211 void http_recv_init(struct receiver
*r
)
213 struct http_recv_args_info dummy
;
215 http_recv_cmdline_parser_init(&dummy
);
216 r
->open
= http_recv_open
;
217 r
->close
= http_recv_close
;
218 r
->pre_select
= http_recv_pre_select
;
219 r
->post_select
= http_recv_post_select
;
220 r
->shutdown
= http_shutdown
;
221 r
->parse_config
= http_recv_parse_config
;
222 r
->help
= (struct ggo_help
) {
223 .short_help
= http_recv_args_info_help
,
224 .detailed_help
= http_recv_args_info_detailed_help
226 http_recv_cmdline_parser_free(&dummy
);