• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

lightningnetwork / lnd / 13980275562

20 Mar 2025 10:06PM UTC coverage: 58.6% (-10.2%) from 68.789%
13980275562

Pull #9623

github

web-flow
Merge b9b960345 into 09b674508
Pull Request #9623: Size msg test msg

0 of 1518 new or added lines in 42 files covered. (0.0%)

26603 existing lines in 443 files now uncovered.

96807 of 165200 relevant lines covered (58.6%)

1.82 hits per line

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

80.98
/feature/manager.go
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.
100
func NewManager(cfg Config) (*Manager, error) {
3✔
101
        return newManager(cfg, defaultSetDesc)
3✔
102
}
3✔
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) {
3✔
108
        // First build the default feature vector for all known sets.
3✔
109
        fsets := make(map[Set]*lnwire.RawFeatureVector)
3✔
110
        for bit, sets := range desc {
6✔
111
                for set := range sets {
6✔
112
                        // Fetch the feature vector for this set, allocating a
3✔
113
                        // new one if it doesn't exist.
3✔
114
                        fv, ok := fsets[set]
3✔
115
                        if !ok {
6✔
116
                                fv = lnwire.NewRawFeatureVector()
3✔
117
                        }
3✔
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)
3✔
123
                        if err != nil {
3✔
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
3✔
130
                }
131
        }
132

133
        // Now, remove any features as directed by the config.
134
        configFeatures := make(map[Set]*lnwire.FeatureVector)
3✔
135
        for set, raw := range fsets {
6✔
136
                if cfg.NoTLVOnion {
6✔
137
                        raw.Unset(lnwire.TLVOnionPayloadOptional)
3✔
138
                        raw.Unset(lnwire.TLVOnionPayloadRequired)
3✔
139
                        raw.Unset(lnwire.PaymentAddrOptional)
3✔
140
                        raw.Unset(lnwire.PaymentAddrRequired)
3✔
141
                        raw.Unset(lnwire.MPPOptional)
3✔
142
                        raw.Unset(lnwire.MPPRequired)
3✔
143
                        raw.Unset(lnwire.RouteBlindingOptional)
3✔
144
                        raw.Unset(lnwire.RouteBlindingRequired)
3✔
145
                        raw.Unset(lnwire.Bolt11BlindedPathsOptional)
3✔
146
                        raw.Unset(lnwire.Bolt11BlindedPathsRequired)
3✔
147
                        raw.Unset(lnwire.AMPOptional)
3✔
148
                        raw.Unset(lnwire.AMPRequired)
3✔
149
                        raw.Unset(lnwire.KeysendOptional)
3✔
150
                        raw.Unset(lnwire.KeysendRequired)
3✔
151
                }
3✔
152
                if cfg.NoStaticRemoteKey {
6✔
153
                        raw.Unset(lnwire.StaticRemoteKeyOptional)
3✔
154
                        raw.Unset(lnwire.StaticRemoteKeyRequired)
3✔
155
                }
3✔
156
                if cfg.NoAnchors {
6✔
157
                        raw.Unset(lnwire.AnchorsZeroFeeHtlcTxOptional)
3✔
158
                        raw.Unset(lnwire.AnchorsZeroFeeHtlcTxRequired)
3✔
159

3✔
160
                        // If anchors are disabled, then we also need to
3✔
161
                        // disable all other features that depend on it as
3✔
162
                        // well, as otherwise we may create an invalid feature
3✔
163
                        // bit set.
3✔
164
                        for bit, depFeatures := range deps {
6✔
165
                                for depFeature := range depFeatures {
6✔
166
                                        switch {
3✔
167
                                        case depFeature == lnwire.AnchorsZeroFeeHtlcTxRequired:
×
168
                                                fallthrough
×
169
                                        case depFeature == lnwire.AnchorsZeroFeeHtlcTxOptional:
3✔
170
                                                raw.Unset(bit)
3✔
171
                                        }
172
                                }
173
                        }
174
                }
175
                if cfg.NoWumbo {
6✔
176
                        raw.Unset(lnwire.WumboChannelsOptional)
3✔
177
                        raw.Unset(lnwire.WumboChannelsRequired)
3✔
178
                }
3✔
179
                if cfg.NoScriptEnforcementLease {
6✔
180
                        raw.Unset(lnwire.ScriptEnforcedLeaseOptional)
3✔
181
                        raw.Unset(lnwire.ScriptEnforcedLeaseRequired)
3✔
182
                }
3✔
183
                if cfg.NoKeysend {
3✔
184
                        raw.Unset(lnwire.KeysendOptional)
×
185
                        raw.Unset(lnwire.KeysendRequired)
×
186
                }
×
187
                if cfg.NoOptionScidAlias {
6✔
188
                        raw.Unset(lnwire.ScidAliasOptional)
3✔
189
                        raw.Unset(lnwire.ScidAliasRequired)
3✔
190
                }
3✔
191
                if cfg.NoZeroConf {
6✔
192
                        raw.Unset(lnwire.ZeroConfOptional)
3✔
193
                        raw.Unset(lnwire.ZeroConfRequired)
3✔
194
                }
3✔
195
                if cfg.NoAnySegwit {
6✔
196
                        raw.Unset(lnwire.ShutdownAnySegwitOptional)
3✔
197
                        raw.Unset(lnwire.ShutdownAnySegwitRequired)
3✔
198
                }
3✔
199
                if cfg.NoTaprootChans {
6✔
200
                        raw.Unset(lnwire.SimpleTaprootChannelsOptionalStaging)
3✔
201
                        raw.Unset(lnwire.SimpleTaprootChannelsRequiredStaging)
3✔
202
                }
3✔
203
                if cfg.NoRouteBlinding {
6✔
204
                        raw.Unset(lnwire.RouteBlindingOptional)
3✔
205
                        raw.Unset(lnwire.RouteBlindingRequired)
3✔
206
                        raw.Unset(lnwire.Bolt11BlindedPathsOptional)
3✔
207
                        raw.Unset(lnwire.Bolt11BlindedPathsRequired)
3✔
208
                }
3✔
209
                if cfg.NoQuiescence {
3✔
210
                        raw.Unset(lnwire.QuiescenceOptional)
×
211
                }
×
212
                if cfg.NoTaprootOverlay {
6✔
213
                        raw.Unset(lnwire.SimpleTaprootOverlayChansOptional)
3✔
214
                        raw.Unset(lnwire.SimpleTaprootOverlayChansRequired)
3✔
215
                }
3✔
216
                if cfg.NoExperimentalEndorsement {
6✔
217
                        raw.Unset(lnwire.ExperimentalEndorsementOptional)
3✔
218
                        raw.Unset(lnwire.ExperimentalEndorsementRequired)
3✔
219
                }
3✔
220
                if cfg.NoRbfCoopClose {
6✔
221
                        raw.Unset(lnwire.RbfCoopCloseOptionalStaging)
3✔
222
                }
3✔
223

224
                for _, custom := range cfg.CustomFeatures[set] {
6✔
225
                        if custom > set.Maximum() {
3✔
226
                                return nil, fmt.Errorf("feature bit: %v "+
×
227
                                        "exceeds set: %v maximum: %v", custom,
×
228
                                        set, set.Maximum())
×
229
                        }
×
230

231
                        if raw.IsSet(custom) {
3✔
232
                                return nil, fmt.Errorf("feature bit: %v "+
×
233
                                        "already set", custom)
×
234
                        }
×
235

236
                        if err := raw.SafeSet(custom); err != nil {
3✔
237
                                return nil, fmt.Errorf("%w: could not set "+
×
238
                                        "feature: %d", err, custom)
×
239
                        }
×
240
                }
241

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

3✔
250
                // Ensure that all of our feature sets properly set any
3✔
251
                // dependent features.
3✔
252
                fv := lnwire.NewFeatureVector(raw, lnwire.Features)
3✔
253
                err := ValidateDeps(fv)
3✔
254
                if err != nil {
3✔
255
                        return nil, fmt.Errorf("invalid feature set %v: %w",
×
256
                                set, err)
×
257
                }
×
258
        }
259

260
        return &Manager{
3✔
261
                fsets:          fsets,
3✔
262
                configFeatures: configFeatures,
3✔
263
        }, nil
3✔
264
}
265

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

UNCOV
273
        return lnwire.NewRawFeatureVector()
×
274
}
275

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

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

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

3✔
292
        for set := range m.fsets {
6✔
293
                sets = append(sets, set)
3✔
294
        }
3✔
295

296
        return sets
3✔
297
}
298

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

3✔
307
        for set, newFeatures := range updates {
6✔
308
                if !set.valid() {
3✔
UNCOV
309
                        return fmt.Errorf("%w: set: %d", ErrUnknownSet, set)
×
UNCOV
310
                }
×
311

312
                if err := newFeatures.ValidatePairs(); err != nil {
3✔
UNCOV
313
                        return err
×
UNCOV
314
                }
×
315

316
                if err := m.Get(set).ValidateUpdate(
3✔
317
                        newFeatures, set.Maximum(),
3✔
318
                ); err != nil {
6✔
319
                        return err
3✔
320
                }
3✔
321

322
                // If any features were configured for this set, ensure that
323
                // they are still set in the new feature vector.
324
                if cfgFeat, haveCfgFeat := m.configFeatures[set]; haveCfgFeat {
6✔
325
                        for feature := range cfgFeat.Features() {
3✔
UNCOV
326
                                if !newFeatures.IsSet(feature) {
×
UNCOV
327
                                        return fmt.Errorf("%w: can't unset: "+
×
UNCOV
328
                                                "%d", ErrFeatureConfigured,
×
UNCOV
329
                                                feature)
×
UNCOV
330
                                }
×
331
                        }
332
                }
333

334
                fv := lnwire.NewFeatureVector(newFeatures, lnwire.Features)
3✔
335
                if err := ValidateDeps(fv); err != nil {
3✔
336
                        return err
×
337
                }
×
338
        }
339

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

347
        return nil
3✔
348
}
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2025 Coveralls, Inc