16683051882

Committed 01 Aug 2025 07:03PM UTC coverage: 54.949% (-12.1%) from 67.047%

Build # 16683051882

Build Type

Pull #9455

github

Committed by

web-flow

Commit Message

Merge 3f1f50be8 into 37523b6cb

Pull Request Pull Request #9455: discovery+lnwire: add support for DNS host name in NodeAnnouncement msg

Run Details

144 of 226 new or added lines in 7 files covered. (63.72%)

23852 existing lines in 290 files now uncovered.

108751 of 197912 relevant lines covered (54.95%)

22080.83 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

62.7

/feature/manager.go

package feature

import (
        "errors"
        "fmt"

        "github.com/lightningnetwork/lnd/lnwire"
)

var (
        // ErrUnknownSet is returned if a proposed feature vector contains a
        // set that is unknown to LND.
        ErrUnknownSet = errors.New("unknown feature bit set")

        // ErrFeatureConfigured is returned if an attempt is made to unset a
        // feature that was configured at startup.
        ErrFeatureConfigured = errors.New("can't unset configured feature")
)

// Config houses any runtime modifications to the default set descriptors. For
// our purposes, this typically means disabling certain features to test legacy
// protocol interoperability or functionality.
type Config struct {
        // NoTLVOnion unsets any optional or required TLVOnionPaylod bits from
        // all feature sets.
        NoTLVOnion bool

        // NoStaticRemoteKey unsets any optional or required StaticRemoteKey
        // bits from all feature sets.
        NoStaticRemoteKey bool

        // NoAnchors unsets any bits signaling support for anchor outputs.
        NoAnchors bool

        // NoWumbo unsets any bits signalling support for wumbo channels.
        NoWumbo bool

        // NoTaprootChans unsets any bits signaling support for taproot
        // channels.
        NoTaprootChans bool

        // NoScriptEnforcementLease unsets any bits signaling support for script
        // enforced leases.
        NoScriptEnforcementLease bool

        // NoKeysend unsets any bits signaling support for accepting keysend
        // payments.
        NoKeysend bool

        // NoOptionScidAlias unsets any bits signalling support for
        // option_scid_alias. This also implicitly disables zero-conf channels.
        NoOptionScidAlias bool

        // NoZeroConf unsets any bits signalling support for zero-conf
        // channels. This should be used instead of NoOptionScidAlias to still
        // keep option-scid-alias support.
        NoZeroConf bool

        // NoAnySegwit unsets any bits that signal support for using other
        // segwit witness versions for co-op closes.
        NoAnySegwit bool

        // NoRouteBlinding unsets route blinding feature bits.
        NoRouteBlinding bool

        // NoQuiescence unsets quiescence feature bits.
        NoQuiescence bool

        // NoTaprootOverlay unsets the taproot overlay channel feature bits.
        NoTaprootOverlay bool

        // NoExperimentalEndorsement unsets any bits that signal support for
        // forwarding experimental endorsement.
        NoExperimentalEndorsement bool

        // NoRbfCoopClose unsets any bits that signal support for using RBF for
        // coop close.
        NoRbfCoopClose bool

        // CustomFeatures is a set of custom features to advertise in each
        // set.
        CustomFeatures map[Set][]lnwire.FeatureBit
}

// Manager is responsible for generating feature vectors for different requested
// feature sets.
type Manager struct {
        // fsets is a static map of feature set to raw feature vectors. Requests
        // are fulfilled by cloning these internal feature vectors.
        fsets map[Set]*lnwire.RawFeatureVector

        // configFeatures is a set of custom features that were "hard set" in
        // lnd's config that cannot be updated at runtime (as is the case with
        // our "standard" features that are defined in LND).
        configFeatures map[Set]*lnwire.FeatureVector
}

// NewManager creates a new feature Manager, applying any custom modifications
// to its feature sets before returning.
func NewManager(cfg Config) (*Manager, error) {
        return newManager(cfg, defaultSetDesc)
}

// newManager creates a new feature Manager, applying any custom modifications
// to its feature sets before returning. This method accepts the setDesc as its
// own parameter so that it can be unit tested.
func newManager(cfg Config, desc setDesc) (*Manager, error) {
        // First build the default feature vector for all known sets.
        fsets := make(map[Set]*lnwire.RawFeatureVector)
        for bit, sets := range desc {
                for set := range sets {
                        // Fetch the feature vector for this set, allocating a
                        // new one if it doesn't exist.
                        fv, ok := fsets[set]
                        if !ok {
                                fv = lnwire.NewRawFeatureVector()
                        }

                        // Set the configured bit on the feature vector,
                        // ensuring that we don't set two feature bits for the
                        // same pair.
                        err := fv.SafeSet(bit)
                        if err != nil {
                                return nil, fmt.Errorf("unable to set "+
                                        "%v in %v: %v", bit, set, err)
                        }

                        // Write the updated feature vector under its set.
                        fsets[set] = fv
                }
        }

        // Now, remove any features as directed by the config.
        configFeatures := make(map[Set]*lnwire.FeatureVector)
        for set, raw := range fsets {
                if cfg.NoTLVOnion {
                        raw.Unset(lnwire.TLVOnionPayloadOptional)
                        raw.Unset(lnwire.TLVOnionPayloadRequired)
                        raw.Unset(lnwire.PaymentAddrOptional)
                        raw.Unset(lnwire.PaymentAddrRequired)
                        raw.Unset(lnwire.MPPOptional)
                        raw.Unset(lnwire.MPPRequired)
                        raw.Unset(lnwire.RouteBlindingOptional)
                        raw.Unset(lnwire.RouteBlindingRequired)
                        raw.Unset(lnwire.Bolt11BlindedPathsOptional)
                        raw.Unset(lnwire.Bolt11BlindedPathsRequired)
                        raw.Unset(lnwire.AMPOptional)
                        raw.Unset(lnwire.AMPRequired)
                        raw.Unset(lnwire.KeysendOptional)
                        raw.Unset(lnwire.KeysendRequired)
                }
                if cfg.NoStaticRemoteKey {
                        raw.Unset(lnwire.StaticRemoteKeyOptional)
                        raw.Unset(lnwire.StaticRemoteKeyRequired)
                }
                if cfg.NoAnchors {
                        raw.Unset(lnwire.AnchorsZeroFeeHtlcTxOptional)
                        raw.Unset(lnwire.AnchorsZeroFeeHtlcTxRequired)

                        // If anchors are disabled, then we also need to
                        // disable all other features that depend on it as
                        // well, as otherwise we may create an invalid feature
                        // bit set.
                        for bit, depFeatures := range deps {
                                for depFeature := range depFeatures {
                                        switch {
                                        case depFeature == lnwire.AnchorsZeroFeeHtlcTxRequired:
                                                fallthrough
                                        case depFeature == lnwire.AnchorsZeroFeeHtlcTxOptional:
                                                raw.Unset(bit)
                                        }
                                }
                        }
                }
                if cfg.NoWumbo {
                        raw.Unset(lnwire.WumboChannelsOptional)
                        raw.Unset(lnwire.WumboChannelsRequired)
                }
                if cfg.NoScriptEnforcementLease {
                        raw.Unset(lnwire.ScriptEnforcedLeaseOptional)
                        raw.Unset(lnwire.ScriptEnforcedLeaseRequired)
                }
                if cfg.NoKeysend {
                        raw.Unset(lnwire.KeysendOptional)
                        raw.Unset(lnwire.KeysendRequired)
                }
                if cfg.NoOptionScidAlias {
                        raw.Unset(lnwire.ScidAliasOptional)
                        raw.Unset(lnwire.ScidAliasRequired)
                }
                if cfg.NoZeroConf {
                        raw.Unset(lnwire.ZeroConfOptional)
                        raw.Unset(lnwire.ZeroConfRequired)
                }
                if cfg.NoAnySegwit {
                        raw.Unset(lnwire.ShutdownAnySegwitOptional)
                        raw.Unset(lnwire.ShutdownAnySegwitRequired)
                }
                if cfg.NoTaprootChans {
                        raw.Unset(lnwire.SimpleTaprootChannelsOptionalStaging)
                        raw.Unset(lnwire.SimpleTaprootChannelsRequiredStaging)
                }
                if cfg.NoRouteBlinding {
                        raw.Unset(lnwire.RouteBlindingOptional)
                        raw.Unset(lnwire.RouteBlindingRequired)
                        raw.Unset(lnwire.Bolt11BlindedPathsOptional)
                        raw.Unset(lnwire.Bolt11BlindedPathsRequired)
                }
                if cfg.NoQuiescence {
                        raw.Unset(lnwire.QuiescenceOptional)
                }
                if cfg.NoTaprootOverlay {
                        raw.Unset(lnwire.SimpleTaprootOverlayChansOptional)
                        raw.Unset(lnwire.SimpleTaprootOverlayChansRequired)
                }
                if cfg.NoExperimentalEndorsement {
                        raw.Unset(lnwire.ExperimentalEndorsementOptional)
                        raw.Unset(lnwire.ExperimentalEndorsementRequired)
                }
                if cfg.NoRbfCoopClose {
                        raw.Unset(lnwire.RbfCoopCloseOptionalStaging)
                        raw.Unset(lnwire.RbfCoopCloseOptional)
                }

                for _, custom := range cfg.CustomFeatures[set] {
                        if custom > set.Maximum() {
                                return nil, fmt.Errorf("feature bit: %v "+
                                        "exceeds set: %v maximum: %v", custom,
                                        set, set.Maximum())
                        }

                        if raw.IsSet(custom) {
                                return nil, fmt.Errorf("feature bit: %v "+
                                        "already set", custom)
                        }

                        if err := raw.SafeSet(custom); err != nil {
                                return nil, fmt.Errorf("%w: could not set "+
                                        "feature: %d", err, custom)
                        }
                }

                // Track custom features separately so that we can check that
                // they aren't unset in subsequent updates. If there is no
                // entry for the set, the vector will just be empty.
                configFeatures[set] = lnwire.NewFeatureVector(
                        lnwire.NewRawFeatureVector(cfg.CustomFeatures[set]...),
                        lnwire.Features,
                )

                // Ensure that all of our feature sets properly set any
                // dependent features.
                fv := lnwire.NewFeatureVector(raw, lnwire.Features)
                err := ValidateDeps(fv)
                if err != nil {
                        return nil, fmt.Errorf("invalid feature set %v: %w",
                                set, err)
                }
        }

        return &Manager{
                fsets:          fsets,
                configFeatures: configFeatures,
        }, nil
}

// GetRaw returns a raw feature vector for the passed set. If no set is known,
// an empty raw feature vector is returned.
func (m *Manager) GetRaw(set Set) *lnwire.RawFeatureVector {
        if fv, ok := m.fsets[set]; ok {
                return fv.Clone()
        }

        return lnwire.NewRawFeatureVector()
}

// setRaw sets a new raw feature vector for the given set.
func (m *Manager) setRaw(set Set, raw *lnwire.RawFeatureVector) {
        m.fsets[set] = raw
}

// Get returns a feature vector for the passed set. If no set is known, an empty
// feature vector is returned.
func (m *Manager) Get(set Set) *lnwire.FeatureVector {
        raw := m.GetRaw(set)
        return lnwire.NewFeatureVector(raw, lnwire.Features)
}

// ListSets returns a list of the feature sets that our node supports.
func (m *Manager) ListSets() []Set {
        var sets []Set

        for set := range m.fsets {
                sets = append(sets, set)
        }

        return sets
}

// UpdateFeatureSets accepts a map of new feature vectors for each of the
// manager's known sets, validates that the update can be applied and modifies
// the feature manager's internal state. If a set is not included in the update
// map, it is left unchanged. The feature vectors provided are expected to
// include the current set of features, updated with desired bits added/removed.
func (m *Manager) UpdateFeatureSets(
        updates map[Set]*lnwire.RawFeatureVector) error {

        for set, newFeatures := range updates {
                if !set.valid() {
                        return fmt.Errorf("%w: set: %d", ErrUnknownSet, set)
                }

                if err := newFeatures.ValidatePairs(); err != nil {
                        return err
                }

                if err := m.Get(set).ValidateUpdate(
                        newFeatures, set.Maximum(),
                ); err != nil {
                        return err
                }

                // If any features were configured for this set, ensure that
                // they are still set in the new feature vector.
                if cfgFeat, haveCfgFeat := m.configFeatures[set]; haveCfgFeat {
                        for feature := range cfgFeat.Features() {
                                if !newFeatures.IsSet(feature) {
                                        return fmt.Errorf("%w: can't unset: "+
                                                "%d", ErrFeatureConfigured,
                                                feature)
                                }
                        }
                }

                fv := lnwire.NewFeatureVector(newFeatures, lnwire.Features)
                if err := ValidateDeps(fv); err != nil {
                        return err
                }
        }

        // Only update the current feature sets once every proposed set has
        // passed validation so that we don't partially update any sets then
        // fail out on a later set's validation.
        for set, features := range updates {
                m.setRaw(set, features.Clone())
        }

        return nil
}

1	package feature
2
3	import (
4	"errors"
5	"fmt"
6
7	"github.com/lightningnetwork/lnd/lnwire"
8	)
9
10	var (
11	// ErrUnknownSet is returned if a proposed feature vector contains a
12	// set that is unknown to LND.
13	ErrUnknownSet = errors.New("unknown feature bit set")
14
15	// ErrFeatureConfigured is returned if an attempt is made to unset a
16	// feature that was configured at startup.
17	ErrFeatureConfigured = errors.New("can't unset configured feature")
18	)
19
20	// Config houses any runtime modifications to the default set descriptors. For
21	// our purposes, this typically means disabling certain features to test legacy
22	// protocol interoperability or functionality.
23	type Config struct {
24	// NoTLVOnion unsets any optional or required TLVOnionPaylod bits from
25	// all feature sets.
26	NoTLVOnion bool
27
28	// NoStaticRemoteKey unsets any optional or required StaticRemoteKey
29	// bits from all feature sets.
30	NoStaticRemoteKey bool
31
32	// NoAnchors unsets any bits signaling support for anchor outputs.
33	NoAnchors bool
34
35	// NoWumbo unsets any bits signalling support for wumbo channels.
36	NoWumbo bool
37
38	// NoTaprootChans unsets any bits signaling support for taproot
39	// channels.
40	NoTaprootChans bool
41
42	// NoScriptEnforcementLease unsets any bits signaling support for script
43	// enforced leases.
44	NoScriptEnforcementLease bool
45
46	// NoKeysend unsets any bits signaling support for accepting keysend
47	// payments.
48	NoKeysend bool
49
50	// NoOptionScidAlias unsets any bits signalling support for
51	// option_scid_alias. This also implicitly disables zero-conf channels.
52	NoOptionScidAlias bool
53
54	// NoZeroConf unsets any bits signalling support for zero-conf
55	// channels. This should be used instead of NoOptionScidAlias to still
56	// keep option-scid-alias support.
57	NoZeroConf bool
58
59	// NoAnySegwit unsets any bits that signal support for using other
60	// segwit witness versions for co-op closes.
61	NoAnySegwit bool
62
63	// NoRouteBlinding unsets route blinding feature bits.
64	NoRouteBlinding bool
65
66	// NoQuiescence unsets quiescence feature bits.
67	NoQuiescence bool
68
69	// NoTaprootOverlay unsets the taproot overlay channel feature bits.
70	NoTaprootOverlay bool
71
72	// NoExperimentalEndorsement unsets any bits that signal support for
73	// forwarding experimental endorsement.
74	NoExperimentalEndorsement bool
75
76	// NoRbfCoopClose unsets any bits that signal support for using RBF for
77	// coop close.
78	NoRbfCoopClose bool
79
80	// CustomFeatures is a set of custom features to advertise in each
81	// set.
82	CustomFeatures map[Set][]lnwire.FeatureBit
83	}
84
85	// Manager is responsible for generating feature vectors for different requested
86	// feature sets.
87	type Manager struct {
88	// fsets is a static map of feature set to raw feature vectors. Requests
89	// are fulfilled by cloning these internal feature vectors.
90	fsets map[Set]*lnwire.RawFeatureVector
91
92	// configFeatures is a set of custom features that were "hard set" in
93	// lnd's config that cannot be updated at runtime (as is the case with
94	// our "standard" features that are defined in LND).
95	configFeatures map[Set]*lnwire.FeatureVector
96	}
97
98	// NewManager creates a new feature Manager, applying any custom modifications
99	// to its feature sets before returning.
UNCOV 100	func NewManager(cfg Config) (*Manager, error) {	×
UNCOV 101	return newManager(cfg, defaultSetDesc)	×
UNCOV 102	}	×
103
104	// newManager creates a new feature Manager, applying any custom modifications
105	// to its feature sets before returning. This method accepts the setDesc as its
106	// own parameter so that it can be unit tested.
107	func newManager(cfg Config, desc setDesc) (*Manager, error) {	16✔
108	// First build the default feature vector for all known sets.	16✔
109	fsets := make(map[Set]*lnwire.RawFeatureVector)	16✔
110	for bit, sets := range desc {	53✔
111	for set := range sets {	95✔
112	// Fetch the feature vector for this set, allocating a	58✔
113	// new one if it doesn't exist.	58✔
114	fv, ok := fsets[set]	58✔
115	if !ok {	90✔
116	fv = lnwire.NewRawFeatureVector()	32✔
117	}	32✔
118
119	// Set the configured bit on the feature vector,
120	// ensuring that we don't set two feature bits for the
121	// same pair.
122	err := fv.SafeSet(bit)	58✔
123	if err != nil {	58✔
124	return nil, fmt.Errorf("unable to set "+	×
125	"%v in %v: %v", bit, set, err)	×
126	}	×
127
128	// Write the updated feature vector under its set.
129	fsets[set] = fv	58✔
130	}
131	}
132
133	// Now, remove any features as directed by the config.
134	configFeatures := make(map[Set]*lnwire.FeatureVector)	16✔
135	for set, raw := range fsets {	48✔
136	if cfg.NoTLVOnion {	36✔
137	raw.Unset(lnwire.TLVOnionPayloadOptional)	4✔
138	raw.Unset(lnwire.TLVOnionPayloadRequired)	4✔
139	raw.Unset(lnwire.PaymentAddrOptional)	4✔
140	raw.Unset(lnwire.PaymentAddrRequired)	4✔
141	raw.Unset(lnwire.MPPOptional)	4✔
142	raw.Unset(lnwire.MPPRequired)	4✔
143	raw.Unset(lnwire.RouteBlindingOptional)	4✔
144	raw.Unset(lnwire.RouteBlindingRequired)	4✔
145	raw.Unset(lnwire.Bolt11BlindedPathsOptional)	4✔
146	raw.Unset(lnwire.Bolt11BlindedPathsRequired)	4✔
147	raw.Unset(lnwire.AMPOptional)	4✔
148	raw.Unset(lnwire.AMPRequired)	4✔
149	raw.Unset(lnwire.KeysendOptional)	4✔
150	raw.Unset(lnwire.KeysendRequired)	4✔
151	}	4✔
152	if cfg.NoStaticRemoteKey {	36✔
153	raw.Unset(lnwire.StaticRemoteKeyOptional)	4✔
154	raw.Unset(lnwire.StaticRemoteKeyRequired)	4✔
155	}	4✔
156	if cfg.NoAnchors {	34✔
157	raw.Unset(lnwire.AnchorsZeroFeeHtlcTxOptional)	2✔
158	raw.Unset(lnwire.AnchorsZeroFeeHtlcTxRequired)	2✔
159		2✔
160	// If anchors are disabled, then we also need to	2✔
161	// disable all other features that depend on it as	2✔
162	// well, as otherwise we may create an invalid feature	2✔
163	// bit set.	2✔
164	for bit, depFeatures := range deps {	28✔
165	for depFeature := range depFeatures {	58✔
166	switch {	32✔
167	case depFeature == lnwire.AnchorsZeroFeeHtlcTxRequired:	×
168	fallthrough	×
169	case depFeature == lnwire.AnchorsZeroFeeHtlcTxOptional:	4✔
170	raw.Unset(bit)	4✔
171	}
172	}
173	}
174	}
175	if cfg.NoWumbo {	32✔
UNCOV 176	raw.Unset(lnwire.WumboChannelsOptional)	×
UNCOV 177	raw.Unset(lnwire.WumboChannelsRequired)	×
UNCOV 178	}	×
179	if cfg.NoScriptEnforcementLease {	32✔
UNCOV 180	raw.Unset(lnwire.ScriptEnforcedLeaseOptional)	×
UNCOV 181	raw.Unset(lnwire.ScriptEnforcedLeaseRequired)	×
UNCOV 182	}	×
183	if cfg.NoKeysend {	32✔
184	raw.Unset(lnwire.KeysendOptional)	×
185	raw.Unset(lnwire.KeysendRequired)	×
186	}	×
187	if cfg.NoOptionScidAlias {	32✔
UNCOV 188	raw.Unset(lnwire.ScidAliasOptional)	×
UNCOV 189	raw.Unset(lnwire.ScidAliasRequired)	×
UNCOV 190	}	×
191	if cfg.NoZeroConf {	32✔
UNCOV 192	raw.Unset(lnwire.ZeroConfOptional)	×
UNCOV 193	raw.Unset(lnwire.ZeroConfRequired)	×
UNCOV 194	}	×
195	if cfg.NoAnySegwit {	32✔
UNCOV 196	raw.Unset(lnwire.ShutdownAnySegwitOptional)	×
UNCOV 197	raw.Unset(lnwire.ShutdownAnySegwitRequired)	×
UNCOV 198	}	×
199	if cfg.NoTaprootChans {	32✔
UNCOV 200	raw.Unset(lnwire.SimpleTaprootChannelsOptionalStaging)	×
UNCOV 201	raw.Unset(lnwire.SimpleTaprootChannelsRequiredStaging)	×
UNCOV 202	}	×
203	if cfg.NoRouteBlinding {	32✔
UNCOV 204	raw.Unset(lnwire.RouteBlindingOptional)	×
UNCOV 205	raw.Unset(lnwire.RouteBlindingRequired)	×
UNCOV 206	raw.Unset(lnwire.Bolt11BlindedPathsOptional)	×
UNCOV 207	raw.Unset(lnwire.Bolt11BlindedPathsRequired)	×
UNCOV 208	}	×
209	if cfg.NoQuiescence {	32✔
210	raw.Unset(lnwire.QuiescenceOptional)	×
211	}	×
212	if cfg.NoTaprootOverlay {	32✔
UNCOV 213	raw.Unset(lnwire.SimpleTaprootOverlayChansOptional)	×
UNCOV 214	raw.Unset(lnwire.SimpleTaprootOverlayChansRequired)	×
UNCOV 215	}	×
216	if cfg.NoExperimentalEndorsement {	32✔
UNCOV 217	raw.Unset(lnwire.ExperimentalEndorsementOptional)	×
UNCOV 218	raw.Unset(lnwire.ExperimentalEndorsementRequired)	×
UNCOV 219	}	×
220	if cfg.NoRbfCoopClose {	32✔
UNCOV 221	raw.Unset(lnwire.RbfCoopCloseOptionalStaging)	×
UNCOV 222	raw.Unset(lnwire.RbfCoopCloseOptional)	×
UNCOV 223	}	×
224
225	for _, custom := range cfg.CustomFeatures[set] {	34✔
226	if custom > set.Maximum() {	2✔
227	return nil, fmt.Errorf("feature bit: %v "+	×
228	"exceeds set: %v maximum: %v", custom,	×
229	set, set.Maximum())	×
230	}	×
231
232	if raw.IsSet(custom) {	2✔
233	return nil, fmt.Errorf("feature bit: %v "+	×
234	"already set", custom)	×
235	}	×
236
237	if err := raw.SafeSet(custom); err != nil {	2✔
238	return nil, fmt.Errorf("%w: could not set "+	×
239	"feature: %d", err, custom)	×
240	}	×
241	}
242
243	// Track custom features separately so that we can check that
244	// they aren't unset in subsequent updates. If there is no
245	// entry for the set, the vector will just be empty.
246	configFeatures[set] = lnwire.NewFeatureVector(	32✔
247	lnwire.NewRawFeatureVector(cfg.CustomFeatures[set]...),	32✔
248	lnwire.Features,	32✔
249	)	32✔
250		32✔
251	// Ensure that all of our feature sets properly set any	32✔
252	// dependent features.	32✔
253	fv := lnwire.NewFeatureVector(raw, lnwire.Features)	32✔
254	err := ValidateDeps(fv)	32✔
255	if err != nil {	32✔
256	return nil, fmt.Errorf("invalid feature set %v: %w",	×
257	set, err)	×
258	}	×
259	}
260
261	return &Manager{	16✔
262	fsets: fsets,	16✔
263	configFeatures: configFeatures,	16✔
264	}, nil	16✔
265	}
266
267	// GetRaw returns a raw feature vector for the passed set. If no set is known,
268	// an empty raw feature vector is returned.
269	func (m Manager) GetRaw(set Set) lnwire.RawFeatureVector {	66✔
270	if fv, ok := m.fsets[set]; ok {	110✔
271	return fv.Clone()	44✔
272	}	44✔
273
274	return lnwire.NewRawFeatureVector()	22✔
275	}
276
277	// setRaw sets a new raw feature vector for the given set.
278	func (m Manager) setRaw(set Set, raw lnwire.RawFeatureVector) {	6✔
279	m.fsets[set] = raw	6✔
280	}	6✔
281
282	// Get returns a feature vector for the passed set. If no set is known, an empty
283	// feature vector is returned.
284	func (m Manager) Get(set Set) lnwire.FeatureVector {	32✔
285	raw := m.GetRaw(set)	32✔
286	return lnwire.NewFeatureVector(raw, lnwire.Features)	32✔
287	}	32✔
288
289	// ListSets returns a list of the feature sets that our node supports.
UNCOV 290	func (m *Manager) ListSets() []Set {	×
UNCOV 291	var sets []Set	×
UNCOV 292		×
UNCOV 293	for set := range m.fsets {	×
UNCOV 294	sets = append(sets, set)	×
UNCOV 295	}	×
296
UNCOV 297	return sets	×
298	}
299
300	// UpdateFeatureSets accepts a map of new feature vectors for each of the
301	// manager's known sets, validates that the update can be applied and modifies
302	// the feature manager's internal state. If a set is not included in the update
303	// map, it is left unchanged. The feature vectors provided are expected to
304	// include the current set of features, updated with desired bits added/removed.
305	func (m *Manager) UpdateFeatureSets(
306	updates map[Set]*lnwire.RawFeatureVector) error {	7✔
307		7✔
308	for set, newFeatures := range updates {	17✔
309	if !set.valid() {	11✔
310	return fmt.Errorf("%w: set: %d", ErrUnknownSet, set)	1✔
311	}	1✔
312
313	if err := newFeatures.ValidatePairs(); err != nil {	11✔
314	return err	2✔
315	}	2✔
316
317	if err := m.Get(set).ValidateUpdate(	7✔
318	newFeatures, set.Maximum(),	7✔
319	); err != nil {	7✔
UNCOV 320	return err	×
UNCOV 321	}	×
322
323	// If any features were configured for this set, ensure that
324	// they are still set in the new feature vector.
325	if cfgFeat, haveCfgFeat := m.configFeatures[set]; haveCfgFeat {	12✔
326	for feature := range cfgFeat.Features() {	6✔
327	if !newFeatures.IsSet(feature) {	2✔
328	return fmt.Errorf("%w: can't unset: "+	1✔
329	"%d", ErrFeatureConfigured,	1✔
330	feature)	1✔
331	}	1✔
332	}
333	}
334
335	fv := lnwire.NewFeatureVector(newFeatures, lnwire.Features)	6✔
336	if err := ValidateDeps(fv); err != nil {	6✔
337	return err	×
338	}	×
339	}
340
341	// Only update the current feature sets once every proposed set has
342	// passed validation so that we don't partially update any sets then
343	// fail out on a later set's validation.
344	for set, features := range updates {	9✔
345	m.setRaw(set, features.Clone())	6✔
346	}	6✔
347
348	return nil	3✔
349	}

lightningnetwork / lnd / 16683051882

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous