Direct proxy works

cli/proxy.go

@@ -0,0 +1,86 @@
+package cli
+
+import (
+	"net"
+	"os"
+	"time"
+
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+
+	"github.com/9seconds/mtg/antireplay"
+	"github.com/9seconds/mtg/config"
+	"github.com/9seconds/mtg/ntp"
+	"github.com/9seconds/mtg/obfuscated2"
+	"github.com/9seconds/mtg/proxy"
+	"github.com/9seconds/mtg/stats"
+	"github.com/9seconds/mtg/telegram"
+)
+
+func Proxy() error {
+	atom := zap.NewAtomicLevel()
+	switch {
+	case config.C.Debug:
+		atom.SetLevel(zapcore.DebugLevel)
+	case config.C.Verbose:
+		atom.SetLevel(zapcore.InfoLevel)
+	default:
+		atom.SetLevel(zapcore.ErrorLevel)
+	}
+
+	encoderCfg := zap.NewProductionEncoderConfig()
+	logger := zap.New(zapcore.NewCore(
+		zapcore.NewJSONEncoder(encoderCfg),
+		zapcore.Lock(os.Stderr),
+		atom,
+	))
+	zap.ReplaceGlobals(logger)
+	defer logger.Sync() // nolint: errcheck
+
+	if err := config.InitPublicAddress(); err != nil {
+		Fatal(err.Error())
+	}
+	zap.S().Debugw("Configuration", "config", config.C)
+
+	if len(config.C.AdTag) > 0 {
+		zap.S().Infow("Use middle proxy connection to Telegram")
+		diff, err := ntp.Fetch()
+		if err != nil {
+			Fatal("Cannot fetch time data from NTP")
+		}
+		if diff > time.Second {
+			Fatal("Your local time is skewed and drift is bigger than a second. Please sync your time.")
+		}
+		go ntp.AutoUpdate()
+	} else {
+		zap.S().Infow("Use direct connection to Telegram")
+	}
+
+	PrintJSONStdout(config.GetURLs())
+
+	if err := antireplay.Init(); err != nil {
+		Fatal(err.Error())
+	}
+	if err := stats.Init(); err != nil {
+		Fatal(err.Error())
+	}
+	proxyListener, err := net.Listen("tcp", config.C.ListenAddr.String())
+	if err != nil {
+		Fatal(err.Error())
+	}
+
+	app := &proxy.Proxy{
+		Logger: zap.S().Named("proxy"),
+	}
+	if len(config.C.AdTag) == 0 {
+		app.TelegramProtocolMaker = obfuscated2.MakeTelegramProtocol
+		app.TelegramDialer = telegram.NewDirectTelegram()
+	}
+	if config.C.SecretMode != config.SecretModeTLS {
+		app.ClientProtocolMaker = obfuscated2.MakeClientProtocol
+	}
+
+	app.Serve(proxyListener)
+
+	return nil
+}