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

htlcswitch/circuit_map: prune stray locally-initiated keystones

Browse Source
This commit is contained in:
Conner Fromknecht 2018-08-18 19:27:19 -07:00
parent c2055d4a9e
commit a213810563
No known key found for this signature in database
GPG Key ID: E7D737B67FA592C7
1 changed files with 39 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
htlcswitch/circuit_map.go
View File

@ -226,7 +226,9 @@ func (cm *circuitMap) initBuckets() error {
// restoreMemState loads the contents of the half circuit and full circuit // restoreMemState loads the contents of the half circuit and full circuit
// buckets from disk and reconstructs the in-memory representation of the // buckets from disk and reconstructs the in-memory representation of the
// circuit map. Afterwards, the state of the hash index is reconstructed using // circuit map. Afterwards, the state of the hash index is reconstructed using
// the recovered set of full circuits. // the recovered set of full circuits. This method will also remove any stray
// keystones, which are those that appear fully-opened, but have no pending
// circuit related to the intended incoming link.
func (cm *circuitMap) restoreMemState() error { func (cm *circuitMap) restoreMemState() error {
log.Infof("Restoring in-memory circuit state from disk") log.Infof("Restoring in-memory circuit state from disk")
@ -235,7 +237,7 @@ func (cm *circuitMap) restoreMemState() error {
pending = make(map[CircuitKey]*PaymentCircuit) pending = make(map[CircuitKey]*PaymentCircuit)
) )
if err := cm.cfg.DB.View(func(tx *bolt.Tx) error { if err := cm.cfg.DB.Update(func(tx *bolt.Tx) error {
// Restore any of the circuits persisted in the circuit bucket // Restore any of the circuits persisted in the circuit bucket
// back into memory. // back into memory.
circuitBkt := tx.Bucket(circuitAddKey) circuitBkt := tx.Bucket(circuitAddKey)
@ -264,6 +266,7 @@ func (cm *circuitMap) restoreMemState() error {
return ErrCorruptedCircuitMap return ErrCorruptedCircuitMap
} }
var strayKeystones []Keystone
if err := keystoneBkt.ForEach(func(k, v []byte) error { if err := keystoneBkt.ForEach(func(k, v []byte) error {
var ( var (
inKey CircuitKey inKey CircuitKey
@ -280,15 +283,45 @@ func (cm *circuitMap) restoreMemState() error {
// Retrieve the pending circuit, set its keystone, then // Retrieve the pending circuit, set its keystone, then
// add it to the opened map. // add it to the opened map.
circuit := pending[inKey] circuit, ok := pending[inKey]
if ok {
circuit.Outgoing = outKey circuit.Outgoing = outKey
opened[*outKey] = circuit opened[*outKey] = circuit
} else {
strayKeystones = append(strayKeystones, Keystone{
InKey: inKey,
OutKey: *outKey,
})
}
return nil return nil
}); err != nil { }); err != nil {
return err return err
} }
// If any stray keystones were found, we'll proceed to prune
// them from the circuit map's persistent storage. This may
// manifest on older nodes that had updated channels before
// their short channel id was set properly. We believe this
// issue has been fixed, though this will allow older nodes to
// recover without additional intervention.
for _, strayKeystone := range strayKeystones {
// As a precaution, we will only cleanup keystones
// related to locally-initiated payments. If a
// documented case of stray keystones emerges for
// forwarded payments, this check should be removed, but
// with extreme caution.
if strayKeystone.OutKey.ChanID != sourceHop {
continue
}
log.Infof("Removing stray keystone: %v", strayKeystone)
err := keystoneBkt.Delete(strayKeystone.OutKey.Bytes())
if err != nil {
return err
}
}
return nil return nil
}); err != nil { }); err != nil {