-
Notifications
You must be signed in to change notification settings - Fork 657
/
ip.go
57 lines (51 loc) · 1.46 KB
/
ip.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
// Copyright (C) 2019-2024, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package ips
import "net/netip"
// IsPublic returns true if the provided address is considered to be a public
// IP.
func IsPublic(addr netip.Addr) bool {
return addr.IsGlobalUnicast() && !addr.IsPrivate()
}
// ParseAddr returns the IP address from the provided string. If the string
// represents an IPv4 address in an IPv6 address, the IPv4 address is returned.
func ParseAddr(s string) (netip.Addr, error) {
addr, err := netip.ParseAddr(s)
if err != nil {
return netip.Addr{}, err
}
if addr.Is4In6() {
addr = addr.Unmap()
}
return addr, nil
}
// ParseAddrPort returns the IP:port address from the provided string. If the
// string represents an IPv4 address in an IPv6 address, the IPv4 address is
// returned.
func ParseAddrPort(s string) (netip.AddrPort, error) {
addrPort, err := netip.ParseAddrPort(s)
if err != nil {
return netip.AddrPort{}, err
}
addr := addrPort.Addr()
if addr.Is4In6() {
addrPort = netip.AddrPortFrom(
addr.Unmap(),
addrPort.Port(),
)
}
return addrPort, nil
}
// AddrFromSlice returns the IP address from the provided byte slice. If the
// byte slice represents an IPv4 address in an IPv6 address, the IPv4 address is
// returned.
func AddrFromSlice(b []byte) (netip.Addr, bool) {
addr, ok := netip.AddrFromSlice(b)
if !ok {
return netip.Addr{}, false
}
if addr.Is4In6() {
addr = addr.Unmap()
}
return addr, true
}