From c636d2e78e464246b3dd7bc8ce0b8e5d8a5bf7a6 Mon Sep 17 00:00:00 2001
From: Mole Shang <135e2@135e2.dev>
Date: Wed, 9 Aug 2023 17:29:15 +0800
Subject: process_url: set referer in the separate function

---
 src/extractors/bilibili.c | 8 ++++----
 src/extractors/haokan.c   | 3 ++-
 2 files changed, 6 insertions(+), 5 deletions(-)

(limited to 'src/extractors')

diff --git a/src/extractors/bilibili.c b/src/extractors/bilibili.c
index aa3fc4e..5b98f26 100644
--- a/src/extractors/bilibili.c
+++ b/src/extractors/bilibili.c
@@ -404,16 +404,16 @@ static int download(Bilibili_options *bilibili_options) {
     callback_struct.videofn = malloc(strlen(callback_struct.filename) + 6);
     sprintf(callback_struct.videofn, "%s[%s]-%s.%s", bilibili_options->title,
             quality_desc, "video", ext);
-    add_url(video->baseUrl, NULL, callback_struct.videofn,
-            "https://www.bilibili.com", NULL, NULL);
+    set_referer("https://www.bilibili.com");
+    add_url(video->baseUrl, NULL, callback_struct.videofn, NULL, NULL);
   }
 
   {
     callback_struct.audiofn = malloc(strlen(callback_struct.filename) + 6);
     sprintf(callback_struct.audiofn, "%s[%s]-%s.%s", bilibili_options->title,
             quality_desc, "audio", mimeType2ext(audio->mimeType));
-    add_url(audio->baseUrl, NULL, callback_struct.audiofn,
-            "https://www.bilibili.com", &bilibili_merge, &callback_struct);
+    add_url(audio->baseUrl, NULL, callback_struct.audiofn, &bilibili_merge,
+            &callback_struct);
   }
   free_and_nullify(resp);
   dash_cleanup(&dash);
diff --git a/src/extractors/haokan.c b/src/extractors/haokan.c
index 5659018..f289c3e 100644
--- a/src/extractors/haokan.c
+++ b/src/extractors/haokan.c
@@ -33,7 +33,8 @@ void haokan_extract(Options *options) {
       char *filename = malloc(strlen(title) + strlen(ct) + 2);
       sprintf(filename, "%s.%s", title, ext);
 
-      add_url(videoURL, NULL, filename, "https://haokan.baidu.com", NULL, NULL);
+      set_referer("https://haokan.baidu.com");
+      add_url(videoURL, NULL, filename, NULL, NULL);
 
     end:
       free_str_array(&results);
-- 
cgit v1.2.3