2 * Multiple format streaming server
3 * Copyright (c) 2000,2001 Gerard Lantau.
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23 #include <netinet/in.h>
26 #include <sys/ioctl.h>
32 #include <sys/types.h>
33 #include <sys/socket.h>
34 #include <arpa/inet.h>
41 /* maximum number of simultaneous HTTP connections */
42 #define HTTP_MAX_CONNECTIONS 2000
45 HTTPSTATE_WAIT_REQUEST,
46 HTTPSTATE_SEND_HEADER,
47 HTTPSTATE_SEND_DATA_HEADER,
49 HTTPSTATE_SEND_DATA_TRAILER,
50 HTTPSTATE_RECEIVE_DATA,
54 const char *http_state[] = {
64 #define IOBUFFER_MAX_SIZE 16384
66 /* coef for exponential mean for bitrate estimation in statistics */
69 /* timeouts are in ms */
70 #define REQUEST_TIMEOUT (15 * 1000)
71 #define SYNC_TIMEOUT (10 * 1000)
73 /* context associated with one connection */
74 typedef struct HTTPContext {
76 int fd; /* socket file descriptor */
77 struct sockaddr_in from_addr; /* origin */
78 struct pollfd *poll_entry; /* used when polling */
80 UINT8 buffer[IOBUFFER_MAX_SIZE];
81 UINT8 *buffer_ptr, *buffer_end;
83 struct HTTPContext *next;
84 int got_key_frame[MAX_STREAMS]; /* for each type */
88 /* input format handling */
89 AVFormatContext *fmt_in;
90 /* output format handling */
91 struct FFStream *stream;
92 AVFormatContext fmt_ctx;
93 int last_packet_sent; /* true if last data packet was sent */
96 /* each generated stream is described here */
102 /* description of each stream of the ffserver.conf file */
103 typedef struct FFStream {
104 enum StreamType stream_type;
105 char filename[1024]; /* stream filename */
106 struct FFStream *feed;
109 AVStream *streams[MAX_STREAMS];
110 int feed_streams[MAX_STREAMS]; /* index of streams in the feed */
111 char feed_filename[1024]; /* file name of the feed storage, or
112 input file name for a stream */
113 struct FFStream *next;
115 int feed_opened; /* true if someone if writing to feed */
116 int is_feed; /* true if it is a feed */
117 INT64 feed_max_size; /* maximum storage size */
118 INT64 feed_write_index; /* current write position in feed (it wraps round) */
119 INT64 feed_size; /* current size of feed */
120 struct FFStream *next_feed;
123 typedef struct FeedData {
124 long long data_count;
125 float avg_frame_size; /* frame size averraged over last frames with exponential mean */
128 struct sockaddr_in my_addr;
129 char logfilename[1024];
130 HTTPContext *first_http_ctx;
131 FFStream *first_feed; /* contains only feeds */
132 FFStream *first_stream; /* contains all streams, including feeds */
134 static int handle_http(HTTPContext *c, long cur_time);
135 static int http_parse_request(HTTPContext *c);
136 static int http_send_data(HTTPContext *c);
137 static void compute_stats(HTTPContext *c);
138 static int open_input_stream(HTTPContext *c, const char *info);
139 static int http_start_receive_data(HTTPContext *c);
140 static int http_receive_data(HTTPContext *c);
142 int nb_max_connections;
145 static long gettime_ms(void)
149 gettimeofday(&tv,NULL);
150 return (long long)tv.tv_sec * 1000 + (tv.tv_usec / 1000);
153 static FILE *logfile = NULL;
155 static void http_log(char *fmt, ...)
161 vfprintf(logfile, fmt, ap);
165 /* main loop of the http server */
166 static int http_server(struct sockaddr_in my_addr)
168 int server_fd, tmp, ret;
169 struct sockaddr_in from_addr;
170 struct pollfd poll_table[HTTP_MAX_CONNECTIONS + 1], *poll_entry;
171 HTTPContext *c, **cp;
174 server_fd = socket(AF_INET,SOCK_STREAM,0);
181 setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &tmp, sizeof(tmp));
183 if (bind (server_fd, (struct sockaddr *) &my_addr, sizeof (my_addr)) < 0) {
189 if (listen (server_fd, 5) < 0) {
195 http_log("ffserver started.\n");
197 fcntl(server_fd, F_SETFL, O_NONBLOCK);
198 first_http_ctx = NULL;
200 first_http_ctx = NULL;
202 poll_entry = poll_table;
203 poll_entry->fd = server_fd;
204 poll_entry->events = POLLIN;
207 /* wait for events on each HTTP handle */
213 case HTTPSTATE_WAIT_REQUEST:
214 c->poll_entry = poll_entry;
216 poll_entry->events = POLLIN;
219 case HTTPSTATE_SEND_HEADER:
220 case HTTPSTATE_SEND_DATA_HEADER:
221 case HTTPSTATE_SEND_DATA:
222 case HTTPSTATE_SEND_DATA_TRAILER:
223 c->poll_entry = poll_entry;
225 poll_entry->events = POLLOUT;
228 case HTTPSTATE_RECEIVE_DATA:
229 c->poll_entry = poll_entry;
231 poll_entry->events = POLLIN;
234 case HTTPSTATE_WAIT_FEED:
235 /* need to catch errors */
236 c->poll_entry = poll_entry;
238 poll_entry->events = 0;
242 c->poll_entry = NULL;
248 /* wait for an event on one connection. We poll at least every
249 second to handle timeouts */
251 ret = poll(poll_table, poll_entry - poll_table, 1000);
254 cur_time = gettime_ms();
256 /* now handle the events */
258 cp = &first_http_ctx;
259 while ((*cp) != NULL) {
261 if (handle_http (c, cur_time) < 0) {
262 /* close and free the connection */
265 av_close_input_file(c->fmt_in);
274 /* new connection request ? */
275 poll_entry = poll_table;
276 if (poll_entry->revents & POLLIN) {
279 len = sizeof(from_addr);
280 fd = accept(server_fd, (struct sockaddr *)&from_addr,
283 fcntl(fd, F_SETFL, O_NONBLOCK);
284 /* XXX: should output a warning page when coming
285 close to the connection limit */
286 if (nb_connections >= nb_max_connections) {
289 /* add a new connection */
290 c = av_mallocz(sizeof(HTTPContext));
291 c->next = first_http_ctx;
294 c->poll_entry = NULL;
295 c->from_addr = from_addr;
296 c->state = HTTPSTATE_WAIT_REQUEST;
297 c->buffer_ptr = c->buffer;
298 c->buffer_end = c->buffer + IOBUFFER_MAX_SIZE;
299 c->timeout = cur_time + REQUEST_TIMEOUT;
308 static int handle_http(HTTPContext *c, long cur_time)
313 case HTTPSTATE_WAIT_REQUEST:
315 if ((c->timeout - cur_time) < 0)
317 if (c->poll_entry->revents & (POLLERR | POLLHUP))
320 /* no need to read if no events */
321 if (!(c->poll_entry->revents & POLLIN))
324 len = read(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr);
326 if (errno != EAGAIN && errno != EINTR)
328 } else if (len == 0) {
331 /* search for end of request. XXX: not fully correct since garbage could come after the end */
333 c->buffer_ptr += len;
335 if ((ptr >= c->buffer + 2 && !memcmp(ptr-2, "\n\n", 2)) ||
336 (ptr >= c->buffer + 4 && !memcmp(ptr-4, "\r\n\r\n", 4))) {
337 /* request found : parse it and reply */
338 if (http_parse_request(c) < 0)
340 } else if (ptr >= c->buffer_end) {
341 /* request too long: cannot do anything */
347 case HTTPSTATE_SEND_HEADER:
348 if (c->poll_entry->revents & (POLLERR | POLLHUP))
351 /* no need to read if no events */
352 if (!(c->poll_entry->revents & POLLOUT))
354 len = write(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr);
356 if (errno != EAGAIN && errno != EINTR) {
357 /* error : close connection */
361 c->buffer_ptr += len;
362 if (c->buffer_ptr >= c->buffer_end) {
366 /* all the buffer was send : synchronize to the incoming stream */
367 c->state = HTTPSTATE_SEND_DATA_HEADER;
368 c->buffer_ptr = c->buffer_end = c->buffer;
373 case HTTPSTATE_SEND_DATA:
374 case HTTPSTATE_SEND_DATA_HEADER:
375 case HTTPSTATE_SEND_DATA_TRAILER:
376 /* no need to read if no events */
377 if (c->poll_entry->revents & (POLLERR | POLLHUP))
380 if (!(c->poll_entry->revents & POLLOUT))
382 if (http_send_data(c) < 0)
385 case HTTPSTATE_RECEIVE_DATA:
386 /* no need to read if no events */
387 if (c->poll_entry->revents & (POLLERR | POLLHUP))
389 if (!(c->poll_entry->revents & POLLIN))
391 if (http_receive_data(c) < 0)
394 case HTTPSTATE_WAIT_FEED:
395 /* no need to read if no events */
396 if (c->poll_entry->revents & (POLLERR | POLLHUP))
399 /* nothing to do, we'll be waken up by incoming feed packets */
407 /* parse http request and prepare header */
408 static int http_parse_request(HTTPContext *c)
413 char info[1024], *filename;
417 const char *mime_type;
422 while (!isspace(*p) && *p != '\0') {
423 if ((q - cmd) < sizeof(cmd) - 1)
428 if (!strcmp(cmd, "GET"))
430 else if (!strcmp(cmd, "POST"))
435 while (isspace(*p)) p++;
437 while (!isspace(*p) && *p != '\0') {
438 if ((q - url) < sizeof(url) - 1)
444 while (isspace(*p)) p++;
446 while (!isspace(*p) && *p != '\0') {
447 if ((q - protocol) < sizeof(protocol) - 1)
452 if (strcmp(protocol, "HTTP/1.0") && strcmp(protocol, "HTTP/1.1"))
455 /* find the filename and the optional info string in the request */
468 stream = first_stream;
469 while (stream != NULL) {
470 if (!strcmp(stream->filename, filename))
472 stream = stream->next;
474 if (stream == NULL) {
475 sprintf(msg, "File '%s' not found", url);
480 /* should do it after so that the size can be computed */
482 char buf1[32], buf2[32], *p;
484 /* XXX: reentrant function ? */
485 p = inet_ntoa(c->from_addr.sin_addr);
490 p = buf2 + strlen(p) - 1;
493 http_log("%s - - [%s] \"%s %s %s\" %d %d\n",
494 buf1, buf2, cmd, url, protocol, 200, 1024);
497 /* XXX: add there authenticate and IP match */
500 /* if post, it means a feed is being sent */
501 if (!stream->is_feed) {
502 sprintf(msg, "POST command not handled");
505 if (http_start_receive_data(c) < 0) {
506 sprintf(msg, "could not open feed");
510 c->state = HTTPSTATE_RECEIVE_DATA;
514 if (c->stream->stream_type == STREAM_TYPE_STATUS)
517 /* open input stream */
518 if (open_input_stream(c, info) < 0) {
519 sprintf(msg, "Input stream corresponding to '%s' not found", url);
523 /* prepare http header */
525 q += sprintf(q, "HTTP/1.0 200 OK\r\n");
526 mime_type = c->stream->fmt->mime_type;
528 mime_type = "application/x-octet_stream";
529 q += sprintf(q, "Content-type: %s\r\n", mime_type);
530 q += sprintf(q, "Pragma: no-cache\r\n");
532 /* for asf, we need extra headers */
533 if (!strcmp(c->stream->fmt->name,"asf")) {
534 q += sprintf(q, "Pragma: features=broadcast\r\n");
536 q += sprintf(q, "\r\n");
538 /* prepare output buffer */
540 c->buffer_ptr = c->buffer;
542 c->state = HTTPSTATE_SEND_HEADER;
547 q += sprintf(q, "HTTP/1.0 404 Not Found\r\n");
548 q += sprintf(q, "Content-type: %s\r\n", "text/html");
549 q += sprintf(q, "\r\n");
550 q += sprintf(q, "<HTML>\n");
551 q += sprintf(q, "<HEAD><TITLE>404 Not Found</TITLE></HEAD>\n");
552 q += sprintf(q, "<BODY>%s</BODY>\n", msg);
553 q += sprintf(q, "</HTML>\n");
555 /* prepare output buffer */
556 c->buffer_ptr = c->buffer;
558 c->state = HTTPSTATE_SEND_HEADER;
562 c->http_error = 200; /* horrible : we use this value to avoid
563 going to the send data state */
564 c->state = HTTPSTATE_SEND_HEADER;
568 static void compute_stats(HTTPContext *c)
577 q += sprintf(q, "HTTP/1.0 200 OK\r\n");
578 q += sprintf(q, "Content-type: %s\r\n", "text/html");
579 q += sprintf(q, "Pragma: no-cache\r\n");
580 q += sprintf(q, "\r\n");
582 q += sprintf(q, "<HEAD><TITLE>FFServer Status</TITLE></HEAD>\n<BODY>");
583 q += sprintf(q, "<H1>FFServer Status</H1>\n");
585 q += sprintf(q, "<H1>Available Streams</H1>\n");
586 q += sprintf(q, "<TABLE>\n");
587 q += sprintf(q, "<TR><TD>Path<TD>Format<TD>Bit rate (kbits/s)<TD>Video<TD>Audio<TD>Feed\n");
588 stream = first_stream;
589 while (stream != NULL) {
590 q += sprintf(q, "<TR><TD><A HREF=\"/%s\">%s</A> ",
591 stream->filename, stream->filename);
592 switch(stream->stream_type) {
593 case STREAM_TYPE_LIVE:
595 int audio_bit_rate = 0;
596 int video_bit_rate = 0;
598 for(i=0;i<stream->nb_streams;i++) {
599 AVStream *st = stream->streams[i];
600 switch(st->codec.codec_type) {
601 case CODEC_TYPE_AUDIO:
602 audio_bit_rate += st->codec.bit_rate;
604 case CODEC_TYPE_VIDEO:
605 video_bit_rate += st->codec.bit_rate;
609 q += sprintf(q, "<TD> %s <TD> %d <TD> %d <TD> %d",
611 (audio_bit_rate + video_bit_rate) / 1000,
612 video_bit_rate / 1000, audio_bit_rate / 1000);
614 q += sprintf(q, "<TD>%s", stream->feed->filename);
616 q += sprintf(q, "<TD>%s", stream->feed_filename);
618 q += sprintf(q, "\n");
622 q += sprintf(q, "<TD> - <TD> - <TD> - <TD> -\n");
625 stream = stream->next;
627 q += sprintf(q, "</TABLE>\n");
637 while (stream != NULL) {
638 q += sprintf(q, "<H1>Feed '%s'</H1>\n", stream->filename);
639 q += sprintf(q, "<TABLE>\n");
640 q += sprintf(q, "<TR><TD>Parameters<TD>Frame count<TD>Size<TD>Avg bitrate (kbits/s)\n");
641 for(i=0;i<stream->nb_streams;i++) {
642 AVStream *st = stream->streams[i];
643 FeedData *fdata = st->priv_data;
646 avcodec_string(buf, sizeof(buf), enc);
647 avg = fdata->avg_frame_size * (float)enc->rate * 8.0;
648 if (enc->codec->type == CODEC_TYPE_AUDIO && enc->frame_size > 0)
649 avg /= enc->frame_size;
650 q += sprintf(q, "<TR><TD>%s <TD> %d <TD> %Ld <TD> %0.1f\n",
651 buf, enc->frame_number, fdata->data_count, avg / 1000.0);
653 q += sprintf(q, "</TABLE>\n");
654 stream = stream->next_feed;
659 /* connection status */
660 q += sprintf(q, "<H1>Connection Status</H1>\n");
662 q += sprintf(q, "Number of connections: %d / %d<BR>\n",
663 nb_connections, nb_max_connections);
665 q += sprintf(q, "<TABLE>\n");
666 q += sprintf(q, "<TR><TD>#<TD>File<TD>IP<TD>State<TD>Size\n");
671 p = inet_ntoa(c1->from_addr.sin_addr);
672 q += sprintf(q, "<TR><TD><B>%d</B><TD>%s%s <TD> %s <TD> %s <TD> %Ld\n",
673 i, c1->stream->filename,
674 c1->state == HTTPSTATE_RECEIVE_DATA ? "(input)" : "",
676 http_state[c1->state],
680 q += sprintf(q, "</TABLE>\n");
685 q += sprintf(q, "<HR>Generated at %s", p);
686 q += sprintf(q, "</BODY>\n</HTML>\n");
688 c->buffer_ptr = c->buffer;
693 static void http_write_packet(void *opaque,
694 unsigned char *buf, int size)
696 HTTPContext *c = opaque;
697 if (size > IOBUFFER_MAX_SIZE)
699 memcpy(c->buffer, buf, size);
700 c->buffer_ptr = c->buffer;
701 c->buffer_end = c->buffer + size;
704 static int open_input_stream(HTTPContext *c, const char *info)
707 char input_filename[1024];
713 if (c->stream->feed) {
714 strcpy(input_filename, c->stream->feed->feed_filename);
715 buf_size = FFM_PACKET_SIZE;
716 /* compute position (absolute time) */
717 if (find_info_tag(buf, sizeof(buf), "date", info)) {
718 stream_pos = parse_date(buf, 0);
720 stream_pos = gettime();
723 strcpy(input_filename, c->stream->feed_filename);
725 /* compute position (relative time) */
726 if (find_info_tag(buf, sizeof(buf), "date", info)) {
727 stream_pos = parse_date(buf, 1);
732 if (input_filename[0] == '\0')
736 s = av_open_input_file(input_filename, buf_size);
741 if (c->fmt_in->format->read_seek) {
742 c->fmt_in->format->read_seek(c->fmt_in, stream_pos);
745 // printf("stream %s opened pos=%0.6f\n", input_filename, stream_pos / 1000000.0);
749 static int http_prepare_data(HTTPContext *c)
754 case HTTPSTATE_SEND_DATA_HEADER:
755 memset(&c->fmt_ctx, 0, sizeof(c->fmt_ctx));
756 if (c->stream->feed) {
757 /* open output stream by using specified codecs */
758 c->fmt_ctx.format = c->stream->fmt;
759 c->fmt_ctx.nb_streams = c->stream->nb_streams;
760 for(i=0;i<c->fmt_ctx.nb_streams;i++) {
762 st = av_mallocz(sizeof(AVStream));
763 c->fmt_ctx.streams[i] = st;
764 memcpy(st, c->stream->streams[i], sizeof(AVStream));
765 st->codec.frame_number = 0; /* XXX: should be done in
766 AVStream, not in codec */
767 c->got_key_frame[i] = 0;
770 /* open output stream by using codecs in specified file */
771 c->fmt_ctx.format = c->stream->fmt;
772 c->fmt_ctx.nb_streams = c->fmt_in->nb_streams;
773 for(i=0;i<c->fmt_ctx.nb_streams;i++) {
775 st = av_mallocz(sizeof(AVStream));
776 c->fmt_ctx.streams[i] = st;
777 memcpy(st, c->fmt_in->streams[i], sizeof(AVStream));
778 st->codec.frame_number = 0; /* XXX: should be done in
779 AVStream, not in codec */
780 c->got_key_frame[i] = 0;
783 init_put_byte(&c->fmt_ctx.pb, c->buffer, IOBUFFER_MAX_SIZE,
784 1, c, NULL, http_write_packet, NULL);
785 c->fmt_ctx.pb.is_streamed = 1;
787 c->fmt_ctx.format->write_header(&c->fmt_ctx);
788 c->state = HTTPSTATE_SEND_DATA;
789 c->last_packet_sent = 0;
791 case HTTPSTATE_SEND_DATA:
792 /* find a new packet */
794 fifo_total_size = http_fifo_write_count - c->last_http_fifo_write_count;
795 if (fifo_total_size >= ((3 * FIFO_MAX_SIZE) / 4)) {
796 /* overflow : resync. We suppose that wptr is at this
797 point a pointer to a valid packet */
798 c->rptr = http_fifo.wptr;
799 for(i=0;i<c->fmt_ctx.nb_streams;i++) {
800 c->got_key_frame[i] = 0;
804 start_rptr = c->rptr;
805 if (fifo_read(&http_fifo, (UINT8 *)&hdr, sizeof(hdr), &c->rptr) < 0)
807 payload_size = ntohs(hdr.payload_size);
808 payload = malloc(payload_size);
809 if (fifo_read(&http_fifo, payload, payload_size, &c->rptr) < 0) {
810 /* cannot read all the payload */
812 c->rptr = start_rptr;
816 c->last_http_fifo_write_count = http_fifo_write_count -
817 fifo_size(&http_fifo, c->rptr);
819 if (c->stream->stream_type != STREAM_TYPE_MASTER) {
820 /* test if the packet can be handled by this format */
822 for(i=0;i<c->fmt_ctx.nb_streams;i++) {
823 AVStream *st = c->fmt_ctx.streams[i];
824 if (test_header(&hdr, &st->codec)) {
825 /* only begin sending when got a key frame */
826 if (st->codec.key_frame)
827 c->got_key_frame[i] = 1;
828 if (c->got_key_frame[i]) {
829 ret = c->fmt_ctx.format->write_packet(&c->fmt_ctx, i,
830 payload, payload_size);
836 /* must send trailer now */
837 c->state = HTTPSTATE_SEND_DATA_TRAILER;
840 /* master case : send everything */
843 memcpy(q, &hdr, sizeof(hdr));
845 memcpy(q, payload, payload_size);
847 c->buffer_ptr = c->buffer;
855 /* read a packet from the input stream */
856 if (c->stream->feed) {
857 ffm_set_write_index(c->fmt_in,
858 c->stream->feed->feed_write_index,
859 c->stream->feed->feed_size);
861 if (av_read_packet(c->fmt_in, &pkt) < 0) {
862 if (c->stream->feed && c->stream->feed->feed_opened) {
863 /* if coming from feed, it means we reached the end of the
864 ffm file, so must wait for more data */
865 c->state = HTTPSTATE_WAIT_FEED;
866 return 1; /* state changed */
868 /* must send trailer now because eof or error */
869 c->state = HTTPSTATE_SEND_DATA_TRAILER;
872 /* send it to the appropriate stream */
873 if (c->stream->feed) {
874 /* if coming from a feed, select the right stream */
875 for(i=0;i<c->stream->nb_streams;i++) {
876 if (c->stream->feed_streams[i] == pkt.stream_index) {
877 pkt.stream_index = i;
883 av_write_packet(&c->fmt_ctx, &pkt);
886 av_free_packet(&pkt);
891 case HTTPSTATE_SEND_DATA_TRAILER:
892 /* last packet test ? */
893 if (c->last_packet_sent)
896 c->fmt_ctx.format->write_trailer(&c->fmt_ctx);
897 c->last_packet_sent = 1;
903 /* should convert the format at the same time */
904 static int http_send_data(HTTPContext *c)
908 while (c->buffer_ptr >= c->buffer_end) {
909 ret = http_prepare_data(c);
915 /* state change requested */
920 len = write(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr);
922 if (errno != EAGAIN && errno != EINTR) {
923 /* error : close connection */
927 c->buffer_ptr += len;
928 c->data_count += len;
933 static int http_start_receive_data(HTTPContext *c)
937 if (c->stream->feed_opened)
941 fd = open(c->stream->feed_filename, O_RDWR);
946 c->stream->feed_write_index = ffm_read_write_index(fd);
947 c->stream->feed_size = lseek(fd, 0, SEEK_END);
948 lseek(fd, 0, SEEK_SET);
950 /* init buffer input */
951 c->buffer_ptr = c->buffer;
952 c->buffer_end = c->buffer + FFM_PACKET_SIZE;
953 c->stream->feed_opened = 1;
957 static int http_receive_data(HTTPContext *c)
962 if (c->buffer_ptr >= c->buffer_end) {
963 /* a packet has been received : write it in the store, except
965 if (c->data_count > FFM_PACKET_SIZE) {
966 FFStream *feed = c->stream;
968 // printf("writing pos=0x%Lx size=0x%Lx\n", feed->feed_write_index, feed->feed_size);
969 /* XXX: use llseek or url_seek */
970 lseek(c->feed_fd, feed->feed_write_index, SEEK_SET);
971 write(c->feed_fd, c->buffer, FFM_PACKET_SIZE);
973 feed->feed_write_index += FFM_PACKET_SIZE;
974 /* update file size */
975 if (feed->feed_write_index > c->stream->feed_size)
976 feed->feed_size = feed->feed_write_index;
978 /* handle wrap around if max file size reached */
979 if (feed->feed_write_index >= c->stream->feed_max_size)
980 feed->feed_write_index = FFM_PACKET_SIZE;
983 ffm_write_write_index(c->feed_fd, feed->feed_write_index);
985 /* wake up any waiting connections */
986 for(c1 = first_http_ctx; c1 != NULL; c1 = c1->next) {
987 if (c1->state == HTTPSTATE_WAIT_FEED &&
988 c1->stream->feed == c->stream->feed) {
989 c1->state = HTTPSTATE_SEND_DATA;
993 c->buffer_ptr = c->buffer;
996 len = read(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr);
998 if (errno != EAGAIN && errno != EINTR) {
999 /* error : close connection */
1002 } else if (len == 0) {
1003 /* end of connection : close it */
1006 c->buffer_ptr += len;
1007 c->data_count += len;
1011 c->stream->feed_opened = 0;
1016 /* return the stream number in the feed */
1017 int add_av_stream(FFStream *feed,
1021 AVCodecContext *av, *av1;
1025 for(i=0;i<feed->nb_streams;i++) {
1026 st = feed->streams[i];
1028 if (av1->codec == av->codec &&
1029 av1->bit_rate == av->bit_rate) {
1031 switch(av->codec_type) {
1032 case CODEC_TYPE_AUDIO:
1033 if (av1->channels == av->channels &&
1034 av1->sample_rate == av->sample_rate)
1037 case CODEC_TYPE_VIDEO:
1038 if (av1->width == av->width &&
1039 av1->height == av->height &&
1040 av1->frame_rate == av->frame_rate &&
1041 av1->gop_size == av->gop_size)
1048 fst = av_mallocz(sizeof(AVStream));
1051 fst->priv_data = av_mallocz(sizeof(FeedData));
1052 memcpy(&fst->codec, av, sizeof(AVCodecContext));
1053 feed->streams[feed->nb_streams++] = fst;
1054 return feed->nb_streams - 1;
1059 /* compute the needed AVStream for each feed */
1060 void build_feed_streams(void)
1062 FFStream *stream, *feed;
1065 /* gather all streams */
1066 for(stream = first_stream; stream != NULL; stream = stream->next) {
1067 feed = stream->feed;
1069 if (!stream->is_feed) {
1070 for(i=0;i<stream->nb_streams;i++) {
1071 stream->feed_streams[i] = add_av_stream(feed, stream->streams[i]);
1074 for(i=0;i<stream->nb_streams;i++) {
1075 stream->feed_streams[i] = i;
1081 /* create feed files if needed */
1082 for(feed = first_feed; feed != NULL; feed = feed->next_feed) {
1085 if (!url_exist(feed->feed_filename)) {
1086 AVFormatContext s1, *s = &s1;
1088 /* only write the header of the ffm file */
1089 if (url_fopen(&s->pb, feed->feed_filename, URL_WRONLY) < 0) {
1090 fprintf(stderr, "Could not open output feed file '%s'\n",
1091 feed->feed_filename);
1094 s->format = feed->fmt;
1095 s->nb_streams = feed->nb_streams;
1096 for(i=0;i<s->nb_streams;i++) {
1098 st = feed->streams[i];
1101 s->format->write_header(s);
1105 /* get feed size and write index */
1106 fd = open(feed->feed_filename, O_RDONLY);
1108 fprintf(stderr, "Could not open output feed file '%s'\n",
1109 feed->feed_filename);
1113 feed->feed_write_index = ffm_read_write_index(fd);
1114 feed->feed_size = lseek(fd, 0, SEEK_END);
1115 /* ensure that we do not wrap before the end of file */
1116 if (feed->feed_max_size < feed->feed_size)
1117 feed->feed_max_size = feed->feed_size;
1123 static void get_arg(char *buf, int buf_size, const char **pp)
1130 while (isspace(*p)) p++;
1133 if (*p == '\"' || *p == '\'')
1145 if ((q - buf) < buf_size - 1)
1150 if (quote && *p == quote)
1155 /* add a codec and set the default parameters */
1156 void add_codec(FFStream *stream, AVCodecContext *av)
1160 /* compute default parameters */
1161 switch(av->codec_type) {
1162 case CODEC_TYPE_AUDIO:
1163 if (av->bit_rate == 0)
1164 av->bit_rate = 64000;
1165 if (av->sample_rate == 0)
1166 av->sample_rate = 22050;
1167 if (av->channels == 0)
1170 case CODEC_TYPE_VIDEO:
1171 if (av->bit_rate == 0)
1172 av->bit_rate = 64000;
1173 if (av->frame_rate == 0)
1174 av->frame_rate = 5 * FRAME_RATE_BASE;
1175 if (av->width == 0 || av->height == 0) {
1182 st = av_mallocz(sizeof(AVStream));
1185 stream->streams[stream->nb_streams++] = st;
1186 memcpy(&st->codec, av, sizeof(AVCodecContext));
1189 int parse_ffconfig(const char *filename)
1196 int val, errors, line_num;
1197 FFStream **last_stream, *stream;
1198 FFStream **last_feed, *feed;
1199 AVCodecContext audio_enc, video_enc;
1200 int audio_id, video_id;
1202 f = fopen(filename, "r");
1210 first_stream = NULL;
1211 last_stream = &first_stream;
1213 last_feed = &first_feed;
1216 audio_id = CODEC_ID_NONE;
1217 video_id = CODEC_ID_NONE;
1219 if (fgets(line, sizeof(line), f) == NULL)
1225 if (*p == '\0' || *p == '#')
1228 get_arg(cmd, sizeof(cmd), &p);
1230 if (!strcasecmp(cmd, "Port")) {
1231 get_arg(arg, sizeof(arg), &p);
1232 my_addr.sin_port = htons (atoi(arg));
1233 } else if (!strcasecmp(cmd, "BindAddress")) {
1234 get_arg(arg, sizeof(arg), &p);
1235 if (!inet_aton(arg, &my_addr.sin_addr)) {
1236 fprintf(stderr, "%s:%d: Invalid IP address: %s\n",
1237 filename, line_num, arg);
1240 } else if (!strcasecmp(cmd, "MaxClients")) {
1241 get_arg(arg, sizeof(arg), &p);
1243 if (val < 1 || val > HTTP_MAX_CONNECTIONS) {
1244 fprintf(stderr, "%s:%d: Invalid MaxClients: %s\n",
1245 filename, line_num, arg);
1248 nb_max_connections = val;
1250 } else if (!strcasecmp(cmd, "CustomLog")) {
1251 get_arg(logfilename, sizeof(logfilename), &p);
1252 } else if (!strcasecmp(cmd, "<Feed")) {
1253 /*********************************************/
1254 /* Feed related options */
1256 if (stream || feed) {
1257 fprintf(stderr, "%s:%d: Already in a tag\n",
1258 filename, line_num);
1260 feed = av_mallocz(sizeof(FFStream));
1261 /* add in stream list */
1262 *last_stream = feed;
1263 last_stream = &feed->next;
1264 /* add in feed list */
1266 last_feed = &feed->next_feed;
1268 get_arg(feed->filename, sizeof(feed->filename), &p);
1269 q = strrchr(feed->filename, '>');
1272 feed->fmt = guess_format("ffm", NULL, NULL);
1273 /* defaut feed file */
1274 snprintf(feed->feed_filename, sizeof(feed->feed_filename),
1275 "/tmp/%s.ffm", feed->filename);
1276 feed->feed_max_size = 5 * 1024 * 1024;
1278 feed->feed = feed; /* self feeding :-) */
1280 } else if (!strcasecmp(cmd, "File")) {
1282 get_arg(feed->feed_filename, sizeof(feed->feed_filename), &p);
1283 } else if (stream) {
1284 get_arg(stream->feed_filename, sizeof(stream->feed_filename), &p);
1286 } else if (!strcasecmp(cmd, "FileMaxSize")) {
1291 get_arg(arg, sizeof(arg), &p);
1293 fsize = strtod(p1, (char **)&p1);
1294 switch(toupper(*p1)) {
1299 fsize *= 1024 * 1024;
1302 fsize *= 1024 * 1024 * 1024;
1305 feed->feed_max_size = (INT64)fsize;
1307 } else if (!strcasecmp(cmd, "</Feed>")) {
1309 fprintf(stderr, "%s:%d: No corresponding <Feed> for </Feed>\n",
1310 filename, line_num);
1314 } else if (!strcasecmp(cmd, "<Stream")) {
1315 /*********************************************/
1316 /* Stream related options */
1318 if (stream || feed) {
1319 fprintf(stderr, "%s:%d: Already in a tag\n",
1320 filename, line_num);
1322 stream = av_mallocz(sizeof(FFStream));
1323 *last_stream = stream;
1324 last_stream = &stream->next;
1326 get_arg(stream->filename, sizeof(stream->filename), &p);
1327 q = strrchr(stream->filename, '>');
1330 stream->fmt = guess_format(NULL, stream->filename, NULL);
1331 memset(&audio_enc, 0, sizeof(AVCodecContext));
1332 memset(&video_enc, 0, sizeof(AVCodecContext));
1333 audio_id = CODEC_ID_NONE;
1334 video_id = CODEC_ID_NONE;
1336 audio_id = stream->fmt->audio_codec;
1337 video_id = stream->fmt->video_codec;
1340 } else if (!strcasecmp(cmd, "Feed")) {
1341 get_arg(arg, sizeof(arg), &p);
1346 while (sfeed != NULL) {
1347 if (!strcmp(sfeed->filename, arg))
1349 sfeed = sfeed->next_feed;
1352 fprintf(stderr, "%s:%d: feed '%s' not defined\n",
1353 filename, line_num, arg);
1355 stream->feed = sfeed;
1358 } else if (!strcasecmp(cmd, "Format")) {
1359 get_arg(arg, sizeof(arg), &p);
1360 if (!strcmp(arg, "status")) {
1361 stream->stream_type = STREAM_TYPE_STATUS;
1364 stream->stream_type = STREAM_TYPE_LIVE;
1365 /* jpeg cannot be used here, so use single frame jpeg */
1366 if (!strcmp(arg, "jpeg"))
1367 strcpy(arg, "singlejpeg");
1368 stream->fmt = guess_format(arg, NULL, NULL);
1370 fprintf(stderr, "%s:%d: Unknown Format: %s\n",
1371 filename, line_num, arg);
1376 audio_id = stream->fmt->audio_codec;
1377 video_id = stream->fmt->video_codec;
1379 } else if (!strcasecmp(cmd, "AudioBitRate")) {
1380 get_arg(arg, sizeof(arg), &p);
1382 audio_enc.bit_rate = atoi(arg) * 1000;
1384 } else if (!strcasecmp(cmd, "AudioChannels")) {
1385 get_arg(arg, sizeof(arg), &p);
1387 audio_enc.channels = atoi(arg);
1389 } else if (!strcasecmp(cmd, "AudioSampleRate")) {
1390 get_arg(arg, sizeof(arg), &p);
1392 audio_enc.sample_rate = atoi(arg);
1394 } else if (!strcasecmp(cmd, "VideoBitRate")) {
1395 get_arg(arg, sizeof(arg), &p);
1397 video_enc.bit_rate = atoi(arg) * 1000;
1399 } else if (!strcasecmp(cmd, "VideoSize")) {
1400 get_arg(arg, sizeof(arg), &p);
1402 parse_image_size(&video_enc.width, &video_enc.height, arg);
1403 if ((video_enc.width % 16) != 0 ||
1404 (video_enc.height % 16) != 0) {
1405 fprintf(stderr, "%s:%d: Image size must be a multiple of 16\n",
1406 filename, line_num);
1410 } else if (!strcasecmp(cmd, "VideoFrameRate")) {
1411 get_arg(arg, sizeof(arg), &p);
1413 video_enc.frame_rate = (int)(strtod(arg, NULL) * FRAME_RATE_BASE);
1415 } else if (!strcasecmp(cmd, "VideoGopSize")) {
1416 get_arg(arg, sizeof(arg), &p);
1418 video_enc.gop_size = atoi(arg);
1420 } else if (!strcasecmp(cmd, "VideoIntraOnly")) {
1422 video_enc.gop_size = 1;
1424 } else if (!strcasecmp(cmd, "NoVideo")) {
1425 video_id = CODEC_ID_NONE;
1426 } else if (!strcasecmp(cmd, "NoAudio")) {
1427 audio_id = CODEC_ID_NONE;
1428 } else if (!strcasecmp(cmd, "</Stream>")) {
1430 fprintf(stderr, "%s:%d: No corresponding <Stream> for </Stream>\n",
1431 filename, line_num);
1434 if (stream->feed && stream->fmt && strcmp(stream->fmt->name, "ffm") != 0) {
1435 if (audio_id != CODEC_ID_NONE) {
1436 audio_enc.codec_type = CODEC_TYPE_AUDIO;
1437 audio_enc.codec_id = audio_id;
1438 add_codec(stream, &audio_enc);
1440 if (video_id != CODEC_ID_NONE) {
1441 video_enc.codec_type = CODEC_TYPE_VIDEO;
1442 video_enc.codec_id = video_id;
1443 add_codec(stream, &video_enc);
1448 fprintf(stderr, "%s:%d: Incorrect keyword: '%s'\n",
1449 filename, line_num, cmd);
1462 void *http_server_thread(void *arg)
1464 http_server(my_addr);
1469 static void write_packet(FFCodec *ffenc,
1470 UINT8 *buf, int size)
1473 AVCodecContext *enc = &ffenc->enc;
1475 mk_header(&hdr, enc, size);
1476 wptr = http_fifo.wptr;
1477 fifo_write(&http_fifo, (UINT8 *)&hdr, sizeof(hdr), &wptr);
1478 fifo_write(&http_fifo, buf, size, &wptr);
1479 /* atomic modification of wptr */
1480 http_fifo.wptr = wptr;
1481 ffenc->data_count += size;
1482 ffenc->avg_frame_size = ffenc->avg_frame_size * AVG_COEF + size * (1.0 - AVG_COEF);
1488 printf("ffserver version " FFMPEG_VERSION ", Copyright (c) 2000,2001 Gerard Lantau\n"
1489 "usage: ffserver [-L] [-h] [-f configfile]\n"
1490 "Hyper fast multi format Audio/Video streaming server\n"
1492 "-L : print the LICENCE\n"
1494 "-f configfile : use configfile instead of /etc/ffserver.conf\n"
1501 "ffserver version " FFMPEG_VERSION "\n"
1502 "Copyright (c) 2000,2001 Gerard Lantau\n"
1503 "This program is free software; you can redistribute it and/or modify\n"
1504 "it under the terms of the GNU General Public License as published by\n"
1505 "the Free Software Foundation; either version 2 of the License, or\n"
1506 "(at your option) any later version.\n"
1508 "This program is distributed in the hope that it will be useful,\n"
1509 "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
1510 "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\n"
1511 "GNU General Public License for more details.\n"
1513 "You should have received a copy of the GNU General Public License\n"
1514 "along with this program; if not, write to the Free Software\n"
1515 "Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.\n"
1519 int main(int argc, char **argv)
1521 const char *config_filename;
1526 config_filename = "/etc/ffserver.conf";
1529 c = getopt_long_only(argc, argv, "Lh?f:", NULL, NULL);
1541 config_filename = optarg;
1548 /* address on which the server will handle connections */
1549 my_addr.sin_family = AF_INET;
1550 my_addr.sin_port = htons (8080);
1551 my_addr.sin_addr.s_addr = htonl (INADDR_ANY);
1552 nb_max_connections = 5;
1553 first_stream = NULL;
1554 logfilename[0] = '\0';
1556 if (parse_ffconfig(config_filename) < 0) {
1557 fprintf(stderr, "Incorrect config file - exiting.\n");
1561 build_feed_streams();
1564 signal(SIGPIPE, SIG_IGN);
1566 /* open log file if needed */
1567 if (logfilename[0] != '\0') {
1568 if (!strcmp(logfilename, "-"))
1571 logfile = fopen(logfilename, "w");
1574 if (http_server(my_addr) < 0) {
1575 fprintf(stderr, "Could start http server\n");