mirrors/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-03-04 01:36:24 +01:00
Code Issues Projects Releases Wiki Activity

lnwire: add randOpaqueAddrs for fuzz test

Browse source 
Let TestLightningWireProtocol also include random OpaqueAddrs.
This commit is contained in:
Elle Mouton 2022-04-21 10:56:38 +02:00
parent 1477c754f5
commit f40e2687ac
No known key found for this signature in database
GPG key ID: D7D916376026F177
1 changed files with 24 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
lnwire/lnwire_test.go
View file

@ -159,6 +159,22 @@ func randV3OnionAddr(r *rand.Rand) (*tor.OnionAddr, error) {
return &tor.OnionAddr{OnionService: onionService, Port: addrPort}, nil
}
func randOpaqueAddr(r *rand.Rand) (*OpaqueAddrs, error) {
payloadLen := r.Int63n(64) + 1
payload := make([]byte, payloadLen)
// The first byte is the address type. So set it to one that we
// definitely don't know about.
payload[0] = math.MaxUint8
// Generate random bytes for the rest of the payload.
if _, err := r.Read(payload[1:]); err != nil {
return nil, err
}
return &OpaqueAddrs{Payload: payload}, nil
}
func randAddrs(r *rand.Rand) ([]net.Addr, error) {
tcp4Addr, err := randTCP4Addr(r)
if err != nil {
@ -180,7 +196,14 @@ func randAddrs(r *rand.Rand) ([]net.Addr, error) {
return nil, err
}
return []net.Addr{tcp4Addr, tcp6Addr, v2OnionAddr, v3OnionAddr}, nil
opaqueAddrs, err := randOpaqueAddr(r)
if err != nil {
return nil, err
}
return []net.Addr{
tcp4Addr, tcp6Addr, v2OnionAddr, v3OnionAddr, opaqueAddrs,
}, nil
}
// TestChanUpdateChanFlags ensures that converting the ChanUpdateChanFlags and