38 lines
611 B
Go
38 lines
611 B
Go
package upstream
|
|
|
|
import (
|
|
"net"
|
|
|
|
"golang.org/x/crypto/ssh"
|
|
|
|
"github.com/tg123/sshpiper/sshpiperd/registry"
|
|
)
|
|
|
|
type UpstreamHandler func(conn ssh.ConnMetadata) (net.Conn, *ssh.SSHPiperAuthPipe, error)
|
|
|
|
type UpstreamProvider interface {
|
|
registry.Plugin
|
|
|
|
GetFindUpstreamHandle() UpstreamHandler
|
|
}
|
|
|
|
var (
|
|
drivers = registry.NewRegistry()
|
|
)
|
|
|
|
func Register(name string, driver UpstreamProvider) {
|
|
drivers.Register(name, driver)
|
|
}
|
|
|
|
func All() []string {
|
|
return drivers.Drivers()
|
|
}
|
|
|
|
func Get(name string) UpstreamProvider {
|
|
if d, ok := drivers.Get(name).(UpstreamProvider); ok {
|
|
return d
|
|
|
|
}
|
|
|
|
return nil
|
|
}
|