98 lines
3.3 KiB
Go
98 lines
3.3 KiB
Go
package net
|
||
|
||
import (
|
||
"b612.me/apps/b612/netforward"
|
||
"b612.me/starlog"
|
||
"fmt"
|
||
"github.com/spf13/cobra"
|
||
"time"
|
||
)
|
||
|
||
var Cmd = &cobra.Command{
|
||
Use: "net",
|
||
Short: "网络工具，包括nat穿透，端口转发等",
|
||
}
|
||
|
||
func init() {
|
||
Cmd.AddCommand(netforward.CmdNetforward)
|
||
}
|
||
|
||
var natc NatClient
|
||
var nats NatServer
|
||
|
||
var dns, ipinfoaddr string
|
||
var timeout int
|
||
var maxHop int
|
||
var disableIpInfo bool
|
||
var bindAddr string
|
||
var hideIncorrect bool
|
||
|
||
func init() {
|
||
CmdNatClient.Flags().StringVarP(&natc.ServiceTarget, "target", "t", "", "forward server target address")
|
||
CmdNatClient.Flags().StringVarP(&natc.CmdTarget, "server", "s", "", "nat server command address")
|
||
CmdNatClient.Flags().StringVarP(&natc.Passwd, "passwd", "p", "", "password")
|
||
CmdNatClient.Flags().BoolVarP(&natc.enableTCP, "enable-tcp", "T", true, "enable tcp forward")
|
||
CmdNatClient.Flags().BoolVarP(&natc.enableUDP, "enable-udp", "U", true, "enable udp forward")
|
||
CmdNatClient.Flags().IntVarP(&natc.DialTimeout, "dial-timeout", "d", 10000, "dial timeout milliseconds")
|
||
CmdNatClient.Flags().IntVarP(&natc.UdpTimeout, "udp-timeout", "D", 60000, "udp connection timeout milliseconds")
|
||
Cmd.AddCommand(CmdNatClient)
|
||
|
||
CmdNatServer.Flags().StringVarP(&nats.ListenAddr, "listen", "l", "", "listen address")
|
||
CmdNatServer.Flags().StringVarP(&nats.Passwd, "passwd", "p", "", "password")
|
||
CmdNatServer.Flags().Int64VarP(&nats.UDPTimeout, "udp-timeout", "D", 60000, "udp connection timeout milliseconds")
|
||
CmdNatServer.Flags().Int64VarP(&nats.NetTimeout, "dial-timeout", "d", 10000, "dial timeout milliseconds")
|
||
CmdNatServer.Flags().BoolVarP(&nats.enableTCP, "enable-tcp", "T", true, "enable tcp forward")
|
||
CmdNatServer.Flags().BoolVarP(&nats.enableUDP, "enable-udp", "U", true, "enable udp forward")
|
||
Cmd.AddCommand(CmdNatServer)
|
||
|
||
CmdNetTrace.Flags().StringVarP(&dns, "dns", "d", "", "自定义dns服务器")
|
||
CmdNetTrace.Flags().StringVarP(&ipinfoaddr, "ipinfo", "i", "https://ip.b612.me/{ip}/detail", "自定义ip信息查询地址")
|
||
CmdNetTrace.Flags().IntVarP(&timeout, "timeout", "t", 800, "超时时间，单位毫秒")
|
||
CmdNetTrace.Flags().IntVarP(&maxHop, "max-hop", "m", 32, "最大跳数")
|
||
CmdNetTrace.Flags().BoolVarP(&disableIpInfo, "disable-ipinfo", "D", false, "禁用ip信息查询")
|
||
CmdNetTrace.Flags().StringVarP(&bindAddr, "bind", "b", "0.0.0.0", "绑定地址")
|
||
CmdNetTrace.Flags().BoolVarP(&hideIncorrect, "hide-incorrect", "H", false, "隐藏错误节点")
|
||
Cmd.AddCommand(CmdNetTrace, cmdSSHJar)
|
||
|
||
}
|
||
|
||
var CmdNatClient = &cobra.Command{
|
||
Use: "natc",
|
||
Short: "nat穿透客户端",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
if natc.ServiceTarget == "" || natc.CmdTarget == "" {
|
||
cmd.Help()
|
||
return
|
||
}
|
||
natc.Run()
|
||
},
|
||
}
|
||
|
||
var CmdNatServer = &cobra.Command{
|
||
Use: "nats",
|
||
Short: "nat穿透服务端",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
|
||
nats.Run()
|
||
},
|
||
}
|
||
|
||
var CmdNetTrace = &cobra.Command{
|
||
Use: "trace",
|
||
Short: "网络路径追踪",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
if len(args) == 0 {
|
||
cmd.Help()
|
||
return
|
||
}
|
||
if disableIpInfo {
|
||
ipinfoaddr = ""
|
||
}
|
||
for _, target := range args {
|
||
starlog.Infoln("Traceroute to ", target)
|
||
Traceroute(target, bindAddr, dns, maxHop, time.Millisecond*time.Duration(timeout), ipinfoaddr, hideIncorrect)
|
||
fmt.Println("-----------------------------")
|
||
}
|
||
},
|
||
}
|