-
Notifications
You must be signed in to change notification settings - Fork 199
/
statusFilters.go
103 lines (84 loc) · 2.79 KB
/
statusFilters.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package filters
import (
"bytes"
"strings"
"github.com/ElrondNetwork/elrond-go/core"
"github.com/ElrondNetwork/elrond-go/data/api"
"github.com/ElrondNetwork/elrond-go/data/block"
"github.com/ElrondNetwork/elrond-go/data/transaction"
)
type statusFilters struct {
selfShardID uint32
}
// NewStatusFilters will create a new instance of a statusFilters
func NewStatusFilters(selfShardID uint32) *statusFilters {
return &statusFilters{
selfShardID: selfShardID,
}
}
// SetStatusIfIsFailedESDTTransfer will set the status if the provided transaction if a failed ESDT transfer
func (sf *statusFilters) SetStatusIfIsFailedESDTTransfer(tx *transaction.ApiTransactionResult) {
if len(tx.SmartContractResults) < 1 {
return
}
isCrossShardTxDestMe := tx.SourceShard != tx.DestinationShard && sf.selfShardID == tx.DestinationShard
if !isCrossShardTxDestMe {
return
}
if !isESDTTransfer(tx) {
return
}
for _, scr := range tx.SmartContractResults {
setStatusBasedOnSCRDataAndNonce(tx, []byte(scr.Data), scr.Nonce)
}
}
// ApplyStatusFilters will apply status filters on the provided miniblocks
func (sf *statusFilters) ApplyStatusFilters(miniblocks []*api.MiniBlock) {
for _, mb := range miniblocks {
if mb.Type != block.TxBlock.String() {
continue
}
isNotCrossShardDestinationMe := mb.SourceShard == mb.DestinationShard || mb.DestinationShard != sf.selfShardID
if isNotCrossShardDestinationMe {
continue
}
iterateMiniblockTxsForESDTTransfer(mb, miniblocks)
}
}
func iterateMiniblockTxsForESDTTransfer(miniblock *api.MiniBlock, miniblocks []*api.MiniBlock) {
for _, tx := range miniblock.Transactions {
if !isESDTTransfer(tx) {
continue
}
searchUnsignedTransaction(tx, miniblocks)
}
}
func searchUnsignedTransaction(tx *transaction.ApiTransactionResult, miniblocks []*api.MiniBlock) {
for _, mb := range miniblocks {
if mb.Type != block.SmartContractResultBlock.String() {
continue
}
shouldCheckTransaction := mb.DestinationShard == tx.SourceShard && mb.SourceShard == tx.DestinationShard
if shouldCheckTransaction {
tryToSetStatusOfESDTTransfer(tx, mb)
}
}
}
func tryToSetStatusOfESDTTransfer(tx *transaction.ApiTransactionResult, miniblock *api.MiniBlock) {
for _, unsignedTx := range miniblock.Transactions {
if unsignedTx.OriginalTransactionHash != tx.Hash {
continue
}
setStatusBasedOnSCRDataAndNonce(tx, unsignedTx.Data, unsignedTx.Nonce)
}
}
func setStatusBasedOnSCRDataAndNonce(tx *transaction.ApiTransactionResult, scrDataField []byte, scrNonce uint64) {
isSCRWithRefund := bytes.HasPrefix(scrDataField, tx.Data) && scrNonce == tx.Nonce
if isSCRWithRefund {
tx.Status = transaction.TxStatusFail
return
}
}
func isESDTTransfer(tx *transaction.ApiTransactionResult) bool {
return strings.HasPrefix(string(tx.Data), core.BuiltInFunctionESDTTransfer)
}