wire: Further reduce transaction allocs.

This commit drastically reduces the number of allocations needed to
deserialize a transaction and its scripts by using the combination of a
free list for initially deserializing the individual scripts along with
copying them into a single contiguous byte slice after the final size is
known and modifying each script in the transaction to point to its
location within the contiguous blob.

The end result is only a single allocation that holds all of the scripts
for a transaction regardless of the total number of scripts it has.

The script free list allows a maximum of 12,500 items with each buffer
being 512 bytes.  This implies it will have a peak usage of 6.1MB.  The
values were chosen based on profiling data and a desire to allow at
least 100 scripts per transaction to be simultaneously deserialized by
125 peers.

Also, while optimizing, decode directly into the existing previous
outpoint structure of each transaction input in order to avoid the extra
allocation per input that is otherwise caused when the local escapes to
the heap.

The following is a before and after comparison of the allocations
with the benchmarks that did not change removed:

benchmark              old allocs     new allocs     delta
-----------------------------------------------------------
ReadTxOut              1              0              -100.00%
ReadTxIn               2              0              -100.00%
DeserializeTxSmall     7              5              -28.57%
DeserializeTxLarge     11146          6              -99.95%
This commit is contained in:
Dave Collins 2016-04-20 21:08:01 -05:00
parent 2adfb3b56a
commit 6229e35835
2 changed files with 198 additions and 14 deletions

View file

@ -285,6 +285,7 @@ func BenchmarkReadTxOut(b *testing.B) {
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
r.Seek(0, 0) r.Seek(0, 0)
readTxOut(r, 0, 0, &txOut) readTxOut(r, 0, 0, &txOut)
scriptPool.Return(txOut.PkScript)
} }
} }
@ -315,6 +316,7 @@ func BenchmarkReadTxIn(b *testing.B) {
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
r.Seek(0, 0) r.Seek(0, 0)
readTxIn(r, 0, 0, &txIn) readTxIn(r, 0, 0, &txIn)
scriptPool.Return(txIn.SignatureScript)
} }
} }

View file

@ -24,14 +24,14 @@ const (
MaxPrevOutIndex uint32 = 0xffffffff MaxPrevOutIndex uint32 = 0xffffffff
) )
// defaultTxInOutAlloc is the default size used for the backing array for
// transaction inputs and outputs. The array will dynamically grow as needed,
// but this figure is intended to provide enough space for the number of
// inputs and outputs in a typical transaction without needing to grow the
// backing array multiple times.
const defaultTxInOutAlloc = 15
const ( const (
// defaultTxInOutAlloc is the default size used for the backing array
// for transaction inputs and outputs. The array will dynamically grow
// as needed, but this figure is intended to provide enough space for
// the number of inputs and outputs in a typical transaction without
// needing to grow the backing array multiple times.
defaultTxInOutAlloc = 15
// minTxInPayload is the minimum payload size for a transaction input. // minTxInPayload is the minimum payload size for a transaction input.
// PreviousOutPoint.Hash + PreviousOutPoint.Index 4 bytes + Varint for // PreviousOutPoint.Hash + PreviousOutPoint.Index 4 bytes + Varint for
// SignatureScript length 1 byte + Sequence 4 bytes. // SignatureScript length 1 byte + Sequence 4 bytes.
@ -57,8 +57,80 @@ const (
// number of transaction outputs 1 byte + LockTime 4 bytes + min input // number of transaction outputs 1 byte + LockTime 4 bytes + min input
// payload + min output payload. // payload + min output payload.
minTxPayload = 10 minTxPayload = 10
// freeListMaxScriptSize is the size of each buffer in the free list
// that is used for deserializing scripts from the wire before they are
// concatenated into a single contiguous buffers. This value was chosen
// because it is slightly more than twice the size of the vast majority
// of all "standard" scripts. Larger scripts are still deserialized
// properly as the free list will simply be bypassed for them.
freeListMaxScriptSize = 512
// freeListMaxItems is the number of buffers to keep in the free list
// to use for script deserialization. This value allows up to 100
// scripts per transaction being simultaneously deserialized by 125
// peers. Thus, the peak usage of the free list is 12,500 * 512 =
// 6,400,000 bytes.
freeListMaxItems = 12500
) )
// scriptFreeList defines a free list of byte slices (up to the maximum number
// defined by the freeListMaxItems constant) that have a cap according to the
// freeListMaxScriptSize constant. It is used to provide temporary buffers for
// deserializing scripts in order to greatly reduce the number of allocations
// required.
//
// The caller can obtain a buffer from the free list by calling the Borrow
// function and should return it via the Return function when done using it.
type scriptFreeList chan []byte
// Borrow returns a byte slice from the free list with a length according the
// provided size. A new buffer is allocated if there are any items available.
//
// When the size is larger than the max size allowed for items on the free list
// a new buffer of the appropriate size is allocated and returned. It is safe
// to attempt to return said buffer via the Return function as it will be
// ignored and allowed to go the garbage collector.
func (c scriptFreeList) Borrow(size uint64) []byte {
if size > freeListMaxScriptSize {
return make([]byte, size, size)
}
var buf []byte
select {
case buf = <-c:
default:
buf = make([]byte, freeListMaxScriptSize)
}
return buf[:size]
}
// Return puts the provided byte slice back on the free list when it has a cap
// of the expected length. The buffer is expected to have been obtained via
// the Borrow function. Any slices that are not of the appropriate size, such
// as those whose size is greater than the largest allowed free list item size
// are simply ignored so they can go to the garbage collector.
func (c scriptFreeList) Return(buf []byte) {
// Ignore any buffers returned that aren't the expected size for the
// free list.
if cap(buf) != freeListMaxScriptSize {
return
}
// Return the buffer to the free list when it's not full. Otherwise let
// it be garbage collected.
select {
case c <- buf:
default:
// Let it go to the garbage collector.
}
}
// Create the concurrent safe free list to use for script deserialization. As
// previously described, this free list is maintained to significantly reduce
// the number of allocations.
var scriptPool scriptFreeList = make(chan []byte, freeListMaxItems)
// OutPoint defines a bitcoin data type that is used to track previous // OutPoint defines a bitcoin data type that is used to track previous
// transaction outputs. // transaction outputs.
type OutPoint struct { type OutPoint struct {
@ -263,20 +335,46 @@ func (msg *MsgTx) BtcDecode(r io.Reader, pver uint32) error {
return messageError("MsgTx.BtcDecode", str) return messageError("MsgTx.BtcDecode", str)
} }
// returnScriptBuffers is a closure that returns any script buffers that
// were borrowed from the pool when there are any deserialization
// errors. This is only valid to call before the final step which
// replaces the scripts with the location in a contiguous buffer and
// returns them.
returnScriptBuffers := func() {
for _, txIn := range msg.TxIn {
if txIn == nil || txIn.SignatureScript == nil {
continue
}
scriptPool.Return(txIn.SignatureScript)
}
for _, txOut := range msg.TxOut {
if txOut == nil || txOut.PkScript == nil {
continue
}
scriptPool.Return(txOut.PkScript)
}
}
// Deserialize the inputs. // Deserialize the inputs.
var totalScriptSize uint64
txIns := make([]TxIn, count) txIns := make([]TxIn, count)
msg.TxIn = make([]*TxIn, count) msg.TxIn = make([]*TxIn, count)
for i := uint64(0); i < count; i++ { for i := uint64(0); i < count; i++ {
// The pointer is set now in case a script buffer is borrowed
// and needs to be returned to the pool on error.
ti := &txIns[i] ti := &txIns[i]
msg.TxIn[i] = ti
err = readTxIn(r, pver, msg.Version, ti) err = readTxIn(r, pver, msg.Version, ti)
if err != nil { if err != nil {
returnScriptBuffers()
return err return err
} }
msg.TxIn[i] = ti totalScriptSize += uint64(len(ti.SignatureScript))
} }
count, err = ReadVarInt(r, pver) count, err = ReadVarInt(r, pver)
if err != nil { if err != nil {
returnScriptBuffers()
return err return err
} }
@ -284,6 +382,7 @@ func (msg *MsgTx) BtcDecode(r io.Reader, pver uint32) error {
// message. It would be possible to cause memory exhaustion and panics // message. It would be possible to cause memory exhaustion and panics
// without a sane upper bound on this count. // without a sane upper bound on this count.
if count > uint64(maxTxOutPerMessage) { if count > uint64(maxTxOutPerMessage) {
returnScriptBuffers()
str := fmt.Sprintf("too many output transactions to fit into "+ str := fmt.Sprintf("too many output transactions to fit into "+
"max message size [count %d, max %d]", count, "max message size [count %d, max %d]", count,
maxTxOutPerMessage) maxTxOutPerMessage)
@ -294,19 +393,73 @@ func (msg *MsgTx) BtcDecode(r io.Reader, pver uint32) error {
txOuts := make([]TxOut, count) txOuts := make([]TxOut, count)
msg.TxOut = make([]*TxOut, count) msg.TxOut = make([]*TxOut, count)
for i := uint64(0); i < count; i++ { for i := uint64(0); i < count; i++ {
// The pointer is set now in case a script buffer is borrowed
// and needs to be returned to the pool on error.
to := &txOuts[i] to := &txOuts[i]
msg.TxOut[i] = to
err = readTxOut(r, pver, msg.Version, to) err = readTxOut(r, pver, msg.Version, to)
if err != nil { if err != nil {
returnScriptBuffers()
return err return err
} }
msg.TxOut[i] = to totalScriptSize += uint64(len(to.PkScript))
} }
msg.LockTime, err = binarySerializer.Uint32(r, littleEndian) msg.LockTime, err = binarySerializer.Uint32(r, littleEndian)
if err != nil { if err != nil {
returnScriptBuffers()
return err return err
} }
// Create a single allocation to house all of the scripts and set each
// input signature script and output public key script to the
// appropriate subslice of the overall contiguous buffer. Then, return
// each individual script buffer back to the pool so they can be reused
// for future deserializations. This is done because it significantly
// reduces the number of allocations the garbage collector needs to
// track, which in turn improves performance and drastically reduces the
// amount of runtime overhead that would otherwise be needed to keep
// track of millions of small allocations.
//
// NOTE: It is no longer valid to call the returnScriptBuffers closure
// after these blocks of code run because it is already done and the
// scripts in the transaction inputs and outputs no longer point to the
// buffers.
var offset uint64
scripts := make([]byte, totalScriptSize)
for i := 0; i < len(msg.TxIn); i++ {
// Copy the signature script into the contiguous buffer at the
// appropriate offset.
signatureScript := msg.TxIn[i].SignatureScript
copy(scripts[offset:], signatureScript)
// Reset the signature script of the transaction input to the
// slice of the contiguous buffer where the script lives.
scriptSize := uint64(len(signatureScript))
end := offset + scriptSize
msg.TxIn[i].SignatureScript = scripts[offset:end:end]
offset += scriptSize
// Return the temporary script buffer to the pool.
scriptPool.Return(signatureScript)
}
for i := 0; i < len(msg.TxOut); i++ {
// Copy the public key script into the contiguous buffer at the
// appropriate offset.
pkScript := msg.TxOut[i].PkScript
copy(scripts[offset:], pkScript)
// Reset the public key script of the transaction output to the
// slice of the contiguous buffer where the script lives.
scriptSize := uint64(len(pkScript))
end := offset + scriptSize
msg.TxOut[i].PkScript = scripts[offset:end:end]
offset += scriptSize
// Return the temporary script buffer to the pool.
scriptPool.Return(pkScript)
}
return nil return nil
} }
@ -500,17 +653,46 @@ func writeOutPoint(w io.Writer, pver uint32, version int32, op *OutPoint) error
return nil return nil
} }
// readScript reads a variable length byte array that represents a transaction
// script. It is encoded as a varInt containing the length of the array
// followed by the bytes themselves. An error is returned if the length is
// greater than the passed maxAllowed parameter which helps protect against
// memory exhuastion attacks and forced panics thorugh malformed messages. The
// fieldName parameter is only used for the error message so it provides more
// context in the error.
func readScript(r io.Reader, pver uint32, maxAllowed uint32, fieldName string) ([]byte, error) {
count, err := ReadVarInt(r, pver)
if err != nil {
return nil, err
}
// Prevent byte array larger than the max message size. It would
// be possible to cause memory exhaustion and panics without a sane
// upper bound on this count.
if count > uint64(maxAllowed) {
str := fmt.Sprintf("%s is larger than the max allowed size "+
"[count %d, max %d]", fieldName, count, maxAllowed)
return nil, messageError("readScript", str)
}
b := scriptPool.Borrow(count)
_, err = io.ReadFull(r, b)
if err != nil {
scriptPool.Return(b)
return nil, err
}
return b, nil
}
// readTxIn reads the next sequence of bytes from r as a transaction input // readTxIn reads the next sequence of bytes from r as a transaction input
// (TxIn). // (TxIn).
func readTxIn(r io.Reader, pver uint32, version int32, ti *TxIn) error { func readTxIn(r io.Reader, pver uint32, version int32, ti *TxIn) error {
var op OutPoint err := readOutPoint(r, pver, version, &ti.PreviousOutPoint)
err := readOutPoint(r, pver, version, &op)
if err != nil { if err != nil {
return err return err
} }
ti.PreviousOutPoint = op
ti.SignatureScript, err = ReadVarBytes(r, pver, MaxMessagePayload, ti.SignatureScript, err = readScript(r, pver, MaxMessagePayload,
"transaction input signature script") "transaction input signature script")
if err != nil { if err != nil {
return err return err
@ -553,7 +735,7 @@ func readTxOut(r io.Reader, pver uint32, version int32, to *TxOut) error {
return err return err
} }
to.PkScript, err = ReadVarBytes(r, pver, MaxMessagePayload, to.PkScript, err = readScript(r, pver, MaxMessagePayload,
"transaction output public key script") "transaction output public key script")
if err != nil { if err != nil {
return err return err