neoneo-go/pkg/network/payload/version.go

91 lines
2.4 KiB
Go
Raw Normal View History

2018-01-27 15:00:28 +00:00
package payload
import (
"encoding/binary"
2018-01-28 15:06:41 +00:00
"io"
2018-01-27 15:00:28 +00:00
)
2018-01-27 15:47:43 +00:00
const (
2018-01-28 15:06:41 +00:00
lenUA = 12
2018-01-28 10:12:05 +00:00
minVersionSize = 27
2018-01-27 15:47:43 +00:00
)
2018-01-27 15:00:28 +00:00
// Version payload.
type Version struct {
// currently the version of the protocol is 0
Version uint32
// currently 1
Services uint64
// timestamp
Timestamp uint32
// port this server is listening on
Port uint16
// it's used to distinguish the node from public IP
Nonce uint32
2018-01-28 10:12:05 +00:00
// client id
2018-01-28 15:06:41 +00:00
UserAgent [lenUA]byte
2018-01-27 15:00:28 +00:00
// Height of the block chain
StartHeight uint32
// Whether to receive and forward
Relay bool
}
// NewVersion returns a pointer to a Version payload.
2018-01-28 07:03:18 +00:00
func NewVersion(id uint32, p uint16, ua string, h uint32, r bool) *Version {
2018-01-27 15:00:28 +00:00
return &Version{
Version: 0,
Services: 1,
Timestamp: 12345,
Port: p,
2018-01-28 07:03:18 +00:00
Nonce: id,
2018-01-28 15:06:41 +00:00
UserAgent: uaToByteArray(ua),
2018-01-27 15:00:28 +00:00
StartHeight: 0,
Relay: r,
}
}
2018-01-28 15:06:41 +00:00
// DecodeBinary implements the Payload interface.
func (p *Version) DecodeBinary(r io.Reader) error {
// TODO: Length of the user agent should be calculated dynamicaly.
2018-01-28 10:12:05 +00:00
// There is no information about the size or format of this.
// the only thing we know is by looking at the #c source code.
// /NEO:{0}/ => /NEO:2.6.0/
2018-01-28 15:06:41 +00:00
err := binary.Read(r, binary.LittleEndian, &p.Version)
err = binary.Read(r, binary.LittleEndian, &p.Services)
err = binary.Read(r, binary.LittleEndian, &p.Timestamp)
err = binary.Read(r, binary.LittleEndian, &p.Port)
err = binary.Read(r, binary.LittleEndian, &p.Nonce)
err = binary.Read(r, binary.LittleEndian, &p.UserAgent)
err = binary.Read(r, binary.LittleEndian, &p.StartHeight)
err = binary.Read(r, binary.LittleEndian, &p.Relay)
2018-01-28 10:12:05 +00:00
2018-01-28 15:06:41 +00:00
return err
2018-01-27 15:00:28 +00:00
}
2018-01-28 15:06:41 +00:00
// EncodeBinary implements the Payload interface.
func (p *Version) EncodeBinary(w io.Writer) error {
err := binary.Write(w, binary.LittleEndian, p.Version)
err = binary.Write(w, binary.LittleEndian, p.Services)
err = binary.Write(w, binary.LittleEndian, p.Timestamp)
err = binary.Write(w, binary.LittleEndian, p.Port)
err = binary.Write(w, binary.LittleEndian, p.Nonce)
err = binary.Write(w, binary.LittleEndian, p.UserAgent)
err = binary.Write(w, binary.LittleEndian, p.StartHeight)
err = binary.Write(w, binary.LittleEndian, p.Relay)
2018-01-28 07:03:18 +00:00
2018-01-28 15:06:41 +00:00
return err
2018-01-28 07:03:18 +00:00
}
2018-01-27 15:00:28 +00:00
2018-01-28 10:12:05 +00:00
// Size implements the payloader interface.
2018-01-28 07:03:18 +00:00
func (p *Version) Size() uint32 {
return uint32(minVersionSize + len(p.UserAgent))
2018-01-27 15:00:28 +00:00
}
2018-01-28 15:06:41 +00:00
func uaToByteArray(ua string) [lenUA]byte {
buf := [lenUA]byte{}
for i := 0; i < lenUA; i++ {
buf[i] = ua[i]
}
return buf
}