81 lines
2.9 KiB
Go
81 lines
2.9 KiB
Go
package net
|
||
|
||
import (
|
||
"b612.me/starlog"
|
||
"github.com/spf13/cobra"
|
||
"os"
|
||
"os/signal"
|
||
"time"
|
||
)
|
||
|
||
var (
|
||
tcps TcpServer
|
||
tcpc TcpClient
|
||
)
|
||
|
||
func init() {
|
||
CmdTcps.Flags().StringVarP(&tcps.LocalAddr, "local", "l", "0.0.0.0:29127", "本地地址")
|
||
CmdTcps.Flags().BoolVarP(&tcps.UsingKeepAlive, "keepalive", "k", true, "启用KeepAlive")
|
||
CmdTcps.Flags().IntVarP(&tcps.KeepAlivePeriod, "keepalive-period", "p", 10, "KeepAlive重试周期（秒）")
|
||
CmdTcps.Flags().IntVarP(&tcps.KeepAliveIdel, "keepalive-idel", "i", 15, "KeepAlive空闲时间（秒）")
|
||
CmdTcps.Flags().IntVarP(&tcps.KeepAliveCount, "keepalive-count", "c", 3, "KeepAlive次数）")
|
||
CmdTcps.Flags().BoolVarP(&tcps.Interactive, "interactive", "I", false, "交互模式")
|
||
CmdTcps.Flags().IntVarP(&tcps.UserTimeout, "user-timeout", "u", 0, "用户超时时间（毫秒）")
|
||
CmdTcps.Flags().BoolVarP(&tcps.ShowRecv, "show-recv", "r", true, "显示接收数据")
|
||
CmdTcps.Flags().BoolVarP(&tcps.ShowAsHex, "show-hex", "H", false, "显示十六进制")
|
||
CmdTcps.Flags().StringVarP(&tcps.SaveToFolder, "save", "s", "", "保存到文件夹")
|
||
CmdTcps.Flags().StringVarP(&tcps.LogPath, "log", "L", "", "日志文件路径")
|
||
Cmd.AddCommand(CmdTcps)
|
||
|
||
CmdTcpc.Flags().StringVarP(&tcpc.LocalAddr, "local", "l", "", "本地地址")
|
||
CmdTcpc.Flags().BoolVarP(&tcpc.UsingKeepAlive, "keepalive", "k", true, "启用KeepAlive")
|
||
CmdTcpc.Flags().IntVarP(&tcpc.KeepAlivePeriod, "keepalive-period", "p", 1, "KeepAlive重试周期（秒）")
|
||
CmdTcpc.Flags().IntVarP(&tcpc.KeepAliveIdel, "keepalive-idel", "i", 15, "KeepAlive空闲时间（秒）")
|
||
CmdTcpc.Flags().IntVarP(&tcpc.KeepAliveCount, "keepalive-count", "c", 3, "KeepAlive次数")
|
||
CmdTcpc.Flags().BoolVarP(&tcpc.Interactive, "interactive", "I", false, "交互模式")
|
||
CmdTcpc.Flags().IntVarP(&tcpc.UserTimeout, "user-timeout", "u", 0, "用户超时时间（毫秒）")
|
||
CmdTcpc.Flags().BoolVarP(&tcpc.ShowRecv, "show-recv", "r", true, "显示接收数据")
|
||
CmdTcpc.Flags().BoolVarP(&tcpc.ShowAsHex, "show-hex", "H", false, "显示十六进制")
|
||
CmdTcpc.Flags().StringVarP(&tcpc.SaveToFolder, "save", "s", "", "保存到文件夹")
|
||
CmdTcpc.Flags().StringVarP(&tcpc.LogPath, "log", "L", "", "日志文件路径")
|
||
|
||
Cmd.AddCommand(CmdTcpc)
|
||
}
|
||
|
||
var CmdTcps = &cobra.Command{
|
||
Use: "tcps",
|
||
Short: "TCP服务端",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
go func() {
|
||
s := make(chan os.Signal, 1)
|
||
signal.Notify(s, os.Interrupt, os.Kill)
|
||
<-s
|
||
tcps.Stop()
|
||
time.Sleep(5 * time.Second)
|
||
os.Exit(0)
|
||
}()
|
||
tcps.Run()
|
||
},
|
||
}
|
||
|
||
var CmdTcpc = &cobra.Command{
|
||
Use: "tcpc",
|
||
Short: "TCP客户端",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
if len(args) == 0 {
|
||
starlog.Errorln("请指定目标地址")
|
||
return
|
||
}
|
||
tcpc.RemoteAddr = args[0]
|
||
go func() {
|
||
s := make(chan os.Signal, 1)
|
||
signal.Notify(s, os.Interrupt, os.Kill)
|
||
<-s
|
||
tcpc.Stop()
|
||
time.Sleep(5 * time.Second)
|
||
os.Exit(0)
|
||
}()
|
||
tcpc.Run()
|
||
},
|
||
}
|