star/net/cmd.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 nattestc NatTesterClient
var nettests NatTesterServer

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() {
	CmdNatPClient.Flags().StringVarP(&natc.ServiceTarget, "target", "t", "", "forward server target address")
	CmdNatPClient.Flags().StringVarP(&natc.CmdTarget, "server", "s", "", "nat server command address")
	CmdNatPClient.Flags().StringVarP(&natc.Passwd, "passwd", "p", "", "password")
	CmdNatPClient.Flags().BoolVarP(&natc.enableTCP, "enable-tcp", "T", true, "enable tcp forward")
	CmdNatPClient.Flags().BoolVarP(&natc.enableUDP, "enable-udp", "U", true, "enable udp forward")
	CmdNatPClient.Flags().IntVarP(&natc.DialTimeout, "dial-timeout", "d", 10000, "dial timeout milliseconds")
	CmdNatPClient.Flags().IntVarP(&natc.UdpTimeout, "udp-timeout", "D", 60000, "udp connection timeout milliseconds")
	Cmd.AddCommand(CmdNatPClient)

	CmdNatPServer.Flags().StringVarP(&nats.ListenAddr, "listen", "l", "", "listen address")
	CmdNatPServer.Flags().StringVarP(&nats.Passwd, "passwd", "p", "", "password")
	CmdNatPServer.Flags().Int64VarP(&nats.UDPTimeout, "udp-timeout", "D", 60000, "udp connection timeout milliseconds")
	CmdNatPServer.Flags().Int64VarP(&nats.NetTimeout, "dial-timeout", "d", 10000, "dial timeout milliseconds")
	CmdNatPServer.Flags().BoolVarP(&nats.enableTCP, "enable-tcp", "T", true, "enable tcp forward")
	CmdNatPServer.Flags().BoolVarP(&nats.enableUDP, "enable-udp", "U", true, "enable udp forward")
	Cmd.AddCommand(CmdNatPServer)

	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)

	CmdNatClient.Flags().IntVarP(&nattestc.RetryTime, "retry", "r", 2, "重试次数")
	CmdNatClient.Flags().IntVarP(&nattestc.Timeout, "timeout", "t", 2, "超时时间")
	CmdNatClient.Flags().StringSliceVarP(&nattestc.dns, "dns", "d", nil, "自定义dns服务器")
	Cmd.AddCommand(CmdNatClient)
}

var CmdNatPClient = &cobra.Command{
	Use:   "natpc",
	Short: "nat穿透客户端",
	Run: func(cmd *cobra.Command, args []string) {
		if natc.ServiceTarget == "" || natc.CmdTarget == "" {
			cmd.Help()
			return
		}
		natc.Run()
	},
}

var CmdNatPServer = &cobra.Command{
	Use:   "natps",
	Short: "nat穿透服务端",
	Run: func(cmd *cobra.Command, args []string) {

		nats.Run()
	},
}

var CmdNatClient = &cobra.Command{
	Use:   "natc",
	Short: "nat类型测试工具",
	Long:  "基于RFC3489的nat类型测试工具",
	Run: func(cmd *cobra.Command, args []string) {
		if nattestc.dns != nil {
			UseCustomeDNS(nattestc.dns)
			for _, v := range nattestc.dns {
				starlog.Infoln("使用自定义DNS:", v)
			}
		}
		natserver := "nat.b612.me"
		if len(args) == 0 {
			starlog.Infoln("使用默认NAT测试服务器：nat.b612.me")
		} else {
			natserver = args[0]
			starlog.Infoln("使用NAT测试服务器：", natserver)
		}
		res, err := nattestc.ServeAndRun(natserver)
		if err != nil {
			starlog.Errorln("测试NAT类型失败", err)
			return
		}
		fmt.Println("-----------------\n")
		starlog.Green("您的NAT类型为%v - %v\nswitch可能显示NAT类型为%v\n%v", res.Code, res.RFC3489, res.NintendoSwitch, res.Desc)
	},
}

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("-----------------------------")
		}
	},
}