Skip to content
Snippets Groups Projects
ffserver.c 150 KiB
Newer Older
  • Learn to ignore specific revisions
  •     const char *p;
    
        for (p = request; *p && *p != '\r' && *p != '\n'; ) {
            if (strncasecmp(p, "Pragma:", 7) == 0) {
                const char *q = p + 7;
    
                while (*q && *q != '\n' && isspace(*q))
                    q++;
    
                if (strncasecmp(q, "stream-switch-entry=", 20) == 0) {
                    int stream_no;
                    int rate_no;
    
                    q += 20;
    
    
                    memset(rates, 0xff, ratelen);
    
                        if (sscanf(q, ":%d:%d", &stream_no, &rate_no) != 2)
    
                        if (stream_no < ratelen && stream_no >= 0)
    
                            rates[stream_no] = rate_no;
    
                        while (*q && *q != '\n' && !isspace(*q))
                            q++;
                    }
    
                    return 1;
                }
            }
            p = strchr(p, '\n');
            if (!p)
                break;
    
            p++;
        }
    
        return 0;
    }
    
    
    static int find_stream_in_feed(FFStream *feed, AVCodecContext *codec, int bit_rate)
    
        int best_bitrate = 100000000;
        int best = -1;
    
        for (i = 0; i < feed->nb_streams; i++) {
    
            AVCodecContext *feed_codec = feed->streams[i]->codec;
    
    
            if (feed_codec->codec_id != codec->codec_id ||
                feed_codec->sample_rate != codec->sample_rate ||
                feed_codec->width != codec->width ||
    
                feed_codec->height != codec->height)
    
                continue;
    
            /* Potential stream */
    
    
            /* We want the fastest stream less than bit_rate, or the slowest
    
             * faster than bit_rate
             */
    
            if (feed_codec->bit_rate <= bit_rate) {
                if (best_bitrate > bit_rate || feed_codec->bit_rate > best_bitrate) {
                    best_bitrate = feed_codec->bit_rate;
                    best = i;
                }
            } else {
                if (feed_codec->bit_rate < best_bitrate) {
                    best_bitrate = feed_codec->bit_rate;
                    best = i;
                }
            }
        }
    
        return best;
    }
    
    static int modify_current_stream(HTTPContext *c, char *rates)
    {
        int i;
        FFStream *req = c->stream;
        int action_required = 0;
    
        /* Not much we can do for a feed */
        if (!req->feed)
            return 0;
    
    
        for (i = 0; i < req->nb_streams; i++) {
    
            AVCodecContext *codec = req->streams[i]->codec;
    
                    c->switch_feed_streams[i] = req->feed_streams[i];
    
                    c->switch_feed_streams[i] = find_stream_in_feed(req->feed, codec, codec->bit_rate / 2);
    
                    /* Wants off or slow */
                    c->switch_feed_streams[i] = find_stream_in_feed(req->feed, codec, codec->bit_rate / 4);
    #ifdef WANTS_OFF
                    /* This doesn't work well when it turns off the only stream! */
                    c->switch_feed_streams[i] = -2;
                    c->feed_streams[i] = -2;
    #endif
    
            if (c->switch_feed_streams[i] >= 0 && c->switch_feed_streams[i] != c->feed_streams[i])
                action_required = 1;
        }
    
        return action_required;
    }
    
    static void do_switch_stream(HTTPContext *c, int i)
    {
        if (c->switch_feed_streams[i] >= 0) {
    
            c->feed_streams[i] = c->switch_feed_streams[i];
    #endif
    
            /* Now update the stream */
    
        c->switch_feed_streams[i] = -1;
    
    /* XXX: factorize in utils.c ? */
    /* XXX: take care with different space meaning */
    static void skip_spaces(const char **pp)
    {
        const char *p;
        p = *pp;
        while (*p == ' ' || *p == '\t')
            p++;
        *pp = p;
    }
    
    static void get_word(char *buf, int buf_size, const char **pp)
    {
        const char *p;
        char *q;
    
        p = *pp;
        skip_spaces(&p);
        q = buf;
        while (!isspace(*p) && *p != '\0') {
            if ((q - buf) < buf_size - 1)
                *q++ = *p;
            p++;
        }
        if (buf_size > 0)
            *q = '\0';
        *pp = p;
    }
    
    
    static int validate_acl(FFStream *stream, HTTPContext *c)
    {
        enum IPAddressAction last_action = IP_DENY;
        IPAddressACL *acl;
        struct in_addr *src = &c->from_addr.sin_addr;
    
        unsigned long src_addr = src->s_addr;
    
    
        for (acl = stream->acl; acl; acl = acl->next) {
    
            if (src_addr >= acl->first.s_addr && src_addr <= acl->last.s_addr)
    
                return (acl->action == IP_ALLOW) ? 1 : 0;
            last_action = acl->action;
        }
    
        /* Nothing matched, so return not the last action */
        return (last_action == IP_DENY) ? 1 : 0;
    }
    
    
    /* compute the real filename of a file by matching it without its
       extensions to all the stream filenames */
    static void compute_real_filename(char *filename, int max_size)
    {
        char file1[1024];
        char file2[1024];
        char *p;
        FFStream *stream;
    
        /* compute filename by matching without the file extensions */
    
        av_strlcpy(file1, filename, sizeof(file1));
    
        p = strrchr(file1, '.');
        if (p)
            *p = '\0';
        for(stream = first_stream; stream != NULL; stream = stream->next) {
    
            av_strlcpy(file2, stream->filename, sizeof(file2));
    
            p = strrchr(file2, '.');
            if (p)
                *p = '\0';
            if (!strcmp(file1, file2)) {
    
                av_strlcpy(filename, stream->filename, max_size);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    /* parse http request and prepare header */
    static int http_parse_request(HTTPContext *c)
    {
        char *p;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        char cmd[32];
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        char url[1024], *q;
        char protocol[32];
        char msg[1024];
        const char *mime_type;
        FFStream *stream;
    
        char *useragent = 0;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
        p = c->buffer;
    
        get_word(cmd, sizeof(cmd), (const char **)&p);
    
        av_strlcpy(c->method, cmd, sizeof(c->method));
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        if (!strcmp(cmd, "GET"))
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        else if (!strcmp(cmd, "POST"))
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        else
            return -1;
    
    
        get_word(url, sizeof(url), (const char **)&p);
    
        av_strlcpy(c->url, url, sizeof(c->url));
    
        get_word(protocol, sizeof(protocol), (const char **)&p);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        if (strcmp(protocol, "HTTP/1.0") && strcmp(protocol, "HTTP/1.1"))
            return -1;
    
        av_strlcpy(c->protocol, protocol, sizeof(c->protocol));
    
            http_log("New connection: %s %s\n", cmd, url);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* find the filename and the optional info string in the request */
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        if (p) {
    
            av_strlcpy(info, p, sizeof(info));
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            *p = '\0';
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            info[0] = '\0';
    
    
        av_strlcpy(filename, url + ((*url == '/') ? 1 : 0), sizeof(filename)-1);
    
        for (p = c->buffer; *p && *p != '\r' && *p != '\n'; ) {
            if (strncasecmp(p, "User-Agent:", 11) == 0) {
                useragent = p + 11;
                if (*useragent && *useragent != '\n' && isspace(*useragent))
                    useragent++;
                break;
            }
            p = strchr(p, '\n');
            if (!p)
                break;
    
            p++;
        }
    
    
        redir_type = REDIR_NONE;
        if (match_ext(filename, "asx")) {
            redir_type = REDIR_ASX;
    
            filename[strlen(filename)-1] = 'f';
    
            (!useragent || strncasecmp(useragent, "NSPlayer", 8) != 0)) {
            /* if this isn't WMP or lookalike, return the redirector file */
    
            redir_type = REDIR_ASF;
        } else if (match_ext(filename, "rpm,ram")) {
            redir_type = REDIR_RAM;
    
            strcpy(filename + strlen(filename)-2, "m");
    
        } else if (match_ext(filename, "rtsp")) {
            redir_type = REDIR_RTSP;
    
            compute_real_filename(filename, sizeof(filename) - 1);
    
        } else if (match_ext(filename, "sdp")) {
            redir_type = REDIR_SDP;
    
            compute_real_filename(filename, sizeof(filename) - 1);
    
        // "redirect" / request to index.html
        if (!strlen(filename))
    
            av_strlcpy(filename, "index.html", sizeof(filename) - 1);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        stream = first_stream;
        while (stream != NULL) {
    
            if (!strcmp(stream->filename, filename) && validate_acl(stream, c))
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                break;
            stream = stream->next;
        }
        if (stream == NULL) {
    
            snprintf(msg, sizeof(msg), "File '%s' not found", url);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            goto send_error;
        }
    
        c->stream = stream;
        memcpy(c->feed_streams, stream->feed_streams, sizeof(c->feed_streams));
        memset(c->switch_feed_streams, -1, sizeof(c->switch_feed_streams));
    
        if (stream->stream_type == STREAM_TYPE_REDIRECT) {
            c->http_error = 301;
            q = c->buffer;
    
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "HTTP/1.0 301 Moved\r\n");
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Location: %s\r\n", stream->feed_filename);
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Content-type: text/html\r\n");
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "\r\n");
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "<html><head><title>Moved</title></head><body>\r\n");
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "You should be <a href=\"%s\">redirected</a>.\r\n", stream->feed_filename);
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "</body></html>\r\n");
    
    
            /* prepare output buffer */
            c->buffer_ptr = c->buffer;
            c->buffer_end = q;
            c->state = HTTPSTATE_SEND_HEADER;
            return 0;
        }
    
    
        /* If this is WMP, get the rate information */
        if (extract_rates(ratebuf, sizeof(ratebuf), c->buffer)) {
    
            if (modify_current_stream(c, ratebuf)) {
                for (i = 0; i < sizeof(c->feed_streams) / sizeof(c->feed_streams[0]); i++) {
                    if (c->switch_feed_streams[i] >= 0)
                        do_switch_stream(c, i);
                }
            }
    
    Diego Biurrun's avatar
    Diego Biurrun committed
        /* If already streaming this feed, do not let start another feeder. */
    
        if (stream->feed_opened) {
            snprintf(msg, sizeof(msg), "This feed is already being received.");
    
            http_log("feed %s already being received\n", stream->feed_filename);
    
        if (c->post == 0 && stream->stream_type == STREAM_TYPE_LIVE)
    
            current_bandwidth += stream->bandwidth;
    
        if (c->post == 0 && max_bandwidth < current_bandwidth) {
    
            c->http_error = 200;
            q = c->buffer;
    
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "HTTP/1.0 200 Server too busy\r\n");
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Content-type: text/html\r\n");
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "\r\n");
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "<html><head><title>Too busy</title></head><body>\r\n");
    
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "<p>The server is too busy to serve your request at this time.</p>\r\n");
    
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "<p>The bandwidth being served (including your stream) is %lldkbit/sec, and this exceeds the limit of %lldkbit/sec.</p>\r\n",
    
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "</body></html>\r\n");
    
    
            /* prepare output buffer */
            c->buffer_ptr = c->buffer;
            c->buffer_end = q;
            c->state = HTTPSTATE_SEND_HEADER;
            return 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;
    
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "HTTP/1.0 200 ASX Follows\r\n");
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Content-type: video/x-ms-asf\r\n");
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "\r\n");
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "<ASX Version=\"3\">\r\n");
    
                            //q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "<!-- Autogenerated by ffserver -->\r\n");
    
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "<ENTRY><REF HREF=\"http://%s/%s%s\"/></ENTRY>\r\n",
    
                                    hostbuf, filename, info);
    
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "</ASX>\r\n");
    
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "HTTP/1.0 200 RAM Follows\r\n");
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Content-type: audio/x-pn-realaudio\r\n");
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "\r\n");
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "# Autogenerated by ffserver\r\n");
    
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "http://%s/%s%s\r\n",
    
                                    hostbuf, filename, info);
    
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "HTTP/1.0 200 ASF Redirect follows\r\n");
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Content-type: video/x-ms-asf\r\n");
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "\r\n");
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "[Reference]\r\n");
    
                            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Ref1=http://%s/%s%s\r\n",
    
                                    hostbuf, filename, info);
    
                            break;
                        case REDIR_RTSP:
                            {
                                char hostname[256], *p;
                                /* extract only hostname */
    
                                av_strlcpy(hostname, hostbuf, sizeof(hostname));
    
                                q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "HTTP/1.0 200 RTSP Redirect follows\r\n");
    
                                q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Content-type: application/x-rtsp\r\n");
                                q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "\r\n");
    
                                q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "rtsp://%s:%d/%s\r\n",
                                             hostname, ntohs(my_rtsp_addr.sin_port),
    
                                uint8_t *sdp_data;
    
                                q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "HTTP/1.0 200 OK\r\n");
                                q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Content-type: application/sdp\r\n");
                                q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "\r\n");
    
    
                                len = sizeof(my_addr);
                                getsockname(c->fd, (struct sockaddr *)&my_addr, &len);
    
                                sdp_data_size = prepare_sdp_description(stream,
                                                                        &sdp_data,
    
                                                                        my_addr.sin_addr);
                                if (sdp_data_size > 0) {
                                    memcpy(q, sdp_data, sdp_data_size);
                                    q += sdp_data_size;
                                    *q = '\0';
                                    av_free(sdp_data);
                                }
                            }
                            break;
                        default:
    
    
                        /* prepare output buffer */
                        c->buffer_ptr = c->buffer;
                        c->buffer_end = q;
                        c->state = HTTPSTATE_SEND_HEADER;
                        return 0;
                    }
                }
            }
    
    
            snprintf(msg, sizeof(msg), "ASX/RAM file not handled");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* XXX: add there authenticate and IP match */
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            /* 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;
                    }
    
                    if (strncasecmp(p, "Pragma: client-id=", 18) == 0)
    
                    p = strchr(p, '\n');
                    if (!p)
                        break;
    
                    p++;
                }
    
                if (logline) {
                    char *eol = strchr(logline, '\n');
    
                    logline += 17;
    
                    if (eol) {
                        if (eol[-1] == '\r')
                            eol--;
    
    Falk Hüffner's avatar
    Falk Hüffner committed
                        http_log("%.*s\n", (int) (eol - logline), logline);
    
    #ifdef DEBUG_WMP
                http_log("\nGot request:\n%s\n", c->buffer);
    
    #endif
    
                if (client_id && extract_rates(ratebuf, sizeof(ratebuf), c->buffer)) {
                    HTTPContext *wmpc;
    
                    /* Now we have to find the client_id */
                    for (wmpc = first_http_ctx; wmpc; wmpc = wmpc->next) {
                        if (wmpc->wmp_client_id == client_id)
                            break;
                    }
    
    
                    if (wmpc && modify_current_stream(wmpc, ratebuf))
                        wmpc->switch_pending = 1;
    
                snprintf(msg, sizeof(msg), "POST command not handled");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                goto send_error;
            }
            if (http_start_receive_data(c) < 0) {
    
                snprintf(msg, sizeof(msg), "could not open feed");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                goto send_error;
            }
            c->http_error = 0;
            c->state = HTTPSTATE_RECEIVE_DATA;
            return 0;
        }
    
    
    #ifdef DEBUG_WMP
    
        if (strcmp(stream->filename + strlen(stream->filename) - 4, ".asf") == 0)
    
            http_log("\nGot request:\n%s\n", c->buffer);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        if (c->stream->stream_type == STREAM_TYPE_STATUS)
    
            goto send_status;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
        /* open input stream */
        if (open_input_stream(c, info) < 0) {
    
            snprintf(msg, sizeof(msg), "Input stream corresponding to '%s' not found", url);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            goto send_error;
        }
    
        /* prepare http header */
        q = c->buffer;
    
        q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "HTTP/1.0 200 OK\r\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        mime_type = c->stream->fmt->mime_type;
        if (!mime_type)
    
            mime_type = "application/x-octet-stream";
    
        q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Pragma: no-cache\r\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
        /* for asf, we need extra headers */
    
        if (!strcmp(c->stream->fmt->name,"asf_stream")) {
    
            c->wmp_client_id = av_random(&random_state) & 0x7fffffff;
    
            q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Server: Cougar 4.1.0.3923\r\nCache-Control: no-cache\r\nPragma: client-id=%d\r\nPragma: features=\"broadcast\"\r\n", c->wmp_client_id);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        }
    
        q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Content-Type: %s\r\n", mime_type);
        q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "\r\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* 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 += snprintf(q, q - (char *) c->buffer + c->buffer_size, "HTTP/1.0 404 Not Found\r\n");
        q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "Content-type: %s\r\n", "text/html");
        q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "\r\n");
        q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "<HTML>\n");
        q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "<HEAD><TITLE>404 Not Found</TITLE></HEAD>\n");
        q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "<BODY>%s</BODY>\n", msg);
        q += snprintf(q, q - (char *) c->buffer + c->buffer_size, "</HTML>\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
        /* prepare output buffer */
        c->buffer_ptr = c->buffer;
        c->buffer_end = q;
        c->state = HTTPSTATE_SEND_HEADER;
        return 0;
    
     send_status:
        compute_status(c);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        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 fmt_bytecount(ByteIOContext *pb, int64_t count)
    
    {
        static const char *suffix = " kMGTP";
        const char *s;
    
    
        for (s = suffix; count >= 100000 && s[1]; count /= 1000, s++);
    
        url_fprintf(pb, "%"PRId64"%c", count, *s);
    
    static void compute_status(HTTPContext *c)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    {
        HTTPContext *c1;
        FFStream *stream;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        time_t ti;
    
        if (url_open_dyn_buf(&pb) < 0) {
    
            c->buffer_ptr = c->buffer;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
        url_fprintf(pb, "HTTP/1.0 200 OK\r\n");
        url_fprintf(pb, "Content-type: %s\r\n", "text/html");
        url_fprintf(pb, "Pragma: no-cache\r\n");
        url_fprintf(pb, "\r\n");
    
        url_fprintf(pb, "<HEAD><TITLE>%s Status</TITLE>\n", program_name);
    
            url_fprintf(pb, "<link rel=\"shortcut icon\" href=\"%s\">\n", c->stream->feed_filename);
        url_fprintf(pb, "</HEAD>\n<BODY>");
    
        url_fprintf(pb, "<H1>%s Status</H1>\n", program_name);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* format status */
    
        url_fprintf(pb, "<H2>Available Streams</H2>\n");
        url_fprintf(pb, "<TABLE cellspacing=0 cellpadding=4>\n");
        url_fprintf(pb, "<TR><Th valign=top>Path<th align=left>Served<br>Conns<Th><br>bytes<Th valign=top>Format<Th>Bit rate<br>kbits/s<Th align=left>Video<br>kbits/s<th><br>Codec<Th align=left>Audio<br>kbits/s<th><br>Codec<Th align=left valign=top>Feed\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        stream = first_stream;
        while (stream != NULL) {
    
            char sfilename[1024];
            char *eosf;
    
    
                av_strlcpy(sfilename, stream->filename, sizeof(sfilename) - 10);
    
                eosf = sfilename + strlen(sfilename);
                if (eosf - sfilename >= 4) {
    
                    if (strcmp(eosf - 4, ".asf") == 0)
    
                    else if (strcmp(eosf - 3, ".rm") == 0)
    
                    else if (stream->fmt && !strcmp(stream->fmt->name, "rtp")) {
    
                        /* generate a sample RTSP director if
                           unicast. Generate an SDP redirector if
                           multicast */
    
                        eosf = strrchr(sfilename, '.');
                        if (!eosf)
                            eosf = sfilename + strlen(sfilename);
    
                        if (stream->is_multicast)
                            strcpy(eosf, ".sdp");
                        else
                            strcpy(eosf, ".rtsp");
    
    
                url_fprintf(pb, "<TR><TD><A HREF=\"/%s\">%s</A> ",
    
                url_fprintf(pb, "<td align=right> %d <td align=right> ",
    
                case STREAM_TYPE_LIVE: {
    
                        int audio_bit_rate = 0;
                        int video_bit_rate = 0;
    
    Zdenek Kabelac's avatar
    Zdenek Kabelac committed
                        const char *audio_codec_name = "";
                        const char *video_codec_name = "";
                        const char *audio_codec_name_extra = "";
                        const char *video_codec_name_extra = "";
    
    
                        for(i=0;i<stream->nb_streams;i++) {
                            AVStream *st = stream->streams[i];
    
                            AVCodec *codec = avcodec_find_encoder(st->codec->codec_id);
                            switch(st->codec->codec_type) {
    
                                audio_bit_rate += st->codec->bit_rate;
    
                                if (codec) {
                                    if (*audio_codec_name)
                                        audio_codec_name_extra = "...";
                                    audio_codec_name = codec->name;
                                }
                                break;
                            case CODEC_TYPE_VIDEO:
    
                                video_bit_rate += st->codec->bit_rate;
    
                                if (codec) {
                                    if (*video_codec_name)
                                        video_codec_name_extra = "...";
                                    video_codec_name = codec->name;
                                }
                                break;
    
                                video_bit_rate += st->codec->bit_rate;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                        }
    
                        url_fprintf(pb, "<TD align=center> %s <TD align=right> %d <TD align=right> %d <TD> %s %s <TD align=right> %d <TD> %s %s",
    
                                     video_bit_rate / 1000, video_codec_name, video_codec_name_extra,
                                     audio_bit_rate / 1000, audio_codec_name, audio_codec_name_extra);
    
                            url_fprintf(pb, "<TD>%s", stream->feed->filename);
    
                            url_fprintf(pb, "<TD>%s", stream->feed_filename);
                        url_fprintf(pb, "\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    }
    
                    url_fprintf(pb, "<TD align=center> - <TD align=right> - <TD align=right> - <td><td align=right> - <TD>\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                }
            }
            stream = stream->next;
        }
    
    
        stream = first_stream;
        while (stream != NULL) {
            if (stream->feed == stream) {
    
                url_fprintf(pb, "<h2>Feed %s</h2>", stream->filename);
    
                if (stream->pid) {
    
                    url_fprintf(pb, "Running as pid %d.\n", stream->pid);
    
    #if defined(linux) && !defined(CONFIG_NOCUTILS)
                    {
                        FILE *pid_stat;
                        char ps_cmd[64];
    
                        /* This is somewhat linux specific I guess */
    
                        snprintf(ps_cmd, sizeof(ps_cmd),
                                 "ps -o \"%%cpu,cputime\" --no-headers %d",
    
                        pid_stat = popen(ps_cmd, "r");
                        if (pid_stat) {
                            char cpuperc[10];
                            char cpuused[64];
    
    
                            if (fscanf(pid_stat, "%10s %64s", cpuperc,
    
                                       cpuused) == 2) {
                                url_fprintf(pb, "Currently using %s%% of the cpu. Total time used %s.\n",
                                             cpuperc, cpuused);
                            }
                            fclose(pid_stat);
    
                url_fprintf(pb, "<table cellspacing=0 cellpadding=4><tr><th>Stream<th>type<th>kbits/s<th align=left>codec<th align=left>Parameters\n");
    
    
                for (i = 0; i < stream->nb_streams; i++) {
                    AVStream *st = stream->streams[i];
    
                    AVCodec *codec = avcodec_find_encoder(st->codec->codec_id);
    
                    const char *type = "unknown";
    
                        snprintf(parameters, sizeof(parameters), "%d channel(s), %d Hz", st->codec->channels, st->codec->sample_rate);
    
                        snprintf(parameters, sizeof(parameters), "%dx%d, q=%d-%d, fps=%d", st->codec->width, st->codec->height,
                                    st->codec->qmin, st->codec->qmax, st->codec->time_base.den / st->codec->time_base.num);
    
                    url_fprintf(pb, "<tr><td align=right>%d<td>%s<td align=right>%d<td>%s<td>%s\n",
    
                            i, type, st->codec->bit_rate/1000, codec ? codec->name : "", parameters);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    #if 0
        {
            float avg;
            AVCodecContext *enc;
            char buf[1024];
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            /* feed status */
            stream = first_feed;
            while (stream != NULL) {
    
                url_fprintf(pb, "<H1>Feed '%s'</H1>\n", stream->filename);
                url_fprintf(pb, "<TABLE>\n");
                url_fprintf(pb, "<TR><TD>Parameters<TD>Frame count<TD>Size<TD>Avg bitrate (kbits/s)\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                for(i=0;i<stream->nb_streams;i++) {
                    AVStream *st = stream->streams[i];
                    FeedData *fdata = st->priv_data;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    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;
    
                    url_fprintf(pb, "<TR><TD>%s <TD> %d <TD> %"PRId64" <TD> %0.1f\n",
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                                 buf, enc->frame_number, fdata->data_count, avg / 1000.0);
                }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                stream = stream->next_feed;
            }
        }
    #endif
    
        /* connection status */
    
        url_fprintf(pb, "<H2>Connection Status</H2>\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
        url_fprintf(pb, "Number of connections: %d / %d<BR>\n",
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                     nb_connections, nb_max_connections);
    
    
        url_fprintf(pb, "Bandwidth in use: %lldk / %lldk<BR>\n",
    
        url_fprintf(pb, "<TABLE>\n");
        url_fprintf(pb, "<TR><th>#<th>File<th>IP<th>Proto<th>State<th>Target bits/sec<th>Actual bits/sec<th>Bytes transferred\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        c1 = first_http_ctx;
        i = 0;
    
            int bitrate;
            int j;
    
            bitrate = 0;
    
            if (c1->stream) {
                for (j = 0; j < c1->stream->nb_streams; j++) {
    
                    if (!c1->stream->feed)
    
                        bitrate += c1->stream->streams[j]->codec->bit_rate;
    
                    else if (c1->feed_streams[j] >= 0)
                        bitrate += c1->stream->feed->streams[c1->feed_streams[j]]->codec->bit_rate;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            i++;
            p = inet_ntoa(c1->from_addr.sin_addr);
    
            url_fprintf(pb, "<TR><TD><B>%d</B><TD>%s%s<TD>%s<TD>%s<TD>%s<td align=right>",
                        i,
                        c1->stream ? c1->stream->filename : "",
    
                        c1->state == HTTPSTATE_RECEIVE_DATA ? "(input)" : "",
    
                        c1->protocol,
                        http_state[c1->state]);
            fmt_bytecount(pb, bitrate);
            url_fprintf(pb, "<td align=right>");
            fmt_bytecount(pb, compute_datarate(&c1->datarate, c1->data_count) * 8);
            url_fprintf(pb, "<td align=right>");
            fmt_bytecount(pb, c1->data_count);
            url_fprintf(pb, "\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            c1 = c1->next;
        }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* date */
        ti = time(NULL);
        p = ctime(&ti);
    
        url_fprintf(pb, "<HR size=1 noshade>Generated at %s", p);
        url_fprintf(pb, "</BODY>\n</HTML>\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
        len = url_close_dyn_buf(pb, &c->pb_buffer);
        c->buffer_ptr = c->pb_buffer;
        c->buffer_end = c->pb_buffer + len;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    }
    
    
    /* check if the parser needs to be opened for stream i */
    static void open_parser(AVFormatContext *s, int i)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    {
    
        if (!st->codec->codec) {
            codec = avcodec_find_decoder(st->codec->codec_id);
    
            if (codec && (codec->capabilities & CODEC_CAP_PARSE_ONLY)) {
    
                if (avcodec_open(st->codec, codec) < 0)
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    }
    
    static int open_input_stream(HTTPContext *c, const char *info)
    {
        char buf[128];
        char input_filename[1024];
        AVFormatContext *s;
    
    Baptiste Coudurier's avatar
    Baptiste Coudurier committed
        int buf_size, i, ret;
    
        int64_t stream_pos;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
        /* 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)) {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                stream_pos = parse_date(buf, 0);
    
                if (stream_pos == INT64_MIN)
                    return -1;
    
            } else if (find_info_tag(buf, sizeof(buf), "buffer", info)) {
    
                stream_pos = av_gettime() - prebuffer * (int64_t)1000000;
    
                stream_pos = av_gettime() - c->stream->prebuffer * (int64_t)1000;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        } else {
            strcpy(input_filename, c->stream->feed_filename);
            buf_size = 0;
            /* compute position (relative time) */
    
            if (find_info_tag(buf, sizeof(buf), "date", info)) {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                stream_pos = parse_date(buf, 1);
    
                if (stream_pos == INT64_MIN)
                    return -1;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                stream_pos = 0;
        }
        if (input_filename[0] == '\0')
            return -1;
    
    
    #if 0
        { time_t when = stream_pos / 1000000;
    
        http_log("Stream pos = %"PRId64", time=%s", stream_pos, ctime(&when));
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* open stream */
    
    Baptiste Coudurier's avatar
    Baptiste Coudurier committed
        if ((ret = av_open_input_file(&s, input_filename, c->stream->ifmt,
                                      buf_size, c->stream->ap_in)) < 0) {
            http_log("could not open %s: %d\n", input_filename, ret);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            return -1;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        c->fmt_in = s;
    
        av_find_stream_info(c->fmt_in);
    
        /* open each parser */
        for(i=0;i<s->nb_streams;i++)
            open_parser(s, i);
    
        /* choose stream as clock source (we favorize video stream if
           present) for packet sending */
        c->pts_stream_index = 0;
        for(i=0;i<c->stream->nb_streams;i++) {
    
            if (c->pts_stream_index == 0 &&
    
                c->stream->streams[i]->codec->codec_type == CODEC_TYPE_VIDEO) {
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    
    
        if (c->fmt_in->iformat->read_seek)
    
            av_seek_frame(c->fmt_in, -1, stream_pos, 0);
    
        /* set the start time (needed for maxtime and RTP packet timing) */
        c->start_time = cur_time;
        c->first_pts = AV_NOPTS_VALUE;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        return 0;
    }