remove BlockWithExtData in favor of WithBlockExtras taking a reca…

…lc argument
ava-labs · Jan 21, 2025 · c068eea · c068eea
1 parent 3922e9e
commit c068eea
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 31 deletions.
diff --git a/core/rawdb/accessors_chain.go b/core/rawdb/accessors_chain.go
@@ -523,7 +523,7 @@ func ReadBlock(db ethdb.Reader, hash common.Hash, number uint64) *types.Block {
 		return nil
 	}
 	block := types.NewBlockWithHeader(header).WithBody(body.Transactions, body.Uncles)
-	block = types.BlockWithExtData(block, body.Version, body.ExtData)
+	block = types.WithBlockExtras(block, body.Version, body.ExtData, false)
 	return block
 }
 

diff --git a/core/types/block_ext.go b/core/types/block_ext.go
@@ -83,33 +83,6 @@ func (b *blockSerializable) updateToExtras(extras *BlockExtra) {
 	extras.extdata = b.ExtData
 }
 
-func BlockWithExtData(b *Block, version uint32, extdata *[]byte) *Block {
-	BlockExtras(b).version = version
-	setExtDataHelper(b, extdata, false)
-	return b
-}
-
-func setExtDataHelper(b *Block, data *[]byte, recalc bool) {
-	if data == nil {
-		setExtData(b, nil, recalc)
-		return
-	}
-	setExtData(b, *data, recalc)
-}
-
-func setExtData(b *Block, data []byte, recalc bool) {
-	_data := make([]byte, len(data))
-	extras := BlockExtras(b)
-	extras.extdata = &_data
-	copy(*extras.extdata, data)
-	if recalc {
-		header := b.Header()
-		headerExtra := HeaderExtras(header)
-		headerExtra.ExtDataHash = CalcExtDataHash(_data)
-		b.SetHeader(header)
-	}
-}
-
 func BlockExtData(b *Block) []byte {
 	extras := BlockExtras(b)
 	if extras.extdata == nil {
@@ -140,7 +113,7 @@ func NewBlockWithExtData(
 	header *Header, txs []*Transaction, uncles []*Header, receipts []*Receipt,
 	hasher TrieHasher, extdata []byte, recalc bool,
 ) *Block {
-	b := ethtypes.NewBlock(header, txs, uncles, receipts, hasher)
+	b := NewBlock(header, txs, uncles, receipts, hasher)
 	const version = 0
 	return WithBlockExtras(b, version, &extdata, recalc)
 }

diff --git a/core/types/libevm.go b/core/types/libevm.go
@@ -35,7 +35,23 @@ func BlockExtras(b *Block) *BlockExtra {
 }
 
 func WithBlockExtras(b *Block, version uint32, extdata *[]byte, recalc bool) *Block {
-	BlockExtras(b).version = version
-	setExtDataHelper(b, extdata, recalc)
+	extras := BlockExtras(b)
+
+	extras.version = version
+
+	cpy := make([]byte, 0)
+	if extdata != nil {
+		cpy = make([]byte, len(*extdata))
+		copy(cpy, *extdata)
+	}
+
+	extras.extdata = &cpy
+	if recalc {
+		header := b.Header()
+		headerExtra := HeaderExtras(header)
+		headerExtra.ExtDataHash = CalcExtDataHash(cpy)
+		b.SetHeader(header)
+	}
+
 	return b
 }