Commit 81306fd4 authored by Anton Khirnov's avatar Anton Khirnov

hls: eliminate ffurl_* usage

Now all IO should go through the IO callbacks and be interceptable by
the caller.
parent 9f61abc8
...@@ -34,7 +34,6 @@ ...@@ -34,7 +34,6 @@
#include "avformat.h" #include "avformat.h"
#include "internal.h" #include "internal.h"
#include "avio_internal.h" #include "avio_internal.h"
#include "url.h"
#define INITIAL_BUFFER_SIZE 32768 #define INITIAL_BUFFER_SIZE 32768
...@@ -73,7 +72,7 @@ struct variant { ...@@ -73,7 +72,7 @@ struct variant {
char url[MAX_URL_SIZE]; char url[MAX_URL_SIZE];
AVIOContext pb; AVIOContext pb;
uint8_t* read_buffer; uint8_t* read_buffer;
URLContext *input; AVIOContext *input;
AVFormatContext *parent; AVFormatContext *parent;
int index; int index;
AVFormatContext *ctx; AVFormatContext *ctx;
...@@ -133,7 +132,7 @@ static void free_variant_list(HLSContext *c) ...@@ -133,7 +132,7 @@ static void free_variant_list(HLSContext *c)
av_packet_unref(&var->pkt); av_packet_unref(&var->pkt);
av_free(var->pb.buffer); av_free(var->pb.buffer);
if (var->input) if (var->input)
ffurl_close(var->input); ff_format_io_close(c->ctx, &var->input);
if (var->ctx) { if (var->ctx) {
var->ctx->pb = NULL; var->ctx->pb = NULL;
avformat_close_input(&var->ctx); avformat_close_input(&var->ctx);
...@@ -214,31 +213,15 @@ static int open_in(HLSContext *c, AVIOContext **in, const char *url) ...@@ -214,31 +213,15 @@ static int open_in(HLSContext *c, AVIOContext **in, const char *url)
return ret; return ret;
} }
static int url_connect(struct variant *var, AVDictionary *opts) static int open_url(AVFormatContext *s, AVIOContext **pb, const char *url,
const AVDictionary *opts)
{ {
AVDictionary *tmp = NULL; AVDictionary *tmp = NULL;
int ret; int ret;
av_dict_copy(&tmp, opts, 0); av_dict_copy(&tmp, opts, 0);
if ((ret = av_opt_set_dict(var->input, &tmp)) < 0 || ret = s->io_open(s, pb, url, AVIO_FLAG_READ, &tmp);
(ret = ffurl_connect(var->input, NULL)) < 0) {
ffurl_close(var->input);
var->input = NULL;
}
av_dict_free(&tmp);
return ret;
}
static int open_url(HLSContext *c, URLContext **uc, const char *url)
{
AVDictionary *tmp = NULL;
int ret;
av_dict_copy(&tmp, c->avio_opts, 0);
ret = ffurl_open(uc, url, AVIO_FLAG_READ, c->interrupt_callback, &tmp);
av_dict_free(&tmp); av_dict_free(&tmp);
...@@ -377,22 +360,23 @@ fail: ...@@ -377,22 +360,23 @@ fail:
static int open_input(struct variant *var) static int open_input(struct variant *var)
{ {
HLSContext *c = var->parent->priv_data;
struct segment *seg = var->segments[var->cur_seq_no - var->start_seq_no]; struct segment *seg = var->segments[var->cur_seq_no - var->start_seq_no];
if (seg->key_type == KEY_NONE) { if (seg->key_type == KEY_NONE) {
return open_url(var->parent->priv_data, &var->input, seg->url); return open_url(var->parent, &var->input, seg->url, c->avio_opts);
} else if (seg->key_type == KEY_AES_128) { } else if (seg->key_type == KEY_AES_128) {
HLSContext *c = var->parent->priv_data; AVDictionary *opts = NULL;
char iv[33], key[33], url[MAX_URL_SIZE]; char iv[33], key[33], url[MAX_URL_SIZE];
int ret; int ret;
if (strcmp(seg->key, var->key_url)) { if (strcmp(seg->key, var->key_url)) {
URLContext *uc; AVIOContext *pb;
if (open_url(var->parent->priv_data, &uc, seg->key) == 0) { if (open_url(var->parent, &pb, seg->key, c->avio_opts) == 0) {
if (ffurl_read_complete(uc, var->key, sizeof(var->key)) ret = avio_read(pb, var->key, sizeof(var->key));
!= sizeof(var->key)) { if (ret != sizeof(var->key)) {
av_log(NULL, AV_LOG_ERROR, "Unable to read key file %s\n", av_log(NULL, AV_LOG_ERROR, "Unable to read key file %s\n",
seg->key); seg->key);
} }
ffurl_close(uc); ff_format_io_close(var->parent, &pb);
} else { } else {
av_log(NULL, AV_LOG_ERROR, "Unable to open key file %s\n", av_log(NULL, AV_LOG_ERROR, "Unable to open key file %s\n",
seg->key); seg->key);
...@@ -406,13 +390,14 @@ static int open_input(struct variant *var) ...@@ -406,13 +390,14 @@ static int open_input(struct variant *var)
snprintf(url, sizeof(url), "crypto+%s", seg->url); snprintf(url, sizeof(url), "crypto+%s", seg->url);
else else
snprintf(url, sizeof(url), "crypto:%s", seg->url); snprintf(url, sizeof(url), "crypto:%s", seg->url);
if ((ret = ffurl_alloc(&var->input, url, AVIO_FLAG_READ,
&var->parent->interrupt_callback)) < 0)
return ret;
av_opt_set(var->input->priv_data, "key", key, 0);
av_opt_set(var->input->priv_data, "iv", iv, 0);
return url_connect(var, c->avio_opts); av_dict_copy(&opts, c->avio_opts, 0);
av_dict_set(&opts, "key", key, 0);
av_dict_set(&opts, "iv", iv, 0);
ret = open_url(var->parent, &var->input, url, opts);
av_dict_free(&opts);
return ret;
} }
return AVERROR(ENOSYS); return AVERROR(ENOSYS);
} }
...@@ -463,11 +448,10 @@ reload: ...@@ -463,11 +448,10 @@ reload:
if (ret < 0) if (ret < 0)
return ret; return ret;
} }
ret = ffurl_read(v->input, buf, buf_size); ret = avio_read(v->input, buf, buf_size);
if (ret > 0) if (ret > 0)
return ret; return ret;
ffurl_close(v->input); ff_format_io_close(c->ctx, &v->input);
v->input = NULL;
v->cur_seq_no++; v->cur_seq_no++;
c->end_of_segment = 1; c->end_of_segment = 1;
...@@ -665,8 +649,7 @@ static int recheck_discard_flags(AVFormatContext *s, int first) ...@@ -665,8 +649,7 @@ static int recheck_discard_flags(AVFormatContext *s, int first)
av_log(s, AV_LOG_INFO, "Now receiving variant %d\n", i); av_log(s, AV_LOG_INFO, "Now receiving variant %d\n", i);
} else if (first && !v->cur_needed && v->needed) { } else if (first && !v->cur_needed && v->needed) {
if (v->input) if (v->input)
ffurl_close(v->input); ff_format_io_close(s, &v->input);
v->input = NULL;
v->needed = 0; v->needed = 0;
changed = 1; changed = 1;
av_log(s, AV_LOG_INFO, "No longer receiving variant %d\n", i); av_log(s, AV_LOG_INFO, "No longer receiving variant %d\n", i);
...@@ -813,10 +796,8 @@ static int hls_read_seek(AVFormatContext *s, int stream_index, ...@@ -813,10 +796,8 @@ static int hls_read_seek(AVFormatContext *s, int stream_index,
struct variant *var = c->variants[i]; struct variant *var = c->variants[i];
int64_t pos = c->first_timestamp == AV_NOPTS_VALUE ? int64_t pos = c->first_timestamp == AV_NOPTS_VALUE ?
0 : c->first_timestamp; 0 : c->first_timestamp;
if (var->input) { if (var->input)
ffurl_close(var->input); ff_format_io_close(s, &var->input);
var->input = NULL;
}
av_packet_unref(&var->pkt); av_packet_unref(&var->pkt);
reset_packet(&var->pkt); reset_packet(&var->pkt);
var->pb.eof_reached = 0; var->pb.eof_reached = 0;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment