p2p: fix close data race

This commit is contained in:
Péter Szilágyi 2015-06-09 16:05:40 +03:00
parent ca8cb65b73
commit ff84352fb7
1 changed files with 1 additions and 0 deletions

View File

@ -102,6 +102,7 @@ func (t *rlpx) doProtoHandshake(our *protoHandshake) (their *protoHandshake, err
werr := make(chan error, 1)
go func() { werr <- Send(t.rw, handshakeMsg, our) }()
if their, err = readProtocolHandshake(t.rw, our); err != nil {
<-werr // make sure the write terminates too
return nil, err
}
if err := <-werr; err != nil {