129 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			129 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * 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,
 | 
						|
};
 |