Chiebot-Mirror
/
FFmpeg
mirror of https://github.com/FFmpeg/FFmpeg.git
8
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update demuxers and protocols for protocol whitelist support

Browse Source 
Reviewed-by: Andreas Cadhalpun <andreas.cadhalpun@googlemail.com>
Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
pull/174/head
Michael Niedermayer 9 years ago
parent 1dba8371d9
commit fe3fed0b14
33 changed files with 176 additions and 110 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 7
      libavdevice/lavfi.c
  2. 2
      libavformat/async.c
  3. 3
      libavformat/cache.c
  4. 6
      libavformat/concat.c
  5. 5
      libavformat/crypto.c
  6. 9
      libavformat/dashenc.c
  7. 2
      libavformat/file.c
  8. 11
      libavformat/ftp.c
  9. 4
      libavformat/gopher.c
  10. 13
      libavformat/hdsenc.c
  11. 16
      libavformat/hls.c
  12. 29
      libavformat/hlsenc.c
  13. 11
      libavformat/hlsproto.c
  14. 18
      libavformat/http.c
  15. 3
      libavformat/icecast.c
  16. 5
      libavformat/md5proto.c
  17. 5
      libavformat/mmst.c
  18. 2
      libavformat/movenc.c
  19. 5
      libavformat/rtmpcrypt.c
  20. 10
      libavformat/rtmpproto.c
  21. 10
      libavformat/rtpproto.c
  22. 20
      libavformat/rtsp.c
  23. 10
      libavformat/rtspdec.c
  24. 5
      libavformat/sapdec.c
  25. 9
      libavformat/sapenc.c
  26. 17
      libavformat/segment.c
  27. 19
      libavformat/smoothstreamingenc.c
  28. 3
      libavformat/srtpproto.c
  29. 4
      libavformat/subfile.c
  30. 5
      libavformat/tee.c
  31. 5
      libavformat/tls.c
  32. 5
      libavformat/tls_securetransport.c
  33. 8
      libavformat/webm_chunk.c

7
libavdevice/lavfi.c
Unescape View File

@ -40,6 +40,7 @@
#include "libavfilter/avfilter.h" #include "libavfilter/avfilter.h"
#include "libavfilter/avfiltergraph.h" #include "libavfilter/avfiltergraph.h"
#include "libavfilter/buffersink.h" #include "libavfilter/buffersink.h"
#include "libavformat/avio_internal.h"
#include "libavformat/internal.h" #include "libavformat/internal.h"
#include "avdevice.h" #include "avdevice.h"
@ -144,7 +145,11 @@ av_cold static int lavfi_read_header(AVFormatContext *avctx)
if (lavfi->graph_filename) { if (lavfi->graph_filename) {
AVBPrint graph_file_pb; AVBPrint graph_file_pb;
AVIOContext *avio = NULL; AVIOContext *avio = NULL;
ret = avio_open(&avio, lavfi->graph_filename, AVIO_FLAG_READ); AVDictionary *options = NULL;
if (avctx->protocol_whitelist && (ret = av_dict_set(&options, "protocol_whitelist", avctx->protocol_whitelist, 0)) < 0)
goto end;
ret = avio_open2(&avio, lavfi->graph_filename, AVIO_FLAG_READ, &avctx->interrupt_callback, &options);
av_dict_set(&options, "protocol_whitelist", NULL, 0);
if (ret < 0) if (ret < 0)
goto end; goto end;
av_bprint_init(&graph_file_pb, 0, AV_BPRINT_SIZE_UNLIMITED); av_bprint_init(&graph_file_pb, 0, AV_BPRINT_SIZE_UNLIMITED);

2
libavformat/async.c
Unescape View File

@ -251,7 +251,7 @@ static int async_open(URLContext *h, const char *arg, int flags, AVDictionary **
/* wrap interrupt callback */ /* wrap interrupt callback */
c->interrupt_callback = h->interrupt_callback; c->interrupt_callback = h->interrupt_callback;
ret = ffurl_open(&c->inner, arg, flags, &interrupt_callback, options); ret = ffurl_open_whitelist(&c->inner, arg, flags, &interrupt_callback, options, h->protocol_whitelist);
if (ret != 0) { if (ret != 0) {
av_log(h, AV_LOG_ERROR, "ffurl_open failed : %s, %s\n", av_err2str(ret), arg); av_log(h, AV_LOG_ERROR, "ffurl_open failed : %s, %s\n", av_err2str(ret), arg);
goto url_fail; goto url_fail;

3
libavformat/cache.c
Unescape View File

@ -86,7 +86,8 @@ static int cache_open(URLContext *h, const char *arg, int flags, AVDictionary **
unlink(buffername); unlink(buffername);
av_freep(&buffername); av_freep(&buffername);
return ffurl_open(&c->inner, arg, flags, &h->interrupt_callback, options); return ffurl_open_whitelist(&c->inner, arg, flags, &h->interrupt_callback,
options, h->protocol_whitelist);
} }
static int add_entry(URLContext *h, const unsigned char *buf, int size) static int add_entry(URLContext *h, const unsigned char *buf, int size)

6
libavformat/concat.c
Unescape View File

@ -97,8 +97,9 @@ static av_cold int concat_open(URLContext *h, const char *uri, int flags)
uri += len + strspn(uri + len, AV_CAT_SEPARATOR); uri += len + strspn(uri + len, AV_CAT_SEPARATOR);
/* creating URLContext */ /* creating URLContext */
if ((err = ffurl_open(&uc, node_uri, flags, err = ffurl_open_whitelist(&uc, node_uri, flags,
&h->interrupt_callback, NULL)) < 0) &h->interrupt_callback, NULL, h->protocol_whitelist);
if (err < 0)
break; break;
/* creating size */ /* creating size */
@ -192,4 +193,5 @@ URLProtocol ff_concat_protocol = {
.url_seek = concat_seek, .url_seek = concat_seek,
.url_close = concat_close, .url_close = concat_close,
.priv_data_size = sizeof(struct concat_data), .priv_data_size = sizeof(struct concat_data),
.default_whitelist = "concat,file,subfile",
}; };

5
libavformat/crypto.c
Unescape View File

@ -136,8 +136,9 @@ static int crypto_open2(URLContext *h, const char *uri, int flags, AVDictionary
goto err; goto err;
} }
if ((ret = ffurl_open(&c->hd, nested_url, flags, if ((ret = ffurl_open_whitelist(&c->hd, nested_url, flags,
&h->interrupt_callback, options)) < 0) { &h->interrupt_callback, options,
h->protocol_whitelist)) < 0) {
av_log(h, AV_LOG_ERROR, "Unable to open resource: %s\n", nested_url); av_log(h, AV_LOG_ERROR, "Unable to open resource: %s\n", nested_url);
goto err; goto err;
} }

9
libavformat/dashenc.c
Unescape View File

@ -448,7 +448,8 @@ static int write_manifest(AVFormatContext *s, int final)
AVDictionaryEntry *title = av_dict_get(s->metadata, "title", NULL, 0); AVDictionaryEntry *title = av_dict_get(s->metadata, "title", NULL, 0);
snprintf(temp_filename, sizeof(temp_filename), "%s.tmp", s->filename); snprintf(temp_filename, sizeof(temp_filename), "%s.tmp", s->filename);
ret = avio_open2(&out, temp_filename, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL); ret = ffio_open_whitelist(&out, temp_filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) { if (ret < 0) {
av_log(s, AV_LOG_ERROR, "Unable to open %s for writing\n", temp_filename); av_log(s, AV_LOG_ERROR, "Unable to open %s for writing\n", temp_filename);
return ret; return ret;
@ -648,7 +649,7 @@ static int dash_write_header(AVFormatContext *s)
dash_fill_tmpl_params(os->initfile, sizeof(os->initfile), c->init_seg_name, i, 0, os->bit_rate, 0); dash_fill_tmpl_params(os->initfile, sizeof(os->initfile), c->init_seg_name, i, 0, os->bit_rate, 0);
} }
snprintf(filename, sizeof(filename), "%s%s", c->dirname, os->initfile); snprintf(filename, sizeof(filename), "%s%s", c->dirname, os->initfile);
ret = ffurl_open(&os->out, filename, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL); ret = ffurl_open_whitelist(&os->out, filename, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) if (ret < 0)
goto fail; goto fail;
os->init_start_pos = 0; os->init_start_pos = 0;
@ -755,7 +756,7 @@ static void find_index_range(AVFormatContext *s, const char *full_path,
URLContext *fd; URLContext *fd;
int ret; int ret;
ret = ffurl_open(&fd, full_path, AVIO_FLAG_READ, &s->interrupt_callback, NULL); ret = ffurl_open_whitelist(&fd, full_path, AVIO_FLAG_READ, &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) if (ret < 0)
return; return;
if (ffurl_seek(fd, pos, SEEK_SET) != pos) { if (ffurl_seek(fd, pos, SEEK_SET) != pos) {
@ -838,7 +839,7 @@ static int dash_flush(AVFormatContext *s, int final, int stream)
dash_fill_tmpl_params(filename, sizeof(filename), c->media_seg_name, i, os->segment_index, os->bit_rate, os->start_pts); dash_fill_tmpl_params(filename, sizeof(filename), c->media_seg_name, i, os->segment_index, os->bit_rate, os->start_pts);
snprintf(full_path, sizeof(full_path), "%s%s", c->dirname, filename); snprintf(full_path, sizeof(full_path), "%s%s", c->dirname, filename);
snprintf(temp_path, sizeof(temp_path), "%s.tmp", full_path); snprintf(temp_path, sizeof(temp_path), "%s.tmp", full_path);
ret = ffurl_open(&os->out, temp_path, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL); ret = ffurl_open_whitelist(&os->out, temp_path, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) if (ret < 0)
break; break;
write_styp(os->ctx->pb); write_styp(os->ctx->pb);

2
libavformat/file.c
Unescape View File

@ -349,6 +349,7 @@ URLProtocol ff_file_protocol = {
.url_open_dir = file_open_dir, .url_open_dir = file_open_dir,
.url_read_dir = file_read_dir, .url_read_dir = file_read_dir,
.url_close_dir = file_close_dir, .url_close_dir = file_close_dir,
.default_whitelist = "file"
}; };
#endif /* CONFIG_FILE_PROTOCOL */ #endif /* CONFIG_FILE_PROTOCOL */
@ -387,6 +388,7 @@ URLProtocol ff_pipe_protocol = {
.url_check = file_check, .url_check = file_check,
.priv_data_size = sizeof(FileContext), .priv_data_size = sizeof(FileContext),
.priv_data_class = &pipe_class, .priv_data_class = &pipe_class,
.default_whitelist = "none"
}; };
#endif /* CONFIG_PIPE_PROTOCOL */ #endif /* CONFIG_PIPE_PROTOCOL */

11
libavformat/ftp.c
Unescape View File

@ -537,8 +537,9 @@ static int ftp_connect_control_connection(URLContext *h)
if (s->rw_timeout != -1) { if (s->rw_timeout != -1) {
av_dict_set_int(&opts, "timeout", s->rw_timeout, 0); av_dict_set_int(&opts, "timeout", s->rw_timeout, 0);
} /* if option is not given, don't pass it and let tcp use its own default */ } /* if option is not given, don't pass it and let tcp use its own default */
err = ffurl_open(&s->conn_control, buf, AVIO_FLAG_READ_WRITE, err = ffurl_open_whitelist(&s->conn_control, buf, AVIO_FLAG_READ_WRITE,
&h->interrupt_callback, &opts); &h->interrupt_callback, &opts,
h->protocol_whitelist);
av_dict_free(&opts); av_dict_free(&opts);
if (err < 0) { if (err < 0) {
av_log(h, AV_LOG_ERROR, "Cannot open control connection\n"); av_log(h, AV_LOG_ERROR, "Cannot open control connection\n");
@ -590,8 +591,9 @@ static int ftp_connect_data_connection(URLContext *h)
if (s->rw_timeout != -1) { if (s->rw_timeout != -1) {
av_dict_set_int(&opts, "timeout", s->rw_timeout, 0); av_dict_set_int(&opts, "timeout", s->rw_timeout, 0);
} /* if option is not given, don't pass it and let tcp use its own default */ } /* if option is not given, don't pass it and let tcp use its own default */
err = ffurl_open(&s->conn_data, buf, h->flags, err = ffurl_open_whitelist(&s->conn_data, buf, h->flags,
&h->interrupt_callback, &opts); &h->interrupt_callback, &opts,
h->protocol_whitelist);
av_dict_free(&opts); av_dict_free(&opts);
if (err < 0) if (err < 0)
return err; return err;
@ -1114,4 +1116,5 @@ URLProtocol ff_ftp_protocol = {
.url_delete = ftp_delete, .url_delete = ftp_delete,
.url_move = ftp_move, .url_move = ftp_move,
.flags = URL_PROTOCOL_FLAG_NETWORK, .flags = URL_PROTOCOL_FLAG_NETWORK,
.default_whitelist = "tcp",
}; };

4
libavformat/gopher.c
Unescape View File

@ -93,8 +93,8 @@ static int gopher_open(URLContext *h, const char *uri, int flags)
ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port, NULL); ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port, NULL);
s->hd = NULL; s->hd = NULL;
err = ffurl_open(&s->hd, buf, AVIO_FLAG_READ_WRITE, err = ffurl_open_whitelist(&s->hd, buf, AVIO_FLAG_READ_WRITE,
&h->interrupt_callback, NULL); &h->interrupt_callback, NULL, h->protocol_whitelist);
if (err < 0) if (err < 0)
goto fail; goto fail;

13
libavformat/hdsenc.c
Unescape View File

@ -26,6 +26,7 @@
#endif #endif
#include "avformat.h" #include "avformat.h"
#include "avio_internal.h"
#include "internal.h" #include "internal.h"
#include "os_support.h" #include "os_support.h"
@ -169,8 +170,8 @@ static int write_manifest(AVFormatContext *s, int final)
snprintf(filename, sizeof(filename), "%s/index.f4m", s->filename); snprintf(filename, sizeof(filename), "%s/index.f4m", s->filename);
snprintf(temp_filename, sizeof(temp_filename), "%s/index.f4m.tmp", s->filename); snprintf(temp_filename, sizeof(temp_filename), "%s/index.f4m.tmp", s->filename);
ret = avio_open2(&out, temp_filename, AVIO_FLAG_WRITE, ret = ffio_open_whitelist(&out, temp_filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL); &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) { if (ret < 0) {
av_log(s, AV_LOG_ERROR, "Unable to open %s for writing\n", temp_filename); av_log(s, AV_LOG_ERROR, "Unable to open %s for writing\n", temp_filename);
return ret; return ret;
@ -238,8 +239,8 @@ static int write_abst(AVFormatContext *s, OutputStream *os, int final)
"%s/stream%d.abst", s->filename, index); "%s/stream%d.abst", s->filename, index);
snprintf(temp_filename, sizeof(temp_filename), snprintf(temp_filename, sizeof(temp_filename),
"%s/stream%d.abst.tmp", s->filename, index); "%s/stream%d.abst.tmp", s->filename, index);
ret = avio_open2(&out, temp_filename, AVIO_FLAG_WRITE, ret = ffio_open_whitelist(&out, temp_filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL); &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) { if (ret < 0) {
av_log(s, AV_LOG_ERROR, "Unable to open %s for writing\n", temp_filename); av_log(s, AV_LOG_ERROR, "Unable to open %s for writing\n", temp_filename);
return ret; return ret;
@ -289,8 +290,8 @@ static int write_abst(AVFormatContext *s, OutputStream *os, int final)
static int init_file(AVFormatContext *s, OutputStream *os, int64_t start_ts) static int init_file(AVFormatContext *s, OutputStream *os, int64_t start_ts)
{ {
int ret, i; int ret, i;
ret = avio_open2(&os->out, os->temp_filename, AVIO_FLAG_WRITE, ret = ffio_open_whitelist(&os->out, os->temp_filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL); &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) if (ret < 0)
return ret; return ret;
avio_wb32(os->out, 0); avio_wb32(os->out, 0);

16
libavformat/hls.c
Unescape View File

@ -180,6 +180,7 @@ struct variant {
typedef struct HLSContext { typedef struct HLSContext {
AVClass *class; AVClass *class;
AVFormatContext *avfmt;
int n_variants; int n_variants;
struct variant **variants; struct variant **variants;
int n_playlists; int n_playlists;
@ -589,6 +590,14 @@ static int url_connect(struct playlist *pls, AVDictionary *opts, AVDictionary *o
av_dict_copy(&tmp, opts, 0); av_dict_copy(&tmp, opts, 0);
av_dict_copy(&tmp, opts2, 0); av_dict_copy(&tmp, opts2, 0);
if (pls->parent->protocol_whitelist) {
pls->input->protocol_whitelist = av_strdup(pls->parent->protocol_whitelist);
if (!pls->input->protocol_whitelist) {
av_dict_free(&tmp);
return AVERROR(ENOMEM);
}
}
if ((ret = ffurl_connect(pls->input, &tmp)) < 0) { if ((ret = ffurl_connect(pls->input, &tmp)) < 0) {
ffurl_close(pls->input); ffurl_close(pls->input);
pls->input = NULL; pls->input = NULL;
@ -626,7 +635,7 @@ static int open_url(HLSContext *c, URLContext **uc, const char *url, AVDictionar
av_dict_copy(&tmp, c->avio_opts, 0); av_dict_copy(&tmp, c->avio_opts, 0);
av_dict_copy(&tmp, opts, 0); av_dict_copy(&tmp, opts, 0);
ret = ffurl_open(uc, url, AVIO_FLAG_READ, c->interrupt_callback, &tmp); ret = ffurl_open_whitelist(uc, url, AVIO_FLAG_READ, c->interrupt_callback, &tmp, c->avfmt->protocol_whitelist);
if( ret >= 0) { if( ret >= 0) {
// update cookies on http response with setcookies. // update cookies on http response with setcookies.
URLContext *u = *uc; URLContext *u = *uc;
@ -671,8 +680,8 @@ static int parse_playlist(HLSContext *c, const char *url,
av_dict_set(&opts, "headers", c->headers, 0); av_dict_set(&opts, "headers", c->headers, 0);
av_dict_set(&opts, "http_proxy", c->http_proxy, 0); av_dict_set(&opts, "http_proxy", c->http_proxy, 0);
ret = avio_open2(&in, url, AVIO_FLAG_READ, ret = ffio_open_whitelist(&in, url, AVIO_FLAG_READ,
c->interrupt_callback, &opts); c->interrupt_callback, &opts, c->avfmt->protocol_whitelist);
av_dict_free(&opts); av_dict_free(&opts);
if (ret < 0) if (ret < 0)
return ret; return ret;
@ -1509,6 +1518,7 @@ static int hls_read_header(AVFormatContext *s)
HLSContext *c = s->priv_data; HLSContext *c = s->priv_data;
int ret = 0, i, j, stream_offset = 0; int ret = 0, i, j, stream_offset = 0;
c->avfmt = s;
c->interrupt_callback = &s->interrupt_callback; c->interrupt_callback = &s->interrupt_callback;
c->strict_std_compliance = s->strict_std_compliance; c->strict_std_compliance = s->strict_std_compliance;

29
libavformat/hlsenc.c
Unescape View File

@ -35,6 +35,7 @@
#include "libavutil/time_internal.h" #include "libavutil/time_internal.h"
#include "avformat.h" #include "avformat.h"
#include "avio_internal.h"
#include "internal.h" #include "internal.h"
#include "os_support.h" #include "os_support.h"
@ -209,8 +210,8 @@ static int hls_encryption_start(AVFormatContext *s)
AVIOContext *pb; AVIOContext *pb;
uint8_t key[KEYSIZE]; uint8_t key[KEYSIZE];
if ((ret = avio_open2(&pb, hls->key_info_file, AVIO_FLAG_READ, if ((ret = ffio_open_whitelist(&pb, hls->key_info_file, AVIO_FLAG_READ,
&s->interrupt_callback, NULL)) < 0) { &s->interrupt_callback, NULL, s->protocol_whitelist)) < 0) {
av_log(hls, AV_LOG_ERROR, av_log(hls, AV_LOG_ERROR,
"error opening key info file %s\n", hls->key_info_file); "error opening key info file %s\n", hls->key_info_file);
return ret; return ret;
@ -237,8 +238,8 @@ static int hls_encryption_start(AVFormatContext *s)
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
if ((ret = avio_open2(&pb, hls->key_file, AVIO_FLAG_READ, if ((ret = ffio_open_whitelist(&pb, hls->key_file, AVIO_FLAG_READ,
&s->interrupt_callback, NULL)) < 0) { &s->interrupt_callback, NULL, s->protocol_whitelist)) < 0) {
av_log(hls, AV_LOG_ERROR, "error opening key file %s\n", hls->key_file); av_log(hls, AV_LOG_ERROR, "error opening key file %s\n", hls->key_file);
return ret; return ret;
} }
@ -394,8 +395,8 @@ static int hls_window(AVFormatContext *s, int last)
set_http_options(&options, hls); set_http_options(&options, hls);
snprintf(temp_filename, sizeof(temp_filename), use_rename ? "%s.tmp" : "%s", s->filename); snprintf(temp_filename, sizeof(temp_filename), use_rename ? "%s.tmp" : "%s", s->filename);
if ((ret = avio_open2(&out, temp_filename, AVIO_FLAG_WRITE, if ((ret = ffio_open_whitelist(&out, temp_filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, &options)) < 0) &s->interrupt_callback, &options, s->protocol_whitelist)) < 0)
goto fail; goto fail;
for (en = hls->segments; en; en = en->next) { for (en = hls->segments; en; en = en->next) {
@ -445,8 +446,8 @@ static int hls_window(AVFormatContext *s, int last)
avio_printf(out, "#EXT-X-ENDLIST\n"); avio_printf(out, "#EXT-X-ENDLIST\n");
if( hls->vtt_m3u8_name ) { if( hls->vtt_m3u8_name ) {
if ((ret = avio_open2(&sub_out, hls->vtt_m3u8_name, AVIO_FLAG_WRITE, if ((ret = ffio_open_whitelist(&sub_out, hls->vtt_m3u8_name, AVIO_FLAG_WRITE,
&s->interrupt_callback, &options)) < 0) &s->interrupt_callback, &options, s->protocol_whitelist)) < 0)
goto fail; goto fail;
avio_printf(sub_out, "#EXTM3U\n"); avio_printf(sub_out, "#EXTM3U\n");
avio_printf(sub_out, "#EXT-X-VERSION:%d\n", version); avio_printf(sub_out, "#EXT-X-VERSION:%d\n", version);
@ -542,20 +543,20 @@ static int hls_start(AVFormatContext *s)
err = AVERROR(ENOMEM); err = AVERROR(ENOMEM);
goto fail; goto fail;
} }
err = avio_open2(&oc->pb, filename, AVIO_FLAG_WRITE, err = ffio_open_whitelist(&oc->pb, filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, &options); &s->interrupt_callback, &options, s->protocol_whitelist);
av_free(filename); av_free(filename);
av_dict_free(&options); av_dict_free(&options);
if (err < 0) if (err < 0)
return err; return err;
} else } else
if ((err = avio_open2(&oc->pb, oc->filename, AVIO_FLAG_WRITE, if ((err = ffio_open_whitelist(&oc->pb, oc->filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, &options)) < 0) &s->interrupt_callback, &options, s->protocol_whitelist)) < 0)
goto fail; goto fail;
if (c->vtt_basename) { if (c->vtt_basename) {
set_http_options(&options, c); set_http_options(&options, c);
if ((err = avio_open2(&vtt_oc->pb, vtt_oc->filename, AVIO_FLAG_WRITE, if ((err = ffio_open_whitelist(&vtt_oc->pb, vtt_oc->filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, &options)) < 0) &s->interrupt_callback, &options, s->protocol_whitelist)) < 0)
goto fail; goto fail;
} }
av_dict_free(&options); av_dict_free(&options);

11
libavformat/hlsproto.c
Unescape View File

@ -28,6 +28,7 @@
#include "libavutil/avstring.h" #include "libavutil/avstring.h"
#include "libavutil/time.h" #include "libavutil/time.h"
#include "avformat.h" #include "avformat.h"
#include "avio_internal.h"
#include "internal.h" #include "internal.h"
#include "url.h" #include "url.h"
#include "version.h" #include "version.h"
@ -116,8 +117,9 @@ static int parse_playlist(URLContext *h, const char *url)
char line[1024]; char line[1024];
const char *ptr; const char *ptr;
if ((ret = avio_open2(&in, url, AVIO_FLAG_READ, if ((ret = ffio_open_whitelist(&in, url, AVIO_FLAG_READ,
&h->interrupt_callback, NULL)) < 0) &h->interrupt_callback, NULL,
h->protocol_whitelist)) < 0)
return ret; return ret;
read_chomp_line(in, line, sizeof(line)); read_chomp_line(in, line, sizeof(line));
@ -303,8 +305,9 @@ retry:
} }
url = s->segments[s->cur_seq_no - s->start_seq_no]->url, url = s->segments[s->cur_seq_no - s->start_seq_no]->url,
av_log(h, AV_LOG_DEBUG, "opening %s\n", url); av_log(h, AV_LOG_DEBUG, "opening %s\n", url);
ret = ffurl_open(&s->seg_hd, url, AVIO_FLAG_READ, ret = ffurl_open_whitelist(&s->seg_hd, url, AVIO_FLAG_READ,
&h->interrupt_callback, NULL); &h->interrupt_callback, NULL,
h->protocol_whitelist);
if (ret < 0) { if (ret < 0) {
if (ff_check_interrupt(&h->interrupt_callback)) if (ff_check_interrupt(&h->interrupt_callback))
return AVERROR_EXIT; return AVERROR_EXIT;

18
libavformat/http.c
Unescape View File

@ -219,8 +219,9 @@ static int http_open_cnx_internal(URLContext *h, AVDictionary **options)
ff_url_join(buf, sizeof(buf), lower_proto, NULL, hostname, port, NULL); ff_url_join(buf, sizeof(buf), lower_proto, NULL, hostname, port, NULL);
if (!s->hd) { if (!s->hd) {
err = ffurl_open(&s->hd, buf, AVIO_FLAG_READ_WRITE, err = ffurl_open_whitelist(&s->hd, buf, AVIO_FLAG_READ_WRITE,
&h->interrupt_callback, options); &h->interrupt_callback, options,
h->protocol_whitelist);
if (err < 0) if (err < 0)
return err; return err;
} }
@ -453,8 +454,10 @@ static int http_listen(URLContext *h, const char *uri, int flags,
NULL); NULL);
if ((ret = av_dict_set_int(options, "listen", s->listen, 0)) < 0) if ((ret = av_dict_set_int(options, "listen", s->listen, 0)) < 0)
goto fail; goto fail;
if ((ret = ffurl_open(&s->hd, lower_url, AVIO_FLAG_READ_WRITE, if ((ret = ffurl_open_whitelist(&s->hd, lower_url, AVIO_FLAG_READ_WRITE,
&h->interrupt_callback, options)) < 0) &h->interrupt_callback, options,
h->protocol_whitelist
)) < 0)
goto fail; goto fail;
s->handshake_step = LOWER_PROTO; s->handshake_step = LOWER_PROTO;
if (s->listen == HTTP_SINGLE) { /* single client */ if (s->listen == HTTP_SINGLE) { /* single client */
@ -1519,6 +1522,7 @@ URLProtocol ff_http_protocol = {
.priv_data_size = sizeof(HTTPContext), .priv_data_size = sizeof(HTTPContext),
.priv_data_class = &http_context_class, .priv_data_class = &http_context_class,
.flags = URL_PROTOCOL_FLAG_NETWORK, .flags = URL_PROTOCOL_FLAG_NETWORK,
.default_whitelist = "http,https,tls,rtp,tcp,udp"
}; };
#endif /* CONFIG_HTTP_PROTOCOL */ #endif /* CONFIG_HTTP_PROTOCOL */
@ -1537,6 +1541,7 @@ URLProtocol ff_https_protocol = {
.priv_data_size = sizeof(HTTPContext), .priv_data_size = sizeof(HTTPContext),
.priv_data_class = &https_context_class, .priv_data_class = &https_context_class,
.flags = URL_PROTOCOL_FLAG_NETWORK, .flags = URL_PROTOCOL_FLAG_NETWORK,
.default_whitelist = "http,https,tls,rtp,tcp,udp"
}; };
#endif /* CONFIG_HTTPS_PROTOCOL */ #endif /* CONFIG_HTTPS_PROTOCOL */
@ -1575,8 +1580,9 @@ static int http_proxy_open(URLContext *h, const char *uri, int flags)
ff_url_join(lower_url, sizeof(lower_url), "tcp", NULL, hostname, port, ff_url_join(lower_url, sizeof(lower_url), "tcp", NULL, hostname, port,
NULL); NULL);
redo: redo:
ret = ffurl_open(&s->hd, lower_url, AVIO_FLAG_READ_WRITE, ret = ffurl_open_whitelist(&s->hd, lower_url, AVIO_FLAG_READ_WRITE,
&h->interrupt_callback, NULL); &h->interrupt_callback, NULL,
h->protocol_whitelist);
if (ret < 0) if (ret < 0)
return ret; return ret;

3
libavformat/icecast.c
Unescape View File

@ -164,7 +164,8 @@ static int icecast_open(URLContext *h, const char *uri, int flags)
// Build new URI for passing to http protocol // Build new URI for passing to http protocol
ff_url_join(h_url, sizeof(h_url), "http", auth, host, port, "%s", path); ff_url_join(h_url, sizeof(h_url), "http", auth, host, port, "%s", path);
// Finally open http proto handler // Finally open http proto handler
ret = ffurl_open(&s->hd, h_url, AVIO_FLAG_READ_WRITE, NULL, &opt_dict); ret = ffurl_open_whitelist(&s->hd, h_url, AVIO_FLAG_READ_WRITE, NULL,
&opt_dict, h->protocol_whitelist);
cleanup: cleanup:
av_freep(&user); av_freep(&user);

5
libavformat/md5proto.c
Unescape View File

@ -69,8 +69,9 @@ static int md5_close(URLContext *h)
av_strstart(filename, "md5:", &filename); av_strstart(filename, "md5:", &filename);
if (*filename) { if (*filename) {
err = ffurl_open(&out, filename, AVIO_FLAG_WRITE, err = ffurl_open_whitelist(&out, filename, AVIO_FLAG_WRITE,
&h->interrupt_callback, NULL); &h->interrupt_callback, NULL,
h->protocol_whitelist);
if (err) if (err)
return err; return err;
err = ffurl_write(out, buf, i*2+1); err = ffurl_write(out, buf, i*2+1);

5
libavformat/mmst.c
Unescape View File

@ -528,8 +528,9 @@ static int mms_open(URLContext *h, const char *uri, int flags)
// establish tcp connection. // establish tcp connection.
ff_url_join(tcpname, sizeof(tcpname), "tcp", NULL, mmst->host, port, NULL); ff_url_join(tcpname, sizeof(tcpname), "tcp", NULL, mmst->host, port, NULL);
err = ffurl_open(&mms->mms_hd, tcpname, AVIO_FLAG_READ_WRITE, err = ffurl_open_whitelist(&mms->mms_hd, tcpname, AVIO_FLAG_READ_WRITE,
&h->interrupt_callback, NULL); &h->interrupt_callback, NULL,
h->protocol_whitelist);
if (err) if (err)
goto fail; goto fail;

2
libavformat/movenc.c
Unescape View File

@ -5534,7 +5534,7 @@ static int shift_data(AVFormatContext *s)
* writing, so we re-open the same output, but for reading. It also avoids * writing, so we re-open the same output, but for reading. It also avoids
* a read/seek/write/seek back and forth. */ * a read/seek/write/seek back and forth. */
avio_flush(s->pb); avio_flush(s->pb);
ret = avio_open(&read_pb, s->filename, AVIO_FLAG_READ); ret = ffio_open_whitelist(&read_pb, s->filename, AVIO_FLAG_READ, &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) { if (ret < 0) {
av_log(s, AV_LOG_ERROR, "Unable to re-open %s output file for " av_log(s, AV_LOG_ERROR, "Unable to re-open %s output file for "
"the second pass (faststart)\n", s->filename); "the second pass (faststart)\n", s->filename);

5
libavformat/rtmpcrypt.c
Unescape View File

@ -264,8 +264,9 @@ static int rtmpe_open(URLContext *h, const char *uri, int flags)
} }
/* open the tcp or ffrtmphttp connection */ /* open the tcp or ffrtmphttp connection */
if ((ret = ffurl_open(&rt->stream, url, AVIO_FLAG_READ_WRITE, if ((ret = ffurl_open_whitelist(&rt->stream, url, AVIO_FLAG_READ_WRITE,
&h->interrupt_callback, NULL)) < 0) { &h->interrupt_callback, NULL,
h->protocol_whitelist)) < 0) {
rtmpe_close(h); rtmpe_close(h);
return ret; return ret;
} }

10
libavformat/rtmpproto.c
Unescape View File

@ -1118,8 +1118,9 @@ static int rtmp_calc_swfhash(URLContext *s)
int ret = 0; int ret = 0;
/* Get the SWF player file. */ /* Get the SWF player file. */
if ((ret = ffurl_open(&stream, rt->swfverify, AVIO_FLAG_READ, if ((ret = ffurl_open_whitelist(&stream, rt->swfverify, AVIO_FLAG_READ,
&s->interrupt_callback, NULL)) < 0) { &s->interrupt_callback, NULL,
s->protocol_whitelist)) < 0) {
av_log(s, AV_LOG_ERROR, "Cannot open connection %s.\n", rt->swfverify); av_log(s, AV_LOG_ERROR, "Cannot open connection %s.\n", rt->swfverify);
goto fail; goto fail;
} }
@ -2647,8 +2648,9 @@ static int rtmp_open(URLContext *s, const char *uri, int flags)
} }
reconnect: reconnect:
if ((ret = ffurl_open(&rt->stream, buf, AVIO_FLAG_READ_WRITE, if ((ret = ffurl_open_whitelist(&rt->stream, buf, AVIO_FLAG_READ_WRITE,
&s->interrupt_callback, &opts)) < 0) { &s->interrupt_callback, &opts,
s->protocol_whitelist)) < 0) {
av_log(s , AV_LOG_ERROR, "Cannot open connection %s\n", buf); av_log(s , AV_LOG_ERROR, "Cannot open connection %s\n", buf);
goto fail; goto fail;
} }

10
libavformat/rtpproto.c
Unescape View File

@ -380,7 +380,8 @@ static int rtp_open(URLContext *h, const char *uri, int flags)
build_udp_url(s, buf, sizeof(buf), build_udp_url(s, buf, sizeof(buf),
hostname, rtp_port, s->local_rtpport, hostname, rtp_port, s->local_rtpport,
sources, block); sources, block);
if (ffurl_open(&s->rtp_hd, buf, flags, &h->interrupt_callback, NULL) < 0) if (ffurl_open_whitelist(&s->rtp_hd, buf, flags, &h->interrupt_callback,
NULL, h->protocol_whitelist) < 0)
goto fail; goto fail;
s->local_rtpport = ff_udp_get_local_port(s->rtp_hd); s->local_rtpport = ff_udp_get_local_port(s->rtp_hd);
if(s->local_rtpport == 65535) { if(s->local_rtpport == 65535) {
@ -392,7 +393,9 @@ static int rtp_open(URLContext *h, const char *uri, int flags)
build_udp_url(s, buf, sizeof(buf), build_udp_url(s, buf, sizeof(buf),
hostname, s->rtcp_port, s->local_rtcpport, hostname, s->rtcp_port, s->local_rtcpport,
sources, block); sources, block);
if (ffurl_open(&s->rtcp_hd, buf, flags, &h->interrupt_callback, NULL) < 0) { if (ffurl_open_whitelist(&s->rtcp_hd, buf, flags,
&h->interrupt_callback, NULL,
h->protocol_whitelist) < 0) {
s->local_rtpport = s->local_rtcpport = -1; s->local_rtpport = s->local_rtcpport = -1;
continue; continue;
} }
@ -401,7 +404,8 @@ static int rtp_open(URLContext *h, const char *uri, int flags)
build_udp_url(s, buf, sizeof(buf), build_udp_url(s, buf, sizeof(buf),
hostname, s->rtcp_port, s->local_rtcpport, hostname, s->rtcp_port, s->local_rtcpport,
sources, block); sources, block);
if (ffurl_open(&s->rtcp_hd, buf, flags, &h->interrupt_callback, NULL) < 0) if (ffurl_open_whitelist(&s->rtcp_hd, buf, flags, &h->interrupt_callback,
NULL, h->protocol_whitelist) < 0)
goto fail; goto fail;
break; break;
} }

20
libavformat/rtsp.c
Unescape View File

@ -1468,8 +1468,8 @@ int ff_rtsp_make_setup_request(AVFormatContext *s, const char *host, int port,
"?localport=%d", j); "?localport=%d", j);
/* we will use two ports per rtp stream (rtp and rtcp) */ /* we will use two ports per rtp stream (rtp and rtcp) */
j += 2; j += 2;
err = ffurl_open(&rtsp_st->rtp_handle, buf, AVIO_FLAG_READ_WRITE, err = ffurl_open_whitelist(&rtsp_st->rtp_handle, buf, AVIO_FLAG_READ_WRITE,
&s->interrupt_callback, &opts); &s->interrupt_callback, &opts, s->protocol_whitelist);
av_dict_free(&opts); av_dict_free(&opts);
@ -1611,8 +1611,8 @@ int ff_rtsp_make_setup_request(AVFormatContext *s, const char *host, int port,
namebuf, sizeof(namebuf), NULL, 0, NI_NUMERICHOST); namebuf, sizeof(namebuf), NULL, 0, NI_NUMERICHOST);
ff_url_join(url, sizeof(url), "rtp", NULL, namebuf, ff_url_join(url, sizeof(url), "rtp", NULL, namebuf,
port, "%s", optbuf); port, "%s", optbuf);
if (ffurl_open(&rtsp_st->rtp_handle, url, AVIO_FLAG_READ_WRITE, if (ffurl_open_whitelist(&rtsp_st->rtp_handle, url, AVIO_FLAG_READ_WRITE,
&s->interrupt_callback, NULL) < 0) { &s->interrupt_callback, NULL, s->protocol_whitelist) < 0) {
err = AVERROR_INVALIDDATA; err = AVERROR_INVALIDDATA;
goto fail; goto fail;
} }
@ -1800,8 +1800,8 @@ redirect:
ff_url_join(tcpname, sizeof(tcpname), lower_rtsp_proto, NULL, ff_url_join(tcpname, sizeof(tcpname), lower_rtsp_proto, NULL,
host, port, host, port,
"?timeout=%d", rt->stimeout); "?timeout=%d", rt->stimeout);
if ((ret = ffurl_open(&rt->rtsp_hd, tcpname, AVIO_FLAG_READ_WRITE, if ((ret = ffurl_open_whitelist(&rt->rtsp_hd, tcpname, AVIO_FLAG_READ_WRITE,
&s->interrupt_callback, NULL)) < 0) { &s->interrupt_callback, NULL, s->protocol_whitelist)) < 0) {
err = ret; err = ret;
goto fail; goto fail;
} }
@ -2316,8 +2316,8 @@ static int sdp_read_header(AVFormatContext *s)
append_source_addrs(url, sizeof(url), "block", append_source_addrs(url, sizeof(url), "block",
rtsp_st->nb_exclude_source_addrs, rtsp_st->nb_exclude_source_addrs,
rtsp_st->exclude_source_addrs); rtsp_st->exclude_source_addrs);
err = ffurl_open(&rtsp_st->rtp_handle, url, AVIO_FLAG_READ_WRITE, err = ffurl_open_whitelist(&rtsp_st->rtp_handle, url, AVIO_FLAG_READ_WRITE,
&s->interrupt_callback, &opts); &s->interrupt_callback, &opts, s->protocol_whitelist);
av_dict_free(&opts); av_dict_free(&opts);
@ -2386,8 +2386,8 @@ static int rtp_read_header(AVFormatContext *s)
if (!ff_network_init()) if (!ff_network_init())
return AVERROR(EIO); return AVERROR(EIO);
ret = ffurl_open(&in, s->filename, AVIO_FLAG_READ, ret = ffurl_open_whitelist(&in, s->filename, AVIO_FLAG_READ,
&s->interrupt_callback, NULL); &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret) if (ret)
goto fail; goto fail;

10
libavformat/rtspdec.c
Unescape View File

@ -294,8 +294,9 @@ static int rtsp_read_setup(AVFormatContext *s, char* host, char *controlurl)
av_dict_set(&opts, "buffer_size", buf, 0); av_dict_set(&opts, "buffer_size", buf, 0);
ff_url_join(url, sizeof(url), "rtp", NULL, host, localport, NULL); ff_url_join(url, sizeof(url), "rtp", NULL, host, localport, NULL);
av_log(s, AV_LOG_TRACE, "Opening: %s", url); av_log(s, AV_LOG_TRACE, "Opening: %s", url);
ret = ffurl_open(&rtsp_st->rtp_handle, url, AVIO_FLAG_READ_WRITE, ret = ffurl_open_whitelist(&rtsp_st->rtp_handle, url, AVIO_FLAG_READ_WRITE,
&s->interrupt_callback, &opts); &s->interrupt_callback, &opts,
s->protocol_whitelist);
av_dict_free(&opts); av_dict_free(&opts);
if (ret) if (ret)
localport += 2; localport += 2;
@ -662,8 +663,9 @@ static int rtsp_listen(AVFormatContext *s)
ff_url_join(tcpname, sizeof(tcpname), lower_proto, NULL, host, port, ff_url_join(tcpname, sizeof(tcpname), lower_proto, NULL, host, port,
"?listen&listen_timeout=%d", rt->initial_timeout * 1000); "?listen&listen_timeout=%d", rt->initial_timeout * 1000);
if (ret = ffurl_open(&rt->rtsp_hd, tcpname, AVIO_FLAG_READ_WRITE, if (ret = ffurl_open_whitelist(&rt->rtsp_hd, tcpname, AVIO_FLAG_READ_WRITE,
&s->interrupt_callback, NULL)) { &s->interrupt_callback, NULL,
s->protocol_whitelist)) {
av_log(s, AV_LOG_ERROR, "Unable to open RTSP for listening\n"); av_log(s, AV_LOG_ERROR, "Unable to open RTSP for listening\n");
return ret; return ret;
} }

5
libavformat/sapdec.c
Unescape View File

@ -85,8 +85,9 @@ static int sap_read_header(AVFormatContext *s)
ff_url_join(url, sizeof(url), "udp", NULL, host, port, "?localport=%d", ff_url_join(url, sizeof(url), "udp", NULL, host, port, "?localport=%d",
port); port);
ret = ffurl_open(&sap->ann_fd, url, AVIO_FLAG_READ, ret = ffurl_open_whitelist(&sap->ann_fd, url, AVIO_FLAG_READ,
&s->interrupt_callback, NULL); &s->interrupt_callback, NULL,
s->protocol_whitelist);
if (ret) if (ret)
goto fail; goto fail;

9
libavformat/sapenc.c
Unescape View File

@ -149,7 +149,9 @@ static int sap_write_header(AVFormatContext *s)
"?ttl=%d", ttl); "?ttl=%d", ttl);
if (!same_port) if (!same_port)
base_port += 2; base_port += 2;
ret = ffurl_open(&fd, url, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL); ret = ffurl_open_whitelist(&fd, url, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL,
s->protocol_whitelist);
if (ret) { if (ret) {
ret = AVERROR(EIO); ret = AVERROR(EIO);
goto fail; goto fail;
@ -167,8 +169,9 @@ static int sap_write_header(AVFormatContext *s)
ff_url_join(url, sizeof(url), "udp", NULL, announce_addr, port, ff_url_join(url, sizeof(url), "udp", NULL, announce_addr, port,
"?ttl=%d&connect=1", ttl); "?ttl=%d&connect=1", ttl);
ret = ffurl_open(&sap->ann_fd, url, AVIO_FLAG_WRITE, ret = ffurl_open_whitelist(&sap->ann_fd, url, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL); &s->interrupt_callback, NULL,
s->protocol_whitelist);
if (ret) { if (ret) {
ret = AVERROR(EIO); ret = AVERROR(EIO);
goto fail; goto fail;

17
libavformat/segment.c
Unescape View File

@ -30,6 +30,7 @@
#include <time.h> #include <time.h>
#include "avformat.h" #include "avformat.h"
#include "avio_internal.h"
#include "internal.h" #include "internal.h"
#include "libavutil/avassert.h" #include "libavutil/avassert.h"
@ -239,8 +240,8 @@ static int segment_start(AVFormatContext *s, int write_header)
if ((err = set_segment_filename(s)) < 0) if ((err = set_segment_filename(s)) < 0)
return err; return err;
if ((err = avio_open2(&oc->pb, oc->filename, AVIO_FLAG_WRITE, if ((err = ffio_open_whitelist(&oc->pb, oc->filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL)) < 0) { &s->interrupt_callback, NULL, s->protocol_whitelist)) < 0) {
av_log(s, AV_LOG_ERROR, "Failed to open segment '%s'\n", oc->filename); av_log(s, AV_LOG_ERROR, "Failed to open segment '%s'\n", oc->filename);
return err; return err;
} }
@ -265,8 +266,8 @@ static int segment_list_open(AVFormatContext *s)
int ret; int ret;
snprintf(seg->temp_list_filename, sizeof(seg->temp_list_filename), seg->use_rename ? "%s.tmp" : "%s", seg->list); snprintf(seg->temp_list_filename, sizeof(seg->temp_list_filename), seg->use_rename ? "%s.tmp" : "%s", seg->list);
ret = avio_open2(&seg->list_pb, seg->temp_list_filename, AVIO_FLAG_WRITE, ret = ffio_open_whitelist(&seg->list_pb, seg->temp_list_filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL); &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) { if (ret < 0) {
av_log(s, AV_LOG_ERROR, "Failed to open segment list '%s'\n", seg->list); av_log(s, AV_LOG_ERROR, "Failed to open segment list '%s'\n", seg->list);
return ret; return ret;
@ -699,8 +700,8 @@ static int seg_write_header(AVFormatContext *s)
goto fail; goto fail;
if (seg->write_header_trailer) { if (seg->write_header_trailer) {
if ((ret = avio_open2(&oc->pb, seg->header_filename ? seg->header_filename : oc->filename, AVIO_FLAG_WRITE, if ((ret = ffio_open_whitelist(&oc->pb, seg->header_filename ? seg->header_filename : oc->filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL)) < 0) { &s->interrupt_callback, NULL, s->protocol_whitelist)) < 0) {
av_log(s, AV_LOG_ERROR, "Failed to open segment '%s'\n", oc->filename); av_log(s, AV_LOG_ERROR, "Failed to open segment '%s'\n", oc->filename);
goto fail; goto fail;
} }
@ -743,8 +744,8 @@ static int seg_write_header(AVFormatContext *s)
} else { } else {
close_null_ctxp(&oc->pb); close_null_ctxp(&oc->pb);
} }
if ((ret = avio_open2(&oc->pb, oc->filename, AVIO_FLAG_WRITE, if ((ret = ffio_open_whitelist(&oc->pb, oc->filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL)) < 0) &s->interrupt_callback, NULL, s->protocol_whitelist)) < 0)
goto fail; goto fail;
if (!seg->individual_header_trailer) if (!seg->individual_header_trailer)
oc->pb->seekable = 0; oc->pb->seekable = 0;

19
libavformat/smoothstreamingenc.c
Unescape View File

@ -26,6 +26,7 @@
#endif #endif
#include "avformat.h" #include "avformat.h"
#include "avio_internal.h"
#include "internal.h" #include "internal.h"
#include "os_support.h" #include "os_support.h"
#include "avc.h" #include "avc.h"
@ -121,7 +122,8 @@ static int64_t ism_seek(void *opaque, int64_t offset, int whence)
AVDictionary *opts = NULL; AVDictionary *opts = NULL;
os->tail_out = os->out; os->tail_out = os->out;
av_dict_set(&opts, "truncate", "0", 0); av_dict_set(&opts, "truncate", "0", 0);
ret = ffurl_open(&os->out, frag->file, AVIO_FLAG_READ_WRITE, &os->ctx->interrupt_callback, &opts); ret = ffurl_open_whitelist(&os->out, frag->file, AVIO_FLAG_READ_WRITE,
&os->ctx->interrupt_callback, &opts, os->ctx->protocol_whitelist);
av_dict_free(&opts); av_dict_free(&opts);
if (ret < 0) { if (ret < 0) {
os->out = os->tail_out; os->out = os->tail_out;
@ -129,7 +131,8 @@ static int64_t ism_seek(void *opaque, int64_t offset, int whence)
return ret; return ret;
} }
av_dict_set(&opts, "truncate", "0", 0); av_dict_set(&opts, "truncate", "0", 0);
ffurl_open(&os->out2, frag->infofile, AVIO_FLAG_READ_WRITE, &os->ctx->interrupt_callback, &opts); ffurl_open_whitelist(&os->out2, frag->infofile, AVIO_FLAG_READ_WRITE,
&os->ctx->interrupt_callback, &opts, os->ctx->protocol_whitelist);
av_dict_free(&opts); av_dict_free(&opts);
ffurl_seek(os->out, offset - frag->start_pos, SEEK_SET); ffurl_seek(os->out, offset - frag->start_pos, SEEK_SET);
if (os->out2) if (os->out2)
@ -220,7 +223,7 @@ static int write_manifest(AVFormatContext *s, int final)
snprintf(filename, sizeof(filename), "%s/Manifest", s->filename); snprintf(filename, sizeof(filename), "%s/Manifest", s->filename);
snprintf(temp_filename, sizeof(temp_filename), "%s/Manifest.tmp", s->filename); snprintf(temp_filename, sizeof(temp_filename), "%s/Manifest.tmp", s->filename);
ret = avio_open2(&out, temp_filename, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL); ret = ffio_open_whitelist(&out, temp_filename, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) { if (ret < 0) {
av_log(s, AV_LOG_ERROR, "Unable to open %s for writing\n", temp_filename); av_log(s, AV_LOG_ERROR, "Unable to open %s for writing\n", temp_filename);
return ret; return ret;
@ -329,7 +332,7 @@ static int ism_write_header(AVFormatContext *s)
} }
ctx = avformat_alloc_context(); ctx = avformat_alloc_context();
if (!ctx) { if (!ctx || ff_copy_whitelists(ctx, s) < 0) {
ret = AVERROR(ENOMEM); ret = AVERROR(ENOMEM);
goto fail; goto fail;
} }
@ -409,7 +412,7 @@ static int parse_fragment(AVFormatContext *s, const char *filename, int64_t *sta
AVIOContext *in; AVIOContext *in;
int ret; int ret;
uint32_t len; uint32_t len;
if ((ret = avio_open2(&in, filename, AVIO_FLAG_READ, &s->interrupt_callback, NULL)) < 0) if ((ret = ffio_open_whitelist(&in, filename, AVIO_FLAG_READ, &s->interrupt_callback, NULL, s->protocol_whitelist)) < 0)
return ret; return ret;
ret = AVERROR(EIO); ret = AVERROR(EIO);
*moof_size = avio_rb32(in); *moof_size = avio_rb32(in);
@ -486,9 +489,9 @@ static int copy_moof(AVFormatContext *s, const char* infile, const char *outfile
{ {
AVIOContext *in, *out; AVIOContext *in, *out;
int ret = 0; int ret = 0;
if ((ret = avio_open2(&in, infile, AVIO_FLAG_READ, &s->interrupt_callback, NULL)) < 0) if ((ret = ffio_open_whitelist(&in, infile, AVIO_FLAG_READ, &s->interrupt_callback, NULL, s->protocol_whitelist)) < 0)
return ret; return ret;
if ((ret = avio_open2(&out, outfile, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL)) < 0) { if ((ret = ffio_open_whitelist(&out, outfile, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL, s->protocol_whitelist)) < 0) {
avio_close(in); avio_close(in);
return ret; return ret;
} }
@ -523,7 +526,7 @@ static int ism_flush(AVFormatContext *s, int final)
continue; continue;
snprintf(filename, sizeof(filename), "%s/temp", os->dirname); snprintf(filename, sizeof(filename), "%s/temp", os->dirname);
ret = ffurl_open(&os->out, filename, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL); ret = ffurl_open_whitelist(&os->out, filename, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) if (ret < 0)
break; break;
os->cur_start_pos = os->tail_pos; os->cur_start_pos = os->tail_pos;

3
libavformat/srtpproto.c
Unescape View File

@ -80,7 +80,8 @@ static int srtp_open(URLContext *h, const char *uri, int flags)
av_url_split(NULL, 0, NULL, 0, hostname, sizeof(hostname), &rtp_port, av_url_split(NULL, 0, NULL, 0, hostname, sizeof(hostname), &rtp_port,
path, sizeof(path), uri); path, sizeof(path), uri);
ff_url_join(buf, sizeof(buf), "rtp", NULL, hostname, rtp_port, "%s", path); ff_url_join(buf, sizeof(buf), "rtp", NULL, hostname, rtp_port, "%s", path);
if ((ret = ffurl_open(&s->rtp_hd, buf, flags, &h->interrupt_callback, NULL)) < 0) if ((ret = ffurl_open_whitelist(&s->rtp_hd, buf, flags, &h->interrupt_callback,
NULL, h->protocol_whitelist)) < 0)
goto fail; goto fail;
h->max_packet_size = FFMIN(s->rtp_hd->max_packet_size, h->max_packet_size = FFMIN(s->rtp_hd->max_packet_size,

4
libavformat/subfile.c
Unescape View File

@ -77,7 +77,8 @@ static int subfile_open(URLContext *h, const char *filename, int flags,
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
av_strstart(filename, "subfile:", &filename); av_strstart(filename, "subfile:", &filename);
ret = ffurl_open(&c->h, filename, flags, &h->interrupt_callback, options); ret = ffurl_open_whitelist(&c->h, filename, flags, &h->interrupt_callback,
options, h->protocol_whitelist);
if (ret < 0) if (ret < 0)
return ret; return ret;
c->pos = c->start; c->pos = c->start;
@ -144,4 +145,5 @@ URLProtocol ff_subfile_protocol = {
.url_close = subfile_close, .url_close = subfile_close,
.priv_data_size = sizeof(SubfileContext), .priv_data_size = sizeof(SubfileContext),
.priv_data_class = &subfile_class, .priv_data_class = &subfile_class,
.default_whitelist = "file",
}; };

5
libavformat/tee.c
Unescape View File

@ -24,6 +24,7 @@
#include "libavutil/avstring.h" #include "libavutil/avstring.h"
#include "libavutil/opt.h" #include "libavutil/opt.h"
#include "avformat.h" #include "avformat.h"
#include "avio_internal.h"
#define MAX_SLAVES 16 #define MAX_SLAVES 16
@ -226,7 +227,9 @@ static int open_slave(AVFormatContext *avf, char *slave, TeeSlave *tee_slave)
} }
if (!(avf2->oformat->flags & AVFMT_NOFILE)) { if (!(avf2->oformat->flags & AVFMT_NOFILE)) {
if ((ret = avio_open(&avf2->pb, filename, AVIO_FLAG_WRITE)) < 0) { if ((ret = ffio_open_whitelist(&avf2->pb, filename, AVIO_FLAG_WRITE,
&avf->interrupt_callback, NULL,
avf->protocol_whitelist)) < 0) {
av_log(avf, AV_LOG_ERROR, "Slave '%s': error opening: %s\n", av_log(avf, AV_LOG_ERROR, "Slave '%s': error opening: %s\n",
slave, av_err2str(ret)); slave, av_err2str(ret));
goto end; goto end;

5
libavformat/tls.c
Unescape View File

@ -104,6 +104,7 @@ int ff_tls_open_underlying(TLSShared *c, URLContext *parent, const char *uri, AV
proxy_port, "/%s", dest); proxy_port, "/%s", dest);
} }
return ffurl_open(&c->tcp, buf, AVIO_FLAG_READ_WRITE, return ffurl_open_whitelist(&c->tcp, buf, AVIO_FLAG_READ_WRITE,
&parent->interrupt_callback, options); &parent->interrupt_callback, options,
parent->protocol_whitelist);
} }

5
libavformat/tls_securetransport.c
Unescape View File

@ -80,8 +80,9 @@ static int import_pem(URLContext *h, char *path, CFArrayRef *array)
goto end; goto end;
} }
if ((ret = avio_open2(&s, path, AVIO_FLAG_READ, if ((ret = ffio_open_whitelist(&s, path, AVIO_FLAG_READ,
&h->interrupt_callback, NULL)) < 0) &h->interrupt_callback, NULL,
h->protocol_whitelist)) < 0)
goto end; goto end;
if ((ret = avio_size(s)) < 0) if ((ret = avio_size(s)) < 0)

8
libavformat/webm_chunk.c
Unescape View File

@ -29,6 +29,7 @@
#include "avformat.h" #include "avformat.h"
#include "avio.h" #include "avio.h"
#include "avio_internal.h"
#include "internal.h" #include "internal.h"
#include "libavutil/avassert.h" #include "libavutil/avassert.h"
@ -125,8 +126,8 @@ static int webm_chunk_write_header(AVFormatContext *s)
ret = get_chunk_filename(s, 1, oc->filename); ret = get_chunk_filename(s, 1, oc->filename);
if (ret < 0) if (ret < 0)
return ret; return ret;
ret = avio_open2(&oc->pb, oc->filename, AVIO_FLAG_WRITE, ret = ffio_open_whitelist(&oc->pb, oc->filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL); &s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) if (ret < 0)
return ret; return ret;
@ -169,7 +170,8 @@ static int chunk_end(AVFormatContext *s)
ret = get_chunk_filename(s, 0, filename); ret = get_chunk_filename(s, 0, filename);
if (ret < 0) if (ret < 0)
goto fail; goto fail;
ret = avio_open2(&pb, filename, AVIO_FLAG_WRITE, &s->interrupt_callback, NULL); ret = ffio_open_whitelist(&pb, filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL, s->protocol_whitelist);
if (ret < 0) if (ret < 0)
goto fail; goto fail;
avio_write(pb, buffer, buffer_size); avio_write(pb, buffer, buffer_size);

Write Preview
Loading…
Cancel
Save