feat(options): add banner callback

gosh.go

@@ -53,6 +53,7 @@ type Client struct {
 	opts     *Options
 	auth     ssh.AuthMethod
 	callback ssh.HostKeyCallback
+	bannercb ssh.BannerCallback
 }
 
 // NewDefault creates a Client with DefaultHostKeyCallback, the host public key must be in known hosts.
@@ -112,6 +113,12 @@ func (c *Client) WithHostKeyCallback(callback ssh.HostKeyCallback) *Client {
 	return c
 }
 
+// WithBannerCallback sets ssh.BannerCallback of Client.
+func (c *Client) WithBannerCallback(callback ssh.BannerCallback) *Client {
+	c.bannercb = callback
+	return c
+}
+
 // Dial starts a client connection to the given SSH server.
 func (c *Client) Dial() error {
 	cli, err := c.dial()
@@ -280,6 +287,7 @@ func (c *Client) dial() (*ssh.Client, error) {
 			Auth:            []ssh.AuthMethod{c.auth},
 			Timeout:         c.opts.Timeout,
 			HostKeyCallback: c.callback,
+			BannerCallback:  c.bannercb,
 		},
 	)
 }

knownhosts.go

@@ -12,10 +12,10 @@ import (
 
 func AutoFixedHostKeyCallback(host string, remote net.Addr, key ssh.PublicKey) error {
 	found, err := VerifyKnownHost("", host, remote, key)
-	if found && err != nil {
+	if err != nil {
 		return err
 	}
-	if found && err == nil {
+	if found {
 		return nil
 	}
 	// Add the new host to known hosts file.