forked from v2ray/v2ray-core
-
Notifications
You must be signed in to change notification settings - Fork 0
/
port.go
76 lines (62 loc) · 1.87 KB
/
port.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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package net
import (
"strconv"
"v2ray.com/core/common/errors"
"v2ray.com/core/common/serial"
)
var (
// ErrInvalidPortRage indicates an error during port range parsing.
ErrInvalidPortRange = errors.New("Invalid port range.")
)
// Port represents a network port in TCP and UDP protocol.
type Port uint16
// PortFromBytes converts a byte array to a Port, assuming bytes are in big endian order.
// @unsafe Caller must ensure that the byte array has at least 2 elements.
func PortFromBytes(port []byte) Port {
return Port(serial.BytesToUint16(port))
}
// PortFromInt converts an integer to a Port.
// @error when the integer is not positive or larger then 65535
func PortFromInt(val uint32) (Port, error) {
if val > 65535 {
return Port(0), ErrInvalidPortRange
}
return Port(val), nil
}
// PortFromString converts a string to a Port.
// @error when the string is not an integer or the integral value is a not a valid Port.
func PortFromString(s string) (Port, error) {
val, err := strconv.ParseUint(s, 10, 32)
if err != nil {
return Port(0), ErrInvalidPortRange
}
return PortFromInt(uint32(val))
}
// Value return the correspoding uint16 value of v Port.
func (v Port) Value() uint16 {
return uint16(v)
}
// Bytes returns the correspoding bytes of v Port, in big endian order.
func (v Port) Bytes(b []byte) []byte {
return serial.Uint16ToBytes(v.Value(), b)
}
// String returns the string presentation of v Port.
func (v Port) String() string {
return serial.Uint16ToString(v.Value())
}
func (v PortRange) FromPort() Port {
return Port(v.From)
}
func (v PortRange) ToPort() Port {
return Port(v.To)
}
// Contains returns true if the given port is within the range of v PortRange.
func (v PortRange) Contains(port Port) bool {
return v.FromPort() <= port && port <= v.ToPort()
}
func SinglePortRange(v Port) *PortRange {
return &PortRange{
From: uint32(v),
To: uint32(v),
}
}