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

lightningnetwork / lnd / 12058234999

27 Nov 2024 09:06PM UTC coverage: 57.847% (-1.1%) from 58.921%
12058234999

Pull #9148

github

ProofOfKeags
lnwire: convert DynPropose and DynCommit to use typed tlv records
Pull Request #9148: DynComms [2/n]: lnwire: add authenticated wire messages for Dyn*

142 of 177 new or added lines in 4 files covered. (80.23%)

19365 existing lines in 251 files now uncovered.

100876 of 174383 relevant lines covered (57.85%)

25338.28 hits per line

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

89.82
/tlv/primitive.go
1
package tlv
2

3
import (
4
        "encoding/binary"
5
        "errors"
6
        "fmt"
7
        "io"
8

9
        "github.com/btcsuite/btcd/btcec/v2"
10
)
11

12
// ErrTypeForEncoding signals that an incorrect type was passed to an Encoder.
13
type ErrTypeForEncoding struct {
14
        val     interface{}
15
        expType string
16
}
17

18
// NewTypeForEncodingErr creates a new ErrTypeForEncoding given the incorrect
19
// val and the expected type.
20
func NewTypeForEncodingErr(val interface{}, expType string) ErrTypeForEncoding {
10✔
21
        return ErrTypeForEncoding{
10✔
22
                val:     val,
10✔
23
                expType: expType,
10✔
24
        }
10✔
25
}
10✔
26

27
// Error returns a human-readable description of the type mismatch.
28
func (e ErrTypeForEncoding) Error() string {
×
29
        return fmt.Sprintf("ErrTypeForEncoding want (type: *%s), "+
×
30
                "got (type: %T)", e.expType, e.val)
×
31
}
×
32

33
// ErrTypeForDecoding signals that an incorrect type was passed to a Decoder or
34
// that the expected length of the encoding is different from that required by
35
// the expected type.
36
type ErrTypeForDecoding struct {
37
        val       interface{}
38
        expType   string
39
        valLength uint64
40
        expLength uint64
41
}
42

43
// NewTypeForDecodingErr creates a new ErrTypeForDecoding given the incorrect
44
// val and expected type, or the mismatch in their expected lengths.
45
func NewTypeForDecodingErr(val interface{}, expType string,
46
        valLength, expLength uint64) ErrTypeForDecoding {
41✔
47

41✔
48
        return ErrTypeForDecoding{
41✔
49
                val:       val,
41✔
50
                expType:   expType,
41✔
51
                valLength: valLength,
41✔
52
                expLength: expLength,
41✔
53
        }
41✔
54
}
41✔
55

56
// Error returns a human-readable description of the type mismatch.
57
func (e ErrTypeForDecoding) Error() string {
×
58
        return fmt.Sprintf("ErrTypeForDecoding want (type: *%s, length: %v), "+
×
59
                "got (type: %T, length: %v)", e.expType, e.expLength, e.val,
×
60
                e.valLength)
×
61
}
×
62

63
var (
64
        byteOrder = binary.BigEndian
65
)
66

67
// EUint8 is an Encoder for uint8 values. An error is returned if val is not a
68
// *uint8.
69
func EUint8(w io.Writer, val interface{}, buf *[8]byte) error {
6,819✔
70
        if i, ok := val.(*uint8); ok {
13,637✔
71
                return EUint8T(w, *i, buf)
6,818✔
72
        }
6,818✔
73
        return NewTypeForEncodingErr(val, "uint8")
1✔
74
}
75

76
// EUint8T encodes a uint8 val to the provided io.Writer. This method is exposed
77
// so that encodings for custom uint8-like types can be created without
78
// incurring an extra heap allocation.
79
func EUint8T(w io.Writer, val uint8, buf *[8]byte) error {
6,818✔
80
        buf[0] = val
6,818✔
81
        _, err := w.Write(buf[:1])
6,818✔
82
        return err
6,818✔
83
}
6,818✔
84

85
// EUint16 is an Encoder for uint16 values. An error is returned if val is not a
86
// *uint16.
87
func EUint16(w io.Writer, val interface{}, buf *[8]byte) error {
23,557✔
88
        if i, ok := val.(*uint16); ok {
47,113✔
89
                return EUint16T(w, *i, buf)
23,556✔
90
        }
23,556✔
91
        return NewTypeForEncodingErr(val, "uint16")
1✔
92
}
93

94
// EUint16T encodes a uint16 val to the provided io.Writer. This method is
95
// exposed so that encodings for custom uint16-like types can be created without
96
// incurring an extra heap allocation.
97
func EUint16T(w io.Writer, val uint16, buf *[8]byte) error {
23,556✔
98
        byteOrder.PutUint16(buf[:2], val)
23,556✔
99
        _, err := w.Write(buf[:2])
23,556✔
100
        return err
23,556✔
101
}
23,556✔
102

103
// EUint32 is an Encoder for uint32 values. An error is returned if val is not a
104
// *uint32.
105
func EUint32(w io.Writer, val interface{}, buf *[8]byte) error {
8,731✔
106
        if i, ok := val.(*uint32); ok {
17,461✔
107
                return EUint32T(w, *i, buf)
8,730✔
108
        }
8,730✔
109
        return NewTypeForEncodingErr(val, "uint32")
1✔
110
}
111

112
// EUint32T encodes a uint32 val to the provided io.Writer. This method is
113
// exposed so that encodings for custom uint32-like types can be created without
114
// incurring an extra heap allocation.
115
func EUint32T(w io.Writer, val uint32, buf *[8]byte) error {
21,868✔
116
        byteOrder.PutUint32(buf[:4], val)
21,868✔
117
        _, err := w.Write(buf[:4])
21,868✔
118
        return err
21,868✔
119
}
21,868✔
120

121
// EUint64 is an Encoder for uint64 values. An error is returned if val is not a
122
// *uint64.
123
func EUint64(w io.Writer, val interface{}, buf *[8]byte) error {
22,973✔
124
        if i, ok := val.(*uint64); ok {
45,945✔
125
                return EUint64T(w, *i, buf)
22,972✔
126
        }
22,972✔
127
        return NewTypeForEncodingErr(val, "uint64")
1✔
128
}
129

130
// EUint64T encodes a uint64 val to the provided io.Writer. This method is
131
// exposed so that encodings for custom uint64-like types can be created without
132
// incurring an extra heap allocation.
133
func EUint64T(w io.Writer, val uint64, buf *[8]byte) error {
26,479✔
134
        byteOrder.PutUint64(buf[:], val)
26,479✔
135
        _, err := w.Write(buf[:])
26,479✔
136
        return err
26,479✔
137
}
26,479✔
138

139
// EBool encodes a boolean. An error is returned if val is not a boolean.
140
func EBool(w io.Writer, val interface{}, buf *[8]byte) error {
3,567✔
141
        if i, ok := val.(*bool); ok {
7,133✔
142
                return EBoolT(w, *i, buf)
3,566✔
143
        }
3,566✔
144
        return NewTypeForEncodingErr(val, "bool")
1✔
145
}
146

147
// EBoolT encodes a bool val to the provided io.Writer. This method is exposed
148
// so that encodings for custom bool-like types can be created without
149
// incurring an extra heap allocation.
150
func EBoolT(w io.Writer, val bool, buf *[8]byte) error {
3,566✔
151
        if val {
5,320✔
152
                buf[0] = 1
1,754✔
153
        } else {
3,566✔
154
                buf[0] = 0
1,812✔
155
        }
1,812✔
156
        _, err := w.Write(buf[:1])
3,566✔
157
        return err
3,566✔
158
}
159

160
// DUint8 is a Decoder for uint8 values. An error is returned if val is not a
161
// *uint8.
162
func DUint8(r io.Reader, val interface{}, buf *[8]byte, l uint64) error {
8,086✔
163
        if i, ok := val.(*uint8); ok && l == 1 {
16,171✔
164
                if _, err := io.ReadFull(r, buf[:1]); err != nil {
8,085✔
165
                        return err
×
166
                }
×
167
                *i = buf[0]
8,085✔
168
                return nil
8,085✔
169
        }
170
        return NewTypeForDecodingErr(val, "uint8", l, 1)
1✔
171
}
172

173
// DUint16 is a Decoder for uint16 values. An error is returned if val is not a
174
// *uint16.
175
func DUint16(r io.Reader, val interface{}, buf *[8]byte, l uint64) error {
10,247✔
176
        if i, ok := val.(*uint16); ok && l == 2 {
20,490✔
177
                if _, err := io.ReadFull(r, buf[:2]); err != nil {
10,243✔
UNCOV
178
                        return err
×
UNCOV
179
                }
×
180
                *i = byteOrder.Uint16(buf[:2])
10,243✔
181
                return nil
10,243✔
182
        }
183
        return NewTypeForDecodingErr(val, "uint16", l, 2)
4✔
184
}
185

186
// DUint32 is a Decoder for uint32 values. An error is returned if val is not a
187
// *uint32.
188
func DUint32(r io.Reader, val interface{}, buf *[8]byte, l uint64) error {
26,378✔
189
        if i, ok := val.(*uint32); ok && l == 4 {
52,752✔
190
                if _, err := io.ReadFull(r, buf[:4]); err != nil {
26,378✔
191
                        return err
4✔
192
                }
4✔
193
                *i = byteOrder.Uint32(buf[:4])
26,370✔
194
                return nil
26,370✔
195
        }
196
        return NewTypeForDecodingErr(val, "uint32", l, 4)
4✔
197
}
198

199
// DUint64 is a Decoder for uint64 values. An error is returned if val is not a
200
// *uint64.
201
func DUint64(r io.Reader, val interface{}, buf *[8]byte, l uint64) error {
44,217✔
202
        if i, ok := val.(*uint64); ok && l == 8 {
88,430✔
203
                if _, err := io.ReadFull(r, buf[:]); err != nil {
44,216✔
204
                        return err
3✔
205
                }
3✔
206
                *i = byteOrder.Uint64(buf[:])
44,210✔
207
                return nil
44,210✔
208
        }
209
        return NewTypeForDecodingErr(val, "uint64", l, 8)
4✔
210
}
211

212
// DBool decodes a boolean. An error is returned if val is not a boolean.
213
func DBool(r io.Reader, val interface{}, buf *[8]byte, l uint64) error {
31✔
214
        if i, ok := val.(*bool); ok && l == 1 {
61✔
215
                if _, err := io.ReadFull(r, buf[:1]); err != nil {
30✔
216
                        return err
×
217
                }
×
218
                if buf[0] != 0 && buf[0] != 1 {
30✔
219
                        return errors.New("corrupted data")
×
220
                }
×
221
                *i = buf[0] != 0
30✔
222
                return nil
30✔
223
        }
224
        return NewTypeForDecodingErr(val, "bool", l, 1)
1✔
225
}
226

227
// EBytes32 is an Encoder for 32-byte arrays. An error is returned if val is not
228
// a *[32]byte.
229
func EBytes32(w io.Writer, val interface{}, _ *[8]byte) error {
16,240✔
230
        if b, ok := val.(*[32]byte); ok {
32,479✔
231
                _, err := w.Write(b[:])
16,239✔
232
                return err
16,239✔
233
        }
16,239✔
234
        return NewTypeForEncodingErr(val, "[32]byte")
1✔
235
}
236

237
// DBytes32 is a Decoder for 32-byte arrays. An error is returned if val is not
238
// a *[32]byte.
239
func DBytes32(r io.Reader, val interface{}, _ *[8]byte, l uint64) error {
13,728✔
240
        if b, ok := val.(*[32]byte); ok && l == 32 {
27,455✔
241
                _, err := io.ReadFull(r, b[:])
13,727✔
242
                return err
13,727✔
243
        }
13,727✔
244
        return NewTypeForDecodingErr(val, "[32]byte", l, 32)
1✔
245
}
246

247
// EBytes33 is an Encoder for 33-byte arrays. An error is returned if val is not
248
// a *[33]byte.
249
func EBytes33(w io.Writer, val interface{}, _ *[8]byte) error {
308✔
250
        if b, ok := val.(*[33]byte); ok {
615✔
251
                _, err := w.Write(b[:])
307✔
252
                return err
307✔
253
        }
307✔
254
        return NewTypeForEncodingErr(val, "[33]byte")
1✔
255
}
256

257
// DBytes33 is a Decoder for 33-byte arrays. An error is returned if val is not
258
// a *[33]byte.
259
func DBytes33(r io.Reader, val interface{}, _ *[8]byte, l uint64) error {
296✔
260
        if b, ok := val.(*[33]byte); ok {
591✔
261
                _, err := io.ReadFull(r, b[:])
295✔
262
                return err
295✔
263
        }
295✔
264
        return NewTypeForDecodingErr(val, "[33]byte", l, 33)
1✔
265
}
266

267
// EBytes64 is an Encoder for 64-byte arrays. An error is returned if val is not
268
// a *[64]byte.
269
func EBytes64(w io.Writer, val interface{}, _ *[8]byte) error {
318✔
270
        if b, ok := val.(*[64]byte); ok {
635✔
271
                _, err := w.Write(b[:])
317✔
272
                return err
317✔
273
        }
317✔
274
        return NewTypeForEncodingErr(val, "[64]byte")
1✔
275
}
276

277
// DBytes64 is an Decoder for 64-byte arrays. An error is returned if val is not
278
// a *[64]byte.
279
func DBytes64(r io.Reader, val interface{}, _ *[8]byte, l uint64) error {
332✔
280
        if b, ok := val.(*[64]byte); ok && l == 64 {
659✔
281
                _, err := io.ReadFull(r, b[:])
327✔
282
                return err
327✔
283
        }
327✔
284
        return NewTypeForDecodingErr(val, "[64]byte", l, 64)
5✔
285
}
286

287
// EPubKey is an Encoder for *btcec.PublicKey values. An error is returned if
288
// val is not a **btcec.PublicKey.
289
func EPubKey(w io.Writer, val interface{}, _ *[8]byte) error {
100✔
290
        if pk, ok := val.(**btcec.PublicKey); ok {
199✔
291
                _, err := w.Write((*pk).SerializeCompressed())
99✔
292
                return err
99✔
293
        }
99✔
294
        return NewTypeForEncodingErr(val, "*btcec.PublicKey")
1✔
295
}
296

297
// DPubKey is a Decoder for *btcec.PublicKey values. An error is returned if val
298
// is not a **btcec.PublicKey.
299
func DPubKey(r io.Reader, val interface{}, _ *[8]byte, l uint64) error {
96✔
300
        if pk, ok := val.(**btcec.PublicKey); ok && l == 33 {
190✔
301
                var b [33]byte
94✔
302
                _, err := io.ReadFull(r, b[:])
94✔
303
                if err != nil {
96✔
304
                        return err
2✔
305
                }
2✔
306

307
                p, err := btcec.ParsePubKey(b[:])
92✔
308
                if err != nil {
96✔
309
                        return err
4✔
310
                }
4✔
311

312
                *pk = p
88✔
313

88✔
314
                return nil
88✔
315
        }
316
        return NewTypeForDecodingErr(val, "*btcec.PublicKey", l, 33)
2✔
317
}
318

319
// EVarBytes is an Encoder for variable byte slices. An error is returned if val
320
// is not *[]byte.
321
func EVarBytes(w io.Writer, val interface{}, _ *[8]byte) error {
12,261✔
322
        if b, ok := val.(*[]byte); ok {
24,521✔
323
                _, err := w.Write(*b)
12,260✔
324
                return err
12,260✔
325
        }
12,260✔
326
        return NewTypeForEncodingErr(val, "[]byte")
1✔
327
}
328

329
// DVarBytes is a Decoder for variable byte slices. An error is returned if val
330
// is not *[]byte.
331
func DVarBytes(r io.Reader, val interface{}, _ *[8]byte, l uint64) error {
13,987✔
332
        if b, ok := val.(*[]byte); ok {
27,973✔
333
                *b = make([]byte, l)
13,986✔
334
                _, err := io.ReadFull(r, *b)
13,986✔
335
                return err
13,986✔
336
        }
13,986✔
337
        return NewTypeForDecodingErr(val, "[]byte", l, l)
1✔
338
}
339

340
// EBigSize encodes an uint32 or an uint64 using BigSize format. An error is
341
// returned if val is not either *uint32 or *uint64.
342
func EBigSize(w io.Writer, val interface{}, buf *[8]byte) error {
19,747✔
343
        if i, ok := val.(*uint32); ok {
19,750✔
344
                return WriteVarInt(w, uint64(*i), buf)
3✔
345
        }
3✔
346

347
        if i, ok := val.(*uint64); ok {
39,488✔
348
                return WriteVarInt(w, uint64(*i), buf)
19,744✔
349
        }
19,744✔
350

351
        return NewTypeForEncodingErr(val, "BigSize")
×
352
}
353

354
// DBigSize decodes an uint32 or an uint64 using BigSize format. An error is
355
// returned if val is not either *uint32 or *uint64.
356
func DBigSize(r io.Reader, val interface{}, buf *[8]byte, l uint64) error {
10,160✔
357
        if i, ok := val.(*uint32); ok {
10,163✔
358
                v, err := ReadVarInt(r, buf)
3✔
359
                if err != nil {
3✔
360
                        return err
×
361
                }
×
362
                *i = uint32(v)
3✔
363
                return nil
3✔
364
        }
365

366
        if i, ok := val.(*uint64); ok {
20,314✔
367
                v, err := ReadVarInt(r, buf)
10,157✔
368
                if err != nil {
10,157✔
369
                        return err
×
370
                }
×
371
                *i = v
10,157✔
372
                return nil
10,157✔
373
        }
374

375
        return NewTypeForDecodingErr(val, "BigSize", l, 8)
×
376
}
377

378
// SizeBigSize returns a SizeFunc that can compute the length of BigSize.
379
func SizeBigSize(val interface{}) SizeFunc {
33,954✔
380
        var size uint64
33,954✔
381

33,954✔
382
        if i, ok := val.(*uint32); ok {
33,960✔
383
                size = VarIntSize(uint64(*i))
6✔
384
        }
6✔
385

386
        if i, ok := val.(*uint64); ok {
67,368✔
387
                size = VarIntSize(uint64(*i))
33,414✔
388
        }
33,414✔
389

390
        return func() uint64 {
53,701✔
391
                return size
19,747✔
392
        }
19,747✔
393
}
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