add customizers for bootstrapping RNG

Author: tim-barry

cmd/bootstrap/cmd/finalize_test.go

@@ -122,7 +122,7 @@ func TestClusterAssignment(t *testing.T) {
 	seed := make([]byte, 32)
 	_, err := cryptoRand.Read(seed)
 	require.NoError(t, err)
-	prng, err := prg.New(seed, nil, nil)
+	prng, err := prg.New(seed, prg.BootstrapClusterAssignment, nil)
 	require.NoError(t, err)
 
 	log := zerolog.Nop()

cmd/bootstrap/cmd/rootblock.go

@@ -250,7 +250,7 @@ func rootBlock(cmd *cobra.Command, args []string) {
 	if err != nil {
 		log.Fatal().Err(err).Msg("unable to generate random seed")
 	}
-	csprg, err := prg.New(randomSeed[:], nil, nil)
+	csprg, err := prg.New(randomSeed[:], prg.BootstrapClusterAssignment, nil)
 	if err != nil {
 		log.Fatal().Err(err).Msg("unable to initialize pseudorandom generator")
 	}
@@ -277,8 +277,12 @@ func rootBlock(cmd *cobra.Command, args []string) {
 	log.Info().Msg("")
 
 	// use the same randomness for the RandomSource of the new epoch
+	randomSourcePrg, err := prg.New(randomSeed[:], prg.BootstrapEpochRandomSource, nil)
+	if err != nil {
+		log.Fatal().Err(err).Msg("failed to initialize pseudorandom generator")
+	}
 	log.Info().Msg("constructing intermediary bootstrapping data")
-	epochSetup, epochCommit, err := constructRootEpochEvents(headerBody.View, participants, assignments, clusterQCs, randomBeaconData, dkgIndexMap, csprg)
+	epochSetup, epochCommit, err := constructRootEpochEvents(headerBody.View, participants, assignments, clusterQCs, randomBeaconData, dkgIndexMap, randomSourcePrg)
 	if err != nil {
 		log.Fatal().Err(err).Msg("failed to construct root epoch events")
 	}

cmd/bootstrap/run/epochs.go

@@ -176,7 +176,7 @@ func GenerateRecoverTxArgsWithDKG(
 		}
 	}
 
-	csprg, err := prg.New(epoch.RandomSource(), nil, nil)
+	csprg, err := prg.New(epoch.RandomSource(), prg.BootstrapClusterAssignment, nil)
 	if err != nil {
 		return nil, fmt.Errorf("could not initialize PRNG: %w", err)
 	}

state/protocol/prg/customizers.go

@@ -25,6 +25,10 @@ var (
 	// ExecutionRandomSourceHistory is the customizer for Flow's transaction execution environment
 	// (used for the source of randomness history core-contract)
 	ExecutionRandomSourceHistory = customizerFromIndices(1, 1)
+	// BootstrapClusterAssignment is the customizer for assignment of collectors to clusters during bootstrapping
+	BootstrapClusterAssignment = customizerFromIndices(2, 0)
+	// BootstrapEpochRandomSource is the customizer for the RandomSource of the new epoch during bootstrapping
+	BootstrapEpochRandomSource = customizerFromIndices(2, 1)
 	//
 	// clusterLeaderSelectionPrefix is the prefix used for CollectorClusterLeaderSelection
 	clusterLeaderSelectionPrefix = []uint16{3}