engineering/sshpiper
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

database test with pk

Browse source
This commit is contained in:
Boshi Lian 2019-01-08 22:40:55 -08:00
parent 21085c2436
commit e544b57dbc
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
sshpiperd/upstream/database/handler.go
View file

@ -31,7 +31,7 @@ func (p *plugin) findUpstream(conn ssh.ConnMetadata) (net.Conn, *ssh.AuthPipe, e
if !d.Upstream.Server.IgnoreHostKey {
key, err := ssh.ParsePublicKey([]byte(d.Upstream.Server.HostKey.Key.Data))
key, _, _, _, err := ssh.ParseAuthorizedKey([]byte(d.Upstream.Server.HostKey.Key.Data))
if err != nil {
return nil, nil, err
}

8
sshpiperd/upstream/database/handler_test.go
View file

@ -127,9 +127,9 @@ func TestFindUpstream(t *testing.T) {
listener, err := createListener(t)
defer listener.Close()
createEntry(t, db, "finddown0", "findup0", listener.Addr().String(), true)
createEntry(t, db, "finddown1", "findup1", listener.Addr().String(), true)
createEntry(t, db, "finddown2", "findup2", listener.Addr().String(), true)
createEntry(t, db, "finddown0", "findup0", listener.Addr().String(), false)
createEntry(t, db, "finddown1", "findup1", listener.Addr().String(), false)
createEntry(t, db, "finddown2", "findup2", listener.Addr().String(), false)
_, auth, err := h(testconn{"finddown0"})
if err != nil {
@ -166,7 +166,7 @@ func TestPublicKeyCallback(t *testing.T) {
listener, err := createListener(t)
defer listener.Close()
pub, _ := createEntry(t, db, "pkdown", "pkdown", listener.Addr().String(), true)
pub, _ := createEntry(t, db, "pkdown", "pkdown", listener.Addr().String(), false)
_, auth, err := h(testconn{"pkdown"})
if err != nil {