From 4e4430e7c5b64ed2899b235a0b51b290305be866 Mon Sep 17 00:00:00 2001
From: dingjun <fangdingjun@gmail.com>
Date: Wed, 4 Aug 2021 16:02:28 +0800
Subject: [PATCH] fix issue when upstream struct change

---
 main.go | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/main.go b/main.go
index 563e63e..902531d 100644
--- a/main.go
+++ b/main.go
@@ -74,13 +74,11 @@ func forward(ctx context.Context, c net.Conn, data []byte, dst string) {
 		raddr := c.RemoteAddr().(*net.TCPAddr)
 		glog.Debugf("connection from %s", raddr)
 		hdr = proxyproto.Header{
-			Version:            1,
-			Command:            proxyproto.PROXY,
-			TransportProtocol:  proxyproto.TCPv4,
-			SourceAddress:      raddr.IP.To4(),
-			DestinationAddress: net.IP{0, 0, 0, 0},
-			SourcePort:         uint16(raddr.Port),
-			DestinationPort:    0,
+			Version:           1,
+			Command:           proxyproto.PROXY,
+			TransportProtocol: proxyproto.TCPv4,
+			SourceAddr:        raddr,
+			DestinationAddr:   &net.TCPAddr{},
 		}
 
 		switch strings.ToLower(ss[1]) {