add channel handler and local port forwarding callback

shellhub-io · Oct 3, 2022 · 333f517 · 333f517
1 parent b368bfa
commit 333f517
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/ssh/server/server.go b/ssh/server/server.go
@@ -46,6 +46,14 @@ func NewServer(opts *Options, tunnel *httptunnel.Tunnel) *Server {
 		SubsystemHandlers: map[string]gliderssh.SubsystemHandler{
 			handler.SFTPSubsystem: handler.SFTPSubsystemHandler(tunnel),
 		},
+                ChannelHandlers: map[string]gliderssh.ChannelHandler{
+                        "session":      gliderssh.DefaultSessionHandler,
+                        "direct-tcpip": gliderssh.DirectTCPIPHandler,
+                },
+                LocalPortForwardingCallback: gliderssh.LocalPortForwardingCallback(func(ctx gliderssh.Context, dhost string, dport uint32) bool {
+                        log.Println("Accepted forward", dhost, dport)
+                        return true
+                }),
 	}
 
 	if _, err := os.Stat(os.Getenv("PRIVATE_KEY")); os.IsNotExist(err) {