Reindent
[ffmpeg.git] / libavformat / gopher.c
1 /*
2  * Gopher protocol
3  *
4  * Copyright (c) 2009 Toshimitsu Kimura
5  *
6  * based on libavformat/http.c, Copyright (c) 2000, 2001 Fabrice Bellard
7  *
8  * This file is part of FFmpeg.
9  *
10  * FFmpeg is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU Lesser General Public
12  * License as published by the Free Software Foundation; either
13  * version 2.1 of the License, or (at your option) any later version.
14  *
15  * FFmpeg is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18  * Lesser General Public License for more details.
19  *
20  * You should have received a copy of the GNU Lesser General Public
21  * License along with FFmpeg; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23  */
24
25 #include "libavutil/avstring.h"
26 #include "avformat.h"
27 #include "network.h"
28
29 typedef struct {
30     URLContext *hd;
31 } GopherContext;
32
33 static int gopher_write(URLContext *h, uint8_t *buf, int size)
34 {
35     GopherContext *s = h->priv_data;
36     return url_write(s->hd, buf, size);
37 }
38
39 static int gopher_connect(URLContext *h, const char *path)
40 {
41     char buffer[1024];
42
43     if (!*path) return AVERROR(EINVAL);
44     switch (*++path) {
45         case '5':
46         case '9':
47             path = strchr(path, '/');
48             if (!path) return AVERROR(EINVAL);
49             break;
50         default:
51             av_log(NULL, AV_LOG_WARNING,
52                    "Gopher protocol type '%c' not supported yet!\n",
53                    *path);
54             return AVERROR(EINVAL);
55     }
56
57     /* send gopher sector */
58     snprintf(buffer, sizeof(buffer), "%s\r\n", path);
59
60     if (gopher_write(h, buffer, strlen(buffer)) < 0)
61         return AVERROR(EIO);
62
63     return 0;
64 }
65
66 static int gopher_close(URLContext *h)
67 {
68     GopherContext *s = h->priv_data;
69     if (s->hd) {
70         url_close(s->hd);
71         s->hd = NULL;
72     }
73     av_freep(&h->priv_data);
74     return 0;
75 }
76
77 static int gopher_open(URLContext *h, const char *uri, int flags)
78 {
79     GopherContext *s;
80     char hostname[1024], auth[1024], path[1024], buf[1024];
81     int port, err;
82
83     h->is_streamed = 1;
84
85     s = av_malloc(sizeof(GopherContext));
86     if (!s) {
87         return AVERROR(ENOMEM);
88     }
89     h->priv_data = s;
90
91     /* needed in any case to build the host string */
92     ff_url_split(NULL, 0, auth, sizeof(auth), hostname, sizeof(hostname), &port,
93                  path, sizeof(path), uri);
94
95     if (port < 0)
96         port = 70;
97
98     ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port, NULL);
99
100     s->hd = NULL;
101     err = url_open(&s->hd, buf, URL_RDWR);
102     if (err < 0)
103         goto fail;
104
105     if ((err = gopher_connect(h, path)) < 0)
106         goto fail;
107     return 0;
108  fail:
109     gopher_close(h);
110     return err;
111 }
112
113 static int gopher_read(URLContext *h, uint8_t *buf, int size)
114 {
115     GopherContext *s = h->priv_data;
116     int len = url_read(s->hd, buf, size);
117     return len;
118 }
119
120
121 URLProtocol gopher_protocol = {
122     "gopher",
123     gopher_open,
124     gopher_read,
125     gopher_write,
126     NULL, /*seek*/
127     gopher_close,
128 };