From 36b310133ce0e1a124fb2c7c0d33d7d0d582d694 Mon Sep 17 00:00:00 2001
From: WeidiDeng <weidi_deng@icloud.com>
Date: Tue, 24 May 2022 11:01:00 +0800
Subject: [PATCH] fix: IP address is empty in unix socket mode (#1314)

---
 main.go              | 1 +
 pkg/conf/conf.go     | 3 ++-
 pkg/conf/defaults.go | 3 ++-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index 815612e..69e628b 100644
--- a/main.go
+++ b/main.go
@@ -76,6 +76,7 @@ func main() {
 			}
 		}
 
+		api.TrustedPlatform = conf.UnixConfig.ProxyHeader
 		util.Log().Info("开始监听 %s", conf.UnixConfig.Listen)
 		if err := api.RunUnix(conf.UnixConfig.Listen); err != nil {
 			util.Log().Error("无法监听[%s]，%s", conf.UnixConfig.Listen, err)
diff --git a/pkg/conf/conf.go b/pkg/conf/conf.go
index 35ed8b6..a9f7d06 100644
--- a/pkg/conf/conf.go
+++ b/pkg/conf/conf.go
@@ -35,7 +35,8 @@ type ssl struct {
 }
 
 type unix struct {
-	Listen string
+	Listen      string
+	ProxyHeader string `validate:"required_with=Listen"`
 }
 
 // slave 作为slave存储端配置
diff --git a/pkg/conf/defaults.go b/pkg/conf/defaults.go
index 44c5371..eabaee5 100644
--- a/pkg/conf/defaults.go
+++ b/pkg/conf/defaults.go
@@ -45,7 +45,8 @@ var SSLConfig = &ssl{
 }
 
 var UnixConfig = &unix{
-	Listen: "",
+	Listen:      "",
+	ProxyHeader: "X-Forwarded-For",
 }
 
 var OptionOverwrite = map[string]interface{}{}