[update] auth code

pkg/auth/server.go

@@ -30,12 +30,17 @@ func checkAuth(ctx ssh.Context, password, publicKey string) (res ssh.AuthResult)
 	if password != "" {
 		authMethod = "password"
 	}
-	remoteAddr, _, _ := net.SplitHostPort(ctx.RemoteAddr().String())
-	userClient := service.NewSessionClient(service.Username(username),
-		service.Password(password), service.PublicKey(publicKey),
-		service.RemoteAddr(remoteAddr), service.LoginType("T"))
-	user, authStatus := userClient.Authenticate(ctx)
+	userClient, ok := ctx.Value(model.ContextKeyClient).(*service.SessionClient)
+	if !ok {
+		remoteAddr, _, _ := net.SplitHostPort(ctx.RemoteAddr().String())
+		sessionClient := service.NewSessionClient(service.Username(username),
+			service.Password(password), service.PublicKey(publicKey),
+			service.RemoteAddr(remoteAddr), service.LoginType("T"))
 
+		userClient = &sessionClient
+
+	}
+	user, authStatus := userClient.Authenticate(ctx)
 	switch authStatus {
 	case service.AuthMFARequired:
 		ctx.SetValue(model.ContextKeyClient, &userClient)