portallocator: rewrite to simplify, removes race condition
Docker-DCO-1.1-Signed-off-by: Erik Hollensbe <github@hollensbe.org> (github: erikh)
This commit is contained in:
parent
a94a87778c
commit
f0489ce3a9
|
@ -2,21 +2,21 @@ package portallocator
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"github.com/dotcloud/docker/pkg/collections"
|
|
||||||
"net"
|
"net"
|
||||||
"sync"
|
"sync"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
portMap map[int]bool
|
||||||
|
protocolMap map[string]portMap
|
||||||
|
ipMapping map[string]protocolMap
|
||||||
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
BeginPortRange = 49153
|
BeginPortRange = 49153
|
||||||
EndPortRange = 65535
|
EndPortRange = 65535
|
||||||
)
|
)
|
||||||
|
|
||||||
type (
|
|
||||||
portMappings map[string]*collections.OrderedIntSet
|
|
||||||
ipMapping map[string]portMappings
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
var (
|
||||||
ErrAllPortsAllocated = errors.New("all ports are allocated")
|
ErrAllPortsAllocated = errors.New("all ports are allocated")
|
||||||
ErrPortAlreadyAllocated = errors.New("port has already been allocated")
|
ErrPortAlreadyAllocated = errors.New("port has already been allocated")
|
||||||
|
@ -24,165 +24,106 @@ var (
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
currentDynamicPort = map[string]int{
|
mutex sync.Mutex
|
||||||
"tcp": BeginPortRange - 1,
|
|
||||||
"udp": BeginPortRange - 1,
|
defaultIP = net.ParseIP("0.0.0.0")
|
||||||
}
|
globalMap = ipMapping{}
|
||||||
defaultIP = net.ParseIP("0.0.0.0")
|
|
||||||
defaultAllocatedPorts = portMappings{}
|
|
||||||
otherAllocatedPorts = ipMapping{}
|
|
||||||
lock = sync.Mutex{}
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
|
||||||
defaultAllocatedPorts["tcp"] = collections.NewOrderedIntSet()
|
|
||||||
defaultAllocatedPorts["udp"] = collections.NewOrderedIntSet()
|
|
||||||
}
|
|
||||||
|
|
||||||
// RequestPort returns an available port if the port is 0
|
|
||||||
// If the provided port is not 0 then it will be checked if
|
|
||||||
// it is available for allocation
|
|
||||||
func RequestPort(ip net.IP, proto string, port int) (int, error) {
|
func RequestPort(ip net.IP, proto string, port int) (int, error) {
|
||||||
lock.Lock()
|
mutex.Lock()
|
||||||
defer lock.Unlock()
|
defer mutex.Unlock()
|
||||||
|
|
||||||
if err := validateProtocol(proto); err != nil {
|
if err := validateProto(proto); err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
|
|
||||||
// If the user requested a specific port to be allocated
|
ip = getDefault(ip)
|
||||||
|
|
||||||
|
mapping := getOrCreate(ip)
|
||||||
|
|
||||||
if port > 0 {
|
if port > 0 {
|
||||||
if err := registerSetPort(ip, proto, port); err != nil {
|
if !mapping[proto][port] {
|
||||||
|
mapping[proto][port] = true
|
||||||
|
return port, nil
|
||||||
|
} else {
|
||||||
|
return 0, ErrPortAlreadyAllocated
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
port, err := findPort(ip, proto)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return port, nil
|
return port, nil
|
||||||
}
|
}
|
||||||
return registerDynamicPort(ip, proto)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ReleasePort will return the provided port back into the
|
|
||||||
// pool for reuse
|
|
||||||
func ReleasePort(ip net.IP, proto string, port int) error {
|
func ReleasePort(ip net.IP, proto string, port int) error {
|
||||||
lock.Lock()
|
mutex.Lock()
|
||||||
defer lock.Unlock()
|
defer mutex.Unlock()
|
||||||
|
|
||||||
if err := validateProtocol(proto); err != nil {
|
ip = getDefault(ip)
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
allocated := defaultAllocatedPorts[proto]
|
mapping := getOrCreate(ip)
|
||||||
allocated.Remove(port)
|
delete(mapping[proto], port)
|
||||||
|
|
||||||
if !equalsDefault(ip) {
|
|
||||||
registerIP(ip)
|
|
||||||
|
|
||||||
// Remove the port for the specific ip address
|
|
||||||
allocated = otherAllocatedPorts[ip.String()][proto]
|
|
||||||
allocated.Remove(port)
|
|
||||||
}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func ReleaseAll() error {
|
func ReleaseAll() error {
|
||||||
lock.Lock()
|
mutex.Lock()
|
||||||
defer lock.Unlock()
|
defer mutex.Unlock()
|
||||||
|
|
||||||
currentDynamicPort["tcp"] = BeginPortRange - 1
|
globalMap = ipMapping{}
|
||||||
currentDynamicPort["udp"] = BeginPortRange - 1
|
|
||||||
|
|
||||||
defaultAllocatedPorts = portMappings{}
|
|
||||||
defaultAllocatedPorts["tcp"] = collections.NewOrderedIntSet()
|
|
||||||
defaultAllocatedPorts["udp"] = collections.NewOrderedIntSet()
|
|
||||||
|
|
||||||
otherAllocatedPorts = ipMapping{}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func registerDynamicPort(ip net.IP, proto string) (int, error) {
|
func getOrCreate(ip net.IP) protocolMap {
|
||||||
|
ipstr := ip.String()
|
||||||
|
|
||||||
if !equalsDefault(ip) {
|
if _, ok := globalMap[ipstr]; !ok {
|
||||||
registerIP(ip)
|
globalMap[ipstr] = protocolMap{
|
||||||
|
"tcp": portMap{},
|
||||||
ipAllocated := otherAllocatedPorts[ip.String()][proto]
|
"udp": portMap{},
|
||||||
|
|
||||||
port, err := findNextPort(proto, ipAllocated)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
}
|
||||||
ipAllocated.Push(port)
|
|
||||||
return port, nil
|
|
||||||
|
|
||||||
} else {
|
|
||||||
|
|
||||||
allocated := defaultAllocatedPorts[proto]
|
|
||||||
|
|
||||||
port, err := findNextPort(proto, allocated)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
allocated.Push(port)
|
|
||||||
return port, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func registerSetPort(ip net.IP, proto string, port int) error {
|
|
||||||
allocated := defaultAllocatedPorts[proto]
|
|
||||||
if allocated.Exists(port) {
|
|
||||||
return ErrPortAlreadyAllocated
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if !equalsDefault(ip) {
|
return globalMap[ipstr]
|
||||||
registerIP(ip)
|
|
||||||
|
|
||||||
ipAllocated := otherAllocatedPorts[ip.String()][proto]
|
|
||||||
if ipAllocated.Exists(port) {
|
|
||||||
return ErrPortAlreadyAllocated
|
|
||||||
}
|
|
||||||
ipAllocated.Push(port)
|
|
||||||
} else {
|
|
||||||
allocated.Push(port)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func equalsDefault(ip net.IP) bool {
|
func findPort(ip net.IP, proto string) (int, error) {
|
||||||
return ip == nil || ip.Equal(defaultIP)
|
port := BeginPortRange
|
||||||
}
|
|
||||||
|
|
||||||
func findNextPort(proto string, allocated *collections.OrderedIntSet) (int, error) {
|
mapping := getOrCreate(ip)
|
||||||
port := nextPort(proto)
|
|
||||||
startSearchPort := port
|
for mapping[proto][port] {
|
||||||
for allocated.Exists(port) {
|
port++
|
||||||
port = nextPort(proto)
|
|
||||||
if startSearchPort == port {
|
if port > EndPortRange {
|
||||||
return 0, ErrAllPortsAllocated
|
return 0, ErrAllPortsAllocated
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
mapping[proto][port] = true
|
||||||
|
|
||||||
return port, nil
|
return port, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func nextPort(proto string) int {
|
func getDefault(ip net.IP) net.IP {
|
||||||
c := currentDynamicPort[proto] + 1
|
if ip == nil {
|
||||||
if c > EndPortRange {
|
return defaultIP
|
||||||
c = BeginPortRange
|
|
||||||
}
|
}
|
||||||
currentDynamicPort[proto] = c
|
|
||||||
return c
|
return ip
|
||||||
}
|
}
|
||||||
|
|
||||||
func registerIP(ip net.IP) {
|
func validateProto(proto string) error {
|
||||||
if _, exists := otherAllocatedPorts[ip.String()]; !exists {
|
if proto != "tcp" && proto != "udp" {
|
||||||
otherAllocatedPorts[ip.String()] = portMappings{
|
|
||||||
"tcp": collections.NewOrderedIntSet(),
|
|
||||||
"udp": collections.NewOrderedIntSet(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func validateProtocol(proto string) error {
|
|
||||||
if _, exists := defaultAllocatedPorts[proto]; !exists {
|
|
||||||
return ErrUnknownProtocol
|
return ErrUnknownProtocol
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue