feature: add banner callback

nuvolaris · Apr 29, 2022 · ee51821 · ee51821
1 parent 90b5fa4
commit ee51821
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/client.go b/client.go
@@ -23,12 +23,13 @@ type Client struct {
 
 // Config for Client.
 type Config struct {
- Auth Auth
- User string
- Addr string
- Port uint
- Timeout time.Duration
- Callback ssh.HostKeyCallback
+ Auth Auth
+ User string
+ Addr string
+ Port uint
+ Timeout time.Duration
+ Callback ssh.HostKeyCallback
+ BannerCallback ssh.BannerCallback
 }
 
 // DefaultTimeout is the timeout of ssh client connection.
@@ -87,6 +88,7 @@ func Dial(proto string, c *Config) (*ssh.Client, error) {
  Auth: c.Auth,
  Timeout: c.Timeout,
  HostKeyCallback: c.Callback,
+ BannerCallback: c.BannerCallback,
  })
 }