Skip to content
Snippets Groups Projects
ffserver.c 83 KiB
Newer Older
  • Learn to ignore specific revisions
  • 
        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;
    }
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
    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, *redirect;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        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;
    
        redirect = NULL;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        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;
                }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            } 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, "Launch")) {
                if (feed) {
                    int i;
    
                    feed->child_argv = (char **) av_mallocz(64 * sizeof(char *));
    
                    feed->child_argv[0] = av_malloc(7);
                    strcpy(feed->child_argv[0], "ffmpeg");
    
                    for (i = 1; i < 62; i++) {
                        char argbuf[256];
    
                        get_arg(argbuf, sizeof(argbuf), &p);
                        if (!argbuf[0])
                            break;
    
                        feed->child_argv[i] = av_malloc(strlen(argbuf + 1));
                        strcpy(feed->child_argv[i], argbuf);
                    }
    
                    feed->child_argv[i] = av_malloc(30 + strlen(feed->filename));
    
                    snprintf(feed->child_argv[i], 256, "http://127.0.0.1:%d/%s", 
                        ntohs(my_addr.sin_port), feed->filename);
                }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            } 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++;
                    }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                }
                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;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    /* jpeg cannot be used here, so use single frame jpeg */
                    if (!strcmp(arg, "jpeg"))
                        strcpy(arg, "singlejpeg");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
                    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, "FaviconURL")) {
                if (stream && stream->stream_type == STREAM_TYPE_STATUS) {
                    get_arg(stream->feed_filename, sizeof(stream->feed_filename), &p);
                } else {
                    fprintf(stderr, "%s:%d: FaviconURL only permitted for status streams\n", 
                                filename, line_num);
                    errors++;
                }
    
            } else if (!strcasecmp(cmd, "Author")) {
                if (stream) {
                    get_arg(stream->author, sizeof(stream->author), &p);
                }
            } else if (!strcasecmp(cmd, "Comment")) {
                if (stream) {
                    get_arg(stream->comment, sizeof(stream->comment), &p);
                }
            } else if (!strcasecmp(cmd, "Copyright")) {
                if (stream) {
                    get_arg(stream->copyright, sizeof(stream->copyright), &p);
                }
            } else if (!strcasecmp(cmd, "Title")) {
                if (stream) {
                    get_arg(stream->title, sizeof(stream->title), &p);
                }
    
            } else if (!strcasecmp(cmd, "Preroll")) {
                get_arg(arg, sizeof(arg), &p);
                if (stream) {
                    stream->prebuffer = atoi(arg) * 1000;
                }
    
            } else if (!strcasecmp(cmd, "StartSendOnKey")) {
                if (stream) {
                    stream->send_on_key = 1;
                }
    
            } 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, "MaxTime")) {
                get_arg(arg, sizeof(arg), &p);
                if (stream) {
    
                    stream->max_time = atoi(arg) * 1000;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            } 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++;
                    }
                }
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            } 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 if (!strcasecmp(cmd, "<Redirect")) {
                /*********************************************/
                char *q;
                if (stream || feed || redirect) {
                    fprintf(stderr, "%s:%d: Already in a tag\n",
                            filename, line_num);
                    errors++;
                } else {
                    redirect = av_mallocz(sizeof(FFStream));
                    *last_stream = redirect;
                    last_stream = &redirect->next;
    
                    get_arg(redirect->filename, sizeof(redirect->filename), &p);
                    q = strrchr(redirect->filename, '>');
                    if (*q)
                        *q = '\0';
                    redirect->stream_type = STREAM_TYPE_REDIRECT;
                }
            } else if (!strcasecmp(cmd, "URL")) {
                if (redirect) {
                    get_arg(redirect->feed_filename, sizeof(redirect->feed_filename), &p);
                }
            } else if (!strcasecmp(cmd, "</Redirect>")) {
                if (!redirect) {
                    fprintf(stderr, "%s:%d: No corresponding <Redirect> for </Redirect>\n",
                            filename, line_num);
                    errors++;
                }
                if (!redirect->feed_filename[0]) {
                    fprintf(stderr, "%s:%d: No URL found for <Redirect>\n",
                            filename, line_num);
                    errors++;
                }
                redirect = NULL;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            } 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, 2002 Fabrice Bellard\n"
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
               "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, 2002 Fabrice Bellard\n"
        "This library is free software; you can redistribute it and/or\n"
        "modify it under the terms of the GNU Lesser General Public\n"
        "License as published by the Free Software Foundation; either\n"
        "version 2 of the License, or (at your option) any later version.\n"
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        "\n"
    
        "This library is distributed in the hope that it will be useful,\n"
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
    
        "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\n"
        "Lesser General Public License for more details.\n"
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        "\n"
    
        "You should have received a copy of the GNU Lesser General Public\n"
        "License along with this library; if not, write to the Free Software\n"
        "Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA\n"
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        );
    }
    
    int main(int argc, char **argv)
    {
        const char *config_filename;
        int c;
    
        register_all();
    
        config_filename = "/etc/ffserver.conf";
    
    
        my_program_name = argv[0];
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        for(;;) {
    
            c = getopt_long_only(argc, argv, "ndLh?f:", NULL, NULL);
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            if (c == -1)
                break;
            switch(c) {
            case 'L':
                licence();
                exit(1);
            case '?':
            case 'h':
                help();
                exit(1);
    
            case 'n':
                no_launch = 1;
                break;
            case 'd':
                ffserver_debug = 1;
                break;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            case 'f':
                config_filename = optarg;
                break;
            default:
                exit(2);
            }
        }
    
    
        putenv("http_proxy");               /* Kill the http_proxy */
    
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        /* 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;
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
        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 not start http server\n");
    
    Fabrice Bellard's avatar
    Fabrice Bellard committed
            exit(1);
        }
    
        return 0;
    }