163 lines
3.3 KiB
Go
163 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"runtime/debug"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/tg123/sshpiper/cmd/sshpiperd/internal/plugin"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
var mainver string = "(devel)"
|
|
|
|
func version() string {
|
|
|
|
var v = mainver
|
|
|
|
bi, ok := debug.ReadBuildInfo()
|
|
if !ok {
|
|
return v
|
|
}
|
|
|
|
for _, s := range bi.Settings {
|
|
switch s.Key {
|
|
case "vcs.revision":
|
|
v = fmt.Sprintf("%v, %v", v, s.Value[:9])
|
|
case "vcs.time":
|
|
v = fmt.Sprintf("%v, %v", v, s.Value)
|
|
}
|
|
}
|
|
|
|
v = fmt.Sprintf("%v, %v", v, bi.GoVersion)
|
|
|
|
return v
|
|
}
|
|
|
|
func splitByDash(args []string) ([]string, []string) {
|
|
for i, arg := range args {
|
|
if arg == "--" {
|
|
return args[:i], args[i+1:]
|
|
}
|
|
}
|
|
|
|
return args, nil
|
|
}
|
|
|
|
func main() {
|
|
|
|
app := &cli.App{
|
|
Name: "sshpiperd",
|
|
Usage: "the missing reverse proxy for ssh scp",
|
|
UsageText: "sshpiperd [options] <plugin1> [plugin options] [-- [plugin2] [plugin options] [-- ...]]",
|
|
Description: "sshpiperd works as a proxy-like ware, and route connections by username, src ip , etc.\nhttps://github.com/tg123/sshpiper",
|
|
Version: version(),
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "address",
|
|
Aliases: []string{"l"},
|
|
Value: "0.0.0.0",
|
|
Usage: "listening address",
|
|
EnvVars: []string{"SSHPIPERD_ADDRESS"},
|
|
},
|
|
&cli.IntFlag{
|
|
Name: "port",
|
|
Aliases: []string{"p"},
|
|
Value: 2222,
|
|
Usage: "listening port",
|
|
EnvVars: []string{"SSHPIPERD_PORT"},
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "server-key",
|
|
Aliases: []string{"i"},
|
|
Usage: "server key files, support wildcard",
|
|
Value: "/etc/ssh/ssh_host_rsa_key",
|
|
EnvVars: []string{"SSHPIPERD_SERVER_KEY"},
|
|
},
|
|
&cli.DurationFlag{
|
|
Name: "login-grace-time",
|
|
Value: 30 * time.Second,
|
|
Usage: "sshpiperd forcely close the connection after this time if the pipe has not successfully established",
|
|
EnvVars: []string{"SSHPIPERD_LOGIN_GRACE_TIME"},
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "log-level",
|
|
Value: "info",
|
|
Usage: "log level, one of: trace, debug, info, warn, error, fatal, panic",
|
|
EnvVars: []string{"SSHPIPERD_LOG_LEVEL"},
|
|
},
|
|
},
|
|
// Commands: []*cli.Command{
|
|
// &cli.Command{
|
|
// Name: "plug",
|
|
|
|
// },
|
|
// // &cli.Command{
|
|
// // Name: "grpc",
|
|
// // Action: func(ctx *cli.Context) error {
|
|
// // return fmt.Errorf("not implemented")
|
|
// // },
|
|
// // },
|
|
// },
|
|
Action: func(ctx *cli.Context) error {
|
|
level, err := log.ParseLevel(ctx.String("log-level"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.SetLevel(level)
|
|
|
|
log.Info("starting sshpiperd version: ", version())
|
|
d, err := newDaemon(ctx)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var plugins []*plugin.GrpcPlugin
|
|
|
|
args := ctx.Args().Slice()
|
|
remain := args
|
|
|
|
for {
|
|
if len(remain) <= 0 {
|
|
break
|
|
}
|
|
|
|
args, remain = splitByDash(remain)
|
|
|
|
if len(args) <= 0 {
|
|
continue
|
|
}
|
|
|
|
exe := args[0]
|
|
cmd := exec.Command(exe)
|
|
cmd.Args = args
|
|
|
|
log.Info("starting plugin: ", cmd.Args)
|
|
|
|
p, err := plugin.DialCmd(cmd)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
go p.RecvLogs(log.StandardLogger().Out)
|
|
|
|
plugins = append(plugins, &p.GrpcPlugin)
|
|
}
|
|
|
|
if err := d.install(plugins...); err != nil {
|
|
return err
|
|
}
|
|
|
|
return d.run()
|
|
},
|
|
}
|
|
|
|
if err := app.Run(os.Args); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|