diff --git a/network/libp2p_impl.go b/network/libp2p_impl.go index 4a19f2f..53427af 100644 --- a/network/libp2p_impl.go +++ b/network/libp2p_impl.go @@ -242,6 +242,7 @@ func (dtnet *libp2pDataTransferNetwork) handleNewStream(s network.Stream) { for { var received datatransfer.Message var err error + _ = s.SetReadDeadline(time.Now().Add(dtnet.sendMessageTimeout)) switch s.Protocol() { case datatransfer.ProtocolDataTransfer1_2: received, err = message.FromNet(s) @@ -253,8 +254,10 @@ func (dtnet *libp2pDataTransferNetwork) handleNewStream(s network.Stream) { go dtnet.receiver.ReceiveError(err) log.Debugf("net handleNewStream from %s error: %s", p, err) } + _ = s.SetReadDeadline(time.Time{}) return } + _ = s.SetReadDeadline(time.Time{}) ctx := context.Background() log.Debugf("net handleNewStream from %s", p)