liubin/icebergs
1
0
Fork 0
forked from x/icebergs
Code Pull Requests Activity

opt some

Browse Source
This commit is contained in:
shaoying 2020-12-03 18:28:49 +08:00
parent 09a602e14b
commit 377b107212
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
base/ssh/connect.go
View File

@ -1,6 +1,8 @@
package ssh package ssh
import ( import (
"fmt"
ice "github.com/shylinux/icebergs" ice "github.com/shylinux/icebergs"
"github.com/shylinux/icebergs/base/aaa" "github.com/shylinux/icebergs/base/aaa"
"github.com/shylinux/icebergs/base/mdb" "github.com/shylinux/icebergs/base/mdb"
@ -25,6 +27,14 @@ func _ssh_conn(m *ice.Message, conn net.Conn, username, hostport string) (*ssh.C
for _, k := range questions { for _, k := range questions {
switch strings.TrimSpace(strings.ToLower(k)) { switch strings.TrimSpace(strings.ToLower(k)) {
case "verification code:": case "verification code:":
if m.Option("verify") == "" {
verify := ""
fmt.Println("verify code: \n")
fmt.Println("verify code: ")
fmt.Scanf("%6s", &verify)
res = append(res, verify)
return
}
res = append(res, aaa.TOTP_GET(m.Option("verify"), 6, 30)) res = append(res, aaa.TOTP_GET(m.Option("verify"), 6, 30))
case "password:": case "password:":
res = append(res, m.Option(aaa.PASSWORD)) res = append(res, m.Option(aaa.PASSWORD))