Skip to content
Snippets Groups Projects
gopher.c 3.08 KiB
Newer Older
  • Learn to ignore specific revisions
  • /*
     * Gopher protocol
     *
     * Copyright (c) 2009 Toshimitsu Kimura
     *
     * based on libavformat/http.c, Copyright (c) 2000, 2001 Fabrice Bellard
     *
     * This file is part of FFmpeg.
     *
     * FFmpeg is free software; you can redistribute it and/or
     * modify it under the terms of the GNU Lesser General Public
     * License as published by the Free Software Foundation; either
     * version 2.1 of the License, or (at your option) any later version.
     *
     * FFmpeg 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
     * Lesser General Public License for more details.
     *
     * You should have received a copy of the GNU Lesser General Public
     * License along with FFmpeg; if not, write to the Free Software
     * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
     */
    
    #include "libavutil/avstring.h"
    #include "avformat.h"
    #include "network.h"
    
    typedef struct {
        URLContext *hd;
    } GopherContext;
    
    static int gopher_write(URLContext *h, uint8_t *buf, int size)
    {
        GopherContext *s = h->priv_data;
        return url_write(s->hd, buf, size);
    }
    
    static int gopher_connect(URLContext *h, const char *path)
    {
        char buffer[1024];
    
        if (!*path) return AVERROR(EINVAL);
        switch (*++path) {
            case '5':
            case '9':
                path = strchr(path, '/');
                if (!path) return AVERROR(EINVAL);
                break;
            default:
                av_log(NULL, AV_LOG_WARNING,
                       "Gopher protocol type '%c' not supported yet!\n",
                       *path);
                return AVERROR(EINVAL);
        }
    
        /* send gopher sector */
        snprintf(buffer, sizeof(buffer), "%s\r\n", path);
    
        if (gopher_write(h, buffer, strlen(buffer)) < 0)
            return AVERROR(EIO);
    
        return 0;
    }
    
    static int gopher_close(URLContext *h)
    {
        GopherContext *s = h->priv_data;
        if (s->hd) {
            url_close(s->hd);
            s->hd = NULL;
        }
        av_freep(&h->priv_data);
        return 0;
    }
    
    static int gopher_open(URLContext *h, const char *uri, int flags)
    {
        GopherContext *s;
        char hostname[1024], auth[1024], path[1024], buf[1024];
        int port, err;
    
        h->is_streamed = 1;
    
        s = av_malloc(sizeof(GopherContext));
        if (!s) {
            return AVERROR(ENOMEM);
        }
        h->priv_data = s;
    
        /* needed in any case to build the host string */
        url_split(NULL, 0, auth, sizeof(auth), hostname, sizeof(hostname), &port,
                  path, sizeof(path), uri);
    
        if (port < 0)
            port = 70;
    
        snprintf(buf, sizeof(buf), "tcp://%s:%d", hostname, port);
    
        s->hd = NULL;
        err = url_open(&s->hd, buf, URL_RDWR);
        if (err < 0)
            goto fail;
    
        if ((err = gopher_connect(h, path)) < 0)
            goto fail;
        return 0;
     fail:
        gopher_close(h);
        return err;
    }
    
    static int gopher_read(URLContext *h, uint8_t *buf, int size)
    {
        GopherContext *s = h->priv_data;
        int len = url_read(s->hd, buf, size);
        return len;
    }
    
    
    URLProtocol gopher_protocol = {
        "gopher",
        gopher_open,
        gopher_read,
        gopher_write,
        NULL, /*seek*/
        gopher_close,
    };