kotovalexarian-likes-github/moby--moby
1
0
Fork 0
mirror of https://github.com/moby/moby.git synced 2022-11-09 12:21:53 -05:00
Code Releases Activity

Fix #3149: port mapping of TCP + UDP

Browse source 
Signed-off-by: Tzu-Jung Lee <roylee17@gmail.com>
This commit is contained in:
Tzu-Jung Lee 2014-01-02 15:51:50 -08:00
parent 81e596e272
commit f560b87a86
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
network.go
View file

@ -248,12 +248,12 @@ type PortMapper struct {
} }
func (mapper *PortMapper) Map(ip net.IP, port int, backendAddr net.Addr) error { func (mapper *PortMapper) Map(ip net.IP, port int, backendAddr net.Addr) error {
mapKey := (&net.TCPAddr{Port: port, IP: ip}).String()
if _, exists := mapper.tcpProxies[mapKey]; exists {
return fmt.Errorf("Port %s is already in use", mapKey)
}
if _, isTCP := backendAddr.(*net.TCPAddr); isTCP { if _, isTCP := backendAddr.(*net.TCPAddr); isTCP {
mapKey := (&net.TCPAddr{Port: port, IP: ip}).String()
if _, exists := mapper.tcpProxies[mapKey]; exists {
return fmt.Errorf("TCP Port %s is already in use", mapKey)
}
backendPort := backendAddr.(*net.TCPAddr).Port backendPort := backendAddr.(*net.TCPAddr).Port
backendIP := backendAddr.(*net.TCPAddr).IP backendIP := backendAddr.(*net.TCPAddr).IP
if mapper.iptables != nil { if mapper.iptables != nil {
@ -270,6 +270,10 @@ func (mapper *PortMapper) Map(ip net.IP, port int, backendAddr net.Addr) error {
mapper.tcpProxies[mapKey] = proxy mapper.tcpProxies[mapKey] = proxy
go proxy.Run() go proxy.Run()
} else { } else {
mapKey := (&net.UDPAddr{Port: port, IP: ip}).String()
if _, exists := mapper.udpProxies[mapKey]; exists {
return fmt.Errorf("UDP: Port %s is already in use", mapKey)
}
backendPort := backendAddr.(*net.UDPAddr).Port backendPort := backendAddr.(*net.UDPAddr).Port
backendIP := backendAddr.(*net.UDPAddr).IP backendIP := backendAddr.(*net.UDPAddr).IP
if mapper.iptables != nil { if mapper.iptables != nil {
@ -290,8 +294,8 @@ func (mapper *PortMapper) Map(ip net.IP, port int, backendAddr net.Addr) error {
} }
func (mapper *PortMapper) Unmap(ip net.IP, port int, proto string) error { func (mapper *PortMapper) Unmap(ip net.IP, port int, proto string) error {
mapKey := (&net.TCPAddr{Port: port, IP: ip}).String()
if proto == "tcp" { if proto == "tcp" {
mapKey := (&net.TCPAddr{Port: port, IP: ip}).String()
backendAddr, ok := mapper.tcpMapping[mapKey] backendAddr, ok := mapper.tcpMapping[mapKey]
if !ok { if !ok {
return fmt.Errorf("Port tcp/%s is not mapped", mapKey) return fmt.Errorf("Port tcp/%s is not mapped", mapKey)
@ -307,6 +311,7 @@ func (mapper *PortMapper) Unmap(ip net.IP, port int, proto string) error {
} }
delete(mapper.tcpMapping, mapKey) delete(mapper.tcpMapping, mapKey)
} else { } else {
mapKey := (&net.UDPAddr{Port: port, IP: ip}).String()
backendAddr, ok := mapper.udpMapping[mapKey] backendAddr, ok := mapper.udpMapping[mapKey]
if !ok { if !ok {
return fmt.Errorf("Port udp/%s is not mapped", mapKey) return fmt.Errorf("Port udp/%s is not mapped", mapKey)