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");
added
removed
ffmpeg/ffserver.c
