context: return preBlock from (*Context).PreBlock()

From the user's PoW PreBlock should be returned from this method, not PreHeader, otherwise the behaviour is misleading and causes bugs. Signed-off-by: Anna Shaleva <[email protected]>
nspcc-dev · Oct 30, 2024 · f25e7ce · f25e7ce
1 parent ba32988
commit f25e7ce
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/context.go b/context.go
@@ -214,8 +214,11 @@ func (c *Context[H]) MoreThanFNodesCommittedOrLost() bool {
 	return c.CountCommitted()+c.CountFailed() > c.F()
 }
 
+// PreBlock returns current PreBlock from context. May be nil in case if no
+// PreBlock is constructed yet (even if PreHeader is already constructed).
+// External changes in the PreBlock will be seen by dBFT.
 func (c *Context[H]) PreBlock() PreBlock[H] {
-	return c.preHeader // without transactions
+	return c.preBlock
 }
 
 func (c *Context[H]) reset(view byte, ts uint64) {