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

htlcswitch/iterator: extract error encrypter from ephemeral key

Browse Source
This commit is contained in:
Conner Fromknecht 2018-03-12 13:37:00 -07:00
parent dccab0a654
commit 3d487e3ca3
No known key found for this signature in database
GPG Key ID: 39DE78FBE6ACB0EF
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
htlcswitch/iterator.go
View File

@ -6,6 +6,7 @@ import (
"github.com/lightningnetwork/lightning-onion" "github.com/lightningnetwork/lightning-onion"
"github.com/lightningnetwork/lnd/lnwire" "github.com/lightningnetwork/lnd/lnwire"
"github.com/roasbeef/btcd/btcec"
) )
// NetworkHop indicates the blockchain network that is intended to be the next // NetworkHop indicates the blockchain network that is intended to be the next
@ -167,7 +168,7 @@ func (r *sphinxHopIterator) ForwardingInstructions() ForwardingInfo {
func (r *sphinxHopIterator) ExtractErrorEncrypter( func (r *sphinxHopIterator) ExtractErrorEncrypter(
extracter ErrorEncrypterExtracter) (ErrorEncrypter, lnwire.FailCode) { extracter ErrorEncrypterExtracter) (ErrorEncrypter, lnwire.FailCode) {
return extracter(r.ogPacket) return extracter(r.ogPacket.EphemeralKey)
} }
// OnionProcessor is responsible for keeping all sphinx dependent parts inside // OnionProcessor is responsible for keeping all sphinx dependent parts inside
@ -401,11 +402,12 @@ func (p *OnionProcessor) DecodeHopIterators(id []byte,
// ErrorEncrypter instance using the derived shared secret. In the case that en // ErrorEncrypter instance using the derived shared secret. In the case that en
// error occurs, a lnwire failure code detailing the parsing failure will be // error occurs, a lnwire failure code detailing the parsing failure will be
// returned. // returned.
func (p *OnionProcessor) ExtractErrorEncrypter(onionPkt *sphinx.OnionPacket) ( func (p *OnionProcessor) ExtractErrorEncrypter(ephemeralKey *btcec.PublicKey) (
ErrorEncrypter, lnwire.FailCode) { ErrorEncrypter, lnwire.FailCode) {
onionObfuscator, err := sphinx.NewOnionErrorEncrypter(p.router, onionObfuscator, err := sphinx.NewOnionErrorEncrypter(
onionPkt.EphemeralKey) p.router, ephemeralKey,
)
if err != nil { if err != nil {
switch err { switch err {
case sphinx.ErrInvalidOnionVersion: case sphinx.ErrInvalidOnionVersion:
@ -422,6 +424,6 @@ func (p *OnionProcessor) ExtractErrorEncrypter(onionPkt *sphinx.OnionPacket) (
return &SphinxErrorEncrypter{ return &SphinxErrorEncrypter{
OnionErrorEncrypter: onionObfuscator, OnionErrorEncrypter: onionObfuscator,
ogPacket: onionPkt, EphemeralKey: ephemeralKey,
}, lnwire.CodeNone }, lnwire.CodeNone
} }