// ReceiveUpdatedUnconfirmedTransactions updates the wallet's unconfirmed
// transaction set.
func (w *Wallet) ReceiveUpdatedUnconfirmedTransactions(txns []types.Transaction, _ modules.ConsensusChange) {
// There are two different situations under which a subscribee calls
// ProcessConsensusChange. The first is when w.subscribed is set to false
// AND the mutex is already locked. The other situation is that subscribed
// is set to true and is not going to be changed. Therefore there is no
// race condition here. If w.subscribed is set to false, trying to grab the
// lock would cause a deadlock.
if w.subscribed {
lockID := w.mu.Lock()
defer w.mu.Unlock(lockID)
}
w.unconfirmedProcessedTransactions = nil
for _, txn := range txns {
// To save on code complexity, relveancy is determined while building
// up the wallet transaction.
relevant := false
pt := modules.ProcessedTransaction{
Transaction: txn,
TransactionID: txn.ID(),
ConfirmationHeight: types.BlockHeight(math.MaxUint64),
ConfirmationTimestamp: types.Timestamp(math.MaxUint64),
}
for _, sci := range txn.SiacoinInputs {
_, exists := w.keys[sci.UnlockConditions.UnlockHash()]
if exists {
relevant = true
}
pt.Inputs = append(pt.Inputs, modules.ProcessedInput{
FundType: types.SpecifierSiacoinInput,
WalletAddress: exists,
RelatedAddress: sci.UnlockConditions.UnlockHash(),
Value: w.historicOutputs[types.OutputID(sci.ParentID)],
})
}
for i, sco := range txn.SiacoinOutputs {
_, exists := w.keys[sco.UnlockHash]
if exists {
relevant = true
}
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierSiacoinOutput,
MaturityHeight: types.BlockHeight(math.MaxUint64),
WalletAddress: exists,
RelatedAddress: sco.UnlockHash,
Value: sco.Value,
})
w.historicOutputs[types.OutputID(txn.SiacoinOutputID(i))] = sco.Value
}
for _, fee := range txn.MinerFees {
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierMinerFee,
Value: fee,
})
}
if relevant {
w.unconfirmedProcessedTransactions = append(w.unconfirmedProcessedTransactions, pt)
}
}
}
// ReceiveUpdatedUnconfirmedTransactions updates the wallet's unconfirmed
// transaction set.
func (w *Wallet) ReceiveUpdatedUnconfirmedTransactions(txns []types.Transaction, _ modules.ConsensusChange) {
if err := w.tg.Add(); err != nil {
// Gracefully reject transactions if the wallet's Close method has
// closed the wallet's ThreadGroup already.
return
}
defer w.tg.Done()
w.mu.Lock()
defer w.mu.Unlock()
w.unconfirmedProcessedTransactions = nil
for _, txn := range txns {
// To save on code complexity, relevancy is determined while building
// up the wallet transaction.
relevant := false
pt := modules.ProcessedTransaction{
Transaction: txn,
TransactionID: txn.ID(),
ConfirmationHeight: types.BlockHeight(math.MaxUint64),
ConfirmationTimestamp: types.Timestamp(math.MaxUint64),
}
for _, sci := range txn.SiacoinInputs {
_, exists := w.keys[sci.UnlockConditions.UnlockHash()]
if exists {
relevant = true
}
pt.Inputs = append(pt.Inputs, modules.ProcessedInput{
FundType: types.SpecifierSiacoinInput,
WalletAddress: exists,
RelatedAddress: sci.UnlockConditions.UnlockHash(),
Value: w.historicOutputs[types.OutputID(sci.ParentID)],
})
}
for i, sco := range txn.SiacoinOutputs {
_, exists := w.keys[sco.UnlockHash]
if exists {
relevant = true
}
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierSiacoinOutput,
MaturityHeight: types.BlockHeight(math.MaxUint64),
WalletAddress: exists,
RelatedAddress: sco.UnlockHash,
Value: sco.Value,
})
w.historicOutputs[types.OutputID(txn.SiacoinOutputID(uint64(i)))] = sco.Value
}
for _, fee := range txn.MinerFees {
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierMinerFee,
Value: fee,
})
}
if relevant {
w.unconfirmedProcessedTransactions = append(w.unconfirmedProcessedTransactions, pt)
}
}
}
// ProcessConsensusChange parses a consensus change to update the set of
// confirmed outputs known to the wallet.
func (w *Wallet) ProcessConsensusChange(cc modules.ConsensusChange) {
// There are two different situations under which a subscribee calls
// ProcessConsensusChange. The first is when w.subscribed is set to false
// AND the mutex is already locked. The other situation is that subscribed
// is set to true and is not going to be changed. Therefore there is no
// race condition here. If w.subscribed is set to false, trying to grab the
// lock would cause a deadlock.
if w.subscribed {
lockID := w.mu.Lock()
defer w.mu.Unlock(lockID)
}
// Iterate through the output diffs (siacoin and siafund) and apply all of
// them. Only apply the outputs that relate to unlock hashes we understand.
for _, diff := range cc.SiacoinOutputDiffs {
// Verify that the diff is relevant to the wallet.
_, exists := w.keys[diff.SiacoinOutput.UnlockHash]
if !exists {
continue
}
_, exists = w.siacoinOutputs[diff.ID]
if diff.Direction == modules.DiffApply {
if exists && build.DEBUG {
panic("adding an existing output to wallet")
}
w.siacoinOutputs[diff.ID] = diff.SiacoinOutput
} else {
if !exists && build.DEBUG {
panic("deleting nonexisting output from wallet")
}
delete(w.siacoinOutputs, diff.ID)
}
}
for _, diff := range cc.SiafundOutputDiffs {
// Verify that the diff is relevant to the wallet.
_, exists := w.keys[diff.SiafundOutput.UnlockHash]
if !exists {
continue
}
_, exists = w.siafundOutputs[diff.ID]
if diff.Direction == modules.DiffApply {
if exists && build.DEBUG {
panic("adding an existing output to wallet")
}
w.siafundOutputs[diff.ID] = diff.SiafundOutput
} else {
if !exists && build.DEBUG {
panic("deleting nonexisting output from wallet")
}
delete(w.siafundOutputs, diff.ID)
}
}
for _, diff := range cc.SiafundPoolDiffs {
if diff.Direction == modules.DiffApply {
w.siafundPool = diff.Adjusted
} else {
w.siafundPool = diff.Previous
}
}
// Iterate through the transactions and find every transaction somehow
// related to the wallet. Wallet transactions must be removed in the same
// order they were added.
for _, block := range cc.RevertedBlocks {
// Remove any transactions that have been reverted.
for i := len(block.Transactions) - 1; i >= 0; i-- {
// If the transaction is relevant to the wallet, it will be the
// most recent transaction appended to w.processedTransactions.
// Relevance can be determined just by looking at the last element
// of w.processedTransactions.
txn := block.Transactions[i]
txid := txn.ID()
if len(w.processedTransactions) > 0 && txid == w.processedTransactions[len(w.processedTransactions)-1].TransactionID {
w.processedTransactions = w.processedTransactions[:len(w.processedTransactions)-1]
delete(w.processedTransactionMap, txid)
}
}
// Remove the miner payout transaction if applicable.
for _, mp := range block.MinerPayouts {
_, exists := w.keys[mp.UnlockHash]
if exists {
w.processedTransactions = w.processedTransactions[:len(w.processedTransactions)-1]
delete(w.processedTransactionMap, types.TransactionID(block.ID()))
break
}
}
w.consensusSetHeight--
}
// Apply all of the new blocks.
for _, block := range cc.AppliedBlocks {
w.consensusSetHeight++
// Apply the miner payout transaction if applicable.
minerPT := modules.ProcessedTransaction{
Transaction: types.Transaction{},
TransactionID: types.TransactionID(block.ID()),
ConfirmationHeight: w.consensusSetHeight,
ConfirmationTimestamp: block.Timestamp,
}
relevant := false
for i, mp := range block.MinerPayouts {
_, exists := w.keys[mp.UnlockHash]
if exists {
relevant = true
}
minerPT.Outputs = append(minerPT.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierMinerPayout,
MaturityHeight: w.consensusSetHeight + types.MaturityDelay,
WalletAddress: exists,
RelatedAddress: mp.UnlockHash,
Value: mp.Value,
})
w.historicOutputs[types.OutputID(block.MinerPayoutID(uint64(i)))] = mp.Value
}
if relevant {
w.processedTransactions = append(w.processedTransactions, minerPT)
w.processedTransactionMap[minerPT.TransactionID] = &w.processedTransactions[len(w.processedTransactions)-1]
}
for _, txn := range block.Transactions {
relevant := false
pt := modules.ProcessedTransaction{
Transaction: txn,
TransactionID: txn.ID(),
ConfirmationHeight: w.consensusSetHeight,
ConfirmationTimestamp: block.Timestamp,
}
for _, sci := range txn.SiacoinInputs {
_, exists := w.keys[sci.UnlockConditions.UnlockHash()]
if exists {
relevant = true
}
pt.Inputs = append(pt.Inputs, modules.ProcessedInput{
FundType: types.SpecifierSiacoinInput,
WalletAddress: exists,
RelatedAddress: sci.UnlockConditions.UnlockHash(),
Value: w.historicOutputs[types.OutputID(sci.ParentID)],
})
}
for i, sco := range txn.SiacoinOutputs {
_, exists := w.keys[sco.UnlockHash]
if exists {
relevant = true
}
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierSiacoinOutput,
MaturityHeight: w.consensusSetHeight,
WalletAddress: exists,
RelatedAddress: sco.UnlockHash,
Value: sco.Value,
})
w.historicOutputs[types.OutputID(txn.SiacoinOutputID(i))] = sco.Value
}
for _, sfi := range txn.SiafundInputs {
_, exists := w.keys[sfi.UnlockConditions.UnlockHash()]
if exists {
relevant = true
}
sfiValue := w.historicOutputs[types.OutputID(sfi.ParentID)]
pt.Inputs = append(pt.Inputs, modules.ProcessedInput{
FundType: types.SpecifierSiafundInput,
WalletAddress: exists,
RelatedAddress: sfi.UnlockConditions.UnlockHash(),
Value: sfiValue,
})
claimValue := w.siafundPool.Sub(w.historicClaimStarts[sfi.ParentID]).Mul(sfiValue)
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierClaimOutput,
MaturityHeight: w.consensusSetHeight + types.MaturityDelay,
WalletAddress: exists,
RelatedAddress: sfi.ClaimUnlockHash,
Value: claimValue,
})
}
for i, sfo := range txn.SiafundOutputs {
_, exists := w.keys[sfo.UnlockHash]
if exists {
relevant = true
}
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierSiafundOutput,
MaturityHeight: w.consensusSetHeight,
WalletAddress: exists,
RelatedAddress: sfo.UnlockHash,
Value: sfo.Value,
})
w.historicOutputs[types.OutputID(txn.SiafundOutputID(i))] = sfo.Value
w.historicClaimStarts[txn.SiafundOutputID(i)] = sfo.ClaimStart
}
for _, fee := range txn.MinerFees {
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierMinerFee,
Value: fee,
})
}
if relevant {
w.processedTransactions = append(w.processedTransactions, pt)
w.processedTransactionMap[pt.TransactionID] = &w.processedTransactions[len(w.processedTransactions)-1]
}
}
}
}
// applyHistory applies any transaction history that was introduced by the
// applied blocks.
func (w *Wallet) applyHistory(cc modules.ConsensusChange) {
for _, block := range cc.AppliedBlocks {
w.consensusSetHeight++
// Apply the miner payout transaction if applicable.
minerPT := modules.ProcessedTransaction{
Transaction: types.Transaction{},
TransactionID: types.TransactionID(block.ID()),
ConfirmationHeight: w.consensusSetHeight,
ConfirmationTimestamp: block.Timestamp,
}
relevant := false
for i, mp := range block.MinerPayouts {
_, exists := w.keys[mp.UnlockHash]
if exists {
relevant = true
}
minerPT.Outputs = append(minerPT.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierMinerPayout,
MaturityHeight: w.consensusSetHeight + types.MaturityDelay,
WalletAddress: exists,
RelatedAddress: mp.UnlockHash,
Value: mp.Value,
})
w.historicOutputs[types.OutputID(block.MinerPayoutID(uint64(i)))] = mp.Value
}
if relevant {
w.processedTransactions = append(w.processedTransactions, minerPT)
w.processedTransactionMap[minerPT.TransactionID] = &w.processedTransactions[len(w.processedTransactions)-1]
}
for _, txn := range block.Transactions {
relevant := false
pt := modules.ProcessedTransaction{
Transaction: txn,
TransactionID: txn.ID(),
ConfirmationHeight: w.consensusSetHeight,
ConfirmationTimestamp: block.Timestamp,
}
for _, sci := range txn.SiacoinInputs {
_, exists := w.keys[sci.UnlockConditions.UnlockHash()]
if exists {
relevant = true
}
pt.Inputs = append(pt.Inputs, modules.ProcessedInput{
FundType: types.SpecifierSiacoinInput,
WalletAddress: exists,
RelatedAddress: sci.UnlockConditions.UnlockHash(),
Value: w.historicOutputs[types.OutputID(sci.ParentID)],
})
}
for i, sco := range txn.SiacoinOutputs {
_, exists := w.keys[sco.UnlockHash]
if exists {
relevant = true
}
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierSiacoinOutput,
MaturityHeight: w.consensusSetHeight,
WalletAddress: exists,
RelatedAddress: sco.UnlockHash,
Value: sco.Value,
})
w.historicOutputs[types.OutputID(txn.SiacoinOutputID(i))] = sco.Value
}
for _, sfi := range txn.SiafundInputs {
_, exists := w.keys[sfi.UnlockConditions.UnlockHash()]
if exists {
relevant = true
}
sfiValue := w.historicOutputs[types.OutputID(sfi.ParentID)]
pt.Inputs = append(pt.Inputs, modules.ProcessedInput{
FundType: types.SpecifierSiafundInput,
WalletAddress: exists,
RelatedAddress: sfi.UnlockConditions.UnlockHash(),
Value: sfiValue,
})
claimValue := w.siafundPool.Sub(w.historicClaimStarts[sfi.ParentID]).Mul(sfiValue)
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierClaimOutput,
MaturityHeight: w.consensusSetHeight + types.MaturityDelay,
WalletAddress: exists,
RelatedAddress: sfi.ClaimUnlockHash,
Value: claimValue,
})
}
for i, sfo := range txn.SiafundOutputs {
_, exists := w.keys[sfo.UnlockHash]
if exists {
relevant = true
}
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierSiafundOutput,
MaturityHeight: w.consensusSetHeight,
WalletAddress: exists,
RelatedAddress: sfo.UnlockHash,
Value: sfo.Value,
})
w.historicOutputs[types.OutputID(txn.SiafundOutputID(i))] = sfo.Value
w.historicClaimStarts[txn.SiafundOutputID(i)] = sfo.ClaimStart
}
for _, fee := range txn.MinerFees {
pt.Outputs = append(pt.Outputs, modules.ProcessedOutput{
FundType: types.SpecifierMinerFee,
Value: fee,
})
}
if relevant {
w.processedTransactions = append(w.processedTransactions, pt)
w.processedTransactionMap[pt.TransactionID] = &w.processedTransactions[len(w.processedTransactions)-1]
}
}
}
}