mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-03-25 17:22:59 +01:00
3cc50ced55
We'll need to pack feature vectors for route blinding, so we pull the encoding/decoding out into separate functions (currently contained in ChannelType). Though it's more lines of code, we keep most of the ChannelType assertions so that we strictly enforce use of the alias.
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
package lnwire
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/lightningnetwork/lnd/tlv"
|
|
)
|
|
|
|
const (
|
|
// ChannelTypeRecordType is the type of the experimental record used
|
|
// to denote which channel type is being negotiated.
|
|
ChannelTypeRecordType tlv.Type = 1
|
|
)
|
|
|
|
// ChannelType represents a specific channel type as a set of feature bits that
|
|
// comprise it.
|
|
type ChannelType RawFeatureVector
|
|
|
|
// featureBitLen returns the length in bytes of the encoded feature bits.
|
|
func (c ChannelType) featureBitLen() uint64 {
|
|
fv := RawFeatureVector(c)
|
|
return fv.sizeFunc()
|
|
}
|
|
|
|
// Record returns a TLV record that can be used to encode/decode the channel
|
|
// type from a given TLV stream.
|
|
func (c *ChannelType) Record() tlv.Record {
|
|
return tlv.MakeDynamicRecord(
|
|
ChannelTypeRecordType, c, c.featureBitLen, channelTypeEncoder,
|
|
channelTypeDecoder,
|
|
)
|
|
}
|
|
|
|
// channelTypeEncoder is a custom TLV encoder for the ChannelType record.
|
|
func channelTypeEncoder(w io.Writer, val interface{}, buf *[8]byte) error {
|
|
if v, ok := val.(*ChannelType); ok {
|
|
fv := RawFeatureVector(*v)
|
|
return rawFeatureEncoder(w, &fv, buf)
|
|
}
|
|
|
|
return tlv.NewTypeForEncodingErr(val, "*lnwire.ChannelType")
|
|
}
|
|
|
|
// channelTypeDecoder is a custom TLV decoder for the ChannelType record.
|
|
func channelTypeDecoder(r io.Reader, val interface{}, buf *[8]byte,
|
|
l uint64) error {
|
|
|
|
if v, ok := val.(*ChannelType); ok {
|
|
fv := NewRawFeatureVector()
|
|
|
|
if err := rawFeatureDecoder(r, fv, buf, l); err != nil {
|
|
return err
|
|
}
|
|
|
|
*v = ChannelType(*fv)
|
|
return nil
|
|
}
|
|
|
|
return tlv.NewTypeForEncodingErr(val, "*lnwire.ChannelType")
|
|
}
|