5bf00db2c9
The logic here is that we'll have all binary encoding/decoding done via our io package, which simplifies error handling. This functionality doesn't belong to util, so it's moved. This also expands BufBinWriter with Reset() method to fit the needs of core package.
99 lines
2.1 KiB
Go
99 lines
2.1 KiB
Go
package payload
|
|
|
|
import (
|
|
"net"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/io"
|
|
)
|
|
|
|
// AddressAndTime payload.
|
|
type AddressAndTime struct {
|
|
Timestamp uint32
|
|
Services uint64
|
|
IP [16]byte
|
|
Port uint16
|
|
}
|
|
|
|
// NewAddressAndTime creates a new AddressAndTime object.
|
|
func NewAddressAndTime(e *net.TCPAddr, t time.Time) *AddressAndTime {
|
|
aat := AddressAndTime{
|
|
Timestamp: uint32(t.UTC().Unix()),
|
|
Services: 1,
|
|
Port: uint16(e.Port),
|
|
}
|
|
copy(aat.IP[:], e.IP)
|
|
return &aat
|
|
}
|
|
|
|
// DecodeBinary implements the Payload interface.
|
|
func (p *AddressAndTime) DecodeBinary(br *io.BinReader) error {
|
|
br.ReadLE(&p.Timestamp)
|
|
br.ReadLE(&p.Services)
|
|
br.ReadBE(&p.IP)
|
|
br.ReadBE(&p.Port)
|
|
return br.Err
|
|
}
|
|
|
|
// EncodeBinary implements the Payload interface.
|
|
func (p *AddressAndTime) EncodeBinary(bw *io.BinWriter) error {
|
|
bw.WriteLE(p.Timestamp)
|
|
bw.WriteLE(p.Services)
|
|
bw.WriteBE(p.IP)
|
|
bw.WriteBE(p.Port)
|
|
return bw.Err
|
|
}
|
|
|
|
// IPPortString makes a string from IP and port specified.
|
|
func (p *AddressAndTime) IPPortString() string {
|
|
var netip net.IP = make(net.IP, 16)
|
|
|
|
copy(netip, p.IP[:])
|
|
port := strconv.Itoa(int(p.Port))
|
|
return netip.String() + ":" + port
|
|
}
|
|
|
|
// AddressList is a list with AddrAndTime.
|
|
type AddressList struct {
|
|
Addrs []*AddressAndTime
|
|
}
|
|
|
|
// NewAddressList creates a list for n AddressAndTime elements.
|
|
func NewAddressList(n int) *AddressList {
|
|
alist := AddressList{
|
|
Addrs: make([]*AddressAndTime, n),
|
|
}
|
|
return &alist
|
|
}
|
|
|
|
// DecodeBinary implements the Payload interface.
|
|
func (p *AddressList) DecodeBinary(br *io.BinReader) error {
|
|
listLen := br.ReadVarUint()
|
|
if br.Err != nil {
|
|
return br.Err
|
|
}
|
|
|
|
p.Addrs = make([]*AddressAndTime, listLen)
|
|
for i := 0; i < int(listLen); i++ {
|
|
p.Addrs[i] = &AddressAndTime{}
|
|
if err := p.Addrs[i].DecodeBinary(br); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// EncodeBinary implements the Payload interface.
|
|
func (p *AddressList) EncodeBinary(bw *io.BinWriter) error {
|
|
bw.WriteVarUint(uint64(len(p.Addrs)))
|
|
if bw.Err != nil {
|
|
return bw.Err
|
|
}
|
|
for _, addr := range p.Addrs {
|
|
if err := addr.EncodeBinary(bw); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|