From d78b4b6da22e982c18125e03c454f858f3b332d1 Mon Sep 17 00:00:00 2001
From: Matthias Schneider <ms@wck.biz>
Date: Tue, 1 Mar 2022 10:43:39 +0100
Subject: [PATCH 1/2] protocol: fixed deadline datarace #86

---
 protocol.go | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/protocol.go b/protocol.go
index bfc24e9..6f5641d 100644
--- a/protocol.go
+++ b/protocol.go
@@ -5,6 +5,7 @@ import (
 	"io"
 	"net"
 	"sync"
+	"sync/atomic"
 	"time"
 )
 
@@ -33,15 +34,15 @@ type Listener struct {
 // return the address of the client instead of the proxy address. Each connection
 // will have its own readHeaderTimeout and readDeadline set by the Accept() call.
 type Conn struct {
-	bufReader         *bufio.Reader
+	readDeadline      atomic.Value // time.Time
+	once              sync.Once
+	readErr           error
 	conn              net.Conn
+	Validate          Validator
+	bufReader         *bufio.Reader
 	header            *Header
-	once              sync.Once
 	ProxyHeaderPolicy Policy
-	Validate          Validator
-	readErr           error
 	readHeaderTimeout time.Duration
-	readDeadline      time.Time
 }
 
 // Validator receives a header and decides whether it is a valid one
@@ -215,7 +216,7 @@ func (p *Conn) UDPConn() (conn *net.UDPConn, ok bool) {
 
 // SetDeadline wraps original conn.SetDeadline
 func (p *Conn) SetDeadline(t time.Time) error {
-	p.readDeadline = t
+	p.readDeadline.Store(t)
 	return p.conn.SetDeadline(t)
 }
 
@@ -224,7 +225,7 @@ func (p *Conn) SetReadDeadline(t time.Time) error {
 	// Set a local var that tells us the desired deadline. This is
 	// needed in order to reset the read deadline to the one that is
 	// desired by the user, rather than an empty deadline.
-	p.readDeadline = t
+	p.readDeadline.Store(t)
 	return p.conn.SetReadDeadline(t)
 }
 
@@ -250,7 +251,11 @@ func (p *Conn) readHeader() error {
 	// Therefore, we check whether the error is a net.Timeout and if it is, we decide
 	// the proxy proto does not exist and set the error accordingly.
 	if p.readHeaderTimeout > 0 {
-		p.conn.SetReadDeadline(p.readDeadline)
+		t := p.readDeadline.Load()
+		if t == nil {
+			t = time.Time{}
+		}
+		p.conn.SetReadDeadline(t.(time.Time))
 		if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
 			err = ErrNoProxyProtocol
 		}

From b4887c17f36348fda09047885675ca4c3002ae9d Mon Sep 17 00:00:00 2001
From: Matthias Schneider <ms@wck.biz>
Date: Tue, 1 Mar 2022 13:24:43 +0100
Subject: [PATCH 2/2] added -race for github action tests

---
 .github/workflows/test.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 122a182..59c47b6 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -33,7 +33,7 @@ jobs:
       run: go vet
 
     - name: Test
-      run: go test -v -covermode=count -coverprofile=coverage.out
+      run: go test -race -v -covermode=atomic -coverprofile=coverage.out
 
     - name: actions-goveralls
       uses: shogo82148/actions-goveralls@v1.2.2