fix data race in client crypto setup

2017-05-06 23:44:15 +08:00
parent b3419633a6
commit 2014ced934
1 changed files with 3 additions and 0 deletions
--- a/handshake/crypto_setup_client.go
+++ b/handshake/crypto_setup_client.go
@@ -323,6 +323,9 @@ func (h *cryptoSetupClient) GetSealer() (protocol.EncryptionLevel, Sealer) {
 }

 func (h *cryptoSetupClient) GetSealerWithEncryptionLevel(encLevel protocol.EncryptionLevel) (Sealer, error) {
+	h.mutex.RLock()
+	defer h.mutex.RUnlock()
+
 	switch encLevel {
 	case protocol.EncryptionUnencrypted:
 		return h.sealUnencrypted, nil