fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-23 20:16:27 +08:00
Code Activity

fix(tcp): set conn on peer's endpoint

Browse Source
This commit is contained in:
源文雨
2024-07-16 22:42:35 +09:00
parent c7bbcb9fb7
commit 39d8d5b755
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gold/p2p/tcp/tcp.go
View File

@@ -67,6 +67,7 @@ func (ep *EndPoint) Listen() (p2p.Conn, error) {
type connrecv struct { type connrecv struct {
addr *EndPoint // cast from tcpconn.RemoteAddr() addr *EndPoint // cast from tcpconn.RemoteAddr()
conn *net.TCPConn
pckt packet pckt packet
} }
@@ -121,6 +122,7 @@ func (conn *Conn) receive(ep *EndPoint) {
if tcpconn == nil { if tcpconn == nil {
return return
} }
r.conn = tcpconn
_, err := io.Copy(&r.pckt, tcpconn) _, err := io.Copy(&r.pckt, tcpconn)
if err != nil { if err != nil {
logrus.Debugln("[tcp] recv from", ep, "err:", err) 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 { if p == nil {
return 0, nil, net.ErrClosed return 0, nil, net.ErrClosed
} }
conn.peers.Set(p.addr.String(), p.conn)
if p.pckt.typ == packetTypeNormal { if p.pckt.typ == packetTypeNormal {
break break
} }