diff mbox series

[FFmpeg-devel,v1] lavf/hls: add option for hls segment request append http get request parameters

Message ID 20220831071308.85780-1-mirsfang@gmail.com
State New
Headers show
Series [FFmpeg-devel,v1] lavf/hls: add option for hls segment request append http get request parameters | expand

Checks

Context Check Description
yinshiyou/make_loongarch64 success Make finished
yinshiyou/make_fate_loongarch64 success Make fate finished
andriy/make_x86 success Make finished
andriy/make_fate_x86 success Make fate finished

Commit Message

mirs fang Aug. 31, 2022, 7:13 a.m. UTC
From: mirs <mirsfang@gmail.com>

Signed-off-by: mirs <mirsfang@gmail.com>

add option for hls when client reqeust segment url maybe use get parameters for user authenticatio

---
 libavformat/hls.c | 36 +++++++++++++++++++++++++++++-------
 1 file changed, 29 insertions(+), 7 deletions(-)

Comments

Steven Liu Aug. 31, 2022, 7:22 a.m. UTC | #1
<mirsfang@gmail.com> 于2022年8月31日周三 15:16写道:
>
> From: mirs <mirsfang@gmail.com>
>
> Signed-off-by: mirs <mirsfang@gmail.com>
>
> add option for hls when client reqeust segment url maybe use get parameters for user authenticatio
>
> ---
>  libavformat/hls.c | 36 +++++++++++++++++++++++++++++-------
>  1 file changed, 29 insertions(+), 7 deletions(-)
>
> diff --git a/libavformat/hls.c b/libavformat/hls.c
> index 3dc7bd3930..06d0a89a95 100644
> --- a/libavformat/hls.c
> +++ b/libavformat/hls.c
> @@ -220,6 +220,7 @@ typedef struct HLSContext {
>      AVIOInterruptCB *interrupt_callback;
>      AVDictionary *avio_opts;
>      AVDictionary *seg_format_opts;
> +    int seg_append_http_get_params;
>      char *allowed_extensions;
>      int max_reload;
>      int http_persistent;
> @@ -307,6 +308,25 @@ static void free_rendition_list(HLSContext *c)
>      c->n_renditions = 0;
>  }
>
> +static void hls_make_url(const HLSContext * c,char *buf, int size, const char *base,
Do you mean this should name about  params concatenation?
> +                          const char *rel)
> +{
> +    ff_make_absolute_url(buf, size, base, rel);
> +
> +    if (c && c->seg_append_http_get_params && base) {
> +        char * ch = strchr(base, '?');
> +        if (!ch) {
> +            return;
> +        }
> +
> +        if (strchr(buf, '?')) {
> +            ch++;
> +            av_strlcat(buf, "&", size);
> +        }
> +        av_strlcat(buf, ch, size);
> +    }
> +}
> +
>  static struct playlist *new_playlist(HLSContext *c, const char *url,
>                                       const char *base)
>  {
> @@ -318,7 +338,7 @@ static struct playlist *new_playlist(HLSContext *c, const char *url,
>          av_free(pls);
>          return NULL;
>      }
> -    ff_make_absolute_url(pls->url, sizeof(pls->url), base, url);
> +    hls_make_url(c,pls->url, sizeof(pls->url), base, url);
>      if (!pls->url[0]) {
>          av_packet_free(&pls->pkt);
>          av_free(pls);
> @@ -411,7 +431,7 @@ struct init_section_info {
>      char byterange[32];
>  };
>
> -static struct segment *new_init_section(struct playlist *pls,
> +static struct segment *new_init_section(const HLSContext * c,struct playlist *pls,
>                                          struct init_section_info *info,
>                                          const char *url_base)
>  {
> @@ -428,7 +448,7 @@ static struct segment *new_init_section(struct playlist *pls,
>      if (!av_strncasecmp(info->uri, "data:", 5)) {
>          ptr = info->uri;
>      } else {
> -        ff_make_absolute_url(tmp_str, sizeof(tmp_str), url_base, info->uri);
> +        hls_make_url(c,tmp_str, sizeof(tmp_str), url_base, info->uri);
>          if (!tmp_str[0]) {
>              av_free(sec);
>              return NULL;
> @@ -861,7 +881,7 @@ static int parse_playlist(HLSContext *c, const char *url,
>                  goto fail;
>              ff_parse_key_value(ptr, (ff_parse_key_val_cb) handle_init_section_args,
>                                 &info);
> -            cur_init_section = new_init_section(pls, &info, url);
> +            cur_init_section = new_init_section(c,pls, &info, url);
>              if (!cur_init_section) {
>                  ret = AVERROR(ENOMEM);
>                  goto fail;
> @@ -876,7 +896,7 @@ static int parse_playlist(HLSContext *c, const char *url,
>              }
>
>              if (key_type != KEY_NONE) {
> -                ff_make_absolute_url(tmp_str, sizeof(tmp_str), url, key);
> +                hls_make_url(c,tmp_str, sizeof(tmp_str), url, key);
>                  if (!tmp_str[0]) {
>                      av_free(cur_init_section);
>                      ret = AVERROR_INVALIDDATA;
> @@ -948,7 +968,7 @@ static int parse_playlist(HLSContext *c, const char *url,
>                  }
>
>                  if (key_type != KEY_NONE) {
> -                    ff_make_absolute_url(tmp_str, sizeof(tmp_str), url, key);
> +                    hls_make_url(c,tmp_str, sizeof(tmp_str), url, key);
>                      if (!tmp_str[0]) {
>                          ret = AVERROR_INVALIDDATA;
>                          av_free(seg);
> @@ -964,7 +984,7 @@ static int parse_playlist(HLSContext *c, const char *url,
>                      seg->key = NULL;
>                  }
>
> -                ff_make_absolute_url(tmp_str, sizeof(tmp_str), url, line);
> +                hls_make_url(c,tmp_str, sizeof(tmp_str), url, line);
>                  if (!tmp_str[0]) {
>                      ret = AVERROR_INVALIDDATA;
>                      if (seg->key)
> @@ -2548,6 +2568,8 @@ static const AVOption hls_options[] = {
>          OFFSET(http_seekable), AV_OPT_TYPE_BOOL, { .i64 = -1}, -1, 1, FLAGS},
>      {"seg_format_options", "Set options for segment demuxer",
>          OFFSET(seg_format_opts), AV_OPT_TYPE_DICT, {.str = NULL}, 0, 0, FLAGS},
> +     {"seg_append_http_get_params", "Segment request append http get parameters 0 = disable 1 = enable",
> +        OFFSET(seg_append_http_get_params), AV_OPT_TYPE_BOOL,{.i64 = 0}, 0, 1, FLAGS},
>      {NULL}
>  };
>
> --
> 2.32.1 (Apple Git-133)
>
> _______________________________________________
> ffmpeg-devel mailing list
> ffmpeg-devel@ffmpeg.org
> https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
>
> To unsubscribe, visit link above, or email
> ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".
diff mbox series

Patch

diff --git a/libavformat/hls.c b/libavformat/hls.c
index 3dc7bd3930..06d0a89a95 100644
--- a/libavformat/hls.c
+++ b/libavformat/hls.c
@@ -220,6 +220,7 @@  typedef struct HLSContext {
     AVIOInterruptCB *interrupt_callback;
     AVDictionary *avio_opts;
     AVDictionary *seg_format_opts;
+    int seg_append_http_get_params;
     char *allowed_extensions;
     int max_reload;
     int http_persistent;
@@ -307,6 +308,25 @@  static void free_rendition_list(HLSContext *c)
     c->n_renditions = 0;
 }
 
+static void hls_make_url(const HLSContext * c,char *buf, int size, const char *base,
+                          const char *rel)
+{
+    ff_make_absolute_url(buf, size, base, rel);
+
+    if (c && c->seg_append_http_get_params && base) {
+        char * ch = strchr(base, '?');
+        if (!ch) {
+            return;
+        }
+
+        if (strchr(buf, '?')) {
+            ch++;
+            av_strlcat(buf, "&", size);
+        }
+        av_strlcat(buf, ch, size);
+    }
+}
+
 static struct playlist *new_playlist(HLSContext *c, const char *url,
                                      const char *base)
 {
@@ -318,7 +338,7 @@  static struct playlist *new_playlist(HLSContext *c, const char *url,
         av_free(pls);
         return NULL;
     }
-    ff_make_absolute_url(pls->url, sizeof(pls->url), base, url);
+    hls_make_url(c,pls->url, sizeof(pls->url), base, url);
     if (!pls->url[0]) {
         av_packet_free(&pls->pkt);
         av_free(pls);
@@ -411,7 +431,7 @@  struct init_section_info {
     char byterange[32];
 };
 
-static struct segment *new_init_section(struct playlist *pls,
+static struct segment *new_init_section(const HLSContext * c,struct playlist *pls,
                                         struct init_section_info *info,
                                         const char *url_base)
 {
@@ -428,7 +448,7 @@  static struct segment *new_init_section(struct playlist *pls,
     if (!av_strncasecmp(info->uri, "data:", 5)) {
         ptr = info->uri;
     } else {
-        ff_make_absolute_url(tmp_str, sizeof(tmp_str), url_base, info->uri);
+        hls_make_url(c,tmp_str, sizeof(tmp_str), url_base, info->uri);
         if (!tmp_str[0]) {
             av_free(sec);
             return NULL;
@@ -861,7 +881,7 @@  static int parse_playlist(HLSContext *c, const char *url,
                 goto fail;
             ff_parse_key_value(ptr, (ff_parse_key_val_cb) handle_init_section_args,
                                &info);
-            cur_init_section = new_init_section(pls, &info, url);
+            cur_init_section = new_init_section(c,pls, &info, url);
             if (!cur_init_section) {
                 ret = AVERROR(ENOMEM);
                 goto fail;
@@ -876,7 +896,7 @@  static int parse_playlist(HLSContext *c, const char *url,
             }
 
             if (key_type != KEY_NONE) {
-                ff_make_absolute_url(tmp_str, sizeof(tmp_str), url, key);
+                hls_make_url(c,tmp_str, sizeof(tmp_str), url, key);
                 if (!tmp_str[0]) {
                     av_free(cur_init_section);
                     ret = AVERROR_INVALIDDATA;
@@ -948,7 +968,7 @@  static int parse_playlist(HLSContext *c, const char *url,
                 }
 
                 if (key_type != KEY_NONE) {
-                    ff_make_absolute_url(tmp_str, sizeof(tmp_str), url, key);
+                    hls_make_url(c,tmp_str, sizeof(tmp_str), url, key);
                     if (!tmp_str[0]) {
                         ret = AVERROR_INVALIDDATA;
                         av_free(seg);
@@ -964,7 +984,7 @@  static int parse_playlist(HLSContext *c, const char *url,
                     seg->key = NULL;
                 }
 
-                ff_make_absolute_url(tmp_str, sizeof(tmp_str), url, line);
+                hls_make_url(c,tmp_str, sizeof(tmp_str), url, line);
                 if (!tmp_str[0]) {
                     ret = AVERROR_INVALIDDATA;
                     if (seg->key)
@@ -2548,6 +2568,8 @@  static const AVOption hls_options[] = {
         OFFSET(http_seekable), AV_OPT_TYPE_BOOL, { .i64 = -1}, -1, 1, FLAGS},
     {"seg_format_options", "Set options for segment demuxer",
         OFFSET(seg_format_opts), AV_OPT_TYPE_DICT, {.str = NULL}, 0, 0, FLAGS},
+     {"seg_append_http_get_params", "Segment request append http get parameters 0 = disable 1 = enable",
+        OFFSET(seg_append_http_get_params), AV_OPT_TYPE_BOOL,{.i64 = 0}, 0, 1, FLAGS},
     {NULL}
 };