de3af9b0c0
In this commit, we modify the existing implementation of the Bandwidth() method on the default ChannelLink implementation to use much tighter accounting. Before this commit, there was a bug wherein if the link restarted with pending un-settled HTLC’s, and one of them was settled, then the bandwidth wouldn’t properly be updated to reflect this fact. To fix this, we’ve done away with the manual accounting and instead grab the current balances from two sources: the set of active HTLC’s within the overflow queue, and the report from the link itself which includes the pending HTLC’s and factors in the amount we’d need to (or not need to) pay in fees for each HTLC.
103 lines
3.1 KiB
Go
103 lines
3.1 KiB
Go
package htlcswitch
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
|
|
"github.com/lightningnetwork/lnd/lnwire"
|
|
)
|
|
|
|
// htlcPacket is a wrapper around htlc lnwire update, which adds additional
|
|
// information which is needed by this package.
|
|
type htlcPacket struct {
|
|
// destNode is the first-hop destination of a local created HTLC add
|
|
// message.
|
|
destNode [33]byte
|
|
|
|
// payHash is the payment hash of the HTLC which was modified by either
|
|
// a settle or fail action.
|
|
//
|
|
// NOTE: This fields is initialized only in settle and fail packets.
|
|
payHash [sha256.Size]byte
|
|
|
|
// dest is the destination of this packet identified by the short
|
|
// channel ID of the target link.
|
|
dest lnwire.ShortChannelID
|
|
|
|
// src is the source of this packet identified by the short channel ID
|
|
// of the target link.
|
|
src lnwire.ShortChannelID
|
|
|
|
// amount is the value of the HTLC that is being created or modified.
|
|
amount lnwire.MilliSatoshi
|
|
|
|
// htlc lnwire message type of which depends on switch request type.
|
|
htlc lnwire.Message
|
|
|
|
// obfuscator contains the necessary state to allow the switch to wrap
|
|
// any forwarded errors in an additional layer of encryption.
|
|
obfuscator ErrorEncrypter
|
|
|
|
// isObfuscated is set to true if an error occurs as soon as the switch
|
|
// forwards a packet to the link. If so, and this is an error packet,
|
|
// then this allows the switch to avoid doubly encrypting the error.
|
|
//
|
|
// TODO(andrew.shvv) revisit after refactoring the way of returning
|
|
// errors inside the htlcswitch packet.
|
|
isObfuscated bool
|
|
}
|
|
|
|
// newInitPacket creates htlc switch add packet which encapsulates the add htlc
|
|
// request and additional information for proper forwarding over htlc switch.
|
|
func newInitPacket(destNode [33]byte, htlc *lnwire.UpdateAddHTLC) *htlcPacket {
|
|
return &htlcPacket{
|
|
destNode: destNode,
|
|
amount: htlc.Amount,
|
|
htlc: htlc,
|
|
}
|
|
}
|
|
|
|
// newAddPacket creates htlc switch add packet which encapsulates the add htlc
|
|
// request and additional information for proper forwarding over htlc switch.
|
|
func newAddPacket(src, dest lnwire.ShortChannelID,
|
|
htlc *lnwire.UpdateAddHTLC, e ErrorEncrypter) *htlcPacket {
|
|
|
|
return &htlcPacket{
|
|
amount: htlc.Amount,
|
|
dest: dest,
|
|
src: src,
|
|
htlc: htlc,
|
|
obfuscator: e,
|
|
}
|
|
}
|
|
|
|
// newSettlePacket creates htlc switch ack/settle packet which encapsulates the
|
|
// settle htlc request which should be created and sent back by last hope in
|
|
// htlc path.
|
|
func newSettlePacket(src lnwire.ShortChannelID, htlc *lnwire.UpdateFufillHTLC,
|
|
payHash [sha256.Size]byte, amount lnwire.MilliSatoshi) *htlcPacket {
|
|
|
|
return &htlcPacket{
|
|
src: src,
|
|
payHash: payHash,
|
|
htlc: htlc,
|
|
amount: amount,
|
|
}
|
|
}
|
|
|
|
// newFailPacket creates htlc switch fail packet which encapsulates the fail
|
|
// htlc request which propagated back to the original hope who sent the htlc
|
|
// add request if something wrong happened on the path to the final
|
|
// destination.
|
|
func newFailPacket(src lnwire.ShortChannelID, htlc *lnwire.UpdateFailHTLC,
|
|
payHash [sha256.Size]byte, amount lnwire.MilliSatoshi,
|
|
isObfuscated bool) *htlcPacket {
|
|
|
|
return &htlcPacket{
|
|
src: src,
|
|
payHash: payHash,
|
|
htlc: htlc,
|
|
amount: amount,
|
|
isObfuscated: isObfuscated,
|
|
}
|
|
}
|