xyz.cssxsh.pixiv.tool.RubySSLSocketFactory.kt Maven / Gradle / Ivy

Go to download
package xyz.cssxsh.pixiv.tool

import java.net.*
import javax.net.ssl.*

public object RubySSLSocketFactory : SSLSocketFactory() {

    internal val regexes = mutableListOf("""(pixiv|pximg)\.net""".toRegex())

    private fun Socket.setServerNames(): Socket = apply {
        if (this !is SSLSocket) return@apply
        sslParameters = sslParameters.apply {
            serverNames = serverNames?.filter { name ->
                if (name !is SNIHostName) return@filter false
                regexes.none { it in name.asciiName }
            }
        }
    }

    private val default: SSLSocketFactory = SSLContext.getDefault().socketFactory

    override fun createSocket(socket: Socket?, host: String?, port: Int, autoClose: Boolean): Socket? =
        default.createSocket(socket, host, port, autoClose)?.setServerNames()

    override fun createSocket(host: String?, port: Int): Socket? =
        default.createSocket(host, port)?.setServerNames()

    override fun createSocket(host: String?, port: Int, localHost: InetAddress?, localPort: Int): Socket? =
        default.createSocket(host, port, localHost, localPort)?.setServerNames()

    override fun createSocket(host: InetAddress?, port: Int): Socket? =
        default.createSocket(host, port)?.setServerNames()

    override fun createSocket(address: InetAddress?, port: Int, localAddress: InetAddress?, localPort: Int): Socket? =
        default.createSocket(address, port, localAddress, localPort)?.setServerNames()

    override fun getDefaultCipherSuites(): Array = default.defaultCipherSuites

    override fun getSupportedCipherSuites(): Array = default.supportedCipherSuites
}