From 39d8d5b75597a43fc57bdc31b9f7518104579a07 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Tue, 16 Jul 2024 22:42:35 +0900
Subject: [PATCH] fix(tcp): set conn on peer's endpoint

---
 gold/p2p/tcp/tcp.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/gold/p2p/tcp/tcp.go b/gold/p2p/tcp/tcp.go
index 883601a..2f641df 100644
--- a/gold/p2p/tcp/tcp.go
+++ b/gold/p2p/tcp/tcp.go
@@ -67,6 +67,7 @@ func (ep *EndPoint) Listen() (p2p.Conn, error) {
 
 type connrecv struct {
 	addr *EndPoint // cast from tcpconn.RemoteAddr()
+	conn *net.TCPConn
 	pckt packet
 }
 
@@ -121,6 +122,7 @@ func (conn *Conn) receive(ep *EndPoint) {
 		if tcpconn == nil {
 			return
 		}
+		r.conn = tcpconn
 		_, err := io.Copy(&r.pckt, tcpconn)
 		if err != nil {
 			logrus.Debugln("[tcp] recv from", ep, "err:", err)
@@ -163,6 +165,7 @@ func (conn *Conn) ReadFromPeer(b []byte) (int, p2p.EndPoint, error) {
 		if p == nil {
 			return 0, nil, net.ErrClosed
 		}
+		conn.peers.Set(p.addr.String(), p.conn)
 		if p.pckt.typ == packetTypeNormal {
 			break
 		}