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

add detail logs to upstream and challenger

Browse source
This commit is contained in:
Boshi Lian 2022-05-14 11:15:21 +00:00
parent 5b4a05e180
commit 7639c1d5a7
1 changed files with 26 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

28
sshpiperd/sshpiperd.go
View file

@ -78,7 +78,17 @@ func installDrivers(piper *ssh.PiperConfig, config *piperdConfig, logger *log.Lo
return fmt.Errorf("upstream driver return nil handler")
}
piper.FindUpstream = handler
piper.FindUpstream = func(conn ssh.ConnMetadata, challengeCtx ssh.AdditionalChallengeContext) (net.Conn, *ssh.AuthPipe, error) {
c, a, err := handler(conn, challengeCtx)
if err != nil {
logger.Errorf("upstream driver [%v] cannot find upstream due to [%v]", plugin.GetName(), err)
} else {
logger.Debugf("upstream driver [%v] found upstream connection [%v]", plugin.GetName(), c.RemoteAddr().String())
}
return c, a, err
}
return nil
},
},
@ -96,7 +106,21 @@ func installDrivers(piper *ssh.PiperConfig, config *piperdConfig, logger *log.Lo
return fmt.Errorf("challenger driver return nil handler")
}
piper.AdditionalChallenge = handler
piper.AdditionalChallenge = func(conn ssh.ConnMetadata, client ssh.KeyboardInteractiveChallenge) (ssh.AdditionalChallengeContext, error) {
c, err := handler(conn, client)
if err != nil {
logger.Errorf("challenger [%v] failed [%v]", plugin.GetName(), err)
} else {
if c != nil {
logger.Infof("challenger [%v] success with challenged username", plugin.GetName(), c.ChallengedUsername())
} else {
logger.Debugf("challenger [%v] success with empty context", plugin.GetName())
}
}
return c, err
}
return nil
},
},