mirror of https://github.com/slackhq/nebula.git
101 lines
1.5 KiB
Go
101 lines
1.5 KiB
Go
package udp
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net"
|
|
"strconv"
|
|
)
|
|
|
|
type m map[string]interface{}
|
|
|
|
type Addr struct {
|
|
IP net.IP
|
|
Port uint16
|
|
}
|
|
|
|
func NewAddr(ip net.IP, port uint16) *Addr {
|
|
addr := Addr{IP: make([]byte, net.IPv6len), Port: port}
|
|
copy(addr.IP, ip.To16())
|
|
return &addr
|
|
}
|
|
|
|
func NewAddrFromString(s string) *Addr {
|
|
ip, port, err := ParseIPAndPort(s)
|
|
//TODO: handle err
|
|
_ = err
|
|
return &Addr{IP: ip.To16(), Port: port}
|
|
}
|
|
|
|
func (ua *Addr) Equals(t *Addr) bool {
|
|
if t == nil || ua == nil {
|
|
return t == nil && ua == nil
|
|
}
|
|
return ua.IP.Equal(t.IP) && ua.Port == t.Port
|
|
}
|
|
|
|
func (ua *Addr) String() string {
|
|
if ua == nil {
|
|
return "<nil>"
|
|
}
|
|
|
|
return net.JoinHostPort(ua.IP.String(), fmt.Sprintf("%v", ua.Port))
|
|
}
|
|
|
|
func (ua *Addr) MarshalJSON() ([]byte, error) {
|
|
if ua == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
return json.Marshal(m{"ip": ua.IP, "port": ua.Port})
|
|
}
|
|
|
|
func (ua *Addr) Copy() *Addr {
|
|
if ua == nil {
|
|
return nil
|
|
}
|
|
|
|
nu := Addr{
|
|
Port: ua.Port,
|
|
IP: make(net.IP, len(ua.IP)),
|
|
}
|
|
|
|
copy(nu.IP, ua.IP)
|
|
return &nu
|
|
}
|
|
|
|
type AddrSlice []*Addr
|
|
|
|
func (a AddrSlice) Equal(b AddrSlice) bool {
|
|
if len(a) != len(b) {
|
|
return false
|
|
}
|
|
|
|
for i := range a {
|
|
if !a[i].Equals(b[i]) {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func ParseIPAndPort(s string) (net.IP, uint16, error) {
|
|
rIp, sPort, err := net.SplitHostPort(s)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
addr, err := net.ResolveIPAddr("ip", rIp)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
iPort, err := strconv.Atoi(sPort)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
return addr.IP, uint16(iPort), nil
|
|
}
|