fix: close conn after read

This commit is contained in:
thinkAfCod 2024-05-30 17:20:29 +08:00 committed by Chen Kai
parent c7e93e70cc
commit c6766840d4

View file

@ -690,6 +690,7 @@ func (p *PortalProtocol) processContent(target *enode.Node, resp []byte) (byte,
}
// Read ALL the data from the connection until EOF and return it
data, err := io.ReadAll(conn)
conn.Close()
if err != nil {
p.Log.Error("failed to read from utp connection", "err", err)
return 0xff, nil, err
@ -1189,6 +1190,7 @@ func (p *PortalProtocol) handleOffer(id enode.ID, addr *net.UDPAddr, request *po
// Read ALL the data from the connection until EOF and return it
var data []byte
data, err = io.ReadAll(conn)
conn.Close()
if err != nil {
p.Log.Error("failed to read from utp connection", "err", err)
return