mirror of https://github.com/v2ray/v2ray-core
77 lines
1.9 KiB
Go
77 lines
1.9 KiB
Go
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),
|
|
}
|
|
}
|