123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968 |
- /*
- * Multiple format streaming server
- * Copyright (c) 2000,2001 Gerard Lantau.
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
- */
- #include <stdarg.h>
- #include <stdlib.h>
- #include <stdio.h>
- #include <string.h>
- #include <netinet/in.h>
- #include <unistd.h>
- #include <fcntl.h>
- #include <sys/ioctl.h>
- #include <sys/poll.h>
- #include <errno.h>
- #include <sys/time.h>
- #include <time.h>
- #include <getopt.h>
- #include <sys/types.h>
- #include <sys/socket.h>
- #include <arpa/inet.h>
- #include <netdb.h>
- #include <ctype.h>
- #include <signal.h>
- #include <assert.h>
- #include "bswap.h" // needed for the bitstream writer in common.h which is included in avformat.h
- #include "avformat.h"
- /* maximum number of simultaneous HTTP connections */
- #define HTTP_MAX_CONNECTIONS 2000
- enum HTTPState {
- HTTPSTATE_WAIT_REQUEST,
- HTTPSTATE_SEND_HEADER,
- HTTPSTATE_SEND_DATA_HEADER,
- HTTPSTATE_SEND_DATA,
- HTTPSTATE_SEND_DATA_TRAILER,
- HTTPSTATE_RECEIVE_DATA,
- HTTPSTATE_WAIT_FEED,
- };
- const char *http_state[] = {
- "WAIT_REQUEST",
- "SEND_HEADER",
- "SEND_DATA_HEADER",
- "SEND_DATA",
- "SEND_DATA_TRAILER",
- "RECEIVE_DATA",
- "WAIT_FEED",
- };
- #define IOBUFFER_MAX_SIZE 32768
- #define PACKET_MAX_SIZE 16384
- /* coef for exponential mean for bitrate estimation in statistics */
- #define AVG_COEF 0.9
- /* timeouts are in ms */
- #define REQUEST_TIMEOUT (15 * 1000)
- #define SYNC_TIMEOUT (10 * 1000)
- /* context associated with one connection */
- typedef struct HTTPContext {
- enum HTTPState state;
- int fd; /* socket file descriptor */
- struct sockaddr_in from_addr; /* origin */
- struct pollfd *poll_entry; /* used when polling */
- long timeout;
- UINT8 *buffer_ptr, *buffer_end;
- int http_error;
- struct HTTPContext *next;
- int got_key_frame; /* stream 0 => 1, stream 1 => 2, stream 2=> 4 */
- INT64 data_count;
- /* feed input */
- int feed_fd;
- /* input format handling */
- AVFormatContext *fmt_in;
- /* output format handling */
- struct FFStream *stream;
- AVFormatContext fmt_ctx;
- int last_packet_sent; /* true if last data packet was sent */
- int suppress_log;
- int bandwidth;
- char protocol[16];
- char method[16];
- char url[128];
- UINT8 buffer[IOBUFFER_MAX_SIZE];
- UINT8 pbuffer[PACKET_MAX_SIZE];
- } HTTPContext;
- /* each generated stream is described here */
- enum StreamType {
- STREAM_TYPE_LIVE,
- STREAM_TYPE_STATUS,
- };
- /* description of each stream of the ffserver.conf file */
- typedef struct FFStream {
- enum StreamType stream_type;
- char filename[1024]; /* stream filename */
- struct FFStream *feed;
- AVFormat *fmt;
- int nb_streams;
- int prebuffer; /* Number of millseconds early to start */
- AVStream *streams[MAX_STREAMS];
- int feed_streams[MAX_STREAMS]; /* index of streams in the feed */
- char feed_filename[1024]; /* file name of the feed storage, or
- input file name for a stream */
- struct FFStream *next;
- /* feed specific */
- int feed_opened; /* true if someone if writing to feed */
- int is_feed; /* true if it is a feed */
- INT64 feed_max_size; /* maximum storage size */
- INT64 feed_write_index; /* current write position in feed (it wraps round) */
- INT64 feed_size; /* current size of feed */
- struct FFStream *next_feed;
- } FFStream;
- typedef struct FeedData {
- long long data_count;
- float avg_frame_size; /* frame size averraged over last frames with exponential mean */
- } FeedData;
- struct sockaddr_in my_addr;
- char logfilename[1024];
- HTTPContext *first_http_ctx;
- FFStream *first_feed; /* contains only feeds */
- FFStream *first_stream; /* contains all streams, including feeds */
- static int handle_http(HTTPContext *c, long cur_time);
- static int http_parse_request(HTTPContext *c);
- static int http_send_data(HTTPContext *c);
- static void compute_stats(HTTPContext *c);
- static int open_input_stream(HTTPContext *c, const char *info);
- static int http_start_receive_data(HTTPContext *c);
- static int http_receive_data(HTTPContext *c);
- int nb_max_connections;
- int nb_connections;
- int nb_max_bandwidth;
- int nb_bandwidth;
- static long gettime_ms(void)
- {
- struct timeval tv;
- gettimeofday(&tv,NULL);
- return (long long)tv.tv_sec * 1000 + (tv.tv_usec / 1000);
- }
- static FILE *logfile = NULL;
- static void http_log(char *fmt, ...)
- {
- va_list ap;
- va_start(ap, fmt);
-
- if (logfile) {
- vfprintf(logfile, fmt, ap);
- fflush(logfile);
- }
- va_end(ap);
- }
- static void log_connection(HTTPContext *c)
- {
- char buf1[32], buf2[32], *p;
- time_t ti;
- if (c->suppress_log)
- return;
- /* XXX: reentrant function ? */
- p = inet_ntoa(c->from_addr.sin_addr);
- strcpy(buf1, p);
- ti = time(NULL);
- p = ctime(&ti);
- strcpy(buf2, p);
- p = buf2 + strlen(p) - 1;
- if (*p == '\n')
- *p = '\0';
- http_log("%s - - [%s] \"%s %s %s\" %d %lld\n",
- buf1, buf2, c->method, c->url, c->protocol, (c->http_error ? c->http_error : 200), c->data_count);
- }
- /* main loop of the http server */
- static int http_server(struct sockaddr_in my_addr)
- {
- int server_fd, tmp, ret;
- struct sockaddr_in from_addr;
- struct pollfd poll_table[HTTP_MAX_CONNECTIONS + 1], *poll_entry;
- HTTPContext *c, **cp;
- long cur_time;
- server_fd = socket(AF_INET,SOCK_STREAM,0);
- if (server_fd < 0) {
- perror ("socket");
- return -1;
- }
-
- tmp = 1;
- setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &tmp, sizeof(tmp));
- if (bind (server_fd, (struct sockaddr *) &my_addr, sizeof (my_addr)) < 0) {
- perror ("bind");
- close(server_fd);
- return -1;
- }
-
- if (listen (server_fd, 5) < 0) {
- perror ("listen");
- close(server_fd);
- return -1;
- }
- http_log("ffserver started.\n");
- fcntl(server_fd, F_SETFL, O_NONBLOCK);
- first_http_ctx = NULL;
- nb_connections = 0;
- first_http_ctx = NULL;
- for(;;) {
- poll_entry = poll_table;
- poll_entry->fd = server_fd;
- poll_entry->events = POLLIN;
- poll_entry++;
- /* wait for events on each HTTP handle */
- c = first_http_ctx;
- while (c != NULL) {
- int fd;
- fd = c->fd;
- switch(c->state) {
- case HTTPSTATE_WAIT_REQUEST:
- c->poll_entry = poll_entry;
- poll_entry->fd = fd;
- poll_entry->events = POLLIN;
- poll_entry++;
- break;
- case HTTPSTATE_SEND_HEADER:
- case HTTPSTATE_SEND_DATA_HEADER:
- case HTTPSTATE_SEND_DATA:
- case HTTPSTATE_SEND_DATA_TRAILER:
- c->poll_entry = poll_entry;
- poll_entry->fd = fd;
- poll_entry->events = POLLOUT;
- poll_entry++;
- break;
- case HTTPSTATE_RECEIVE_DATA:
- c->poll_entry = poll_entry;
- poll_entry->fd = fd;
- poll_entry->events = POLLIN;
- poll_entry++;
- break;
- case HTTPSTATE_WAIT_FEED:
- /* need to catch errors */
- c->poll_entry = poll_entry;
- poll_entry->fd = fd;
- poll_entry->events = 0;
- poll_entry++;
- break;
- default:
- c->poll_entry = NULL;
- break;
- }
- c = c->next;
- }
- /* wait for an event on one connection. We poll at least every
- second to handle timeouts */
- do {
- ret = poll(poll_table, poll_entry - poll_table, 1000);
- } while (ret == -1);
-
- cur_time = gettime_ms();
- /* now handle the events */
- cp = &first_http_ctx;
- while ((*cp) != NULL) {
- c = *cp;
- if (handle_http (c, cur_time) < 0) {
- /* close and free the connection */
- log_connection(c);
- close(c->fd);
- if (c->fmt_in)
- av_close_input_file(c->fmt_in);
- *cp = c->next;
- nb_bandwidth -= c->bandwidth;
- free(c);
- nb_connections--;
- } else {
- cp = &c->next;
- }
- }
- /* new connection request ? */
- poll_entry = poll_table;
- if (poll_entry->revents & POLLIN) {
- int fd, len;
- len = sizeof(from_addr);
- fd = accept(server_fd, (struct sockaddr *)&from_addr,
- &len);
- if (fd >= 0) {
- fcntl(fd, F_SETFL, O_NONBLOCK);
- /* XXX: should output a warning page when coming
- close to the connection limit */
- if (nb_connections >= nb_max_connections) {
- close(fd);
- } else {
- /* add a new connection */
- c = av_mallocz(sizeof(HTTPContext));
- c->next = first_http_ctx;
- first_http_ctx = c;
- c->fd = fd;
- c->poll_entry = NULL;
- c->from_addr = from_addr;
- c->state = HTTPSTATE_WAIT_REQUEST;
- c->buffer_ptr = c->buffer;
- c->buffer_end = c->buffer + IOBUFFER_MAX_SIZE;
- c->timeout = cur_time + REQUEST_TIMEOUT;
- nb_connections++;
- }
- }
- }
- poll_entry++;
- }
- }
- static int handle_http(HTTPContext *c, long cur_time)
- {
- int len;
-
- switch(c->state) {
- case HTTPSTATE_WAIT_REQUEST:
- /* timeout ? */
- if ((c->timeout - cur_time) < 0)
- return -1;
- if (c->poll_entry->revents & (POLLERR | POLLHUP))
- return -1;
- /* no need to read if no events */
- if (!(c->poll_entry->revents & POLLIN))
- return 0;
- /* read the data */
- len = read(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr);
- if (len < 0) {
- if (errno != EAGAIN && errno != EINTR)
- return -1;
- } else if (len == 0) {
- return -1;
- } else {
- /* search for end of request. XXX: not fully correct since garbage could come after the end */
- UINT8 *ptr;
- c->buffer_ptr += len;
- ptr = c->buffer_ptr;
- if ((ptr >= c->buffer + 2 && !memcmp(ptr-2, "\n\n", 2)) ||
- (ptr >= c->buffer + 4 && !memcmp(ptr-4, "\r\n\r\n", 4))) {
- /* request found : parse it and reply */
- if (http_parse_request(c) < 0)
- return -1;
- } else if (ptr >= c->buffer_end) {
- /* request too long: cannot do anything */
- return -1;
- }
- }
- break;
- case HTTPSTATE_SEND_HEADER:
- if (c->poll_entry->revents & (POLLERR | POLLHUP))
- return -1;
- /* no need to read if no events */
- if (!(c->poll_entry->revents & POLLOUT))
- return 0;
- len = write(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr);
- if (len < 0) {
- if (errno != EAGAIN && errno != EINTR) {
- /* error : close connection */
- return -1;
- }
- } else {
- c->buffer_ptr += len;
- if (c->buffer_ptr >= c->buffer_end) {
- /* if error, exit */
- if (c->http_error)
- return -1;
- /* all the buffer was send : synchronize to the incoming stream */
- c->state = HTTPSTATE_SEND_DATA_HEADER;
- c->buffer_ptr = c->buffer_end = c->buffer;
- }
- }
- break;
- case HTTPSTATE_SEND_DATA:
- case HTTPSTATE_SEND_DATA_HEADER:
- case HTTPSTATE_SEND_DATA_TRAILER:
- /* no need to read if no events */
- if (c->poll_entry->revents & (POLLERR | POLLHUP))
- return -1;
-
- if (!(c->poll_entry->revents & POLLOUT))
- return 0;
- if (http_send_data(c) < 0)
- return -1;
- break;
- case HTTPSTATE_RECEIVE_DATA:
- /* no need to read if no events */
- if (c->poll_entry->revents & (POLLERR | POLLHUP))
- return -1;
- if (!(c->poll_entry->revents & POLLIN))
- return 0;
- if (http_receive_data(c) < 0)
- return -1;
- break;
- case HTTPSTATE_WAIT_FEED:
- /* no need to read if no events */
- if (c->poll_entry->revents & (POLLERR | POLLHUP))
- return -1;
- /* nothing to do, we'll be waken up by incoming feed packets */
- break;
- default:
- return -1;
- }
- return 0;
- }
- /* parse http request and prepare header */
- static int http_parse_request(HTTPContext *c)
- {
- char *p;
- int post;
- int doing_asx;
- int doing_ram;
- char cmd[32];
- char info[1024], *filename;
- char url[1024], *q;
- char protocol[32];
- char msg[1024];
- const char *mime_type;
- FFStream *stream;
- int i;
- p = c->buffer;
- q = cmd;
- while (!isspace(*p) && *p != '\0') {
- if ((q - cmd) < sizeof(cmd) - 1)
- *q++ = *p;
- p++;
- }
- *q = '\0';
- strlcpy(c->method, cmd, sizeof(c->method));
- if (!strcmp(cmd, "GET"))
- post = 0;
- else if (!strcmp(cmd, "POST"))
- post = 1;
- else
- return -1;
- while (isspace(*p)) p++;
- q = url;
- while (!isspace(*p) && *p != '\0') {
- if ((q - url) < sizeof(url) - 1)
- *q++ = *p;
- p++;
- }
- *q = '\0';
- strlcpy(c->url, url, sizeof(c->url));
- while (isspace(*p)) p++;
- q = protocol;
- while (!isspace(*p) && *p != '\0') {
- if ((q - protocol) < sizeof(protocol) - 1)
- *q++ = *p;
- p++;
- }
- *q = '\0';
- if (strcmp(protocol, "HTTP/1.0") && strcmp(protocol, "HTTP/1.1"))
- return -1;
- strlcpy(c->protocol, protocol, sizeof(c->protocol));
-
- /* find the filename and the optional info string in the request */
- p = url;
- if (*p == '/')
- p++;
- filename = p;
- p = strchr(p, '?');
- if (p) {
- strlcpy(info, p, sizeof(info));
- *p = '\0';
- } else {
- info[0] = '\0';
- }
- if (strlen(filename) > 4 && strcmp(".asx", filename + strlen(filename) - 4) == 0) {
- doing_asx = 1;
- filename[strlen(filename)-1] = 'f';
- } else {
- doing_asx = 0;
- }
- if (strlen(filename) > 4 &&
- (strcmp(".rpm", filename + strlen(filename) - 4) == 0 ||
- strcmp(".ram", filename + strlen(filename) - 4) == 0)) {
- doing_ram = 1;
- strcpy(filename + strlen(filename)-2, "m");
- } else {
- doing_ram = 0;
- }
- stream = first_stream;
- while (stream != NULL) {
- if (!strcmp(stream->filename, filename))
- break;
- stream = stream->next;
- }
- if (stream == NULL) {
- sprintf(msg, "File '%s' not found", url);
- goto send_error;
- }
- if (post == 0 && stream->stream_type == STREAM_TYPE_LIVE) {
- /* See if we meet the bandwidth requirements */
- for(i=0;i<stream->nb_streams;i++) {
- AVStream *st = stream->streams[i];
- switch(st->codec.codec_type) {
- case CODEC_TYPE_AUDIO:
- c->bandwidth += st->codec.bit_rate;
- break;
- case CODEC_TYPE_VIDEO:
- c->bandwidth += st->codec.bit_rate;
- break;
- default:
- abort();
- }
- }
- }
- c->bandwidth /= 1000;
- nb_bandwidth += c->bandwidth;
- if (post == 0 && nb_max_bandwidth < nb_bandwidth) {
- c->http_error = 200;
- q = c->buffer;
- q += sprintf(q, "HTTP/1.0 200 Server too busy\r\n");
- q += sprintf(q, "Content-type: text/html\r\n");
- q += sprintf(q, "\r\n");
- q += sprintf(q, "<html><head><title>Too busy</title></head><body>\r\n");
- q += sprintf(q, "The server is too busy to serve your request at this time.<p>\r\n");
- q += sprintf(q, "The bandwidth being served (including your stream) is %dkbit/sec, and this exceeds the limit of %dkbit/sec\r\n",
- nb_bandwidth, nb_max_bandwidth);
- q += sprintf(q, "</body></html>\r\n");
- /* prepare output buffer */
- c->buffer_ptr = c->buffer;
- c->buffer_end = q;
- c->state = HTTPSTATE_SEND_HEADER;
- return 0;
- }
-
- if (doing_asx || doing_ram) {
- char *hostinfo = 0;
-
- for (p = c->buffer; *p && *p != '\r' && *p != '\n'; ) {
- if (strncasecmp(p, "Host:", 5) == 0) {
- hostinfo = p + 5;
- break;
- }
- p = strchr(p, '\n');
- if (!p)
- break;
- p++;
- }
- if (hostinfo) {
- char *eoh;
- char hostbuf[260];
- while (isspace(*hostinfo))
- hostinfo++;
- eoh = strchr(hostinfo, '\n');
- if (eoh) {
- if (eoh[-1] == '\r')
- eoh--;
- if (eoh - hostinfo < sizeof(hostbuf) - 1) {
- memcpy(hostbuf, hostinfo, eoh - hostinfo);
- hostbuf[eoh - hostinfo] = 0;
- c->http_error = 200;
- q = c->buffer;
- if (doing_asx) {
- q += sprintf(q, "HTTP/1.0 200 ASX Follows\r\n");
- q += sprintf(q, "Content-type: video/x-ms-asf\r\n");
- q += sprintf(q, "\r\n");
- q += sprintf(q, "<ASX Version=\"3\">\r\n");
- q += sprintf(q, "<!-- Autogenerated by ffserver -->\r\n");
- q += sprintf(q, "<ENTRY><REF HREF=\"http://%s/%s%s\"/></ENTRY>\r\n",
- hostbuf, filename, info);
- q += sprintf(q, "</ASX>\r\n");
- } else if (doing_ram) {
- q += sprintf(q, "HTTP/1.0 200 RAM Follows\r\n");
- q += sprintf(q, "Content-type: audio/x-pn-realaudio\r\n");
- q += sprintf(q, "\r\n");
- q += sprintf(q, "# Autogenerated by ffserver\r\n");
- q += sprintf(q, "http://%s/%s%s\r\n",
- hostbuf, filename, info);
- } else
- abort();
- /* prepare output buffer */
- c->buffer_ptr = c->buffer;
- c->buffer_end = q;
- c->state = HTTPSTATE_SEND_HEADER;
- return 0;
- }
- }
- }
- sprintf(msg, "ASX/RAM file not handled");
- goto send_error;
- }
- c->stream = stream;
- /* XXX: add there authenticate and IP match */
- if (post) {
- /* if post, it means a feed is being sent */
- if (!stream->is_feed) {
- /* However it might be a status report from WMP! Lets log the data
- * as it might come in handy one day
- */
- char *logline = 0;
-
- for (p = c->buffer; *p && *p != '\r' && *p != '\n'; ) {
- if (strncasecmp(p, "Pragma: log-line=", 17) == 0) {
- logline = p;
- break;
- }
- p = strchr(p, '\n');
- if (!p)
- break;
- p++;
- }
- if (logline) {
- char *eol = strchr(logline, '\n');
- logline += 17;
- if (eol) {
- if (eol[-1] == '\r')
- eol--;
- http_log("%.*s\n", eol - logline, logline);
- c->suppress_log = 1;
- }
- }
-
- sprintf(msg, "POST command not handled");
- goto send_error;
- }
- if (http_start_receive_data(c) < 0) {
- sprintf(msg, "could not open feed");
- goto send_error;
- }
- c->http_error = 0;
- c->state = HTTPSTATE_RECEIVE_DATA;
- return 0;
- }
- if (c->stream->stream_type == STREAM_TYPE_STATUS)
- goto send_stats;
- /* open input stream */
- if (open_input_stream(c, info) < 0) {
- sprintf(msg, "Input stream corresponding to '%s' not found", url);
- goto send_error;
- }
- /* prepare http header */
- q = c->buffer;
- q += sprintf(q, "HTTP/1.0 200 OK\r\n");
- mime_type = c->stream->fmt->mime_type;
- if (!mime_type)
- mime_type = "application/x-octet_stream";
- q += sprintf(q, "Pragma: no-cache\r\n");
- /* for asf, we need extra headers */
- if (!strcmp(c->stream->fmt->name,"asf")) {
- q += sprintf(q, "Server: Cougar 4.1.0.3923\r\nCache-Control: no-cache\r\nPragma: client-id=1234\r\nPragma: features=\"broadcast\"\r\n");
- /* mime_type = "application/octet-stream"; */
- /* video/x-ms-asf seems better -- netscape doesn't crash any more! */
- mime_type = "video/x-ms-asf";
- }
- q += sprintf(q, "Content-Type: %s\r\n", mime_type);
- q += sprintf(q, "\r\n");
-
- /* prepare output buffer */
- c->http_error = 0;
- c->buffer_ptr = c->buffer;
- c->buffer_end = q;
- c->state = HTTPSTATE_SEND_HEADER;
- return 0;
- send_error:
- c->http_error = 404;
- q = c->buffer;
- q += sprintf(q, "HTTP/1.0 404 Not Found\r\n");
- q += sprintf(q, "Content-type: %s\r\n", "text/html");
- q += sprintf(q, "\r\n");
- q += sprintf(q, "<HTML>\n");
- q += sprintf(q, "<HEAD><TITLE>404 Not Found</TITLE></HEAD>\n");
- q += sprintf(q, "<BODY>%s</BODY>\n", msg);
- q += sprintf(q, "</HTML>\n");
- /* prepare output buffer */
- c->buffer_ptr = c->buffer;
- c->buffer_end = q;
- c->state = HTTPSTATE_SEND_HEADER;
- return 0;
- send_stats:
- compute_stats(c);
- c->http_error = 200; /* horrible : we use this value to avoid
- going to the send data state */
- c->state = HTTPSTATE_SEND_HEADER;
- return 0;
- }
- static void compute_stats(HTTPContext *c)
- {
- HTTPContext *c1;
- FFStream *stream;
- char *q, *p;
- time_t ti;
- int i;
- q = c->buffer;
- q += sprintf(q, "HTTP/1.0 200 OK\r\n");
- q += sprintf(q, "Content-type: %s\r\n", "text/html");
- q += sprintf(q, "Pragma: no-cache\r\n");
- q += sprintf(q, "\r\n");
-
- q += sprintf(q, "<HEAD><TITLE>FFServer Status</TITLE></HEAD>\n<BODY>");
- q += sprintf(q, "<H1>FFServer Status</H1>\n");
- /* format status */
- q += sprintf(q, "<H1>Available Streams</H1>\n");
- q += sprintf(q, "<TABLE>\n");
- q += sprintf(q, "<TR><TD>Path<TD>Format<TD>Bit rate (kbits/s)<TD>Video<TD>Audio<TD>Feed\n");
- stream = first_stream;
- while (stream != NULL) {
- char sfilename[1024];
- char *eosf;
- strlcpy(sfilename, stream->filename, sizeof(sfilename) - 1);
- eosf = sfilename + strlen(sfilename);
- if (eosf - sfilename >= 4) {
- if (strcmp(eosf - 4, ".asf") == 0) {
- strcpy(eosf - 4, ".asx");
- } else if (strcmp(eosf - 3, ".rm") == 0) {
- strcpy(eosf - 3, ".ram");
- }
- }
-
- q += sprintf(q, "<TR><TD><A HREF=\"/%s\">%s</A> ",
- sfilename, stream->filename);
- switch(stream->stream_type) {
- case STREAM_TYPE_LIVE:
- {
- int audio_bit_rate = 0;
- int video_bit_rate = 0;
- for(i=0;i<stream->nb_streams;i++) {
- AVStream *st = stream->streams[i];
- switch(st->codec.codec_type) {
- case CODEC_TYPE_AUDIO:
- audio_bit_rate += st->codec.bit_rate;
- break;
- case CODEC_TYPE_VIDEO:
- video_bit_rate += st->codec.bit_rate;
- break;
- default:
- abort();
- }
- }
- q += sprintf(q, "<TD> %s <TD> %d <TD> %d <TD> %d",
- stream->fmt->name,
- (audio_bit_rate + video_bit_rate) / 1000,
- video_bit_rate / 1000, audio_bit_rate / 1000);
- if (stream->feed) {
- q += sprintf(q, "<TD>%s", stream->feed->filename);
- } else {
- q += sprintf(q, "<TD>%s", stream->feed_filename);
- }
- q += sprintf(q, "\n");
- }
- break;
- default:
- q += sprintf(q, "<TD> - <TD> - <TD> - <TD> -\n");
- break;
- }
- stream = stream->next;
- }
- q += sprintf(q, "</TABLE>\n");
-
- #if 0
- {
- float avg;
- AVCodecContext *enc;
- char buf[1024];
-
- /* feed status */
- stream = first_feed;
- while (stream != NULL) {
- q += sprintf(q, "<H1>Feed '%s'</H1>\n", stream->filename);
- q += sprintf(q, "<TABLE>\n");
- q += sprintf(q, "<TR><TD>Parameters<TD>Frame count<TD>Size<TD>Avg bitrate (kbits/s)\n");
- for(i=0;i<stream->nb_streams;i++) {
- AVStream *st = stream->streams[i];
- FeedData *fdata = st->priv_data;
- enc = &st->codec;
-
- avcodec_string(buf, sizeof(buf), enc);
- avg = fdata->avg_frame_size * (float)enc->rate * 8.0;
- if (enc->codec->type == CODEC_TYPE_AUDIO && enc->frame_size > 0)
- avg /= enc->frame_size;
- q += sprintf(q, "<TR><TD>%s <TD> %d <TD> %Ld <TD> %0.1f\n",
- buf, enc->frame_number, fdata->data_count, avg / 1000.0);
- }
- q += sprintf(q, "</TABLE>\n");
- stream = stream->next_feed;
- }
- }
- #endif
- /* connection status */
- q += sprintf(q, "<H1>Connection Status</H1>\n");
- q += sprintf(q, "Number of connections: %d / %d<BR>\n",
- nb_connections, nb_max_connections);
- q += sprintf(q, "Bandwidth in use: %dk / %dk<BR>\n",
- nb_bandwidth, nb_max_bandwidth);
- q += sprintf(q, "<TABLE>\n");
- q += sprintf(q, "<TR><TD>#<TD>File<TD>IP<TD>State<TD>Size\n");
- c1 = first_http_ctx;
- i = 0;
- while (c1 != NULL) {
- i++;
- p = inet_ntoa(c1->from_addr.sin_addr);
- q += sprintf(q, "<TR><TD><B>%d</B><TD>%s%s <TD> %s <TD> %s <TD> %Ld\n",
- i, c1->stream->filename,
- c1->state == HTTPSTATE_RECEIVE_DATA ? "(input)" : "",
- p,
- http_state[c1->state],
- c1->data_count);
- c1 = c1->next;
- }
- q += sprintf(q, "</TABLE>\n");
-
- /* date */
- ti = time(NULL);
- p = ctime(&ti);
- q += sprintf(q, "<HR>Generated at %s", p);
- q += sprintf(q, "</BODY>\n</HTML>\n");
- c->buffer_ptr = c->buffer;
- c->buffer_end = q;
- }
- static void http_write_packet(void *opaque,
- unsigned char *buf, int size)
- {
- HTTPContext *c = opaque;
- if (c->buffer_ptr == c->buffer_end || !c->buffer_ptr)
- c->buffer_ptr = c->buffer_end = c->buffer;
- if (c->buffer_end - c->buffer + size > IOBUFFER_MAX_SIZE)
- abort();
- memcpy(c->buffer_end, buf, size);
- c->buffer_end += size;
- }
- static int open_input_stream(HTTPContext *c, const char *info)
- {
- char buf[128];
- char input_filename[1024];
- AVFormatContext *s;
- int buf_size;
- INT64 stream_pos;
- /* find file name */
- if (c->stream->feed) {
- strcpy(input_filename, c->stream->feed->feed_filename);
- buf_size = FFM_PACKET_SIZE;
- /* compute position (absolute time) */
- if (find_info_tag(buf, sizeof(buf), "date", info)) {
- stream_pos = parse_date(buf, 0);
- } else if (find_info_tag(buf, sizeof(buf), "buffer", info)) {
- int prebuffer = strtol(buf, 0, 10);
- stream_pos = gettime() - prebuffer * 1000000;
- } else {
- stream_pos = gettime() - c->stream->prebuffer * 1000;
- }
- } else {
- strcpy(input_filename, c->stream->feed_filename);
- buf_size = 0;
- /* compute position (relative time) */
- if (find_info_tag(buf, sizeof(buf), "date", info)) {
- stream_pos = parse_date(buf, 1);
- } else {
- stream_pos = 0;
- }
- }
- if (input_filename[0] == '\0')
- return -1;
- /* open stream */
- s = av_open_input_file(input_filename, NULL, buf_size, NULL);
- if (!s)
- return -1;
- c->fmt_in = s;
- if (c->fmt_in->format->read_seek) {
- c->fmt_in->format->read_seek(c->fmt_in, stream_pos);
- }
-
- // printf("stream %s opened pos=%0.6f\n", input_filename, stream_pos / 1000000.0);
- return 0;
- }
- static int http_prepare_data(HTTPContext *c)
- {
- int i;
- switch(c->state) {
- case HTTPSTATE_SEND_DATA_HEADER:
- memset(&c->fmt_ctx, 0, sizeof(c->fmt_ctx));
- if (c->stream->feed) {
- /* open output stream by using specified codecs */
- c->fmt_ctx.format = c->stream->fmt;
- c->fmt_ctx.nb_streams = c->stream->nb_streams;
- for(i=0;i<c->fmt_ctx.nb_streams;i++) {
- AVStream *st;
- st = av_mallocz(sizeof(AVStream));
- c->fmt_ctx.streams[i] = st;
- if (c->stream->feed == c->stream)
- memcpy(st, c->stream->streams[i], sizeof(AVStream));
- else
- memcpy(st, c->stream->feed->streams[c->stream->feed_streams[i]], sizeof(AVStream));
- st->codec.frame_number = 0; /* XXX: should be done in
- AVStream, not in codec */
- }
- c->got_key_frame = 0;
- } else {
- /* open output stream by using codecs in specified file */
- c->fmt_ctx.format = c->stream->fmt;
- c->fmt_ctx.nb_streams = c->fmt_in->nb_streams;
- for(i=0;i<c->fmt_ctx.nb_streams;i++) {
- AVStream *st;
- st = av_mallocz(sizeof(AVStream));
- c->fmt_ctx.streams[i] = st;
- memcpy(st, c->fmt_in->streams[i], sizeof(AVStream));
- st->codec.frame_number = 0; /* XXX: should be done in
- AVStream, not in codec */
- }
- c->got_key_frame = 0;
- }
- init_put_byte(&c->fmt_ctx.pb, c->pbuffer, PACKET_MAX_SIZE,
- 1, c, NULL, http_write_packet, NULL);
- c->fmt_ctx.pb.is_streamed = 1;
- /* prepare header */
- c->fmt_ctx.format->write_header(&c->fmt_ctx);
- c->state = HTTPSTATE_SEND_DATA;
- c->last_packet_sent = 0;
- break;
- case HTTPSTATE_SEND_DATA:
- /* find a new packet */
- #if 0
- fifo_total_size = http_fifo_write_count - c->last_http_fifo_write_count;
- if (fifo_total_size >= ((3 * FIFO_MAX_SIZE) / 4)) {
- /* overflow : resync. We suppose that wptr is at this
- point a pointer to a valid packet */
- c->rptr = http_fifo.wptr;
- c->got_key_frame = 0;
- }
-
- start_rptr = c->rptr;
- if (fifo_read(&http_fifo, (UINT8 *)&hdr, sizeof(hdr), &c->rptr) < 0)
- return 0;
- payload_size = ntohs(hdr.payload_size);
- payload = malloc(payload_size);
- if (fifo_read(&http_fifo, payload, payload_size, &c->rptr) < 0) {
- /* cannot read all the payload */
- free(payload);
- c->rptr = start_rptr;
- return 0;
- }
-
- c->last_http_fifo_write_count = http_fifo_write_count -
- fifo_size(&http_fifo, c->rptr);
-
- if (c->stream->stream_type != STREAM_TYPE_MASTER) {
- /* test if the packet can be handled by this format */
- ret = 0;
- for(i=0;i<c->fmt_ctx.nb_streams;i++) {
- AVStream *st = c->fmt_ctx.streams[i];
- if (test_header(&hdr, &st->codec)) {
- /* only begin sending when got a key frame */
- if (st->codec.key_frame)
- c->got_key_frame |= 1 << i;
- if (c->got_key_frame & (1 << i)) {
- ret = c->fmt_ctx.format->write_packet(&c->fmt_ctx, i,
- payload, payload_size);
- }
- break;
- }
- }
- if (ret) {
- /* must send trailer now */
- c->state = HTTPSTATE_SEND_DATA_TRAILER;
- }
- } else {
- /* master case : send everything */
- char *q;
- q = c->buffer;
- memcpy(q, &hdr, sizeof(hdr));
- q += sizeof(hdr);
- memcpy(q, payload, payload_size);
- q += payload_size;
- c->buffer_ptr = c->buffer;
- c->buffer_end = q;
- }
- free(payload);
- #endif
- {
- AVPacket pkt;
- /* read a packet from the input stream */
- if (c->stream->feed) {
- ffm_set_write_index(c->fmt_in,
- c->stream->feed->feed_write_index,
- c->stream->feed->feed_size);
- }
- if (av_read_packet(c->fmt_in, &pkt) < 0) {
- if (c->stream->feed && c->stream->feed->feed_opened) {
- /* if coming from feed, it means we reached the end of the
- ffm file, so must wait for more data */
- c->state = HTTPSTATE_WAIT_FEED;
- return 1; /* state changed */
- } else {
- /* must send trailer now because eof or error */
- c->state = HTTPSTATE_SEND_DATA_TRAILER;
- }
- } else {
- /* send it to the appropriate stream */
- if (c->stream->feed) {
- /* if coming from a feed, select the right stream */
- for(i=0;i<c->stream->nb_streams;i++) {
- if (c->stream->feed_streams[i] == pkt.stream_index) {
- pkt.stream_index = i;
- if (pkt.flags & PKT_FLAG_KEY) {
- c->got_key_frame |= 1 << i;
- }
- /* See if we have all the key frames, then
- * we start to send. This logic is not quite
- * right, but it works for the case of a
- * single video stream with one or more
- * audio streams (for which every frame is
- * typically a key frame).
- */
- if ((c->got_key_frame + 1) >> c->stream->nb_streams) {
- goto send_it;
- }
- }
- }
- } else {
- AVCodecContext *codec;
- send_it:
- /* Fudge here */
- codec = &c->fmt_ctx.streams[pkt.stream_index]->codec;
- codec->key_frame = ((pkt.flags & PKT_FLAG_KEY) != 0);
- #ifdef PJSG
- if (codec->codec_type == CODEC_TYPE_AUDIO) {
- codec->frame_size = (codec->sample_rate * pkt.duration + 500000) / 1000000;
- /* printf("Calculated size %d, from sr %d, duration %d\n", codec->frame_size, codec->sample_rate, pkt.duration); */
- }
- #endif
- if (av_write_packet(&c->fmt_ctx, &pkt, 0))
- c->state = HTTPSTATE_SEND_DATA_TRAILER;
- codec->frame_number++;
- }
- av_free_packet(&pkt);
- }
- }
- break;
- default:
- case HTTPSTATE_SEND_DATA_TRAILER:
- /* last packet test ? */
- if (c->last_packet_sent)
- return -1;
- /* prepare header */
- c->fmt_ctx.format->write_trailer(&c->fmt_ctx);
- c->last_packet_sent = 1;
- break;
- }
- return 0;
- }
- /* should convert the format at the same time */
- static int http_send_data(HTTPContext *c)
- {
- int len, ret;
- while (c->buffer_ptr >= c->buffer_end) {
- ret = http_prepare_data(c);
- if (ret < 0)
- return -1;
- else if (ret == 0) {
- break;
- } else {
- /* state change requested */
- return 0;
- }
- }
- if (c->buffer_end > c->buffer_ptr) {
- len = write(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr);
- if (len < 0) {
- if (errno != EAGAIN && errno != EINTR) {
- /* error : close connection */
- return -1;
- }
- } else {
- c->buffer_ptr += len;
- c->data_count += len;
- }
- }
- return 0;
- }
- static int http_start_receive_data(HTTPContext *c)
- {
- int fd;
- if (c->stream->feed_opened)
- return -1;
- /* open feed */
- fd = open(c->stream->feed_filename, O_RDWR);
- if (fd < 0)
- return -1;
- c->feed_fd = fd;
-
- c->stream->feed_write_index = ffm_read_write_index(fd);
- c->stream->feed_size = lseek(fd, 0, SEEK_END);
- lseek(fd, 0, SEEK_SET);
- /* init buffer input */
- c->buffer_ptr = c->buffer;
- c->buffer_end = c->buffer + FFM_PACKET_SIZE;
- c->stream->feed_opened = 1;
- return 0;
- }
-
- static int http_receive_data(HTTPContext *c)
- {
- int len;
- HTTPContext *c1;
- if (c->buffer_ptr >= c->buffer_end) {
- FFStream *feed = c->stream;
- /* a packet has been received : write it in the store, except
- if header */
- if (c->data_count > FFM_PACKET_SIZE) {
-
- // printf("writing pos=0x%Lx size=0x%Lx\n", feed->feed_write_index, feed->feed_size);
- /* XXX: use llseek or url_seek */
- lseek(c->feed_fd, feed->feed_write_index, SEEK_SET);
- write(c->feed_fd, c->buffer, FFM_PACKET_SIZE);
-
- feed->feed_write_index += FFM_PACKET_SIZE;
- /* update file size */
- if (feed->feed_write_index > c->stream->feed_size)
- feed->feed_size = feed->feed_write_index;
- /* handle wrap around if max file size reached */
- if (feed->feed_write_index >= c->stream->feed_max_size)
- feed->feed_write_index = FFM_PACKET_SIZE;
- /* write index */
- ffm_write_write_index(c->feed_fd, feed->feed_write_index);
- /* wake up any waiting connections */
- for(c1 = first_http_ctx; c1 != NULL; c1 = c1->next) {
- if (c1->state == HTTPSTATE_WAIT_FEED &&
- c1->stream->feed == c->stream->feed) {
- c1->state = HTTPSTATE_SEND_DATA;
- }
- }
- } else {
- /* We have a header in our hands that contains useful data */
- AVFormatContext s;
- ByteIOContext *pb = &s.pb;
- int i;
- memset(&s, 0, sizeof(s));
- url_open_buf(pb, c->buffer, c->buffer_end - c->buffer, URL_RDONLY);
- pb->buf_end = c->buffer_end; /* ?? */
- pb->is_streamed = 1;
- if (feed->fmt->read_header(&s, 0) < 0) {
- goto fail;
- }
- /* Now we have the actual streams */
- if (s.nb_streams != feed->nb_streams) {
- goto fail;
- }
- for (i = 0; i < s.nb_streams; i++) {
- memcpy(&feed->streams[i]->codec, &s.streams[i]->codec, sizeof(AVCodecContext));
- }
- }
- c->buffer_ptr = c->buffer;
- }
- len = read(c->fd, c->buffer_ptr, c->buffer_end - c->buffer_ptr);
- if (len < 0) {
- if (errno != EAGAIN && errno != EINTR) {
- /* error : close connection */
- goto fail;
- }
- } else if (len == 0) {
- /* end of connection : close it */
- goto fail;
- } else {
- c->buffer_ptr += len;
- c->data_count += len;
- }
- return 0;
- fail:
- c->stream->feed_opened = 0;
- close(c->feed_fd);
- return -1;
- }
- /* return the stream number in the feed */
- int add_av_stream(FFStream *feed,
- AVStream *st)
- {
- AVStream *fst;
- AVCodecContext *av, *av1;
- int i;
- av = &st->codec;
- for(i=0;i<feed->nb_streams;i++) {
- st = feed->streams[i];
- av1 = &st->codec;
- if (av1->codec_id == av->codec_id &&
- av1->codec_type == av->codec_type &&
- av1->bit_rate == av->bit_rate) {
- switch(av->codec_type) {
- case CODEC_TYPE_AUDIO:
- if (av1->channels == av->channels &&
- av1->sample_rate == av->sample_rate)
- goto found;
- break;
- case CODEC_TYPE_VIDEO:
- if (av1->width == av->width &&
- av1->height == av->height &&
- av1->frame_rate == av->frame_rate &&
- av1->gop_size == av->gop_size)
- goto found;
- break;
- default:
- abort();
- }
- }
- }
-
- fst = av_mallocz(sizeof(AVStream));
- if (!fst)
- return -1;
- fst->priv_data = av_mallocz(sizeof(FeedData));
- memcpy(&fst->codec, av, sizeof(AVCodecContext));
- feed->streams[feed->nb_streams++] = fst;
- return feed->nb_streams - 1;
- found:
- return i;
- }
- /* compute the needed AVStream for each feed */
- void build_feed_streams(void)
- {
- FFStream *stream, *feed;
- int i;
- /* gather all streams */
- for(stream = first_stream; stream != NULL; stream = stream->next) {
- feed = stream->feed;
- if (feed) {
- if (!stream->is_feed) {
- for(i=0;i<stream->nb_streams;i++) {
- stream->feed_streams[i] = add_av_stream(feed, stream->streams[i]);
- }
- } else {
- for(i=0;i<stream->nb_streams;i++) {
- stream->feed_streams[i] = i;
- }
- }
- }
- }
- /* create feed files if needed */
- for(feed = first_feed; feed != NULL; feed = feed->next_feed) {
- int fd;
- if (!url_exist(feed->feed_filename)) {
- AVFormatContext s1, *s = &s1;
- /* only write the header of the ffm file */
- if (url_fopen(&s->pb, feed->feed_filename, URL_WRONLY) < 0) {
- fprintf(stderr, "Could not open output feed file '%s'\n",
- feed->feed_filename);
- exit(1);
- }
- s->format = feed->fmt;
- s->nb_streams = feed->nb_streams;
- for(i=0;i<s->nb_streams;i++) {
- AVStream *st;
- st = feed->streams[i];
- s->streams[i] = st;
- }
- s->format->write_header(s);
- url_fclose(&s->pb);
- }
- /* get feed size and write index */
- fd = open(feed->feed_filename, O_RDONLY);
- if (fd < 0) {
- fprintf(stderr, "Could not open output feed file '%s'\n",
- feed->feed_filename);
- exit(1);
- }
- feed->feed_write_index = ffm_read_write_index(fd);
- feed->feed_size = lseek(fd, 0, SEEK_END);
- /* ensure that we do not wrap before the end of file */
- if (feed->feed_max_size < feed->feed_size)
- feed->feed_max_size = feed->feed_size;
- close(fd);
- }
- }
- static void get_arg(char *buf, int buf_size, const char **pp)
- {
- const char *p;
- char *q;
- int quote;
- p = *pp;
- while (isspace(*p)) p++;
- q = buf;
- quote = 0;
- if (*p == '\"' || *p == '\'')
- quote = *p++;
- for(;;) {
- if (quote) {
- if (*p == quote)
- break;
- } else {
- if (isspace(*p))
- break;
- }
- if (*p == '\0')
- break;
- if ((q - buf) < buf_size - 1)
- *q++ = *p;
- p++;
- }
- *q = '\0';
- if (quote && *p == quote)
- p++;
- *pp = p;
- }
- /* add a codec and set the default parameters */
- void add_codec(FFStream *stream, AVCodecContext *av)
- {
- AVStream *st;
- /* compute default parameters */
- switch(av->codec_type) {
- case CODEC_TYPE_AUDIO:
- if (av->bit_rate == 0)
- av->bit_rate = 64000;
- if (av->sample_rate == 0)
- av->sample_rate = 22050;
- if (av->channels == 0)
- av->channels = 1;
- break;
- case CODEC_TYPE_VIDEO:
- if (av->bit_rate == 0)
- av->bit_rate = 64000;
- if (av->frame_rate == 0)
- av->frame_rate = 5 * FRAME_RATE_BASE;
- if (av->width == 0 || av->height == 0) {
- av->width = 160;
- av->height = 128;
- }
- /* Bitrate tolerance is less for streaming */
- if (av->bit_rate_tolerance == 0)
- av->bit_rate_tolerance = av->bit_rate / 4;
- if (av->qmin == 0)
- av->qmin = 3;
- if (av->qmax == 0)
- av->qmax = 31;
- if (av->max_qdiff == 0)
- av->max_qdiff = 3;
- av->qcompress = 0.5;
- av->qblur = 0.5;
- break;
- default:
- abort();
- }
- st = av_mallocz(sizeof(AVStream));
- if (!st)
- return;
- stream->streams[stream->nb_streams++] = st;
- memcpy(&st->codec, av, sizeof(AVCodecContext));
- }
- int opt_audio_codec(const char *arg)
- {
- AVCodec *p;
- p = first_avcodec;
- while (p) {
- if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_AUDIO)
- break;
- p = p->next;
- }
- if (p == NULL) {
- return CODEC_ID_NONE;
- }
- return p->id;
- }
- int opt_video_codec(const char *arg)
- {
- AVCodec *p;
- p = first_avcodec;
- while (p) {
- if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_VIDEO)
- break;
- p = p->next;
- }
- if (p == NULL) {
- return CODEC_ID_NONE;
- }
- return p->id;
- }
- int parse_ffconfig(const char *filename)
- {
- FILE *f;
- char line[1024];
- char cmd[64];
- char arg[1024];
- const char *p;
- int val, errors, line_num;
- FFStream **last_stream, *stream;
- FFStream **last_feed, *feed;
- AVCodecContext audio_enc, video_enc;
- int audio_id, video_id;
- f = fopen(filename, "r");
- if (!f) {
- perror(filename);
- return -1;
- }
-
- errors = 0;
- line_num = 0;
- first_stream = NULL;
- last_stream = &first_stream;
- first_feed = NULL;
- last_feed = &first_feed;
- stream = NULL;
- feed = NULL;
- audio_id = CODEC_ID_NONE;
- video_id = CODEC_ID_NONE;
- for(;;) {
- if (fgets(line, sizeof(line), f) == NULL)
- break;
- line_num++;
- p = line;
- while (isspace(*p))
- p++;
- if (*p == '\0' || *p == '#')
- continue;
- get_arg(cmd, sizeof(cmd), &p);
-
- if (!strcasecmp(cmd, "Port")) {
- get_arg(arg, sizeof(arg), &p);
- my_addr.sin_port = htons (atoi(arg));
- } else if (!strcasecmp(cmd, "BindAddress")) {
- get_arg(arg, sizeof(arg), &p);
- if (!inet_aton(arg, &my_addr.sin_addr)) {
- fprintf(stderr, "%s:%d: Invalid IP address: %s\n",
- filename, line_num, arg);
- errors++;
- }
- } else if (!strcasecmp(cmd, "MaxClients")) {
- get_arg(arg, sizeof(arg), &p);
- val = atoi(arg);
- if (val < 1 || val > HTTP_MAX_CONNECTIONS) {
- fprintf(stderr, "%s:%d: Invalid MaxClients: %s\n",
- filename, line_num, arg);
- errors++;
- } else {
- nb_max_connections = val;
- }
- } else if (!strcasecmp(cmd, "MaxBandwidth")) {
- get_arg(arg, sizeof(arg), &p);
- val = atoi(arg);
- if (val < 10 || val > 100000) {
- fprintf(stderr, "%s:%d: Invalid MaxBandwidth: %s\n",
- filename, line_num, arg);
- errors++;
- } else {
- nb_max_bandwidth = val;
- }
- } else if (!strcasecmp(cmd, "CustomLog")) {
- get_arg(logfilename, sizeof(logfilename), &p);
- } else if (!strcasecmp(cmd, "<Feed")) {
- /*********************************************/
- /* Feed related options */
- char *q;
- if (stream || feed) {
- fprintf(stderr, "%s:%d: Already in a tag\n",
- filename, line_num);
- } else {
- feed = av_mallocz(sizeof(FFStream));
- /* add in stream list */
- *last_stream = feed;
- last_stream = &feed->next;
- /* add in feed list */
- *last_feed = feed;
- last_feed = &feed->next_feed;
-
- get_arg(feed->filename, sizeof(feed->filename), &p);
- q = strrchr(feed->filename, '>');
- if (*q)
- *q = '\0';
- feed->fmt = guess_format("ffm", NULL, NULL);
- /* defaut feed file */
- snprintf(feed->feed_filename, sizeof(feed->feed_filename),
- "/tmp/%s.ffm", feed->filename);
- feed->feed_max_size = 5 * 1024 * 1024;
- feed->is_feed = 1;
- feed->feed = feed; /* self feeding :-) */
- }
- } else if (!strcasecmp(cmd, "File")) {
- if (feed) {
- get_arg(feed->feed_filename, sizeof(feed->feed_filename), &p);
- } else if (stream) {
- get_arg(stream->feed_filename, sizeof(stream->feed_filename), &p);
- }
- } else if (!strcasecmp(cmd, "FileMaxSize")) {
- if (feed) {
- const char *p1;
- double fsize;
- get_arg(arg, sizeof(arg), &p);
- p1 = arg;
- fsize = strtod(p1, (char **)&p1);
- switch(toupper(*p1)) {
- case 'K':
- fsize *= 1024;
- break;
- case 'M':
- fsize *= 1024 * 1024;
- break;
- case 'G':
- fsize *= 1024 * 1024 * 1024;
- break;
- }
- feed->feed_max_size = (INT64)fsize;
- }
- } else if (!strcasecmp(cmd, "</Feed>")) {
- if (!feed) {
- fprintf(stderr, "%s:%d: No corresponding <Feed> for </Feed>\n",
- filename, line_num);
- errors++;
- } else {
- /* Make sure that we start out clean */
- if (unlink(feed->feed_filename) < 0
- && errno != ENOENT) {
- fprintf(stderr, "%s:%d: Unable to clean old feed file '%s': %s\n",
- filename, line_num, feed->feed_filename, strerror(errno));
- errors++;
- }
- }
- feed = NULL;
- } else if (!strcasecmp(cmd, "<Stream")) {
- /*********************************************/
- /* Stream related options */
- char *q;
- if (stream || feed) {
- fprintf(stderr, "%s:%d: Already in a tag\n",
- filename, line_num);
- } else {
- stream = av_mallocz(sizeof(FFStream));
- *last_stream = stream;
- last_stream = &stream->next;
- get_arg(stream->filename, sizeof(stream->filename), &p);
- q = strrchr(stream->filename, '>');
- if (*q)
- *q = '\0';
- stream->fmt = guess_format(NULL, stream->filename, NULL);
- memset(&audio_enc, 0, sizeof(AVCodecContext));
- memset(&video_enc, 0, sizeof(AVCodecContext));
- audio_id = CODEC_ID_NONE;
- video_id = CODEC_ID_NONE;
- if (stream->fmt) {
- audio_id = stream->fmt->audio_codec;
- video_id = stream->fmt->video_codec;
- }
- }
- } else if (!strcasecmp(cmd, "Feed")) {
- get_arg(arg, sizeof(arg), &p);
- if (stream) {
- FFStream *sfeed;
-
- sfeed = first_feed;
- while (sfeed != NULL) {
- if (!strcmp(sfeed->filename, arg))
- break;
- sfeed = sfeed->next_feed;
- }
- if (!sfeed) {
- fprintf(stderr, "%s:%d: feed '%s' not defined\n",
- filename, line_num, arg);
- } else {
- stream->feed = sfeed;
- }
- }
- } else if (!strcasecmp(cmd, "Format")) {
- get_arg(arg, sizeof(arg), &p);
- if (!strcmp(arg, "status")) {
- stream->stream_type = STREAM_TYPE_STATUS;
- stream->fmt = NULL;
- } else {
- stream->stream_type = STREAM_TYPE_LIVE;
- /* jpeg cannot be used here, so use single frame jpeg */
- if (!strcmp(arg, "jpeg"))
- strcpy(arg, "singlejpeg");
- stream->fmt = guess_format(arg, NULL, NULL);
- if (!stream->fmt) {
- fprintf(stderr, "%s:%d: Unknown Format: %s\n",
- filename, line_num, arg);
- errors++;
- }
- }
- if (stream->fmt) {
- audio_id = stream->fmt->audio_codec;
- video_id = stream->fmt->video_codec;
- }
- } else if (!strcasecmp(cmd, "Preroll")) {
- get_arg(arg, sizeof(arg), &p);
- if (stream) {
- stream->prebuffer = atoi(arg) * 1000;
- }
- } else if (!strcasecmp(cmd, "AudioCodec")) {
- get_arg(arg, sizeof(arg), &p);
- audio_id = opt_audio_codec(arg);
- if (audio_id == CODEC_ID_NONE) {
- fprintf(stderr, "%s:%d: Unknown AudioCodec: %s\n",
- filename, line_num, arg);
- errors++;
- }
- } else if (!strcasecmp(cmd, "VideoCodec")) {
- get_arg(arg, sizeof(arg), &p);
- video_id = opt_video_codec(arg);
- if (video_id == CODEC_ID_NONE) {
- fprintf(stderr, "%s:%d: Unknown VideoCodec: %s\n",
- filename, line_num, arg);
- errors++;
- }
- } else if (!strcasecmp(cmd, "AudioBitRate")) {
- get_arg(arg, sizeof(arg), &p);
- if (stream) {
- audio_enc.bit_rate = atoi(arg) * 1000;
- }
- } else if (!strcasecmp(cmd, "AudioChannels")) {
- get_arg(arg, sizeof(arg), &p);
- if (stream) {
- audio_enc.channels = atoi(arg);
- }
- } else if (!strcasecmp(cmd, "AudioSampleRate")) {
- get_arg(arg, sizeof(arg), &p);
- if (stream) {
- audio_enc.sample_rate = atoi(arg);
- }
- } else if (!strcasecmp(cmd, "VideoBitRate")) {
- get_arg(arg, sizeof(arg), &p);
- if (stream) {
- video_enc.bit_rate = atoi(arg) * 1000;
- }
- } else if (!strcasecmp(cmd, "VideoSize")) {
- get_arg(arg, sizeof(arg), &p);
- if (stream) {
- parse_image_size(&video_enc.width, &video_enc.height, arg);
- if ((video_enc.width % 16) != 0 ||
- (video_enc.height % 16) != 0) {
- fprintf(stderr, "%s:%d: Image size must be a multiple of 16\n",
- filename, line_num);
- errors++;
- }
- }
- } else if (!strcasecmp(cmd, "VideoFrameRate")) {
- get_arg(arg, sizeof(arg), &p);
- if (stream) {
- video_enc.frame_rate = (int)(strtod(arg, NULL) * FRAME_RATE_BASE);
- }
- } else if (!strcasecmp(cmd, "VideoGopSize")) {
- get_arg(arg, sizeof(arg), &p);
- if (stream) {
- video_enc.gop_size = atoi(arg);
- }
- } else if (!strcasecmp(cmd, "VideoIntraOnly")) {
- if (stream) {
- video_enc.gop_size = 1;
- }
- } else if (!strcasecmp(cmd, "VideoHighQuality")) {
- if (stream) {
- video_enc.flags |= CODEC_FLAG_HQ;
- }
- } else if (!strcasecmp(cmd, "VideoQDiff")) {
- if (stream) {
- video_enc.max_qdiff = atoi(arg);
- if (video_enc.max_qdiff < 1 || video_enc.max_qdiff > 31) {
- fprintf(stderr, "%s:%d: VideoQDiff out of range\n",
- filename, line_num);
- errors++;
- }
- }
- } else if (!strcasecmp(cmd, "VideoQMax")) {
- if (stream) {
- video_enc.qmax = atoi(arg);
- if (video_enc.qmax < 1 || video_enc.qmax > 31) {
- fprintf(stderr, "%s:%d: VideoQMax out of range\n",
- filename, line_num);
- errors++;
- }
- }
- } else if (!strcasecmp(cmd, "VideoQMin")) {
- if (stream) {
- video_enc.qmin = atoi(arg);
- if (video_enc.qmin < 1 || video_enc.qmin > 31) {
- fprintf(stderr, "%s:%d: VideoQMin out of range\n",
- filename, line_num);
- errors++;
- }
- }
- } else if (!strcasecmp(cmd, "NoVideo")) {
- video_id = CODEC_ID_NONE;
- } else if (!strcasecmp(cmd, "NoAudio")) {
- audio_id = CODEC_ID_NONE;
- } else if (!strcasecmp(cmd, "</Stream>")) {
- if (!stream) {
- fprintf(stderr, "%s:%d: No corresponding <Stream> for </Stream>\n",
- filename, line_num);
- errors++;
- }
- if (stream->feed && stream->fmt && strcmp(stream->fmt->name, "ffm") != 0) {
- if (audio_id != CODEC_ID_NONE) {
- audio_enc.codec_type = CODEC_TYPE_AUDIO;
- audio_enc.codec_id = audio_id;
- add_codec(stream, &audio_enc);
- }
- if (video_id != CODEC_ID_NONE) {
- video_enc.codec_type = CODEC_TYPE_VIDEO;
- video_enc.codec_id = video_id;
- add_codec(stream, &video_enc);
- }
- }
- stream = NULL;
- } else {
- fprintf(stderr, "%s:%d: Incorrect keyword: '%s'\n",
- filename, line_num, cmd);
- errors++;
- }
- }
- fclose(f);
- if (errors)
- return -1;
- else
- return 0;
- }
- void *http_server_thread(void *arg)
- {
- http_server(my_addr);
- return NULL;
- }
- #if 0
- static void write_packet(FFCodec *ffenc,
- UINT8 *buf, int size)
- {
- PacketHeader hdr;
- AVCodecContext *enc = &ffenc->enc;
- UINT8 *wptr;
- mk_header(&hdr, enc, size);
- wptr = http_fifo.wptr;
- fifo_write(&http_fifo, (UINT8 *)&hdr, sizeof(hdr), &wptr);
- fifo_write(&http_fifo, buf, size, &wptr);
- /* atomic modification of wptr */
- http_fifo.wptr = wptr;
- ffenc->data_count += size;
- ffenc->avg_frame_size = ffenc->avg_frame_size * AVG_COEF + size * (1.0 - AVG_COEF);
- }
- #endif
- void help(void)
- {
- printf("ffserver version " FFMPEG_VERSION ", Copyright (c) 2000,2001 Gerard Lantau\n"
- "usage: ffserver [-L] [-h] [-f configfile]\n"
- "Hyper fast multi format Audio/Video streaming server\n"
- "\n"
- "-L : print the LICENCE\n"
- "-h : this help\n"
- "-f configfile : use configfile instead of /etc/ffserver.conf\n"
- );
- }
- void licence(void)
- {
- printf(
- "ffserver version " FFMPEG_VERSION "\n"
- "Copyright (c) 2000,2001 Gerard Lantau\n"
- "This program is free software; you can redistribute it and/or modify\n"
- "it under the terms of the GNU General Public License as published by\n"
- "the Free Software Foundation; either version 2 of the License, or\n"
- "(at your option) any later version.\n"
- "\n"
- "This program is distributed in the hope that it will be useful,\n"
- "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
- "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\n"
- "GNU General Public License for more details.\n"
- "\n"
- "You should have received a copy of the GNU General Public License\n"
- "along with this program; if not, write to the Free Software\n"
- "Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.\n"
- );
- }
- int main(int argc, char **argv)
- {
- const char *config_filename;
- int c;
- register_all();
- config_filename = "/etc/ffserver.conf";
- for(;;) {
- c = getopt_long_only(argc, argv, "Lh?f:", NULL, NULL);
- if (c == -1)
- break;
- switch(c) {
- case 'L':
- licence();
- exit(1);
- case '?':
- case 'h':
- help();
- exit(1);
- case 'f':
- config_filename = optarg;
- break;
- default:
- exit(2);
- }
- }
- /* address on which the server will handle connections */
- my_addr.sin_family = AF_INET;
- my_addr.sin_port = htons (8080);
- my_addr.sin_addr.s_addr = htonl (INADDR_ANY);
- nb_max_connections = 5;
- nb_max_bandwidth = 1000;
- first_stream = NULL;
- logfilename[0] = '\0';
- if (parse_ffconfig(config_filename) < 0) {
- fprintf(stderr, "Incorrect config file - exiting.\n");
- exit(1);
- }
- build_feed_streams();
- /* signal init */
- signal(SIGPIPE, SIG_IGN);
- /* open log file if needed */
- if (logfilename[0] != '\0') {
- if (!strcmp(logfilename, "-"))
- logfile = stdout;
- else
- logfile = fopen(logfilename, "w");
- }
- if (http_server(my_addr) < 0) {
- fprintf(stderr, "Could start http server\n");
- exit(1);
- }
- return 0;
- }
|