package transfer import ( "context" "math/big" "sync/atomic" "time" "github.com/ethereum/go-ethereum/accounts/abi/bind" "github.com/ethereum/go-ethereum/common" "github.com/ethereum/go-ethereum/core/types" "github.com/ethereum/go-ethereum/event" "github.com/ethereum/go-ethereum/log" "github.com/status-im/status-go/contracts" nodetypes "github.com/status-im/status-go/eth-node/types" "github.com/status-im/status-go/multiaccounts/accounts" "github.com/status-im/status-go/rpc/chain" "github.com/status-im/status-go/services/wallet/async" "github.com/status-im/status-go/services/wallet/balance" "github.com/status-im/status-go/services/wallet/blockchainstate" "github.com/status-im/status-go/services/wallet/token" "github.com/status-im/status-go/services/wallet/walletevent" "github.com/status-im/status-go/transactions" ) var findBlocksRetryInterval = 5 * time.Second const ( transferHistoryTag = "transfer_history" newTransferHistoryTag = "new_transfer_history" transferHistoryLimit = 10000 transferHistoryLimitPerAccount = 5000 transferHistoryLimitPeriod = 24 * time.Hour ) type nonceInfo struct { nonce *int64 blockNumber *big.Int } type findNewBlocksCommand struct { *findBlocksCommand contractMaker *contracts.ContractMaker iteration int blockChainState *blockchainstate.BlockChainState lastNonces map[common.Address]nonceInfo nonceCheckIntervalIterations int logsCheckIntervalIterations int } func (c *findNewBlocksCommand) Command() async.Command { return async.InfiniteCommand{ Interval: 2 * time.Minute, Runable: c.Run, }.Run } var requestTimeout = 20 * time.Second func (c *findNewBlocksCommand) detectTransfers(parent context.Context, accounts []common.Address) (*big.Int, []common.Address, error) { bc, err := c.contractMaker.NewBalanceChecker(c.chainClient.NetworkID()) if err != nil { log.Error("findNewBlocksCommand error creating balance checker", "error", err, "chain", c.chainClient.NetworkID()) return nil, nil, err } tokens, err := c.tokenManager.GetTokens(c.chainClient.NetworkID()) if err != nil { return nil, nil, err } tokenAddresses := []common.Address{} nilAddress := common.Address{} for _, token := range tokens { if token.Address != nilAddress { tokenAddresses = append(tokenAddresses, token.Address) } } log.Debug("findNewBlocksCommand detectTransfers", "cnt", len(tokenAddresses)) ctx, cancel := context.WithTimeout(parent, requestTimeout) defer cancel() blockNum, hashes, err := bc.BalancesHash(&bind.CallOpts{Context: ctx}, c.accounts, tokenAddresses) if err != nil { log.Error("findNewBlocksCommand can't get balances hashes", "error", err) return nil, nil, err } addressesToCheck := []common.Address{} for idx, account := range accounts { blockRange, _, err := c.blockRangeDAO.getBlockRange(c.chainClient.NetworkID(), account) if err != nil { log.Error("findNewBlocksCommand can't get block range", "error", err, "account", account, "chain", c.chainClient.NetworkID()) return nil, nil, err } checkHash := common.BytesToHash(hashes[idx][:]) log.Debug("findNewBlocksCommand comparing hashes", "account", account, "network", c.chainClient.NetworkID(), "old hash", blockRange.balanceCheckHash, "new hash", checkHash.String()) if checkHash.String() != blockRange.balanceCheckHash { addressesToCheck = append(addressesToCheck, account) } blockRange.balanceCheckHash = checkHash.String() err = c.blockRangeDAO.upsertRange(c.chainClient.NetworkID(), account, blockRange) if err != nil { log.Error("findNewBlocksCommand can't update balance check", "error", err, "account", account, "chain", c.chainClient.NetworkID()) return nil, nil, err } } return blockNum, addressesToCheck, nil } func (c *findNewBlocksCommand) detectNonceChange(parent context.Context, to *big.Int, accounts []common.Address) (map[common.Address]*big.Int, error) { addressesWithChange := map[common.Address]*big.Int{} for _, account := range accounts { var oldNonce *int64 blockRange, _, err := c.blockRangeDAO.getBlockRange(c.chainClient.NetworkID(), account) if err != nil { log.Error("findNewBlocksCommand can't get block range", "error", err, "account", account, "chain", c.chainClient.NetworkID()) return nil, err } lastNonceInfo, ok := c.lastNonces[account] if !ok || lastNonceInfo.blockNumber.Cmp(blockRange.eth.LastKnown) != 0 { log.Debug("Fetching old nonce", "at", blockRange.eth.LastKnown, "acc", account) if blockRange.eth.LastKnown == nil { blockRange.eth.LastKnown = big.NewInt(0) oldNonce = new(int64) // At 0 block nonce is 0 } else { oldNonce, err = c.balanceCacher.NonceAt(parent, c.chainClient, account, blockRange.eth.LastKnown) if err != nil { log.Error("findNewBlocksCommand can't get nonce", "error", err, "account", account, "chain", c.chainClient.NetworkID()) return nil, err } } } else { oldNonce = lastNonceInfo.nonce } newNonce, err := c.balanceCacher.NonceAt(parent, c.chainClient, account, to) if err != nil { log.Error("findNewBlocksCommand can't get nonce", "error", err, "account", account, "chain", c.chainClient.NetworkID()) return nil, err } log.Debug("Comparing nonces", "oldNonce", *oldNonce, "newNonce", *newNonce, "to", to, "acc", account) if *newNonce != *oldNonce { addressesWithChange[account] = blockRange.eth.LastKnown } if c.lastNonces == nil { c.lastNonces = map[common.Address]nonceInfo{} } c.lastNonces[account] = nonceInfo{ nonce: newNonce, blockNumber: to, } } return addressesWithChange, nil } var nonceCheckIntervalIterations = 30 var logsCheckIntervalIterations = 5 func (c *findNewBlocksCommand) Run(parent context.Context) error { mnemonicWasNotShown, err := c.accountsDB.GetMnemonicWasNotShown() if err != nil { return err } accountsToCheck := []common.Address{} // accounts which might have outgoing transfers initiated outside // the application, e.g. watch only or restored from mnemonic phrase accountsWithOutsideTransfers := []common.Address{} for _, account := range c.accounts { acc, err := c.accountsDB.GetAccountByAddress(nodetypes.Address(account)) if err != nil { return err } if mnemonicWasNotShown { if acc.AddressWasNotShown { log.Info("skip findNewBlocksCommand, mnemonic has not been shown and the address has not been shared yet", "address", account) continue } } if !mnemonicWasNotShown || acc.Type != accounts.AccountTypeGenerated { accountsWithOutsideTransfers = append(accountsWithOutsideTransfers, account) } accountsToCheck = append(accountsToCheck, account) } if len(accountsToCheck) == 0 { return nil } headNum, accountsWithDetectedChanges, err := c.detectTransfers(parent, accountsToCheck) if err != nil { log.Error("findNewBlocksCommand error on transfer detection", "error", err, "chain", c.chainClient.NetworkID()) return err } c.blockChainState.SetLastBlockNumber(c.chainClient.NetworkID(), headNum.Uint64()) if len(accountsWithDetectedChanges) != 0 { log.Debug("findNewBlocksCommand detected accounts with changes, proceeding", "accounts", accountsWithDetectedChanges, "from", c.fromBlockNumber) err = c.findAndSaveEthBlocks(parent, c.fromBlockNumber, headNum, accountsToCheck) if err != nil { return err } } else if c.iteration%c.nonceCheckIntervalIterations == 0 && len(accountsWithOutsideTransfers) > 0 { log.Debug("findNewBlocksCommand nonce check", "accounts", accountsWithOutsideTransfers) accountsWithNonceChanges, err := c.detectNonceChange(parent, headNum, accountsWithOutsideTransfers) if err != nil { return err } if len(accountsWithNonceChanges) > 0 { log.Debug("findNewBlocksCommand detected nonce diff", "accounts", accountsWithNonceChanges) for account, from := range accountsWithNonceChanges { err = c.findAndSaveEthBlocks(parent, from, headNum, []common.Address{account}) if err != nil { return err } } } for _, account := range accountsToCheck { if _, ok := accountsWithNonceChanges[account]; ok { continue } err := c.markEthBlockRangeChecked(account, &BlockRange{nil, c.fromBlockNumber, headNum}) if err != nil { return err } } } if len(accountsWithDetectedChanges) != 0 || c.iteration%c.logsCheckIntervalIterations == 0 { from := c.fromBlockNumber if c.logsCheckLastKnownBlock != nil { from = c.logsCheckLastKnownBlock } err = c.findAndSaveTokenBlocks(parent, from, headNum) if err != nil { return err } c.logsCheckLastKnownBlock = headNum } c.fromBlockNumber = headNum c.iteration++ return nil } func (c *findNewBlocksCommand) findAndSaveEthBlocks(parent context.Context, fromNum, headNum *big.Int, accounts []common.Address) error { // Check ETH transfers for each account independently mnemonicWasNotShown, err := c.accountsDB.GetMnemonicWasNotShown() if err != nil { return err } for _, account := range accounts { if mnemonicWasNotShown { acc, err := c.accountsDB.GetAccountByAddress(nodetypes.Address(account)) if err != nil { return err } if acc.AddressWasNotShown { log.Info("skip findNewBlocksCommand, mnemonic has not been shown and the address has not been shared yet", "address", account) continue } } log.Debug("start findNewBlocksCommand", "account", account, "chain", c.chainClient.NetworkID(), "noLimit", c.noLimit, "from", fromNum, "to", headNum) headers, startBlockNum, err := c.findBlocksWithEthTransfers(parent, account, fromNum, headNum) if err != nil { return err } if len(headers) > 0 { log.Debug("findNewBlocksCommand saving headers", "len", len(headers), "lastBlockNumber", headNum, "balance", c.balanceCacher.Cache().GetBalance(account, c.chainClient.NetworkID(), headNum), "nonce", c.balanceCacher.Cache().GetNonce(account, c.chainClient.NetworkID(), headNum)) err := c.db.SaveBlocks(c.chainClient.NetworkID(), headers) if err != nil { return err } c.blocksFound(headers) } err = c.markEthBlockRangeChecked(account, &BlockRange{startBlockNum, fromNum, headNum}) if err != nil { return err } log.Debug("end findNewBlocksCommand", "account", account, "chain", c.chainClient.NetworkID(), "noLimit", c.noLimit, "from", fromNum, "to", headNum) } return nil } func (c *findNewBlocksCommand) findAndSaveTokenBlocks(parent context.Context, fromNum, headNum *big.Int) error { // Check token transfers for all accounts. // Each account's last checked block can be different, so we can get duplicated headers, // so we need to deduplicate them const incomingOnly = false erc20Headers, err := c.fastIndexErc20(parent, fromNum, headNum, incomingOnly) if err != nil { log.Error("findNewBlocksCommand fastIndexErc20", "err", err, "account", c.accounts, "chain", c.chainClient.NetworkID()) return err } if len(erc20Headers) > 0 { log.Debug("findNewBlocksCommand saving headers", "len", len(erc20Headers), "from", fromNum, "to", headNum) // get not loaded headers from DB for all accs and blocks preLoadedTransactions, err := c.db.GetTransactionsToLoad(c.chainClient.NetworkID(), common.Address{}, nil) if err != nil { return err } tokenBlocksFiltered := filterNewPreloadedTransactions(erc20Headers, preLoadedTransactions) err = c.db.SaveBlocks(c.chainClient.NetworkID(), tokenBlocksFiltered) if err != nil { return err } c.blocksFound(tokenBlocksFiltered) } return c.markTokenBlockRangeChecked(c.accounts, fromNum, headNum) } func (c *findBlocksCommand) markTokenBlockRangeChecked(accounts []common.Address, from, to *big.Int) error { log.Debug("markTokenBlockRangeChecked", "chain", c.chainClient.NetworkID(), "from", from.Uint64(), "to", to.Uint64()) for _, account := range accounts { err := c.blockRangeDAO.updateTokenRange(c.chainClient.NetworkID(), account, &BlockRange{FirstKnown: from, LastKnown: to}) if err != nil { log.Error("findNewBlocksCommand upsertTokenRange", "error", err) return err } } return nil } func filterNewPreloadedTransactions(erc20Headers []*DBHeader, preLoadedTransfers []*PreloadedTransaction) []*DBHeader { var uniqueErc20Headers []*DBHeader for _, header := range erc20Headers { loaded := false for _, transfer := range preLoadedTransfers { if header.PreloadedTransactions[0].ID == transfer.ID { loaded = true break } } if !loaded { uniqueErc20Headers = append(uniqueErc20Headers, header) } } return uniqueErc20Headers } func (c *findNewBlocksCommand) findBlocksWithEthTransfers(parent context.Context, account common.Address, fromOrig, toOrig *big.Int) (headers []*DBHeader, startBlockNum *big.Int, err error) { log.Debug("start findNewBlocksCommand::findBlocksWithEthTransfers", "account", account, "chain", c.chainClient.NetworkID(), "noLimit", c.noLimit, "from", c.fromBlockNumber, "to", c.toBlockNumber) rangeSize := big.NewInt(int64(c.defaultNodeBlockChunkSize)) from, to := new(big.Int).Set(fromOrig), new(big.Int).Set(toOrig) // Limit the range size to DefaultNodeBlockChunkSize if new(big.Int).Sub(to, from).Cmp(rangeSize) > 0 { from.Sub(to, rangeSize) } for { if from.Cmp(to) == 0 { log.Debug("findNewBlocksCommand empty range", "from", from, "to", to) break } fromBlock := &Block{Number: from} var newFromBlock *Block var ethHeaders []*DBHeader newFromBlock, ethHeaders, startBlockNum, err = c.fastIndex(parent, account, c.balanceCacher, fromBlock, to) if err != nil { log.Error("findNewBlocksCommand checkRange fastIndex", "err", err, "account", account, "chain", c.chainClient.NetworkID()) return nil, nil, err } log.Debug("findNewBlocksCommand checkRange", "chainID", c.chainClient.NetworkID(), "account", account, "startBlock", startBlockNum, "newFromBlock", newFromBlock.Number, "toBlockNumber", to, "noLimit", c.noLimit) headers = append(headers, ethHeaders...) if startBlockNum != nil && startBlockNum.Cmp(from) >= 0 { log.Debug("Checked all ranges, stop execution", "startBlock", startBlockNum, "from", from, "to", to) break } nextFrom, nextTo := nextRange(c.defaultNodeBlockChunkSize, newFromBlock.Number, fromOrig) if nextFrom.Cmp(from) == 0 && nextTo.Cmp(to) == 0 { log.Debug("findNewBlocksCommand empty next range", "from", from, "to", to) break } from = nextFrom to = nextTo } log.Debug("end findNewBlocksCommand::findBlocksWithEthTransfers", "account", account, "chain", c.chainClient.NetworkID(), "noLimit", c.noLimit) return headers, startBlockNum, nil } // TODO NewFindBlocksCommand type findBlocksCommand struct { accounts []common.Address db *Database accountsDB *accounts.Database blockRangeDAO BlockRangeDAOer chainClient chain.ClientInterface balanceCacher balance.Cacher feed *event.Feed noLimit bool tokenManager *token.Manager fromBlockNumber *big.Int logsCheckLastKnownBlock *big.Int toBlockNumber *big.Int blocksLoadedCh chan<- []*DBHeader defaultNodeBlockChunkSize int // Not to be set by the caller resFromBlock *Block startBlockNumber *big.Int reachedETHHistoryStart bool } func (c *findBlocksCommand) Runner(interval ...time.Duration) async.Runner { intvl := findBlocksRetryInterval if len(interval) > 0 { intvl = interval[0] } return async.FiniteCommandWithErrorCounter{ FiniteCommand: async.FiniteCommand{ Interval: intvl, Runable: c.Run, }, ErrorCounter: async.NewErrorCounter(3, "findBlocksCommand"), } } func (c *findBlocksCommand) Command(interval ...time.Duration) async.Command { return c.Runner(interval...).Run } type ERC20BlockRange struct { from *big.Int to *big.Int } func (c *findBlocksCommand) ERC20ScanByBalance(parent context.Context, account common.Address, fromBlock, toBlock *big.Int, token common.Address) ([]ERC20BlockRange, error) { var err error batchSize := getErc20BatchSize(c.chainClient.NetworkID()) ranges := [][]*big.Int{{fromBlock, toBlock}} foundRanges := []ERC20BlockRange{} cache := map[int64]*big.Int{} for { nextRanges := [][]*big.Int{} for _, blockRange := range ranges { from, to := blockRange[0], blockRange[1] fromBalance, ok := cache[from.Int64()] if !ok { fromBalance, err = c.tokenManager.GetTokenBalanceAt(parent, c.chainClient, account, token, from) if err != nil { return nil, err } if fromBalance == nil { fromBalance = big.NewInt(0) } cache[from.Int64()] = fromBalance } toBalance, ok := cache[to.Int64()] if !ok { toBalance, err = c.tokenManager.GetTokenBalanceAt(parent, c.chainClient, account, token, to) if err != nil { return nil, err } if toBalance == nil { toBalance = big.NewInt(0) } cache[to.Int64()] = toBalance } if fromBalance.Cmp(toBalance) != 0 { diff := new(big.Int).Sub(to, from) if diff.Cmp(batchSize) <= 0 { foundRanges = append(foundRanges, ERC20BlockRange{from, to}) continue } halfOfDiff := new(big.Int).Div(diff, big.NewInt(2)) mid := new(big.Int).Add(from, halfOfDiff) nextRanges = append(nextRanges, []*big.Int{from, mid}) nextRanges = append(nextRanges, []*big.Int{mid, to}) } } if len(nextRanges) == 0 { break } ranges = nextRanges } return foundRanges, nil } func (c *findBlocksCommand) checkERC20Tail(parent context.Context, account common.Address) ([]*DBHeader, error) { log.Debug("checkERC20Tail", "account", account, "to block", c.startBlockNumber, "from", c.resFromBlock.Number) tokens, err := c.tokenManager.GetTokens(c.chainClient.NetworkID()) if err != nil { return nil, err } addresses := make([]common.Address, len(tokens)) for i, token := range tokens { addresses[i] = token.Address } from := new(big.Int).Sub(c.resFromBlock.Number, big.NewInt(1)) clients := make(map[uint64]chain.ClientInterface, 1) clients[c.chainClient.NetworkID()] = c.chainClient atBlocks := make(map[uint64]*big.Int, 1) atBlocks[c.chainClient.NetworkID()] = from balances, err := c.tokenManager.GetBalancesAtByChain(parent, clients, []common.Address{account}, addresses, atBlocks) if err != nil { return nil, err } foundRanges := []ERC20BlockRange{} for token, balance := range balances[c.chainClient.NetworkID()][account] { bigintBalance := big.NewInt(balance.ToInt().Int64()) if bigintBalance.Cmp(big.NewInt(0)) <= 0 { continue } result, err := c.ERC20ScanByBalance(parent, account, big.NewInt(0), from, token) if err != nil { return nil, err } foundRanges = append(foundRanges, result...) } uniqRanges := []ERC20BlockRange{} rangesMap := map[string]bool{} for _, rangeItem := range foundRanges { key := rangeItem.from.String() + "-" + rangeItem.to.String() if _, ok := rangesMap[key]; !ok { rangesMap[key] = true uniqRanges = append(uniqRanges, rangeItem) } } foundHeaders := []*DBHeader{} for _, rangeItem := range uniqRanges { headers, err := c.fastIndexErc20(parent, rangeItem.from, rangeItem.to, true) if err != nil { return nil, err } foundHeaders = append(foundHeaders, headers...) } return foundHeaders, nil } func (c *findBlocksCommand) Run(parent context.Context) (err error) { log.Debug("start findBlocksCommand", "accounts", c.accounts, "chain", c.chainClient.NetworkID(), "noLimit", c.noLimit, "from", c.fromBlockNumber, "to", c.toBlockNumber) account := c.accounts[0] // For now this command supports only 1 account mnemonicWasNotShown, err := c.accountsDB.GetMnemonicWasNotShown() if err != nil { return err } if mnemonicWasNotShown { account, err := c.accountsDB.GetAccountByAddress(nodetypes.BytesToAddress(account.Bytes())) if err != nil { return err } if account.AddressWasNotShown { log.Info("skip findBlocksCommand, mnemonic has not been shown and the address has not been shared yet", "address", account) return nil } } rangeSize := big.NewInt(int64(c.defaultNodeBlockChunkSize)) from, to := new(big.Int).Set(c.fromBlockNumber), new(big.Int).Set(c.toBlockNumber) // Limit the range size to DefaultNodeBlockChunkSize if new(big.Int).Sub(to, from).Cmp(rangeSize) > 0 { from.Sub(to, rangeSize) } for { if from.Cmp(to) == 0 { log.Debug("findBlocksCommand empty range", "from", from, "to", to) break } var headers []*DBHeader if c.reachedETHHistoryStart { if c.fromBlockNumber.Cmp(zero) == 0 && c.startBlockNumber != nil && c.startBlockNumber.Cmp(zero) == 1 { headers, err = c.checkERC20Tail(parent, account) if err != nil { log.Error("findBlocksCommand checkERC20Tail", "err", err, "account", account, "chain", c.chainClient.NetworkID()) break } } } else { headers, err = c.checkRange(parent, from, to) if err != nil { break } } if len(headers) > 0 { log.Debug("findBlocksCommand saving headers", "len", len(headers), "lastBlockNumber", to, "balance", c.balanceCacher.Cache().GetBalance(account, c.chainClient.NetworkID(), to), "nonce", c.balanceCacher.Cache().GetNonce(account, c.chainClient.NetworkID(), to)) err = c.db.SaveBlocks(c.chainClient.NetworkID(), headers) if err != nil { break } c.blocksFound(headers) } if c.reachedETHHistoryStart { err = c.markTokenBlockRangeChecked([]common.Address{account}, big.NewInt(0), to) if err != nil { break } log.Debug("findBlocksCommand reached first ETH transfer and checked erc20 tail", "chain", c.chainClient.NetworkID(), "account", account) break } err = c.markEthBlockRangeChecked(account, &BlockRange{c.startBlockNumber, c.resFromBlock.Number, to}) if err != nil { break } err = c.markTokenBlockRangeChecked([]common.Address{account}, c.resFromBlock.Number, to) if err != nil { break } // if we have found first ETH block and we have not reached the start of ETH history yet if c.startBlockNumber != nil && c.fromBlockNumber.Cmp(from) == -1 { log.Debug("ERC20 tail should be checked", "initial from", c.fromBlockNumber, "actual from", from, "first ETH block", c.startBlockNumber) c.reachedETHHistoryStart = true continue } if c.startBlockNumber != nil && c.startBlockNumber.Cmp(from) >= 0 { log.Debug("Checked all ranges, stop execution", "startBlock", c.startBlockNumber, "from", from, "to", to) break } nextFrom, nextTo := nextRange(c.defaultNodeBlockChunkSize, c.resFromBlock.Number, c.fromBlockNumber) if nextFrom.Cmp(from) == 0 && nextTo.Cmp(to) == 0 { log.Debug("findBlocksCommand empty next range", "from", from, "to", to) break } from = nextFrom to = nextTo } log.Debug("end findBlocksCommand", "account", account, "chain", c.chainClient.NetworkID(), "noLimit", c.noLimit, "err", err) return err } func (c *findBlocksCommand) blocksFound(headers []*DBHeader) { c.blocksLoadedCh <- headers } func (c *findBlocksCommand) markEthBlockRangeChecked(account common.Address, blockRange *BlockRange) error { log.Debug("upsert block range", "Start", blockRange.Start, "FirstKnown", blockRange.FirstKnown, "LastKnown", blockRange.LastKnown, "chain", c.chainClient.NetworkID(), "account", account) err := c.blockRangeDAO.upsertEthRange(c.chainClient.NetworkID(), account, blockRange) if err != nil { log.Error("findBlocksCommand upsertRange", "error", err) return err } return nil } func (c *findBlocksCommand) checkRange(parent context.Context, from *big.Int, to *big.Int) ( foundHeaders []*DBHeader, err error) { account := c.accounts[0] fromBlock := &Block{Number: from} newFromBlock, ethHeaders, startBlock, err := c.fastIndex(parent, account, c.balanceCacher, fromBlock, to) if err != nil { log.Error("findBlocksCommand checkRange fastIndex", "err", err, "account", account, "chain", c.chainClient.NetworkID()) return nil, err } log.Debug("findBlocksCommand checkRange", "chainID", c.chainClient.NetworkID(), "account", account, "startBlock", startBlock, "newFromBlock", newFromBlock.Number, "toBlockNumber", to, "noLimit", c.noLimit) // There could be incoming ERC20 transfers which don't change the balance // and nonce of ETH account, so we keep looking for them erc20Headers, err := c.fastIndexErc20(parent, newFromBlock.Number, to, false) if err != nil { log.Error("findBlocksCommand checkRange fastIndexErc20", "err", err, "account", account, "chain", c.chainClient.NetworkID()) return nil, err } allHeaders := append(ethHeaders, erc20Headers...) if len(allHeaders) > 0 { foundHeaders = uniqueHeaderPerBlockHash(allHeaders) } c.resFromBlock = newFromBlock c.startBlockNumber = startBlock log.Debug("end findBlocksCommand checkRange", "chainID", c.chainClient.NetworkID(), "account", account, "c.startBlock", c.startBlockNumber, "newFromBlock", newFromBlock.Number, "toBlockNumber", to, "c.resFromBlock", c.resFromBlock.Number) return } func loadBlockRangeInfo(chainID uint64, account common.Address, blockDAO BlockRangeDAOer) ( *ethTokensBlockRanges, error) { blockRange, _, err := blockDAO.getBlockRange(chainID, account) if err != nil { log.Error("failed to load block ranges from database", "chain", chainID, "account", account, "error", err) return nil, err } return blockRange, nil } // Returns if all blocks are loaded, which means that start block (beginning of account history) // has been found and all block headers saved to the DB func areAllHistoryBlocksLoaded(blockInfo *BlockRange) bool { if blockInfo != nil && blockInfo.FirstKnown != nil && ((blockInfo.Start != nil && blockInfo.Start.Cmp(blockInfo.FirstKnown) >= 0) || blockInfo.FirstKnown.Cmp(zero) == 0) { return true } return false } func areAllHistoryBlocksLoadedForAddress(blockRangeDAO BlockRangeDAOer, chainID uint64, address common.Address) (bool, error) { blockRange, _, err := blockRangeDAO.getBlockRange(chainID, address) if err != nil { log.Error("findBlocksCommand getBlockRange", "error", err) return false, err } return areAllHistoryBlocksLoaded(blockRange.eth) && areAllHistoryBlocksLoaded(blockRange.tokens), nil } // run fast indexing for every accont up to canonical chain head minus safety depth. // every account will run it from last synced header. func (c *findBlocksCommand) fastIndex(ctx context.Context, account common.Address, bCacher balance.Cacher, fromBlock *Block, toBlockNumber *big.Int) (resultingFrom *Block, headers []*DBHeader, startBlock *big.Int, err error) { log.Debug("fast index started", "chainID", c.chainClient.NetworkID(), "account", account, "from", fromBlock.Number, "to", toBlockNumber) start := time.Now() group := async.NewGroup(ctx) command := ðHistoricalCommand{ chainClient: c.chainClient, balanceCacher: bCacher, address: account, feed: c.feed, from: fromBlock, to: toBlockNumber, noLimit: c.noLimit, threadLimit: SequentialThreadLimit, } group.Add(command.Command()) select { case <-ctx.Done(): err = ctx.Err() log.Debug("fast indexer ctx Done", "error", err) return case <-group.WaitAsync(): if command.error != nil { err = command.error return } resultingFrom = &Block{Number: command.resultingFrom} headers = command.foundHeaders startBlock = command.startBlock log.Debug("fast indexer finished", "chainID", c.chainClient.NetworkID(), "account", account, "in", time.Since(start), "startBlock", command.startBlock, "resultingFrom", resultingFrom.Number, "headers", len(headers)) return } } // run fast indexing for every accont up to canonical chain head minus safety depth. // every account will run it from last synced header. func (c *findBlocksCommand) fastIndexErc20(ctx context.Context, fromBlockNumber *big.Int, toBlockNumber *big.Int, incomingOnly bool) ([]*DBHeader, error) { start := time.Now() group := async.NewGroup(ctx) erc20 := &erc20HistoricalCommand{ erc20: NewERC20TransfersDownloader(c.chainClient, c.accounts, types.LatestSignerForChainID(c.chainClient.ToBigInt()), incomingOnly), chainClient: c.chainClient, feed: c.feed, from: fromBlockNumber, to: toBlockNumber, foundHeaders: []*DBHeader{}, } group.Add(erc20.Command()) select { case <-ctx.Done(): return nil, ctx.Err() case <-group.WaitAsync(): headers := erc20.foundHeaders log.Debug("fast indexer Erc20 finished", "chainID", c.chainClient.NetworkID(), "in", time.Since(start), "headers", len(headers)) return headers, nil } } // Start transfers loop to load transfers for new blocks func (c *loadBlocksAndTransfersCommand) startTransfersLoop(ctx context.Context) { c.incLoops() go func() { defer func() { c.decLoops() }() log.Debug("loadTransfersLoop start", "chain", c.chainClient.NetworkID()) for { select { case <-ctx.Done(): log.Debug("startTransfersLoop done", "chain", c.chainClient.NetworkID(), "error", ctx.Err()) return case dbHeaders := <-c.blocksLoadedCh: log.Debug("loadTransfersOnDemand transfers received", "chain", c.chainClient.NetworkID(), "headers", len(dbHeaders)) blocksByAddress := map[common.Address][]*big.Int{} // iterate over headers and group them by address for _, dbHeader := range dbHeaders { blocksByAddress[dbHeader.Address] = append(blocksByAddress[dbHeader.Address], dbHeader.Number) } go func() { _ = loadTransfers(ctx, c.blockDAO, c.db, c.chainClient, noBlockLimit, blocksByAddress, c.transactionManager, c.pendingTxManager, c.tokenManager, c.feed) }() } } }() } func newLoadBlocksAndTransfersCommand(accounts []common.Address, db *Database, accountsDB *accounts.Database, blockDAO *BlockDAO, blockRangesSeqDAO BlockRangeDAOer, chainClient chain.ClientInterface, feed *event.Feed, transactionManager *TransactionManager, pendingTxManager *transactions.PendingTxTracker, tokenManager *token.Manager, balanceCacher balance.Cacher, omitHistory bool, blockChainState *blockchainstate.BlockChainState) *loadBlocksAndTransfersCommand { return &loadBlocksAndTransfersCommand{ accounts: accounts, db: db, blockRangeDAO: blockRangesSeqDAO, accountsDB: accountsDB, blockDAO: blockDAO, chainClient: chainClient, feed: feed, balanceCacher: balanceCacher, transactionManager: transactionManager, pendingTxManager: pendingTxManager, tokenManager: tokenManager, blocksLoadedCh: make(chan []*DBHeader, 100), omitHistory: omitHistory, contractMaker: tokenManager.ContractMaker, blockChainState: blockChainState, } } type loadBlocksAndTransfersCommand struct { accounts []common.Address db *Database accountsDB *accounts.Database blockRangeDAO BlockRangeDAOer blockDAO *BlockDAO chainClient chain.ClientInterface feed *event.Feed balanceCacher balance.Cacher // nonArchivalRPCNode bool // TODO Make use of it transactionManager *TransactionManager pendingTxManager *transactions.PendingTxTracker tokenManager *token.Manager blocksLoadedCh chan []*DBHeader omitHistory bool contractMaker *contracts.ContractMaker blockChainState *blockchainstate.BlockChainState // Not to be set by the caller transfersLoaded map[common.Address]bool // For event RecentHistoryReady to be sent only once per account during app lifetime loops atomic.Int32 } func (c *loadBlocksAndTransfersCommand) incLoops() { c.loops.Add(1) } func (c *loadBlocksAndTransfersCommand) decLoops() { c.loops.Add(-1) } func (c *loadBlocksAndTransfersCommand) isStarted() bool { return c.loops.Load() > 0 } func (c *loadBlocksAndTransfersCommand) Run(parent context.Context) (err error) { log.Debug("start load all transfers command", "chain", c.chainClient.NetworkID(), "accounts", c.accounts) // Finite processes (to be restarted on error, but stopped on success or context cancel): // fetching transfers for loaded blocks // fetching history blocks // Infinite processes (to be restarted on error), but stopped on context cancel: // fetching new blocks // fetching transfers for new blocks ctx := parent finiteGroup := async.NewAtomicGroup(ctx) finiteGroup.SetName("finiteGroup") defer func() { finiteGroup.Stop() finiteGroup.Wait() }() blockRanges, err := c.blockRangeDAO.getBlockRanges(c.chainClient.NetworkID(), c.accounts) if err != nil { return err } firstScan := false var headNum *big.Int for _, address := range c.accounts { blockRange, ok := blockRanges[address] if !ok || blockRange.tokens.LastKnown == nil { firstScan = true break } if headNum == nil || blockRange.tokens.LastKnown.Cmp(headNum) < 0 { headNum = blockRange.tokens.LastKnown } } fromNum := big.NewInt(0) if firstScan { headNum, err = getHeadBlockNumber(ctx, c.chainClient) if err != nil { return err } } // It will start loadTransfersCommand which will run until all transfers from DB are loaded or any one failed to load err = c.startFetchingTransfersForLoadedBlocks(finiteGroup) if err != nil { log.Error("loadBlocksAndTransfersCommand fetchTransfersForLoadedBlocks", "error", err) return err } if !c.isStarted() { c.startTransfersLoop(ctx) c.startFetchingNewBlocks(ctx, c.accounts, headNum, c.blocksLoadedCh) } // It will start findBlocksCommands which will run until success when all blocks are loaded err = c.fetchHistoryBlocks(finiteGroup, c.accounts, fromNum, headNum, c.blocksLoadedCh) if err != nil { log.Error("loadBlocksAndTransfersCommand fetchHistoryBlocks", "error", err) return err } select { case <-ctx.Done(): log.Debug("loadBlocksAndTransfers command cancelled", "chain", c.chainClient.NetworkID(), "accounts", c.accounts, "error", ctx.Err()) case <-finiteGroup.WaitAsync(): err = finiteGroup.Error() // if there was an error, rerun the command log.Debug("end loadBlocksAndTransfers command", "chain", c.chainClient.NetworkID(), "accounts", c.accounts, "error", err, "group", finiteGroup.Name()) } return err } func (c *loadBlocksAndTransfersCommand) Runner(interval ...time.Duration) async.Runner { // 30s - default interval for Infura's delay returned in error. That should increase chances // for request to succeed with the next attempt for now until we have a proper retry mechanism intvl := 30 * time.Second if len(interval) > 0 { intvl = interval[0] } return async.FiniteCommand{ Interval: intvl, Runable: c.Run, } } func (c *loadBlocksAndTransfersCommand) Command(interval ...time.Duration) async.Command { return c.Runner(interval...).Run } func (c *loadBlocksAndTransfersCommand) fetchHistoryBlocks(group *async.AtomicGroup, accounts []common.Address, fromNum, toNum *big.Int, blocksLoadedCh chan []*DBHeader) (err error) { for _, account := range accounts { err = c.fetchHistoryBlocksForAccount(group, account, fromNum, toNum, c.blocksLoadedCh) if err != nil { return err } } return nil } func (c *loadBlocksAndTransfersCommand) fetchHistoryBlocksForAccount(group *async.AtomicGroup, account common.Address, fromNum, toNum *big.Int, blocksLoadedCh chan []*DBHeader) error { log.Debug("fetchHistoryBlocks start", "chainID", c.chainClient.NetworkID(), "account", account, "omit", c.omitHistory) if c.omitHistory { blockRange := ðTokensBlockRanges{eth: &BlockRange{nil, big.NewInt(0), toNum}, tokens: &BlockRange{nil, big.NewInt(0), toNum}} err := c.blockRangeDAO.upsertRange(c.chainClient.NetworkID(), account, blockRange) log.Error("fetchHistoryBlocks upsertRange", "error", err) return err } blockRange, err := loadBlockRangeInfo(c.chainClient.NetworkID(), account, c.blockRangeDAO) if err != nil { log.Error("fetchHistoryBlocks loadBlockRangeInfo", "error", err) return err } ranges := [][]*big.Int{} // There are 2 history intervals: // 1) from 0 to FirstKnown // 2) from LastKnown to `toNum`` (head) // If we blockRange is nil, we need to load all blocks from `fromNum` to `toNum` // As current implementation checks ETH first then tokens, tokens ranges maybe behind ETH ranges in // cases when block searching was interrupted, so we use tokens ranges if blockRange.tokens.LastKnown != nil || blockRange.tokens.FirstKnown != nil { if blockRange.tokens.LastKnown != nil && toNum.Cmp(blockRange.tokens.LastKnown) > 0 { ranges = append(ranges, []*big.Int{blockRange.tokens.LastKnown, toNum}) } if blockRange.tokens.FirstKnown != nil { if fromNum.Cmp(blockRange.tokens.FirstKnown) < 0 { ranges = append(ranges, []*big.Int{fromNum, blockRange.tokens.FirstKnown}) } else { if !c.transfersLoaded[account] { transfersLoaded, err := c.areAllTransfersLoaded(account) if err != nil { return err } if transfersLoaded { if c.transfersLoaded == nil { c.transfersLoaded = make(map[common.Address]bool) } c.transfersLoaded[account] = true c.notifyHistoryReady(account) } } } } } else { ranges = append(ranges, []*big.Int{fromNum, toNum}) } if len(ranges) > 0 { storage := chain.NewLimitsDBStorage(c.db.client) limiter := chain.NewRequestLimiter(storage) chainClient, _ := createChainClientWithLimiter(c.chainClient, account, limiter) if chainClient == nil { chainClient = c.chainClient } for _, rangeItem := range ranges { log.Debug("range item", "r", rangeItem, "n", c.chainClient.NetworkID(), "a", account) fbc := &findBlocksCommand{ accounts: []common.Address{account}, db: c.db, accountsDB: c.accountsDB, blockRangeDAO: c.blockRangeDAO, chainClient: chainClient, balanceCacher: c.balanceCacher, feed: c.feed, noLimit: false, fromBlockNumber: rangeItem[0], toBlockNumber: rangeItem[1], tokenManager: c.tokenManager, blocksLoadedCh: blocksLoadedCh, defaultNodeBlockChunkSize: DefaultNodeBlockChunkSize, } group.Add(fbc.Command()) } } return nil } func (c *loadBlocksAndTransfersCommand) startFetchingNewBlocks(ctx context.Context, addresses []common.Address, fromNum *big.Int, blocksLoadedCh chan<- []*DBHeader) { log.Debug("startFetchingNewBlocks start", "chainID", c.chainClient.NetworkID(), "accounts", addresses) c.incLoops() go func() { defer func() { c.decLoops() }() newBlocksCmd := &findNewBlocksCommand{ findBlocksCommand: &findBlocksCommand{ accounts: addresses, db: c.db, accountsDB: c.accountsDB, blockRangeDAO: c.blockRangeDAO, chainClient: c.chainClient, balanceCacher: c.balanceCacher, feed: c.feed, noLimit: false, fromBlockNumber: fromNum, tokenManager: c.tokenManager, blocksLoadedCh: blocksLoadedCh, defaultNodeBlockChunkSize: DefaultNodeBlockChunkSize, }, contractMaker: c.contractMaker, blockChainState: c.blockChainState, nonceCheckIntervalIterations: nonceCheckIntervalIterations, logsCheckIntervalIterations: logsCheckIntervalIterations, } group := async.NewGroup(ctx) group.Add(newBlocksCmd.Command()) // No need to wait for the group since it is infinite <-ctx.Done() log.Debug("startFetchingNewBlocks end", "chainID", c.chainClient.NetworkID(), "accounts", addresses, "error", ctx.Err()) }() } func (c *loadBlocksAndTransfersCommand) getBlocksToLoad() (map[common.Address][]*big.Int, error) { blocksMap := make(map[common.Address][]*big.Int) for _, account := range c.accounts { blocks, err := c.blockDAO.GetBlocksToLoadByAddress(c.chainClient.NetworkID(), account, numberOfBlocksCheckedPerIteration) if err != nil { log.Error("loadBlocksAndTransfersCommand GetBlocksToLoadByAddress", "error", err) return nil, err } if len(blocks) == 0 { log.Debug("fetchTransfers no blocks to load", "chainID", c.chainClient.NetworkID(), "account", account) continue } blocksMap[account] = blocks } if len(blocksMap) == 0 { log.Debug("fetchTransfers no blocks to load", "chainID", c.chainClient.NetworkID()) } return blocksMap, nil } func (c *loadBlocksAndTransfersCommand) startFetchingTransfersForLoadedBlocks(group *async.AtomicGroup) error { log.Debug("fetchTransfers start", "chainID", c.chainClient.NetworkID(), "accounts", c.accounts) blocksMap, err := c.getBlocksToLoad() if err != nil { return err } go func() { txCommand := &loadTransfersCommand{ accounts: c.accounts, db: c.db, blockDAO: c.blockDAO, chainClient: c.chainClient, transactionManager: c.transactionManager, pendingTxManager: c.pendingTxManager, tokenManager: c.tokenManager, blocksByAddress: blocksMap, feed: c.feed, } group.Add(txCommand.Command()) log.Debug("fetchTransfers end", "chainID", c.chainClient.NetworkID(), "accounts", c.accounts) }() return nil } func (c *loadBlocksAndTransfersCommand) notifyHistoryReady(account common.Address) { if c.feed != nil { c.feed.Send(walletevent.Event{ Type: EventRecentHistoryReady, Accounts: []common.Address{account}, ChainID: c.chainClient.NetworkID(), }) } } func (c *loadBlocksAndTransfersCommand) areAllTransfersLoaded(account common.Address) (bool, error) { allBlocksLoaded, err := areAllHistoryBlocksLoadedForAddress(c.blockRangeDAO, c.chainClient.NetworkID(), account) if err != nil { log.Error("loadBlockAndTransfersCommand allHistoryBlocksLoaded", "error", err) return false, err } if allBlocksLoaded { headers, err := c.blockDAO.GetBlocksToLoadByAddress(c.chainClient.NetworkID(), account, 1) if err != nil { log.Error("loadBlocksAndTransfersCommand GetFirstSavedBlock", "error", err) return false, err } if len(headers) == 0 { return true, nil } } return false, nil } // TODO - make it a common method for every service that wants head block number, that will cache the latest block // and updates it on timeout func getHeadBlockNumber(parent context.Context, chainClient chain.ClientInterface) (*big.Int, error) { ctx, cancel := context.WithTimeout(parent, 3*time.Second) head, err := chainClient.HeaderByNumber(ctx, nil) cancel() if err != nil { log.Error("getHeadBlockNumber", "error", err) return nil, err } return head.Number, err } func nextRange(maxRangeSize int, prevFrom, zeroBlockNumber *big.Int) (*big.Int, *big.Int) { log.Debug("next range start", "from", prevFrom, "zeroBlockNumber", zeroBlockNumber) rangeSize := big.NewInt(int64(maxRangeSize)) to := big.NewInt(0).Set(prevFrom) from := big.NewInt(0).Sub(to, rangeSize) if from.Cmp(zeroBlockNumber) < 0 { from = new(big.Int).Set(zeroBlockNumber) } log.Debug("next range end", "from", from, "to", to, "zeroBlockNumber", zeroBlockNumber) return from, to } func accountLimiterTag(account common.Address) string { return transferHistoryTag + "_" + account.String() } func createChainClientWithLimiter(client chain.ClientInterface, account common.Address, limiter chain.RequestLimiter) (chain.ClientInterface, error) { // Each account has its own limit and a global limit for all accounts accountTag := accountLimiterTag(account) chainClient := chain.ClientWithTag(client, accountTag, transferHistoryTag) // Check if limit is already reached, then skip the comamnd if allow, err := limiter.Allow(accountTag); !allow { log.Info("fetchHistoryBlocksForAccount limit reached", "account", account, "chain", chainClient.NetworkID(), "error", err) return nil, err } if allow, err := limiter.Allow(transferHistoryTag); !allow { log.Info("fetchHistoryBlocksForAccount common limit reached", "chain", chainClient.NetworkID(), "error", err) return nil, err } limit, _ := limiter.GetLimit(accountTag) if limit == nil { err := limiter.SetLimit(accountTag, transferHistoryLimitPerAccount, chain.LimitInfinitely) if err != nil { log.Error("fetchHistoryBlocksForAccount SetLimit", "error", err, "accountTag", accountTag) } } // Here total limit per day is overwriten on each app start, that still saves us RPC calls, but allows to proceed // after app restart if the limit was reached. Currently there is no way to reset the limit from UI err := limiter.SetLimit(transferHistoryTag, transferHistoryLimit, transferHistoryLimitPeriod) if err != nil { log.Error("fetchHistoryBlocksForAccount SetLimit", "error", err, "groupTag", transferHistoryTag) } chainClient.SetLimiter(limiter) return chainClient, nil }