Cloudpods/backend/vendor/github.com/anacrolix/dht/v2/krpc/nodeinfo.go

package krpc

import (
	"bytes"
	"encoding"
	"encoding/binary"
	"fmt"
	"math"
	"math/rand"
	"net"
)

type NodeInfo struct {
	ID   ID
	Addr NodeAddr
}

func (me NodeInfo) String() string {
	return fmt.Sprintf("{%x at %s}", me.ID, me.Addr)
}

func RandomNodeInfo(ipLen int) (ni NodeInfo) {
	rand.Read(ni.ID[:])
	ni.Addr.IP = make(net.IP, ipLen)
	rand.Read(ni.Addr.IP)
	ni.Addr.Port = rand.Intn(math.MaxUint16 + 1)
	return
}

var _ interface {
	encoding.BinaryMarshaler
	encoding.BinaryUnmarshaler
} = (*NodeInfo)(nil)

func (ni NodeInfo) MarshalBinary() ([]byte, error) {
	var w bytes.Buffer
	w.Write(ni.ID[:])
	w.Write(ni.Addr.IP)
	binary.Write(&w, binary.BigEndian, uint16(ni.Addr.Port))
	return w.Bytes(), nil
}

func (ni *NodeInfo) UnmarshalBinary(b []byte) error {
	copy(ni.ID[:], b)
	return ni.Addr.UnmarshalBinary(b[20:])
}