akovalenko/lnd.xprv
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

watchtower/wtmock/client_db: adjust mock clientdb behavior

Browse Source 
In advance of the upcoming wtdb.ClientDB, we'll modify the behavior
of the mockdb to be more like the final bbolt backed one, and assert
that all or our tests are still passing.
This commit is contained in:
Conner Fromknecht 2019-05-23 20:48:23 -07:00
parent 2a904cb69f
commit 440ae7818a
No known key found for this signature in database
GPG Key ID: E7D737B67FA592C7
2 changed files with 25 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
watchtower/wtdb/client_session.go
View File

@ -39,6 +39,12 @@ var (
// created because session key index differs from the reserved key
// index.
ErrIncorrectKeyIndex = errors.New("incorrect key index")
// ErrClientSessionAlreadyExists signals an attempt to reinsert
// a client session that has already been created.
ErrClientSessionAlreadyExists = errors.New(
"client session already exists",
)
)
// ClientSession encapsulates a SessionInfo returned from a successful

23
watchtower/wtmock/client_db.go
View File

@ -65,7 +65,7 @@ func (m *ClientDB) CreateTower(lnAddr *lnwire.NetAddress) (*wtdb.Tower, error) {
m.towerIndex[towerPubKey] = towerID
m.towers[towerID] = tower
return tower, nil
return copyTower(tower), nil
}
// LoadTower retrieves a tower by its tower ID.
@ -74,7 +74,7 @@ func (m *ClientDB) LoadTower(towerID wtdb.TowerID) (*wtdb.Tower, error) {
defer m.mu.Unlock()
if tower, ok := m.towers[towerID]; ok {
return tower, nil
return copyTower(tower), nil
}
return nil, wtdb.ErrTowerNotFound
@ -106,6 +106,11 @@ func (m *ClientDB) CreateClientSession(session *wtdb.ClientSession) error {
m.mu.Lock()
defer m.mu.Unlock()
// Ensure that we aren't overwriting an existing session.
if _, ok := m.activeSessions[session.ID]; ok {
return wtdb.ErrClientSessionAlreadyExists
}
// Ensure that a session key index has been reserved for this tower.
keyIndex, ok := m.indexes[session.TowerID]
if !ok {
@ -151,11 +156,10 @@ func (m *ClientDB) NextSessionKeyIndex(towerID wtdb.TowerID) (uint32, error) {
return index, nil
}
m.nextIndex++
index := m.nextIndex
m.indexes[towerID] = index
m.nextIndex++
return index, nil
}
@ -286,3 +290,14 @@ func cloneBytes(b []byte) []byte {
return bb
}
func copyTower(tower *wtdb.Tower) *wtdb.Tower {
t := &wtdb.Tower{
ID: tower.ID,
IdentityKey: tower.IdentityKey,
Addresses: make([]net.Addr, len(tower.Addresses)),
}
copy(t.Addresses, tower.Addresses)
return t
}