Add Setter functions for SSH.sshconfig and SSH.listener

Closes #35
sosedoff · Mar 1, 2023 · 90ab74a · 90ab74a
1 parent 7a361c7
commit 90ab74a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/ssh.go b/ssh.go
@@ -231,6 +231,9 @@ func (s *SSH) createServerKey() error {
 }
 
 func (s *SSH) setup() error {
+	if s.sshconfig != nil {
+		return nil
+	}
 	config := &ssh.ServerConfig{
 		ServerVersion: fmt.Sprintf("SSH-2.0-gitkit %s", Version),
 	}
@@ -375,3 +378,13 @@ func (s *SSH) Address() string {
 	}
 	return ""
 }
+
+// SetSSHConfig can be used to set custom SSH Server settings.
+func (s *SSH) SetSSHConfig(cfg *ssh.ServerConfig) {
+	s.sshconfig = cfg
+}
+
+// SetListener can be used to set custom Listener.
+func (s *SSH) SetListener(l net.Listener) {
+	s.listener = l
+}