apps/ffmpeg-rockchip
1
0
Fork 0
mirror of https://github.com/nyanmisaka/ffmpeg-rockchip.git synced 2025-10-16 05:41:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

avformat: add a concat protocol that takes a line break delimited list of resources

Browse Source 
Suggested-by: ffmpeg@fb.com
Reviewed-by: Nicolas George <george@nsup.org>
Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
James Almer
2021-06-29 12:07:44 -03:00
parent da0abbbb01
commit 8c2c0135e5
6 changed files with 146 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

109
libavformat/concat.c
View File

@@ -22,9 +22,11 @@
*/
#include "libavutil/avstring.h"
#include "libavutil/bprint.h"
#include "libavutil/mem.h"
#include "avformat.h"
#include "avio_internal.h"
#include "url.h"
#define AV_CAT_SEPARATOR "|"
@@ -56,6 +58,7 @@ static av_cold int concat_close(URLContext *h)
return err < 0 ? -1 : 0;
}
#if CONFIG_CONCAT_PROTOCOL
static av_cold int concat_open(URLContext *h, const char *uri, int flags)
{
char *node_uri = NULL;
@@ -124,6 +127,7 @@ static av_cold int concat_open(URLContext *h, const char *uri, int flags)
data->total_size = total_size;
return err;
}
#endif
static int concat_read(URLContext *h, unsigned char *buf, int size)
{
@@ -188,6 +192,7 @@ static int64_t concat_seek(URLContext *h, int64_t pos, int whence)
return result;
}
#if CONFIG_CONCAT_PROTOCOL
const URLProtocol ff_concat_protocol = {
.name = "concat",
.url_open = concat_open,
@@ -197,3 +202,107 @@ const URLProtocol ff_concat_protocol = {
.priv_data_size = sizeof(struct concat_data),
.default_whitelist = "concat,file,subfile",
};
#endif
#if CONFIG_CONCATF_PROTOCOL
static av_cold int concatf_open(URLContext *h, const char *uri, int flags)
{
AVBPrint bp;
struct concat_data *data = h->priv_data;
AVIOContext *in = NULL;
const char *cursor;
int64_t total_size = 0;
unsigned int nodes_size = 0;
size_t i = 0;
int err;
if (!av_strstart(uri, "concatf:", &uri)) {
av_log(h, AV_LOG_ERROR, "URL %s lacks prefix\n", uri);
return AVERROR(EINVAL);
}
/* handle input */
if (!*uri)
return AVERROR(ENOENT);
err = ffio_open_whitelist(&in, uri, AVIO_FLAG_READ, &h->interrupt_callback,
NULL, h->protocol_whitelist, h->protocol_blacklist);
if (err < 0)
return err;
av_bprint_init(&bp, 0, AV_BPRINT_SIZE_UNLIMITED);
err = avio_read_to_bprint(in, &bp, SIZE_MAX);
avio_closep(&in);
if (err < 0) {
av_bprint_finalize(&bp, NULL);
return err;
}
cursor = bp.str;
while (*cursor) {
struct concat_nodes *nodes;
URLContext *uc;
char *node_uri;
int64_t size;
size_t len = i;
node_uri = av_get_token(&cursor, "\r\n");
if (!node_uri) {
err = AVERROR(ENOMEM);
break;
}
cursor++;
if (++len == SIZE_MAX / sizeof(*nodes)) {
av_free(node_uri);
err = AVERROR(ENAMETOOLONG);
break;
}
/* creating URLContext */
err = ffurl_open_whitelist(&uc, node_uri, flags,
&h->interrupt_callback, NULL, h->protocol_whitelist, h->protocol_blacklist, h);
av_free(node_uri);
if (err < 0)
break;
/* creating size */
if ((size = ffurl_size(uc)) < 0) {
ffurl_close(uc);
err = AVERROR(ENOSYS);
break;
}
nodes = av_fast_realloc(data->nodes, &nodes_size, sizeof(*nodes) * len);
if (!nodes) {
ffurl_close(uc);
err = AVERROR(ENOMEM);
break;
}
data->nodes = nodes;
/* assembling */
data->nodes[i].uc = uc;
data->nodes[i++].size = size;
total_size += size;
}
av_bprint_finalize(&bp, NULL);
data->length = i;
if (err < 0)
concat_close(h);
data->total_size = total_size;
return err;
}
const URLProtocol ff_concatf_protocol = {
.name = "concatf",
.url_open = concatf_open,
.url_read = concat_read,
.url_seek = concat_seek,
.url_close = concat_close,
.priv_data_size = sizeof(struct concat_data),
.default_whitelist = "concatf,concat,file,subfile",
};
#endif