X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/a7da936b6106bff17a7830124b1fd87ff01ad91e..b3e4886cbbe195347179d0664621da9bc34e6170:/lib/dispatchcloud/test/ssh_service.go

diff --git a/lib/dispatchcloud/test/ssh_service.go b/lib/dispatchcloud/test/ssh_service.go
index ed5995f4c5..31919b566d 100644
--- a/lib/dispatchcloud/test/ssh_service.go
+++ b/lib/dispatchcloud/test/ssh_service.go
@@ -18,6 +18,8 @@ import (
 	check "gopkg.in/check.v1"
 )
 
+// LoadTestKey returns a public/private ssh keypair, read from the files
+// identified by the path of the private key.
 func LoadTestKey(c *check.C, fnm string) (ssh.PublicKey, ssh.Signer) {
 	rawpubkey, err := ioutil.ReadFile(fnm + ".pub")
 	c.Assert(err, check.IsNil)
@@ -39,6 +41,7 @@ type SSHExecFunc func(env map[string]string, command string, stdin io.Reader, st
 type SSHService struct {
 	Exec           SSHExecFunc
 	HostKey        ssh.Signer
+	AuthorizedUser string
 	AuthorizedKeys []ssh.PublicKey
 
 	listener net.Listener
@@ -64,6 +67,11 @@ func (ss *SSHService) Address() string {
 	return ln.Addr().String()
 }
 
+// RemoteUser returns the username that will be accepted.
+func (ss *SSHService) RemoteUser() string {
+	return ss.AuthorizedUser
+}
+
 // Close shuts down the server and releases resources. Established
 // connections are unaffected.
 func (ss *SSHService) Close() {
@@ -103,7 +111,7 @@ func (ss *SSHService) run() {
 	}
 	config.AddHostKey(ss.HostKey)
 
-	listener, err := net.Listen("tcp", ":")
+	listener, err := net.Listen("tcp", "127.0.0.1:")
 	if err != nil {
 		ss.err = err
 		return