nps/main.go

package main

import (
	"flag"
	"fmt"
	"log"
)

var (
	configPath   = flag.String("config", "config.json", "配置文件路径")
	tcpPort      = flag.Int("tcpport", 8284, "Socket连接或者监听的端口")
	httpPort     = flag.Int("httpport", 8024, "当mode为server时为服务端监听端口，当为mode为client时为转发至本地客户端的端口")
	rpMode       = flag.String("mode", "client", "启动模式，可选为client|server")
	tunnelTarget = flag.String("target", "10.1.50.203:80", "远程目标")
	verifyKey    = flag.String("vkey", "", "验证密钥")
	u            = flag.String("u", "", "socks5验证用户名")
	p            = flag.String("p", "", "socks5验证密码")
	compress     = flag.String("compress", "", "数据压缩（gzip|snappy）")
	config       Config
	err          error
	DataEncode   int
	DataDecode   int
)

func main() {
	flag.Parse()
	log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
	switch *compress {
	case "":
		DataDecode = COMPRESS_NONE
		DataEncode = COMPRESS_NONE
	case "gzip":
		DataDecode = COMPRESS_GZIP_DECODE
		DataEncode = COMPRESS_GZIP_ENCODE
	case "snnapy":
		DataDecode = COMPRESS_SNAPY_DECODE
		DataEncode = COMPRESS_SNAPY_ENCODE
	default:
		log.Fatalln("数据压缩格式错误")
	}
	if *rpMode == "client" {
		JsonParse := NewJsonStruct()
		config, err = JsonParse.Load(*configPath)
		if err != nil {
			log.Fatalln(err)
		}
		*verifyKey = config.Server.Vkey
		log.Println("客户端启动，连接：", config.Server.Ip, "， 端口：", config.Server.Tcp)
		cli := NewRPClient(fmt.Sprintf("%s:%d", config.Server.Ip, config.Server.Tcp), config.Server.Num)
		cli.Start()
	} else {
		if *verifyKey == "" {
			log.Fatalln("必须输入一个验证的key")
		}
		if *tcpPort <= 0 || *tcpPort >= 65536 {
			log.Fatalln("请输入正确的tcp端口。")
		}
		if *httpPort <= 0 || *httpPort >= 65536 {
			log.Fatalln("请输入正确的http端口。")
		}
		log.Println("服务端启动，监听tcp服务端端口：", *tcpPort, "， 外部服务端端口：", *httpPort)
		if *rpMode == "httpServer" {
			svr := NewHttpModeServer(*tcpPort, *httpPort)
			if err := svr.Start(); err != nil {
				log.Fatalln(err)
			}
		} else if *rpMode == "tunnelServer" {
			svr := NewTunnelModeServer(*tcpPort, *httpPort, *tunnelTarget, ProcessTunnel)
			svr.Start()
		} else if *rpMode == "sock5Server" {
			svr := NewSock5ModeServer(*tcpPort, *httpPort, *u, *p)
			svr.Start()
		} else if *rpMode == "httpProxyServer" {
			svr := NewTunnelModeServer(*tcpPort, *httpPort, *tunnelTarget, ProcessHttp)
			svr.Start()
		} else if *rpMode == "udpServer" {
			svr := NewUdpModeServer(*tcpPort, *httpPort, *tunnelTarget)
			svr.Start()
		}
	}
}