bug fix: lock the chunk map

addresses #204
schollz · Mar 18, 2020 · aae5604 · aae5604
1 parent 09d35d2
commit aae5604
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/croc/croc.go b/src/croc/croc.go
@@ -775,10 +775,12 @@ func (c *Client) processMessage(payload []byte) (done bool, err error) {
 		c.CurrentFileChunkRanges = remoteFile.CurrentFileChunkRanges
 		c.CurrentFileChunks = utils.ChunkRangesToChunks(c.CurrentFileChunkRanges)
 		log.Debugf("current file chunks: %+v", c.CurrentFileChunks)
+		c.mutex.Lock()
 		c.chunkMap = make(map[uint64]struct{})
 		for _, chunk := range c.CurrentFileChunks {
 			c.chunkMap[uint64(chunk)] = struct{}{}
 		}
+		c.mutex.Unlock()
 		c.Step3RecipientRequestFile = true
 
 		if c.Options.Ask {