mirror of https://github.com/FFmpeg/FFmpeg.git
parent
bc6901c949
commit
97e87e09c8
6 changed files with 176 additions and 2 deletions
@ -0,0 +1,147 @@ |
|||||||
|
/*
|
||||||
|
* Copyright (c) 2014 Nicolas George |
||||||
|
* |
||||||
|
* 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/avassert.h" |
||||||
|
#include "libavutil/avstring.h" |
||||||
|
#include "libavutil/opt.h" |
||||||
|
#include "avformat.h" |
||||||
|
#include "url.h" |
||||||
|
|
||||||
|
typedef struct SubfileContext { |
||||||
|
const AVClass *class; |
||||||
|
URLContext *h; |
||||||
|
int64_t start; |
||||||
|
int64_t end; |
||||||
|
int64_t pos; |
||||||
|
} SubfileContext; |
||||||
|
|
||||||
|
#define OFFSET(field) offsetof(SubfileContext, field) |
||||||
|
#define D AV_OPT_FLAG_DECODING_PARAM |
||||||
|
|
||||||
|
static const AVOption subfile_options[] = { |
||||||
|
{ "start", "start offset", OFFSET(start), AV_OPT_TYPE_INT64, {.i64 = 0}, 0, INT64_MAX, D }, |
||||||
|
{ "end", "end offset", OFFSET(end), AV_OPT_TYPE_INT64, {.i64 = 0}, 0, INT64_MAX, D }, |
||||||
|
{ NULL } |
||||||
|
}; |
||||||
|
|
||||||
|
#undef OFFSET |
||||||
|
#undef D |
||||||
|
|
||||||
|
static const AVClass subfile_class = { |
||||||
|
.class_name = "subfile", |
||||||
|
.item_name = av_default_item_name, |
||||||
|
.option = subfile_options, |
||||||
|
.version = LIBAVUTIL_VERSION_INT, |
||||||
|
}; |
||||||
|
|
||||||
|
static int slave_seek(URLContext *h) |
||||||
|
{ |
||||||
|
SubfileContext *c = h->priv_data; |
||||||
|
int64_t ret; |
||||||
|
|
||||||
|
if ((ret = ffurl_seek(c->h, c->pos, SEEK_SET)) != c->pos) { |
||||||
|
if (ret >= 0) |
||||||
|
ret = AVERROR_BUG; |
||||||
|
av_log(h, AV_LOG_ERROR, "Impossible to seek in file: %s\n", |
||||||
|
av_err2str(ret)); |
||||||
|
return ret; |
||||||
|
} |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static int subfile_open(URLContext *h, const char *filename, int flags, |
||||||
|
AVDictionary **options) |
||||||
|
{ |
||||||
|
SubfileContext *c = h->priv_data; |
||||||
|
int ret; |
||||||
|
|
||||||
|
if (c->end <= c->start) { |
||||||
|
av_log(h, AV_LOG_ERROR, "end before start\n"); |
||||||
|
return AVERROR(EINVAL); |
||||||
|
} |
||||||
|
av_strstart(filename, "subfile:", &filename); |
||||||
|
ret = ffurl_open(&c->h, filename, flags, &h->interrupt_callback, options); |
||||||
|
if (ret < 0) |
||||||
|
return ret; |
||||||
|
c->pos = c->start; |
||||||
|
if ((ret = slave_seek(h)) < 0) { |
||||||
|
ffurl_close(c->h); |
||||||
|
return ret; |
||||||
|
} |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static int subfile_close(URLContext *h) |
||||||
|
{ |
||||||
|
SubfileContext *c = h->priv_data; |
||||||
|
return ffurl_close(c->h); |
||||||
|
} |
||||||
|
|
||||||
|
static int subfile_read(URLContext *h, unsigned char *buf, int size) |
||||||
|
{ |
||||||
|
SubfileContext *c = h->priv_data; |
||||||
|
int64_t rest = c->end - c->pos; |
||||||
|
int ret; |
||||||
|
|
||||||
|
if (rest <= 0) |
||||||
|
return 0; |
||||||
|
size = FFMIN(size, rest); |
||||||
|
ret = ffurl_read(c->h, buf, size); |
||||||
|
if (ret >= 0) |
||||||
|
c->pos += ret; |
||||||
|
return ret; |
||||||
|
} |
||||||
|
|
||||||
|
static int64_t subfile_seek(URLContext *h, int64_t pos, int whence) |
||||||
|
{ |
||||||
|
SubfileContext *c = h->priv_data; |
||||||
|
int64_t new_pos = -1; |
||||||
|
int ret; |
||||||
|
|
||||||
|
if (whence == AVSEEK_SIZE) |
||||||
|
return c->end - c->start; |
||||||
|
switch (whence) { |
||||||
|
case SEEK_SET: |
||||||
|
new_pos = c->start + pos; |
||||||
|
break; |
||||||
|
case SEEK_CUR: |
||||||
|
new_pos += pos; |
||||||
|
break; |
||||||
|
case SEEK_END: |
||||||
|
new_pos = c->end + c->pos; |
||||||
|
break; |
||||||
|
} |
||||||
|
if (new_pos < c->start) |
||||||
|
return AVERROR(EINVAL); |
||||||
|
c->pos = new_pos; |
||||||
|
if ((ret = slave_seek(h)) < 0) |
||||||
|
return ret; |
||||||
|
return c->pos - c->start; |
||||||
|
} |
||||||
|
|
||||||
|
URLProtocol ff_subfile_protocol = { |
||||||
|
.name = "subfile", |
||||||
|
.url_open2 = subfile_open, |
||||||
|
.url_read = subfile_read, |
||||||
|
.url_seek = subfile_seek, |
||||||
|
.url_close = subfile_close, |
||||||
|
.priv_data_size = sizeof(SubfileContext), |
||||||
|
.priv_data_class = &subfile_class, |
||||||
|
}; |
Loading…
Reference in new issue