3 * Copyright (c) 2006 Steve Lhomme
4 * Copyright (c) 2007 Wolfram Gloger
5 * Copyright (c) 2010 Michele OrrĂ¹
7 * This file is part of FFmpeg.
9 * FFmpeg is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
14 * FFmpeg is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Lesser General Public License for more details.
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with FFmpeg; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
24 #include "libavutil/avstring.h"
25 #include "libavutil/mem.h"
30 #define AV_CAT_SEPARATOR "|"
33 URLContext *uc; ///< node's URLContext
34 int64_t size; ///< url filesize
38 struct concat_nodes *nodes; ///< list of nodes to concat
39 size_t length; ///< number of cat'ed nodes
40 size_t current; ///< index of currently read node
43 static av_cold int concat_close(URLContext *h)
47 struct concat_data *data = h->priv_data;
48 struct concat_nodes *nodes = data->nodes;
50 for (i = 0; i != data->length; i++)
51 err |= ffurl_close(nodes[i].uc);
53 av_freep(&data->nodes);
55 return err < 0 ? -1 : 0;
58 static av_cold int concat_open(URLContext *h, const char *uri, int flags)
60 char *node_uri = NULL;
65 struct concat_data *data = h->priv_data;
66 struct concat_nodes *nodes;
68 if (!av_strstart(uri, "concat:", &uri)) {
69 av_log(h, AV_LOG_ERROR, "URL %s lacks prefix\n", uri);
70 return AVERROR(EINVAL);
73 for (i = 0, len = 1; uri[i]; i++) {
74 if (uri[i] == *AV_CAT_SEPARATOR) {
75 /* integer overflow */
76 if (++len == UINT_MAX / sizeof(*nodes)) {
77 av_freep(&h->priv_data);
78 return AVERROR(ENAMETOOLONG);
83 if (!(nodes = av_realloc(NULL, sizeof(*nodes) * len)))
84 return AVERROR(ENOMEM);
90 err = AVERROR(ENOENT);
91 for (i = 0; *uri; i++) {
93 len = strcspn(uri, AV_CAT_SEPARATOR);
94 if ((err = av_reallocp(&node_uri, len + 1)) < 0)
96 av_strlcpy(node_uri, uri, len + 1);
97 uri += len + strspn(uri + len, AV_CAT_SEPARATOR);
99 /* creating URLContext */
100 err = ffurl_open_whitelist(&uc, node_uri, flags,
101 &h->interrupt_callback, NULL, h->protocol_whitelist, h->protocol_blacklist);
106 if ((size = ffurl_size(uc)) < 0) {
108 err = AVERROR(ENOSYS);
114 nodes[i].size = size;
121 else if (!(nodes = av_realloc(nodes, data->length * sizeof(*nodes)))) {
123 err = AVERROR(ENOMEM);
129 static int concat_read(URLContext *h, unsigned char *buf, int size)
131 int result, total = 0;
132 struct concat_data *data = h->priv_data;
133 struct concat_nodes *nodes = data->nodes;
134 size_t i = data->current;
137 result = ffurl_read(nodes[i].uc, buf, size);
139 return total ? total : result;
141 if (i + 1 == data->length ||
142 ffurl_seek(nodes[++i].uc, 0, SEEK_SET) < 0)
153 static int64_t concat_seek(URLContext *h, int64_t pos, int whence)
156 struct concat_data *data = h->priv_data;
157 struct concat_nodes *nodes = data->nodes;
162 for (i = data->length - 1; i && pos < -nodes[i].size; i--)
163 pos += nodes[i].size;
166 /* get the absolute position */
167 for (i = 0; i != data->current; i++)
168 pos += nodes[i].size;
169 pos += ffurl_seek(nodes[i].uc, 0, SEEK_CUR);
171 /* fall through with the absolute position */
173 for (i = 0; i != data->length - 1 && pos >= nodes[i].size; i++)
174 pos -= nodes[i].size;
177 return AVERROR(EINVAL);
180 result = ffurl_seek(nodes[i].uc, pos, whence);
184 result += nodes[--i].size;
189 const URLProtocol ff_concat_protocol = {
191 .url_open = concat_open,
192 .url_read = concat_read,
193 .url_seek = concat_seek,
194 .url_close = concat_close,
195 .priv_data_size = sizeof(struct concat_data),
196 .default_whitelist = "concat,file,subfile",