This repository has been archived on 2024-08-25. You can view files and clone it, but cannot push or open issues or pull requests.
Comet/app/src/main/java/dev/lowrespalmtree/comet/Request.kt
2021-12-05 21:49:24 +01:00

72 lines
2.6 KiB
Kotlin

package dev.lowrespalmtree.comet
import android.annotation.SuppressLint
import android.net.Uri
import android.util.Log
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.channels.Channel
import kotlinx.coroutines.launch
import java.io.BufferedInputStream
import java.net.SocketTimeoutException
import java.security.cert.X509Certificate
import javax.net.ssl.SSLContext
import javax.net.ssl.SSLSocket
import javax.net.ssl.X509TrustManager
class Request(private val uri: Uri) {
private val port get() = if (uri.port > 0) uri.port else 1965
fun connect(): SSLSocket {
Log.d(TAG, "connect")
val context = SSLContext.getInstance("TLSv1.2")
context.init(null, arrayOf(TrustManager()), null)
val socket = context.socketFactory.createSocket(uri.host, port) as SSLSocket
socket.soTimeout = 10000
socket.startHandshake()
return socket
}
fun proceed(socket: SSLSocket, scope: CoroutineScope): Channel<ByteArray> {
Log.d(TAG, "proceed")
socket.outputStream.write("$uri\r\n".toByteArray())
val channel = Channel<ByteArray>()
scope.launch {
val buffer = ByteArray(1024)
var numRead: Int
socket.inputStream.use { socket_input_stream ->
BufferedInputStream(socket_input_stream).use { bis ->
try {
@Suppress("BlockingMethodInNonBlockingContext") // what u gonna do
while ((bis.read(buffer).also { numRead = it }) >= 0) {
Log.d(TAG, "proceed coroutine: received $numRead bytes")
val received = buffer.sliceArray(0 until numRead)
channel.send(received)
}
} catch (e: SocketTimeoutException) {
Log.i(TAG, "Socket timeout.")
channel.cancel()
}
}
}
Log.d(TAG, "proceed coroutine: reading completed")
}
return channel
}
@SuppressLint("CustomX509TrustManager")
class TrustManager: X509TrustManager {
@SuppressLint("TrustAllX509TrustManager")
override fun checkClientTrusted(chain: Array<out X509Certificate>?, authType: String?) {}
override fun checkServerTrusted(chain: Array<out X509Certificate>?, authType: String?) {
Log.d(TAG, "cool cert, please continue")
}
override fun getAcceptedIssuers(): Array<out X509Certificate> = arrayOf()
}
companion object {
const val TAG = "Request"
}
}