Add collective sig message and broadcasting it after prepare phase of cosi

blockchain/block.go

@@ -23,7 +23,7 @@ type Block struct {
 	ShardId         uint32
 	Hash            [32]byte
 	// Signature...
-	bitmap    []byte   // Contains which validator signed the block.
+	Bitmap    []byte   // Contains which validator signed the block.
 	Signature [66]byte // Schnorr collective signature
 }
 

consensus/consensus_leader.go

@@ -247,15 +247,26 @@ func (consensus *Consensus) processResponseMessage(payload []byte) {
 				log.Error("Failed to aggregate responses")
 				return
 			}
-			collectiveSig, err := crypto.Sign(crypto.Ed25519Curve, consensus.aggregatedCommitment, aggResponse, consensus.bitmap)
+			collectiveSigAndBitmap, err := crypto.Sign(crypto.Ed25519Curve, consensus.aggregatedCommitment, aggResponse, consensus.bitmap)
 
 			if err != nil {
 				log.Error("Failed to create collective signature")
 				return
 			} else {
-				log.Info("CollectiveSig created.", "size", len(collectiveSig))
+				log.Info("CollectiveSig and Bitmap created.", "size", len(collectiveSigAndBitmap))
 			}
 
+			// Start the second round of Cosi
+			collectiveSig := [64]byte{}
+			copy(collectiveSig[:], collectiveSigAndBitmap[:64])
+			bitmap := collectiveSigAndBitmap[64:]
+			msgToSend := consensus.constructCollectiveSigMessage(collectiveSig, bitmap)
+
+			p2p.BroadcastMessage(consensus.getValidatorPeers(), msgToSend)
+
+			// Set state to CHALLENGE_DONE
+			consensus.state = COLLECTIVE_SIG_DONE
+
 			consensus.Log.Debug("Consensus reached with signatures.", "numOfSignatures", len(consensus.responses))
 			// Reset state to FINISHED, and clear other data.
 			consensus.ResetState()
@@ -273,7 +284,8 @@ func (consensus *Consensus) processResponseMessage(payload []byte) {
 
 			// Sign the block
 			// TODO(RJ): populate bitmap
-			copy(blockHeaderObj.Signature[:], collectiveSig)
+			copy(blockHeaderObj.Signature[:], collectiveSig[:])
+			copy(blockHeaderObj.Bitmap[:], bitmap)
 			consensus.OnConsensusDone(&blockHeaderObj)
 
 			// TODO: @ricl these logic are irrelevant to consensus, move them to another file, say profiler.