Readd customizable timeout

antoniomika · May 2, 2020 · 6e70b2a · 6e70b2a
1 parent 00fbac4
commit 6e70b2a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.dockerignore b/.dockerignore
@@ -1,8 +1,8 @@
 /.git/
 /.github/
 /.gitignore
-/deploy/
 !/deploy/pubkeys/.gitkeep
+/deploy/
 /dist/
 /sish
 /.vscode/
diff --git a/cmd/sish.go b/cmd/sish.go
@@ -90,6 +90,7 @@ func init() {
 
 	rootCmd.PersistentFlags().DurationP("connection-idle-timeout", "", 5*time.Second, "Number of seconds to wait for activity before closing a connection")
 	rootCmd.PersistentFlags().DurationP("ping-client-interval", "", 5*time.Second, "Interval in seconds to ping a client to ensure it is up.")
+	rootCmd.PersistentFlags().DurationP("cleanup-unbound-timeout", "", 5*time.Second, "Interval in seconds to wait before cleaning up an unbound connection.")
 }
 
 func initConfig() {

diff --git a/sshmuxer/sshmuxer.go b/sshmuxer/sshmuxer.go
@@ -180,7 +180,7 @@ func Start() {
 
 		if viper.GetBool("cleanup-unbound") {
 			go func() {
-				<-time.After(5 * time.Second)
+				<-time.After(viper.GetDuration("cleanup-unbound-timeout"))
 				if !clientLoggedIn {
 					conn.Close()
 				}