watchtower/wtmock/client_db: clone bytes w/ nil

This commit is contained in:
Conner Fromknecht committed 2019-04-23 20:04:23 -07:00
1 parent c6f95c5667
commit df390df966
1 file changed
+6 -1
+6 -1
View File
@@ -110,7 +110,7 @@ func (m *ClientDB) CreateClientSession(session *wtdb.ClientSession) error {
Policy: session.Policy, Policy: session.Policy,
SeqNum: session.SeqNum, SeqNum: session.SeqNum,
TowerLastApplied: session.TowerLastApplied, TowerLastApplied: session.TowerLastApplied,
RewardPkScript: session.RewardPkScript, RewardPkScript: cloneBytes(session.RewardPkScript),
CommittedUpdates: make(map[uint16]*wtdb.CommittedUpdate), CommittedUpdates: make(map[uint16]*wtdb.CommittedUpdate),
AckedUpdates: make(map[uint16]wtdb.BackupID), AckedUpdates: make(map[uint16]wtdb.BackupID),
} }
@@ -228,7 +228,12 @@ func (m *ClientDB) AddChanPkScript(chanID lnwire.ChannelID, pkScript []byte) err
} }
func cloneBytes(b []byte) []byte { func cloneBytes(b []byte) []byte {
if b == nil {
return nil
}
bb := make([]byte, len(b)) bb := make([]byte, len(b))
copy(bb, b) copy(bb, b)
return bb return bb
} }