[NOD-1308] Don't call wg.done() on handshake if flow failed (#896)

This commit is contained in:
Ori Newman 2020-08-24 14:03:58 +03:00 committed by GitHub
parent b668d98942
commit 8dd7b95423
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -49,22 +49,22 @@ func HandleHandshake(context HandleHandshakeContext, netConnection *netadapter.N
var peerAddress *appmessage.NetAddress
spawn("HandleHandshake-ReceiveVersion", func() {
defer wg.Done()
address, err := ReceiveVersion(context, receiveVersionRoute, outgoingRoute, peer)
if err != nil {
handleError(err, "ReceiveVersion", &isStopping, errChan)
return
}
peerAddress = address
wg.Done()
})
spawn("HandleHandshake-SendVersion", func() {
defer wg.Done()
err := SendVersion(context, sendVersionRoute, outgoingRoute, peer)
if err != nil {
handleError(err, "SendVersion", &isStopping, errChan)
return
}
wg.Done()
})
select {