services: fix changing of main tx size of Notary request

This commit is contained in:
Anna Shaleva 2022-04-05 15:12:40 +03:00
parent 20c0e2f2e2
commit 5771bbb52a

View file

@ -218,8 +218,15 @@ func (n *Notary) OnNewRequest(payload *payload.P2PNotaryRequest) {
r.minNotValidBefore = nvbFallback r.minNotValidBefore = nvbFallback
} }
} else { } else {
// Avoid changes in main transaction witnesses got from notary request pool to
// keep the pooled tx valid. We will update its copy => the copy's size will be changed.
cp := *payload.MainTransaction
cp.Scripts = make([]transaction.Witness, len(payload.MainTransaction.Scripts))
for i, w := range payload.MainTransaction.Scripts {
cp.Scripts[i] = w
}
r = &request{ r = &request{
main: payload.MainTransaction, main: &cp,
minNotValidBefore: nvbFallback, minNotValidBefore: nvbFallback,
} }
n.requests[payload.MainTransaction.Hash()] = r n.requests[payload.MainTransaction.Hash()] = r
@ -227,6 +234,8 @@ func (n *Notary) OnNewRequest(payload *payload.P2PNotaryRequest) {
if r.witnessInfo == nil && validationErr == nil { if r.witnessInfo == nil && validationErr == nil {
r.witnessInfo = newInfo r.witnessInfo = newInfo
} }
// Allow modification of fallback transaction got from notary request pool.
// It has dummy Notary witness attached => its size won't be changed.
r.fallbacks = append(r.fallbacks, payload.FallbackTransaction) r.fallbacks = append(r.fallbacks, payload.FallbackTransaction)
if exists && r.isMainCompleted() || validationErr != nil { if exists && r.isMainCompleted() || validationErr != nil {
return return