Skip to content

K
koko
Commit 69c04033 authored by Eric's avatar Eric
Browse files
Options

fix addr bugs

parent 98f4154c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
pkg/auth/server.go
View file @ 69c04033
package auth package auth
import ( import (
"strings" "net"
"github.com/gliderlabs/ssh" "github.com/gliderlabs/ssh"
gossh "golang.org/x/crypto/ssh" gossh "golang.org/x/crypto/ssh"
...@@ -30,7 +30,7 @@ func checkAuth(ctx ssh.Context, password, publicKey string) (res ssh.AuthResult) ...@@ -30,7 +30,7 @@ func checkAuth(ctx ssh.Context, password, publicKey string) (res ssh.AuthResult)
if password != "" { if password != "" {
authMethod = "password" authMethod = "password"
} }
remoteAddr := strings.Split(ctx.RemoteAddr().String(), ":")[0] remoteAddr, _, _ := net.SplitHostPort(ctx.RemoteAddr().String())
resp, err := service.Authenticate(username, password, publicKey, remoteAddr, "T") resp, err := service.Authenticate(username, password, publicKey, remoteAddr, "T")
if err != nil { if err != nil {
...@@ -73,7 +73,7 @@ func CheckUserPublicKey(ctx ssh.Context, key ssh.PublicKey) ssh.AuthResult { ...@@ -73,7 +73,7 @@ func CheckUserPublicKey(ctx ssh.Context, key ssh.PublicKey) ssh.AuthResult {
func CheckMFA(ctx ssh.Context, challenger gossh.KeyboardInteractiveChallenge) (res ssh.AuthResult) { func CheckMFA(ctx ssh.Context, challenger gossh.KeyboardInteractiveChallenge) (res ssh.AuthResult) {
username := ctx.User() username := ctx.User()
remoteAddr := strings.Split(ctx.RemoteAddr().String(), ":")[0] remoteAddr, _, _ := net.SplitHostPort(ctx.RemoteAddr().String())
res = ssh.AuthFailed res = ssh.AuthFailed
defer func() { defer func() {
authMethod := "MFA" authMethod := "MFA"
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment