From 4c5f1a5636901e4d1e9b8d58252bea2af889a875 Mon Sep 17 00:00:00 2001
From: 9seconds <nineseconds@yandex.ru>
Date: Tue, 5 Nov 2019 10:44:23 +0300
Subject: [PATCH] Propagate cloak_port setting

---
 config/config.go | 6 ++++++
 main.go          | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/config/config.go b/config/config.go
index 7bb12d6..bea6989 100644
--- a/config/config.go
+++ b/config/config.go
@@ -55,6 +55,8 @@ const (
 	OptionTypeWriteBufferSize
 	OptionTypeReadBufferSize
 
+	OptionTypeCloakPort
+
 	OptionTypeAntiReplayMaxSize
 	OptionTypeAntiReplayEvictionTime
 
@@ -71,10 +73,12 @@ type Config struct {
 
 	StatsNamespace string            `json:"stats_namespace"`
 	StatsdNetwork  string            `json:"statsd_network"`
+	CloakHost      string            `json:"cloak_host"`
 	StatsdTags     map[string]string `json:"statsd_tags"`
 
 	WriteBuffer int `json:"write_buffer"`
 	ReadBuffer  int `json:"read_buffer"`
+	CloakPort   int `json:"cloak_port"`
 
 	AntiReplayMaxSize      int           `json:"anti_replay_max_size"`
 	AntiReplayEvictionTime time.Duration `json:"anti_replay_eviction_time"`
@@ -144,6 +148,8 @@ func Init(options ...Opt) error { // nolint: gocyclo, funlen
 			C.WriteBuffer = int(opt.Value.(units.Base2Bytes))
 		case OptionTypeReadBufferSize:
 			C.ReadBuffer = int(opt.Value.(units.Base2Bytes))
+		case OptionTypeCloakPort:
+			C.CloakPort = int(opt.Value.(uint16))
 		case OptionTypeAntiReplayMaxSize:
 			C.AntiReplayMaxSize = opt.Value.(int)
 		case OptionTypeAntiReplayEvictionTime:
diff --git a/main.go b/main.go
index 7d0fb31..194545b 100644
--- a/main.go
+++ b/main.go
@@ -93,6 +93,11 @@ var (
 		Envar("MTG_BUFFER_READ").
 		Default("131072KB").
 		Bytes()
+	proxyTLSCloakPort = proxyCommand.Flag("cloak-port",
+		"Port which should be used for host cloaking.").
+		Envar("MTG_CLOAK_PORT").
+		Default("443").
+		Uint16()
 	proxyAntiReplayMaxSize = proxyCommand.Flag("anti-replay-max-size",
 		"Max size of antireplay cache in megabytes.").
 		Envar("MTG_ANTIREPLAY_MAXSIZE").
@@ -134,6 +139,7 @@ func main() {
 			config.Opt{Option: config.OptionTypeStatsdTags, Value: *proxyStatsdTags},
 			config.Opt{Option: config.OptionTypeWriteBufferSize, Value: *proxyWriteBufferSize},
 			config.Opt{Option: config.OptionTypeReadBufferSize, Value: *proxyReadBufferSize},
+			config.Opt{Option: config.OptionTypeCloakPort, Value: *proxyTLSCloakPort},
 			config.Opt{Option: config.OptionTypeAntiReplayMaxSize, Value: *proxyAntiReplayMaxSize},
 			config.Opt{Option: config.OptionTypeAntiReplayEvictionTime, Value: *proxyAntiReplayEvictionTime},
 			config.Opt{Option: config.OptionTypeSecret, Value: *proxySecret},