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

lightningnetwork / lnd / 11170835610

03 Oct 2024 10:41PM UTC coverage: 49.188% (-9.6%) from 58.738%
11170835610

push

github

web-flow
Merge pull request #9154 from ziggie1984/master

multi: bump btcd version.

3 of 6 new or added lines in 6 files covered. (50.0%)

26110 existing lines in 428 files now uncovered.

97359 of 197934 relevant lines covered (49.19%)

1.04 hits per line

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

55.83
/zpay32/encode.go
1
package zpay32
2

3
import (
4
        "bytes"
5
        "encoding/binary"
6
        "fmt"
7

8
        "github.com/btcsuite/btcd/btcutil"
9
        "github.com/btcsuite/btcd/btcutil/bech32"
10
        "github.com/btcsuite/btcd/chaincfg"
11
        "github.com/btcsuite/btcd/chaincfg/chainhash"
12
        "github.com/lightningnetwork/lnd/fn"
13
        "github.com/lightningnetwork/lnd/lnwire"
14
)
15

16
// Encode takes the given MessageSigner and returns a string encoding this
17
// invoice signed by the node key of the signer.
18
func (invoice *Invoice) Encode(signer MessageSigner) (string, error) {
2✔
19
        // First check that this invoice is valid before starting the encoding.
2✔
20
        if err := validateInvoice(invoice); err != nil {
2✔
UNCOV
21
                return "", err
×
UNCOV
22
        }
×
23

24
        // The buffer will encoded the invoice data using 5-bit groups (base32).
25
        var bufferBase32 bytes.Buffer
2✔
26

2✔
27
        // The timestamp will be encoded using 35 bits, in base32.
2✔
28
        timestampBase32 := uint64ToBase32(uint64(invoice.Timestamp.Unix()))
2✔
29

2✔
30
        // The timestamp must be exactly 35 bits, which means 7 groups. If it
2✔
31
        // can fit into fewer groups we add leading zero groups, if it is too
2✔
32
        // big we fail early, as there is not possible to encode it.
2✔
33
        if len(timestampBase32) > timestampBase32Len {
2✔
34
                return "", fmt.Errorf("timestamp too big: %d",
×
35
                        invoice.Timestamp.Unix())
×
36
        }
×
37

38
        // Add zero bytes to the first timestampBase32Len-len(timestampBase32)
39
        // groups, then add the non-zero groups.
40
        zeroes := make([]byte, timestampBase32Len-len(timestampBase32))
2✔
41
        _, err := bufferBase32.Write(zeroes)
2✔
42
        if err != nil {
2✔
43
                return "", fmt.Errorf("unable to write to buffer: %w", err)
×
44
        }
×
45
        _, err = bufferBase32.Write(timestampBase32)
2✔
46
        if err != nil {
2✔
47
                return "", fmt.Errorf("unable to write to buffer: %w", err)
×
48
        }
×
49

50
        // We now write the tagged fields to the buffer, which will fill the
51
        // rest of the data part before the signature.
52
        if err := writeTaggedFields(&bufferBase32, invoice); err != nil {
2✔
53
                return "", err
×
54
        }
×
55

56
        // The human-readable part (hrp) is "ln" + net hrp + optional amount,
57
        // except for signet where we add an additional "s" to differentiate it
58
        // from the older testnet3 (Core devs decided to use the same hrp for
59
        // signet as for testnet3 which is not optimal for LN). See
60
        // https://github.com/lightningnetwork/lightning-rfc/pull/844 for more
61
        // information.
62
        hrp := "ln" + invoice.Net.Bech32HRPSegwit
2✔
63
        if invoice.Net.Name == chaincfg.SigNetParams.Name {
2✔
UNCOV
64
                hrp = "lntbs"
×
UNCOV
65
        }
×
66
        if invoice.MilliSat != nil {
4✔
67
                // Encode the amount using the fewest possible characters.
2✔
68
                am, err := encodeAmount(*invoice.MilliSat)
2✔
69
                if err != nil {
2✔
70
                        return "", err
×
71
                }
×
72
                hrp += am
2✔
73
        }
74

75
        // The signature is over the single SHA-256 hash of the hrp + the
76
        // tagged fields encoded in base256.
77
        taggedFieldsBytes, err := bech32.ConvertBits(bufferBase32.Bytes(), 5, 8, true)
2✔
78
        if err != nil {
2✔
79
                return "", err
×
80
        }
×
81

82
        toSign := append([]byte(hrp), taggedFieldsBytes...)
2✔
83

2✔
84
        // We use compact signature format, and also encoded the recovery ID
2✔
85
        // such that a reader of the invoice can recover our pubkey from the
2✔
86
        // signature.
2✔
87
        sign, err := signer.SignCompact(toSign)
2✔
88
        if err != nil {
2✔
89
                return "", err
×
90
        }
×
91

92
        // From the header byte we can extract the recovery ID, and the last 64
93
        // bytes encode the signature.
94
        recoveryID := sign[0] - 27 - 4
2✔
95
        sig, err := lnwire.NewSigFromWireECDSA(sign[1:])
2✔
96
        if err != nil {
2✔
97
                return "", err
×
98
        }
×
99

100
        // If the pubkey field was explicitly set, it must be set to the pubkey
101
        // used to create the signature.
102
        if invoice.Destination != nil {
2✔
UNCOV
103
                signature, err := sig.ToSignature()
×
UNCOV
104
                if err != nil {
×
105
                        return "", fmt.Errorf("unable to deserialize "+
×
106
                                "signature: %v", err)
×
107
                }
×
108

UNCOV
109
                hash := chainhash.HashB(toSign)
×
UNCOV
110
                valid := signature.Verify(hash, invoice.Destination)
×
UNCOV
111
                if !valid {
×
UNCOV
112
                        return "", fmt.Errorf("signature does not match " +
×
UNCOV
113
                                "provided pubkey")
×
UNCOV
114
                }
×
115
        }
116

117
        // Convert the signature to base32 before writing it to the buffer.
118
        signBase32, err := bech32.ConvertBits(
2✔
119
                append(sig.RawBytes(), recoveryID),
2✔
120
                8, 5, true,
2✔
121
        )
2✔
122
        if err != nil {
2✔
123
                return "", err
×
124
        }
×
125
        bufferBase32.Write(signBase32)
2✔
126

2✔
127
        // Now we can create the bech32 encoded string from the base32 buffer.
2✔
128
        b32, err := bech32.Encode(hrp, bufferBase32.Bytes())
2✔
129
        if err != nil {
2✔
130
                return "", err
×
131
        }
×
132

133
        // Before returning, check that the bech32 encoded string is not greater
134
        // than our largest supported invoice size.
135
        if len(b32) > maxInvoiceLength {
2✔
136
                return "", ErrInvoiceTooLarge
×
137
        }
×
138

139
        return b32, nil
2✔
140
}
141

142
// writeTaggedFields writes the non-nil tagged fields of the Invoice to the
143
// base32 buffer.
144
func writeTaggedFields(bufferBase32 *bytes.Buffer, invoice *Invoice) error {
2✔
145
        if invoice.PaymentHash != nil {
4✔
146
                err := writeBytes32(bufferBase32, fieldTypeP, *invoice.PaymentHash)
2✔
147
                if err != nil {
2✔
148
                        return err
×
149
                }
×
150
        }
151

152
        if invoice.Description != nil {
4✔
153
                base32, err := bech32.ConvertBits([]byte(*invoice.Description),
2✔
154
                        8, 5, true)
2✔
155
                if err != nil {
2✔
156
                        return err
×
157
                }
×
158
                err = writeTaggedField(bufferBase32, fieldTypeD, base32)
2✔
159
                if err != nil {
2✔
160
                        return err
×
161
                }
×
162
        }
163

164
        if invoice.DescriptionHash != nil {
2✔
UNCOV
165
                err := writeBytes32(
×
UNCOV
166
                        bufferBase32, fieldTypeH, *invoice.DescriptionHash,
×
UNCOV
167
                )
×
UNCOV
168
                if err != nil {
×
169
                        return err
×
170
                }
×
171
        }
172

173
        if invoice.Metadata != nil {
2✔
UNCOV
174
                base32, err := bech32.ConvertBits(invoice.Metadata, 8, 5, true)
×
UNCOV
175
                if err != nil {
×
176
                        return err
×
177
                }
×
UNCOV
178
                err = writeTaggedField(bufferBase32, fieldTypeM, base32)
×
UNCOV
179
                if err != nil {
×
180
                        return err
×
181
                }
×
182
        }
183

184
        if invoice.minFinalCLTVExpiry != nil {
4✔
185
                finalDelta := uint64ToBase32(*invoice.minFinalCLTVExpiry)
2✔
186
                err := writeTaggedField(bufferBase32, fieldTypeC, finalDelta)
2✔
187
                if err != nil {
2✔
188
                        return err
×
189
                }
×
190
        }
191

192
        if invoice.expiry != nil {
4✔
193
                seconds := invoice.expiry.Seconds()
2✔
194
                expiry := uint64ToBase32(uint64(seconds))
2✔
195
                err := writeTaggedField(bufferBase32, fieldTypeX, expiry)
2✔
196
                if err != nil {
2✔
197
                        return err
×
198
                }
×
199
        }
200

201
        if invoice.FallbackAddr != nil {
2✔
UNCOV
202
                var version byte
×
UNCOV
203
                switch addr := invoice.FallbackAddr.(type) {
×
UNCOV
204
                case *btcutil.AddressPubKeyHash:
×
UNCOV
205
                        version = 17
×
UNCOV
206
                case *btcutil.AddressScriptHash:
×
UNCOV
207
                        version = 18
×
UNCOV
208
                case *btcutil.AddressWitnessPubKeyHash:
×
UNCOV
209
                        version = addr.WitnessVersion()
×
UNCOV
210
                case *btcutil.AddressWitnessScriptHash:
×
UNCOV
211
                        version = addr.WitnessVersion()
×
212
                default:
×
213
                        return fmt.Errorf("unknown fallback address type")
×
214
                }
UNCOV
215
                base32Addr, err := bech32.ConvertBits(
×
UNCOV
216
                        invoice.FallbackAddr.ScriptAddress(), 8, 5, true)
×
UNCOV
217
                if err != nil {
×
218
                        return err
×
219
                }
×
220

UNCOV
221
                err = writeTaggedField(bufferBase32, fieldTypeF,
×
UNCOV
222
                        append([]byte{version}, base32Addr...))
×
UNCOV
223
                if err != nil {
×
224
                        return err
×
225
                }
×
226
        }
227

228
        for _, routeHint := range invoice.RouteHints {
4✔
229
                // Each hop hint is encoded using 51 bytes, so we'll make to
2✔
230
                // sure to allocate enough space for the whole route hint.
2✔
231
                routeHintBase256 := make([]byte, 0, hopHintLen*len(routeHint))
2✔
232

2✔
233
                for _, hopHint := range routeHint {
4✔
234
                        hopHintBase256 := make([]byte, hopHintLen)
2✔
235
                        copy(hopHintBase256[:33], hopHint.NodeID.SerializeCompressed())
2✔
236
                        binary.BigEndian.PutUint64(
2✔
237
                                hopHintBase256[33:41], hopHint.ChannelID,
2✔
238
                        )
2✔
239
                        binary.BigEndian.PutUint32(
2✔
240
                                hopHintBase256[41:45], hopHint.FeeBaseMSat,
2✔
241
                        )
2✔
242
                        binary.BigEndian.PutUint32(
2✔
243
                                hopHintBase256[45:49], hopHint.FeeProportionalMillionths,
2✔
244
                        )
2✔
245
                        binary.BigEndian.PutUint16(
2✔
246
                                hopHintBase256[49:51], hopHint.CLTVExpiryDelta,
2✔
247
                        )
2✔
248
                        routeHintBase256 = append(routeHintBase256, hopHintBase256...)
2✔
249
                }
2✔
250

251
                routeHintBase32, err := bech32.ConvertBits(
2✔
252
                        routeHintBase256, 8, 5, true,
2✔
253
                )
2✔
254
                if err != nil {
2✔
255
                        return err
×
256
                }
×
257

258
                err = writeTaggedField(bufferBase32, fieldTypeR, routeHintBase32)
2✔
259
                if err != nil {
2✔
260
                        return err
×
261
                }
×
262
        }
263

264
        for _, path := range invoice.BlindedPaymentPaths {
4✔
265
                var buf bytes.Buffer
2✔
266

2✔
267
                err := path.Encode(&buf)
2✔
268
                if err != nil {
2✔
269
                        return err
×
270
                }
×
271

272
                blindedPathBase32, err := bech32.ConvertBits(
2✔
273
                        buf.Bytes(), 8, 5, true,
2✔
274
                )
2✔
275
                if err != nil {
2✔
276
                        return err
×
277
                }
×
278

279
                err = writeTaggedField(
2✔
280
                        bufferBase32, fieldTypeB, blindedPathBase32,
2✔
281
                )
2✔
282
                if err != nil {
2✔
283
                        return err
×
284
                }
×
285
        }
286

287
        if invoice.Destination != nil {
2✔
UNCOV
288
                // Convert 33 byte pubkey to 53 5-bit groups.
×
UNCOV
289
                pubKeyBase32, err := bech32.ConvertBits(
×
UNCOV
290
                        invoice.Destination.SerializeCompressed(), 8, 5, true)
×
UNCOV
291
                if err != nil {
×
292
                        return err
×
293
                }
×
294

UNCOV
295
                if len(pubKeyBase32) != pubKeyBase32Len {
×
296
                        return fmt.Errorf("invalid pubkey length: %d",
×
297
                                len(invoice.Destination.SerializeCompressed()))
×
298
                }
×
299

UNCOV
300
                err = writeTaggedField(bufferBase32, fieldTypeN, pubKeyBase32)
×
UNCOV
301
                if err != nil {
×
302
                        return err
×
303
                }
×
304
        }
305

306
        err := fn.MapOptionZ(invoice.PaymentAddr, func(addr [32]byte) error {
4✔
307
                return writeBytes32(bufferBase32, fieldTypeS, addr)
2✔
308
        })
2✔
309
        if err != nil {
2✔
310
                return err
×
311
        }
×
312

313
        if invoice.Features.SerializeSize32() > 0 {
4✔
314
                var b bytes.Buffer
2✔
315
                err := invoice.Features.RawFeatureVector.EncodeBase32(&b)
2✔
316
                if err != nil {
2✔
317
                        return err
×
318
                }
×
319

320
                err = writeTaggedField(bufferBase32, fieldType9, b.Bytes())
2✔
321
                if err != nil {
2✔
322
                        return err
×
323
                }
×
324
        }
325

326
        return nil
2✔
327
}
328

329
// writeBytes32 encodes a 32-byte array as base32 and writes it to bufferBase32
330
// under the passed fieldType.
331
func writeBytes32(bufferBase32 *bytes.Buffer, fieldType byte, b [32]byte) error {
2✔
332
        // Convert 32 byte hash to 52 5-bit groups.
2✔
333
        base32, err := bech32.ConvertBits(b[:], 8, 5, true)
2✔
334
        if err != nil {
2✔
335
                return err
×
336
        }
×
337

338
        return writeTaggedField(bufferBase32, fieldType, base32)
2✔
339
}
340

341
// writeTaggedField takes the type of a tagged data field, and the data of
342
// the tagged field (encoded in base32), and writes the type, length and data
343
// to the buffer.
344
func writeTaggedField(bufferBase32 *bytes.Buffer, dataType byte, data []byte) error {
2✔
345
        // Length must be exactly 10 bits, so add leading zero groups if
2✔
346
        // needed.
2✔
347
        lenBase32 := uint64ToBase32(uint64(len(data)))
2✔
348
        for len(lenBase32) < 2 {
4✔
349
                lenBase32 = append([]byte{0}, lenBase32...)
2✔
350
        }
2✔
351

352
        if len(lenBase32) != 2 {
2✔
353
                return fmt.Errorf("data length too big to fit within 10 bits: %d",
×
354
                        len(data))
×
355
        }
×
356

357
        err := bufferBase32.WriteByte(dataType)
2✔
358
        if err != nil {
2✔
359
                return fmt.Errorf("unable to write to buffer: %w", err)
×
360
        }
×
361
        _, err = bufferBase32.Write(lenBase32)
2✔
362
        if err != nil {
2✔
363
                return fmt.Errorf("unable to write to buffer: %w", err)
×
364
        }
×
365
        _, err = bufferBase32.Write(data)
2✔
366
        if err != nil {
2✔
367
                return fmt.Errorf("unable to write to buffer: %w", err)
×
368
        }
×
369

370
        return nil
2✔
371
}
372

373
// uint64ToBase32 converts a uint64 to a base32 encoded integer encoded using
374
// as few 5-bit groups as possible.
375
func uint64ToBase32(num uint64) []byte {
2✔
376
        // Return at least one group.
2✔
377
        if num == 0 {
4✔
378
                return []byte{0}
2✔
379
        }
2✔
380

381
        // To fit an uint64, we need at most is ceil(64 / 5) = 13 groups.
382
        arr := make([]byte, 13)
2✔
383
        i := 13
2✔
384
        for num > 0 {
4✔
385
                i--
2✔
386
                arr[i] = byte(num & uint64(31)) // 0b11111 in binary
2✔
387
                num >>= 5
2✔
388
        }
2✔
389

390
        // We only return non-zero leading groups.
391
        return arr[i:]
2✔
392
}
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