fix: apply review feedback

filecoin-project · Oct 25, 2021 · b8529ce · b8529ce
1 parent 8bc0cd3
commit b8529ce
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/filecoin-proofs/src/api/post_util.rs b/filecoin-proofs/src/api/post_util.rs
@@ -363,7 +363,7 @@ pub fn single_partition_vanilla_proofs<Tree: MerkleTreeTrait>(
     info!("single_partition_vanilla_proofs:finish");
 
     ensure!(
-        FallbackPoSt::<Tree>::verify(pub_params, pub_inputs, &partition_proof,)?,
+        FallbackPoSt::<Tree>::verify(pub_params, pub_inputs, &partition_proof)?,
         "partitioned vanilla proofs failed to verify"
     );
 
@@ -384,7 +384,7 @@ pub fn merge_window_post_partition_proofs(
 pub fn get_num_partition_for_fallback_post(config: &PoStConfig, num_sectors: usize) -> usize {
     match config.typ {
         PoStType::Window => {
-            let partitions = ((num_sectors) as f32 / config.sector_count as f32).ceil() as usize;
+            let partitions = (num_sectors as f32 / config.sector_count as f32).ceil() as usize;
             if partitions > 1 {
                 partitions
             } else {

diff --git a/filecoin-proofs/src/api/window_post.rs b/filecoin-proofs/src/api/window_post.rs
@@ -267,6 +267,7 @@ pub fn generate_single_window_post_with_vanilla<Tree: 'static + MerkleTreeTrait>
 
     let vanilla_params = window_post_setup_params(&post_config);
     let partitions = get_partitions_for_window_post(vanilla_proofs.len(), &post_config);
+    ensure!(partition_index < partitions.unwrap_or(1), "Invalid partition index specified");
 
     let setup_params = compound_proof::SetupParams {
         vanilla_params,