2016-10-03 21:54:17 +00:00
|
|
|
package ssh
|
|
|
|
|
|
|
|
import "io/ioutil"
|
|
|
|
|
2016-11-29 23:23:42 +00:00
|
|
|
// PasswordAuth returns a functional option that sets PasswordHandler on the server.
|
2016-10-03 21:54:17 +00:00
|
|
|
func PasswordAuth(fn PasswordHandler) Option {
|
|
|
|
return func(srv *Server) error {
|
|
|
|
srv.PasswordHandler = fn
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-11-29 23:23:42 +00:00
|
|
|
// PublicKeyAuth returns a functional option that sets PublicKeyHandler on the server.
|
2016-10-03 21:54:17 +00:00
|
|
|
func PublicKeyAuth(fn PublicKeyHandler) Option {
|
|
|
|
return func(srv *Server) error {
|
|
|
|
srv.PublicKeyHandler = fn
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-11-29 23:23:42 +00:00
|
|
|
// HostKeyFile returns a functional option that adds HostSigners to the server
|
|
|
|
// from a PEM file at filepath.
|
2016-10-03 21:54:17 +00:00
|
|
|
func HostKeyFile(filepath string) Option {
|
|
|
|
return func(srv *Server) error {
|
|
|
|
pemBytes, err := ioutil.ReadFile(filepath)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
for _, block := range decodePemBlocks(pemBytes) {
|
|
|
|
signer, err := signerFromBlock(block)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2016-11-29 23:23:42 +00:00
|
|
|
srv.AddHostKey(signer)
|
2016-10-03 21:54:17 +00:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-11-29 23:23:42 +00:00
|
|
|
// HostKeyPEM returns a functional option that adds HostSigners to the server
|
|
|
|
// from a PEM file as bytes.
|
2016-10-03 21:54:17 +00:00
|
|
|
func HostKeyPEM(bytes []byte) Option {
|
|
|
|
return func(srv *Server) error {
|
|
|
|
for _, block := range decodePemBlocks(bytes) {
|
|
|
|
signer, err := signerFromBlock(block)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2016-11-29 23:23:42 +00:00
|
|
|
srv.AddHostKey(signer)
|
2016-10-03 21:54:17 +00:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-11-29 23:23:42 +00:00
|
|
|
// NoPty returns a functional option that sets PtyCallback to return false,
|
|
|
|
// denying PTY requests.
|
2016-10-03 21:54:17 +00:00
|
|
|
func NoPty() Option {
|
|
|
|
return func(srv *Server) error {
|
2017-03-14 19:13:03 +00:00
|
|
|
srv.PtyCallback = func(ctx Context, pty Pty) bool {
|
2016-10-03 21:54:17 +00:00
|
|
|
return false
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
2017-07-12 17:27:56 +00:00
|
|
|
|
|
|
|
// WrapConn returns a functional option that sets ConnCallback on the server.
|
|
|
|
func WrapConn(fn ConnCallback) Option {
|
|
|
|
return func(srv *Server) error {
|
|
|
|
srv.ConnCallback = fn
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|