bug: context switching on blocking socket operations

mailserver/imap-agent/build.gradle.kts

@@ -18,7 +18,7 @@ kotlin {
         implementation(project(":utils"))
         implementation("io.github.microutils:kotlin-logging:3.0.2")
 
-        implementation("dev.sitar:kio:1.1.1")
+        implementation("dev.sitar:kio:1.1.2")
         implementation("dev.sitar:keystone:0.1-SNAPSHOT")
         implementation("io.ktor:ktor-network:2.1.3")
         implementation("io.ktor:ktor-network-tls:2.1.3")

mailserver/imap/build.gradle.kts

@@ -15,7 +15,7 @@ kotlin {
         api(project(":sasl"))
         implementation("io.github.microutils:kotlin-logging:3.0.2")
         implementation("org.jetbrains.kotlinx:kotlinx-coroutines-core:1.6.4")
-        implementation("dev.sitar:kio:1.1.1")
+        implementation("dev.sitar:kio:1.1.2")
         implementation("io.ktor:ktor-network:2.1.3")
     }
 }

mailserver/message-format/build.gradle.kts

@@ -12,7 +12,7 @@ kotlin {
 
     sourceSets["commonMain"].dependencies {
         api("org.jetbrains.kotlinx:kotlinx-datetime:0.4.0")
-        implementation("dev.sitar:kio:1.1.1")
+        implementation("dev.sitar:kio:1.1.2")
     }
 
     sourceSets["commonTest"].dependencies {

mailserver/pop3-agent/build.gradle.kts

@@ -17,7 +17,7 @@ kotlin {
         implementation("io.github.microutils:kotlin-logging:3.0.2")
 
         implementation("dev.sitar:keystone:0.1-SNAPSHOT")
-        implementation("dev.sitar:kio:1.1.1")
+        implementation("dev.sitar:kio:1.1.2")
         implementation("io.ktor:ktor-network:2.1.3")
         implementation("io.ktor:ktor-network-tls:2.1.3")
         implementation("dev.sitar:dnskotlin:0.2.1")

mailserver/pop3/build.gradle.kts

@@ -16,7 +16,7 @@ kotlin {
         implementation(project(":message-format"))
         implementation("io.github.microutils:kotlin-logging:3.0.2")
         implementation("org.jetbrains.kotlinx:kotlinx-coroutines-core:1.6.4")
-        implementation("dev.sitar:kio:1.1.1")
+        implementation("dev.sitar:kio:1.1.2")
         implementation("io.ktor:ktor-network:2.1.3")
     }
 }

mailserver/runner/src/main/kotlin/imap.kt

@@ -9,6 +9,7 @@ import dev.sitar.kmail.runner.storage.formats.MailboxFolder
 import dev.sitar.kmail.runner.storage.formats.MailboxMessage
 import dev.sitar.kmail.sasl.SaslChallenge
 import dev.sitar.kmail.utils.server.ServerSocketFactory
+import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.launch
 import kotlinx.datetime.Clock

mailserver/runner/src/main/kotlin/launcher.kt

@@ -1,6 +1,12 @@
 package dev.sitar.kmail.runner
 
+import dev.sitar.kmail.imap.agent.ImapAgent
+import dev.sitar.kmail.imap.agent.ImapLayer
+import dev.sitar.kmail.imap.agent.ImapMailbox
+import dev.sitar.kmail.imap.agent.ImapServer
+import dev.sitar.kmail.imap.agent.transports.ImapServerTransport
 import dev.sitar.kmail.runner.storage.mailbox
+import dev.sitar.kmail.sasl.SaslChallenge
 import dev.sitar.kmail.smtp.InternetMessage
 import dev.sitar.kmail.utils.connection.ConnectionFactory
 import dev.sitar.kmail.utils.server.ServerSocketFactory

mailserver/runner/src/main/kotlin/launcherJvm.kt

@@ -4,7 +4,7 @@ import dev.sitar.kmail.utils.connection.TlsCapableConnectionFactory
 import dev.sitar.kmail.utils.server.TlsCapableServerSocketFactory
 import kotlinx.coroutines.coroutineScope
 
-suspend fun main(): Unit = coroutineScope {
+suspend fun main() {
     dns()
 
 //    System.setProperty("javax.net.debug", "all")

mailserver/runner/src/main/kotlin/pop3.kt

@@ -8,6 +8,7 @@ import dev.sitar.kmail.runner.storage.formats.Mailbox
 import dev.sitar.kmail.runner.storage.formats.MailboxMessage
 import dev.sitar.kmail.runner.storage.StorageLayer
 import dev.sitar.kmail.utils.server.ServerSocketFactory
+import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.launch
 import mu.KotlinLogging

mailserver/runner/src/main/kotlin/transfer.kt

@@ -5,6 +5,7 @@ import dev.sitar.kmail.agents.smtp.transfer.*
 import dev.sitar.kmail.smtp.InternetMessage
 import dev.sitar.kmail.utils.connection.ConnectionFactory
 import dev.sitar.kmail.utils.server.ServerSocketFactory
+import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.MutableSharedFlow

mailserver/sasl/build.gradle.kts

@@ -11,6 +11,6 @@ kotlin {
 
     sourceSets["commonMain"].dependencies {
         api("org.jetbrains.kotlinx:kotlinx-datetime:0.4.0")
-        implementation("dev.sitar:kio:1.1.1")
+        implementation("dev.sitar:kio:1.1.2")
     }
 }

mailserver/smtp-agent/build.gradle.kts

@@ -17,7 +17,6 @@ kotlin {
         implementation("io.github.microutils:kotlin-logging:3.0.2")
 
         implementation("dev.sitar:keystone:0.1-SNAPSHOT")
-        implementation("dev.sitar:kio:1.1.1")
         implementation("io.ktor:ktor-network:2.1.3")
         implementation("io.ktor:ktor-network-tls:2.1.3")
         implementation("dev.sitar:dnskotlin:0.2.1")

mailserver/smtp/build.gradle.kts

@@ -14,7 +14,7 @@ kotlin {
     sourceSets["commonMain"].dependencies {
         api(project(":message-format"))
         api(project(":utils"))
-        api("dev.sitar:kio:1.1.1")
+        api("dev.sitar:kio:1.1.2")
         api("dev.sitar:dnskotlin:0.2.1")
         implementation("io.ktor:ktor-utils:2.2.4")
     }

mailserver/utils/build.gradle.kts

@@ -12,7 +12,7 @@ kotlin {
     sourceSets["commonMain"].dependencies {
         implementation("io.github.microutils:kotlin-logging:3.0.2")
 
-        api("dev.sitar:kio:1.1.1")
+        api("dev.sitar:kio:1.1.2")
         api("io.ktor:ktor-io:2.2.4")
         api("io.ktor:ktor-network:2.1.3")
         api("io.ktor:ktor-network-tls:2.1.3")

mailserver/utils/src/commonMain/kotlin/io/OutputStreamAsyncWriterStream.kt

@@ -4,6 +4,8 @@ import dev.sitar.kio.Pool
 import dev.sitar.kio.Slice
 import dev.sitar.kio.buffers.Buffer
 import dev.sitar.kio.buffers.DefaultBufferPool
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 import java.io.OutputStream
 
 class OutputStreamAsyncWriterStream(val outputStream: OutputStream) : AsyncWriterStream {
@@ -14,7 +16,7 @@ class OutputStreamAsyncWriterStream(val outputStream: OutputStream) : AsyncWrite
 
     override suspend fun writeBytes(slice: Slice): Int {
         val (arr, off, len) = slice
-        outputStream.write(arr, off, len)
+        withContext(Dispatchers.IO) { outputStream.write(arr, off, len) }
         return len
     }
 

mailserver/utils/src/jvmMain/kotlin/connection/TlsCapableConnectionFactory.kt

@@ -1,11 +1,15 @@
 package dev.sitar.kmail.utils.connection
 
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 import javax.net.SocketFactory
 import javax.net.ssl.SSLContext
 
 actual class TlsCapableConnectionFactory(val sslContext: SSLContext) : ConnectionFactory {
     override suspend fun connect(host: String, port: Int): Connection {
-        val socket = SocketFactory.getDefault().createSocket(host, port)
+        val socket = withContext(Dispatchers.IO) {
+            SocketFactory.getDefault().createSocket(host, port)
+        }
         return TlsCapableConnection(socket, sslContext, isSecure = false)
     }
 }

mailserver/utils/src/jvmMain/kotlin/server/TlsCapableServerSocket.kt

@@ -2,10 +2,13 @@ package dev.sitar.kmail.utils.server
 
 import dev.sitar.kmail.utils.connection.Connection
 import dev.sitar.kmail.utils.connection.TlsCapableConnection
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 import javax.net.ssl.SSLContext
 
 actual class TlsCapableServerSocket(val socket: java.net.ServerSocket, val sslContext: SSLContext) : ServerSocket {
     override suspend fun accept(): Connection {
-        return TlsCapableConnection(socket.accept(), sslContext, isSecure = false)
+        val socket = withContext(Dispatchers.IO) { socket.accept() }
+        return TlsCapableConnection(socket, sslContext, isSecure = false)
     }
 }

mailserver/utils/src/jvmMain/kotlin/server/TlsCapableServerSocketFactory.kt

@@ -1,5 +1,7 @@
 package dev.sitar.kmail.utils.server
 
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 import mu.KotlinLogging
 import javax.net.ssl.SSLContext
 
@@ -8,7 +10,7 @@ private val logger = KotlinLogging.logger { }
 actual class TlsCapableServerSocketFactory(val sslContext: SSLContext) : ServerSocketFactory {
     override suspend fun bind(port: Int): ServerSocket {
         logger.debug { "Creating a secure server socket on port $port." }
-        val serverSocket = javax.net.ServerSocketFactory.getDefault().createServerSocket(port)
+        val serverSocket = withContext(Dispatchers.IO) { javax.net.ServerSocketFactory.getDefault().createServerSocket(port) }
         logger.debug { "Created a secure server socket on $port." }
         return TlsCapableServerSocket(serverSocket, sslContext)
     }