highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-09-07 19:30:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

htlcswitch+channel: add channel states synchronization

Browse Source 
In this commit BOLT№2 retranmission logic for the channel link have
been added. Now if channel link have been initialised with the
'SyncState' field than it will send the lnwire.ChannelReestablish
message and will be waiting for receiving the same message from remote
side. Exchange of this message allow both sides understand which
updates they should exchange with each other in order sync their
states.
This commit is contained in:
Andrey Samokhvalov
2017-07-09 02:30:20 +03:00
committed by Olaoluwa Osuntokun
parent bea9c0b52b
commit d70ffe93e4
8 changed files with 1133 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
htlcswitch/mock.go
View File

@@ -62,15 +62,21 @@ func newMockServer(name string, errChan chan error) *mockServer {
func (s *mockServer) Start() error {
if !atomic.CompareAndSwapInt32(&s.started, 0, 1) {
return nil
return errors.New("mock server already started")
}
s.htlcSwitch.Start()
if err := s.htlcSwitch.Start(); err != nil {
return err
}
s.wg.Add(1)
go func() {
defer s.wg.Done()
defer func() {
s.htlcSwitch.Stop()
}()
for {
select {
case msg := <-s.messages:
@@ -79,8 +85,8 @@ func (s *mockServer) Start() error {
for _, interceptor := range s.interceptorFuncs {
skip, err := interceptor(msg)
if err != nil {
s.errChan <- errors.Errorf("%v: error in the "+
"interceptor: %v", s.name, err)
s.fail(errors.Errorf("%v: error in the "+
"interceptor: %v", s.name, err))
return
}
shouldSkip = shouldSkip || skip
@@ -91,7 +97,8 @@ func (s *mockServer) Start() error {
}
if err := s.readHandler(msg); err != nil {
s.errChan <- errors.Errorf("%v server error: %v", s.name, err)
s.fail(err)
return
}
case <-s.quit:
return
@@ -102,6 +109,16 @@ func (s *mockServer) Start() error {
return nil
}
func (s *mockServer) fail(err error) {
go func() {
s.Stop()
}()
go func() {
s.errChan <- errors.Errorf("%v server error: %v", s.name, err)
}()
}
// mockHopIterator represents the test version of hop iterator which instead
// of encrypting the path in onion blob just stores the path as a list of hops.
type mockHopIterator struct {
@@ -266,6 +283,7 @@ func (s *mockServer) SendMessage(message lnwire.Message) error {
select {
case s.messages <- message:
case <-s.quit:
return errors.New("server is stopped")
}
return nil
@@ -290,6 +308,8 @@ func (s *mockServer) readHandler(message lnwire.Message) error {
case *lnwire.FundingLocked:
// Ignore
return nil
case *lnwire.ChannelReestablish:
targetChan = msg.ChanID
default:
return errors.New("unknown message type")
}
@@ -323,24 +343,22 @@ func (s *mockServer) PubKey() [33]byte {
func (s *mockServer) Disconnect(reason error) {
fmt.Printf("server %v disconnected due to %v\n", s.name, reason)
s.Stop()
s.errChan <- errors.Errorf("server %v was disconnected: %v", s.name,
reason)
s.fail(errors.Errorf("server %v was disconnected: %v", s.name, reason))
}
func (s *mockServer) WipeChannel(*lnwallet.LightningChannel) error {
return nil
}
func (s *mockServer) Stop() {
func (s *mockServer) Stop() error {
if !atomic.CompareAndSwapInt32(&s.shutdown, 0, 1) {
return
return nil
}
s.htlcSwitch.Stop()
close(s.quit)
s.wg.Wait()
return nil
}
func (s *mockServer) String() string {