nps/cmd/proxy_server/proxy_server.go

package main

import (
	"flag"
	"github.com/astaxie/beego"
	"github.com/cnlh/easyProxy/server"
	"github.com/cnlh/easyProxy/utils"
	_ "github.com/cnlh/easyProxy/web/routers"
	"os"
)

const VERSION = "v0.0.13"

var (
	TcpPort      = flag.Int("tcpport", 0, "客户端与服务端通信端口")
	httpPort     = flag.Int("httpport", 8024, "对外监听的端口")
	rpMode       = flag.String("mode", "webServer", "启动模式")
	tunnelTarget = flag.String("target", "127.0.0.1:80", "远程目标")
	VerifyKey    = flag.String("vkey", "", "验证密钥")
	u            = flag.String("u", "", "验证用户名(socks5和web)")
	p            = flag.String("p", "", "验证密码(socks5和web)")
	compress     = flag.String("compress", "", "数据压缩方式（snappy）")
	crypt        = flag.String("crypt", "false", "是否加密(true|false)")
	logType      = flag.String("log", "stdout", "日志输出方式（stdout|file）")
)

func main() {
	flag.Parse()
	var test bool
	if len(os.Args) > 1 && os.Args[1] == "test" {
		test = true
	}
	utils.InitDaemon("server")
	if *logType == "stdout" || test {
		utils.InitLogFile("server", true)
	} else {
		utils.InitLogFile("server", false)
	}
	task := &utils.Tunnel{
		TcpPort: *httpPort,
		Mode:    *rpMode,
		Target:  *tunnelTarget,
		Config: &utils.Config{
			U:        *u,
			P:        *p,
			Compress: *compress,
			Crypt:    utils.GetBoolByStr(*crypt),
		},
		Flow:         &utils.Flow{},
		UseClientCnf: false,
	}
	if *VerifyKey != "" {
		c := &utils.Client{
			Id:        0,
			VerifyKey: *VerifyKey,
			Addr:      "",
			Remark:    "",
			Status:    true,
			IsConnect: false,
			Cnf:       &utils.Config{},
			Flow:      &utils.Flow{},
		}
		c.Cnf.CompressDecode, c.Cnf.CompressEncode = utils.GetCompressType(c.Cnf.Compress)
		server.CsvDb.Clients[0] = c
		task.Client = c
	}
	if *TcpPort == 0 {
		p, err := beego.AppConfig.Int("tcpport")
		if err == nil && *rpMode == "webServer" {
			*TcpPort = p
		} else {
			*TcpPort = 8284
		}
	}
	utils.Println("服务端启动，监听tcp服务端端口：", *TcpPort)
	task.Config.CompressDecode, task.Config.CompressEncode = utils.GetCompressType(task.Config.Compress)
	if *rpMode != "webServer" {
		server.CsvDb.Tasks[0] = task
	}
	server.StartNewServer(*TcpPort, task, test)
}