Invinciblelee
6 years ago
parent
654d510efd
commit
b8bc0e8b59
@ -0,0 +1,58 @@ |
|||||||
|
package io.legado.app.base |
||||||
|
|
||||||
|
import android.app.Application |
||||||
|
import androidx.lifecycle.AndroidViewModel |
||||||
|
import kotlinx.coroutines.* |
||||||
|
import kotlin.coroutines.CoroutineContext |
||||||
|
|
||||||
|
open class BaseViewModel(application: Application) : AndroidViewModel(application), CoroutineScope { |
||||||
|
override val coroutineContext: CoroutineContext |
||||||
|
get() = Dispatchers.Main |
||||||
|
|
||||||
|
|
||||||
|
private val launchManager: MutableList<Job> = mutableListOf() |
||||||
|
|
||||||
|
protected fun launchOnUI( |
||||||
|
tryBlock: suspend CoroutineScope.() -> Unit, |
||||||
|
cacheBlock: suspend CoroutineScope.(Throwable) -> Unit, |
||||||
|
finallyBlock: suspend CoroutineScope.() -> Unit, |
||||||
|
handleCancellationExceptionManually: Boolean |
||||||
|
) { |
||||||
|
launchOnUI { |
||||||
|
tryCatch(tryBlock, cacheBlock, finallyBlock, handleCancellationExceptionManually) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* add launch task to [launchManager] |
||||||
|
*/ |
||||||
|
private fun launchOnUI(block: suspend CoroutineScope.() -> Unit) { |
||||||
|
val job = launch { block() } |
||||||
|
launchManager.add(job) |
||||||
|
job.invokeOnCompletion { launchManager.remove(job) } |
||||||
|
} |
||||||
|
|
||||||
|
private suspend fun tryCatch( |
||||||
|
tryBlock: suspend CoroutineScope.() -> Unit, |
||||||
|
catchBlock: suspend CoroutineScope.(Throwable) -> Unit, |
||||||
|
finallyBlock: suspend CoroutineScope.() -> Unit, |
||||||
|
handleCancellationExceptionManually: Boolean = false |
||||||
|
) { |
||||||
|
try { |
||||||
|
coroutineScope { tryBlock() } |
||||||
|
} catch (e: Throwable) { |
||||||
|
if (e !is CancellationException || handleCancellationExceptionManually) { |
||||||
|
coroutineScope { catchBlock(e) } |
||||||
|
} else { |
||||||
|
throw e |
||||||
|
} |
||||||
|
} finally { |
||||||
|
coroutineScope { finallyBlock() } |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
override fun onCleared() { |
||||||
|
super.onCleared() |
||||||
|
launchManager.clear() |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,107 @@ |
|||||||
|
package io.legado.app.help.http |
||||||
|
|
||||||
|
import kotlinx.coroutines.CompletableDeferred |
||||||
|
import kotlinx.coroutines.Deferred |
||||||
|
import retrofit2.* |
||||||
|
import java.lang.reflect.ParameterizedType |
||||||
|
import java.lang.reflect.Type |
||||||
|
|
||||||
|
class CoroutineCallAdapterFactory private constructor() : CallAdapter.Factory() { |
||||||
|
companion object { |
||||||
|
@JvmStatic @JvmName("create") |
||||||
|
operator fun invoke() = CoroutineCallAdapterFactory() |
||||||
|
} |
||||||
|
|
||||||
|
override fun get( |
||||||
|
returnType: Type, |
||||||
|
annotations: Array<out Annotation>, |
||||||
|
retrofit: Retrofit |
||||||
|
): CallAdapter<*, *>? { |
||||||
|
if (Deferred::class.java != getRawType(returnType)) { |
||||||
|
return null |
||||||
|
} |
||||||
|
if (returnType !is ParameterizedType) { |
||||||
|
throw IllegalStateException( |
||||||
|
"Deferred return type must be parameterized as Deferred<Foo> or Deferred<out Foo>") |
||||||
|
} |
||||||
|
val responseType = getParameterUpperBound(0, returnType) |
||||||
|
|
||||||
|
val rawDeferredType = getRawType(responseType) |
||||||
|
return if (rawDeferredType == Response::class.java) { |
||||||
|
if (responseType !is ParameterizedType) { |
||||||
|
throw IllegalStateException( |
||||||
|
"Response must be parameterized as Response<Foo> or Response<out Foo>") |
||||||
|
} |
||||||
|
ResponseCallAdapter<Any>( |
||||||
|
getParameterUpperBound( |
||||||
|
0, |
||||||
|
responseType |
||||||
|
) |
||||||
|
) |
||||||
|
} else { |
||||||
|
BodyCallAdapter<Any>(responseType) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private class BodyCallAdapter<T>( |
||||||
|
private val responseType: Type |
||||||
|
) : CallAdapter<T, Deferred<T>> { |
||||||
|
|
||||||
|
override fun responseType() = responseType |
||||||
|
|
||||||
|
override fun adapt(call: Call<T>): Deferred<T> { |
||||||
|
val deferred = CompletableDeferred<T>() |
||||||
|
|
||||||
|
deferred.invokeOnCompletion { |
||||||
|
if (deferred.isCancelled) { |
||||||
|
call.cancel() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
call.enqueue(object : Callback<T> { |
||||||
|
override fun onFailure(call: Call<T>, t: Throwable) { |
||||||
|
deferred.completeExceptionally(t) |
||||||
|
} |
||||||
|
|
||||||
|
override fun onResponse(call: Call<T>, response: Response<T>) { |
||||||
|
if (response.isSuccessful) { |
||||||
|
deferred.complete(response.body()!!) |
||||||
|
} else { |
||||||
|
deferred.completeExceptionally(HttpException(response)) |
||||||
|
} |
||||||
|
} |
||||||
|
}) |
||||||
|
|
||||||
|
return deferred |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private class ResponseCallAdapter<T>( |
||||||
|
private val responseType: Type |
||||||
|
) : CallAdapter<T, Deferred<Response<T>>> { |
||||||
|
|
||||||
|
override fun responseType() = responseType |
||||||
|
|
||||||
|
override fun adapt(call: Call<T>): Deferred<Response<T>> { |
||||||
|
val deferred = CompletableDeferred<Response<T>>() |
||||||
|
|
||||||
|
deferred.invokeOnCompletion { |
||||||
|
if (deferred.isCancelled) { |
||||||
|
call.cancel() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
call.enqueue(object : Callback<T> { |
||||||
|
override fun onFailure(call: Call<T>, t: Throwable) { |
||||||
|
deferred.completeExceptionally(t) |
||||||
|
} |
||||||
|
|
||||||
|
override fun onResponse(call: Call<T>, response: Response<T>) { |
||||||
|
deferred.complete(response) |
||||||
|
} |
||||||
|
}) |
||||||
|
|
||||||
|
return deferred |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,49 @@ |
|||||||
|
package io.legado.app.help.http |
||||||
|
|
||||||
|
import okhttp3.* |
||||||
|
import java.util.* |
||||||
|
import java.util.concurrent.TimeUnit |
||||||
|
|
||||||
|
object HttpHelper { |
||||||
|
|
||||||
|
val client: OkHttpClient = getOkHttpClient() |
||||||
|
|
||||||
|
|
||||||
|
private fun getOkHttpClient(): OkHttpClient { |
||||||
|
val cs = ConnectionSpec.Builder(ConnectionSpec.MODERN_TLS) |
||||||
|
.tlsVersions(TlsVersion.TLS_1_2) |
||||||
|
.build() |
||||||
|
|
||||||
|
val specs = ArrayList<ConnectionSpec>() |
||||||
|
specs.add(cs) |
||||||
|
specs.add(ConnectionSpec.COMPATIBLE_TLS) |
||||||
|
specs.add(ConnectionSpec.CLEARTEXT) |
||||||
|
|
||||||
|
val sslParams = SSLHelper.getSslSocketFactory() |
||||||
|
return OkHttpClient.Builder() |
||||||
|
.connectTimeout(15, TimeUnit.SECONDS) |
||||||
|
.writeTimeout(15, TimeUnit.SECONDS) |
||||||
|
.readTimeout(15, TimeUnit.SECONDS) |
||||||
|
.retryOnConnectionFailure(true) |
||||||
|
.sslSocketFactory(sslParams.sSLSocketFactory, sslParams.trustManager) |
||||||
|
.hostnameVerifier(SSLHelper.unsafeHostnameVerifier) |
||||||
|
.connectionSpecs(specs) |
||||||
|
.followRedirects(true) |
||||||
|
.followSslRedirects(true) |
||||||
|
.protocols(listOf(Protocol.HTTP_1_1)) |
||||||
|
.addInterceptor(getHeaderInterceptor()) |
||||||
|
.build() |
||||||
|
} |
||||||
|
|
||||||
|
private fun getHeaderInterceptor(): Interceptor { |
||||||
|
return Interceptor { chain -> |
||||||
|
val request = chain.request() |
||||||
|
.newBuilder() |
||||||
|
.addHeader("Keep-Alive", "300") |
||||||
|
.addHeader("Connection", "Keep-Alive") |
||||||
|
.addHeader("Cache-Control", "no-cache") |
||||||
|
.build() |
||||||
|
chain.proceed(request) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,175 @@ |
|||||||
|
package io.legado.app.help.http |
||||||
|
|
||||||
|
import javax.net.ssl.* |
||||||
|
import java.io.IOException |
||||||
|
import java.io.InputStream |
||||||
|
import java.security.KeyManagementException |
||||||
|
import java.security.KeyStore |
||||||
|
import java.security.NoSuchAlgorithmException |
||||||
|
import java.security.cert.CertificateException |
||||||
|
import java.security.cert.CertificateFactory |
||||||
|
import java.security.cert.X509Certificate |
||||||
|
|
||||||
|
object SSLHelper { |
||||||
|
|
||||||
|
val sslSocketFactory: SSLParams |
||||||
|
get() = getSslSocketFactoryBase(null, null, null) |
||||||
|
|
||||||
|
/** |
||||||
|
* 为了解决客户端不信任服务器数字证书的问题，网络上大部分的解决方案都是让客户端不对证书做任何检查， |
||||||
|
* 这是一种有很大安全漏洞的办法 |
||||||
|
*/ |
||||||
|
val unsafeTrustManager: X509TrustManager = object : X509TrustManager { |
||||||
|
@Throws(CertificateException::class) |
||||||
|
override fun checkClientTrusted(chain: Array<X509Certificate>, authType: String) { |
||||||
|
} |
||||||
|
|
||||||
|
@Throws(CertificateException::class) |
||||||
|
override fun checkServerTrusted(chain: Array<X509Certificate>, authType: String) { |
||||||
|
} |
||||||
|
|
||||||
|
override fun getAcceptedIssuers(): Array<X509Certificate> { |
||||||
|
return arrayOf() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 此类是用于主机名验证的基接口。 在握手期间，如果 URL 的主机名和服务器的标识主机名不匹配， |
||||||
|
* 则验证机制可以回调此接口的实现程序来确定是否应该允许此连接。策略可以是基于证书的或依赖于其他验证方案。 |
||||||
|
* 当验证 URL 主机名使用的默认规则失败时使用这些回调。如果主机名是可接受的，则返回 true |
||||||
|
*/ |
||||||
|
val unsafeHostnameVerifier: HostnameVerifier = HostnameVerifier { _, _ -> true } |
||||||
|
|
||||||
|
class SSLParams { |
||||||
|
var sSLSocketFactory: SSLSocketFactory? = null |
||||||
|
var trustManager: X509TrustManager? = null |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* https单向认证 |
||||||
|
* 可以额外配置信任服务端的证书策略，否则默认是按CA证书去验证的，若不是CA可信任的证书，则无法通过验证 |
||||||
|
*/ |
||||||
|
fun getSslSocketFactory(trustManager: X509TrustManager): SSLParams { |
||||||
|
return getSslSocketFactoryBase(trustManager, null, null) |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* https单向认证 |
||||||
|
* 用含有服务端公钥的证书校验服务端证书 |
||||||
|
*/ |
||||||
|
fun getSslSocketFactory(vararg certificates: InputStream): SSLParams { |
||||||
|
return getSslSocketFactoryBase(null, null, null, *certificates) |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* https双向认证 |
||||||
|
* bksFile 和 password -> 客户端使用bks证书校验服务端证书 |
||||||
|
* certificates -> 用含有服务端公钥的证书校验服务端证书 |
||||||
|
*/ |
||||||
|
fun getSslSocketFactory(bksFile: InputStream, password: String, vararg certificates: InputStream): SSLParams { |
||||||
|
return getSslSocketFactoryBase(null, bksFile, password, *certificates) |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* https双向认证 |
||||||
|
* bksFile 和 password -> 客户端使用bks证书校验服务端证书 |
||||||
|
* X509TrustManager -> 如果需要自己校验，那么可以自己实现相关校验，如果不需要自己校验，那么传null即可 |
||||||
|
*/ |
||||||
|
fun getSslSocketFactory(bksFile: InputStream, password: String, trustManager: X509TrustManager): SSLParams { |
||||||
|
return getSslSocketFactoryBase(trustManager, bksFile, password) |
||||||
|
} |
||||||
|
|
||||||
|
private fun getSslSocketFactoryBase( |
||||||
|
trustManager: X509TrustManager?, |
||||||
|
bksFile: InputStream?, |
||||||
|
password: String?, |
||||||
|
vararg certificates: InputStream |
||||||
|
): SSLParams { |
||||||
|
val sslParams = SSLParams() |
||||||
|
try { |
||||||
|
val keyManagers = prepareKeyManager(bksFile, password) |
||||||
|
val trustManagers = prepareTrustManager(*certificates) |
||||||
|
val manager: X509TrustManager? |
||||||
|
manager = //优先使用用户自定义的TrustManager |
||||||
|
trustManager ?: if (trustManagers != null) { |
||||||
|
//然后使用默认的TrustManager |
||||||
|
chooseTrustManager(trustManagers) |
||||||
|
} else { |
||||||
|
//否则使用不安全的TrustManager |
||||||
|
unsafeTrustManager |
||||||
|
} |
||||||
|
// 创建TLS类型的SSLContext对象， that uses our TrustManager |
||||||
|
val sslContext = SSLContext.getInstance("TLS") |
||||||
|
// 用上面得到的trustManagers初始化SSLContext，这样sslContext就会信任keyStore中的证书 |
||||||
|
// 第一个参数是授权的密钥管理器，用来授权验证，比如授权自签名的证书验证。第二个是被授权的证书管理器，用来验证服务器端的证书 |
||||||
|
sslContext.init(keyManagers, manager?.let { arrayOf<TrustManager>(it) }, null) |
||||||
|
// 通过sslContext获取SSLSocketFactory对象 |
||||||
|
sslParams.sSLSocketFactory = sslContext.socketFactory |
||||||
|
sslParams.trustManager = manager |
||||||
|
return sslParams |
||||||
|
} catch (e: NoSuchAlgorithmException) { |
||||||
|
throw AssertionError(e) |
||||||
|
} catch (e: KeyManagementException) { |
||||||
|
throw AssertionError(e) |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
private fun prepareKeyManager(bksFile: InputStream?, password: String?): Array<KeyManager>? { |
||||||
|
try { |
||||||
|
if (bksFile == null || password == null) return null |
||||||
|
val clientKeyStore = KeyStore.getInstance("BKS") |
||||||
|
clientKeyStore.load(bksFile, password.toCharArray()) |
||||||
|
val kmf = KeyManagerFactory.getInstance(KeyManagerFactory.getDefaultAlgorithm()) |
||||||
|
kmf.init(clientKeyStore, password.toCharArray()) |
||||||
|
return kmf.keyManagers |
||||||
|
} catch (e: Exception) { |
||||||
|
e.printStackTrace() |
||||||
|
} |
||||||
|
|
||||||
|
return null |
||||||
|
} |
||||||
|
|
||||||
|
private fun prepareTrustManager(vararg certificates: InputStream): Array<TrustManager>? { |
||||||
|
if (certificates.isEmpty()) return null |
||||||
|
try { |
||||||
|
val certificateFactory = CertificateFactory.getInstance("X.509") |
||||||
|
// 创建一个默认类型的KeyStore，存储我们信任的证书 |
||||||
|
val keyStore = KeyStore.getInstance(KeyStore.getDefaultType()) |
||||||
|
keyStore.load(null) |
||||||
|
var index = 0 |
||||||
|
for (certStream in certificates) { |
||||||
|
val certificateAlias = Integer.toString(index++) |
||||||
|
// 证书工厂根据证书文件的流生成证书 cert |
||||||
|
val cert = certificateFactory.generateCertificate(certStream) |
||||||
|
// 将 cert 作为可信证书放入到keyStore中 |
||||||
|
keyStore.setCertificateEntry(certificateAlias, cert) |
||||||
|
try { |
||||||
|
certStream?.close() |
||||||
|
} catch (e: IOException) { |
||||||
|
e.printStackTrace() |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
|
//我们创建一个默认类型的TrustManagerFactory |
||||||
|
val tmf = TrustManagerFactory.getInstance(TrustManagerFactory.getDefaultAlgorithm()) |
||||||
|
//用我们之前的keyStore实例初始化TrustManagerFactory，这样tmf就会信任keyStore中的证书 |
||||||
|
tmf.init(keyStore) |
||||||
|
//通过tmf获取TrustManager数组，TrustManager也会信任keyStore中的证书 |
||||||
|
return tmf.trustManagers |
||||||
|
} catch (e: Exception) { |
||||||
|
e.printStackTrace() |
||||||
|
} |
||||||
|
|
||||||
|
return null |
||||||
|
} |
||||||
|
|
||||||
|
private fun chooseTrustManager(trustManagers: Array<TrustManager>): X509TrustManager? { |
||||||
|
for (trustManager in trustManagers) { |
||||||
|
if (trustManager is X509TrustManager) { |
||||||
|
return trustManager |
||||||
|
} |
||||||
|
} |
||||||
|
return null |
||||||
|
} |
||||||
|
} |
||||||
@ -1,6 +1,18 @@ |
|||||||
package io.legado.app.ui.search |
package io.legado.app.ui.search |
||||||
|
|
||||||
import android.app.Application |
import android.app.Application |
||||||
import androidx.lifecycle.AndroidViewModel |
import androidx.lifecycle.LiveData |
||||||
|
import androidx.lifecycle.MutableLiveData |
||||||
|
import io.legado.app.base.BaseViewModel |
||||||
|
import io.legado.app.data.entities.SearchBook |
||||||
|
|
||||||
class SearchViewModel(application: Application) : AndroidViewModel(application) |
class SearchViewModel(application: Application) : BaseViewModel(application){ |
||||||
|
|
||||||
|
val searchBooks: LiveData<List<SearchBook>> = MutableLiveData() |
||||||
|
|
||||||
|
public fun search(){ |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
|
|||||||
Loading…
Reference in new issue