52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"strconv"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/tg123/sshpiper/libplugin"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
func main() {
|
|
|
|
libplugin.CreateAndRunPluginTemplate(&libplugin.PluginTemplate{
|
|
Name: "simplemath",
|
|
Usage: "sshpiperd simplemath plugin, do math before ssh login",
|
|
CreateConfig: func(_ *cli.Context) (*libplugin.SshPiperPluginConfig, error) {
|
|
return &libplugin.SshPiperPluginConfig{
|
|
KeyboardInteractiveCallback: func(conn libplugin.ConnMetadata, client libplugin.KeyboardInteractiveChallenge) (*libplugin.Upstream, error) {
|
|
client("", "lets do math", "", false)
|
|
|
|
for {
|
|
|
|
a := rand.Intn(10)
|
|
b := rand.Intn(10)
|
|
|
|
ans, err := client("", "", fmt.Sprintf("what is %v + %v = ", a, b), true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
log.Printf("got ans = %v", ans)
|
|
|
|
if ans == fmt.Sprintf("%v", a+b) {
|
|
|
|
log.Printf("got ans = %v", ans)
|
|
|
|
return &libplugin.Upstream{
|
|
Auth: libplugin.CreateNextPluginAuth(map[string]string{
|
|
"a": strconv.Itoa(a),
|
|
"b": strconv.Itoa(b),
|
|
"ans": ans,
|
|
}),
|
|
}, nil
|
|
}
|
|
}
|
|
},
|
|
}, nil
|
|
},
|
|
})
|
|
}
|