Skip to content

Commit

Permalink
feat(en/aniwave): Use webview hack for vidsrc (aniyomiorg#2762)
Browse files Browse the repository at this point in the history
  • Loading branch information
@Secozzi
Secozzi authored Jan 16, 2024
1 parent 727b4bc commit 8936f77
Show file tree
Hide file tree
Showing 2 changed files with 96 additions and 96 deletions.
2 changes: 1 addition & 1 deletion src/en/aniwave/build.gradle
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ ext {
extName = 'Aniwave'
pkgNameSuffix = 'en.nineanime'
extClass = '.Aniwave'
extVersionCode = 62
extVersionCode = 63
}

dependencies {
Expand Down
190 changes: 95 additions & 95 deletions ...aniwave/src/eu/kanade/tachiyomi/animeextension/en/nineanime/extractors/VidsrcExtractor.kt
View file
Original file line number Diff line number Diff line change
@@ -1,47 +1,42 @@
package eu.kanade.tachiyomi.animeextension.en.nineanime.extractors

import android.util.Base64
import app.cash.quickjs.QuickJs
import android.app.Application
import android.os.Handler
import android.os.Looper
import android.webkit.JavascriptInterface
import android.webkit.WebResourceRequest
import android.webkit.WebResourceResponse
import android.webkit.WebSettings
import android.webkit.WebView
import android.webkit.WebViewClient
import eu.kanade.tachiyomi.animeextension.en.nineanime.MediaResponseBody
import eu.kanade.tachiyomi.animesource.model.Track
import eu.kanade.tachiyomi.animesource.model.Video
import eu.kanade.tachiyomi.lib.playlistutils.PlaylistUtils
import eu.kanade.tachiyomi.network.GET
import eu.kanade.tachiyomi.util.parseAs
import kotlinx.serialization.json.Json
import okhttp3.CacheControl
import okhttp3.Headers
import okhttp3.HttpUrl.Companion.toHttpUrl
import okhttp3.OkHttpClient
import uy.kohesive.injekt.injectLazy
import java.io.ByteArrayInputStream
import java.net.URLDecoder
import javax.crypto.Cipher
import javax.crypto.spec.SecretKeySpec
import java.util.concurrent.CountDownLatch
import java.util.concurrent.TimeUnit

class VidsrcExtractor(private val client: OkHttpClient, private val headers: Headers) {

private val json: Json by injectLazy()

private val playlistUtils by lazy { PlaylistUtils(client, headers) }

private val cacheControl = CacheControl.Builder().noStore().build()
private val noCacheClient = client.newBuilder()
.cache(null)
.build()

private val keys by lazy {
noCacheClient.newCall(
GET("https://raw.githubusercontent.com/Claudemirovsky/worstsource-keys/keys/keys.json", cache = cacheControl),
).execute().parseAs<List<String>>()
}

fun videosFromUrl(embedLink: String, name: String, type: String): List<Video> {
val hosterName = when (name) {
"vidplay" -> "VidPlay"
else -> "MyCloud"
}
val host = embedLink.toHttpUrl().host
val apiUrl = getApiUrl(embedLink, keys)
val apiSlug = runCatching {
extractFromUrl(embedLink)
}.getOrElse { return emptyList() }

val apiHeaders = headers.newBuilder().apply {
add("Accept", "application/json, text/javascript, */*; q=0.01")
Expand All @@ -51,20 +46,10 @@ class VidsrcExtractor(private val client: OkHttpClient, private val headers: Hea
}.build()

val response = client.newCall(
GET(apiUrl, apiHeaders),
GET("https://$host/$apiSlug", apiHeaders),
).execute()

val data = runCatching {
response.parseAs<MediaResponseBody>()
}.getOrElse { // Keys are out of date
val newKeys = noCacheClient.newCall(
GET("https://raw.githubusercontent.com/Claudemirovsky/worstsource-keys/keys/keys.json", cache = cacheControl),
).execute().parseAs<List<String>>()
val newApiUrL = getApiUrl(embedLink, newKeys)
client.newCall(
GET(newApiUrL, apiHeaders),
).execute().parseAs()
}
val data = response.parseAs<MediaResponseBody>()

return playlistUtils.extractFromHls(
data.result.sources.first().file,
Expand All @@ -74,69 +59,6 @@ class VidsrcExtractor(private val client: OkHttpClient, private val headers: Hea
)
}

private fun getApiUrl(embedLink: String, keyList: List<String>): String {
val host = embedLink.toHttpUrl().host
val params = embedLink.toHttpUrl().let { url ->
url.queryParameterNames.map {
Pair(it, url.queryParameter(it) ?: "")
}
}
val vidId = embedLink.substringAfterLast("/").substringBefore("?")
val encodedID = encodeID(vidId, keyList)
val apiSlug = callFromFuToken(host, encodedID)

return buildString {
append("https://")
append(host)
append("/")
append(apiSlug)
if (params.isNotEmpty()) {
append("?")
append(
params.joinToString("&") {
"${it.first}=${it.second}"
},
)
}
}
}

private fun encodeID(videoID: String, keyList: List<String>): String {
val rc4Key1 = SecretKeySpec(keyList[0].toByteArray(), "RC4")
val rc4Key2 = SecretKeySpec(keyList[1].toByteArray(), "RC4")
val cipher1 = Cipher.getInstance("RC4")
val cipher2 = Cipher.getInstance("RC4")
cipher1.init(Cipher.DECRYPT_MODE, rc4Key1, cipher1.parameters)
cipher2.init(Cipher.DECRYPT_MODE, rc4Key2, cipher2.parameters)
var encoded = videoID.toByteArray()

encoded = cipher1.doFinal(encoded)
encoded = cipher2.doFinal(encoded)
encoded = Base64.encode(encoded, Base64.DEFAULT)
return encoded.toString(Charsets.UTF_8).replace("/", "_").trim()
}

private fun callFromFuToken(host: String, data: String): String {
val fuTokenScript = client.newCall(
GET("https://$host/futoken"),
).execute().use { it.body.string() }

val js = buildString {
append("(function")
append(
fuTokenScript.substringAfter("window")
.substringAfter("function")
.replace("jQuery.ajax(", "")
.substringBefore("+location.search"),
)
append("}(\"$data\"))")
}

return QuickJs.create().use {
it.evaluate(js)?.toString()!!
}
}

private fun List<MediaResponseBody.Result.SubTrack>.toTracks(): List<Track> {
return filter {
it.kind == "captions"
Expand All @@ -149,4 +71,82 @@ class VidsrcExtractor(private val client: OkHttpClient, private val headers: Hea
}.getOrNull()
}
}

private val context: Application by injectLazy()
private val handler by lazy { Handler(Looper.getMainLooper()) }

class JsObject(private val latch: CountDownLatch) {
var payload: String = ""

@JavascriptInterface
fun passPayload(passedPayload: String) {
payload = passedPayload
latch.countDown()
}
}

fun extractFromUrl(episodeUrl: String): String {
val latch = CountDownLatch(1)

var webView: WebView? = null

val jsinterface = JsObject(latch)

handler.post {
val webview = WebView(context)

webView = webview
with(webview.settings) {
javaScriptEnabled = true
domStorageEnabled = true
databaseEnabled = true
useWideViewPort = false
loadWithOverviewMode = false
cacheMode = WebSettings.LOAD_NO_CACHE
}

webview.addJavascriptInterface(jsinterface, "ihatetheantichrist")
webview.webViewClient = object : WebViewClient() {
override fun onPageFinished(view: WebView?, url: String?) {
view?.clearCache(true)
view?.clearFormData()
}

override fun shouldInterceptRequest(view: WebView, request: WebResourceRequest): WebResourceResponse? {
val reqUrl = request.url.toString()
if ("futoken" in reqUrl) {
return patchScript(reqUrl)
}
return super.shouldInterceptRequest(view, request)
}
}

webview.loadUrl(episodeUrl)
}

latch.await(5, TimeUnit.SECONDS)

handler.post {
webView?.stopLoading()
webView?.destroy()
webView = null
}

return jsinterface.payload
}

private fun patchScript(scriptUrl: String): WebResourceResponse {
val scriptBody = client.newCall(GET(scriptUrl)).execute().use { it.body.string() }
val newBody = scriptBody.replace("return", "ihatetheantichrist.passPayload('mediainfo/'+a.join(',')+location.search);return")
return WebResourceResponse(
"application/javascript", // mimeType
"utf-8", // encoding
200, // status code
"ok", // reason phrase
mapOf( // response headers
"server" to "cloudflare",
),
ByteArrayInputStream(newBody.toByteArray()), // data
)
}
}

0 comments on commit 8936f77

Please sign in to comment.