add ghost if pruning occurs

starcoinorg · Dec 6, 2024 · 81dbbf5 · 81dbbf5
1 parent 8f01900
commit 81dbbf5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/flexidag/src/blockdag.rs b/flexidag/src/blockdag.rs
@@ -564,8 +564,12 @@ impl BlockDAG {
         blue_blocks: &[BlockHeader],
         header: &BlockHeader,
     ) -> Result<GhostdagData, anyhow::Error> {
-        self.ghost_dag_manager()
-            .verify_and_ghostdata(blue_blocks, header)
+        if header.pruning_point() != HashValue::zero() {
+            self.ghost_dag_manager().ghostdag(&header.parents())
+        } else {
+            self.ghost_dag_manager()
+                .verify_and_ghostdata(blue_blocks, header)
+        }
     }
     pub fn check_upgrade(&self, main: &BlockHeader, genesis_id: HashValue) -> anyhow::Result<()> {
         // set the state with key 0