Implement url_open_protocol(), which is basiclly the former url_open()
[libav.git] / libavformat / avio.c
CommitLineData
de6d9b64
FB
1/*
2 * Unbuffered io for ffmpeg system
19720f15 3 * Copyright (c) 2001 Fabrice Bellard
de6d9b64 4 *
b78e7197
DB
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
19720f15
FB
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
b78e7197 10 * version 2.1 of the License, or (at your option) any later version.
de6d9b64 11 *
b78e7197 12 * FFmpeg is distributed in the hope that it will be useful,
de6d9b64 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19720f15
FB
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
de6d9b64 16 *
19720f15 17 * You should have received a copy of the GNU Lesser General Public
b78e7197 18 * License along with FFmpeg; if not, write to the Free Software
5509bffa 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
de6d9b64 20 */
245976da
DB
21
22#include "libavutil/avstring.h"
23#include "libavcodec/opt.h"
de6d9b64 24#include "avformat.h"
5acef35f
BA
25
26#if LIBAVFORMAT_VERSION_MAJOR >= 53
27/** @name Logging context. */
28/*@{*/
29static const char *urlcontext_to_name(void *ptr)
30{
31 URLContext *h = (URLContext *)ptr;
32 if(h->prot) return h->prot->name;
33 else return "NULL";
34}
35static const AVOption options[] = {{NULL}};
36static const AVClass urlcontext_class =
37 { "URLContext", urlcontext_to_name, options };
38/*@}*/
39#endif
de6d9b64 40
019ac05a
FB
41static int default_interrupt_cb(void);
42
de6d9b64 43URLProtocol *first_protocol = NULL;
019ac05a 44URLInterruptCB *url_interrupt_cb = default_interrupt_cb;
de6d9b64 45
84be6e72
MN
46URLProtocol *av_protocol_next(URLProtocol *p)
47{
48 if(p) return p->next;
49 else return first_protocol;
50}
51
de6d9b64
FB
52int register_protocol(URLProtocol *protocol)
53{
54 URLProtocol **p;
55 p = &first_protocol;
56 while (*p != NULL) p = &(*p)->next;
57 *p = protocol;
58 protocol->next = NULL;
59 return 0;
60}
61
ba99cfc2
RB
62int url_open_protocol (URLContext **puc, struct URLProtocol *up,
63 const char *filename, int flags)
de6d9b64
FB
64{
65 URLContext *uc;
de6d9b64
FB
66 int err;
67
62181004 68 uc = av_malloc(sizeof(URLContext) + strlen(filename) + 1);
8a9488b5 69 if (!uc) {
8fa36ae0 70 err = AVERROR(ENOMEM);
8a9488b5
FB
71 goto fail;
72 }
5acef35f
BA
73#if LIBAVFORMAT_VERSION_MAJOR >= 53
74 uc->av_class = &urlcontext_class;
75#endif
62181004 76 uc->filename = (char *) &uc[1];
f746a046 77 strcpy(uc->filename, filename);
de6d9b64
FB
78 uc->prot = up;
79 uc->flags = flags;
80 uc->is_streamed = 0; /* default = not streamed */
8a9488b5 81 uc->max_packet_size = 0; /* default: stream file */
de6d9b64
FB
82 err = up->url_open(uc, filename, flags);
83 if (err < 0) {
1ea4f593 84 av_free(uc);
de6d9b64
FB
85 *puc = NULL;
86 return err;
87 }
abbae514
MN
88
89 //We must be carefull here as url_seek() could be slow, for example for http
90 if( (flags & (URL_WRONLY | URL_RDWR))
ba99cfc2 91 || !strcmp(up->name, "file"))
abbae514
MN
92 if(!uc->is_streamed && url_seek(uc, 0, SEEK_SET) < 0)
93 uc->is_streamed= 1;
de6d9b64
FB
94 *puc = uc;
95 return 0;
8a9488b5
FB
96 fail:
97 *puc = NULL;
98 return err;
de6d9b64
FB
99}
100
ba99cfc2
RB
101int url_open(URLContext **puc, const char *filename, int flags)
102{
103 URLProtocol *up;
104 const char *p;
105 char proto_str[128], *q;
106
107 p = filename;
108 q = proto_str;
109 while (*p != '\0' && *p != ':') {
110 /* protocols can only contain alphabetic chars */
111 if (!isalpha(*p))
112 goto file_proto;
113 if ((q - proto_str) < sizeof(proto_str) - 1)
114 *q++ = *p;
115 p++;
116 }
117 /* if the protocol has length 1, we consider it is a dos drive */
118 if (*p == '\0' || (q - proto_str) <= 1) {
119 file_proto:
120 strcpy(proto_str, "file");
121 } else {
122 *q = '\0';
123 }
124
125 up = first_protocol;
126 while (up != NULL) {
127 if (!strcmp(proto_str, up->name))
128 return url_open_protocol (puc, up, filename, flags);
129 up = up->next;
130 }
131 *puc = NULL;
132 return AVERROR(ENOENT);
133}
134
de6d9b64
FB
135int url_read(URLContext *h, unsigned char *buf, int size)
136{
137 int ret;
138 if (h->flags & URL_WRONLY)
6f3e0b21 139 return AVERROR(EIO);
de6d9b64
FB
140 ret = h->prot->url_read(h, buf, size);
141 return ret;
142}
143
144int url_write(URLContext *h, unsigned char *buf, int size)
145{
146 int ret;
8a9488b5 147 if (!(h->flags & (URL_WRONLY | URL_RDWR)))
6f3e0b21 148 return AVERROR(EIO);
8a9488b5
FB
149 /* avoid sending too big packets */
150 if (h->max_packet_size && size > h->max_packet_size)
6f3e0b21 151 return AVERROR(EIO);
de6d9b64
FB
152 ret = h->prot->url_write(h, buf, size);
153 return ret;
154}
155
156offset_t url_seek(URLContext *h, offset_t pos, int whence)
157{
158 offset_t ret;
159
160 if (!h->prot->url_seek)
8fa36ae0 161 return AVERROR(EPIPE);
de6d9b64
FB
162 ret = h->prot->url_seek(h, pos, whence);
163 return ret;
164}
165
de6d9b64
FB
166int url_close(URLContext *h)
167{
7a813b36 168 int ret = 0;
346db900 169 if (!h) return 0; /* can happen when url_open fails */
de6d9b64 170
7a813b36
AB
171 if (h->prot->url_close)
172 ret = h->prot->url_close(h);
1ea4f593 173 av_free(h);
de6d9b64
FB
174 return ret;
175}
176
177int url_exist(const char *filename)
178{
179 URLContext *h;
180 if (url_open(&h, filename, URL_RDONLY) < 0)
181 return 0;
182 url_close(h);
183 return 1;
184}
185
186offset_t url_filesize(URLContext *h)
187{
188 offset_t pos, size;
115329f1 189
8e287af0
MN
190 size= url_seek(h, 0, AVSEEK_SIZE);
191 if(size<0){
1ae2c5f2 192 pos = url_seek(h, 0, SEEK_CUR);
eabbae73
RB
193 if ((size = url_seek(h, -1, SEEK_END)) < 0)
194 return size;
195 size++;
1ae2c5f2 196 url_seek(h, pos, SEEK_SET);
8e287af0 197 }
de6d9b64
FB
198 return size;
199}
8a9488b5 200
8a9488b5
FB
201int url_get_max_packet_size(URLContext *h)
202{
203 return h->max_packet_size;
204}
f746a046
FB
205
206void url_get_filename(URLContext *h, char *buf, int buf_size)
207{
75e61b0e 208 av_strlcpy(buf, h->filename, buf_size);
f746a046 209}
019ac05a
FB
210
211
212static int default_interrupt_cb(void)
213{
214 return 0;
215}
216
019ac05a
FB
217void url_set_interrupt_cb(URLInterruptCB *interrupt_cb)
218{
219 if (!interrupt_cb)
220 interrupt_cb = default_interrupt_cb;
221 url_interrupt_cb = interrupt_cb;
222}
536333a0 223
502bdf68 224int av_url_read_pause(URLContext *h, int pause)
536333a0
BA
225{
226 if (!h->prot->url_read_pause)
227 return AVERROR(ENOSYS);
502bdf68 228 return h->prot->url_read_pause(h, pause);
536333a0
BA
229}
230
b075e520 231offset_t av_url_read_seek(URLContext *h,
536333a0
BA
232 int stream_index, int64_t timestamp, int flags)
233{
234 if (!h->prot->url_read_seek)
235 return AVERROR(ENOSYS);
236 return h->prot->url_read_seek(h, stream_index, timestamp, flags);
237}