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

[mirsfang at gmail.com]

From: mirs <mirsfang at gmail.com>

Signed-off-by: mirs <mirsfang at 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,
+                          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)