?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Add some MIME types found in other players
[vlc.git] / src / network / httpd.c
blob5eec4a2a8c7df6a552e7750452690d1333224bef
1 /*****************************************************************************
2 * httpd.c
3 *****************************************************************************
4 * Copyright (C) 2004-2006 VLC authors and VideoLAN
5 * Copyright © 2004-2007 Rémi Denis-Courmont
6 * $Id$
7 *
8 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
9 * Rémi Denis-Courmont <rem # videolan.org>
10 *
11 * This program is free software; you can redistribute it and/or modify it
12 * under the terms of the GNU Lesser General Public License as published by
13 * the Free Software Foundation; either version 2.1 of the License, or
14 * (at your option) any later version.
15 *
16 * This program is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 * GNU Lesser General Public License for more details.
20 *
21 * You should have received a copy of the GNU Lesser General Public License
22 * along with this program; if not, write to the Free Software Foundation,
23 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
24 *****************************************************************************/
25
26 #ifdef HAVE_CONFIG_H
27 # include "config.h"
28 #endif
29
30 #include <vlc_common.h>
31 #include <vlc_httpd.h>
32
33 #include <assert.h>
34
35 #include <vlc_network.h>
36 #include <vlc_tls.h>
37 #include <vlc_strings.h>
38 #include <vlc_rand.h>
39 #include <vlc_charset.h>
40 #include <vlc_url.h>
41 #include <vlc_mime.h>
42 #include <vlc_block.h>
43 #include "../libvlc.h"
44
45 #include <string.h>
46 #include <errno.h>
47 #include <unistd.h>
48
49 #ifdef HAVE_POLL
50 # include <poll.h>
51 #endif
52
53 #if defined(_WIN32)
54 # include <winsock2.h>
55 #else
56 # include <sys/socket.h>
57 #endif
58
59 #if defined(_WIN32)
60 /* We need HUGE buffer otherwise TCP throughput is very limited */
61 #define HTTPD_CL_BUFSIZE 1000000
62 #else
63 #define HTTPD_CL_BUFSIZE 10000
64 #endif
65
66 static void httpd_ClientDestroy(httpd_client_t *cl);
67 static void httpd_AppendData(httpd_stream_t *stream, uint8_t *p_data, int i_data);
68
69 /* each host run in his own thread */
70 struct httpd_host_t
71 {
72 VLC_COMMON_MEMBERS
73
74 /* ref count */
75 unsigned i_ref;
76
77 /* address/port and socket for listening at connections */
78 int *fds;
79 unsigned nfd;
80 unsigned port;
81
82 vlc_thread_t thread;
83 vlc_mutex_t lock;
84 vlc_cond_t wait;
85
86 /* all registered url (becarefull that 2 httpd_url_t could point at the same url)
87 * This will slow down the url research but make my live easier
88 * All url will have their cb trigger, but only the first one can answer
89 * */
90 int i_url;
91 httpd_url_t **url;
92
93 int i_client;
94 httpd_client_t **client;
95
96 /* TLS data */
97 vlc_tls_creds_t *p_tls;
98 };
99
100
101 struct httpd_url_t
102 {
103 httpd_host_t *host;
104
105 vlc_mutex_t lock;
106
107 char *psz_url;
108 char *psz_user;
109 char *psz_password;
110
111 struct
112 {
113 httpd_callback_t cb;
114 httpd_callback_sys_t *p_sys;
115 } catch[HTTPD_MSG_MAX];
116 };
117
118 /* status */
119 enum
120 {
121 HTTPD_CLIENT_RECEIVING,
122 HTTPD_CLIENT_RECEIVE_DONE,
123
124 HTTPD_CLIENT_SENDING,
125 HTTPD_CLIENT_SEND_DONE,
126
127 HTTPD_CLIENT_WAITING,
128
129 HTTPD_CLIENT_DEAD,
130
131 HTTPD_CLIENT_TLS_HS_IN,
132 HTTPD_CLIENT_TLS_HS_OUT
133 };
134
135 /* mode */
136 enum
137 {
138 HTTPD_CLIENT_FILE, /* default */
139 HTTPD_CLIENT_STREAM, /* regulary get data from cb */
140 };
141
142 struct httpd_client_t
143 {
144 httpd_url_t *url;
145
146 int i_ref;
147
148 int fd;
149
150 bool b_stream_mode;
151 uint8_t i_state;
152
153 mtime_t i_activity_date;
154 mtime_t i_activity_timeout;
155
156 /* buffer for reading header */
157 int i_buffer_size;
158 int i_buffer;
159 uint8_t *p_buffer;
160
161 /*
162 * If waiting for a keyframe, this is the position (in bytes) of the
163 * last keyframe the stream saw before this client connected.
164 * Otherwise, -1.
165 */
166 int64_t i_keyframe_wait_to_pass;
167
168 /* */
169 httpd_message_t query; /* client -> httpd */
170 httpd_message_t answer; /* httpd -> client */
171
172 /* TLS data */
173 vlc_tls_t *p_tls;
174 };
175
176
177 /*****************************************************************************
178 * Various functions
179 *****************************************************************************/
180 static const char *httpd_ReasonFromCode(unsigned i_code)
181 {
182 typedef struct
183 {
184 unsigned i_code;
185 const char psz_reason[36];
186 } http_status_info;
187
188 static const http_status_info http_reason[] =
189 {
190 /*{ 100, "Continue" },
191 { 101, "Switching Protocols" },*/
192 { 200, "OK" },
193 /*{ 201, "Created" },
194 { 202, "Accepted" },
195 { 203, "Non-authoritative information" },
196 { 204, "No content" },
197 { 205, "Reset content" },
198 { 206, "Partial content" },
199 { 250, "Low on storage space" },
200 { 300, "Multiple choices" },*/
201 { 301, "Moved permanently" },
202 /*{ 302, "Moved temporarily" },
203 { 303, "See other" },
204 { 304, "Not modified" },
205 { 305, "Use proxy" },
206 { 307, "Temporary redirect" },
207 { 400, "Bad request" },*/
208 { 401, "Unauthorized" },
209 /*{ 402, "Payment Required" },*/
210 { 403, "Forbidden" },
211 { 404, "Not found" },
212 { 405, "Method not allowed" },
213 /*{ 406, "Not acceptable" },
214 { 407, "Proxy authentication required" },
215 { 408, "Request time-out" },
216 { 409, "Conflict" },
217 { 410, "Gone" },
218 { 411, "Length required" },
219 { 412, "Precondition failed" },
220 { 413, "Request entity too large" },
221 { 414, "Request-URI too large" },
222 { 415, "Unsupported media Type" },
223 { 416, "Requested range not satisfiable" },
224 { 417, "Expectation failed" },
225 { 451, "Parameter not understood" },
226 { 452, "Conference not found" },
227 { 453, "Not enough bandwidth" },*/
228 { 454, "Session not found" },
229 { 455, "Method not valid in this State" },
230 { 456, "Header field not valid for resource" },
231 { 457, "Invalid range" },
232 /*{ 458, "Read-only parameter" },*/
233 { 459, "Aggregate operation not allowed" },
234 { 460, "Non-aggregate operation not allowed" },
235 { 461, "Unsupported transport" },
236 /*{ 462, "Destination unreachable" },*/
237 { 500, "Internal server error" },
238 { 501, "Not implemented" },
239 /*{ 502, "Bad gateway" },*/
240 { 503, "Service unavailable" },
241 /*{ 504, "Gateway time-out" },*/
242 { 505, "Protocol version not supported" },
243 { 551, "Option not supported" },
244 { 999, "" }
245 };
246
247 static const char psz_fallback_reason[5][16] = {
248 "Continue", "OK", "Found", "Client error", "Server error"
249 };
250
251 assert((i_code >= 100) && (i_code <= 599));
252
253 const http_status_info *p = http_reason;
254 while (i_code < p->i_code)
255 p++;
256
257 if (p->i_code == i_code)
258 return p->psz_reason;
259
260 return psz_fallback_reason[(i_code / 100) - 1];
261 }
262
263 static size_t httpd_HtmlError (char **body, int code, const char *url)
264 {
265 const char *errname = httpd_ReasonFromCode (code);
266 assert (errname);
267
268 char *url_Encoded = vlc_xml_encode (url ? url : "");
269
270 int res = asprintf (body,
271 "<?xml version=\"1.0\" encoding=\"ascii\" ?>\n"
272 "<!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Strict//EN\""
273 " \"http://www.w3.org/TR/xhtml10/DTD/xhtml10strict.dtd\">\n"
274 "<html lang=\"en\">\n"
275 "<head>\n"
276 "<title>%s</title>\n"
277 "</head>\n"
278 "<body>\n"
279 "<h1>%d %s%s%s%s</h1>\n"
280 "<hr />\n"
281 "<a href=\"http://www.videolan.org\">VideoLAN</a>\n"
282 "</body>\n"
283 "</html>\n", errname, code, errname,
284 (url_Encoded ? " (" : ""), (url_Encoded ? url_Encoded : ""), (url_Encoded ? ")" : ""));
285
286 free (url_Encoded);
287
288 if (res == -1) {
289 *body = NULL;
290 return 0;
291 }
292
293 return (size_t)res;
294 }
295
296
297 /*****************************************************************************
298 * High Level Functions: httpd_file_t
299 *****************************************************************************/
300 struct httpd_file_t
301 {
302 httpd_url_t *url;
303 httpd_file_callback_t pf_fill;
304 httpd_file_sys_t *p_sys;
305 char mime[1];
306 };
307
308 static int
309 httpd_FileCallBack(httpd_callback_sys_t *p_sys, httpd_client_t *cl,
310 httpd_message_t *answer, const httpd_message_t *query)
311 {
312 httpd_file_t *file = (httpd_file_t*)p_sys;
313 uint8_t **pp_body, *p_body; const char *psz_connection;
314 int *pi_body, i_body;
315
316 if (!answer || !query )
317 return VLC_SUCCESS;
318
319 answer->i_proto = HTTPD_PROTO_HTTP;
320 answer->i_version= 1;
321 answer->i_type = HTTPD_MSG_ANSWER;
322
323 answer->i_status = 200;
324
325 httpd_MsgAdd(answer, "Content-type", "%s", file->mime);
326 httpd_MsgAdd(answer, "Cache-Control", "%s", "no-cache");
327
328 if (query->i_type != HTTPD_MSG_HEAD) {
329 pp_body = &answer->p_body;
330 pi_body = &answer->i_body;
331 } else {
332 /* The file still needs to be executed. */
333 p_body = NULL;
334 i_body = 0;
335 pp_body = &p_body;
336 pi_body = &i_body;
337 }
338
339 if (query->i_type == HTTPD_MSG_POST) {
340 /* msg_Warn not supported */
341 }
342
343 uint8_t *psz_args = query->psz_args;
344 file->pf_fill(file->p_sys, file, psz_args, pp_body, pi_body);
345
346 if (query->i_type == HTTPD_MSG_HEAD)
347 free(p_body);
348
349 /* We respect client request */
350 psz_connection = httpd_MsgGet(&cl->query, "Connection");
351 if (psz_connection)
352 httpd_MsgAdd(answer, "Connection", "%s", psz_connection);
353
354 httpd_MsgAdd(answer, "Content-Length", "%d", answer->i_body);
355
356 return VLC_SUCCESS;
357 }
358
359 httpd_file_t *httpd_FileNew(httpd_host_t *host,
360 const char *psz_url, const char *psz_mime,
361 const char *psz_user, const char *psz_password,
362 httpd_file_callback_t pf_fill,
363 httpd_file_sys_t *p_sys)
364 {
365 const char *mime = psz_mime;
366 if (mime == NULL || mime[0] == '\0')
367 mime = vlc_mime_Ext2Mime(psz_url);
368
369 size_t mimelen = strlen(mime);
370 httpd_file_t *file = malloc(sizeof(*file) + mimelen);
371 if (unlikely(file == NULL))
372 return NULL;
373
374 file->url = httpd_UrlNew(host, psz_url, psz_user, psz_password);
375 if (!file->url) {
376 free(file);
377 return NULL;
378 }
379
380 file->pf_fill = pf_fill;
381 file->p_sys = p_sys;
382 memcpy(file->mime, mime, mimelen + 1);
383
384 httpd_UrlCatch(file->url, HTTPD_MSG_HEAD, httpd_FileCallBack,
385 (httpd_callback_sys_t*)file);
386 httpd_UrlCatch(file->url, HTTPD_MSG_GET, httpd_FileCallBack,
387 (httpd_callback_sys_t*)file);
388 httpd_UrlCatch(file->url, HTTPD_MSG_POST, httpd_FileCallBack,
389 (httpd_callback_sys_t*)file);
390
391 return file;
392 }
393
394 httpd_file_sys_t *httpd_FileDelete(httpd_file_t *file)
395 {
396 httpd_file_sys_t *p_sys = file->p_sys;
397
398 httpd_UrlDelete(file->url);
399 free(file);
400 return p_sys;
401 }
402
403 /*****************************************************************************
404 * High Level Functions: httpd_handler_t (for CGIs)
405 *****************************************************************************/
406 struct httpd_handler_t
407 {
408 httpd_url_t *url;
409
410 httpd_handler_callback_t pf_fill;
411 httpd_handler_sys_t *p_sys;
412
413 };
414
415 static int
416 httpd_HandlerCallBack(httpd_callback_sys_t *p_sys, httpd_client_t *cl,
417 httpd_message_t *answer, const httpd_message_t *query)
418 {
419 httpd_handler_t *handler = (httpd_handler_t*)p_sys;
420 char psz_remote_addr[NI_MAXNUMERICHOST];
421
422 if (!answer || !query)
423 return VLC_SUCCESS;
424
425 answer->i_proto = HTTPD_PROTO_NONE;
426 answer->i_type = HTTPD_MSG_ANSWER;
427
428 /* We do it ourselves, thanks */
429 answer->i_status = 0;
430
431 if (!httpd_ClientIP(cl, psz_remote_addr, NULL))
432 *psz_remote_addr = '\0';
433
434 uint8_t *psz_args = query->psz_args;
435 handler->pf_fill(handler->p_sys, handler, query->psz_url, psz_args,
436 query->i_type, query->p_body, query->i_body,
437 psz_remote_addr, NULL,
438 &answer->p_body, &answer->i_body);
439
440 if (query->i_type == HTTPD_MSG_HEAD) {
441 char *p = (char *)answer->p_body;
442
443 /* Looks for end of header (i.e. one empty line) */
444 while ((p = strchr(p, '\r')))
445 if (p[1] == '\n' && p[2] == '\r' && p[3] == '\n')
446 break;
447
448 if (p) {
449 p[4] = '\0';
450 answer->i_body = strlen((char*)answer->p_body) + 1;
451 answer->p_body = xrealloc(answer->p_body, answer->i_body);
452 }
453 }
454
455 if (strncmp((char *)answer->p_body, "HTTP/1.", 7)) {
456 int i_status, i_headers;
457 char *psz_headers, *psz_new;
458 const char *psz_status;
459
460 if (!strncmp((char *)answer->p_body, "Status: ", 8)) {
461 /* Apache-style */
462 i_status = strtol((char *)&answer->p_body[8], &psz_headers, 0);
463 if (*psz_headers == '\r' || *psz_headers == '\n') psz_headers++;
464 if (*psz_headers == '\n') psz_headers++;
465 i_headers = answer->i_body - (psz_headers - (char *)answer->p_body);
466 } else {
467 i_status = 200;
468 psz_headers = (char *)answer->p_body;
469 i_headers = answer->i_body;
470 }
471
472 psz_status = httpd_ReasonFromCode(i_status);
473 answer->i_body = sizeof("HTTP/1.0 xxx \r\n")
474 + strlen(psz_status) + i_headers - 1;
475 psz_new = (char *)xmalloc(answer->i_body + 1);
476 sprintf(psz_new, "HTTP/1.0 %03d %s\r\n", i_status, psz_status);
477 memcpy(&psz_new[strlen(psz_new)], psz_headers, i_headers);
478 free(answer->p_body);
479 answer->p_body = (uint8_t *)psz_new;
480 }
481
482 return VLC_SUCCESS;
483 }
484
485 httpd_handler_t *httpd_HandlerNew(httpd_host_t *host, const char *psz_url,
486 const char *psz_user,
487 const char *psz_password,
488 httpd_handler_callback_t pf_fill,
489 httpd_handler_sys_t *p_sys)
490 {
491 httpd_handler_t *handler = malloc(sizeof(*handler));
492 if (!handler)
493 return NULL;
494
495 handler->url = httpd_UrlNew(host, psz_url, psz_user, psz_password);
496 if (!handler->url) {
497 free(handler);
498 return NULL;
499 }
500
501 handler->pf_fill = pf_fill;
502 handler->p_sys = p_sys;
503
504 httpd_UrlCatch(handler->url, HTTPD_MSG_HEAD, httpd_HandlerCallBack,
505 (httpd_callback_sys_t*)handler);
506 httpd_UrlCatch(handler->url, HTTPD_MSG_GET, httpd_HandlerCallBack,
507 (httpd_callback_sys_t*)handler);
508 httpd_UrlCatch(handler->url, HTTPD_MSG_POST, httpd_HandlerCallBack,
509 (httpd_callback_sys_t*)handler);
510
511 return handler;
512 }
513
514 httpd_handler_sys_t *httpd_HandlerDelete(httpd_handler_t *handler)
515 {
516 httpd_handler_sys_t *p_sys = handler->p_sys;
517 httpd_UrlDelete(handler->url);
518 free(handler);
519 return p_sys;
520 }
521
522 /*****************************************************************************
523 * High Level Functions: httpd_redirect_t
524 *****************************************************************************/
525 struct httpd_redirect_t
526 {
527 httpd_url_t *url;
528 char dst[1];
529 };
530
531 static int httpd_RedirectCallBack(httpd_callback_sys_t *p_sys,
532 httpd_client_t *cl, httpd_message_t *answer,
533 const httpd_message_t *query)
534 {
535 httpd_redirect_t *rdir = (httpd_redirect_t*)p_sys;
536 char *p_body;
537 (void)cl;
538
539 if (!answer || !query)
540 return VLC_SUCCESS;
541
542 answer->i_proto = HTTPD_PROTO_HTTP;
543 answer->i_version= 1;
544 answer->i_type = HTTPD_MSG_ANSWER;
545 answer->i_status = 301;
546
547 answer->i_body = httpd_HtmlError (&p_body, 301, rdir->dst);
548 answer->p_body = (unsigned char *)p_body;
549
550 /* XXX check if it's ok or we need to set an absolute url */
551 httpd_MsgAdd(answer, "Location", "%s", rdir->dst);
552
553 httpd_MsgAdd(answer, "Content-Length", "%d", answer->i_body);
554
555 return VLC_SUCCESS;
556 }
557
558 httpd_redirect_t *httpd_RedirectNew(httpd_host_t *host, const char *psz_url_dst,
559 const char *psz_url_src)
560 {
561 size_t dstlen = strlen(psz_url_dst);
562
563 httpd_redirect_t *rdir = malloc(sizeof(*rdir) + dstlen);
564 if (unlikely(rdir == NULL))
565 return NULL;
566
567 rdir->url = httpd_UrlNew(host, psz_url_src, NULL, NULL);
568 if (!rdir->url) {
569 free(rdir);
570 return NULL;
571 }
572 memcpy(rdir->dst, psz_url_dst, dstlen + 1);
573
574 /* Redirect apply for all HTTP request and RTSP DESCRIBE resquest */
575 httpd_UrlCatch(rdir->url, HTTPD_MSG_HEAD, httpd_RedirectCallBack,
576 (httpd_callback_sys_t*)rdir);
577 httpd_UrlCatch(rdir->url, HTTPD_MSG_GET, httpd_RedirectCallBack,
578 (httpd_callback_sys_t*)rdir);
579 httpd_UrlCatch(rdir->url, HTTPD_MSG_POST, httpd_RedirectCallBack,
580 (httpd_callback_sys_t*)rdir);
581 httpd_UrlCatch(rdir->url, HTTPD_MSG_DESCRIBE, httpd_RedirectCallBack,
582 (httpd_callback_sys_t*)rdir);
583
584 return rdir;
585 }
586 void httpd_RedirectDelete(httpd_redirect_t *rdir)
587 {
588 httpd_UrlDelete(rdir->url);
589 free(rdir);
590 }
591
592 /*****************************************************************************
593 * High Level Funtions: httpd_stream_t
594 *****************************************************************************/
595 struct httpd_stream_t
596 {
597 vlc_mutex_t lock;
598 httpd_url_t *url;
599
600 char *psz_mime;
601
602 /* Header to send as first packet */
603 uint8_t *p_header;
604 int i_header;
605
606 /* Some muxes, in particular the avformat mux, can mark given blocks
607 * as keyframes, to ensure that the stream starts with one.
608 * (This is particularly important for WebM streaming to certain
609 * browsers.) Store if we've ever seen any such keyframe blocks,
610 * and if so, the byte position of the start of the last one. */
611 bool b_has_keyframes;
612 int64_t i_last_keyframe_seen_pos;
613
614 /* circular buffer */
615 int i_buffer_size; /* buffer size, can't be reallocated smaller */
616 uint8_t *p_buffer; /* buffer */
617 int64_t i_buffer_pos; /* absolute position from beginning */
618 int64_t i_buffer_last_pos; /* a new connection will start with that */
619
620 /* custom headers */
621 size_t i_http_headers;
622 httpd_header * p_http_headers;
623 };
624
625 static int httpd_StreamCallBack(httpd_callback_sys_t *p_sys,
626 httpd_client_t *cl, httpd_message_t *answer,
627 const httpd_message_t *query)
628 {
629 httpd_stream_t *stream = (httpd_stream_t*)p_sys;
630
631 if (!answer || !query || !cl)
632 return VLC_SUCCESS;
633
634 if (answer->i_body_offset > 0) {
635 int i_pos;
636
637 if (answer->i_body_offset >= stream->i_buffer_pos)
638 return VLC_EGENERIC; /* wait, no data available */
639
640 if (cl->i_keyframe_wait_to_pass >= 0) {
641 if (stream->i_last_keyframe_seen_pos <= cl->i_keyframe_wait_to_pass)
642 /* still waiting for the next keyframe */
643 return VLC_EGENERIC;
644
645 /* seek to the new keyframe */
646 answer->i_body_offset = stream->i_last_keyframe_seen_pos;
647 cl->i_keyframe_wait_to_pass = -1;
648 }
649
650 if (answer->i_body_offset + stream->i_buffer_size < stream->i_buffer_pos)
651 answer->i_body_offset = stream->i_buffer_last_pos; /* this client isn't fast enough */
652
653 i_pos = answer->i_body_offset % stream->i_buffer_size;
654 int64_t i_write = stream->i_buffer_pos - answer->i_body_offset;
655
656 if (i_write > HTTPD_CL_BUFSIZE)
657 i_write = HTTPD_CL_BUFSIZE;
658 else if (i_write <= 0)
659 return VLC_EGENERIC; /* wait, no data available */
660
661 /* Don't go past the end of the circular buffer */
662 i_write = __MIN(i_write, stream->i_buffer_size - i_pos);
663
664 /* using HTTPD_MSG_ANSWER -> data available */
665 answer->i_proto = HTTPD_PROTO_HTTP;
666 answer->i_version= 0;
667 answer->i_type = HTTPD_MSG_ANSWER;
668
669 answer->i_body = i_write;
670 answer->p_body = xmalloc(i_write);
671 memcpy(answer->p_body, &stream->p_buffer[i_pos], i_write);
672
673 answer->i_body_offset += i_write;
674
675 return VLC_SUCCESS;
676 } else {
677 answer->i_proto = HTTPD_PROTO_HTTP;
678 answer->i_version= 0;
679 answer->i_type = HTTPD_MSG_ANSWER;
680
681 answer->i_status = 200;
682
683 bool b_has_content_type = false;
684 bool b_has_cache_control = false;
685
686 vlc_mutex_lock(&stream->lock);
687 for (size_t i = 0; i < stream->i_http_headers; i++)
688 if (strncasecmp(stream->p_http_headers[i].name, "Content-Length", 14)) {
689 httpd_MsgAdd(answer, stream->p_http_headers[i].name, "%s",
690 stream->p_http_headers[i].value);
691
692 if (!strncasecmp(stream->p_http_headers[i].name, "Content-Type", 12))
693 b_has_content_type = true;
694 else if (!strncasecmp(stream->p_http_headers[i].name, "Cache-Control", 13))
695 b_has_cache_control = true;
696 }
697 vlc_mutex_unlock(&stream->lock);
698
699 if (query->i_type != HTTPD_MSG_HEAD) {
700 cl->b_stream_mode = true;
701 vlc_mutex_lock(&stream->lock);
702 /* Send the header */
703 if (stream->i_header > 0) {
704 answer->i_body = stream->i_header;
705 answer->p_body = xmalloc(stream->i_header);
706 memcpy(answer->p_body, stream->p_header, stream->i_header);
707 }
708 answer->i_body_offset = stream->i_buffer_last_pos;
709 if (stream->b_has_keyframes)
710 cl->i_keyframe_wait_to_pass = stream->i_last_keyframe_seen_pos;
711 else
712 cl->i_keyframe_wait_to_pass = -1;
713 vlc_mutex_unlock(&stream->lock);
714 } else {
715 httpd_MsgAdd(answer, "Content-Length", "0");
716 answer->i_body_offset = 0;
717 }
718
719 /* FIXME: move to http access_output */
720 if (!strcmp(stream->psz_mime, "video/x-ms-asf-stream")) {
721 bool b_xplaystream = false;
722
723 httpd_MsgAdd(answer, "Content-type", "application/octet-stream");
724 httpd_MsgAdd(answer, "Server", "Cougar 4.1.0.3921");
725 httpd_MsgAdd(answer, "Pragma", "no-cache");
726 httpd_MsgAdd(answer, "Pragma", "client-id=%lu",
727 vlc_mrand48()&0x7fff);
728 httpd_MsgAdd(answer, "Pragma", "features=\"broadcast\"");
729
730 /* Check if there is a xPlayStrm=1 */
731 for (size_t i = 0; i < query->i_headers; i++)
732 if (!strcasecmp(query->p_headers[i].name, "Pragma") &&
733 strstr(query->p_headers[i].value, "xPlayStrm=1"))
734 b_xplaystream = true;
735
736 if (!b_xplaystream)
737 answer->i_body_offset = 0;
738 } else if (!b_has_content_type)
739 httpd_MsgAdd(answer, "Content-type", "%s", stream->psz_mime);
740
741 if (!b_has_cache_control)
742 httpd_MsgAdd(answer, "Cache-Control", "no-cache");
743 return VLC_SUCCESS;
744 }
745 }
746
747 httpd_stream_t *httpd_StreamNew(httpd_host_t *host,
748 const char *psz_url, const char *psz_mime,
749 const char *psz_user, const char *psz_password)
750 {
751 httpd_stream_t *stream = malloc(sizeof(*stream));
752 if (!stream)
753 return NULL;
754
755 stream->url = httpd_UrlNew(host, psz_url, psz_user, psz_password);
756 if (!stream->url) {
757 free(stream);
758 return NULL;
759 }
760
761 vlc_mutex_init(&stream->lock);
762 if (psz_mime == NULL || psz_mime[0] == '\0')
763 psz_mime = vlc_mime_Ext2Mime(psz_url);
764 stream->psz_mime = xstrdup(psz_mime);
765
766 stream->i_header = 0;
767 stream->p_header = NULL;
768 stream->i_buffer_size = 5000000; /* 5 Mo per stream */
769 stream->p_buffer = xmalloc(stream->i_buffer_size);
770 /* We set to 1 to make life simpler
771 * (this way i_body_offset can never be 0) */
772 stream->i_buffer_pos = 1;
773 stream->i_buffer_last_pos = 1;
774 stream->b_has_keyframes = false;
775 stream->i_last_keyframe_seen_pos = 0;
776 stream->i_http_headers = 0;
777 stream->p_http_headers = NULL;
778
779 httpd_UrlCatch(stream->url, HTTPD_MSG_HEAD, httpd_StreamCallBack,
780 (httpd_callback_sys_t*)stream);
781 httpd_UrlCatch(stream->url, HTTPD_MSG_GET, httpd_StreamCallBack,
782 (httpd_callback_sys_t*)stream);
783 httpd_UrlCatch(stream->url, HTTPD_MSG_POST, httpd_StreamCallBack,
784 (httpd_callback_sys_t*)stream);
785
786 return stream;
787 }
788
789 int httpd_StreamHeader(httpd_stream_t *stream, uint8_t *p_data, int i_data)
790 {
791 vlc_mutex_lock(&stream->lock);
792 free(stream->p_header);
793 stream->p_header = NULL;
794
795 stream->i_header = i_data;
796 if (i_data > 0) {
797 stream->p_header = xmalloc(i_data);
798 memcpy(stream->p_header, p_data, i_data);
799 }
800 vlc_mutex_unlock(&stream->lock);
801
802 return VLC_SUCCESS;
803 }
804
805 static void httpd_AppendData(httpd_stream_t *stream, uint8_t *p_data, int i_data)
806 {
807 int i_pos = stream->i_buffer_pos % stream->i_buffer_size;
808 int i_count = i_data;
809 while (i_count > 0) {
810 int i_copy = __MIN(i_count, stream->i_buffer_size - i_pos);
811
812 /* Ok, we can't go past the end of our buffer */
813 memcpy(&stream->p_buffer[i_pos], p_data, i_copy);
814
815 i_pos = (i_pos + i_copy) % stream->i_buffer_size;
816 i_count -= i_copy;
817 p_data += i_copy;
818 }
819
820 stream->i_buffer_pos += i_data;
821 }
822
823 int httpd_StreamSend(httpd_stream_t *stream, const block_t *p_block)
824 {
825 if (!p_block || !p_block->p_buffer)
826 return VLC_SUCCESS;
827
828 vlc_mutex_lock(&stream->lock);
829
830 /* save this pointer (to be used by new connection) */
831 stream->i_buffer_last_pos = stream->i_buffer_pos;
832
833 if (p_block->i_flags & BLOCK_FLAG_TYPE_I) {
834 stream->b_has_keyframes = true;
835 stream->i_last_keyframe_seen_pos = stream->i_buffer_pos;
836 }
837
838 httpd_AppendData(stream, p_block->p_buffer, p_block->i_buffer);
839
840 vlc_mutex_unlock(&stream->lock);
841 return VLC_SUCCESS;
842 }
843
844 void httpd_StreamDelete(httpd_stream_t *stream)
845 {
846 httpd_UrlDelete(stream->url);
847 for (size_t i = 0; i < stream->i_http_headers; i++) {
848 free(stream->p_http_headers[i].name);
849 free(stream->p_http_headers[i].value);
850 }
851 free(stream->p_http_headers);
852 vlc_mutex_destroy(&stream->lock);
853 free(stream->psz_mime);
854 free(stream->p_header);
855 free(stream->p_buffer);
856 free(stream);
857 }
858
859 /*****************************************************************************
860 * Low level
861 *****************************************************************************/
862 static void* httpd_HostThread(void *);
863 static httpd_host_t *httpd_HostCreate(vlc_object_t *, const char *,
864 const char *, vlc_tls_creds_t *);
865
866 /* create a new host */
867 httpd_host_t *vlc_http_HostNew(vlc_object_t *p_this)
868 {
869 return httpd_HostCreate(p_this, "http-host", "http-port", NULL);
870 }
871
872 httpd_host_t *vlc_https_HostNew(vlc_object_t *obj)
873 {
874 char *cert = var_InheritString(obj, "http-cert");
875 if (!cert) {
876 msg_Err(obj, "HTTP/TLS certificate not specified!");
877 return NULL;
878 }
879
880 char *key = var_InheritString(obj, "http-key");
881 vlc_tls_creds_t *tls = vlc_tls_ServerCreate(obj, cert, key);
882
883 if (!tls) {
884 msg_Err(obj, "HTTP/TLS certificate error (%s and %s)",
885 cert, key ? key : cert);
886 free(key);
887 free(cert);
888 return NULL;
889 }
890 free(key);
891 free(cert);
892
893 return httpd_HostCreate(obj, "http-host", "https-port", tls);
894 }
895
896 httpd_host_t *vlc_rtsp_HostNew(vlc_object_t *p_this)
897 {
898 return httpd_HostCreate(p_this, "rtsp-host", "rtsp-port", NULL);
899 }
900
901 static struct httpd
902 {
903 vlc_mutex_t mutex;
904
905 httpd_host_t **host;
906 int i_host;
907 } httpd = { VLC_STATIC_MUTEX, NULL, 0 };
908
909 static httpd_host_t *httpd_HostCreate(vlc_object_t *p_this,
910 const char *hostvar,
911 const char *portvar,
912 vlc_tls_creds_t *p_tls)
913 {
914 httpd_host_t *host;
915 char *hostname = var_InheritString(p_this, hostvar);
916 unsigned port = var_InheritInteger(p_this, portvar);
917
918 vlc_url_t url;
919 vlc_UrlParse(&url, hostname);
920 free(hostname);
921 if (url.i_port != 0) {
922 msg_Err(p_this, "Ignoring port %d (using %d)", url.i_port, port);
923 msg_Info(p_this, "Specify port %d separately with the "
924 "%s option instead.", url.i_port, portvar);
925 }
926
927 /* to be sure to avoid multiple creation */
928 vlc_mutex_lock(&httpd.mutex);
929
930 /* verify if it already exist */
931 for (int i = 0; i < httpd.i_host; i++) {
932 host = httpd.host[i];
933
934 /* cannot mix TLS and non-TLS hosts */
935 if (host->port != port
936 || (host->p_tls != NULL) != (p_tls != NULL))
937 continue;
938
939 /* Increase existing matching host reference count.
940 * The reference count is written under both the global httpd and the
941 * host lock. It is read with either or both locks held. The global
942 * lock is always acquired first. */
943 vlc_mutex_lock(&host->lock);
944 host->i_ref++;
945 vlc_mutex_unlock(&host->lock);
946
947 vlc_mutex_unlock(&httpd.mutex);
948 vlc_UrlClean(&url);
949 vlc_tls_Delete(p_tls);
950 return host;
951 }
952
953 /* create the new host */
954 host = (httpd_host_t *)vlc_custom_create(p_this, sizeof (*host),
955 "http host");
956 if (!host)
957 goto error;
958
959 vlc_mutex_init(&host->lock);
960 vlc_cond_init(&host->wait);
961 host->i_ref = 1;
962
963 host->fds = net_ListenTCP(p_this, url.psz_host, port);
964 if (!host->fds) {
965 msg_Err(p_this, "cannot create socket(s) for HTTP host");
966 goto error;
967 }
968 for (host->nfd = 0; host->fds[host->nfd] != -1; host->nfd++);
969
970 host->port = port;
971 host->i_url = 0;
972 host->url = NULL;
973 host->i_client = 0;
974 host->client = NULL;
975 host->p_tls = p_tls;
976
977 /* create the thread */
978 if (vlc_clone(&host->thread, httpd_HostThread, host,
979 VLC_THREAD_PRIORITY_LOW)) {
980 msg_Err(p_this, "cannot spawn http host thread");
981 goto error;
982 }
983
984 /* now add it to httpd */
985 TAB_APPEND(httpd.i_host, httpd.host, host);
986 vlc_mutex_unlock(&httpd.mutex);
987
988 vlc_UrlClean(&url);
989
990 return host;
991
992 error:
993 vlc_mutex_unlock(&httpd.mutex);
994
995 if (host) {
996 net_ListenClose(host->fds);
997 vlc_cond_destroy(&host->wait);
998 vlc_mutex_destroy(&host->lock);
999 vlc_object_release(host);
1000 }
1001
1002 vlc_UrlClean(&url);
1003 vlc_tls_Delete(p_tls);
1004 return NULL;
1005 }
1006
1007 /* delete a host */
1008 void httpd_HostDelete(httpd_host_t *host)
1009 {
1010 bool delete = false;
1011
1012 vlc_mutex_lock(&httpd.mutex);
1013
1014 vlc_mutex_lock(&host->lock);
1015 host->i_ref--;
1016 if (host->i_ref == 0)
1017 delete = true;
1018 vlc_mutex_unlock(&host->lock);
1019 if (!delete) {
1020 /* still used */
1021 vlc_mutex_unlock(&httpd.mutex);
1022 msg_Dbg(host, "httpd_HostDelete: host still in use");
1023 return;
1024 }
1025 TAB_REMOVE(httpd.i_host, httpd.host, host);
1026
1027 vlc_cancel(host->thread);
1028 vlc_join(host->thread, NULL);
1029
1030 msg_Dbg(host, "HTTP host removed");
1031
1032 for (int i = 0; i < host->i_url; i++)
1033 msg_Err(host, "url still registered: %s", host->url[i]->psz_url);
1034
1035 for (int i = 0; i < host->i_client; i++) {
1036 msg_Warn(host, "client still connected");
1037 httpd_ClientDestroy(host->client[i]);
1038 }
1039 TAB_CLEAN(host->i_client, host->client);
1040
1041 vlc_tls_Delete(host->p_tls);
1042 net_ListenClose(host->fds);
1043 vlc_cond_destroy(&host->wait);
1044 vlc_mutex_destroy(&host->lock);
1045 vlc_object_release(host);
1046 vlc_mutex_unlock(&httpd.mutex);
1047 }
1048
1049 /* register a new url */
1050 httpd_url_t *httpd_UrlNew(httpd_host_t *host, const char *psz_url,
1051 const char *psz_user, const char *psz_password)
1052 {
1053 httpd_url_t *url;
1054
1055 assert(psz_url);
1056
1057 vlc_mutex_lock(&host->lock);
1058 for (int i = 0; i < host->i_url; i++)
1059 if (!strcmp(psz_url, host->url[i]->psz_url)) {
1060 msg_Warn(host, "cannot add '%s' (url already defined)", psz_url);
1061 vlc_mutex_unlock(&host->lock);
1062 return NULL;
1063 }
1064
1065 url = xmalloc(sizeof(httpd_url_t));
1066 url->host = host;
1067
1068 vlc_mutex_init(&url->lock);
1069 url->psz_url = xstrdup(psz_url);
1070 url->psz_user = xstrdup(psz_user ? psz_user : "");
1071 url->psz_password = xstrdup(psz_password ? psz_password : "");
1072 for (int i = 0; i < HTTPD_MSG_MAX; i++) {
1073 url->catch[i].cb = NULL;
1074 url->catch[i].p_sys = NULL;
1075 }
1076
1077 TAB_APPEND(host->i_url, host->url, url);
1078 vlc_cond_signal(&host->wait);
1079 vlc_mutex_unlock(&host->lock);
1080
1081 return url;
1082 }
1083
1084 /* register callback on a url */
1085 int httpd_UrlCatch(httpd_url_t *url, int i_msg, httpd_callback_t cb,
1086 httpd_callback_sys_t *p_sys)
1087 {
1088 vlc_mutex_lock(&url->lock);
1089 url->catch[i_msg].cb = cb;
1090 url->catch[i_msg].p_sys= p_sys;
1091 vlc_mutex_unlock(&url->lock);
1092
1093 return VLC_SUCCESS;
1094 }
1095
1096 /* delete a url */
1097 void httpd_UrlDelete(httpd_url_t *url)
1098 {
1099 httpd_host_t *host = url->host;
1100
1101 vlc_mutex_lock(&host->lock);
1102 TAB_REMOVE(host->i_url, host->url, url);
1103
1104 vlc_mutex_destroy(&url->lock);
1105 free(url->psz_url);
1106 free(url->psz_user);
1107 free(url->psz_password);
1108
1109 for (int i = 0; i < host->i_client; i++) {
1110 httpd_client_t *client = host->client[i];
1111
1112 if (client->url != url)
1113 continue;
1114
1115 /* TODO complete it */
1116 msg_Warn(host, "force closing connections");
1117 TAB_REMOVE(host->i_client, host->client, client);
1118 httpd_ClientDestroy(client);
1119 i--;
1120 }
1121 free(url);
1122 vlc_mutex_unlock(&host->lock);
1123 }
1124
1125 static void httpd_MsgInit(httpd_message_t *msg)
1126 {
1127 msg->cl = NULL;
1128 msg->i_type = HTTPD_MSG_NONE;
1129 msg->i_proto = HTTPD_PROTO_NONE;
1130 msg->i_version = -1; /* FIXME */
1131
1132 msg->i_status = 0;
1133
1134 msg->psz_url = NULL;
1135 msg->psz_args = NULL;
1136
1137 msg->i_headers = 0;
1138 msg->p_headers = NULL;
1139
1140 msg->i_body_offset = 0;
1141 msg->i_body = 0;
1142 msg->p_body = NULL;
1143 }
1144
1145 static void httpd_MsgClean(httpd_message_t *msg)
1146 {
1147 free(msg->psz_url);
1148 free(msg->psz_args);
1149 for (size_t i = 0; i < msg->i_headers; i++) {
1150 free(msg->p_headers[i].name);
1151 free(msg->p_headers[i].value);
1152 }
1153 free(msg->p_headers);
1154 free(msg->p_body);
1155 httpd_MsgInit(msg);
1156 }
1157
1158 const char *httpd_MsgGet(const httpd_message_t *msg, const char *name)
1159 {
1160 for (size_t i = 0; i < msg->i_headers; i++)
1161 if (!strcasecmp(msg->p_headers[i].name, name))
1162 return msg->p_headers[i].value;
1163 return NULL;
1164 }
1165
1166 void httpd_MsgAdd(httpd_message_t *msg, const char *name, const char *psz_value, ...)
1167 {
1168 httpd_header *p_tmp = realloc(msg->p_headers, sizeof(httpd_header) * (msg->i_headers + 1));
1169 if (!p_tmp)
1170 return;
1171
1172 msg->p_headers = p_tmp;
1173
1174 httpd_header *h = &msg->p_headers[msg->i_headers];
1175 h->name = strdup(name);
1176 if (!h->name)
1177 return;
1178
1179 h->value = NULL;
1180
1181 va_list args;
1182 va_start(args, psz_value);
1183 int ret = us_vasprintf(&h->value, psz_value, args);
1184 va_end(args);
1185
1186 if (ret == -1) {
1187 free(h->name);
1188 return;
1189 }
1190
1191 msg->i_headers++;
1192 }
1193
1194 static void httpd_ClientInit(httpd_client_t *cl, mtime_t now)
1195 {
1196 cl->i_state = HTTPD_CLIENT_RECEIVING;
1197 cl->i_activity_date = now;
1198 cl->i_activity_timeout = INT64_C(10000000);
1199 cl->i_buffer_size = HTTPD_CL_BUFSIZE;
1200 cl->i_buffer = 0;
1201 cl->p_buffer = xmalloc(cl->i_buffer_size);
1202 cl->i_keyframe_wait_to_pass = -1;
1203 cl->b_stream_mode = false;
1204
1205 httpd_MsgInit(&cl->query);
1206 httpd_MsgInit(&cl->answer);
1207 }
1208
1209 char* httpd_ClientIP(const httpd_client_t *cl, char *ip, int *port)
1210 {
1211 return net_GetPeerAddress(cl->fd, ip, port) ? NULL : ip;
1212 }
1213
1214 char* httpd_ServerIP(const httpd_client_t *cl, char *ip, int *port)
1215 {
1216 return net_GetSockAddress(cl->fd, ip, port) ? NULL : ip;
1217 }
1218
1219 static void httpd_ClientDestroy(httpd_client_t *cl)
1220 {
1221 if (cl->p_tls != NULL)
1222 vlc_tls_Close(cl->p_tls);
1223 else
1224 net_Close(cl->fd);
1225
1226 httpd_MsgClean(&cl->answer);
1227 httpd_MsgClean(&cl->query);
1228
1229 free(cl->p_buffer);
1230 free(cl);
1231 }
1232
1233 static httpd_client_t *httpd_ClientNew(int fd, vlc_tls_t *p_tls, mtime_t now)
1234 {
1235 httpd_client_t *cl = malloc(sizeof(httpd_client_t));
1236
1237 if (!cl) return NULL;
1238
1239 cl->i_ref = 0;
1240 cl->fd = fd;
1241 cl->url = NULL;
1242 cl->p_tls = p_tls;
1243
1244 httpd_ClientInit(cl, now);
1245 if (p_tls)
1246 cl->i_state = HTTPD_CLIENT_TLS_HS_OUT;
1247
1248 return cl;
1249 }
1250
1251 static
1252 ssize_t httpd_NetRecv (httpd_client_t *cl, uint8_t *p, size_t i_len)
1253 {
1254 vlc_tls_t *p_tls;
1255 ssize_t val;
1256
1257 p_tls = cl->p_tls;
1258 do
1259 val = p_tls ? tls_Recv (p_tls, p, i_len)
1260 : recv (cl->fd, p, i_len, 0);
1261 while (val == -1 && errno == EINTR);
1262 return val;
1263 }
1264
1265 static
1266 ssize_t httpd_NetSend (httpd_client_t *cl, const uint8_t *p, size_t i_len)
1267 {
1268 vlc_tls_t *p_tls;
1269 ssize_t val;
1270
1271 p_tls = cl->p_tls;
1272 do
1273 val = p_tls ? tls_Send(p_tls, p, i_len)
1274 : send (cl->fd, p, i_len, MSG_NOSIGNAL);
1275 while (val == -1 && errno == EINTR);
1276 return val;
1277 }
1278
1279
1280 static const struct
1281 {
1282 const char name[16];
1283 int i_type;
1284 int i_proto;
1285 }
1286 msg_type[] =
1287 {
1288 { "OPTIONS", HTTPD_MSG_OPTIONS, HTTPD_PROTO_RTSP },
1289 { "DESCRIBE", HTTPD_MSG_DESCRIBE, HTTPD_PROTO_RTSP },
1290 { "SETUP", HTTPD_MSG_SETUP, HTTPD_PROTO_RTSP },
1291 { "PLAY", HTTPD_MSG_PLAY, HTTPD_PROTO_RTSP },
1292 { "PAUSE", HTTPD_MSG_PAUSE, HTTPD_PROTO_RTSP },
1293 { "GET_PARAMETER", HTTPD_MSG_GETPARAMETER, HTTPD_PROTO_RTSP },
1294 { "TEARDOWN", HTTPD_MSG_TEARDOWN, HTTPD_PROTO_RTSP },
1295 { "GET", HTTPD_MSG_GET, HTTPD_PROTO_HTTP },
1296 { "HEAD", HTTPD_MSG_HEAD, HTTPD_PROTO_HTTP },
1297 { "POST", HTTPD_MSG_POST, HTTPD_PROTO_HTTP },
1298 { "", HTTPD_MSG_NONE, HTTPD_PROTO_NONE }
1299 };
1300
1301
1302 static void httpd_ClientRecv(httpd_client_t *cl)
1303 {
1304 int i_len;
1305
1306 /* ignore leading whites */
1307 if (cl->query.i_proto == HTTPD_PROTO_NONE && cl->i_buffer == 0) {
1308 unsigned char c;
1309
1310 i_len = httpd_NetRecv(cl, &c, 1);
1311
1312 if (i_len > 0 && !strchr("\r\n\t ", c)) {
1313 cl->p_buffer[0] = c;
1314 cl->i_buffer++;
1315 }
1316 } else if (cl->query.i_proto == HTTPD_PROTO_NONE) {
1317 /* enough to see if it's Interleaved RTP over RTSP or RTSP/HTTP */
1318 i_len = httpd_NetRecv(cl, &cl->p_buffer[cl->i_buffer],
1319 7 - cl->i_buffer);
1320 if (i_len > 0)
1321 cl->i_buffer += i_len;
1322
1323 /* The smallest legal request is 7 bytes ("GET /\r\n"),
1324 * this is the maximum we can ask at this point. */
1325 if (cl->i_buffer >= 7) {
1326 if (!memcmp(cl->p_buffer, "HTTP/1.", 7)) {
1327 cl->query.i_proto = HTTPD_PROTO_HTTP;
1328 cl->query.i_type = HTTPD_MSG_ANSWER;
1329 } else if (!memcmp(cl->p_buffer, "RTSP/1.", 7)) {
1330 cl->query.i_proto = HTTPD_PROTO_RTSP;
1331 cl->query.i_type = HTTPD_MSG_ANSWER;
1332 } else {
1333 /* We need the full request line to determine the protocol. */
1334 cl->query.i_proto = HTTPD_PROTO_HTTP0;
1335 cl->query.i_type = HTTPD_MSG_NONE;
1336 }
1337 }
1338 } else if (cl->query.i_body > 0) {
1339 /* we are reading the body of a request or a channel */
1340 assert (cl->query.p_body != NULL);
1341 i_len = httpd_NetRecv(cl, &cl->query.p_body[cl->i_buffer],
1342 cl->query.i_body - cl->i_buffer);
1343 if (i_len > 0)
1344 cl->i_buffer += i_len;
1345
1346 if (cl->i_buffer >= cl->query.i_body)
1347 cl->i_state = HTTPD_CLIENT_RECEIVE_DONE;
1348 } else for (;;) { /* we are reading a header -> char by char */
1349 if (cl->i_buffer == cl->i_buffer_size) {
1350 uint8_t *newbuf = realloc(cl->p_buffer, cl->i_buffer_size + 1024);
1351 if (!newbuf) {
1352 i_len = 0;
1353 break;
1354 }
1355
1356 cl->p_buffer = newbuf;
1357 cl->i_buffer_size += 1024;
1358 }
1359
1360 i_len = httpd_NetRecv (cl, &cl->p_buffer[cl->i_buffer], 1);
1361 if (i_len <= 0)
1362 break;
1363
1364 cl->i_buffer++;
1365
1366 if ((cl->query.i_proto == HTTPD_PROTO_HTTP0)
1367 && (cl->p_buffer[cl->i_buffer - 1] == '\n'))
1368 {
1369 /* Request line is now complete */
1370 const char *p = memchr(cl->p_buffer, ' ', cl->i_buffer);
1371 size_t len;
1372
1373 assert(cl->query.i_type == HTTPD_MSG_NONE);
1374
1375 if (!p) { /* no URI: evil guy */
1376 i_len = 0; /* drop connection */
1377 break;
1378 }
1379
1380 do
1381 p++; /* skips extra spaces */
1382 while (*p == ' ');
1383
1384 p = memchr(p, ' ', ((char *)cl->p_buffer) + cl->i_buffer - p);
1385 if (!p) { /* no explicit protocol: HTTP/0.9 */
1386 i_len = 0; /* not supported currently -> drop */
1387 break;
1388 }
1389
1390 do
1391 p++; /* skips extra spaces ever again */
1392 while (*p == ' ');
1393
1394 len = ((char *)cl->p_buffer) + cl->i_buffer - p;
1395 if (len < 7) /* foreign protocol */
1396 i_len = 0; /* I don't understand -> drop */
1397 else if (!memcmp(p, "HTTP/1.", 7)) {
1398 cl->query.i_proto = HTTPD_PROTO_HTTP;
1399 cl->query.i_version = atoi(p + 7);
1400 } else if (!memcmp(p, "RTSP/1.", 7)) {
1401 cl->query.i_proto = HTTPD_PROTO_RTSP;
1402 cl->query.i_version = atoi(p + 7);
1403 } else if (!memcmp(p, "HTTP/", 5)) {
1404 const uint8_t sorry[] =
1405 "HTTP/1.1 505 Unknown HTTP version\r\n\r\n";
1406 httpd_NetSend(cl, sorry, sizeof(sorry) - 1);
1407 i_len = 0; /* drop */
1408 } else if (!memcmp(p, "RTSP/", 5)) {
1409 const uint8_t sorry[] =
1410 "RTSP/1.0 505 Unknown RTSP version\r\n\r\n";
1411 httpd_NetSend(cl, sorry, sizeof(sorry) - 1);
1412 i_len = 0; /* drop */
1413 } else /* yet another foreign protocol */
1414 i_len = 0;
1415
1416 if (i_len == 0)
1417 break;
1418 }
1419
1420 if ((cl->i_buffer >= 2 && !memcmp(&cl->p_buffer[cl->i_buffer-2], "\n\n", 2))||
1421 (cl->i_buffer >= 4 && !memcmp(&cl->p_buffer[cl->i_buffer-4], "\r\n\r\n", 4)))
1422 {
1423 char *p;
1424
1425 /* we have finished the header so parse it and set i_body */
1426 cl->p_buffer[cl->i_buffer] = '\0';
1427
1428 if (cl->query.i_type == HTTPD_MSG_ANSWER) {
1429 /* FIXME:
1430 * assume strlen("HTTP/1.x") = 8
1431 */
1432 cl->query.i_status =
1433 strtol((char *)&cl->p_buffer[8],
1434 &p, 0);
1435 while (*p == ' ')
1436 p++;
1437 } else {
1438 p = NULL;
1439 cl->query.i_type = HTTPD_MSG_NONE;
1440
1441 for (unsigned i = 0; msg_type[i].name[0]; i++)
1442 if (!strncmp((char *)cl->p_buffer, msg_type[i].name,
1443 strlen(msg_type[i].name))) {
1444 p = (char *)&cl->p_buffer[strlen(msg_type[i].name) + 1 ];
1445 cl->query.i_type = msg_type[i].i_type;
1446 if (cl->query.i_proto != msg_type[i].i_proto) {
1447 p = NULL;
1448 cl->query.i_proto = HTTPD_PROTO_NONE;
1449 cl->query.i_type = HTTPD_MSG_NONE;
1450 }
1451 break;
1452 }
1453
1454 if (!p) {
1455 if (strstr((char *)cl->p_buffer, "HTTP/1."))
1456 cl->query.i_proto = HTTPD_PROTO_HTTP;
1457 else if (strstr((char *)cl->p_buffer, "RTSP/1."))
1458 cl->query.i_proto = HTTPD_PROTO_RTSP;
1459 } else {
1460 char *p2;
1461 char *p3;
1462
1463 while (*p == ' ')
1464 p++;
1465
1466 p2 = strchr(p, ' ');
1467 if (p2)
1468 *p2++ = '\0';
1469
1470 if (!strncasecmp(p, (cl->query.i_proto == HTTPD_PROTO_HTTP) ? "http:" : "rtsp:", 5)) {
1471 /* Skip hier-part of URL (if present) */
1472 p += 5;
1473 if (!strncmp(p, "//", 2)) { /* skip authority */
1474 /* see RFC3986 §3.2 */
1475 p += 2;
1476 p += strcspn(p, "/?#");
1477 }
1478 }
1479 else if (!strncasecmp(p, (cl->query.i_proto == HTTPD_PROTO_HTTP) ? "https:" : "rtsps:", 6)) {
1480 /* Skip hier-part of URL (if present) */
1481 p += 6;
1482 if (!strncmp(p, "//", 2)) { /* skip authority */
1483 /* see RFC3986 §3.2 */
1484 p += 2;
1485 p += strcspn(p, "/?#");
1486 }
1487 }
1488
1489 cl->query.psz_url = strdup(p);
1490 if ((p3 = strchr(cl->query.psz_url, '?')) ) {
1491 *p3++ = '\0';
1492 cl->query.psz_args = (uint8_t *)strdup(p3);
1493 }
1494 p = p2;
1495 }
1496 }
1497 if (p)
1498 p = strchr(p, '\n');
1499
1500 if (p) {
1501 while (*p == '\n' || *p == '\r')
1502 p++;
1503
1504 while (p && *p) {
1505 char *line = p;
1506 char *eol = p = strchr(p, '\n');
1507 char *colon;
1508
1509 while (eol && eol >= line && (*eol == '\n' || *eol == '\r'))
1510 *eol-- = '\0';
1511
1512 if ((colon = strchr(line, ':'))) {
1513 *colon++ = '\0';
1514 while (*colon == ' ')
1515 colon++;
1516 httpd_MsgAdd(&cl->query, line, "%s", colon);
1517
1518 if (!strcasecmp(line, "Content-Length"))
1519 cl->query.i_body = atol(colon);
1520 }
1521
1522 if (p) {
1523 p++;
1524 while (*p == '\n' || *p == '\r')
1525 p++;
1526 }
1527 }
1528 }
1529 if (cl->query.i_body > 0) {
1530 /* TODO Mhh, handle the case where the client only
1531 * sends a request and closes the connection to
1532 * mark the end of the body (probably only RTSP) */
1533 if (cl->query.i_body >= 65536)
1534 cl->query.p_body = malloc(cl->query.i_body);
1535 else
1536 cl->query.p_body = NULL;
1537 cl->i_buffer = 0;
1538 if (!cl->query.p_body) {
1539 switch (cl->query.i_proto) {
1540 case HTTPD_PROTO_HTTP: {
1541 const uint8_t sorry[] = "HTTP/1.1 413 Request Entity Too Large\r\n\r\n";
1542 httpd_NetSend(cl, sorry, sizeof(sorry) - 1);
1543 break;
1544 }
1545 case HTTPD_PROTO_RTSP: {
1546 const uint8_t sorry[] = "RTSP/1.0 413 Request Entity Too Large\r\n\r\n";
1547 httpd_NetSend(cl, sorry, sizeof(sorry) - 1);
1548 break;
1549 }
1550 default:
1551 vlc_assert_unreachable();
1552 }
1553 i_len = 0; /* drop */
1554 }
1555 break;
1556 } else
1557 cl->i_state = HTTPD_CLIENT_RECEIVE_DONE;
1558 }
1559 }
1560
1561 /* check if the client is to be set to dead */
1562 #if defined(_WIN32)
1563 if ((i_len < 0 && WSAGetLastError() != WSAEWOULDBLOCK) || (i_len == 0))
1564 #else
1565 if ((i_len < 0 && errno != EAGAIN) || (i_len == 0))
1566 #endif
1567 {
1568 if (cl->query.i_proto != HTTPD_PROTO_NONE && cl->query.i_type != HTTPD_MSG_NONE) {
1569 /* connection closed -> end of data */
1570 if (cl->query.i_body > 0)
1571 cl->query.i_body = cl->i_buffer;
1572 cl->i_state = HTTPD_CLIENT_RECEIVE_DONE;
1573 }
1574 else
1575 cl->i_state = HTTPD_CLIENT_DEAD;
1576 }
1577
1578 /* XXX: for QT I have to disable timeout. Try to find why */
1579 if (cl->query.i_proto == HTTPD_PROTO_RTSP)
1580 cl->i_activity_timeout = 0;
1581 }
1582
1583 static void httpd_ClientSend(httpd_client_t *cl)
1584 {
1585 int i_len;
1586
1587 if (cl->i_buffer < 0) {
1588 /* We need to create the header */
1589 int i_size = 0;
1590 char *p;
1591 const char *psz_status = httpd_ReasonFromCode(cl->answer.i_status);
1592
1593 i_size = strlen("HTTP/1.") + 10 + 10 + strlen(psz_status) + 5;
1594 for (size_t i = 0; i < cl->answer.i_headers; i++)
1595 i_size += strlen(cl->answer.p_headers[i].name) + 2 +
1596 strlen(cl->answer.p_headers[i].value) + 2;
1597
1598 if (cl->i_buffer_size < i_size) {
1599 cl->i_buffer_size = i_size;
1600 free(cl->p_buffer);
1601 cl->p_buffer = xmalloc(i_size);
1602 }
1603 p = (char *)cl->p_buffer;
1604
1605 p += sprintf(p, "%s.%u %d %s\r\n",
1606 cl->answer.i_proto == HTTPD_PROTO_HTTP ? "HTTP/1" : "RTSP/1",
1607 cl->answer.i_version,
1608 cl->answer.i_status, psz_status);
1609 for (size_t i = 0; i < cl->answer.i_headers; i++)
1610 p += sprintf(p, "%s: %s\r\n", cl->answer.p_headers[i].name,
1611 cl->answer.p_headers[i].value);
1612 p += sprintf(p, "\r\n");
1613
1614 cl->i_buffer = 0;
1615 cl->i_buffer_size = (uint8_t*)p - cl->p_buffer;
1616 }
1617
1618 i_len = httpd_NetSend(cl, &cl->p_buffer[cl->i_buffer],
1619 cl->i_buffer_size - cl->i_buffer);
1620 if (i_len >= 0) {
1621 cl->i_buffer += i_len;
1622
1623 if (cl->i_buffer >= cl->i_buffer_size) {
1624 if (cl->answer.i_body == 0 && cl->answer.i_body_offset > 0) {
1625 /* catch more body data */
1626 int i_msg = cl->query.i_type;
1627 int64_t i_offset = cl->answer.i_body_offset;
1628
1629 httpd_MsgClean(&cl->answer);
1630 cl->answer.i_body_offset = i_offset;
1631
1632 cl->url->catch[i_msg].cb(cl->url->catch[i_msg].p_sys, cl,
1633 &cl->answer, &cl->query);
1634 }
1635
1636 if (cl->answer.i_body > 0) {
1637 /* send the body data */
1638 free(cl->p_buffer);
1639 cl->p_buffer = cl->answer.p_body;
1640 cl->i_buffer_size = cl->answer.i_body;
1641 cl->i_buffer = 0;
1642
1643 cl->answer.i_body = 0;
1644 cl->answer.p_body = NULL;
1645 } else /* send finished */
1646 cl->i_state = HTTPD_CLIENT_SEND_DONE;
1647 }
1648 } else {
1649 #if defined(_WIN32)
1650 if ((i_len < 0 && WSAGetLastError() != WSAEWOULDBLOCK) || (i_len == 0))
1651 #else
1652 if ((i_len < 0 && errno != EAGAIN) || (i_len == 0))
1653 #endif
1654 {
1655 /* error */
1656 cl->i_state = HTTPD_CLIENT_DEAD;
1657 }
1658 }
1659 }
1660
1661 static void httpd_ClientTlsHandshake(httpd_host_t *host, httpd_client_t *cl)
1662 {
1663 switch (vlc_tls_SessionHandshake(host->p_tls, cl->p_tls))
1664 {
1665 case -1: cl->i_state = HTTPD_CLIENT_DEAD; break;
1666 case 0: cl->i_state = HTTPD_CLIENT_RECEIVING; break;
1667 case 1: cl->i_state = HTTPD_CLIENT_TLS_HS_IN; break;
1668 case 2: cl->i_state = HTTPD_CLIENT_TLS_HS_OUT; break;
1669 }
1670 }
1671
1672 static bool httpdAuthOk(const char *user, const char *pass, const char *b64)
1673 {
1674 if (!*user && !*pass)
1675 return true;
1676
1677 if (!b64)
1678 return false;
1679
1680 if (strncasecmp(b64, "BASIC", 5))
1681 return false;
1682
1683 b64 += 5;
1684 while (*b64 == ' ')
1685 b64++;
1686
1687 char *given_user = vlc_b64_decode(b64);
1688 if (!given_user)
1689 return false;
1690
1691 char *given_pass = NULL;
1692 given_pass = strchr (given_user, ':');
1693 if (!given_pass)
1694 goto auth_failed;
1695
1696 *given_pass++ = '\0';
1697
1698 if (strcmp (given_user, user))
1699 goto auth_failed;
1700
1701 if (strcmp (given_pass, pass))
1702 goto auth_failed;
1703
1704 free(given_user);
1705 return true;
1706
1707 auth_failed:
1708 free(given_user);
1709 return false;
1710 }
1711
1712 static void httpdLoop(httpd_host_t *host)
1713 {
1714 struct pollfd ufd[host->nfd + host->i_client];
1715 unsigned nfd;
1716 for (nfd = 0; nfd < host->nfd; nfd++) {
1717 ufd[nfd].fd = host->fds[nfd];
1718 ufd[nfd].events = POLLIN;
1719 ufd[nfd].revents = 0;
1720 }
1721
1722 /* add all socket that should be read/write and close dead connection */
1723 while (host->i_url <= 0) {
1724 mutex_cleanup_push(&host->lock);
1725 vlc_cond_wait(&host->wait, &host->lock);
1726 vlc_cleanup_pop();
1727 }
1728
1729 mtime_t now = mdate();
1730 bool b_low_delay = false;
1731
1732 int canc = vlc_savecancel();
1733 for (int i_client = 0; i_client < host->i_client; i_client++) {
1734 int64_t i_offset;
1735 httpd_client_t *cl = host->client[i_client];
1736 if (cl->i_ref < 0 || (cl->i_ref == 0 &&
1737 (cl->i_state == HTTPD_CLIENT_DEAD ||
1738 (cl->i_activity_timeout > 0 &&
1739 cl->i_activity_date+cl->i_activity_timeout < now)))) {
1740 TAB_REMOVE(host->i_client, host->client, cl);
1741 i_client--;
1742 httpd_ClientDestroy(cl);
1743 continue;
1744 }
1745
1746 struct pollfd *pufd = ufd + nfd;
1747 assert (pufd < ufd + (sizeof (ufd) / sizeof (ufd[0])));
1748
1749 pufd->fd = cl->fd;
1750 pufd->events = pufd->revents = 0;
1751
1752 switch (cl->i_state) {
1753 case HTTPD_CLIENT_RECEIVING:
1754 case HTTPD_CLIENT_TLS_HS_IN:
1755 pufd->events = POLLIN;
1756 break;
1757
1758 case HTTPD_CLIENT_SENDING:
1759 case HTTPD_CLIENT_TLS_HS_OUT:
1760 pufd->events = POLLOUT;
1761 break;
1762
1763 case HTTPD_CLIENT_RECEIVE_DONE: {
1764 httpd_message_t *answer = &cl->answer;
1765 httpd_message_t *query = &cl->query;
1766
1767 httpd_MsgInit(answer);
1768
1769 /* Handle what we received */
1770 switch (query->i_type) {
1771 case HTTPD_MSG_ANSWER:
1772 cl->url = NULL;
1773 cl->i_state = HTTPD_CLIENT_DEAD;
1774 break;
1775
1776 case HTTPD_MSG_OPTIONS:
1777 answer->i_type = HTTPD_MSG_ANSWER;
1778 answer->i_proto = query->i_proto;
1779 answer->i_status = 200;
1780 answer->i_body = 0;
1781 answer->p_body = NULL;
1782
1783 httpd_MsgAdd(answer, "Server", "VLC/%s", VERSION);
1784 httpd_MsgAdd(answer, "Content-Length", "0");
1785
1786 switch(query->i_proto) {
1787 case HTTPD_PROTO_HTTP:
1788 answer->i_version = 1;
1789 httpd_MsgAdd(answer, "Allow", "GET,HEAD,POST,OPTIONS");
1790 break;
1791
1792 case HTTPD_PROTO_RTSP:
1793 answer->i_version = 0;
1794
1795 const char *p = httpd_MsgGet(query, "Cseq");
1796 if (p)
1797 httpd_MsgAdd(answer, "Cseq", "%s", p);
1798 p = httpd_MsgGet(query, "Timestamp");
1799 if (p)
1800 httpd_MsgAdd(answer, "Timestamp", "%s", p);
1801
1802 p = httpd_MsgGet(query, "Require");
1803 if (p) {
1804 answer->i_status = 551;
1805 httpd_MsgAdd(query, "Unsupported", "%s", p);
1806 }
1807
1808 httpd_MsgAdd(answer, "Public", "DESCRIBE,SETUP,"
1809 "TEARDOWN,PLAY,PAUSE,GET_PARAMETER");
1810 break;
1811 }
1812
1813 cl->i_buffer = -1; /* Force the creation of the answer in
1814 * httpd_ClientSend */
1815 cl->i_state = HTTPD_CLIENT_SENDING;
1816 break;
1817
1818 case HTTPD_MSG_NONE:
1819 if (query->i_proto == HTTPD_PROTO_NONE) {
1820 cl->url = NULL;
1821 cl->i_state = HTTPD_CLIENT_DEAD;
1822 } else {
1823 /* unimplemented */
1824 answer->i_proto = query->i_proto ;
1825 answer->i_type = HTTPD_MSG_ANSWER;
1826 answer->i_version= 0;
1827 answer->i_status = 501;
1828
1829 char *p;
1830 answer->i_body = httpd_HtmlError (&p, 501, NULL);
1831 answer->p_body = (uint8_t *)p;
1832 httpd_MsgAdd(answer, "Content-Length", "%d", answer->i_body);
1833
1834 cl->i_buffer = -1; /* Force the creation of the answer in httpd_ClientSend */
1835 cl->i_state = HTTPD_CLIENT_SENDING;
1836 }
1837 break;
1838
1839 default: {
1840 int i_msg = query->i_type;
1841 bool b_auth_failed = false;
1842
1843 /* Search the url and trigger callbacks */
1844 for (int i = 0; i < host->i_url; i++) {
1845 httpd_url_t *url = host->url[i];
1846
1847 if (strcmp(url->psz_url, query->psz_url))
1848 continue;
1849 if (!url->catch[i_msg].cb)
1850 continue;
1851
1852 if (answer) {
1853 b_auth_failed = !httpdAuthOk(url->psz_user,
1854 url->psz_password,
1855 httpd_MsgGet(query, "Authorization")); /* BASIC id */
1856 if (b_auth_failed)
1857 break;
1858 }
1859
1860 if (url->catch[i_msg].cb(url->catch[i_msg].p_sys, cl, answer, query))
1861 continue;
1862
1863 if (answer->i_proto == HTTPD_PROTO_NONE)
1864 cl->i_buffer = cl->i_buffer_size; /* Raw answer from a CGI */
1865 else
1866 cl->i_buffer = -1;
1867
1868 /* only one url can answer */
1869 answer = NULL;
1870 if (!cl->url)
1871 cl->url = url;
1872 }
1873
1874 if (answer) {
1875 answer->i_proto = query->i_proto;
1876 answer->i_type = HTTPD_MSG_ANSWER;
1877 answer->i_version= 0;
1878
1879 if (b_auth_failed) {
1880 httpd_MsgAdd(answer, "WWW-Authenticate",
1881 "Basic realm=\"VLC stream\"");
1882 answer->i_status = 401;
1883 } else
1884 answer->i_status = 404; /* no url registered */
1885
1886 char *p;
1887 answer->i_body = httpd_HtmlError (&p, answer->i_status,
1888 query->psz_url);
1889 answer->p_body = (uint8_t *)p;
1890
1891 cl->i_buffer = -1; /* Force the creation of the answer in httpd_ClientSend */
1892 httpd_MsgAdd(answer, "Content-Length", "%d", answer->i_body);
1893 httpd_MsgAdd(answer, "Content-Type", "%s", "text/html");
1894 }
1895
1896 cl->i_state = HTTPD_CLIENT_SENDING;
1897 }
1898 }
1899 break;
1900 }
1901
1902 case HTTPD_CLIENT_SEND_DONE:
1903 if (!cl->b_stream_mode || cl->answer.i_body_offset == 0) {
1904 const char *psz_connection = httpd_MsgGet(&cl->answer, "Connection");
1905 const char *psz_query = httpd_MsgGet(&cl->query, "Connection");
1906 bool b_connection = false;
1907 bool b_keepalive = false;
1908 bool b_query = false;
1909
1910 cl->url = NULL;
1911 if (psz_connection) {
1912 b_connection = (strcasecmp(psz_connection, "Close") == 0);
1913 b_keepalive = (strcasecmp(psz_connection, "Keep-Alive") == 0);
1914 }
1915
1916 if (psz_query)
1917 b_query = (strcasecmp(psz_query, "Close") == 0);
1918
1919 if (((cl->query.i_proto == HTTPD_PROTO_HTTP) &&
1920 ((cl->query.i_version == 0 && b_keepalive) ||
1921 (cl->query.i_version == 1 && !b_connection))) ||
1922 ((cl->query.i_proto == HTTPD_PROTO_RTSP) &&
1923 !b_query && !b_connection)) {
1924 httpd_MsgClean(&cl->query);
1925 httpd_MsgInit(&cl->query);
1926
1927 cl->i_buffer = 0;
1928 cl->i_buffer_size = 1000;
1929 free(cl->p_buffer);
1930 cl->p_buffer = xmalloc(cl->i_buffer_size);
1931 cl->i_state = HTTPD_CLIENT_RECEIVING;
1932 } else
1933 cl->i_state = HTTPD_CLIENT_DEAD;
1934 httpd_MsgClean(&cl->answer);
1935 } else {
1936 i_offset = cl->answer.i_body_offset;
1937 httpd_MsgClean(&cl->answer);
1938
1939 cl->answer.i_body_offset = i_offset;
1940 free(cl->p_buffer);
1941 cl->p_buffer = NULL;
1942 cl->i_buffer = 0;
1943 cl->i_buffer_size = 0;
1944
1945 cl->i_state = HTTPD_CLIENT_WAITING;
1946 }
1947 break;
1948
1949 case HTTPD_CLIENT_WAITING:
1950 i_offset = cl->answer.i_body_offset;
1951 int i_msg = cl->query.i_type;
1952
1953 httpd_MsgInit(&cl->answer);
1954 cl->answer.i_body_offset = i_offset;
1955
1956 cl->url->catch[i_msg].cb(cl->url->catch[i_msg].p_sys, cl,
1957 &cl->answer, &cl->query);
1958 if (cl->answer.i_type != HTTPD_MSG_NONE) {
1959 /* we have new data, so re-enter send mode */
1960 cl->i_buffer = 0;
1961 cl->p_buffer = cl->answer.p_body;
1962 cl->i_buffer_size = cl->answer.i_body;
1963 cl->answer.p_body = NULL;
1964 cl->answer.i_body = 0;
1965 cl->i_state = HTTPD_CLIENT_SENDING;
1966 }
1967 }
1968
1969 if (pufd->events != 0)
1970 nfd++;
1971 else
1972 b_low_delay = true;
1973 }
1974 vlc_mutex_unlock(&host->lock);
1975 vlc_restorecancel(canc);
1976
1977 /* we will wait 20ms (not too big) if HTTPD_CLIENT_WAITING */
1978 int ret = poll(ufd, nfd, b_low_delay ? 20 : -1);
1979
1980 canc = vlc_savecancel();
1981 vlc_mutex_lock(&host->lock);
1982 switch(ret) {
1983 case -1:
1984 if (errno != EINTR) {
1985 /* Kernel on low memory or a bug: pace */
1986 msg_Err(host, "polling error: %s", vlc_strerror_c(errno));
1987 msleep(100000);
1988 }
1989 case 0:
1990 vlc_restorecancel(canc);
1991 return;
1992 }
1993
1994 /* Handle client sockets */
1995 now = mdate();
1996 nfd = host->nfd;
1997
1998 for (int i_client = 0; i_client < host->i_client; i_client++) {
1999 httpd_client_t *cl = host->client[i_client];
2000 const struct pollfd *pufd = &ufd[nfd];
2001
2002 assert(pufd < &ufd[sizeof(ufd) / sizeof(ufd[0])]);
2003
2004 if (cl->fd != pufd->fd)
2005 continue; // we were not waiting for this client
2006 ++nfd;
2007 if (pufd->revents == 0)
2008 continue; // no event received
2009
2010 cl->i_activity_date = now;
2011
2012 switch (cl->i_state) {
2013 case HTTPD_CLIENT_RECEIVING: httpd_ClientRecv(cl); break;
2014 case HTTPD_CLIENT_SENDING: httpd_ClientSend(cl); break;
2015 case HTTPD_CLIENT_TLS_HS_IN:
2016 case HTTPD_CLIENT_TLS_HS_OUT:
2017 httpd_ClientTlsHandshake(host, cl);
2018 break;
2019 }
2020 }
2021
2022 /* Handle server sockets (accept new connections) */
2023 for (nfd = 0; nfd < host->nfd; nfd++) {
2024 httpd_client_t *cl;
2025 int fd = ufd[nfd].fd;
2026
2027 assert (fd == host->fds[nfd]);
2028
2029 if (ufd[nfd].revents == 0)
2030 continue;
2031
2032 /* */
2033 fd = vlc_accept (fd, NULL, NULL, true);
2034 if (fd == -1)
2035 continue;
2036 setsockopt (fd, SOL_SOCKET, SO_REUSEADDR,
2037 &(int){ 1 }, sizeof(int));
2038
2039 vlc_tls_t *p_tls;
2040
2041 if (host->p_tls != NULL)
2042 {
2043 const char *alpn[] = { "http/1.1", NULL };
2044
2045 p_tls = vlc_tls_ServerSessionCreate(host->p_tls, fd, alpn);
2046 }
2047 else
2048 p_tls = NULL;
2049
2050 cl = httpd_ClientNew(fd, p_tls, now);
2051
2052 TAB_APPEND(host->i_client, host->client, cl);
2053 }
2054
2055 vlc_restorecancel(canc);
2056 }
2057
2058 static void* httpd_HostThread(void *data)
2059 {
2060 httpd_host_t *host = data;
2061
2062 vlc_mutex_lock(&host->lock);
2063 while (host->i_ref > 0)
2064 httpdLoop(host);
2065 vlc_mutex_unlock(&host->lock);
2066 return NULL;
2067 }
2068
2069 int httpd_StreamSetHTTPHeaders(httpd_stream_t * p_stream, httpd_header * p_headers, size_t i_headers)
2070 {
2071 if (!p_stream)
2072 return VLC_EGENERIC;
2073
2074 vlc_mutex_lock(&p_stream->lock);
2075 if (p_stream->p_http_headers) {
2076 for (size_t i = 0; i < p_stream->i_http_headers; i++) {
2077 free(p_stream->p_http_headers[i].name);
2078 free(p_stream->p_http_headers[i].value);
2079 }
2080 free(p_stream->p_http_headers);
2081 p_stream->p_http_headers = NULL;
2082 p_stream->i_http_headers = 0;
2083 }
2084
2085 if (!p_headers || !i_headers) {
2086 vlc_mutex_unlock(&p_stream->lock);
2087 return VLC_SUCCESS;
2088 }
2089
2090 p_stream->p_http_headers = malloc(sizeof(httpd_header) * i_headers);
2091 if (!p_stream->p_http_headers) {
2092 vlc_mutex_unlock(&p_stream->lock);
2093 return VLC_ENOMEM;
2094 }
2095
2096 size_t j = 0;
2097 for (size_t i = 0; i < i_headers; i++) {
2098 if (unlikely(!p_headers[i].name || !p_headers[i].value))
2099 continue;
2100
2101 p_stream->p_http_headers[j].name = strdup(p_headers[i].name);
2102 p_stream->p_http_headers[j].value = strdup(p_headers[i].value);
2103
2104 if (unlikely(!p_stream->p_http_headers[j].name ||
2105 !p_stream->p_http_headers[j].value)) {
2106 free(p_stream->p_http_headers[j].name);
2107 free(p_stream->p_http_headers[j].value);
2108 break;
2109 }
2110 j++;
2111 }
2112 p_stream->i_http_headers = j;
2113 vlc_mutex_unlock(&p_stream->lock);
2114 return VLC_SUCCESS;
2115 }
VLC Development tree mirror