47 lines
835 B
Go
47 lines
835 B
Go
|
package krpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding"
|
||
|
|
"encoding/binary"
|
||
|
|
"fmt"
|
||
|
|
"math"
|
||
|
|
"math/rand"
|
||
|
|
"net"
|
||
|
|
)
|
||
|
|
|
||
|
|
type NodeInfo struct {
|
||
|
|
ID [20]byte
|
||
|
|
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:])
|
||
|
|
}
|