v2ray-core/common/net/port.go

68 lines
1.8 KiB
Go

package net
import (
"errors"
"strconv"
"github.com/v2ray/v2ray-core/common/serial"
)
var (
// ErrorInvalidPortRage indicates an error during port range parsing.
ErrorInvalidPortRange = errors.New("Invalid port range.")
)
// Port represents a network port in TCP and UDP protocol.
type Port serial.Uint16Literal
// 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.BytesLiteral(port).Uint16Value())
}
// PortFromInt converts an integer to a Port.
// @error when the integer is not positive or larger then 65535
func PortFromInt(v int) (Port, error) {
if v <= 0 || v > 65535 {
return Port(0), ErrorInvalidPortRange
}
return Port(v), 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) {
v, err := strconv.Atoi(s)
if err != nil {
return Port(0), ErrorInvalidPortRange
}
return PortFromInt(v)
}
// Value return the correspoding uint16 value of this Port.
func (this Port) Value() uint16 {
return uint16(this)
}
// Bytes returns the correspoding bytes of this Port, in big endian order.
func (this Port) Bytes() []byte {
return serial.Uint16Literal(this).Bytes()
}
// String returns the string presentation of this Port.
func (this Port) String() string {
return serial.Uint16Literal(this).String()
}
// PortRange represents a range of ports.
type PortRange struct {
From Port
To Port
}
// Contains returns true if the given port is within the range of this PortRange.
func (this PortRange) Contains(port Port) bool {
return this.From <= port && port <= this.To
}