TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-12-05 03:58:23 +00:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #1901 from nspcc-dev/nep5-tracking-fixes

Browse source 
NEP5 tracking fixes
This commit is contained in:
Roman Khimov 2021-04-12 17:04:55 +03:00 committed by GitHub
commit d8c8593410
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
pkg/core/blockchain.go
View file

@ -945,15 +945,14 @@ func (bc *Blockchain) processNEP5Transfer(cache *dao.Cached, transfer *state.NEP
return
}
bs := balances.Trackers[transfer.Asset]
if bs.Balance == nil {
return
}
bs.Balance.Sub(bs.Balance, transfer.Amount)
if bs.Balance.Sign() > 0 {
bs.LastUpdatedBlock = transfer.Block
balances.Trackers[transfer.Asset] = bs
} else {
delete(balances.Trackers, transfer.Asset)
if bs.Balance != nil {
bs.Balance.Sub(bs.Balance, transfer.Amount)
if bs.Balance.Sign() > 0 {
bs.LastUpdatedBlock = transfer.Block
balances.Trackers[transfer.Asset] = bs
} else {
delete(balances.Trackers, transfer.Asset)
}
}
transfer.Amount.Neg(transfer.Amount)

3
pkg/core/state/nep5.go
View file

@ -188,6 +188,9 @@ func NEP5TransferFromNotification(ne NotificationEvent, txHash util.Uint256, hei
if !ok {
return nil, errors.New("wrong amount type")
}
if len(bs) > amountSize {
return nil, errors.New("integer overflow")
}
amount = emit.BytesToInt(bs)
}
toAddr := parseUint160(to)