mirror of
https://github.com/kevwan/tproxy.git
synced 2026-05-23 07:40:35 +00:00
Kevin Wan
45 lines
1.3 KiB
Go
45 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/fatih/color"
|
|
)
|
|
|
|
var settings Settings
|
|
|
|
func main() {
|
|
var (
|
|
localPort = flag.Int("p", 0, "Local port to listen on, default to pick a random port")
|
|
localHost = flag.String("l", "localhost", "Local address to listen on")
|
|
remote = flag.String("r", "", "Remote address (host:port) to connect")
|
|
delay = flag.Duration("d", 0, "the delay to relay packets")
|
|
protocol = flag.String("t", "", "The type of protocol, currently support text, http2, grpc, mysql, redis, mongodb and mqtt")
|
|
stat = flag.Bool("s", false, "Enable statistics")
|
|
quiet = flag.Bool("q", false, "Quiet mode, only prints connection open/close and stats, default false")
|
|
upLimit = flag.Int64("up", 0, "Upward speed limit(bytes/second)")
|
|
downLimit = flag.Int64("down", 0, "Downward speed limit(bytes/second)")
|
|
)
|
|
|
|
if len(os.Args) <= 1 {
|
|
flag.Usage()
|
|
return
|
|
}
|
|
|
|
flag.Parse()
|
|
saveSettings(*localHost, *localPort, *remote, *delay, *protocol, *stat, *quiet, *upLimit, *downLimit)
|
|
|
|
if len(settings.Remote) == 0 {
|
|
fmt.Fprintln(os.Stderr, color.HiRedString("[x] Remote target required"))
|
|
flag.PrintDefaults()
|
|
os.Exit(1)
|
|
}
|
|
|
|
if err := startListener(); err != nil {
|
|
fmt.Fprintln(os.Stderr, color.HiRedString("[x] Failed to start listener: %v", err))
|
|
os.Exit(1)
|
|
}
|
|
}
|