diff mbox series

[FFmpeg-devel,v1] avformat/aviobuf: ffio_copy_url_options

Message ID 20211215003514.23851-1-pal@sandflow.com
State Accepted
Commit c8b5f2848dcdc7103a5b85c50c4c3082382d1f82
Headers show
Series [FFmpeg-devel,v1] avformat/aviobuf: ffio_copy_url_options | expand

Checks

Context Check Description
andriy/make_x86 success Make finished
andriy/make_fate_x86 success Make fate finished
andriy/make_ppc success Make finished
andriy/make_fate_ppc success Make fate finished

Commit Message

Pierre-Anthony Lemieux Dec. 15, 2021, 12:35 a.m. UTC
From: Pierre-Anthony Lemieux <pal@palemieux.com>

Signed-off-by: Pierre-Anthony Lemieux <pal@palemieux.com>
---

Notes:
    Refactors save_avio_options() from dashdec.c and hls.c
    into a common ffio_copy_url_options() in libavformat/aviobuf.c.
    
    Co-authored: Nicholas Vanderzwet <https://github.com/nvanderzwet-ssimwave>

 libavformat/avio_internal.h |  6 ++++++
 libavformat/aviobuf.c       | 24 ++++++++++++++++++++++++
 libavformat/dashdec.c       | 27 +--------------------------
 libavformat/hls.c           | 24 +-----------------------
 4 files changed, 32 insertions(+), 49 deletions(-)

Comments

Zane van Iperen Dec. 16, 2021, 8:09 a.m. UTC | #1
Dedup is always good, lgtm.


On 15/12/21 10:35, pal@sandflow.com wrote:
> From: Pierre-Anthony Lemieux <pal@palemieux.com>
> 
> Signed-off-by: Pierre-Anthony Lemieux <pal@palemieux.com>
> ---
> 
> Notes:
>      Refactors save_avio_options() from dashdec.c and hls.c
>      into a common ffio_copy_url_options() in libavformat/aviobuf.c.
>      
>      Co-authored: Nicholas Vanderzwet <https://github.com/nvanderzwet-ssimwave>
> 
>   libavformat/avio_internal.h |  6 ++++++
>   libavformat/aviobuf.c       | 24 ++++++++++++++++++++++++
>   libavformat/dashdec.c       | 27 +--------------------------
>   libavformat/hls.c           | 24 +-----------------------
>   4 files changed, 32 insertions(+), 49 deletions(-)
> 
> diff --git a/libavformat/avio_internal.h b/libavformat/avio_internal.h
> index 187433f283..1f5e3d474b 100644
> --- a/libavformat/avio_internal.h
> +++ b/libavformat/avio_internal.h
> @@ -206,6 +206,12 @@ int ffio_fdopen(AVIOContext **s, URLContext *h);
>    */
>   URLContext *ffio_geturlcontext(AVIOContext *s);
>   
> +
> +/**
> + * Read url related dictionary options from the AVIOContext and write to the given dictionary
> + */
> +int ffio_copy_url_options(AVIOContext* pb, AVDictionary** avio_opts);
> +
>   /**
>    * Open a write-only fake memory stream. The written data is not stored
>    * anywhere - this is only used for measuring the amount of data
> diff --git a/libavformat/aviobuf.c b/libavformat/aviobuf.c
> index 969c127b23..096f37ae23 100644
> --- a/libavformat/aviobuf.c
> +++ b/libavformat/aviobuf.c
> @@ -1022,6 +1022,30 @@ URLContext* ffio_geturlcontext(AVIOContext *s)
>           return NULL;
>   }
>   
> +int ffio_copy_url_options(AVIOContext* pb, AVDictionary** avio_opts)
> +{
> +    const char *opts[] = {
> +        "headers", "user_agent", "cookies", "http_proxy", "referer", "rw_timeout", "icy", NULL };
> +    const char **opt = opts;
> +    uint8_t *buf = NULL;
> +    int ret = 0;
> +
> +    while (*opt) {
> +        if (av_opt_get(pb, *opt, AV_OPT_SEARCH_CHILDREN, &buf) >= 0) {
> +            if (buf[0] != '\0') {
> +                ret = av_dict_set(avio_opts, *opt, buf, AV_DICT_DONT_STRDUP_VAL);
> +                if (ret < 0)
> +                    return ret;
> +            } else {
> +                av_freep(&buf);
> +            }
> +        }
> +        opt++;
> +    }
> +
> +    return ret;
> +}
> +
>   static void update_checksum(AVIOContext *s)
>   {
>       if (s->update_checksum && s->buf_ptr > s->checksum_ptr) {
> diff --git a/libavformat/dashdec.c b/libavformat/dashdec.c
> index 983dc85d65..797fe74157 100644
> --- a/libavformat/dashdec.c
> +++ b/libavformat/dashdec.c
> @@ -1833,31 +1833,6 @@ end:
>       return ret;
>   }
>   
> -static int save_avio_options(AVFormatContext *s)
> -{
> -    DASHContext *c = s->priv_data;
> -    const char *opts[] = {
> -        "headers", "user_agent", "cookies", "http_proxy", "referer", "rw_timeout", "icy", NULL };
> -    const char **opt = opts;
> -    uint8_t *buf = NULL;
> -    int ret = 0;
> -
> -    while (*opt) {
> -        if (av_opt_get(s->pb, *opt, AV_OPT_SEARCH_CHILDREN, &buf) >= 0) {
> -            if (buf[0] != '\0') {
> -                ret = av_dict_set(&c->avio_opts, *opt, buf, AV_DICT_DONT_STRDUP_VAL);
> -                if (ret < 0)
> -                    return ret;
> -            } else {
> -                av_freep(&buf);
> -            }
> -        }
> -        opt++;
> -    }
> -
> -    return ret;
> -}
> -
>   static int nested_io_open(AVFormatContext *s, AVIOContext **pb, const char *url,
>                             int flags, AVDictionary **opts)
>   {
> @@ -2057,7 +2032,7 @@ static int dash_read_header(AVFormatContext *s)
>   
>       c->interrupt_callback = &s->interrupt_callback;
>   
> -    if ((ret = save_avio_options(s)) < 0)
> +    if ((ret = ffio_copy_url_options(s->pb, &c->avio_opts)) < 0)
>           return ret;
>   
>       if ((ret = parse_manifest(s, s->url, s->pb)) < 0)
> diff --git a/libavformat/hls.c b/libavformat/hls.c
> index 557faf8e8d..8c526f748f 100644
> --- a/libavformat/hls.c
> +++ b/libavformat/hls.c
> @@ -1717,28 +1717,6 @@ static int64_t select_cur_seq_no(HLSContext *c, struct playlist *pls)
>       return pls->start_seq_no;
>   }
>   
> -static int save_avio_options(AVFormatContext *s)
> -{
> -    HLSContext *c = s->priv_data;
> -    static const char * const opts[] = {
> -        "headers", "http_proxy", "user_agent", "cookies", "referer", "rw_timeout", "icy", NULL };
> -    const char * const * opt = opts;
> -    uint8_t *buf;
> -    int ret = 0;
> -
> -    while (*opt) {
> -        if (av_opt_get(s->pb, *opt, AV_OPT_SEARCH_CHILDREN | AV_OPT_ALLOW_NULL, &buf) >= 0) {
> -            ret = av_dict_set(&c->avio_opts, *opt, buf,
> -                              AV_DICT_DONT_STRDUP_VAL);
> -            if (ret < 0)
> -                return ret;
> -        }
> -        opt++;
> -    }
> -
> -    return ret;
> -}
> -
>   static int nested_io_open(AVFormatContext *s, AVIOContext **pb, const char *url,
>                             int flags, AVDictionary **opts)
>   {
> @@ -1884,7 +1862,7 @@ static int hls_read_header(AVFormatContext *s)
>       c->first_timestamp = AV_NOPTS_VALUE;
>       c->cur_timestamp = AV_NOPTS_VALUE;
>   
> -    if ((ret = save_avio_options(s)) < 0)
> +    if ((ret = ffio_copy_url_options(s->pb, &c->avio_opts)) < 0)
>           return ret;
>   
>       /* XXX: Some HLS servers don't like being sent the range header,
>
Zane van Iperen Dec. 17, 2021, 11:26 a.m. UTC | #2
Will apply tomorrow unless there are objections.



On 15/12/21 10:35, pal@sandflow.com wrote:
> From: Pierre-Anthony Lemieux <pal@palemieux.com>
> 
> Signed-off-by: Pierre-Anthony Lemieux <pal@palemieux.com>
> ---
> 
> Notes:
>      Refactors save_avio_options() from dashdec.c and hls.c
>      into a common ffio_copy_url_options() in libavformat/aviobuf.c.
>      
>      Co-authored: Nicholas Vanderzwet <https://github.com/nvanderzwet-ssimwave>
> 
>   libavformat/avio_internal.h |  6 ++++++
>   libavformat/aviobuf.c       | 24 ++++++++++++++++++++++++
>   libavformat/dashdec.c       | 27 +--------------------------
>   libavformat/hls.c           | 24 +-----------------------
>   4 files changed, 32 insertions(+), 49 deletions(-)
> 
> diff --git a/libavformat/avio_internal.h b/libavformat/avio_internal.h
> index 187433f283..1f5e3d474b 100644
> --- a/libavformat/avio_internal.h
> +++ b/libavformat/avio_internal.h
> @@ -206,6 +206,12 @@ int ffio_fdopen(AVIOContext **s, URLContext *h);
>    */
>   URLContext *ffio_geturlcontext(AVIOContext *s);
>   
> +
> +/**
> + * Read url related dictionary options from the AVIOContext and write to the given dictionary
> + */
> +int ffio_copy_url_options(AVIOContext* pb, AVDictionary** avio_opts);
> +
>   /**
>    * Open a write-only fake memory stream. The written data is not stored
>    * anywhere - this is only used for measuring the amount of data
> diff --git a/libavformat/aviobuf.c b/libavformat/aviobuf.c
> index 969c127b23..096f37ae23 100644
> --- a/libavformat/aviobuf.c
> +++ b/libavformat/aviobuf.c
> @@ -1022,6 +1022,30 @@ URLContext* ffio_geturlcontext(AVIOContext *s)
>           return NULL;
>   }
>   
> +int ffio_copy_url_options(AVIOContext* pb, AVDictionary** avio_opts)
> +{
> +    const char *opts[] = {
> +        "headers", "user_agent", "cookies", "http_proxy", "referer", "rw_timeout", "icy", NULL };
> +    const char **opt = opts;
> +    uint8_t *buf = NULL;
> +    int ret = 0;
> +
> +    while (*opt) {
> +        if (av_opt_get(pb, *opt, AV_OPT_SEARCH_CHILDREN, &buf) >= 0) {
> +            if (buf[0] != '\0') {
> +                ret = av_dict_set(avio_opts, *opt, buf, AV_DICT_DONT_STRDUP_VAL);
> +                if (ret < 0)
> +                    return ret;
> +            } else {
> +                av_freep(&buf);
> +            }
> +        }
> +        opt++;
> +    }
> +
> +    return ret;
> +}
> +
>   static void update_checksum(AVIOContext *s)
>   {
>       if (s->update_checksum && s->buf_ptr > s->checksum_ptr) {
> diff --git a/libavformat/dashdec.c b/libavformat/dashdec.c
> index 983dc85d65..797fe74157 100644
> --- a/libavformat/dashdec.c
> +++ b/libavformat/dashdec.c
> @@ -1833,31 +1833,6 @@ end:
>       return ret;
>   }
>   
> -static int save_avio_options(AVFormatContext *s)
> -{
> -    DASHContext *c = s->priv_data;
> -    const char *opts[] = {
> -        "headers", "user_agent", "cookies", "http_proxy", "referer", "rw_timeout", "icy", NULL };
> -    const char **opt = opts;
> -    uint8_t *buf = NULL;
> -    int ret = 0;
> -
> -    while (*opt) {
> -        if (av_opt_get(s->pb, *opt, AV_OPT_SEARCH_CHILDREN, &buf) >= 0) {
> -            if (buf[0] != '\0') {
> -                ret = av_dict_set(&c->avio_opts, *opt, buf, AV_DICT_DONT_STRDUP_VAL);
> -                if (ret < 0)
> -                    return ret;
> -            } else {
> -                av_freep(&buf);
> -            }
> -        }
> -        opt++;
> -    }
> -
> -    return ret;
> -}
> -
>   static int nested_io_open(AVFormatContext *s, AVIOContext **pb, const char *url,
>                             int flags, AVDictionary **opts)
>   {
> @@ -2057,7 +2032,7 @@ static int dash_read_header(AVFormatContext *s)
>   
>       c->interrupt_callback = &s->interrupt_callback;
>   
> -    if ((ret = save_avio_options(s)) < 0)
> +    if ((ret = ffio_copy_url_options(s->pb, &c->avio_opts)) < 0)
>           return ret;
>   
>       if ((ret = parse_manifest(s, s->url, s->pb)) < 0)
> diff --git a/libavformat/hls.c b/libavformat/hls.c
> index 557faf8e8d..8c526f748f 100644
> --- a/libavformat/hls.c
> +++ b/libavformat/hls.c
> @@ -1717,28 +1717,6 @@ static int64_t select_cur_seq_no(HLSContext *c, struct playlist *pls)
>       return pls->start_seq_no;
>   }
>   
> -static int save_avio_options(AVFormatContext *s)
> -{
> -    HLSContext *c = s->priv_data;
> -    static const char * const opts[] = {
> -        "headers", "http_proxy", "user_agent", "cookies", "referer", "rw_timeout", "icy", NULL };
> -    const char * const * opt = opts;
> -    uint8_t *buf;
> -    int ret = 0;
> -
> -    while (*opt) {
> -        if (av_opt_get(s->pb, *opt, AV_OPT_SEARCH_CHILDREN | AV_OPT_ALLOW_NULL, &buf) >= 0) {
> -            ret = av_dict_set(&c->avio_opts, *opt, buf,
> -                              AV_DICT_DONT_STRDUP_VAL);
> -            if (ret < 0)
> -                return ret;
> -        }
> -        opt++;
> -    }
> -
> -    return ret;
> -}
> -
>   static int nested_io_open(AVFormatContext *s, AVIOContext **pb, const char *url,
>                             int flags, AVDictionary **opts)
>   {
> @@ -1884,7 +1862,7 @@ static int hls_read_header(AVFormatContext *s)
>       c->first_timestamp = AV_NOPTS_VALUE;
>       c->cur_timestamp = AV_NOPTS_VALUE;
>   
> -    if ((ret = save_avio_options(s)) < 0)
> +    if ((ret = ffio_copy_url_options(s->pb, &c->avio_opts)) < 0)
>           return ret;
>   
>       /* XXX: Some HLS servers don't like being sent the range header,
>
diff mbox series

Patch

diff --git a/libavformat/avio_internal.h b/libavformat/avio_internal.h
index 187433f283..1f5e3d474b 100644
--- a/libavformat/avio_internal.h
+++ b/libavformat/avio_internal.h
@@ -206,6 +206,12 @@  int ffio_fdopen(AVIOContext **s, URLContext *h);
  */
 URLContext *ffio_geturlcontext(AVIOContext *s);
 
+
+/**
+ * Read url related dictionary options from the AVIOContext and write to the given dictionary
+ */
+int ffio_copy_url_options(AVIOContext* pb, AVDictionary** avio_opts);
+
 /**
  * Open a write-only fake memory stream. The written data is not stored
  * anywhere - this is only used for measuring the amount of data
diff --git a/libavformat/aviobuf.c b/libavformat/aviobuf.c
index 969c127b23..096f37ae23 100644
--- a/libavformat/aviobuf.c
+++ b/libavformat/aviobuf.c
@@ -1022,6 +1022,30 @@  URLContext* ffio_geturlcontext(AVIOContext *s)
         return NULL;
 }
 
+int ffio_copy_url_options(AVIOContext* pb, AVDictionary** avio_opts)
+{
+    const char *opts[] = {
+        "headers", "user_agent", "cookies", "http_proxy", "referer", "rw_timeout", "icy", NULL };
+    const char **opt = opts;
+    uint8_t *buf = NULL;
+    int ret = 0;
+
+    while (*opt) {
+        if (av_opt_get(pb, *opt, AV_OPT_SEARCH_CHILDREN, &buf) >= 0) {
+            if (buf[0] != '\0') {
+                ret = av_dict_set(avio_opts, *opt, buf, AV_DICT_DONT_STRDUP_VAL);
+                if (ret < 0)
+                    return ret;
+            } else {
+                av_freep(&buf);
+            }
+        }
+        opt++;
+    }
+
+    return ret;
+}
+
 static void update_checksum(AVIOContext *s)
 {
     if (s->update_checksum && s->buf_ptr > s->checksum_ptr) {
diff --git a/libavformat/dashdec.c b/libavformat/dashdec.c
index 983dc85d65..797fe74157 100644
--- a/libavformat/dashdec.c
+++ b/libavformat/dashdec.c
@@ -1833,31 +1833,6 @@  end:
     return ret;
 }
 
-static int save_avio_options(AVFormatContext *s)
-{
-    DASHContext *c = s->priv_data;
-    const char *opts[] = {
-        "headers", "user_agent", "cookies", "http_proxy", "referer", "rw_timeout", "icy", NULL };
-    const char **opt = opts;
-    uint8_t *buf = NULL;
-    int ret = 0;
-
-    while (*opt) {
-        if (av_opt_get(s->pb, *opt, AV_OPT_SEARCH_CHILDREN, &buf) >= 0) {
-            if (buf[0] != '\0') {
-                ret = av_dict_set(&c->avio_opts, *opt, buf, AV_DICT_DONT_STRDUP_VAL);
-                if (ret < 0)
-                    return ret;
-            } else {
-                av_freep(&buf);
-            }
-        }
-        opt++;
-    }
-
-    return ret;
-}
-
 static int nested_io_open(AVFormatContext *s, AVIOContext **pb, const char *url,
                           int flags, AVDictionary **opts)
 {
@@ -2057,7 +2032,7 @@  static int dash_read_header(AVFormatContext *s)
 
     c->interrupt_callback = &s->interrupt_callback;
 
-    if ((ret = save_avio_options(s)) < 0)
+    if ((ret = ffio_copy_url_options(s->pb, &c->avio_opts)) < 0)
         return ret;
 
     if ((ret = parse_manifest(s, s->url, s->pb)) < 0)
diff --git a/libavformat/hls.c b/libavformat/hls.c
index 557faf8e8d..8c526f748f 100644
--- a/libavformat/hls.c
+++ b/libavformat/hls.c
@@ -1717,28 +1717,6 @@  static int64_t select_cur_seq_no(HLSContext *c, struct playlist *pls)
     return pls->start_seq_no;
 }
 
-static int save_avio_options(AVFormatContext *s)
-{
-    HLSContext *c = s->priv_data;
-    static const char * const opts[] = {
-        "headers", "http_proxy", "user_agent", "cookies", "referer", "rw_timeout", "icy", NULL };
-    const char * const * opt = opts;
-    uint8_t *buf;
-    int ret = 0;
-
-    while (*opt) {
-        if (av_opt_get(s->pb, *opt, AV_OPT_SEARCH_CHILDREN | AV_OPT_ALLOW_NULL, &buf) >= 0) {
-            ret = av_dict_set(&c->avio_opts, *opt, buf,
-                              AV_DICT_DONT_STRDUP_VAL);
-            if (ret < 0)
-                return ret;
-        }
-        opt++;
-    }
-
-    return ret;
-}
-
 static int nested_io_open(AVFormatContext *s, AVIOContext **pb, const char *url,
                           int flags, AVDictionary **opts)
 {
@@ -1884,7 +1862,7 @@  static int hls_read_header(AVFormatContext *s)
     c->first_timestamp = AV_NOPTS_VALUE;
     c->cur_timestamp = AV_NOPTS_VALUE;
 
-    if ((ret = save_avio_options(s)) < 0)
+    if ((ret = ffio_copy_url_options(s->pb, &c->avio_opts)) < 0)
         return ret;
 
     /* XXX: Some HLS servers don't like being sent the range header,