mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-24 15:02:32 +00:00
42e50e6dc2
* [NOD-1191] Convert wire protocol to 100% protobuf * [NOD-1191] Simplify wire interface and remove redundant messages * [NOD-1191] Map all proto to wire conversions * [NOD-1203] Create netadapter outside of protocol manager * [NOD-1191] Fix nil errors * [NOD-1191] Fix comments * [NOD-1191] Add converter interface * [NOD-1191] Add missing GetBlockLocator message * [NOD-1191] Change message names that starts with 'get' to 'request' * [NOD-1191] Change message commands values * [NOD-1191] Remove redundant methods * [NOD-1191] Rename message constructors * [NOD-1191] Change message commands to use iota * [NOD-1191] Add missing outputs to protobuf conversion * [NOD-1191] Make block header a required field * [NOD-1191] Rename variables * [NOD-1212] Fix test names * [NOD-1191] Rename flow names * [NOD-1191] Fix infinite loop
75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
// Copyright (c) 2013-2015 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package wire
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/util/mstime"
|
|
"net"
|
|
)
|
|
|
|
// NetAddress defines information about a peer on the network including the time
|
|
// it was last seen, the services it supports, its IP address, and port.
|
|
type NetAddress struct {
|
|
// Last time the address was seen.
|
|
Timestamp mstime.Time
|
|
|
|
// Bitfield which identifies the services supported by the address.
|
|
Services ServiceFlag
|
|
|
|
// IP address of the peer.
|
|
IP net.IP
|
|
|
|
// Port the peer is using. This is encoded in big endian on the wire
|
|
// which differs from most everything else.
|
|
Port uint16
|
|
}
|
|
|
|
// HasService returns whether the specified service is supported by the address.
|
|
func (na *NetAddress) HasService(service ServiceFlag) bool {
|
|
return na.Services&service == service
|
|
}
|
|
|
|
// AddService adds service as a supported service by the peer generating the
|
|
// message.
|
|
func (na *NetAddress) AddService(service ServiceFlag) {
|
|
na.Services |= service
|
|
}
|
|
|
|
// TCPAddress converts the NetAddress to *net.TCPAddr
|
|
func (na *NetAddress) TCPAddress() *net.TCPAddr {
|
|
return &net.TCPAddr{
|
|
IP: na.IP,
|
|
Port: int(na.Port),
|
|
}
|
|
}
|
|
|
|
// NewNetAddressIPPort returns a new NetAddress using the provided IP, port, and
|
|
// supported services with defaults for the remaining fields.
|
|
func NewNetAddressIPPort(ip net.IP, port uint16, services ServiceFlag) *NetAddress {
|
|
return NewNetAddressTimestamp(mstime.Now(), services, ip, port)
|
|
}
|
|
|
|
// NewNetAddressTimestamp returns a new NetAddress using the provided
|
|
// timestamp, IP, port, and supported services. The timestamp is rounded to
|
|
// single millisecond precision.
|
|
func NewNetAddressTimestamp(
|
|
timestamp mstime.Time, services ServiceFlag, ip net.IP, port uint16) *NetAddress {
|
|
// Limit the timestamp to one millisecond precision since the protocol
|
|
// doesn't support better.
|
|
na := NetAddress{
|
|
Timestamp: timestamp,
|
|
Services: services,
|
|
IP: ip,
|
|
Port: port,
|
|
}
|
|
return &na
|
|
}
|
|
|
|
// NewNetAddress returns a new NetAddress using the provided TCP address and
|
|
// supported services with defaults for the remaining fields.
|
|
func NewNetAddress(addr *net.TCPAddr, services ServiceFlag) *NetAddress {
|
|
return NewNetAddressIPPort(addr.IP, uint16(addr.Port), services)
|
|
}
|