flik/termora
1
0
Fork 0
mirror of https://github.com/TermoraDev/termora.git synced 2026-01-16 02:12:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: remove proxy when session closes

Browse Source
This commit is contained in:
hstyi
2025-03-31 10:53:40 +08:00
committed by hstyi
parent ca82704738
commit 54044625ea
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/kotlin/app/termora/SshClients.kt
View File

@@ -41,6 +41,8 @@ import org.apache.sshd.common.io.IoServiceEventListener
import org.apache.sshd.common.io.IoSession import org.apache.sshd.common.io.IoSession
import org.apache.sshd.common.kex.BuiltinDHFactories import org.apache.sshd.common.kex.BuiltinDHFactories
import org.apache.sshd.common.keyprovider.KeyIdentityProvider import org.apache.sshd.common.keyprovider.KeyIdentityProvider
import org.apache.sshd.common.session.Session
import org.apache.sshd.common.session.SessionListener
import org.apache.sshd.common.util.net.SshdSocketAddress import org.apache.sshd.common.util.net.SshdSocketAddress
import org.apache.sshd.core.CoreModuleProperties import org.apache.sshd.core.CoreModuleProperties
import org.apache.sshd.server.forward.AcceptAllForwardingFilter import org.apache.sshd.server.forward.AcceptAllForwardingFilter
@@ -519,7 +521,17 @@ object SshClients {
override fun getClientProxyConnector(): ClientProxyConnector? { override fun getClientProxyConnector(): ClientProxyConnector? {
val entry = getAttribute(HOST_CONFIG_ENTRY) ?: return null val entry = getAttribute(HOST_CONFIG_ENTRY) ?: return null
val clientProxyConnectorId = entry.getProperty(CLIENT_PROXY_CONNECTOR) ?: return null val clientProxyConnectorId = entry.getProperty(CLIENT_PROXY_CONNECTOR) ?: return null
return sshClient.clientProxyConnectors.remove(clientProxyConnectorId) val clientProxyConnector = sshClient.clientProxyConnectors[clientProxyConnectorId]
if (clientProxyConnector != null) {
addSessionListener(object : SessionListener {
override fun sessionClosed(session: Session) {
clientProxyConnectors.remove(clientProxyConnectorId)
}
})
}
return clientProxyConnector
} }
} }
} }