neo-go/pkg/network/payload/addr.go

package payload

import (
	"bytes"
	"encoding/binary"
)

// Endpoint host + port of a node.
type Endpoint struct {
	IP   [16]byte // TODO: make a uint128 type
	Port uint16
}

// AddrWithTime payload
type AddrWithTime struct {
	Timestamp uint32
	Services  uint64
	Addr      Endpoint
}

// Size implements the payloader interface.
func (p *AddrWithTime) Size() uint32 {
	return 30
}

// UnmarshalBinary implements the Payloader interface.
func (p *AddrWithTime) UnmarshalBinary(b []byte) error {
	p.Timestamp = binary.LittleEndian.Uint32(b[0:4])
	p.Services = binary.LittleEndian.Uint64(b[4:12])
	binary.Read(bytes.NewReader(b[12:28]), binary.BigEndian, &p.Addr.IP)
	p.Addr.Port = binary.LittleEndian.Uint16(b[28:30])
	return nil
}

// MarshalBinary implements the Payloader interface.
func (p *AddrWithTime) MarshalBinary() ([]byte, error) {
	return nil, nil
}

// AddressList contains a slice of AddrWithTime.
type AddressList struct {
	Addrs []*AddrWithTime
}

// UnmarshalBinary implements the Payloader interface.
func (p *AddressList) UnmarshalBinary(b []byte) error {
	var lenList uint8
	binary.Read(bytes.NewReader(b[0:1]), binary.LittleEndian, &lenList)

	offset := 1 // skip the uint8 length byte.
	size := 30  // size of AddrWithTime
	for i := 0; i < int(lenList); i++ {
		address := &AddrWithTime{}
		address.UnmarshalBinary(b[offset : offset+size])
		p.Addrs = append(p.Addrs, address)
		offset += size
	}

	return nil
}

// MarshalBinary implements the Payloader interface.
func (p *AddressList) MarshalBinary() ([]byte, error) {
	return nil, nil
}

// Size implements the Payloader interface.
func (p *AddressList) Size() uint32 {
	return uint32(len(p.Addrs) * 30)
}