feat: salt

cssxsh · May 23, 2023 · 7a78d69 · 7a78d69
1 parent e20d89e
commit 7a78d69
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/main/kotlin/xyz/cssxsh/arknights/bilibili/VideoDataHolder.kt b/src/main/kotlin/xyz/cssxsh/arknights/bilibili/VideoDataHolder.kt
@@ -11,6 +11,7 @@ import kotlinx.serialization.*
 import kotlinx.serialization.json.*
 import xyz.cssxsh.arknights.*
 import java.io.*
+import java.security.*
 import java.util.*
 import kotlin.collections.*
 import kotlin.properties.*
@@ -26,6 +27,16 @@ public class VideoDataHolder(override val folder: File, override val ignore: sus
 
     private val AcceptAllCookiesStorage.container: MutableList<Cookie> by reflect()
 
+    private val salt: String
+        get() = kotlin.run {
+            val file = File("data/xyz.cssxsh.mirai.plugin.bilibili-helper/salt.txt")
+            if (file.exists()) {
+                file.readText()
+            } else {
+                ""
+            }
+        }
+
     init {
         http.launch {
             val cookies = with(File("data/xyz.cssxsh.mirai.plugin.bilibili-helper/cookies.json")) {
@@ -51,6 +62,16 @@ public class VideoDataHolder(override val folder: File, override val ignore: sus
                 parameter("order", "pubdate")
                 parameter("tid", key.tid)
                 parameter("jsonp", "jsonp")
+                parameter("wts", System.currentTimeMillis() / 1000)
+
+                val digest = MessageDigest.getInstance("MD5")
+                val parameters = url.parameters.entries()
+                    .flatMap { e -> e.value.map { e.key to it } }
+                    .sortedBy { it.first }.formUrlEncode()
+
+                val md5 = digest.digest((parameters + salt).encodeToByteArray())
+
+                parameter("w_rid", md5.joinToString("") { "%02x".format(it) })
 
                 header(HttpHeaders.Origin, "https://space.bilibili.com")
                 header(HttpHeaders.Referrer, "https://space.bilibili.com/161775300/video")