blob: 2dfdda892812eea4dc83d77a9ea9991275d20596 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
package com.pitchedapps.frost.web
import android.webkit.WebResourceRequest
import android.webkit.WebResourceResponse
import android.webkit.WebView
import ca.allanwang.kau.kotlin.LazyContext
import ca.allanwang.kau.kotlin.lazyContext
import ca.allanwang.kau.utils.use
import com.pitchedapps.frost.utils.L
import com.pitchedapps.frost.utils.Prefs
import okhttp3.HttpUrl
import java.io.ByteArrayInputStream
/**
* Created by Allan Wang on 2017-07-13.
*
* Handler to decide when a request should be done by us
* This is the crux of Frost's optimizations for the web browser
*/
private val blankResource: WebResourceResponse by lazy { WebResourceResponse("text/plain", "utf-8", ByteArrayInputStream("".toByteArray())) }
//these hosts will redirect to a blank resource
private val blacklistHost: Set<String> =
setOf(
"edge-chat.facebook.com"
)
//these hosts will return null and skip logging
private val whitelistHost: Set<String> =
setOf(
"static.xx.fbcdn.net",
"m.facebook.com",
"touch.facebook.com"
)
//these hosts will skip ad inspection
//this list does not have to include anything from the two above
private val adWhitelistHost: Set<String> =
setOf(
"scontent-sea1-1.xx.fbcdn.net"
)
private val adblock: LazyContext<Set<String>> = lazyContext {
it.assets.open("adblock.txt").bufferedReader().use { it.readLines().toSet() }
}
fun WebView.shouldFrostInterceptRequest(request: WebResourceRequest): WebResourceResponse? {
val httpUrl = HttpUrl.parse(request.url?.toString() ?: return null) ?: return null
val host = httpUrl.host()
val url = httpUrl.toString()
if (blacklistHost.contains(host)) return blankResource
if (whitelistHost.contains(host)) return null
if (!adWhitelistHost.contains(host) && adblock(context).any { url.contains(it) }) return blankResource
if (!shouldLoadImages && !Prefs.loadMediaOnMeteredNetwork && request.isMedia) return blankResource
L.v("Intercept Request", "$host $url")
return null
}
/**
* Wrapper to ensure that null exceptions are not reached
*/
fun WebResourceRequest.query(action: (url: String) -> Boolean): Boolean {
return action(url?.path ?: return false)
}
val WebResourceRequest.isImage: Boolean
get() = query { it.contains(".jpg") || it.contains(".png") }
val WebResourceRequest.isMedia: Boolean
get() = query { it.contains(".jpg") || it.contains(".png") || it.contains("video") }
/**
* Generic filter passthrough
* If Resource is already nonnull, pass it, otherwise check if filter is met and override the response accordingly
*/
fun WebResourceResponse?.filter(request: WebResourceRequest, filter: (url: String) -> Boolean)
= filter(request.query { filter(it) })
fun WebResourceResponse?.filter(filter: Boolean): WebResourceResponse?
= this ?: if (filter) blankResource else null
fun WebResourceResponse?.filterCss(request: WebResourceRequest): WebResourceResponse?
= filter(request) { it.endsWith(".css") }
fun WebResourceResponse?.filterImage(request: WebResourceRequest): WebResourceResponse?
= filter(request.isImage)
|