From 2d8e90b31102fa784f1657153db99d6fe52b4e9d Mon Sep 17 00:00:00 2001
From: Chris Kuiper <ckuiper@google.com>
Date: Fri, 3 May 2019 07:01:38 -0700
Subject: Proper cleanup of sockets that used REUSEPORT

Fixed a small logic error that broke proper accounting of MultiPortEndpoints.

PiperOrigin-RevId: 246502126
Change-Id: I1a7d6ea134f811612e545676212899a3707bc2c2
---
 pkg/tcpip/stack/transport_demuxer.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'pkg/tcpip/stack')

diff --git a/pkg/tcpip/stack/transport_demuxer.go b/pkg/tcpip/stack/transport_demuxer.go
index 66c564613..807c3ba5e 100644
--- a/pkg/tcpip/stack/transport_demuxer.go
+++ b/pkg/tcpip/stack/transport_demuxer.go
@@ -171,7 +171,7 @@ func (ep *multiPortEndpoint) singleRegisterEndpoint(t TransportEndpoint) {
 	// A new endpoint is added into endpointsArr and its index there is
 	// saved in endpointsMap. This will allows to remove endpoint from
 	// the array fast.
-	ep.endpointsMap[ep] = len(ep.endpointsArr)
+	ep.endpointsMap[t] = len(ep.endpointsArr)
 	ep.endpointsArr = append(ep.endpointsArr, t)
 }
 
-- 
cgit v1.2.3