gopher.c 3.33 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
/*
 * 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"
27
#include "internal.h"
28
#include "network.h"
29
#include "url.h"
30

31
typedef struct GopherContext {
32 33 34
    URLContext *hd;
} GopherContext;

35
static int gopher_write(URLContext *h, const uint8_t *buf, int size)
36 37
{
    GopherContext *s = h->priv_data;
38
    return ffurl_write(s->hd, buf, size);
39 40 41 42 43 44 45 46 47 48 49 50 51 52
}

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:
53
            av_log(h, AV_LOG_WARNING,
54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
                   "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) {
72
        ffurl_close(s->hd);
73 74 75 76 77 78 79
        s->hd = NULL;
    }
    return 0;
}

static int gopher_open(URLContext *h, const char *uri, int flags)
{
80
    GopherContext *s = h->priv_data;
81 82 83 84 85 86
    char hostname[1024], auth[1024], path[1024], buf[1024];
    int port, err;

    h->is_streamed = 1;

    /* needed in any case to build the host string */
87
    av_url_split(NULL, 0, auth, sizeof(auth), hostname, sizeof(hostname), &port,
Martin Storsjö's avatar
Martin Storsjö committed
88
                 path, sizeof(path), uri);
89 90 91 92

    if (port < 0)
        port = 70;

93
    ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port, NULL);
94 95

    s->hd = NULL;
96
    err = ffurl_open_whitelist(&s->hd, buf, AVIO_FLAG_READ_WRITE,
97
                               &h->interrupt_callback, NULL, h->protocol_whitelist, h->protocol_blacklist, h);
98 99 100 101 102 103 104 105 106 107 108 109 110 111
    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;
112
    int len = ffurl_read(s->hd, buf, size);
113 114 115 116
    return len;
}


117
const URLProtocol ff_gopher_protocol = {
118 119 120 121 122
    .name           = "gopher",
    .url_open       = gopher_open,
    .url_read       = gopher_read,
    .url_write      = gopher_write,
    .url_close      = gopher_close,
123
    .priv_data_size = sizeof(GopherContext),
124
    .flags          = URL_PROTOCOL_FLAG_NETWORK,
125
};