2018-01-17 06:37:09 +03:00
|
|
|
package contractcourt
|
2018-03-19 16:11:13 +03:00
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
2018-07-18 05:42:43 +03:00
|
|
|
"github.com/btcsuite/btcd/chaincfg/chainhash"
|
|
|
|
"github.com/btcsuite/btcd/wire"
|
2018-03-19 16:11:13 +03:00
|
|
|
"github.com/lightningnetwork/lnd/chainntnfs"
|
2018-08-21 13:21:15 +03:00
|
|
|
"github.com/lightningnetwork/lnd/channeldb"
|
2018-03-19 16:11:13 +03:00
|
|
|
"github.com/lightningnetwork/lnd/lnwallet"
|
|
|
|
"github.com/lightningnetwork/lnd/lnwire"
|
|
|
|
)
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
type mockArbitratorLog struct {
|
2018-08-21 13:21:16 +03:00
|
|
|
state ArbitratorState
|
|
|
|
newStates chan ArbitratorState
|
|
|
|
failLog bool
|
|
|
|
failFetch error
|
|
|
|
failCommit bool
|
|
|
|
failCommitState ArbitratorState
|
2018-08-21 13:21:14 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// A compile time check to ensure mockArbitratorLog meets the ArbitratorLog
|
|
|
|
// interface.
|
|
|
|
var _ ArbitratorLog = (*mockArbitratorLog)(nil)
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) CurrentState() (ArbitratorState, error) {
|
|
|
|
return b.state, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) CommitState(s ArbitratorState) error {
|
2018-08-21 13:21:16 +03:00
|
|
|
if b.failCommit && s == b.failCommitState {
|
|
|
|
return fmt.Errorf("intentional commit error at state %v",
|
|
|
|
b.failCommitState)
|
|
|
|
}
|
2018-08-21 13:21:14 +03:00
|
|
|
b.state = s
|
|
|
|
b.newStates <- s
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) FetchUnresolvedContracts() ([]ContractResolver, error) {
|
|
|
|
var contracts []ContractResolver
|
|
|
|
return contracts, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) InsertUnresolvedContracts(resolvers ...ContractResolver) error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) SwapContract(oldContract, newContract ContractResolver) error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) ResolveContract(res ContractResolver) error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) LogContractResolutions(c *ContractResolutions) error {
|
2018-08-21 13:21:15 +03:00
|
|
|
if b.failLog {
|
|
|
|
return fmt.Errorf("intentional log failure")
|
|
|
|
}
|
2018-08-21 13:21:14 +03:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) FetchContractResolutions() (*ContractResolutions, error) {
|
2018-08-21 13:21:15 +03:00
|
|
|
if b.failFetch != nil {
|
|
|
|
return nil, b.failFetch
|
|
|
|
}
|
2018-08-21 13:21:14 +03:00
|
|
|
c := &ContractResolutions{}
|
|
|
|
|
|
|
|
return c, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) LogChainActions(actions ChainActionMap) error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) FetchChainActions() (ChainActionMap, error) {
|
|
|
|
actionsMap := make(ChainActionMap)
|
|
|
|
return actionsMap, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *mockArbitratorLog) WipeHistory() error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-03-19 16:11:13 +03:00
|
|
|
type mockChainIO struct{}
|
|
|
|
|
|
|
|
func (*mockChainIO) GetBestBlock() (*chainhash.Hash, int32, error) {
|
|
|
|
return nil, 0, nil
|
|
|
|
}
|
|
|
|
|
2018-07-18 05:42:43 +03:00
|
|
|
func (*mockChainIO) GetUtxo(op *wire.OutPoint, _ []byte,
|
2018-03-19 16:11:13 +03:00
|
|
|
heightHint uint32) (*wire.TxOut, error) {
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (*mockChainIO) GetBlockHash(blockHeight int64) (*chainhash.Hash, error) {
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (*mockChainIO) GetBlock(blockHash *chainhash.Hash) (*wire.MsgBlock, error) {
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
func createTestChannelArbitrator(log ArbitratorLog) (*ChannelArbitrator,
|
|
|
|
chan struct{}, error) {
|
2018-03-19 16:11:13 +03:00
|
|
|
blockEpoch := &chainntnfs.BlockEpochEvent{
|
|
|
|
Cancel: func() {},
|
|
|
|
}
|
|
|
|
|
|
|
|
chanPoint := wire.OutPoint{}
|
|
|
|
shortChanID := lnwire.ShortChannelID{}
|
|
|
|
chanEvents := &ChainEventSubscription{
|
|
|
|
RemoteUnilateralClosure: make(chan *lnwallet.UnilateralCloseSummary, 1),
|
|
|
|
LocalUnilateralClosure: make(chan *LocalUnilateralCloseInfo, 1),
|
2018-08-21 13:21:15 +03:00
|
|
|
CooperativeClosure: make(chan *CooperativeCloseInfo, 1),
|
2018-03-19 16:11:13 +03:00
|
|
|
ContractBreach: make(chan *lnwallet.BreachRetribution, 1),
|
|
|
|
}
|
|
|
|
|
|
|
|
chainIO := &mockChainIO{}
|
|
|
|
chainArbCfg := ChainArbitratorConfig{
|
|
|
|
ChainIO: chainIO,
|
|
|
|
PublishTx: func(*wire.MsgTx) error {
|
|
|
|
return nil
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
// We'll use the resolvedChan to synchronize on call to
|
|
|
|
// MarkChannelResolved.
|
|
|
|
resolvedChan := make(chan struct{}, 1)
|
|
|
|
|
|
|
|
// Next we'll create the matching configuration struct that contains
|
|
|
|
// all interfaces and methods the arbitrator needs to do its job.
|
|
|
|
arbCfg := ChannelArbitratorConfig{
|
|
|
|
ChanPoint: chanPoint,
|
|
|
|
ShortChanID: shortChanID,
|
|
|
|
BlockEpochs: blockEpoch,
|
|
|
|
MarkChannelResolved: func() error {
|
|
|
|
resolvedChan <- struct{}{}
|
|
|
|
return nil
|
|
|
|
},
|
|
|
|
ForceCloseChan: func() (*lnwallet.LocalForceCloseSummary, error) {
|
|
|
|
summary := &lnwallet.LocalForceCloseSummary{
|
|
|
|
CloseTx: &wire.MsgTx{},
|
|
|
|
HtlcResolutions: &lnwallet.HtlcResolutions{},
|
|
|
|
}
|
|
|
|
return summary, nil
|
|
|
|
},
|
|
|
|
MarkCommitmentBroadcasted: func() error {
|
|
|
|
return nil
|
|
|
|
},
|
2018-08-21 13:21:15 +03:00
|
|
|
MarkChannelClosed: func(*channeldb.ChannelCloseSummary) error {
|
|
|
|
return nil
|
|
|
|
},
|
2018-08-21 13:21:16 +03:00
|
|
|
IsPendingClose: false,
|
2018-03-19 16:11:13 +03:00
|
|
|
ChainArbitratorConfig: chainArbCfg,
|
|
|
|
ChainEvents: chanEvents,
|
|
|
|
}
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
return NewChannelArbitrator(arbCfg, nil, log), resolvedChan, nil
|
2018-03-19 16:11:13 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// assertState checks that the ChannelArbitrator is in the state we expect it
|
|
|
|
// to be.
|
|
|
|
func assertState(t *testing.T, c *ChannelArbitrator, expected ArbitratorState) {
|
|
|
|
if c.state != expected {
|
|
|
|
t.Fatalf("expected state %v, was %v", expected, c.state)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// TestChannelArbitratorCooperativeClose tests that the ChannelArbitertor
|
2018-08-21 13:21:15 +03:00
|
|
|
// correctly marks the channel resolved in case a cooperative close is
|
|
|
|
// confirmed.
|
2018-03-19 16:11:13 +03:00
|
|
|
func TestChannelArbitratorCooperativeClose(t *testing.T) {
|
2018-08-21 13:21:14 +03:00
|
|
|
log := &mockArbitratorLog{
|
|
|
|
state: StateDefault,
|
|
|
|
newStates: make(chan ArbitratorState, 5),
|
|
|
|
}
|
|
|
|
|
2018-08-21 13:21:15 +03:00
|
|
|
chanArb, resolved, err := createTestChannelArbitrator(log)
|
2018-03-19 16:11:13 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
defer chanArb.Stop()
|
|
|
|
|
|
|
|
// It should start out in the default state.
|
|
|
|
assertState(t, chanArb, StateDefault)
|
|
|
|
|
2018-08-21 13:21:15 +03:00
|
|
|
// We set up a channel to detect when MarkChannelClosed is called.
|
|
|
|
closeInfos := make(chan *channeldb.ChannelCloseSummary)
|
|
|
|
chanArb.cfg.MarkChannelClosed = func(
|
|
|
|
closeInfo *channeldb.ChannelCloseSummary) error {
|
|
|
|
closeInfos <- closeInfo
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Cooperative close should do trigger a MarkChannelClosed +
|
|
|
|
// MarkChannelResolved.
|
|
|
|
closeInfo := &CooperativeCloseInfo{
|
|
|
|
&channeldb.ChannelCloseSummary{},
|
|
|
|
}
|
|
|
|
chanArb.cfg.ChainEvents.CooperativeClosure <- closeInfo
|
|
|
|
|
|
|
|
select {
|
|
|
|
case c := <-closeInfos:
|
|
|
|
if c.CloseType != channeldb.CooperativeClose {
|
|
|
|
t.Fatalf("expected cooperative close, got %v", c.CloseType)
|
|
|
|
}
|
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("timeout waiting for channel close")
|
|
|
|
}
|
|
|
|
|
|
|
|
// It should mark the channel as resolved.
|
|
|
|
select {
|
|
|
|
case <-resolved:
|
|
|
|
// Expected.
|
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("contract was not resolved")
|
|
|
|
}
|
2018-03-19 16:11:13 +03:00
|
|
|
}
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
func assertStateTransitions(t *testing.T, newStates <-chan ArbitratorState,
|
|
|
|
expectedStates ...ArbitratorState) {
|
|
|
|
t.Helper()
|
|
|
|
|
|
|
|
for _, exp := range expectedStates {
|
|
|
|
var state ArbitratorState
|
|
|
|
select {
|
|
|
|
case state = <-newStates:
|
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("new state not received")
|
|
|
|
}
|
|
|
|
|
|
|
|
if state != exp {
|
|
|
|
t.Fatalf("expected new state %v, got %v", exp, state)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// TestChannelArbitratorRemoteForceClose checks that the ChannelArbitrator goes
|
2018-03-19 16:11:13 +03:00
|
|
|
// through the expected states if a remote force close is observed in the
|
|
|
|
// chain.
|
|
|
|
func TestChannelArbitratorRemoteForceClose(t *testing.T) {
|
2018-08-21 13:21:14 +03:00
|
|
|
log := &mockArbitratorLog{
|
|
|
|
state: StateDefault,
|
|
|
|
newStates: make(chan ArbitratorState, 5),
|
|
|
|
}
|
|
|
|
|
|
|
|
chanArb, resolved, err := createTestChannelArbitrator(log)
|
2018-03-19 16:11:13 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
defer chanArb.Stop()
|
|
|
|
|
|
|
|
// It should start out in the default state.
|
|
|
|
assertState(t, chanArb, StateDefault)
|
|
|
|
|
|
|
|
// Send a remote force close event.
|
|
|
|
commitSpend := &chainntnfs.SpendDetail{
|
|
|
|
SpenderTxHash: &chainhash.Hash{},
|
|
|
|
}
|
|
|
|
|
|
|
|
uniClose := &lnwallet.UnilateralCloseSummary{
|
|
|
|
SpendDetail: commitSpend,
|
|
|
|
HtlcResolutions: &lnwallet.HtlcResolutions{},
|
|
|
|
}
|
|
|
|
chanArb.cfg.ChainEvents.RemoteUnilateralClosure <- uniClose
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
// It should transition StateDefault -> StateContractClosed ->
|
|
|
|
// StateFullyResolved.
|
|
|
|
assertStateTransitions(
|
|
|
|
t, log.newStates, StateContractClosed, StateFullyResolved,
|
|
|
|
)
|
|
|
|
|
|
|
|
// It should alos mark the channel as resolved.
|
2018-03-19 16:11:13 +03:00
|
|
|
select {
|
|
|
|
case <-resolved:
|
|
|
|
// Expected.
|
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("contract was not resolved")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// TestChannelArbitratorLocalForceClose tests that the ChannelArbitrator goes
|
|
|
|
// through the expected states in case we request it to force close the channel,
|
|
|
|
// and the local force close event is observed in chain.
|
|
|
|
func TestChannelArbitratorLocalForceClose(t *testing.T) {
|
2018-08-21 13:21:14 +03:00
|
|
|
log := &mockArbitratorLog{
|
|
|
|
state: StateDefault,
|
|
|
|
newStates: make(chan ArbitratorState, 5),
|
|
|
|
}
|
|
|
|
|
|
|
|
chanArb, resolved, err := createTestChannelArbitrator(log)
|
2018-03-19 16:11:13 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
defer chanArb.Stop()
|
|
|
|
|
|
|
|
// It should start out in the default state.
|
|
|
|
assertState(t, chanArb, StateDefault)
|
|
|
|
|
|
|
|
// We create a channel we can use to pause the ChannelArbitrator at the
|
|
|
|
// point where it broadcasts the close tx, and check its state.
|
|
|
|
stateChan := make(chan ArbitratorState)
|
|
|
|
chanArb.cfg.PublishTx = func(*wire.MsgTx) error {
|
|
|
|
// When the force close tx is being broadcasted, check that the
|
|
|
|
// state is correct at that point.
|
|
|
|
select {
|
|
|
|
case stateChan <- chanArb.state:
|
|
|
|
case <-chanArb.quit:
|
|
|
|
return fmt.Errorf("exiting")
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
errChan := make(chan error, 1)
|
|
|
|
respChan := make(chan *wire.MsgTx, 1)
|
|
|
|
|
|
|
|
// With the channel found, and the request crafted, we'll send over a
|
|
|
|
// force close request to the arbitrator that watches this channel.
|
|
|
|
chanArb.forceCloseReqs <- &forceCloseReq{
|
|
|
|
errResp: errChan,
|
|
|
|
closeTx: respChan,
|
|
|
|
}
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
// It should transition to StateBroadcastCommit.
|
|
|
|
assertStateTransitions(t, log.newStates, StateBroadcastCommit)
|
|
|
|
|
2018-03-19 16:11:13 +03:00
|
|
|
// When it is broadcasting the force close, its state should be
|
|
|
|
// StateBroadcastCommit.
|
|
|
|
select {
|
|
|
|
case state := <-stateChan:
|
|
|
|
if state != StateBroadcastCommit {
|
|
|
|
t.Fatalf("state during PublishTx was %v", state)
|
|
|
|
}
|
|
|
|
case <-time.After(15 * time.Second):
|
|
|
|
t.Fatalf("did not get state update")
|
|
|
|
}
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
// After broadcasting, transition should be to
|
|
|
|
// StateCommitmentBroadcasted.
|
|
|
|
assertStateTransitions(t, log.newStates, StateCommitmentBroadcasted)
|
|
|
|
|
2018-03-19 16:11:13 +03:00
|
|
|
select {
|
|
|
|
case <-respChan:
|
2018-08-21 13:21:14 +03:00
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("no response received")
|
|
|
|
}
|
|
|
|
|
|
|
|
select {
|
2018-03-19 16:11:13 +03:00
|
|
|
case err := <-errChan:
|
2018-08-21 13:21:14 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("error force closing channel: %v", err)
|
|
|
|
}
|
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("no response received")
|
2018-03-19 16:11:13 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// After broadcasting the close tx, it should be in state
|
|
|
|
// StateCommitmentBroadcasted.
|
|
|
|
assertState(t, chanArb, StateCommitmentBroadcasted)
|
|
|
|
|
|
|
|
// Now notify about the local force close getting confirmed.
|
|
|
|
chanArb.cfg.ChainEvents.LocalUnilateralClosure <- &LocalUnilateralCloseInfo{
|
|
|
|
&chainntnfs.SpendDetail{},
|
|
|
|
&lnwallet.LocalForceCloseSummary{
|
|
|
|
CloseTx: &wire.MsgTx{},
|
|
|
|
HtlcResolutions: &lnwallet.HtlcResolutions{},
|
|
|
|
},
|
2018-08-21 13:21:15 +03:00
|
|
|
&channeldb.ChannelCloseSummary{},
|
2018-03-19 16:11:13 +03:00
|
|
|
}
|
2018-08-21 13:21:14 +03:00
|
|
|
|
|
|
|
// It should transition StateContractClosed -> StateFullyResolved.
|
|
|
|
assertStateTransitions(t, log.newStates, StateContractClosed,
|
|
|
|
StateFullyResolved)
|
|
|
|
|
|
|
|
// It should also mark the channel as resolved.
|
2018-03-19 16:11:13 +03:00
|
|
|
select {
|
|
|
|
case <-resolved:
|
|
|
|
// Expected.
|
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("contract was not resolved")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// TestChannelArbitratorLocalForceCloseRemoteConfiremd tests that the
|
|
|
|
// ChannelArbitrator behaves as expected in the case where we request a local
|
|
|
|
// force close, but a remote commitment ends up being confirmed in chain.
|
|
|
|
func TestChannelArbitratorLocalForceCloseRemoteConfirmed(t *testing.T) {
|
2018-08-21 13:21:14 +03:00
|
|
|
log := &mockArbitratorLog{
|
|
|
|
state: StateDefault,
|
|
|
|
newStates: make(chan ArbitratorState, 5),
|
|
|
|
}
|
|
|
|
|
|
|
|
chanArb, resolved, err := createTestChannelArbitrator(log)
|
2018-03-19 16:11:13 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
defer chanArb.Stop()
|
|
|
|
|
|
|
|
// It should start out in the default state.
|
|
|
|
assertState(t, chanArb, StateDefault)
|
|
|
|
|
|
|
|
// Create a channel we can use to assert the state when it publishes
|
|
|
|
// the close tx.
|
|
|
|
stateChan := make(chan ArbitratorState)
|
|
|
|
chanArb.cfg.PublishTx = func(*wire.MsgTx) error {
|
|
|
|
// When the force close tx is being broadcasted, check that the
|
|
|
|
// state is correct at that point.
|
|
|
|
select {
|
|
|
|
case stateChan <- chanArb.state:
|
|
|
|
case <-chanArb.quit:
|
|
|
|
return fmt.Errorf("exiting")
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
errChan := make(chan error, 1)
|
|
|
|
respChan := make(chan *wire.MsgTx, 1)
|
|
|
|
|
|
|
|
// With the channel found, and the request crafted, we'll send over a
|
|
|
|
// force close request to the arbitrator that watches this channel.
|
|
|
|
chanArb.forceCloseReqs <- &forceCloseReq{
|
|
|
|
errResp: errChan,
|
|
|
|
closeTx: respChan,
|
|
|
|
}
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
// It should transition to StateBroadcastCommit.
|
|
|
|
assertStateTransitions(t, log.newStates, StateBroadcastCommit)
|
|
|
|
|
2018-03-19 16:11:13 +03:00
|
|
|
// We expect it to be in state StateBroadcastCommit when publishing
|
|
|
|
// the force close.
|
|
|
|
select {
|
|
|
|
case state := <-stateChan:
|
|
|
|
if state != StateBroadcastCommit {
|
|
|
|
t.Fatalf("state during PublishTx was %v", state)
|
|
|
|
}
|
|
|
|
case <-time.After(15 * time.Second):
|
2018-05-14 15:21:06 +03:00
|
|
|
t.Fatalf("no state update received")
|
|
|
|
}
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
// After broadcasting, transition should be to
|
|
|
|
// StateCommitmentBroadcasted.
|
|
|
|
assertStateTransitions(t, log.newStates, StateCommitmentBroadcasted)
|
|
|
|
|
2018-05-14 15:21:06 +03:00
|
|
|
// Wait for a response to the force close.
|
|
|
|
select {
|
|
|
|
case <-respChan:
|
2018-08-21 13:21:14 +03:00
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("no response received")
|
|
|
|
}
|
|
|
|
|
|
|
|
select {
|
2018-05-14 15:21:06 +03:00
|
|
|
case err := <-errChan:
|
2018-08-21 13:21:14 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("error force closing channel: %v", err)
|
|
|
|
}
|
|
|
|
case <-time.After(5 * time.Second):
|
2018-05-14 15:21:06 +03:00
|
|
|
t.Fatalf("no response received")
|
|
|
|
}
|
|
|
|
|
|
|
|
// The state should be StateCommitmentBroadcasted.
|
|
|
|
assertState(t, chanArb, StateCommitmentBroadcasted)
|
|
|
|
|
|
|
|
// Now notify about the _REMOTE_ commitment getting confirmed.
|
|
|
|
commitSpend := &chainntnfs.SpendDetail{
|
|
|
|
SpenderTxHash: &chainhash.Hash{},
|
|
|
|
}
|
|
|
|
uniClose := &lnwallet.UnilateralCloseSummary{
|
|
|
|
SpendDetail: commitSpend,
|
|
|
|
HtlcResolutions: &lnwallet.HtlcResolutions{},
|
|
|
|
}
|
|
|
|
chanArb.cfg.ChainEvents.RemoteUnilateralClosure <- uniClose
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
// It should transition StateContractClosed -> StateFullyResolved.
|
|
|
|
assertStateTransitions(t, log.newStates, StateContractClosed,
|
|
|
|
StateFullyResolved)
|
|
|
|
|
2018-05-14 15:21:06 +03:00
|
|
|
// It should resolve.
|
|
|
|
select {
|
|
|
|
case <-resolved:
|
|
|
|
// Expected.
|
|
|
|
case <-time.After(15 * time.Second):
|
|
|
|
t.Fatalf("contract was not resolved")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// TestChannelArbitratorLocalForceCloseDoubleSpend tests that the
|
|
|
|
// ChannelArbitrator behaves as expected in the case where we request a local
|
|
|
|
// force close, but we fail broadcasting our commitment because a remote
|
|
|
|
// commitment has already been published.
|
|
|
|
func TestChannelArbitratorLocalForceDoubleSpend(t *testing.T) {
|
2018-08-21 13:21:14 +03:00
|
|
|
log := &mockArbitratorLog{
|
|
|
|
state: StateDefault,
|
|
|
|
newStates: make(chan ArbitratorState, 5),
|
|
|
|
}
|
|
|
|
|
|
|
|
chanArb, resolved, err := createTestChannelArbitrator(log)
|
2018-05-14 15:21:06 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
defer chanArb.Stop()
|
|
|
|
|
|
|
|
// It should start out in the default state.
|
|
|
|
assertState(t, chanArb, StateDefault)
|
|
|
|
|
|
|
|
// Return ErrDoubleSpend when attempting to publish the tx.
|
|
|
|
stateChan := make(chan ArbitratorState)
|
|
|
|
chanArb.cfg.PublishTx = func(*wire.MsgTx) error {
|
|
|
|
// When the force close tx is being broadcasted, check that the
|
|
|
|
// state is correct at that point.
|
|
|
|
select {
|
|
|
|
case stateChan <- chanArb.state:
|
|
|
|
case <-chanArb.quit:
|
|
|
|
return fmt.Errorf("exiting")
|
|
|
|
}
|
|
|
|
return lnwallet.ErrDoubleSpend
|
|
|
|
}
|
|
|
|
|
|
|
|
errChan := make(chan error, 1)
|
|
|
|
respChan := make(chan *wire.MsgTx, 1)
|
|
|
|
|
|
|
|
// With the channel found, and the request crafted, we'll send over a
|
|
|
|
// force close request to the arbitrator that watches this channel.
|
|
|
|
chanArb.forceCloseReqs <- &forceCloseReq{
|
|
|
|
errResp: errChan,
|
|
|
|
closeTx: respChan,
|
|
|
|
}
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
// It should transition to StateBroadcastCommit.
|
|
|
|
assertStateTransitions(t, log.newStates, StateBroadcastCommit)
|
|
|
|
|
2018-05-14 15:21:06 +03:00
|
|
|
// We expect it to be in state StateBroadcastCommit when publishing
|
|
|
|
// the force close.
|
|
|
|
select {
|
|
|
|
case state := <-stateChan:
|
|
|
|
if state != StateBroadcastCommit {
|
|
|
|
t.Fatalf("state during PublishTx was %v", state)
|
|
|
|
}
|
|
|
|
case <-time.After(15 * time.Second):
|
2018-03-19 16:11:13 +03:00
|
|
|
t.Fatalf("no state update received")
|
|
|
|
}
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
// After broadcasting, transition should be to
|
|
|
|
// StateCommitmentBroadcasted.
|
|
|
|
assertStateTransitions(t, log.newStates, StateCommitmentBroadcasted)
|
|
|
|
|
2018-03-19 16:11:13 +03:00
|
|
|
// Wait for a response to the force close.
|
|
|
|
select {
|
|
|
|
case <-respChan:
|
2018-08-21 13:21:14 +03:00
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("no response received")
|
|
|
|
}
|
|
|
|
|
|
|
|
select {
|
2018-03-19 16:11:13 +03:00
|
|
|
case err := <-errChan:
|
2018-08-21 13:21:14 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("error force closing channel: %v", err)
|
|
|
|
}
|
|
|
|
case <-time.After(5 * time.Second):
|
2018-03-19 16:11:13 +03:00
|
|
|
t.Fatalf("no response received")
|
|
|
|
}
|
|
|
|
|
|
|
|
// The state should be StateCommitmentBroadcasted.
|
|
|
|
assertState(t, chanArb, StateCommitmentBroadcasted)
|
|
|
|
|
|
|
|
// Now notify about the _REMOTE_ commitment getting confirmed.
|
|
|
|
commitSpend := &chainntnfs.SpendDetail{
|
|
|
|
SpenderTxHash: &chainhash.Hash{},
|
|
|
|
}
|
|
|
|
uniClose := &lnwallet.UnilateralCloseSummary{
|
|
|
|
SpendDetail: commitSpend,
|
|
|
|
HtlcResolutions: &lnwallet.HtlcResolutions{},
|
|
|
|
}
|
|
|
|
chanArb.cfg.ChainEvents.RemoteUnilateralClosure <- uniClose
|
|
|
|
|
2018-08-21 13:21:14 +03:00
|
|
|
// It should transition StateContractClosed -> StateFullyResolved.
|
|
|
|
assertStateTransitions(t, log.newStates, StateContractClosed,
|
|
|
|
StateFullyResolved)
|
|
|
|
|
2018-03-19 16:11:13 +03:00
|
|
|
// It should resolve.
|
|
|
|
select {
|
|
|
|
case <-resolved:
|
|
|
|
// Expected.
|
|
|
|
case <-time.After(15 * time.Second):
|
|
|
|
t.Fatalf("contract was not resolved")
|
|
|
|
}
|
|
|
|
}
|
2018-08-21 13:21:15 +03:00
|
|
|
|
|
|
|
// TestChannelArbitratorPersistence tests that the ChannelArbitrator is able to
|
|
|
|
// keep advancing the state machine from various states after restart.
|
|
|
|
func TestChannelArbitratorPersistence(t *testing.T) {
|
|
|
|
// Start out with a log that will fail writing the set of resolutions.
|
|
|
|
log := &mockArbitratorLog{
|
|
|
|
state: StateDefault,
|
|
|
|
newStates: make(chan ArbitratorState, 5),
|
|
|
|
failLog: true,
|
|
|
|
}
|
|
|
|
|
|
|
|
chanArb, resolved, err := createTestChannelArbitrator(log)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// It should start in StateDefault.
|
|
|
|
assertState(t, chanArb, StateDefault)
|
|
|
|
|
|
|
|
// Send a remote force close event.
|
|
|
|
commitSpend := &chainntnfs.SpendDetail{
|
|
|
|
SpenderTxHash: &chainhash.Hash{},
|
|
|
|
}
|
|
|
|
|
|
|
|
uniClose := &lnwallet.UnilateralCloseSummary{
|
|
|
|
SpendDetail: commitSpend,
|
|
|
|
HtlcResolutions: &lnwallet.HtlcResolutions{},
|
|
|
|
}
|
|
|
|
chanArb.cfg.ChainEvents.RemoteUnilateralClosure <- uniClose
|
|
|
|
|
|
|
|
// Since writing the resolutions fail, the arbitrator should not
|
|
|
|
// advance to the next state.
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
if log.state != StateDefault {
|
|
|
|
t.Fatalf("expected to stay in StateDefault")
|
|
|
|
}
|
|
|
|
chanArb.Stop()
|
|
|
|
|
|
|
|
// Create a new arbitrator with the same log.
|
|
|
|
chanArb, resolved, err = createTestChannelArbitrator(log)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Again, it should start up in the default state.
|
|
|
|
assertState(t, chanArb, StateDefault)
|
|
|
|
|
|
|
|
// Now we make the log succeed writing the resolutions, but fail when
|
|
|
|
// attempting to close the channel.
|
|
|
|
log.failLog = false
|
|
|
|
chanArb.cfg.MarkChannelClosed = func(*channeldb.ChannelCloseSummary) error {
|
|
|
|
return fmt.Errorf("intentional close error")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Send a new remote force close event.
|
|
|
|
chanArb.cfg.ChainEvents.RemoteUnilateralClosure <- uniClose
|
|
|
|
|
|
|
|
// Since closing the channel failed, the arbitrator should stay in the
|
|
|
|
// default state.
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
if log.state != StateDefault {
|
|
|
|
t.Fatalf("expected to stay in StateDefault")
|
|
|
|
}
|
|
|
|
chanArb.Stop()
|
|
|
|
|
|
|
|
// Create yet another arbitrator with the same log.
|
|
|
|
chanArb, resolved, err = createTestChannelArbitrator(log)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Starts out in StateDefault.
|
|
|
|
assertState(t, chanArb, StateDefault)
|
|
|
|
|
|
|
|
// Now make fetching the resolutions fail.
|
|
|
|
log.failFetch = fmt.Errorf("intentional fetch failure")
|
|
|
|
chanArb.cfg.ChainEvents.RemoteUnilateralClosure <- uniClose
|
|
|
|
|
|
|
|
// Since logging the resolutions and closing the channel now succeeds,
|
|
|
|
// it should advance to StateContractClosed.
|
|
|
|
assertStateTransitions(
|
|
|
|
t, log.newStates, StateContractClosed,
|
|
|
|
)
|
|
|
|
|
|
|
|
// It should not advance further, however, as fetching resolutions
|
|
|
|
// failed.
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
if log.state != StateContractClosed {
|
|
|
|
t.Fatalf("expected to stay in StateContractClosed")
|
|
|
|
}
|
|
|
|
chanArb.Stop()
|
|
|
|
|
|
|
|
// Create a new arbitrator, and now make fetching resolutions succeed.
|
|
|
|
log.failFetch = nil
|
|
|
|
chanArb, resolved, err = createTestChannelArbitrator(log)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
defer chanArb.Stop()
|
|
|
|
|
|
|
|
// Finally it should advance to StateFullyResolved.
|
|
|
|
assertStateTransitions(
|
|
|
|
t, log.newStates, StateFullyResolved,
|
|
|
|
)
|
|
|
|
|
|
|
|
// It should also mark the channel as resolved.
|
|
|
|
select {
|
|
|
|
case <-resolved:
|
|
|
|
// Expected.
|
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("contract was not resolved")
|
|
|
|
}
|
|
|
|
}
|
2018-08-21 13:21:16 +03:00
|
|
|
|
|
|
|
// TestChannelArbitratorCommitFailure tests that the channel arbitrator is able
|
|
|
|
// to recover from a failed CommitState call at restart.
|
|
|
|
func TestChannelArbitratorCommitFailure(t *testing.T) {
|
|
|
|
// Start out with a log that will fail committing to StateContractClosed.
|
|
|
|
log := &mockArbitratorLog{
|
|
|
|
state: StateDefault,
|
|
|
|
newStates: make(chan ArbitratorState, 5),
|
|
|
|
failCommit: true,
|
|
|
|
failCommitState: StateContractClosed,
|
|
|
|
}
|
|
|
|
|
|
|
|
chanArb, resolved, err := createTestChannelArbitrator(log)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// It should start in StateDefault.
|
|
|
|
assertState(t, chanArb, StateDefault)
|
|
|
|
|
|
|
|
closed := make(chan struct{})
|
|
|
|
chanArb.cfg.MarkChannelClosed = func(*channeldb.ChannelCloseSummary) error {
|
|
|
|
close(closed)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Send a remote force close event.
|
|
|
|
commitSpend := &chainntnfs.SpendDetail{
|
|
|
|
SpenderTxHash: &chainhash.Hash{},
|
|
|
|
}
|
|
|
|
|
|
|
|
uniClose := &lnwallet.UnilateralCloseSummary{
|
|
|
|
SpendDetail: commitSpend,
|
|
|
|
HtlcResolutions: &lnwallet.HtlcResolutions{},
|
|
|
|
}
|
|
|
|
chanArb.cfg.ChainEvents.RemoteUnilateralClosure <- uniClose
|
|
|
|
|
|
|
|
select {
|
|
|
|
case <-closed:
|
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("channel was not marked closed")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Since the channel was marked closed in the database, but the commit
|
|
|
|
// to the next state failed, the state should still be StateDefault.
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
if log.state != StateDefault {
|
|
|
|
t.Fatalf("expected to stay in StateDefault")
|
|
|
|
}
|
|
|
|
chanArb.Stop()
|
|
|
|
|
|
|
|
// Start the arbitrator again, with IsPendingClose reporting the
|
|
|
|
// channel closed in the database.
|
|
|
|
chanArb, resolved, err = createTestChannelArbitrator(log)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
log.failCommit = false
|
|
|
|
|
|
|
|
chanArb.cfg.IsPendingClose = true
|
|
|
|
chanArb.cfg.ClosingHeight = 100
|
|
|
|
chanArb.cfg.CloseType = channeldb.RemoteForceClose
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Since the channel is marked closed in the database, it should
|
|
|
|
// advance to StateContractClosed and StateFullyResolved.
|
|
|
|
assertStateTransitions(
|
|
|
|
t, log.newStates, StateContractClosed, StateFullyResolved,
|
|
|
|
)
|
|
|
|
|
|
|
|
// It should also mark the channel as resolved.
|
|
|
|
select {
|
|
|
|
case <-resolved:
|
|
|
|
// Expected.
|
|
|
|
case <-time.After(5 * time.Second):
|
|
|
|
t.Fatalf("contract was not resolved")
|
|
|
|
}
|
|
|
|
}
|
2018-08-21 13:55:19 +03:00
|
|
|
|
|
|
|
// TestChannelArbitratorEmptyResolutions makes sure that a channel that is
|
|
|
|
// pending close in the database, but haven't had any resolutions logged will
|
|
|
|
// not be marked resolved. This situation must be handled to avoid closing
|
|
|
|
// channels from earlier versions of the ChannelArbitrator, which didn't have a
|
|
|
|
// proper handoff from the ChainWatcher, and we could risk ending up in a state
|
|
|
|
// where the channel was closed in the DB, but the resolutions weren't properly
|
|
|
|
// written.
|
|
|
|
func TestChannelArbitratorEmptyResolutions(t *testing.T) {
|
|
|
|
// Start out with a log that will fail writing the set of resolutions.
|
|
|
|
log := &mockArbitratorLog{
|
|
|
|
state: StateDefault,
|
|
|
|
newStates: make(chan ArbitratorState, 5),
|
|
|
|
failFetch: errNoResolutions,
|
|
|
|
}
|
|
|
|
|
|
|
|
chanArb, _, err := createTestChannelArbitrator(log)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to create ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
chanArb.cfg.IsPendingClose = true
|
|
|
|
chanArb.cfg.ClosingHeight = 100
|
|
|
|
chanArb.cfg.CloseType = channeldb.RemoteForceClose
|
|
|
|
|
|
|
|
if err := chanArb.Start(); err != nil {
|
|
|
|
t.Fatalf("unable to start ChannelArbitrator: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// It should not advance its state beyond StateContractClosed, since
|
|
|
|
// fetching resolutions fails.
|
|
|
|
assertStateTransitions(
|
|
|
|
t, log.newStates, StateContractClosed,
|
|
|
|
)
|
|
|
|
|
|
|
|
// It should not advance further, however, as fetching resolutions
|
|
|
|
// failed.
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
if log.state != StateContractClosed {
|
|
|
|
t.Fatalf("expected to stay in StateContractClosed")
|
|
|
|
}
|
|
|
|
chanArb.Stop()
|
|
|
|
}
|