2 * Multiple format streaming server
3 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
24 #define closesocket close
29 #include "libavformat/avformat.h"
30 #include "libavformat/ffm.h"
31 #include "libavformat/network.h"
32 #include "libavformat/os_support.h"
33 #include "libavformat/rtpdec.h"
34 #include "libavformat/rtsp.h"
35 // XXX for ffio_open_dyn_packet_buffer, to be removed
36 #include "libavformat/avio_internal.h"
37 #include "libavutil/avstring.h"
38 #include "libavutil/lfg.h"
39 #include "libavutil/dict.h"
40 #include "libavutil/mathematics.h"
41 #include "libavutil/random_seed.h"
42 #include "libavutil/parseutils.h"
43 #include "libavutil/opt.h"
47 #include <sys/ioctl.h>
62 const char program_name[] = "ffserver";
63 const int program_birth_year = 2000;
65 static const OptionDef options[];
68 HTTPSTATE_WAIT_REQUEST,
69 HTTPSTATE_SEND_HEADER,
70 HTTPSTATE_SEND_DATA_HEADER,
71 HTTPSTATE_SEND_DATA, /* sending TCP or UDP data */
72 HTTPSTATE_SEND_DATA_TRAILER,
73 HTTPSTATE_RECEIVE_DATA,
74 HTTPSTATE_WAIT_FEED, /* wait for data from the feed */
77 RTSPSTATE_WAIT_REQUEST,
79 RTSPSTATE_SEND_PACKET,
82 static const char *http_state[] = {
98 #define MAX_STREAMS 20
100 #define IOBUFFER_INIT_SIZE 8192
102 /* timeouts are in ms */
103 #define HTTP_REQUEST_TIMEOUT (15 * 1000)
104 #define RTSP_REQUEST_TIMEOUT (3600 * 24 * 1000)
106 #define SYNC_TIMEOUT (10 * 1000)
108 typedef struct RTSPActionServerSetup {
110 char transport_option[512];
111 } RTSPActionServerSetup;
114 int64_t count1, count2;
115 int64_t time1, time2;
118 /* context associated with one connection */
119 typedef struct HTTPContext {
120 enum HTTPState state;
121 int fd; /* socket file descriptor */
122 struct sockaddr_in from_addr; /* origin */
123 struct pollfd *poll_entry; /* used when polling */
125 uint8_t *buffer_ptr, *buffer_end;
128 int chunked_encoding;
129 int chunk_size; /* 0 if it needs to be read */
130 struct HTTPContext *next;
131 int got_key_frame; /* stream 0 => 1, stream 1 => 2, stream 2=> 4 */
135 /* input format handling */
136 AVFormatContext *fmt_in;
137 int64_t start_time; /* In milliseconds - this wraps fairly often */
138 int64_t first_pts; /* initial pts value */
139 int64_t cur_pts; /* current pts value from the stream in us */
140 int64_t cur_frame_duration; /* duration of the current frame in us */
141 int cur_frame_bytes; /* output frame size, needed to compute
142 the time at which we send each
144 int pts_stream_index; /* stream we choose as clock reference */
145 int64_t cur_clock; /* current clock reference value in us */
146 /* output format handling */
147 struct FFStream *stream;
148 /* -1 is invalid stream */
149 int feed_streams[MAX_STREAMS]; /* index of streams in the feed */
150 int switch_feed_streams[MAX_STREAMS]; /* index of streams in the feed */
152 AVFormatContext fmt_ctx; /* instance of FFStream for one user */
153 int last_packet_sent; /* true if last data packet was sent */
155 DataRateData datarate;
162 int is_packetized; /* if true, the stream is packetized */
163 int packet_stream_index; /* current stream for output in state machine */
165 /* RTSP state specific */
166 uint8_t *pb_buffer; /* XXX: use that in all the code */
168 int seq; /* RTSP sequence number */
170 /* RTP state specific */
171 enum RTSPLowerTransport rtp_protocol;
172 char session_id[32]; /* session id */
173 AVFormatContext *rtp_ctx[MAX_STREAMS];
175 /* RTP/UDP specific */
176 URLContext *rtp_handles[MAX_STREAMS];
178 /* RTP/TCP specific */
179 struct HTTPContext *rtsp_c;
180 uint8_t *packet_buffer, *packet_buffer_ptr, *packet_buffer_end;
183 /* each generated stream is described here */
187 STREAM_TYPE_REDIRECT,
190 enum IPAddressAction {
195 typedef struct IPAddressACL {
196 struct IPAddressACL *next;
197 enum IPAddressAction action;
198 /* These are in host order */
199 struct in_addr first;
203 /* description of each stream of the ffserver.conf file */
204 typedef struct FFStream {
205 enum StreamType stream_type;
206 char filename[1024]; /* stream filename */
207 struct FFStream *feed; /* feed we are using (can be null if
209 AVDictionary *in_opts; /* input parameters */
210 AVInputFormat *ifmt; /* if non NULL, force input format */
213 char dynamic_acl[1024];
215 int prebuffer; /* Number of millseconds early to start */
216 int64_t max_time; /* Number of milliseconds to run */
218 AVStream *streams[MAX_STREAMS];
219 int feed_streams[MAX_STREAMS]; /* index of streams in the feed */
220 char feed_filename[1024]; /* file name of the feed storage, or
221 input file name for a stream */
226 pid_t pid; /* Of ffmpeg process */
227 time_t pid_start; /* Of ffmpeg process */
229 struct FFStream *next;
230 unsigned bandwidth; /* bandwidth, in kbits/s */
233 /* multicast specific */
235 struct in_addr multicast_ip;
236 int multicast_port; /* first port used for multicast */
238 int loop; /* if true, send the stream in loops (only meaningful if file) */
241 int feed_opened; /* true if someone is writing to the feed */
242 int is_feed; /* true if it is a feed */
243 int readonly; /* True if writing is prohibited to the file */
244 int truncate; /* True if feeder connection truncate the feed file */
246 int64_t bytes_served;
247 int64_t feed_max_size; /* maximum storage size, zero means unlimited */
248 int64_t feed_write_index; /* current write position in feed (it wraps around) */
249 int64_t feed_size; /* current size of feed */
250 struct FFStream *next_feed;
253 typedef struct FeedData {
254 long long data_count;
255 float avg_frame_size; /* frame size averaged over last frames with exponential mean */
258 static struct sockaddr_in my_http_addr;
259 static struct sockaddr_in my_rtsp_addr;
261 static char logfilename[1024];
262 static HTTPContext *first_http_ctx;
263 static FFStream *first_feed; /* contains only feeds */
264 static FFStream *first_stream; /* contains all streams, including feeds */
266 static void new_connection(int server_fd, int is_rtsp);
267 static void close_connection(HTTPContext *c);
270 static int handle_connection(HTTPContext *c);
271 static int http_parse_request(HTTPContext *c);
272 static int http_send_data(HTTPContext *c);
273 static void compute_status(HTTPContext *c);
274 static int open_input_stream(HTTPContext *c, const char *info);
275 static int http_start_receive_data(HTTPContext *c);
276 static int http_receive_data(HTTPContext *c);
279 static int rtsp_parse_request(HTTPContext *c);
280 static void rtsp_cmd_describe(HTTPContext *c, const char *url);
281 static void rtsp_cmd_options(HTTPContext *c, const char *url);
282 static void rtsp_cmd_setup(HTTPContext *c, const char *url, RTSPMessageHeader *h);
283 static void rtsp_cmd_play(HTTPContext *c, const char *url, RTSPMessageHeader *h);
284 static void rtsp_cmd_pause(HTTPContext *c, const char *url, RTSPMessageHeader *h);
285 static void rtsp_cmd_teardown(HTTPContext *c, const char *url, RTSPMessageHeader *h);
288 static int prepare_sdp_description(FFStream *stream, uint8_t **pbuffer,
289 struct in_addr my_ip);
292 static HTTPContext *rtp_new_connection(struct sockaddr_in *from_addr,
293 FFStream *stream, const char *session_id,
294 enum RTSPLowerTransport rtp_protocol);
295 static int rtp_new_av_stream(HTTPContext *c,
296 int stream_index, struct sockaddr_in *dest_addr,
297 HTTPContext *rtsp_c);
299 static const char *my_program_name;
300 static const char *my_program_dir;
302 static const char *config_filename = "/etc/ffserver.conf";
304 static int ffserver_debug;
305 static int ffserver_daemon;
306 static int no_launch;
307 static int need_to_start_children;
309 /* maximum number of simultaneous HTTP connections */
310 static unsigned int nb_max_http_connections = 2000;
311 static unsigned int nb_max_connections = 5;
312 static unsigned int nb_connections;
314 static uint64_t max_bandwidth = 1000;
315 static uint64_t current_bandwidth;
317 static int64_t cur_time; // Making this global saves on passing it around everywhere
319 static AVLFG random_state;
321 static FILE *logfile = NULL;
323 /* FIXME: make ffserver work with IPv6 */
324 /* resolve host with also IP address parsing */
325 static int resolve_host(struct in_addr *sin_addr, const char *hostname)
328 if (!ff_inet_aton(hostname, sin_addr)) {
330 struct addrinfo *ai, *cur;
331 struct addrinfo hints;
332 memset(&hints, 0, sizeof(hints));
333 hints.ai_family = AF_INET;
334 if (getaddrinfo(hostname, NULL, &hints, &ai))
336 /* getaddrinfo returns a linked list of addrinfo structs.
337 * Even if we set ai_family = AF_INET above, make sure
338 * that the returned one actually is of the correct type. */
339 for (cur = ai; cur; cur = cur->ai_next) {
340 if (cur->ai_family == AF_INET) {
341 *sin_addr = ((struct sockaddr_in *)cur->ai_addr)->sin_addr;
350 hp = gethostbyname(hostname);
353 memcpy(sin_addr, hp->h_addr_list[0], sizeof(struct in_addr));
359 static char *ctime1(char *buf2)
367 p = buf2 + strlen(p) - 1;
373 static void http_vlog(const char *fmt, va_list vargs)
375 static int print_prefix = 1;
380 fprintf(logfile, "%s ", buf);
382 print_prefix = strstr(fmt, "\n") != NULL;
383 vfprintf(logfile, fmt, vargs);
389 __attribute__ ((format (printf, 1, 2)))
391 static void http_log(const char *fmt, ...)
394 va_start(vargs, fmt);
395 http_vlog(fmt, vargs);
399 static void http_av_log(void *ptr, int level, const char *fmt, va_list vargs)
401 static int print_prefix = 1;
402 AVClass *avc = ptr ? *(AVClass**)ptr : NULL;
403 if (level > av_log_get_level())
405 if (print_prefix && avc)
406 http_log("[%s @ %p]", avc->item_name(ptr), ptr);
407 print_prefix = strstr(fmt, "\n") != NULL;
408 http_vlog(fmt, vargs);
411 static void log_connection(HTTPContext *c)
416 http_log("%s - - [%s] \"%s %s\" %d %"PRId64"\n",
417 inet_ntoa(c->from_addr.sin_addr), c->method, c->url,
418 c->protocol, (c->http_error ? c->http_error : 200), c->data_count);
421 static void update_datarate(DataRateData *drd, int64_t count)
423 if (!drd->time1 && !drd->count1) {
424 drd->time1 = drd->time2 = cur_time;
425 drd->count1 = drd->count2 = count;
426 } else if (cur_time - drd->time2 > 5000) {
427 drd->time1 = drd->time2;
428 drd->count1 = drd->count2;
429 drd->time2 = cur_time;
434 /* In bytes per second */
435 static int compute_datarate(DataRateData *drd, int64_t count)
437 if (cur_time == drd->time1)
440 return ((count - drd->count1) * 1000) / (cur_time - drd->time1);
444 static void start_children(FFStream *feed)
449 for (; feed; feed = feed->next) {
450 if (feed->child_argv && !feed->pid) {
451 feed->pid_start = time(0);
456 http_log("Unable to create children\n");
465 av_strlcpy(pathname, my_program_name, sizeof(pathname));
467 slash = strrchr(pathname, '/');
472 strcpy(slash, "ffmpeg");
474 http_log("Launch commandline: ");
475 http_log("%s ", pathname);
476 for (i = 1; feed->child_argv[i] && feed->child_argv[i][0]; i++)
477 http_log("%s ", feed->child_argv[i]);
480 for (i = 3; i < 256; i++)
483 if (!ffserver_debug) {
484 i = open("/dev/null", O_RDWR);
493 /* This is needed to make relative pathnames work */
494 chdir(my_program_dir);
496 signal(SIGPIPE, SIG_DFL);
498 execvp(pathname, feed->child_argv);
506 /* open a listening socket */
507 static int socket_open_listen(struct sockaddr_in *my_addr)
511 server_fd = socket(AF_INET,SOCK_STREAM,0);
518 setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &tmp, sizeof(tmp));
520 if (bind (server_fd, (struct sockaddr *) my_addr, sizeof (*my_addr)) < 0) {
522 snprintf(bindmsg, sizeof(bindmsg), "bind(port %d)", ntohs(my_addr->sin_port));
524 closesocket(server_fd);
528 if (listen (server_fd, 5) < 0) {
530 closesocket(server_fd);
533 ff_socket_nonblock(server_fd, 1);
538 /* start all multicast streams */
539 static void start_multicast(void)
544 struct sockaddr_in dest_addr;
545 int default_port, stream_index;
548 for(stream = first_stream; stream != NULL; stream = stream->next) {
549 if (stream->is_multicast) {
550 /* open the RTP connection */
551 snprintf(session_id, sizeof(session_id), "%08x%08x",
552 av_lfg_get(&random_state), av_lfg_get(&random_state));
554 /* choose a port if none given */
555 if (stream->multicast_port == 0) {
556 stream->multicast_port = default_port;
560 dest_addr.sin_family = AF_INET;
561 dest_addr.sin_addr = stream->multicast_ip;
562 dest_addr.sin_port = htons(stream->multicast_port);
564 rtp_c = rtp_new_connection(&dest_addr, stream, session_id,
565 RTSP_LOWER_TRANSPORT_UDP_MULTICAST);
569 if (open_input_stream(rtp_c, "") < 0) {
570 http_log("Could not open input stream for stream '%s'\n",
575 /* open each RTP stream */
576 for(stream_index = 0; stream_index < stream->nb_streams;
578 dest_addr.sin_port = htons(stream->multicast_port +
580 if (rtp_new_av_stream(rtp_c, stream_index, &dest_addr, NULL) < 0) {
581 http_log("Could not open output stream '%s/streamid=%d'\n",
582 stream->filename, stream_index);
587 /* change state to send data */
588 rtp_c->state = HTTPSTATE_SEND_DATA;
593 /* main loop of the http server */
594 static int http_server(void)
596 int server_fd = 0, rtsp_server_fd = 0;
597 int ret, delay, delay1;
598 struct pollfd *poll_table, *poll_entry;
599 HTTPContext *c, *c_next;
601 if(!(poll_table = av_mallocz((nb_max_http_connections + 2)*sizeof(*poll_table)))) {
602 http_log("Impossible to allocate a poll table handling %d connections.\n", nb_max_http_connections);
606 if (my_http_addr.sin_port) {
607 server_fd = socket_open_listen(&my_http_addr);
612 if (my_rtsp_addr.sin_port) {
613 rtsp_server_fd = socket_open_listen(&my_rtsp_addr);
614 if (rtsp_server_fd < 0)
618 if (!rtsp_server_fd && !server_fd) {
619 http_log("HTTP and RTSP disabled.\n");
623 http_log("FFserver started.\n");
625 start_children(first_feed);
630 poll_entry = poll_table;
632 poll_entry->fd = server_fd;
633 poll_entry->events = POLLIN;
636 if (rtsp_server_fd) {
637 poll_entry->fd = rtsp_server_fd;
638 poll_entry->events = POLLIN;
642 /* wait for events on each HTTP handle */
649 case HTTPSTATE_SEND_HEADER:
650 case RTSPSTATE_SEND_REPLY:
651 case RTSPSTATE_SEND_PACKET:
652 c->poll_entry = poll_entry;
654 poll_entry->events = POLLOUT;
657 case HTTPSTATE_SEND_DATA_HEADER:
658 case HTTPSTATE_SEND_DATA:
659 case HTTPSTATE_SEND_DATA_TRAILER:
660 if (!c->is_packetized) {
661 /* for TCP, we output as much as we can (may need to put a limit) */
662 c->poll_entry = poll_entry;
664 poll_entry->events = POLLOUT;
667 /* when ffserver is doing the timing, we work by
668 looking at which packet need to be sent every
670 delay1 = 10; /* one tick wait XXX: 10 ms assumed */
675 case HTTPSTATE_WAIT_REQUEST:
676 case HTTPSTATE_RECEIVE_DATA:
677 case HTTPSTATE_WAIT_FEED:
678 case RTSPSTATE_WAIT_REQUEST:
679 /* need to catch errors */
680 c->poll_entry = poll_entry;
682 poll_entry->events = POLLIN;/* Maybe this will work */
686 c->poll_entry = NULL;
692 /* wait for an event on one connection. We poll at least every
693 second to handle timeouts */
695 ret = poll(poll_table, poll_entry - poll_table, delay);
696 if (ret < 0 && ff_neterrno() != AVERROR(EAGAIN) &&
697 ff_neterrno() != AVERROR(EINTR))
701 cur_time = av_gettime() / 1000;
703 if (need_to_start_children) {
704 need_to_start_children = 0;
705 start_children(first_feed);
708 /* now handle the events */
709 for(c = first_http_ctx; c != NULL; c = c_next) {
711 if (handle_connection(c) < 0) {
712 /* close and free the connection */
718 poll_entry = poll_table;
720 /* new HTTP connection request ? */
721 if (poll_entry->revents & POLLIN)
722 new_connection(server_fd, 0);
725 if (rtsp_server_fd) {
726 /* new RTSP connection request ? */
727 if (poll_entry->revents & POLLIN)
728 new_connection(rtsp_server_fd, 1);
733 /* start waiting for a new HTTP/RTSP request */
734 static void start_wait_request(HTTPContext *c, int is_rtsp)
736 c->buffer_ptr = c->buffer;
737 c->buffer_end = c->buffer + c->buffer_size - 1; /* leave room for '\0' */
740 c->timeout = cur_time + RTSP_REQUEST_TIMEOUT;
741 c->state = RTSPSTATE_WAIT_REQUEST;
743 c->timeout = cur_time + HTTP_REQUEST_TIMEOUT;
744 c->state = HTTPSTATE_WAIT_REQUEST;
748 static void http_send_too_busy_reply(int fd)
751 int len = snprintf(buffer, sizeof(buffer),
752 "HTTP/1.0 503 Server too busy\r\n"
753 "Content-type: text/html\r\n"
755 "<html><head><title>Too busy</title></head><body>\r\n"
756 "<p>The server is too busy to serve your request at this time.</p>\r\n"
757 "<p>The number of current connections is %d, and this exceeds the limit of %d.</p>\r\n"
758 "</body></html>\r\n",
759 nb_connections, nb_max_connections);
760 send(fd, buffer, len, 0);
764 static void new_connection(int server_fd, int is_rtsp)
766 struct sockaddr_in from_addr;
768 HTTPContext *c = NULL;
770 len = sizeof(from_addr);
771 fd = accept(server_fd, (struct sockaddr *)&from_addr,
774 http_log("error during accept %s\n", strerror(errno));
777 ff_socket_nonblock(fd, 1);
779 if (nb_connections >= nb_max_connections) {
780 http_send_too_busy_reply(fd);
784 /* add a new connection */
785 c = av_mallocz(sizeof(HTTPContext));
790 c->poll_entry = NULL;
791 c->from_addr = from_addr;
792 c->buffer_size = IOBUFFER_INIT_SIZE;
793 c->buffer = av_malloc(c->buffer_size);
797 c->next = first_http_ctx;
801 start_wait_request(c, is_rtsp);
813 static void close_connection(HTTPContext *c)
815 HTTPContext **cp, *c1;
817 AVFormatContext *ctx;
821 /* remove connection from list */
822 cp = &first_http_ctx;
823 while ((*cp) != NULL) {
831 /* remove references, if any (XXX: do it faster) */
832 for(c1 = first_http_ctx; c1 != NULL; c1 = c1->next) {
837 /* remove connection associated resources */
841 /* close each frame parser */
842 for(i=0;i<c->fmt_in->nb_streams;i++) {
843 st = c->fmt_in->streams[i];
844 if (st->codec->codec)
845 avcodec_close(st->codec);
847 av_close_input_file(c->fmt_in);
850 /* free RTP output streams if any */
853 nb_streams = c->stream->nb_streams;
855 for(i=0;i<nb_streams;i++) {
858 av_write_trailer(ctx);
859 av_dict_free(&ctx->metadata);
860 av_free(ctx->streams[0]);
863 h = c->rtp_handles[i];
870 if (!c->last_packet_sent && c->state == HTTPSTATE_SEND_DATA_TRAILER) {
873 if (avio_open_dyn_buf(&ctx->pb) >= 0) {
874 av_write_trailer(ctx);
875 av_freep(&c->pb_buffer);
876 avio_close_dyn_buf(ctx->pb, &c->pb_buffer);
881 for(i=0; i<ctx->nb_streams; i++)
882 av_free(ctx->streams[i]);
884 if (c->stream && !c->post && c->stream->stream_type == STREAM_TYPE_LIVE)
885 current_bandwidth -= c->stream->bandwidth;
887 /* signal that there is no feed if we are the feeder socket */
888 if (c->state == HTTPSTATE_RECEIVE_DATA && c->stream) {
889 c->stream->feed_opened = 0;
893 av_freep(&c->pb_buffer);
894 av_freep(&c->packet_buffer);
900 static int handle_connection(HTTPContext *c)
905 case HTTPSTATE_WAIT_REQUEST:
906 case RTSPSTATE_WAIT_REQUEST:
908 if ((c->timeout - cur_time) < 0)
910 if (c->poll_entry->revents & (POLLERR | POLLHUP))
913 /* no need to read if no events */
914 if (!(c->poll_entry->revents & POLLIN))
918 len = recv(c->fd, c->buffer_ptr, 1, 0);
920 if (ff_neterrno() != AVERROR(EAGAIN) &&
921 ff_neterrno() != AVERROR(EINTR))
923 } else if (len == 0) {
926 /* search for end of request. */
928 c->buffer_ptr += len;
930 if ((ptr >= c->buffer + 2 && !memcmp(ptr-2, "\n\n", 2)) ||
931 (ptr >= c->buffer + 4 && !memcmp(ptr-4, "\r\n\r\n", 4))) {
932 /* request found : parse it and reply */
933 if (c->state == HTTPSTATE_WAIT_REQUEST) {
934 ret = http_parse_request(c);
936 ret = rtsp_parse_request(c);
940 } else if (ptr >= c->buffer_end) {
941 /* request too long: cannot do anything */
943 } else goto read_loop;
947 case HTTPSTATE_SEND_HEADER:
948 if (c->poll_entry->revents & (POLLERR | POLLHUP))
951 /* no need to write if no events */
952 if (!(c->poll_entry->revents & POLLOUT))
954 len = send(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr, 0);
956 if (ff_neterrno() != AVERROR(EAGAIN) &&
957 ff_neterrno() != AVERROR(EINTR)) {
958 /* error : close connection */
959 av_freep(&c->pb_buffer);
963 c->buffer_ptr += len;
965 c->stream->bytes_served += len;
966 c->data_count += len;
967 if (c->buffer_ptr >= c->buffer_end) {
968 av_freep(&c->pb_buffer);
972 /* all the buffer was sent : synchronize to the incoming stream */
973 c->state = HTTPSTATE_SEND_DATA_HEADER;
974 c->buffer_ptr = c->buffer_end = c->buffer;
979 case HTTPSTATE_SEND_DATA:
980 case HTTPSTATE_SEND_DATA_HEADER:
981 case HTTPSTATE_SEND_DATA_TRAILER:
982 /* for packetized output, we consider we can always write (the
983 input streams sets the speed). It may be better to verify
984 that we do not rely too much on the kernel queues */
985 if (!c->is_packetized) {
986 if (c->poll_entry->revents & (POLLERR | POLLHUP))
989 /* no need to read if no events */
990 if (!(c->poll_entry->revents & POLLOUT))
993 if (http_send_data(c) < 0)
995 /* close connection if trailer sent */
996 if (c->state == HTTPSTATE_SEND_DATA_TRAILER)
999 case HTTPSTATE_RECEIVE_DATA:
1000 /* no need to read if no events */
1001 if (c->poll_entry->revents & (POLLERR | POLLHUP))
1003 if (!(c->poll_entry->revents & POLLIN))
1005 if (http_receive_data(c) < 0)
1008 case HTTPSTATE_WAIT_FEED:
1009 /* no need to read if no events */
1010 if (c->poll_entry->revents & (POLLIN | POLLERR | POLLHUP))
1013 /* nothing to do, we'll be waken up by incoming feed packets */
1016 case RTSPSTATE_SEND_REPLY:
1017 if (c->poll_entry->revents & (POLLERR | POLLHUP)) {
1018 av_freep(&c->pb_buffer);
1021 /* no need to write if no events */
1022 if (!(c->poll_entry->revents & POLLOUT))
1024 len = send(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr, 0);
1026 if (ff_neterrno() != AVERROR(EAGAIN) &&
1027 ff_neterrno() != AVERROR(EINTR)) {
1028 /* error : close connection */
1029 av_freep(&c->pb_buffer);
1033 c->buffer_ptr += len;
1034 c->data_count += len;
1035 if (c->buffer_ptr >= c->buffer_end) {
1036 /* all the buffer was sent : wait for a new request */
1037 av_freep(&c->pb_buffer);
1038 start_wait_request(c, 1);
1042 case RTSPSTATE_SEND_PACKET:
1043 if (c->poll_entry->revents & (POLLERR | POLLHUP)) {
1044 av_freep(&c->packet_buffer);
1047 /* no need to write if no events */
1048 if (!(c->poll_entry->revents & POLLOUT))
1050 len = send(c->fd, c->packet_buffer_ptr,
1051 c->packet_buffer_end - c->packet_buffer_ptr, 0);
1053 if (ff_neterrno() != AVERROR(EAGAIN) &&
1054 ff_neterrno() != AVERROR(EINTR)) {
1055 /* error : close connection */
1056 av_freep(&c->packet_buffer);
1060 c->packet_buffer_ptr += len;
1061 if (c->packet_buffer_ptr >= c->packet_buffer_end) {
1062 /* all the buffer was sent : wait for a new request */
1063 av_freep(&c->packet_buffer);
1064 c->state = RTSPSTATE_WAIT_REQUEST;
1068 case HTTPSTATE_READY:
1077 static int extract_rates(char *rates, int ratelen, const char *request)
1081 for (p = request; *p && *p != '\r' && *p != '\n'; ) {
1082 if (strncasecmp(p, "Pragma:", 7) == 0) {
1083 const char *q = p + 7;
1085 while (*q && *q != '\n' && isspace(*q))
1088 if (strncasecmp(q, "stream-switch-entry=", 20) == 0) {
1094 memset(rates, 0xff, ratelen);
1097 while (*q && *q != '\n' && *q != ':')
1100 if (sscanf(q, ":%d:%d", &stream_no, &rate_no) != 2)
1104 if (stream_no < ratelen && stream_no >= 0)
1105 rates[stream_no] = rate_no;
1107 while (*q && *q != '\n' && !isspace(*q))
1114 p = strchr(p, '\n');
1124 static int find_stream_in_feed(FFStream *feed, AVCodecContext *codec, int bit_rate)
1127 int best_bitrate = 100000000;
1130 for (i = 0; i < feed->nb_streams; i++) {
1131 AVCodecContext *feed_codec = feed->streams[i]->codec;
1133 if (feed_codec->codec_id != codec->codec_id ||
1134 feed_codec->sample_rate != codec->sample_rate ||
1135 feed_codec->width != codec->width ||
1136 feed_codec->height != codec->height)
1139 /* Potential stream */
1141 /* We want the fastest stream less than bit_rate, or the slowest
1142 * faster than bit_rate
1145 if (feed_codec->bit_rate <= bit_rate) {
1146 if (best_bitrate > bit_rate || feed_codec->bit_rate > best_bitrate) {
1147 best_bitrate = feed_codec->bit_rate;
1151 if (feed_codec->bit_rate < best_bitrate) {
1152 best_bitrate = feed_codec->bit_rate;
1161 static int modify_current_stream(HTTPContext *c, char *rates)
1164 FFStream *req = c->stream;
1165 int action_required = 0;
1167 /* Not much we can do for a feed */
1171 for (i = 0; i < req->nb_streams; i++) {
1172 AVCodecContext *codec = req->streams[i]->codec;
1176 c->switch_feed_streams[i] = req->feed_streams[i];
1179 c->switch_feed_streams[i] = find_stream_in_feed(req->feed, codec, codec->bit_rate / 2);
1182 /* Wants off or slow */
1183 c->switch_feed_streams[i] = find_stream_in_feed(req->feed, codec, codec->bit_rate / 4);
1185 /* This doesn't work well when it turns off the only stream! */
1186 c->switch_feed_streams[i] = -2;
1187 c->feed_streams[i] = -2;
1192 if (c->switch_feed_streams[i] >= 0 && c->switch_feed_streams[i] != c->feed_streams[i])
1193 action_required = 1;
1196 return action_required;
1199 /* XXX: factorize in utils.c ? */
1200 /* XXX: take care with different space meaning */
1201 static void skip_spaces(const char **pp)
1205 while (*p == ' ' || *p == '\t')
1210 static void get_word(char *buf, int buf_size, const char **pp)
1218 while (!isspace(*p) && *p != '\0') {
1219 if ((q - buf) < buf_size - 1)
1228 static void get_arg(char *buf, int buf_size, const char **pp)
1235 while (isspace(*p)) p++;
1238 if (*p == '\"' || *p == '\'')
1250 if ((q - buf) < buf_size - 1)
1255 if (quote && *p == quote)
1260 static void parse_acl_row(FFStream *stream, FFStream* feed, IPAddressACL *ext_acl,
1261 const char *p, const char *filename, int line_num)
1267 get_arg(arg, sizeof(arg), &p);
1268 if (strcasecmp(arg, "allow") == 0)
1269 acl.action = IP_ALLOW;
1270 else if (strcasecmp(arg, "deny") == 0)
1271 acl.action = IP_DENY;
1273 fprintf(stderr, "%s:%d: ACL action '%s' is not ALLOW or DENY\n",
1274 filename, line_num, arg);
1278 get_arg(arg, sizeof(arg), &p);
1280 if (resolve_host(&acl.first, arg) != 0) {
1281 fprintf(stderr, "%s:%d: ACL refers to invalid host or ip address '%s'\n",
1282 filename, line_num, arg);
1285 acl.last = acl.first;
1287 get_arg(arg, sizeof(arg), &p);
1290 if (resolve_host(&acl.last, arg) != 0) {
1291 fprintf(stderr, "%s:%d: ACL refers to invalid host or ip address '%s'\n",
1292 filename, line_num, arg);
1298 IPAddressACL *nacl = av_mallocz(sizeof(*nacl));
1299 IPAddressACL **naclp = 0;
1305 naclp = &stream->acl;
1311 fprintf(stderr, "%s:%d: ACL found not in <stream> or <feed>\n",
1312 filename, line_num);
1318 naclp = &(*naclp)->next;
1326 static IPAddressACL* parse_dynamic_acl(FFStream *stream, HTTPContext *c)
1331 IPAddressACL *acl = NULL;
1335 f = fopen(stream->dynamic_acl, "r");
1337 perror(stream->dynamic_acl);
1341 acl = av_mallocz(sizeof(IPAddressACL));
1345 if (fgets(line, sizeof(line), f) == NULL)
1351 if (*p == '\0' || *p == '#')
1353 get_arg(cmd, sizeof(cmd), &p);
1355 if (!strcasecmp(cmd, "ACL"))
1356 parse_acl_row(NULL, NULL, acl, p, stream->dynamic_acl, line_num);
1363 static void free_acl_list(IPAddressACL *in_acl)
1365 IPAddressACL *pacl,*pacl2;
1375 static int validate_acl_list(IPAddressACL *in_acl, HTTPContext *c)
1377 enum IPAddressAction last_action = IP_DENY;
1379 struct in_addr *src = &c->from_addr.sin_addr;
1380 unsigned long src_addr = src->s_addr;
1382 for (acl = in_acl; acl; acl = acl->next) {
1383 if (src_addr >= acl->first.s_addr && src_addr <= acl->last.s_addr)
1384 return (acl->action == IP_ALLOW) ? 1 : 0;
1385 last_action = acl->action;
1388 /* Nothing matched, so return not the last action */
1389 return (last_action == IP_DENY) ? 1 : 0;
1392 static int validate_acl(FFStream *stream, HTTPContext *c)
1398 /* if stream->acl is null validate_acl_list will return 1 */
1399 ret = validate_acl_list(stream->acl, c);
1401 if (stream->dynamic_acl[0]) {
1402 acl = parse_dynamic_acl(stream, c);
1404 ret = validate_acl_list(acl, c);
1412 /* compute the real filename of a file by matching it without its
1413 extensions to all the stream filenames */
1414 static void compute_real_filename(char *filename, int max_size)
1421 /* compute filename by matching without the file extensions */
1422 av_strlcpy(file1, filename, sizeof(file1));
1423 p = strrchr(file1, '.');
1426 for(stream = first_stream; stream != NULL; stream = stream->next) {
1427 av_strlcpy(file2, stream->filename, sizeof(file2));
1428 p = strrchr(file2, '.');
1431 if (!strcmp(file1, file2)) {
1432 av_strlcpy(filename, stream->filename, max_size);
1447 /* parse http request and prepare header */
1448 static int http_parse_request(HTTPContext *c)
1451 enum RedirType redir_type;
1453 char info[1024], filename[1024];
1457 const char *mime_type;
1461 char *useragent = 0;
1464 get_word(cmd, sizeof(cmd), (const char **)&p);
1465 av_strlcpy(c->method, cmd, sizeof(c->method));
1467 if (!strcmp(cmd, "GET"))
1469 else if (!strcmp(cmd, "POST"))
1474 get_word(url, sizeof(url), (const char **)&p);
1475 av_strlcpy(c->url, url, sizeof(c->url));
1477 get_word(protocol, sizeof(protocol), (const char **)&p);
1478 if (strcmp(protocol, "HTTP/1.0") && strcmp(protocol, "HTTP/1.1"))
1481 av_strlcpy(c->protocol, protocol, sizeof(c->protocol));
1484 http_log("%s - - New connection: %s %s\n", inet_ntoa(c->from_addr.sin_addr), cmd, url);
1486 /* find the filename and the optional info string in the request */
1487 p = strchr(url, '?');
1489 av_strlcpy(info, p, sizeof(info));
1494 av_strlcpy(filename, url + ((*url == '/') ? 1 : 0), sizeof(filename)-1);
1496 for (p = c->buffer; *p && *p != '\r' && *p != '\n'; ) {
1497 if (strncasecmp(p, "User-Agent:", 11) == 0) {
1499 if (*useragent && *useragent != '\n' && isspace(*useragent))
1503 p = strchr(p, '\n');
1510 redir_type = REDIR_NONE;
1511 if (av_match_ext(filename, "asx")) {
1512 redir_type = REDIR_ASX;
1513 filename[strlen(filename)-1] = 'f';
1514 } else if (av_match_ext(filename, "asf") &&
1515 (!useragent || strncasecmp(useragent, "NSPlayer", 8) != 0)) {
1516 /* if this isn't WMP or lookalike, return the redirector file */
1517 redir_type = REDIR_ASF;
1518 } else if (av_match_ext(filename, "rpm,ram")) {
1519 redir_type = REDIR_RAM;
1520 strcpy(filename + strlen(filename)-2, "m");
1521 } else if (av_match_ext(filename, "rtsp")) {
1522 redir_type = REDIR_RTSP;
1523 compute_real_filename(filename, sizeof(filename) - 1);
1524 } else if (av_match_ext(filename, "sdp")) {
1525 redir_type = REDIR_SDP;
1526 compute_real_filename(filename, sizeof(filename) - 1);
1529 // "redirect" / request to index.html
1530 if (!strlen(filename))
1531 av_strlcpy(filename, "index.html", sizeof(filename) - 1);
1533 stream = first_stream;
1534 while (stream != NULL) {
1535 if (!strcmp(stream->filename, filename) && validate_acl(stream, c))
1537 stream = stream->next;
1539 if (stream == NULL) {
1540 snprintf(msg, sizeof(msg), "File '%s' not found", url);
1541 http_log("File '%s' not found\n", url);
1546 memcpy(c->feed_streams, stream->feed_streams, sizeof(c->feed_streams));
1547 memset(c->switch_feed_streams, -1, sizeof(c->switch_feed_streams));
1549 if (stream->stream_type == STREAM_TYPE_REDIRECT) {
1550 c->http_error = 301;
1552 q += snprintf(q, c->buffer_size,
1553 "HTTP/1.0 301 Moved\r\n"
1555 "Content-type: text/html\r\n"
1557 "<html><head><title>Moved</title></head><body>\r\n"
1558 "You should be <a href=\"%s\">redirected</a>.\r\n"
1559 "</body></html>\r\n", stream->feed_filename, stream->feed_filename);
1560 /* prepare output buffer */
1561 c->buffer_ptr = c->buffer;
1563 c->state = HTTPSTATE_SEND_HEADER;
1567 /* If this is WMP, get the rate information */
1568 if (extract_rates(ratebuf, sizeof(ratebuf), c->buffer)) {
1569 if (modify_current_stream(c, ratebuf)) {
1570 for (i = 0; i < FF_ARRAY_ELEMS(c->feed_streams); i++) {
1571 if (c->switch_feed_streams[i] >= 0)
1572 c->switch_feed_streams[i] = -1;
1577 if (c->post == 0 && stream->stream_type == STREAM_TYPE_LIVE)
1578 current_bandwidth += stream->bandwidth;
1580 /* If already streaming this feed, do not let start another feeder. */
1581 if (stream->feed_opened) {
1582 snprintf(msg, sizeof(msg), "This feed is already being received.");
1583 http_log("Feed '%s' already being received\n", stream->feed_filename);
1587 if (c->post == 0 && max_bandwidth < current_bandwidth) {
1588 c->http_error = 503;
1590 q += snprintf(q, c->buffer_size,
1591 "HTTP/1.0 503 Server too busy\r\n"
1592 "Content-type: text/html\r\n"
1594 "<html><head><title>Too busy</title></head><body>\r\n"
1595 "<p>The server is too busy to serve your request at this time.</p>\r\n"
1596 "<p>The bandwidth being served (including your stream) is %"PRIu64"kbit/sec, "
1597 "and this exceeds the limit of %"PRIu64"kbit/sec.</p>\r\n"
1598 "</body></html>\r\n", current_bandwidth, max_bandwidth);
1599 /* prepare output buffer */
1600 c->buffer_ptr = c->buffer;
1602 c->state = HTTPSTATE_SEND_HEADER;
1606 if (redir_type != REDIR_NONE) {
1609 for (p = c->buffer; *p && *p != '\r' && *p != '\n'; ) {
1610 if (strncasecmp(p, "Host:", 5) == 0) {
1614 p = strchr(p, '\n');
1625 while (isspace(*hostinfo))
1628 eoh = strchr(hostinfo, '\n');
1630 if (eoh[-1] == '\r')
1633 if (eoh - hostinfo < sizeof(hostbuf) - 1) {
1634 memcpy(hostbuf, hostinfo, eoh - hostinfo);
1635 hostbuf[eoh - hostinfo] = 0;
1637 c->http_error = 200;
1639 switch(redir_type) {
1641 q += snprintf(q, c->buffer_size,
1642 "HTTP/1.0 200 ASX Follows\r\n"
1643 "Content-type: video/x-ms-asf\r\n"
1645 "<ASX Version=\"3\">\r\n"
1646 //"<!-- Autogenerated by ffserver -->\r\n"
1647 "<ENTRY><REF HREF=\"http://%s/%s%s\"/></ENTRY>\r\n"
1648 "</ASX>\r\n", hostbuf, filename, info);
1651 q += snprintf(q, c->buffer_size,
1652 "HTTP/1.0 200 RAM Follows\r\n"
1653 "Content-type: audio/x-pn-realaudio\r\n"
1655 "# Autogenerated by ffserver\r\n"
1656 "http://%s/%s%s\r\n", hostbuf, filename, info);
1659 q += snprintf(q, c->buffer_size,
1660 "HTTP/1.0 200 ASF Redirect follows\r\n"
1661 "Content-type: video/x-ms-asf\r\n"
1664 "Ref1=http://%s/%s%s\r\n", hostbuf, filename, info);
1668 char hostname[256], *p;
1669 /* extract only hostname */
1670 av_strlcpy(hostname, hostbuf, sizeof(hostname));
1671 p = strrchr(hostname, ':');
1674 q += snprintf(q, c->buffer_size,
1675 "HTTP/1.0 200 RTSP Redirect follows\r\n"
1676 /* XXX: incorrect mime type ? */
1677 "Content-type: application/x-rtsp\r\n"
1679 "rtsp://%s:%d/%s\r\n", hostname, ntohs(my_rtsp_addr.sin_port), filename);
1685 int sdp_data_size, len;
1686 struct sockaddr_in my_addr;
1688 q += snprintf(q, c->buffer_size,
1689 "HTTP/1.0 200 OK\r\n"
1690 "Content-type: application/sdp\r\n"
1693 len = sizeof(my_addr);
1694 getsockname(c->fd, (struct sockaddr *)&my_addr, &len);
1696 /* XXX: should use a dynamic buffer */
1697 sdp_data_size = prepare_sdp_description(stream,
1700 if (sdp_data_size > 0) {
1701 memcpy(q, sdp_data, sdp_data_size);
1713 /* prepare output buffer */
1714 c->buffer_ptr = c->buffer;
1716 c->state = HTTPSTATE_SEND_HEADER;
1722 snprintf(msg, sizeof(msg), "ASX/RAM file not handled");
1726 stream->conns_served++;
1728 /* XXX: add there authenticate and IP match */
1731 /* if post, it means a feed is being sent */
1732 if (!stream->is_feed) {
1733 /* However it might be a status report from WMP! Let us log the
1734 * data as it might come in handy one day. */
1738 for (p = c->buffer; *p && *p != '\r' && *p != '\n'; ) {
1739 if (strncasecmp(p, "Pragma: log-line=", 17) == 0) {
1743 if (strncasecmp(p, "Pragma: client-id=", 18) == 0)
1744 client_id = strtol(p + 18, 0, 10);
1745 p = strchr(p, '\n');
1753 char *eol = strchr(logline, '\n');
1758 if (eol[-1] == '\r')
1760 http_log("%.*s\n", (int) (eol - logline), logline);
1761 c->suppress_log = 1;
1766 http_log("\nGot request:\n%s\n", c->buffer);
1769 if (client_id && extract_rates(ratebuf, sizeof(ratebuf), c->buffer)) {
1772 /* Now we have to find the client_id */
1773 for (wmpc = first_http_ctx; wmpc; wmpc = wmpc->next) {
1774 if (wmpc->wmp_client_id == client_id)
1778 if (wmpc && modify_current_stream(wmpc, ratebuf))
1779 wmpc->switch_pending = 1;
1782 snprintf(msg, sizeof(msg), "POST command not handled");
1786 if (http_start_receive_data(c) < 0) {
1787 snprintf(msg, sizeof(msg), "could not open feed");
1791 c->state = HTTPSTATE_RECEIVE_DATA;
1796 if (strcmp(stream->filename + strlen(stream->filename) - 4, ".asf") == 0)
1797 http_log("\nGot request:\n%s\n", c->buffer);
1800 if (c->stream->stream_type == STREAM_TYPE_STATUS)
1803 /* open input stream */
1804 if (open_input_stream(c, info) < 0) {
1805 snprintf(msg, sizeof(msg), "Input stream corresponding to '%s' not found", url);
1809 /* prepare http header */
1811 q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "HTTP/1.0 200 OK\r\n");
1812 mime_type = c->stream->fmt->mime_type;
1814 mime_type = "application/x-octet-stream";
1815 q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Pragma: no-cache\r\n");
1817 /* for asf, we need extra headers */
1818 if (!strcmp(c->stream->fmt->name,"asf_stream")) {
1819 /* Need to allocate a client id */
1821 c->wmp_client_id = av_lfg_get(&random_state);
1823 q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Server: Cougar 4.1.0.3923\r\nCache-Control: no-cache\r\nPragma: client-id=%d\r\nPragma: features=\"broadcast\"\r\n", c->wmp_client_id);
1825 q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Content-Type: %s\r\n", mime_type);
1826 q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "\r\n");
1828 /* prepare output buffer */
1830 c->buffer_ptr = c->buffer;
1832 c->state = HTTPSTATE_SEND_HEADER;
1835 c->http_error = 404;
1837 q += snprintf(q, c->buffer_size,
1838 "HTTP/1.0 404 Not Found\r\n"
1839 "Content-type: text/html\r\n"
1842 "<head><title>404 Not Found</title></head>\n"
1845 /* prepare output buffer */
1846 c->buffer_ptr = c->buffer;
1848 c->state = HTTPSTATE_SEND_HEADER;
1852 c->http_error = 200; /* horrible : we use this value to avoid
1853 going to the send data state */
1854 c->state = HTTPSTATE_SEND_HEADER;
1858 static void fmt_bytecount(AVIOContext *pb, int64_t count)
1860 static const char *suffix = " kMGTP";
1863 for (s = suffix; count >= 100000 && s[1]; count /= 1000, s++);
1865 avio_printf(pb, "%"PRId64"%c", count, *s);
1868 static void compute_status(HTTPContext *c)
1877 if (avio_open_dyn_buf(&pb) < 0) {
1878 /* XXX: return an error ? */
1879 c->buffer_ptr = c->buffer;
1880 c->buffer_end = c->buffer;
1884 avio_printf(pb, "HTTP/1.0 200 OK\r\n");
1885 avio_printf(pb, "Content-type: %s\r\n", "text/html");
1886 avio_printf(pb, "Pragma: no-cache\r\n");
1887 avio_printf(pb, "\r\n");
1889 avio_printf(pb, "<html><head><title>%s Status</title>\n", program_name);
1890 if (c->stream->feed_filename[0])
1891 avio_printf(pb, "<link rel=\"shortcut icon\" href=\"%s\">\n", c->stream->feed_filename);
1892 avio_printf(pb, "</head>\n<body>");
1893 avio_printf(pb, "<h1>%s Status</h1>\n", program_name);
1895 avio_printf(pb, "<h2>Available Streams</h2>\n");
1896 avio_printf(pb, "<table cellspacing=0 cellpadding=4>\n");
1897 avio_printf(pb, "<tr><th valign=top>Path<th align=left>Served<br>Conns<th><br>bytes<th valign=top>Format<th>Bit rate<br>kbits/s<th align=left>Video<br>kbits/s<th><br>Codec<th align=left>Audio<br>kbits/s<th><br>Codec<th align=left valign=top>Feed\n");
1898 stream = first_stream;
1899 while (stream != NULL) {
1900 char sfilename[1024];
1903 if (stream->feed != stream) {
1904 av_strlcpy(sfilename, stream->filename, sizeof(sfilename) - 10);
1905 eosf = sfilename + strlen(sfilename);
1906 if (eosf - sfilename >= 4) {
1907 if (strcmp(eosf - 4, ".asf") == 0)
1908 strcpy(eosf - 4, ".asx");
1909 else if (strcmp(eosf - 3, ".rm") == 0)
1910 strcpy(eosf - 3, ".ram");
1911 else if (stream->fmt && !strcmp(stream->fmt->name, "rtp")) {
1912 /* generate a sample RTSP director if
1913 unicast. Generate an SDP redirector if
1915 eosf = strrchr(sfilename, '.');
1917 eosf = sfilename + strlen(sfilename);
1918 if (stream->is_multicast)
1919 strcpy(eosf, ".sdp");
1921 strcpy(eosf, ".rtsp");
1925 avio_printf(pb, "<tr><td><a href=\"/%s\">%s</a> ",
1926 sfilename, stream->filename);
1927 avio_printf(pb, "<td align=right> %d <td align=right> ",
1928 stream->conns_served);
1929 fmt_bytecount(pb, stream->bytes_served);
1930 switch(stream->stream_type) {
1931 case STREAM_TYPE_LIVE: {
1932 int audio_bit_rate = 0;
1933 int video_bit_rate = 0;
1934 const char *audio_codec_name = "";
1935 const char *video_codec_name = "";
1936 const char *audio_codec_name_extra = "";
1937 const char *video_codec_name_extra = "";
1939 for(i=0;i<stream->nb_streams;i++) {
1940 AVStream *st = stream->streams[i];
1941 AVCodec *codec = avcodec_find_encoder(st->codec->codec_id);
1942 switch(st->codec->codec_type) {
1943 case AVMEDIA_TYPE_AUDIO:
1944 audio_bit_rate += st->codec->bit_rate;
1946 if (*audio_codec_name)
1947 audio_codec_name_extra = "...";
1948 audio_codec_name = codec->name;
1951 case AVMEDIA_TYPE_VIDEO:
1952 video_bit_rate += st->codec->bit_rate;
1954 if (*video_codec_name)
1955 video_codec_name_extra = "...";
1956 video_codec_name = codec->name;
1959 case AVMEDIA_TYPE_DATA:
1960 video_bit_rate += st->codec->bit_rate;
1966 avio_printf(pb, "<td align=center> %s <td align=right> %d <td align=right> %d <td> %s %s <td align=right> %d <td> %s %s",
1969 video_bit_rate / 1000, video_codec_name, video_codec_name_extra,
1970 audio_bit_rate / 1000, audio_codec_name, audio_codec_name_extra);
1972 avio_printf(pb, "<td>%s", stream->feed->filename);
1974 avio_printf(pb, "<td>%s", stream->feed_filename);
1975 avio_printf(pb, "\n");
1979 avio_printf(pb, "<td align=center> - <td align=right> - <td align=right> - <td><td align=right> - <td>\n");
1983 stream = stream->next;
1985 avio_printf(pb, "</table>\n");
1987 stream = first_stream;
1988 while (stream != NULL) {
1989 if (stream->feed == stream) {
1990 avio_printf(pb, "<h2>Feed %s</h2>", stream->filename);
1992 avio_printf(pb, "Running as pid %d.\n", stream->pid);
1994 #if defined(linux) && !defined(CONFIG_NOCUTILS)
1999 /* This is somewhat linux specific I guess */
2000 snprintf(ps_cmd, sizeof(ps_cmd),
2001 "ps -o \"%%cpu,cputime\" --no-headers %d",
2004 pid_stat = popen(ps_cmd, "r");
2009 if (fscanf(pid_stat, "%10s %64s", cpuperc,
2011 avio_printf(pb, "Currently using %s%% of the cpu. Total time used %s.\n",
2019 avio_printf(pb, "<p>");
2021 avio_printf(pb, "<table cellspacing=0 cellpadding=4><tr><th>Stream<th>type<th>kbits/s<th align=left>codec<th align=left>Parameters\n");
2023 for (i = 0; i < stream->nb_streams; i++) {
2024 AVStream *st = stream->streams[i];
2025 AVCodec *codec = avcodec_find_encoder(st->codec->codec_id);
2026 const char *type = "unknown";
2027 char parameters[64];
2031 switch(st->codec->codec_type) {
2032 case AVMEDIA_TYPE_AUDIO:
2034 snprintf(parameters, sizeof(parameters), "%d channel(s), %d Hz", st->codec->channels, st->codec->sample_rate);
2036 case AVMEDIA_TYPE_VIDEO:
2038 snprintf(parameters, sizeof(parameters), "%dx%d, q=%d-%d, fps=%d", st->codec->width, st->codec->height,
2039 st->codec->qmin, st->codec->qmax, st->codec->time_base.den / st->codec->time_base.num);
2044 avio_printf(pb, "<tr><td align=right>%d<td>%s<td align=right>%d<td>%s<td>%s\n",
2045 i, type, st->codec->bit_rate/1000, codec ? codec->name : "", parameters);
2047 avio_printf(pb, "</table>\n");
2050 stream = stream->next;
2053 /* connection status */
2054 avio_printf(pb, "<h2>Connection Status</h2>\n");
2056 avio_printf(pb, "Number of connections: %d / %d<br>\n",
2057 nb_connections, nb_max_connections);
2059 avio_printf(pb, "Bandwidth in use: %"PRIu64"k / %"PRIu64"k<br>\n",
2060 current_bandwidth, max_bandwidth);
2062 avio_printf(pb, "<table>\n");
2063 avio_printf(pb, "<tr><th>#<th>File<th>IP<th>Proto<th>State<th>Target bits/sec<th>Actual bits/sec<th>Bytes transferred\n");
2064 c1 = first_http_ctx;
2066 while (c1 != NULL) {
2072 for (j = 0; j < c1->stream->nb_streams; j++) {
2073 if (!c1->stream->feed)
2074 bitrate += c1->stream->streams[j]->codec->bit_rate;
2075 else if (c1->feed_streams[j] >= 0)
2076 bitrate += c1->stream->feed->streams[c1->feed_streams[j]]->codec->bit_rate;
2081 p = inet_ntoa(c1->from_addr.sin_addr);
2082 avio_printf(pb, "<tr><td><b>%d</b><td>%s%s<td>%s<td>%s<td>%s<td align=right>",
2084 c1->stream ? c1->stream->filename : "",
2085 c1->state == HTTPSTATE_RECEIVE_DATA ? "(input)" : "",
2088 http_state[c1->state]);
2089 fmt_bytecount(pb, bitrate);
2090 avio_printf(pb, "<td align=right>");
2091 fmt_bytecount(pb, compute_datarate(&c1->datarate, c1->data_count) * 8);
2092 avio_printf(pb, "<td align=right>");
2093 fmt_bytecount(pb, c1->data_count);
2094 avio_printf(pb, "\n");
2097 avio_printf(pb, "</table>\n");
2102 avio_printf(pb, "<hr size=1 noshade>Generated at %s", p);
2103 avio_printf(pb, "</body>\n</html>\n");
2105 len = avio_close_dyn_buf(pb, &c->pb_buffer);
2106 c->buffer_ptr = c->pb_buffer;
2107 c->buffer_end = c->pb_buffer + len;
2110 /* check if the parser needs to be opened for stream i */
2111 static void open_parser(AVFormatContext *s, int i)
2113 AVStream *st = s->streams[i];
2116 if (!st->codec->codec) {
2117 codec = avcodec_find_decoder(st->codec->codec_id);
2118 if (codec && (codec->capabilities & CODEC_CAP_PARSE_ONLY)) {
2119 st->codec->parse_only = 1;
2120 if (avcodec_open2(st->codec, codec, NULL) < 0)
2121 st->codec->parse_only = 0;
2126 static int open_input_stream(HTTPContext *c, const char *info)
2129 char input_filename[1024];
2130 AVFormatContext *s = NULL;
2134 /* find file name */
2135 if (c->stream->feed) {
2136 strcpy(input_filename, c->stream->feed->feed_filename);
2137 /* compute position (absolute time) */
2138 if (av_find_info_tag(buf, sizeof(buf), "date", info)) {
2139 if ((ret = av_parse_time(&stream_pos, buf, 0)) < 0)
2141 } else if (av_find_info_tag(buf, sizeof(buf), "buffer", info)) {
2142 int prebuffer = strtol(buf, 0, 10);
2143 stream_pos = av_gettime() - prebuffer * (int64_t)1000000;
2145 stream_pos = av_gettime() - c->stream->prebuffer * (int64_t)1000;
2147 strcpy(input_filename, c->stream->feed_filename);
2148 /* compute position (relative time) */
2149 if (av_find_info_tag(buf, sizeof(buf), "date", info)) {
2150 if ((ret = av_parse_time(&stream_pos, buf, 1)) < 0)
2155 if (input_filename[0] == '\0')
2159 if ((ret = avformat_open_input(&s, input_filename, c->stream->ifmt, &c->stream->in_opts)) < 0) {
2160 http_log("could not open %s: %d\n", input_filename, ret);
2163 s->flags |= AVFMT_FLAG_GENPTS;
2165 if (strcmp(s->iformat->name, "ffm") && avformat_find_stream_info(c->fmt_in, NULL) < 0) {
2166 http_log("Could not find stream info '%s'\n", input_filename);
2167 av_close_input_file(s);
2171 /* open each parser */
2172 for(i=0;i<s->nb_streams;i++)
2175 /* choose stream as clock source (we favorize video stream if
2176 present) for packet sending */
2177 c->pts_stream_index = 0;
2178 for(i=0;i<c->stream->nb_streams;i++) {
2179 if (c->pts_stream_index == 0 &&
2180 c->stream->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
2181 c->pts_stream_index = i;
2185 if (c->fmt_in->iformat->read_seek)
2186 av_seek_frame(c->fmt_in, -1, stream_pos, 0);
2187 /* set the start time (needed for maxtime and RTP packet timing) */
2188 c->start_time = cur_time;
2189 c->first_pts = AV_NOPTS_VALUE;
2193 /* return the server clock (in us) */
2194 static int64_t get_server_clock(HTTPContext *c)
2196 /* compute current pts value from system time */
2197 return (cur_time - c->start_time) * 1000;
2200 /* return the estimated time at which the current packet must be sent
2202 static int64_t get_packet_send_clock(HTTPContext *c)
2204 int bytes_left, bytes_sent, frame_bytes;
2206 frame_bytes = c->cur_frame_bytes;
2207 if (frame_bytes <= 0)
2210 bytes_left = c->buffer_end - c->buffer_ptr;
2211 bytes_sent = frame_bytes - bytes_left;
2212 return c->cur_pts + (c->cur_frame_duration * bytes_sent) / frame_bytes;
2217 static int http_prepare_data(HTTPContext *c)
2220 AVFormatContext *ctx;
2222 av_freep(&c->pb_buffer);
2224 case HTTPSTATE_SEND_DATA_HEADER:
2225 memset(&c->fmt_ctx, 0, sizeof(c->fmt_ctx));
2226 av_dict_set(&c->fmt_ctx.metadata, "author" , c->stream->author , 0);
2227 av_dict_set(&c->fmt_ctx.metadata, "comment" , c->stream->comment , 0);
2228 av_dict_set(&c->fmt_ctx.metadata, "copyright", c->stream->copyright, 0);
2229 av_dict_set(&c->fmt_ctx.metadata, "title" , c->stream->title , 0);
2231 c->fmt_ctx.streams = av_mallocz(sizeof(AVStream *) * c->stream->nb_streams);
2233 for(i=0;i<c->stream->nb_streams;i++) {
2235 c->fmt_ctx.streams[i] = av_mallocz(sizeof(AVStream));
2236 /* if file or feed, then just take streams from FFStream struct */
2237 if (!c->stream->feed ||
2238 c->stream->feed == c->stream)
2239 src = c->stream->streams[i];
2241 src = c->stream->feed->streams[c->stream->feed_streams[i]];
2243 *(c->fmt_ctx.streams[i]) = *src;
2244 c->fmt_ctx.streams[i]->priv_data = 0;
2245 c->fmt_ctx.streams[i]->codec->frame_number = 0; /* XXX: should be done in
2246 AVStream, not in codec */
2248 /* set output format parameters */
2249 c->fmt_ctx.oformat = c->stream->fmt;
2250 c->fmt_ctx.nb_streams = c->stream->nb_streams;
2252 c->got_key_frame = 0;
2254 /* prepare header and save header data in a stream */
2255 if (avio_open_dyn_buf(&c->fmt_ctx.pb) < 0) {
2256 /* XXX: potential leak */
2259 c->fmt_ctx.pb->seekable = 0;
2262 * HACK to avoid mpeg ps muxer to spit many underflow errors
2263 * Default value from FFmpeg
2264 * Try to set it use configuration option
2266 c->fmt_ctx.preload = (int)(0.5*AV_TIME_BASE);
2267 c->fmt_ctx.max_delay = (int)(0.7*AV_TIME_BASE);
2269 if (avformat_write_header(&c->fmt_ctx, NULL) < 0) {
2270 http_log("Error writing output header\n");
2273 av_dict_free(&c->fmt_ctx.metadata);
2275 len = avio_close_dyn_buf(c->fmt_ctx.pb, &c->pb_buffer);
2276 c->buffer_ptr = c->pb_buffer;
2277 c->buffer_end = c->pb_buffer + len;
2279 c->state = HTTPSTATE_SEND_DATA;
2280 c->last_packet_sent = 0;
2282 case HTTPSTATE_SEND_DATA:
2283 /* find a new packet */
2284 /* read a packet from the input stream */
2285 if (c->stream->feed)
2286 ffm_set_write_index(c->fmt_in,
2287 c->stream->feed->feed_write_index,
2288 c->stream->feed->feed_size);
2290 if (c->stream->max_time &&
2291 c->stream->max_time + c->start_time - cur_time < 0)
2292 /* We have timed out */
2293 c->state = HTTPSTATE_SEND_DATA_TRAILER;
2297 ret = av_read_frame(c->fmt_in, &pkt);
2299 if (c->stream->feed) {
2300 /* if coming from feed, it means we reached the end of the
2301 ffm file, so must wait for more data */
2302 c->state = HTTPSTATE_WAIT_FEED;
2303 return 1; /* state changed */
2304 } else if (ret == AVERROR(EAGAIN)) {
2305 /* input not ready, come back later */
2308 if (c->stream->loop) {
2309 av_close_input_file(c->fmt_in);
2311 if (open_input_stream(c, "") < 0)
2316 /* must send trailer now because eof or error */
2317 c->state = HTTPSTATE_SEND_DATA_TRAILER;
2321 int source_index = pkt.stream_index;
2322 /* update first pts if needed */
2323 if (c->first_pts == AV_NOPTS_VALUE) {
2324 c->first_pts = av_rescale_q(pkt.dts, c->fmt_in->streams[pkt.stream_index]->time_base, AV_TIME_BASE_Q);
2325 c->start_time = cur_time;
2327 /* send it to the appropriate stream */
2328 if (c->stream->feed) {
2329 /* if coming from a feed, select the right stream */
2330 if (c->switch_pending) {
2331 c->switch_pending = 0;
2332 for(i=0;i<c->stream->nb_streams;i++) {
2333 if (c->switch_feed_streams[i] == pkt.stream_index)
2334 if (pkt.flags & AV_PKT_FLAG_KEY)
2335 c->switch_feed_streams[i] = -1;
2336 if (c->switch_feed_streams[i] >= 0)
2337 c->switch_pending = 1;
2340 for(i=0;i<c->stream->nb_streams;i++) {
2341 if (c->stream->feed_streams[i] == pkt.stream_index) {
2342 AVStream *st = c->fmt_in->streams[source_index];
2343 pkt.stream_index = i;
2344 if (pkt.flags & AV_PKT_FLAG_KEY &&
2345 (st->codec->codec_type == AVMEDIA_TYPE_VIDEO ||
2346 c->stream->nb_streams == 1))
2347 c->got_key_frame = 1;
2348 if (!c->stream->send_on_key || c->got_key_frame)
2353 AVCodecContext *codec;
2354 AVStream *ist, *ost;
2356 ist = c->fmt_in->streams[source_index];
2357 /* specific handling for RTP: we use several
2358 output stream (one for each RTP
2359 connection). XXX: need more abstract handling */
2360 if (c->is_packetized) {
2361 /* compute send time and duration */
2362 c->cur_pts = av_rescale_q(pkt.dts, ist->time_base, AV_TIME_BASE_Q);
2363 c->cur_pts -= c->first_pts;
2364 c->cur_frame_duration = av_rescale_q(pkt.duration, ist->time_base, AV_TIME_BASE_Q);
2365 /* find RTP context */
2366 c->packet_stream_index = pkt.stream_index;
2367 ctx = c->rtp_ctx[c->packet_stream_index];
2369 av_free_packet(&pkt);
2372 codec = ctx->streams[0]->codec;
2373 /* only one stream per RTP connection */
2374 pkt.stream_index = 0;
2378 codec = ctx->streams[pkt.stream_index]->codec;
2381 if (c->is_packetized) {
2382 int max_packet_size;
2383 if (c->rtp_protocol == RTSP_LOWER_TRANSPORT_TCP)
2384 max_packet_size = RTSP_TCP_MAX_PACKET_SIZE;
2386 max_packet_size = url_get_max_packet_size(c->rtp_handles[c->packet_stream_index]);
2387 ret = ffio_open_dyn_packet_buf(&ctx->pb, max_packet_size);
2389 ret = avio_open_dyn_buf(&ctx->pb);
2392 /* XXX: potential leak */
2395 ost = ctx->streams[pkt.stream_index];
2397 ctx->pb->seekable = 0;
2398 if (pkt.dts != AV_NOPTS_VALUE)
2399 pkt.dts = av_rescale_q(pkt.dts, ist->time_base, ost->time_base);
2400 if (pkt.pts != AV_NOPTS_VALUE)
2401 pkt.pts = av_rescale_q(pkt.pts, ist->time_base, ost->time_base);
2402 pkt.duration = av_rescale_q(pkt.duration, ist->time_base, ost->time_base);
2403 if (av_write_frame(ctx, &pkt) < 0) {
2404 http_log("Error writing frame to output\n");
2405 c->state = HTTPSTATE_SEND_DATA_TRAILER;
2408 len = avio_close_dyn_buf(ctx->pb, &c->pb_buffer);
2409 c->cur_frame_bytes = len;
2410 c->buffer_ptr = c->pb_buffer;
2411 c->buffer_end = c->pb_buffer + len;
2413 codec->frame_number++;
2415 av_free_packet(&pkt);
2419 av_free_packet(&pkt);
2424 case HTTPSTATE_SEND_DATA_TRAILER:
2425 /* last packet test ? */
2426 if (c->last_packet_sent || c->is_packetized)
2429 /* prepare header */
2430 if (avio_open_dyn_buf(&ctx->pb) < 0) {
2431 /* XXX: potential leak */
2434 c->fmt_ctx.pb->seekable = 0;
2435 av_write_trailer(ctx);
2436 len = avio_close_dyn_buf(ctx->pb, &c->pb_buffer);
2437 c->buffer_ptr = c->pb_buffer;
2438 c->buffer_end = c->pb_buffer + len;
2440 c->last_packet_sent = 1;
2446 /* should convert the format at the same time */
2447 /* send data starting at c->buffer_ptr to the output connection
2448 (either UDP or TCP connection) */
2449 static int http_send_data(HTTPContext *c)
2454 if (c->buffer_ptr >= c->buffer_end) {
2455 ret = http_prepare_data(c);
2459 /* state change requested */
2462 if (c->is_packetized) {
2463 /* RTP data output */
2464 len = c->buffer_end - c->buffer_ptr;
2466 /* fail safe - should never happen */
2468 c->buffer_ptr = c->buffer_end;
2471 len = (c->buffer_ptr[0] << 24) |
2472 (c->buffer_ptr[1] << 16) |
2473 (c->buffer_ptr[2] << 8) |
2475 if (len > (c->buffer_end - c->buffer_ptr))
2477 if ((get_packet_send_clock(c) - get_server_clock(c)) > 0) {
2478 /* nothing to send yet: we can wait */
2482 c->data_count += len;
2483 update_datarate(&c->datarate, c->data_count);
2485 c->stream->bytes_served += len;
2487 if (c->rtp_protocol == RTSP_LOWER_TRANSPORT_TCP) {
2488 /* RTP packets are sent inside the RTSP TCP connection */
2490 int interleaved_index, size;
2492 HTTPContext *rtsp_c;
2495 /* if no RTSP connection left, error */
2498 /* if already sending something, then wait. */
2499 if (rtsp_c->state != RTSPSTATE_WAIT_REQUEST)
2501 if (avio_open_dyn_buf(&pb) < 0)
2503 interleaved_index = c->packet_stream_index * 2;
2504 /* RTCP packets are sent at odd indexes */
2505 if (c->buffer_ptr[1] == 200)
2506 interleaved_index++;
2507 /* write RTSP TCP header */
2509 header[1] = interleaved_index;
2510 header[2] = len >> 8;
2512 avio_write(pb, header, 4);
2513 /* write RTP packet data */
2515 avio_write(pb, c->buffer_ptr, len);
2516 size = avio_close_dyn_buf(pb, &c->packet_buffer);
2517 /* prepare asynchronous TCP sending */
2518 rtsp_c->packet_buffer_ptr = c->packet_buffer;
2519 rtsp_c->packet_buffer_end = c->packet_buffer + size;
2520 c->buffer_ptr += len;
2522 /* send everything we can NOW */
2523 len = send(rtsp_c->fd, rtsp_c->packet_buffer_ptr,
2524 rtsp_c->packet_buffer_end - rtsp_c->packet_buffer_ptr, 0);
2526 rtsp_c->packet_buffer_ptr += len;
2527 if (rtsp_c->packet_buffer_ptr < rtsp_c->packet_buffer_end) {
2528 /* if we could not send all the data, we will
2529 send it later, so a new state is needed to
2530 "lock" the RTSP TCP connection */
2531 rtsp_c->state = RTSPSTATE_SEND_PACKET;
2534 /* all data has been sent */
2535 av_freep(&c->packet_buffer);
2537 /* send RTP packet directly in UDP */
2539 url_write(c->rtp_handles[c->packet_stream_index],
2540 c->buffer_ptr, len);
2541 c->buffer_ptr += len;
2542 /* here we continue as we can send several packets per 10 ms slot */
2545 /* TCP data output */
2546 len = send(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr, 0);
2548 if (ff_neterrno() != AVERROR(EAGAIN) &&
2549 ff_neterrno() != AVERROR(EINTR))
2550 /* error : close connection */
2555 c->buffer_ptr += len;
2557 c->data_count += len;
2558 update_datarate(&c->datarate, c->data_count);
2560 c->stream->bytes_served += len;
2568 static int http_start_receive_data(HTTPContext *c)
2572 if (c->stream->feed_opened)
2575 /* Don't permit writing to this one */
2576 if (c->stream->readonly)
2580 fd = open(c->stream->feed_filename, O_RDWR);
2582 http_log("Error opening feeder file: %s\n", strerror(errno));
2587 if (c->stream->truncate) {
2588 /* truncate feed file */
2589 ffm_write_write_index(c->feed_fd, FFM_PACKET_SIZE);
2590 ftruncate(c->feed_fd, FFM_PACKET_SIZE);
2591 http_log("Truncating feed file '%s'\n", c->stream->feed_filename);
2593 if ((c->stream->feed_write_index = ffm_read_write_index(fd)) < 0) {
2594 http_log("Error reading write index from feed file: %s\n", strerror(errno));
2599 c->stream->feed_write_index = FFMAX(ffm_read_write_index(fd), FFM_PACKET_SIZE);
2600 c->stream->feed_size = lseek(fd, 0, SEEK_END);
2601 lseek(fd, 0, SEEK_SET);
2603 /* init buffer input */
2604 c->buffer_ptr = c->buffer;
2605 c->buffer_end = c->buffer + FFM_PACKET_SIZE;
2606 c->stream->feed_opened = 1;
2607 c->chunked_encoding = !!av_stristr(c->buffer, "Transfer-Encoding: chunked");
2611 static int http_receive_data(HTTPContext *c)
2614 int len, loop_run = 0;
2616 while (c->chunked_encoding && !c->chunk_size &&
2617 c->buffer_end > c->buffer_ptr) {
2618 /* read chunk header, if present */
2619 len = recv(c->fd, c->buffer_ptr, 1, 0);
2622 if (ff_neterrno() != AVERROR(EAGAIN) &&
2623 ff_neterrno() != AVERROR(EINTR))
2624 /* error : close connection */
2627 } else if (len == 0) {
2628 /* end of connection : close it */
2630 } else if (c->buffer_ptr - c->buffer >= 2 &&
2631 !memcmp(c->buffer_ptr - 1, "\r\n", 2)) {
2632 c->chunk_size = strtol(c->buffer, 0, 16);
2633 if (c->chunk_size == 0) // end of stream
2635 c->buffer_ptr = c->buffer;
2637 } else if (++loop_run > 10) {
2638 /* no chunk header, abort */
2645 if (c->buffer_end > c->buffer_ptr) {
2646 len = recv(c->fd, c->buffer_ptr,
2647 FFMIN(c->chunk_size, c->buffer_end - c->buffer_ptr), 0);
2649 if (ff_neterrno() != AVERROR(EAGAIN) &&
2650 ff_neterrno() != AVERROR(EINTR))
2651 /* error : close connection */
2653 } else if (len == 0)
2654 /* end of connection : close it */
2657 c->chunk_size -= len;
2658 c->buffer_ptr += len;
2659 c->data_count += len;
2660 update_datarate(&c->datarate, c->data_count);
2664 if (c->buffer_ptr - c->buffer >= 2 && c->data_count > FFM_PACKET_SIZE) {
2665 if (c->buffer[0] != 'f' ||
2666 c->buffer[1] != 'm') {
2667 http_log("Feed stream has become desynchronized -- disconnecting\n");
2672 if (c->buffer_ptr >= c->buffer_end) {
2673 FFStream *feed = c->stream;
2674 /* a packet has been received : write it in the store, except
2676 if (c->data_count > FFM_PACKET_SIZE) {
2678 // printf("writing pos=0x%"PRIx64" size=0x%"PRIx64"\n", feed->feed_write_index, feed->feed_size);
2679 /* XXX: use llseek or url_seek */
2680 lseek(c->feed_fd, feed->feed_write_index, SEEK_SET);
2681 if (write(c->feed_fd, c->buffer, FFM_PACKET_SIZE) < 0) {
2682 http_log("Error writing to feed file: %s\n", strerror(errno));
2686 feed->feed_write_index += FFM_PACKET_SIZE;
2687 /* update file size */
2688 if (feed->feed_write_index > c->stream->feed_size)
2689 feed->feed_size = feed->feed_write_index;
2691 /* handle wrap around if max file size reached */
2692 if (c->stream->feed_max_size && feed->feed_write_index >= c->stream->feed_max_size)
2693 feed->feed_write_index = FFM_PACKET_SIZE;
2696 if (ffm_write_write_index(c->feed_fd, feed->feed_write_index) < 0) {
2697 http_log("Error writing index to feed file: %s\n", strerror(errno));
2701 /* wake up any waiting connections */
2702 for(c1 = first_http_ctx; c1 != NULL; c1 = c1->next) {
2703 if (c1->state == HTTPSTATE_WAIT_FEED &&
2704 c1->stream->feed == c->stream->feed)
2705 c1->state = HTTPSTATE_SEND_DATA;
2708 /* We have a header in our hands that contains useful data */
2709 AVFormatContext *s = avformat_alloc_context();
2711 AVInputFormat *fmt_in;
2717 /* use feed output format name to find corresponding input format */
2718 fmt_in = av_find_input_format(feed->fmt->name);
2722 pb = avio_alloc_context(c->buffer, c->buffer_end - c->buffer,
2723 0, NULL, NULL, NULL, NULL);
2727 if (avformat_open_input(&s, c->stream->feed_filename, fmt_in, NULL) < 0) {
2732 /* Now we have the actual streams */
2733 if (s->nb_streams != feed->nb_streams) {
2734 av_close_input_stream(s);
2736 http_log("Feed '%s' stream number does not match registered feed\n",
2737 c->stream->feed_filename);
2741 for (i = 0; i < s->nb_streams; i++) {
2742 AVStream *fst = feed->streams[i];
2743 AVStream *st = s->streams[i];
2744 avcodec_copy_context(fst->codec, st->codec);
2747 av_close_input_stream(s);
2750 c->buffer_ptr = c->buffer;
2755 c->stream->feed_opened = 0;
2757 /* wake up any waiting connections to stop waiting for feed */
2758 for(c1 = first_http_ctx; c1 != NULL; c1 = c1->next) {
2759 if (c1->state == HTTPSTATE_WAIT_FEED &&
2760 c1->stream->feed == c->stream->feed)
2761 c1->state = HTTPSTATE_SEND_DATA_TRAILER;
2766 /********************************************************************/
2769 static void rtsp_reply_header(HTTPContext *c, enum RTSPStatusCode error_number)
2776 switch(error_number) {
2777 case RTSP_STATUS_OK:
2780 case RTSP_STATUS_METHOD:
2781 str = "Method Not Allowed";
2783 case RTSP_STATUS_BANDWIDTH:
2784 str = "Not Enough Bandwidth";
2786 case RTSP_STATUS_SESSION:
2787 str = "Session Not Found";
2789 case RTSP_STATUS_STATE:
2790 str = "Method Not Valid in This State";
2792 case RTSP_STATUS_AGGREGATE:
2793 str = "Aggregate operation not allowed";
2795 case RTSP_STATUS_ONLY_AGGREGATE:
2796 str = "Only aggregate operation allowed";
2798 case RTSP_STATUS_TRANSPORT:
2799 str = "Unsupported transport";
2801 case RTSP_STATUS_INTERNAL:
2802 str = "Internal Server Error";
2804 case RTSP_STATUS_SERVICE:
2805 str = "Service Unavailable";
2807 case RTSP_STATUS_VERSION:
2808 str = "RTSP Version not supported";
2811 str = "Unknown Error";
2815 avio_printf(c->pb, "RTSP/1.0 %d %s\r\n", error_number, str);
2816 avio_printf(c->pb, "CSeq: %d\r\n", c->seq);
2818 /* output GMT time */
2821 strftime(buf2, sizeof(buf2), "%a, %d %b %Y %H:%M:%S", tm);
2822 avio_printf(c->pb, "Date: %s GMT\r\n", buf2);
2825 static void rtsp_reply_error(HTTPContext *c, enum RTSPStatusCode error_number)
2827 rtsp_reply_header(c, error_number);
2828 avio_printf(c->pb, "\r\n");
2831 static int rtsp_parse_request(HTTPContext *c)
2833 const char *p, *p1, *p2;
2839 RTSPMessageHeader header1, *header = &header1;
2841 c->buffer_ptr[0] = '\0';
2844 get_word(cmd, sizeof(cmd), &p);
2845 get_word(url, sizeof(url), &p);
2846 get_word(protocol, sizeof(protocol), &p);
2848 av_strlcpy(c->method, cmd, sizeof(c->method));
2849 av_strlcpy(c->url, url, sizeof(c->url));
2850 av_strlcpy(c->protocol, protocol, sizeof(c->protocol));
2852 if (avio_open_dyn_buf(&c->pb) < 0) {
2853 /* XXX: cannot do more */
2854 c->pb = NULL; /* safety */
2858 /* check version name */
2859 if (strcmp(protocol, "RTSP/1.0") != 0) {
2860 rtsp_reply_error(c, RTSP_STATUS_VERSION);
2864 /* parse each header line */
2865 memset(header, 0, sizeof(*header));
2866 /* skip to next line */
2867 while (*p != '\n' && *p != '\0')
2871 while (*p != '\0') {
2872 p1 = memchr(p, '\n', (char *)c->buffer_ptr - p);
2876 if (p2 > p && p2[-1] == '\r')
2878 /* skip empty line */
2882 if (len > sizeof(line) - 1)
2883 len = sizeof(line) - 1;
2884 memcpy(line, p, len);
2886 ff_rtsp_parse_line(header, line, NULL, NULL);
2890 /* handle sequence number */
2891 c->seq = header->seq;
2893 if (!strcmp(cmd, "DESCRIBE"))
2894 rtsp_cmd_describe(c, url);
2895 else if (!strcmp(cmd, "OPTIONS"))
2896 rtsp_cmd_options(c, url);
2897 else if (!strcmp(cmd, "SETUP"))
2898 rtsp_cmd_setup(c, url, header);
2899 else if (!strcmp(cmd, "PLAY"))
2900 rtsp_cmd_play(c, url, header);
2901 else if (!strcmp(cmd, "PAUSE"))
2902 rtsp_cmd_pause(c, url, header);
2903 else if (!strcmp(cmd, "TEARDOWN"))
2904 rtsp_cmd_teardown(c, url, header);
2906 rtsp_reply_error(c, RTSP_STATUS_METHOD);
2909 len = avio_close_dyn_buf(c->pb, &c->pb_buffer);
2910 c->pb = NULL; /* safety */
2912 /* XXX: cannot do more */
2915 c->buffer_ptr = c->pb_buffer;
2916 c->buffer_end = c->pb_buffer + len;
2917 c->state = RTSPSTATE_SEND_REPLY;
2921 static int prepare_sdp_description(FFStream *stream, uint8_t **pbuffer,
2922 struct in_addr my_ip)
2924 AVFormatContext *avc;
2925 AVStream *avs = NULL;
2928 avc = avformat_alloc_context();
2932 av_dict_set(&avc->metadata, "title",
2933 stream->title[0] ? stream->title : "No Title", 0);
2934 avc->nb_streams = stream->nb_streams;
2935 if (stream->is_multicast) {
2936 snprintf(avc->filename, 1024, "rtp://%s:%d?multicast=1?ttl=%d",
2937 inet_ntoa(stream->multicast_ip),
2938 stream->multicast_port, stream->multicast_ttl);
2940 snprintf(avc->filename, 1024, "rtp://0.0.0.0");
2943 if (avc->nb_streams >= INT_MAX/sizeof(*avc->streams) ||
2944 !(avc->streams = av_malloc(avc->nb_streams * sizeof(*avc->streams))))
2946 if (avc->nb_streams >= INT_MAX/sizeof(*avs) ||
2947 !(avs = av_malloc(avc->nb_streams * sizeof(*avs))))
2950 for(i = 0; i < stream->nb_streams; i++) {
2951 avc->streams[i] = &avs[i];
2952 avc->streams[i]->codec = stream->streams[i]->codec;
2954 *pbuffer = av_mallocz(2048);
2955 av_sdp_create(&avc, 1, *pbuffer, 2048);
2958 av_free(avc->streams);
2959 av_dict_free(&avc->metadata);
2963 return strlen(*pbuffer);
2966 static void rtsp_cmd_options(HTTPContext *c, const char *url)
2968 // rtsp_reply_header(c, RTSP_STATUS_OK);
2969 avio_printf(c->pb, "RTSP/1.0 %d %s\r\n", RTSP_STATUS_OK, "OK");
2970 avio_printf(c->pb, "CSeq: %d\r\n", c->seq);
2971 avio_printf(c->pb, "Public: %s\r\n", "OPTIONS, DESCRIBE, SETUP, TEARDOWN, PLAY, PAUSE");
2972 avio_printf(c->pb, "\r\n");
2975 static void rtsp_cmd_describe(HTTPContext *c, const char *url)
2981 int content_length, len;
2982 struct sockaddr_in my_addr;
2984 /* find which url is asked */
2985 av_url_split(NULL, 0, NULL, 0, NULL, 0, NULL, path1, sizeof(path1), url);
2990 for(stream = first_stream; stream != NULL; stream = stream->next) {
2991 if (!stream->is_feed &&
2992 stream->fmt && !strcmp(stream->fmt->name, "rtp") &&
2993 !strcmp(path, stream->filename)) {
2997 /* no stream found */
2998 rtsp_reply_error(c, RTSP_STATUS_SERVICE); /* XXX: right error ? */
3002 /* prepare the media description in sdp format */
3004 /* get the host IP */
3005 len = sizeof(my_addr);
3006 getsockname(c->fd, (struct sockaddr *)&my_addr, &len);
3007 content_length = prepare_sdp_description(stream, &content, my_addr.sin_addr);
3008 if (content_length < 0) {
3009 rtsp_reply_error(c, RTSP_STATUS_INTERNAL);
3012 rtsp_reply_header(c, RTSP_STATUS_OK);
3013 avio_printf(c->pb, "Content-Base: %s/\r\n", url);
3014 avio_printf(c->pb, "Content-Type: application/sdp\r\n");
3015 avio_printf(c->pb, "Content-Length: %d\r\n", content_length);
3016 avio_printf(c->pb, "\r\n");
3017 avio_write(c->pb, content, content_length);
3021 static HTTPContext *find_rtp_session(const char *session_id)
3025 if (session_id[0] == '\0')
3028 for(c = first_http_ctx; c != NULL; c = c->next) {
3029 if (!strcmp(c->session_id, session_id))
3035 static RTSPTransportField *find_transport(RTSPMessageHeader *h, enum RTSPLowerTransport lower_transport)
3037 RTSPTransportField *th;
3040 for(i=0;i<h->nb_transports;i++) {
3041 th = &h->transports[i];
3042 if (th->lower_transport == lower_transport)
3048 static void rtsp_cmd_setup(HTTPContext *c, const char *url,
3049 RTSPMessageHeader *h)
3052 int stream_index, rtp_port, rtcp_port;
3057 RTSPTransportField *th;
3058 struct sockaddr_in dest_addr;
3059 RTSPActionServerSetup setup;
3061 /* find which url is asked */
3062 av_url_split(NULL, 0, NULL, 0, NULL, 0, NULL, path1, sizeof(path1), url);
3067 /* now check each stream */
3068 for(stream = first_stream; stream != NULL; stream = stream->next) {
3069 if (!stream->is_feed &&
3070 stream->fmt && !strcmp(stream->fmt->name, "rtp")) {
3071 /* accept aggregate filenames only if single stream */
3072 if (!strcmp(path, stream->filename)) {
3073 if (stream->nb_streams != 1) {
3074 rtsp_reply_error(c, RTSP_STATUS_AGGREGATE);
3081 for(stream_index = 0; stream_index < stream->nb_streams;
3083 snprintf(buf, sizeof(buf), "%s/streamid=%d",
3084 stream->filename, stream_index);
3085 if (!strcmp(path, buf))
3090 /* no stream found */
3091 rtsp_reply_error(c, RTSP_STATUS_SERVICE); /* XXX: right error ? */
3095 /* generate session id if needed */
3096 if (h->session_id[0] == '\0')
3097 snprintf(h->session_id, sizeof(h->session_id), "%08x%08x",
3098 av_lfg_get(&random_state), av_lfg_get(&random_state));
3100 /* find rtp session, and create it if none found */
3101 rtp_c = find_rtp_session(h->session_id);
3103 /* always prefer UDP */
3104 th = find_transport(h, RTSP_LOWER_TRANSPORT_UDP);
3106 th = find_transport(h, RTSP_LOWER_TRANSPORT_TCP);
3108 rtsp_reply_error(c, RTSP_STATUS_TRANSPORT);
3113 rtp_c = rtp_new_connection(&c->from_addr, stream, h->session_id,
3114 th->lower_transport);
3116 rtsp_reply_error(c, RTSP_STATUS_BANDWIDTH);
3120 /* open input stream */
3121 if (open_input_stream(rtp_c, "") < 0) {
3122 rtsp_reply_error(c, RTSP_STATUS_INTERNAL);
3127 /* test if stream is OK (test needed because several SETUP needs
3128 to be done for a given file) */
3129 if (rtp_c->stream != stream) {
3130 rtsp_reply_error(c, RTSP_STATUS_SERVICE);
3134 /* test if stream is already set up */
3135 if (rtp_c->rtp_ctx[stream_index]) {
3136 rtsp_reply_error(c, RTSP_STATUS_STATE);
3140 /* check transport */
3141 th = find_transport(h, rtp_c->rtp_protocol);
3142 if (!th || (th->lower_transport == RTSP_LOWER_TRANSPORT_UDP &&
3143 th->client_port_min <= 0)) {
3144 rtsp_reply_error(c, RTSP_STATUS_TRANSPORT);
3148 /* setup default options */
3149 setup.transport_option[0] = '\0';
3150 dest_addr = rtp_c->from_addr;
3151 dest_addr.sin_port = htons(th->client_port_min);
3154 if (rtp_new_av_stream(rtp_c, stream_index, &dest_addr, c) < 0) {
3155 rtsp_reply_error(c, RTSP_STATUS_TRANSPORT);
3159 /* now everything is OK, so we can send the connection parameters */
3160 rtsp_reply_header(c, RTSP_STATUS_OK);
3162 avio_printf(c->pb, "Session: %s\r\n", rtp_c->session_id);
3164 switch(rtp_c->rtp_protocol) {
3165 case RTSP_LOWER_TRANSPORT_UDP:
3166 rtp_port = rtp_get_local_rtp_port(rtp_c->rtp_handles[stream_index]);
3167 rtcp_port = rtp_get_local_rtcp_port(rtp_c->rtp_handles[stream_index]);
3168 avio_printf(c->pb, "Transport: RTP/AVP/UDP;unicast;"
3169 "client_port=%d-%d;server_port=%d-%d",
3170 th->client_port_min, th->client_port_max,
3171 rtp_port, rtcp_port);
3173 case RTSP_LOWER_TRANSPORT_TCP:
3174 avio_printf(c->pb, "Transport: RTP/AVP/TCP;interleaved=%d-%d",
3175 stream_index * 2, stream_index * 2 + 1);
3180 if (setup.transport_option[0] != '\0')
3181 avio_printf(c->pb, ";%s", setup.transport_option);
3182 avio_printf(c->pb, "\r\n");
3185 avio_printf(c->pb, "\r\n");
3189 /* find an rtp connection by using the session ID. Check consistency
3191 static HTTPContext *find_rtp_session_with_url(const char *url,
3192 const char *session_id)
3200 rtp_c = find_rtp_session(session_id);
3204 /* find which url is asked */
3205 av_url_split(NULL, 0, NULL, 0, NULL, 0, NULL, path1, sizeof(path1), url);
3209 if(!strcmp(path, rtp_c->stream->filename)) return rtp_c;
3210 for(s=0; s<rtp_c->stream->nb_streams; ++s) {
3211 snprintf(buf, sizeof(buf), "%s/streamid=%d",
3212 rtp_c->stream->filename, s);
3213 if(!strncmp(path, buf, sizeof(buf))) {
3214 // XXX: Should we reply with RTSP_STATUS_ONLY_AGGREGATE if nb_streams>1?
3219 if (len > 0 && path[len - 1] == '/' &&
3220 !strncmp(path, rtp_c->stream->filename, len - 1))
3225 static void rtsp_cmd_play(HTTPContext *c, const char *url, RTSPMessageHeader *h)
3229 rtp_c = find_rtp_session_with_url(url, h->session_id);
3231 rtsp_reply_error(c, RTSP_STATUS_SESSION);
3235 if (rtp_c->state != HTTPSTATE_SEND_DATA &&
3236 rtp_c->state != HTTPSTATE_WAIT_FEED &&
3237 rtp_c->state != HTTPSTATE_READY) {
3238 rtsp_reply_error(c, RTSP_STATUS_STATE);
3242 rtp_c->state = HTTPSTATE_SEND_DATA;
3244 /* now everything is OK, so we can send the connection parameters */
3245 rtsp_reply_header(c, RTSP_STATUS_OK);
3247 avio_printf(c->pb, "Session: %s\r\n", rtp_c->session_id);
3248 avio_printf(c->pb, "\r\n");
3251 static void rtsp_cmd_pause(HTTPContext *c, const char *url, RTSPMessageHeader *h)
3255 rtp_c = find_rtp_session_with_url(url, h->session_id);
3257 rtsp_reply_error(c, RTSP_STATUS_SESSION);
3261 if (rtp_c->state != HTTPSTATE_SEND_DATA &&
3262 rtp_c->state != HTTPSTATE_WAIT_FEED) {
3263 rtsp_reply_error(c, RTSP_STATUS_STATE);
3267 rtp_c->state = HTTPSTATE_READY;
3268 rtp_c->first_pts = AV_NOPTS_VALUE;
3269 /* now everything is OK, so we can send the connection parameters */
3270 rtsp_reply_header(c, RTSP_STATUS_OK);
3272 avio_printf(c->pb, "Session: %s\r\n", rtp_c->session_id);
3273 avio_printf(c->pb, "\r\n");
3276 static void rtsp_cmd_teardown(HTTPContext *c, const char *url, RTSPMessageHeader *h)
3280 rtp_c = find_rtp_session_with_url(url, h->session_id);
3282 rtsp_reply_error(c, RTSP_STATUS_SESSION);
3286 /* now everything is OK, so we can send the connection parameters */
3287 rtsp_reply_header(c, RTSP_STATUS_OK);
3289 avio_printf(c->pb, "Session: %s\r\n", rtp_c->session_id);
3290 avio_printf(c->pb, "\r\n");
3292 /* abort the session */
3293 close_connection(rtp_c);
3297 /********************************************************************/
3300 static HTTPContext *rtp_new_connection(struct sockaddr_in *from_addr,
3301 FFStream *stream, const char *session_id,
3302 enum RTSPLowerTransport rtp_protocol)
3304 HTTPContext *c = NULL;
3305 const char *proto_str;
3307 /* XXX: should output a warning page when coming
3308 close to the connection limit */
3309 if (nb_connections >= nb_max_connections)
3312 /* add a new connection */
3313 c = av_mallocz(sizeof(HTTPContext));
3318 c->poll_entry = NULL;
3319 c->from_addr = *from_addr;
3320 c->buffer_size = IOBUFFER_INIT_SIZE;
3321 c->buffer = av_malloc(c->buffer_size);
3326 av_strlcpy(c->session_id, session_id, sizeof(c->session_id));
3327 c->state = HTTPSTATE_READY;
3328 c->is_packetized = 1;
3329 c->rtp_protocol = rtp_protocol;
3331 /* protocol is shown in statistics */
3332 switch(c->rtp_protocol) {
3333 case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
3334 proto_str = "MCAST";
3336 case RTSP_LOWER_TRANSPORT_UDP:
3339 case RTSP_LOWER_TRANSPORT_TCP:
3346 av_strlcpy(c->protocol, "RTP/", sizeof(c->protocol));
3347 av_strlcat(c->protocol, proto_str, sizeof(c->protocol));
3349 current_bandwidth += stream->bandwidth;
3351 c->next = first_http_ctx;
3363 /* add a new RTP stream in an RTP connection (used in RTSP SETUP
3364 command). If RTP/TCP protocol is used, TCP connection 'rtsp_c' is
3366 static int rtp_new_av_stream(HTTPContext *c,
3367 int stream_index, struct sockaddr_in *dest_addr,
3368 HTTPContext *rtsp_c)
3370 AVFormatContext *ctx;
3373 URLContext *h = NULL;
3375 int max_packet_size;
3377 /* now we can open the relevant output stream */
3378 ctx = avformat_alloc_context();
3381 ctx->oformat = av_guess_format("rtp", NULL, NULL);
3383 st = av_mallocz(sizeof(AVStream));
3386 ctx->nb_streams = 1;
3387 ctx->streams = av_mallocz(sizeof(AVStream *) * ctx->nb_streams);
3390 ctx->streams[0] = st;
3392 if (!c->stream->feed ||
3393 c->stream->feed == c->stream)
3394 memcpy(st, c->stream->streams[stream_index], sizeof(AVStream));
3397 c->stream->feed->streams[c->stream->feed_streams[stream_index]],
3399 st->priv_data = NULL;
3401 /* build destination RTP address */
3402 ipaddr = inet_ntoa(dest_addr->sin_addr);
3404 switch(c->rtp_protocol) {
3405 case RTSP_LOWER_TRANSPORT_UDP:
3406 case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
3409 /* XXX: also pass as parameter to function ? */
3410 if (c->stream->is_multicast) {
3412 ttl = c->stream->multicast_ttl;
3415 snprintf(ctx->filename, sizeof(ctx->filename),
3416 "rtp://%s:%d?multicast=1&ttl=%d",
3417 ipaddr, ntohs(dest_addr->sin_port), ttl);
3419 snprintf(ctx->filename, sizeof(ctx->filename),
3420 "rtp://%s:%d", ipaddr, ntohs(dest_addr->sin_port));
3423 if (url_open(&h, ctx->filename, AVIO_FLAG_WRITE) < 0)
3425 c->rtp_handles[stream_index] = h;
3426 max_packet_size = url_get_max_packet_size(h);
3428 case RTSP_LOWER_TRANSPORT_TCP:
3431 max_packet_size = RTSP_TCP_MAX_PACKET_SIZE;
3437 http_log("%s:%d - - \"PLAY %s/streamid=%d %s\"\n",
3438 ipaddr, ntohs(dest_addr->sin_port),
3439 c->stream->filename, stream_index, c->protocol);
3441 /* normally, no packets should be output here, but the packet size may be checked */
3442 if (ffio_open_dyn_packet_buf(&ctx->pb, max_packet_size) < 0) {
3443 /* XXX: close stream */
3446 if (avformat_write_header(ctx, NULL) < 0) {
3453 avio_close_dyn_buf(ctx->pb, &dummy_buf);
3456 c->rtp_ctx[stream_index] = ctx;
3460 /********************************************************************/
3461 /* ffserver initialization */
3463 static AVStream *add_av_stream1(FFStream *stream, AVCodecContext *codec, int copy)
3467 fst = av_mallocz(sizeof(AVStream));
3471 fst->codec = avcodec_alloc_context3(NULL);
3472 memcpy(fst->codec, codec, sizeof(AVCodecContext));
3473 if (codec->extradata_size) {
3474 fst->codec->extradata = av_malloc(codec->extradata_size);
3475 memcpy(fst->codec->extradata, codec->extradata,
3476 codec->extradata_size);
3479 /* live streams must use the actual feed's codec since it may be
3480 * updated later to carry extradata needed by the streams.
3484 fst->priv_data = av_mallocz(sizeof(FeedData));
3485 fst->index = stream->nb_streams;
3486 av_set_pts_info(fst, 33, 1, 90000);
3487 fst->sample_aspect_ratio = codec->sample_aspect_ratio;
3488 stream->streams[stream->nb_streams++] = fst;
3492 /* return the stream number in the feed */
3493 static int add_av_stream(FFStream *feed, AVStream *st)
3496 AVCodecContext *av, *av1;
3500 for(i=0;i<feed->nb_streams;i++) {
3501 st = feed->streams[i];
3503 if (av1->codec_id == av->codec_id &&
3504 av1->codec_type == av->codec_type &&
3505 av1->bit_rate == av->bit_rate) {
3507 switch(av->codec_type) {
3508 case AVMEDIA_TYPE_AUDIO:
3509 if (av1->channels == av->channels &&
3510 av1->sample_rate == av->sample_rate)
3513 case AVMEDIA_TYPE_VIDEO:
3514 if (av1->width == av->width &&
3515 av1->height == av->height &&
3516 av1->time_base.den == av->time_base.den &&
3517 av1->time_base.num == av->time_base.num &&
3518 av1->gop_size == av->gop_size)
3527 fst = add_av_stream1(feed, av, 0);
3530 return feed->nb_streams - 1;
3533 static void remove_stream(FFStream *stream)
3537 while (*ps != NULL) {
3545 /* specific mpeg4 handling : we extract the raw parameters */
3546 static void extract_mpeg4_header(AVFormatContext *infile)
3548 int mpeg4_count, i, size;
3554 for(i=0;i<infile->nb_streams;i++) {
3555 st = infile->streams[i];
3556 if (st->codec->codec_id == CODEC_ID_MPEG4 &&
3557 st->codec->extradata_size == 0) {
3564 printf("MPEG4 without extra data: trying to find header in %s\n", infile->filename);
3565 while (mpeg4_count > 0) {
3566 if (av_read_packet(infile, &pkt) < 0)
3568 st = infile->streams[pkt.stream_index];
3569 if (st->codec->codec_id == CODEC_ID_MPEG4 &&
3570 st->codec->extradata_size == 0) {
3571 av_freep(&st->codec->extradata);
3572 /* fill extradata with the header */
3573 /* XXX: we make hard suppositions here ! */
3575 while (p < pkt.data + pkt.size - 4) {
3576 /* stop when vop header is found */
3577 if (p[0] == 0x00 && p[1] == 0x00 &&
3578 p[2] == 0x01 && p[3] == 0xb6) {
3579 size = p - pkt.data;
3580 // av_hex_dump_log(infile, AV_LOG_DEBUG, pkt.data, size);
3581 st->codec->extradata = av_malloc(size);
3582 st->codec->extradata_size = size;
3583 memcpy(st->codec->extradata, pkt.data, size);
3590 av_free_packet(&pkt);
3594 /* compute the needed AVStream for each file */
3595 static void build_file_streams(void)
3597 FFStream *stream, *stream_next;
3600 /* gather all streams */
3601 for(stream = first_stream; stream != NULL; stream = stream_next) {
3602 AVFormatContext *infile = NULL;
3603 stream_next = stream->next;
3604 if (stream->stream_type == STREAM_TYPE_LIVE &&
3606 /* the stream comes from a file */
3607 /* try to open the file */
3609 if (stream->fmt && !strcmp(stream->fmt->name, "rtp")) {
3610 /* specific case : if transport stream output to RTP,
3611 we use a raw transport stream reader */
3612 av_dict_set(&stream->in_opts, "mpeg2ts_compute_pcr", "1", 0);
3615 http_log("Opening file '%s'\n", stream->feed_filename);
3616 if ((ret = avformat_open_input(&infile, stream->feed_filename, stream->ifmt, &stream->in_opts)) < 0) {
3617 http_log("Could not open '%s': %d\n", stream->feed_filename, ret);
3618 /* remove stream (no need to spend more time on it) */
3620 remove_stream(stream);
3622 /* find all the AVStreams inside and reference them in
3624 if (avformat_find_stream_info(infile, NULL) < 0) {
3625 http_log("Could not find codec parameters from '%s'\n",
3626 stream->feed_filename);
3627 av_close_input_file(infile);
3630 extract_mpeg4_header(infile);
3632 for(i=0;i<infile->nb_streams;i++)
3633 add_av_stream1(stream, infile->streams[i]->codec, 1);
3635 av_close_input_file(infile);
3641 /* compute the needed AVStream for each feed */
3642 static void build_feed_streams(void)
3644 FFStream *stream, *feed;
3647 /* gather all streams */
3648 for(stream = first_stream; stream != NULL; stream = stream->next) {
3649 feed = stream->feed;
3651 if (stream->is_feed) {
3652 for(i=0;i<stream->nb_streams;i++)
3653 stream->feed_streams[i] = i;
3655 /* we handle a stream coming from a feed */
3656 for(i=0;i<stream->nb_streams;i++)
3657 stream->feed_streams[i] = add_av_stream(feed, stream->streams[i]);
3662 /* create feed files if needed */
3663 for(feed = first_feed; feed != NULL; feed = feed->next_feed) {
3666 if (avio_check(feed->feed_filename, AVIO_FLAG_READ) > 0) {
3667 /* See if it matches */
3668 AVFormatContext *s = NULL;
3671 if (avformat_open_input(&s, feed->feed_filename, NULL, NULL) >= 0) {
3672 /* Now see if it matches */
3673 if (s->nb_streams == feed->nb_streams) {
3675 for(i=0;i<s->nb_streams;i++) {
3677 sf = feed->streams[i];
3680 if (sf->index != ss->index ||
3682 http_log("Index & Id do not match for stream %d (%s)\n",
3683 i, feed->feed_filename);
3686 AVCodecContext *ccf, *ccs;
3690 #define CHECK_CODEC(x) (ccf->x != ccs->x)
3692 if (CHECK_CODEC(codec_id) || CHECK_CODEC(codec_type)) {
3693 http_log("Codecs do not match for stream %d\n", i);
3695 } else if (CHECK_CODEC(bit_rate) || CHECK_CODEC(flags)) {
3696 http_log("Codec bitrates do not match for stream %d\n", i);
3698 } else if (ccf->codec_type == AVMEDIA_TYPE_VIDEO) {
3699 if (CHECK_CODEC(time_base.den) ||
3700 CHECK_CODEC(time_base.num) ||
3701 CHECK_CODEC(width) ||
3702 CHECK_CODEC(height)) {
3703 http_log("Codec width, height and framerate do not match for stream %d\n", i);
3706 } else if (ccf->codec_type == AVMEDIA_TYPE_AUDIO) {
3707 if (CHECK_CODEC(sample_rate) ||
3708 CHECK_CODEC(channels) ||
3709 CHECK_CODEC(frame_size)) {
3710 http_log("Codec sample_rate, channels, frame_size do not match for stream %d\n", i);
3714 http_log("Unknown codec type\n");
3722 http_log("Deleting feed file '%s' as stream counts differ (%d != %d)\n",
3723 feed->feed_filename, s->nb_streams, feed->nb_streams);
3725 av_close_input_file(s);
3727 http_log("Deleting feed file '%s' as it appears to be corrupt\n",
3728 feed->feed_filename);
3731 if (feed->readonly) {
3732 http_log("Unable to delete feed file '%s' as it is marked readonly\n",
3733 feed->feed_filename);
3736 unlink(feed->feed_filename);
3739 if (avio_check(feed->feed_filename, AVIO_FLAG_WRITE) <= 0) {
3740 AVFormatContext s1 = {0}, *s = &s1;
3742 if (feed->readonly) {
3743 http_log("Unable to create feed file '%s' as it is marked readonly\n",
3744 feed->feed_filename);
3748 /* only write the header of the ffm file */
3749 if (avio_open(&s->pb, feed->feed_filename, AVIO_FLAG_WRITE) < 0) {
3750 http_log("Could not open output feed file '%s'\n",
3751 feed->feed_filename);
3754 s->oformat = feed->fmt;
3755 s->nb_streams = feed->nb_streams;
3756 s->streams = feed->streams;
3757 if (avformat_write_header(s, NULL) < 0) {
3758 http_log("Container doesn't supports the required parameters\n");
3761 /* XXX: need better api */
3762 av_freep(&s->priv_data);
3765 /* get feed size and write index */
3766 fd = open(feed->feed_filename, O_RDONLY);
3768 http_log("Could not open output feed file '%s'\n",
3769 feed->feed_filename);
3773 feed->feed_write_index = FFMAX(ffm_read_write_index(fd), FFM_PACKET_SIZE);
3774 feed->feed_size = lseek(fd, 0, SEEK_END);
3775 /* ensure that we do not wrap before the end of file */
3776 if (feed->feed_max_size && feed->feed_max_size < feed->feed_size)
3777 feed->feed_max_size = feed->feed_size;
3783 /* compute the bandwidth used by each stream */
3784 static void compute_bandwidth(void)
3790 for(stream = first_stream; stream != NULL; stream = stream->next) {
3792 for(i=0;i<stream->nb_streams;i++) {
3793 AVStream *st = stream->streams[i];
3794 switch(st->codec->codec_type) {
3795 case AVMEDIA_TYPE_AUDIO:
3796 case AVMEDIA_TYPE_VIDEO:
3797 bandwidth += st->codec->bit_rate;
3803 stream->bandwidth = (bandwidth + 999) / 1000;
3807 /* add a codec and set the default parameters */
3808 static void add_codec(FFStream *stream, AVCodecContext *av)
3812 /* compute default parameters */
3813 switch(av->codec_type) {
3814 case AVMEDIA_TYPE_AUDIO:
3815 if (av->bit_rate == 0)
3816 av->bit_rate = 64000;
3817 if (av->sample_rate == 0)
3818 av->sample_rate = 22050;
3819 if (av->channels == 0)
3822 case AVMEDIA_TYPE_VIDEO:
3823 if (av->bit_rate == 0)
3824 av->bit_rate = 64000;
3825 if (av->time_base.num == 0){
3826 av->time_base.den = 5;
3827 av->time_base.num = 1;
3829 if (av->width == 0 || av->height == 0) {
3833 /* Bitrate tolerance is less for streaming */
3834 if (av->bit_rate_tolerance == 0)
3835 av->bit_rate_tolerance = FFMAX(av->bit_rate / 4,
3836 (int64_t)av->bit_rate*av->time_base.num/av->time_base.den);
3841 if (av->max_qdiff == 0)
3843 av->qcompress = 0.5;
3846 if (!av->nsse_weight)
3847 av->nsse_weight = 8;
3849 av->frame_skip_cmp = FF_CMP_DCTMAX;
3851 av->me_method = ME_EPZS;
3852 av->rc_buffer_aggressivity = 1.0;
3855 av->rc_eq = "tex^qComp";
3856 if (!av->i_quant_factor)
3857 av->i_quant_factor = -0.8;
3858 if (!av->b_quant_factor)
3859 av->b_quant_factor = 1.25;
3860 if (!av->b_quant_offset)
3861 av->b_quant_offset = 1.25;
3862 if (!av->rc_max_rate)
3863 av->rc_max_rate = av->bit_rate * 2;
3865 if (av->rc_max_rate && !av->rc_buffer_size) {
3866 av->rc_buffer_size = av->rc_max_rate;
3875 st = av_mallocz(sizeof(AVStream));
3878 st->codec = avcodec_alloc_context3(NULL);
3879 stream->streams[stream->nb_streams++] = st;
3880 memcpy(st->codec, av, sizeof(AVCodecContext));
3883 static enum CodecID opt_audio_codec(const char *arg)
3885 AVCodec *p= avcodec_find_encoder_by_name(arg);
3887 if (p == NULL || p->type != AVMEDIA_TYPE_AUDIO)
3888 return CODEC_ID_NONE;
3893 static enum CodecID opt_video_codec(const char *arg)
3895 AVCodec *p= avcodec_find_encoder_by_name(arg);
3897 if (p == NULL || p->type != AVMEDIA_TYPE_VIDEO)
3898 return CODEC_ID_NONE;
3903 /* simplistic plugin support */
3906 static void load_module(const char *filename)
3909 void (*init_func)(void);
3910 dll = dlopen(filename, RTLD_NOW);
3912 fprintf(stderr, "Could not load module '%s' - %s\n",
3913 filename, dlerror());
3917 init_func = dlsym(dll, "ffserver_module_init");
3920 "%s: init function 'ffserver_module_init()' not found\n",
3929 static int ffserver_opt_default(const char *opt, const char *arg,
3930 AVCodecContext *avctx, int type)
3933 const AVOption *o = av_opt_find(avctx, opt, NULL, type, 0);
3935 ret = av_set_string3(avctx, opt, arg, 1, NULL);
3939 static int ffserver_opt_preset(const char *arg,
3940 AVCodecContext *avctx, int type,
3941 enum CodecID *audio_id, enum CodecID *video_id)
3944 char filename[1000], tmp[1000], tmp2[1000], line[1000];
3946 AVCodec *codec = avcodec_find_encoder(avctx->codec_id);
3948 if (!(f = get_preset_file(filename, sizeof(filename), arg, 0,
3949 codec ? codec->name : NULL))) {
3950 fprintf(stderr, "File for preset '%s' not found\n", arg);
3955 int e= fscanf(f, "%999[^\n]\n", line) - 1;
3956 if(line[0] == '#' && !e)
3958 e|= sscanf(line, "%999[^=]=%999[^\n]\n", tmp, tmp2) - 2;
3960 fprintf(stderr, "%s: Invalid syntax: '%s'\n", filename, line);
3964 if(!strcmp(tmp, "acodec")){
3965 *audio_id = opt_audio_codec(tmp2);
3966 }else if(!strcmp(tmp, "vcodec")){
3967 *video_id = opt_video_codec(tmp2);
3968 }else if(!strcmp(tmp, "scodec")){
3969 /* opt_subtitle_codec(tmp2); */
3970 }else if(ffserver_opt_default(tmp, tmp2, avctx, type) < 0){
3971 fprintf(stderr, "%s: Invalid option or argument: '%s', parsed as '%s' = '%s'\n", filename, line, tmp, tmp2);
3982 static AVOutputFormat *ffserver_guess_format(const char *short_name, const char *filename,
3983 const char *mime_type)
3985 AVOutputFormat *fmt = av_guess_format(short_name, filename, mime_type);
3988 AVOutputFormat *stream_fmt;
3989 char stream_format_name[64];
3991 snprintf(stream_format_name, sizeof(stream_format_name), "%s_stream", fmt->name);
3992 stream_fmt = av_guess_format(stream_format_name, NULL, NULL);
4001 static void report_config_error(const char *filename, int line_num, int *errors, const char *fmt, ...)
4005 fprintf(stderr, "%s:%d: ", filename, line_num);
4006 vfprintf(stderr, fmt, vl);
4012 static int parse_ffconfig(const char *filename)
4019 int val, errors, line_num;
4020 FFStream **last_stream, *stream, *redirect;
4021 FFStream **last_feed, *feed, *s;
4022 AVCodecContext audio_enc, video_enc;
4023 enum CodecID audio_id, video_id;
4025 f = fopen(filename, "r");
4033 first_stream = NULL;
4034 last_stream = &first_stream;
4036 last_feed = &first_feed;
4040 audio_id = CODEC_ID_NONE;
4041 video_id = CODEC_ID_NONE;
4043 #define ERROR(...) report_config_error(filename, line_num, &errors, __VA_ARGS__)
4045 if (fgets(line, sizeof(line), f) == NULL)
4051 if (*p == '\0' || *p == '#')
4054 get_arg(cmd, sizeof(cmd), &p);
4056 if (!strcasecmp(cmd, "Port")) {
4057 get_arg(arg, sizeof(arg), &p);
4059 if (val < 1 || val > 65536) {
4060 ERROR("Invalid_port: %s\n", arg);
4062 my_http_addr.sin_port = htons(val);
4063 } else if (!strcasecmp(cmd, "BindAddress")) {
4064 get_arg(arg, sizeof(arg), &p);
4065 if (resolve_host(&my_http_addr.sin_addr, arg) != 0) {
4066 ERROR("%s:%d: Invalid host/IP address: %s\n", arg);
4068 } else if (!strcasecmp(cmd, "NoDaemon")) {
4069 ffserver_daemon = 0;
4070 } else if (!strcasecmp(cmd, "RTSPPort")) {
4071 get_arg(arg, sizeof(arg), &p);
4073 if (val < 1 || val > 65536) {
4074 ERROR("%s:%d: Invalid port: %s\n", arg);
4076 my_rtsp_addr.sin_port = htons(atoi(arg));
4077 } else if (!strcasecmp(cmd, "RTSPBindAddress")) {
4078 get_arg(arg, sizeof(arg), &p);
4079 if (resolve_host(&my_rtsp_addr.sin_addr, arg) != 0) {
4080 ERROR("Invalid host/IP address: %s\n", arg);
4082 } else if (!strcasecmp(cmd, "MaxHTTPConnections")) {
4083 get_arg(arg, sizeof(arg), &p);
4085 if (val < 1 || val > 65536) {
4086 ERROR("Invalid MaxHTTPConnections: %s\n", arg);
4088 nb_max_http_connections = val;
4089 } else if (!strcasecmp(cmd, "MaxClients")) {
4090 get_arg(arg, sizeof(arg), &p);
4092 if (val < 1 || val > nb_max_http_connections) {
4093 ERROR("Invalid MaxClients: %s\n", arg);
4095 nb_max_connections = val;
4097 } else if (!strcasecmp(cmd, "MaxBandwidth")) {
4099 get_arg(arg, sizeof(arg), &p);
4101 if (llval < 10 || llval > 10000000) {
4102 ERROR("Invalid MaxBandwidth: %s\n", arg);
4104 max_bandwidth = llval;
4105 } else if (!strcasecmp(cmd, "CustomLog")) {
4106 if (!ffserver_debug)
4107 get_arg(logfilename, sizeof(logfilename), &p);
4108 } else if (!strcasecmp(cmd, "<Feed")) {
4109 /*********************************************/
4110 /* Feed related options */
4112 if (stream || feed) {
4113 ERROR("Already in a tag\n");
4115 feed = av_mallocz(sizeof(FFStream));
4116 get_arg(feed->filename, sizeof(feed->filename), &p);
4117 q = strrchr(feed->filename, '>');
4121 for (s = first_feed; s; s = s->next) {
4122 if (!strcmp(feed->filename, s->filename)) {
4123 ERROR("Feed '%s' already registered\n", s->filename);
4127 feed->fmt = av_guess_format("ffm", NULL, NULL);
4128 /* defaut feed file */
4129 snprintf(feed->feed_filename, sizeof(feed->feed_filename),
4130 "/tmp/%s.ffm", feed->filename);
4131 feed->feed_max_size = 5 * 1024 * 1024;
4133 feed->feed = feed; /* self feeding :-) */
4135 /* add in stream list */
4136 *last_stream = feed;
4137 last_stream = &feed->next;
4138 /* add in feed list */
4140 last_feed = &feed->next_feed;
4142 } else if (!strcasecmp(cmd, "Launch")) {
4146 feed->child_argv = av_mallocz(64 * sizeof(char *));
4148 for (i = 0; i < 62; i++) {
4149 get_arg(arg, sizeof(arg), &p);
4153 feed->child_argv[i] = av_strdup(arg);
4156 feed->child_argv[i] = av_malloc(30 + strlen(feed->filename));
4158 snprintf(feed->child_argv[i], 30+strlen(feed->filename),
4160 (my_http_addr.sin_addr.s_addr == INADDR_ANY) ? "127.0.0.1" :
4161 inet_ntoa(my_http_addr.sin_addr),
4162 ntohs(my_http_addr.sin_port), feed->filename);
4164 } else if (!strcasecmp(cmd, "ReadOnlyFile")) {
4166 get_arg(feed->feed_filename, sizeof(feed->feed_filename), &p);
4168 } else if (stream) {
4169 get_arg(stream->feed_filename, sizeof(stream->feed_filename), &p);
4171 } else if (!strcasecmp(cmd, "File")) {
4173 get_arg(feed->feed_filename, sizeof(feed->feed_filename), &p);
4175 get_arg(stream->feed_filename, sizeof(stream->feed_filename), &p);
4176 } else if (!strcasecmp(cmd, "Truncate")) {
4178 get_arg(arg, sizeof(arg), &p);
4179 feed->truncate = strtod(arg, NULL);
4181 } else if (!strcasecmp(cmd, "FileMaxSize")) {
4186 get_arg(arg, sizeof(arg), &p);
4188 fsize = strtod(p1, &p1);
4189 switch(toupper(*p1)) {
4194 fsize *= 1024 * 1024;
4197 fsize *= 1024 * 1024 * 1024;
4200 feed->feed_max_size = (int64_t)fsize;
4201 if (feed->feed_max_size < FFM_PACKET_SIZE*4) {
4202 ERROR("Feed max file size is too small, must be at least %d\n", FFM_PACKET_SIZE*4);
4205 } else if (!strcasecmp(cmd, "</Feed>")) {
4207 ERROR("No corresponding <Feed> for </Feed>\n");
4210 } else if (!strcasecmp(cmd, "<Stream")) {
4211 /*********************************************/
4212 /* Stream related options */
4214 if (stream || feed) {
4215 ERROR("Already in a tag\n");
4218 stream = av_mallocz(sizeof(FFStream));
4219 get_arg(stream->filename, sizeof(stream->filename), &p);
4220 q = strrchr(stream->filename, '>');
4224 for (s = first_stream; s; s = s->next) {
4225 if (!strcmp(stream->filename, s->filename)) {
4226 ERROR("Stream '%s' already registered\n", s->filename);
4230 stream->fmt = ffserver_guess_format(NULL, stream->filename, NULL);
4231 avcodec_get_context_defaults2(&video_enc, AVMEDIA_TYPE_VIDEO);
4232 avcodec_get_context_defaults2(&audio_enc, AVMEDIA_TYPE_AUDIO);
4233 audio_id = CODEC_ID_NONE;
4234 video_id = CODEC_ID_NONE;
4236 audio_id = stream->fmt->audio_codec;
4237 video_id = stream->fmt->video_codec;
4240 *last_stream = stream;
4241 last_stream = &stream->next;
4243 } else if (!strcasecmp(cmd, "Feed")) {
4244 get_arg(arg, sizeof(arg), &p);
4249 while (sfeed != NULL) {
4250 if (!strcmp(sfeed->filename, arg))
4252 sfeed = sfeed->next_feed;
4255 ERROR("feed '%s' not defined\n", arg);
4257 stream->feed = sfeed;
4259 } else if (!strcasecmp(cmd, "Format")) {
4260 get_arg(arg, sizeof(arg), &p);
4262 if (!strcmp(arg, "status")) {
4263 stream->stream_type = STREAM_TYPE_STATUS;
4266 stream->stream_type = STREAM_TYPE_LIVE;
4267 /* jpeg cannot be used here, so use single frame jpeg */
4268 if (!strcmp(arg, "jpeg"))
4269 strcpy(arg, "mjpeg");
4270 stream->fmt = ffserver_guess_format(arg, NULL, NULL);
4272 ERROR("Unknown Format: %s\n", arg);
4276 audio_id = stream->fmt->audio_codec;
4277 video_id = stream->fmt->video_codec;
4280 } else if (!strcasecmp(cmd, "InputFormat")) {
4281 get_arg(arg, sizeof(arg), &p);
4283 stream->ifmt = av_find_input_format(arg);
4284 if (!stream->ifmt) {
4285 ERROR("Unknown input format: %s\n", arg);
4288 } else if (!strcasecmp(cmd, "FaviconURL")) {
4289 if (stream && stream->stream_type == STREAM_TYPE_STATUS) {
4290 get_arg(stream->feed_filename, sizeof(stream->feed_filename), &p);
4292 ERROR("FaviconURL only permitted for status streams\n");
4294 } else if (!strcasecmp(cmd, "Author")) {
4296 get_arg(stream->author, sizeof(stream->author), &p);
4297 } else if (!strcasecmp(cmd, "Comment")) {
4299 get_arg(stream->comment, sizeof(stream->comment), &p);
4300 } else if (!strcasecmp(cmd, "Copyright")) {
4302 get_arg(stream->copyright, sizeof(stream->copyright), &p);
4303 } else if (!strcasecmp(cmd, "Title")) {
4305 get_arg(stream->title, sizeof(stream->title), &p);
4306 } else if (!strcasecmp(cmd, "Preroll")) {
4307 get_arg(arg, sizeof(arg), &p);
4309 stream->prebuffer = atof(arg) * 1000;
4310 } else if (!strcasecmp(cmd, "StartSendOnKey")) {
4312 stream->send_on_key = 1;
4313 } else if (!strcasecmp(cmd, "AudioCodec")) {
4314 get_arg(arg, sizeof(arg), &p);
4315 audio_id = opt_audio_codec(arg);
4316 if (audio_id == CODEC_ID_NONE) {
4317 ERROR("Unknown AudioCodec: %s\n", arg);
4319 } else if (!strcasecmp(cmd, "VideoCodec")) {
4320 get_arg(arg, sizeof(arg), &p);
4321 video_id = opt_video_codec(arg);
4322 if (video_id == CODEC_ID_NONE) {
4323 ERROR("Unknown VideoCodec: %s\n", arg);
4325 } else if (!strcasecmp(cmd, "MaxTime")) {
4326 get_arg(arg, sizeof(arg), &p);
4328 stream->max_time = atof(arg) * 1000;
4329 } else if (!strcasecmp(cmd, "AudioBitRate")) {
4330 get_arg(arg, sizeof(arg), &p);
4332 audio_enc.bit_rate = lrintf(atof(arg) * 1000);
4333 } else if (!strcasecmp(cmd, "AudioChannels")) {
4334 get_arg(arg, sizeof(arg), &p);
4336 audio_enc.channels = atoi(arg);
4337 } else if (!strcasecmp(cmd, "AudioSampleRate")) {
4338 get_arg(arg, sizeof(arg), &p);
4340 audio_enc.sample_rate = atoi(arg);
4341 } else if (!strcasecmp(cmd, "AudioQuality")) {
4342 get_arg(arg, sizeof(arg), &p);
4344 // audio_enc.quality = atof(arg) * 1000;
4346 } else if (!strcasecmp(cmd, "VideoBitRateRange")) {
4348 int minrate, maxrate;
4350 get_arg(arg, sizeof(arg), &p);
4352 if (sscanf(arg, "%d-%d", &minrate, &maxrate) == 2) {
4353 video_enc.rc_min_rate = minrate * 1000;
4354 video_enc.rc_max_rate = maxrate * 1000;
4356 ERROR("Incorrect format for VideoBitRateRange -- should be <min>-<max>: %s\n", arg);
4359 } else if (!strcasecmp(cmd, "Debug")) {
4361 get_arg(arg, sizeof(arg), &p);
4362 video_enc.debug = strtol(arg,0,0);
4364 } else if (!strcasecmp(cmd, "Strict")) {
4366 get_arg(arg, sizeof(arg), &p);
4367 video_enc.strict_std_compliance = atoi(arg);
4369 } else if (!strcasecmp(cmd, "VideoBufferSize")) {
4371 get_arg(arg, sizeof(arg), &p);
4372 video_enc.rc_buffer_size = atoi(arg) * 8*1024;
4374 } else if (!strcasecmp(cmd, "VideoBitRateTolerance")) {
4376 get_arg(arg, sizeof(arg), &p);
4377 video_enc.bit_rate_tolerance = atoi(arg) * 1000;
4379 } else if (!strcasecmp(cmd, "VideoBitRate")) {
4380 get_arg(arg, sizeof(arg), &p);
4382 video_enc.bit_rate = atoi(arg) * 1000;
4384 } else if (!strcasecmp(cmd, "VideoSize")) {
4385 get_arg(arg, sizeof(arg), &p);
4387 av_parse_video_size(&video_enc.width, &video_enc.height, arg);
4388 if ((video_enc.width % 16) != 0 ||
4389 (video_enc.height % 16) != 0) {
4390 ERROR("Image size must be a multiple of 16\n");
4393 } else if (!strcasecmp(cmd, "VideoFrameRate")) {
4394 get_arg(arg, sizeof(arg), &p);
4396 AVRational frame_rate;
4397 if (av_parse_video_rate(&frame_rate, arg) < 0) {
4398 ERROR("Incorrect frame rate: %s\n", arg);
4400 video_enc.time_base.num = frame_rate.den;
4401 video_enc.time_base.den = frame_rate.num;
4404 } else if (!strcasecmp(cmd, "VideoGopSize")) {
4405 get_arg(arg, sizeof(arg), &p);
4407 video_enc.gop_size = atoi(arg);
4408 } else if (!strcasecmp(cmd, "VideoIntraOnly")) {
4410 video_enc.gop_size = 1;
4411 } else if (!strcasecmp(cmd, "VideoHighQuality")) {
4413 video_enc.mb_decision = FF_MB_DECISION_BITS;
4414 } else if (!strcasecmp(cmd, "Video4MotionVector")) {
4416 video_enc.mb_decision = FF_MB_DECISION_BITS; //FIXME remove
4417 video_enc.flags |= CODEC_FLAG_4MV;
4419 } else if (!strcasecmp(cmd, "AVOptionVideo") ||
4420 !strcasecmp(cmd, "AVOptionAudio")) {
4422 AVCodecContext *avctx;
4424 get_arg(arg, sizeof(arg), &p);
4425 get_arg(arg2, sizeof(arg2), &p);
4426 if (!strcasecmp(cmd, "AVOptionVideo")) {
4428 type = AV_OPT_FLAG_VIDEO_PARAM;
4431 type = AV_OPT_FLAG_AUDIO_PARAM;
4433 if (ffserver_opt_default(arg, arg2, avctx, type|AV_OPT_FLAG_ENCODING_PARAM)) {
4434 ERROR("AVOption error: %s %s\n", arg, arg2);
4436 } else if (!strcasecmp(cmd, "AVPresetVideo") ||
4437 !strcasecmp(cmd, "AVPresetAudio")) {
4438 AVCodecContext *avctx;
4440 get_arg(arg, sizeof(arg), &p);
4441 if (!strcasecmp(cmd, "AVPresetVideo")) {
4443 video_enc.codec_id = video_id;
4444 type = AV_OPT_FLAG_VIDEO_PARAM;
4447 audio_enc.codec_id = audio_id;
4448 type = AV_OPT_FLAG_AUDIO_PARAM;
4450 if (ffserver_opt_preset(arg, avctx, type|AV_OPT_FLAG_ENCODING_PARAM, &audio_id, &video_id)) {
4451 ERROR("AVPreset error: %s\n", arg);
4453 } else if (!strcasecmp(cmd, "VideoTag")) {
4454 get_arg(arg, sizeof(arg), &p);
4455 if ((strlen(arg) == 4) && stream)
4456 video_enc.codec_tag = MKTAG(arg[0], arg[1], arg[2], arg[3]);
4457 } else if (!strcasecmp(cmd, "BitExact")) {
4459 video_enc.flags |= CODEC_FLAG_BITEXACT;
4460 } else if (!strcasecmp(cmd, "DctFastint")) {
4462 video_enc.dct_algo = FF_DCT_FASTINT;
4463 } else if (!strcasecmp(cmd, "IdctSimple")) {
4465 video_enc.idct_algo = FF_IDCT_SIMPLE;
4466 } else if (!strcasecmp(cmd, "Qscale")) {
4467 get_arg(arg, sizeof(arg), &p);
4469 video_enc.flags |= CODEC_FLAG_QSCALE;
4470 video_enc.global_quality = FF_QP2LAMBDA * atoi(arg);
4472 } else if (!strcasecmp(cmd, "VideoQDiff")) {
4473 get_arg(arg, sizeof(arg), &p);
4475 video_enc.max_qdiff = atoi(arg);
4476 if (video_enc.max_qdiff < 1 || video_enc.max_qdiff > 31) {
4477 ERROR("VideoQDiff out of range\n");
4480 } else if (!strcasecmp(cmd, "VideoQMax")) {
4481 get_arg(arg, sizeof(arg), &p);
4483 video_enc.qmax = atoi(arg);
4484 if (video_enc.qmax < 1 || video_enc.qmax > 31) {
4485 ERROR("VideoQMax out of range\n");
4488 } else if (!strcasecmp(cmd, "VideoQMin")) {
4489 get_arg(arg, sizeof(arg), &p);
4491 video_enc.qmin = atoi(arg);
4492 if (video_enc.qmin < 1 || video_enc.qmin > 31) {
4493 ERROR("VideoQMin out of range\n");
4496 } else if (!strcasecmp(cmd, "LumaElim")) {
4497 get_arg(arg, sizeof(arg), &p);
4499 video_enc.luma_elim_threshold = atoi(arg);
4500 } else if (!strcasecmp(cmd, "ChromaElim")) {
4501 get_arg(arg, sizeof(arg), &p);
4503 video_enc.chroma_elim_threshold = atoi(arg);
4504 } else if (!strcasecmp(cmd, "LumiMask")) {
4505 get_arg(arg, sizeof(arg), &p);
4507 video_enc.lumi_masking = atof(arg);
4508 } else if (!strcasecmp(cmd, "DarkMask")) {
4509 get_arg(arg, sizeof(arg), &p);
4511 video_enc.dark_masking = atof(arg);
4512 } else if (!strcasecmp(cmd, "NoVideo")) {
4513 video_id = CODEC_ID_NONE;
4514 } else if (!strcasecmp(cmd, "NoAudio")) {
4515 audio_id = CODEC_ID_NONE;
4516 } else if (!strcasecmp(cmd, "ACL")) {
4517 parse_acl_row(stream, feed, NULL, p, filename, line_num);
4518 } else if (!strcasecmp(cmd, "DynamicACL")) {
4520 get_arg(stream->dynamic_acl, sizeof(stream->dynamic_acl), &p);
4522 } else if (!strcasecmp(cmd, "RTSPOption")) {
4523 get_arg(arg, sizeof(arg), &p);
4525 av_freep(&stream->rtsp_option);
4526 stream->rtsp_option = av_strdup(arg);
4528 } else if (!strcasecmp(cmd, "MulticastAddress")) {
4529 get_arg(arg, sizeof(arg), &p);
4531 if (resolve_host(&stream->multicast_ip, arg) != 0) {
4532 ERROR("Invalid host/IP address: %s\n", arg);
4534 stream->is_multicast = 1;
4535 stream->loop = 1; /* default is looping */
4537 } else if (!strcasecmp(cmd, "MulticastPort")) {
4538 get_arg(arg, sizeof(arg), &p);
4540 stream->multicast_port = atoi(arg);
4541 } else if (!strcasecmp(cmd, "MulticastTTL")) {
4542 get_arg(arg, sizeof(arg), &p);
4544 stream->multicast_ttl = atoi(arg);
4545 } else if (!strcasecmp(cmd, "NoLoop")) {
4548 } else if (!strcasecmp(cmd, "</Stream>")) {
4550 ERROR("No corresponding <Stream> for </Stream>\n");
4552 if (stream->feed && stream->fmt && strcmp(stream->fmt->name, "ffm") != 0) {
4553 if (audio_id != CODEC_ID_NONE) {
4554 audio_enc.codec_type = AVMEDIA_TYPE_AUDIO;
4555 audio_enc.codec_id = audio_id;
4556 add_codec(stream, &audio_enc);
4558 if (video_id != CODEC_ID_NONE) {
4559 video_enc.codec_type = AVMEDIA_TYPE_VIDEO;
4560 video_enc.codec_id = video_id;
4561 add_codec(stream, &video_enc);
4566 } else if (!strcasecmp(cmd, "<Redirect")) {
4567 /*********************************************/
4569 if (stream || feed || redirect) {
4570 ERROR("Already in a tag\n");
4572 redirect = av_mallocz(sizeof(FFStream));
4573 *last_stream = redirect;
4574 last_stream = &redirect->next;
4576 get_arg(redirect->filename, sizeof(redirect->filename), &p);
4577 q = strrchr(redirect->filename, '>');
4580 redirect->stream_type = STREAM_TYPE_REDIRECT;
4582 } else if (!strcasecmp(cmd, "URL")) {
4584 get_arg(redirect->feed_filename, sizeof(redirect->feed_filename), &p);
4585 } else if (!strcasecmp(cmd, "</Redirect>")) {
4587 ERROR("No corresponding <Redirect> for </Redirect>\n");
4589 if (!redirect->feed_filename[0]) {
4590 ERROR("No URL found for <Redirect>\n");
4594 } else if (!strcasecmp(cmd, "LoadModule")) {
4595 get_arg(arg, sizeof(arg), &p);
4599 ERROR("Module support not compiled into this version: '%s'\n", arg);
4602 ERROR("Incorrect keyword: '%s'\n", cmd);
4614 static void handle_child_exit(int sig)
4619 while ((pid = waitpid(-1, &status, WNOHANG)) > 0) {
4622 for (feed = first_feed; feed; feed = feed->next) {
4623 if (feed->pid == pid) {
4624 int uptime = time(0) - feed->pid_start;
4627 fprintf(stderr, "%s: Pid %d exited with status %d after %d seconds\n", feed->filename, pid, status, uptime);
4630 /* Turn off any more restarts */
4631 feed->child_argv = 0;
4636 need_to_start_children = 1;
4639 static void opt_debug(void)
4642 ffserver_daemon = 0;
4643 logfilename[0] = '-';
4646 static int opt_help(const char *opt, const char *arg)
4648 printf("usage: ffserver [options]\n"
4649 "Hyper fast multi format Audio/Video streaming server\n");
4651 show_help_options(options, "Main options:\n", 0, 0);
4655 static const OptionDef options[] = {
4656 #include "cmdutils_common_opts.h"
4657 { "n", OPT_BOOL, {(void *)&no_launch }, "enable no-launch mode" },
4658 { "d", 0, {(void*)opt_debug}, "enable debug mode" },
4659 { "f", HAS_ARG | OPT_STRING, {(void*)&config_filename }, "use configfile instead of /etc/ffserver.conf", "configfile" },
4663 int main(int argc, char **argv)
4665 struct sigaction sigact;
4671 my_program_name = argv[0];
4672 my_program_dir = getcwd(0, 0);
4673 ffserver_daemon = 1;
4675 parse_options(argc, argv, options, NULL);
4677 unsetenv("http_proxy"); /* Kill the http_proxy */
4679 av_lfg_init(&random_state, av_get_random_seed());
4681 memset(&sigact, 0, sizeof(sigact));
4682 sigact.sa_handler = handle_child_exit;
4683 sigact.sa_flags = SA_NOCLDSTOP | SA_RESTART;
4684 sigaction(SIGCHLD, &sigact, 0);
4686 if (parse_ffconfig(config_filename) < 0) {
4687 fprintf(stderr, "Incorrect config file - exiting.\n");
4691 /* open log file if needed */
4692 if (logfilename[0] != '\0') {
4693 if (!strcmp(logfilename, "-"))
4696 logfile = fopen(logfilename, "a");
4697 av_log_set_callback(http_av_log);
4700 build_file_streams();
4702 build_feed_streams();
4704 compute_bandwidth();
4706 /* put the process in background and detach it from its TTY */
4707 if (ffserver_daemon) {
4714 } else if (pid > 0) {
4721 open("/dev/null", O_RDWR);
4722 if (strcmp(logfilename, "-") != 0) {
4732 signal(SIGPIPE, SIG_IGN);
4734 if (ffserver_daemon)
4737 if (http_server() < 0) {
4738 http_log("Could not start server\n");