Commit 25c7db7c authored by Derek Buitenhuis's avatar Derek Buitenhuis

avio: Check for memory allocation failure of private data

Signed-off-by: 's avatarDerek Buitenhuis <derek.buitenhuis@gmail.com>
parent 327c439f
...@@ -135,6 +135,10 @@ static int url_alloc_for_protocol(URLContext **puc, struct URLProtocol *up, ...@@ -135,6 +135,10 @@ static int url_alloc_for_protocol(URLContext **puc, struct URLProtocol *up,
uc->max_packet_size = 0; /* default: stream file */ uc->max_packet_size = 0; /* default: stream file */
if (up->priv_data_size) { if (up->priv_data_size) {
uc->priv_data = av_mallocz(up->priv_data_size); uc->priv_data = av_mallocz(up->priv_data_size);
if (!uc->priv_data) {
err = AVERROR(ENOMEM);
goto fail;
}
if (up->priv_data_class) { if (up->priv_data_class) {
*(const AVClass **)uc->priv_data = up->priv_data_class; *(const AVClass **)uc->priv_data = up->priv_data_class;
av_opt_set_defaults(uc->priv_data); av_opt_set_defaults(uc->priv_data);
...@@ -147,6 +151,9 @@ static int url_alloc_for_protocol(URLContext **puc, struct URLProtocol *up, ...@@ -147,6 +151,9 @@ static int url_alloc_for_protocol(URLContext **puc, struct URLProtocol *up,
return 0; return 0;
fail: fail:
*puc = NULL; *puc = NULL;
if (uc)
av_freep(&uc->priv_data);
av_freep(&uc);
#if CONFIG_NETWORK #if CONFIG_NETWORK
if (up->flags & URL_PROTOCOL_FLAG_NETWORK) if (up->flags & URL_PROTOCOL_FLAG_NETWORK)
ff_network_close(); ff_network_close();
......
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