mapper_test.go 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152
  1. package portmapper
  2. import (
  3. "net"
  4. "testing"
  5. "github.com/docker/docker/daemon/networkdriver/portallocator"
  6. "github.com/docker/docker/pkg/iptables"
  7. )
  8. func init() {
  9. // override this func to mock out the proxy server
  10. NewProxy = NewMockProxyCommand
  11. }
  12. func reset() {
  13. chain = nil
  14. currentMappings = make(map[string]*mapping)
  15. }
  16. func TestSetIptablesChain(t *testing.T) {
  17. defer reset()
  18. c := &iptables.Chain{
  19. Name: "TEST",
  20. Bridge: "192.168.1.1",
  21. }
  22. if chain != nil {
  23. t.Fatal("chain should be nil at init")
  24. }
  25. SetIptablesChain(c)
  26. if chain == nil {
  27. t.Fatal("chain should not be nil after set")
  28. }
  29. }
  30. func TestMapPorts(t *testing.T) {
  31. dstIp1 := net.ParseIP("192.168.0.1")
  32. dstIp2 := net.ParseIP("192.168.0.2")
  33. dstAddr1 := &net.TCPAddr{IP: dstIp1, Port: 80}
  34. dstAddr2 := &net.TCPAddr{IP: dstIp2, Port: 80}
  35. srcAddr1 := &net.TCPAddr{Port: 1080, IP: net.ParseIP("172.16.0.1")}
  36. srcAddr2 := &net.TCPAddr{Port: 1080, IP: net.ParseIP("172.16.0.2")}
  37. addrEqual := func(addr1, addr2 net.Addr) bool {
  38. return (addr1.Network() == addr2.Network()) && (addr1.String() == addr2.String())
  39. }
  40. if host, err := Map(srcAddr1, dstIp1, 80); err != nil {
  41. t.Fatalf("Failed to allocate port: %s", err)
  42. } else if !addrEqual(dstAddr1, host) {
  43. t.Fatalf("Incorrect mapping result: expected %s:%s, got %s:%s",
  44. dstAddr1.String(), dstAddr1.Network(), host.String(), host.Network())
  45. }
  46. if _, err := Map(srcAddr1, dstIp1, 80); err == nil {
  47. t.Fatalf("Port is in use - mapping should have failed")
  48. }
  49. if _, err := Map(srcAddr2, dstIp1, 80); err == nil {
  50. t.Fatalf("Port is in use - mapping should have failed")
  51. }
  52. if _, err := Map(srcAddr2, dstIp2, 80); err != nil {
  53. t.Fatalf("Failed to allocate port: %s", err)
  54. }
  55. if Unmap(dstAddr1) != nil {
  56. t.Fatalf("Failed to release port")
  57. }
  58. if Unmap(dstAddr2) != nil {
  59. t.Fatalf("Failed to release port")
  60. }
  61. if Unmap(dstAddr2) == nil {
  62. t.Fatalf("Port already released, but no error reported")
  63. }
  64. }
  65. func TestGetUDPKey(t *testing.T) {
  66. addr := &net.UDPAddr{IP: net.ParseIP("192.168.1.5"), Port: 53}
  67. key := getKey(addr)
  68. if expected := "192.168.1.5:53/udp"; key != expected {
  69. t.Fatalf("expected key %s got %s", expected, key)
  70. }
  71. }
  72. func TestGetTCPKey(t *testing.T) {
  73. addr := &net.TCPAddr{IP: net.ParseIP("192.168.1.5"), Port: 80}
  74. key := getKey(addr)
  75. if expected := "192.168.1.5:80/tcp"; key != expected {
  76. t.Fatalf("expected key %s got %s", expected, key)
  77. }
  78. }
  79. func TestGetUDPIPAndPort(t *testing.T) {
  80. addr := &net.UDPAddr{IP: net.ParseIP("192.168.1.5"), Port: 53}
  81. ip, port := getIPAndPort(addr)
  82. if expected := "192.168.1.5"; ip.String() != expected {
  83. t.Fatalf("expected ip %s got %s", expected, ip)
  84. }
  85. if ep := 53; port != ep {
  86. t.Fatalf("expected port %d got %d", ep, port)
  87. }
  88. }
  89. func TestMapAllPortsSingleInterface(t *testing.T) {
  90. dstIp1 := net.ParseIP("0.0.0.0")
  91. srcAddr1 := &net.TCPAddr{Port: 1080, IP: net.ParseIP("172.16.0.1")}
  92. hosts := []net.Addr{}
  93. var host net.Addr
  94. var err error
  95. defer func() {
  96. for _, val := range hosts {
  97. Unmap(val)
  98. }
  99. }()
  100. for i := 0; i < 10; i++ {
  101. for i := portallocator.BeginPortRange; i < portallocator.EndPortRange; i++ {
  102. if host, err = Map(srcAddr1, dstIp1, 0); err != nil {
  103. t.Fatal(err)
  104. }
  105. hosts = append(hosts, host)
  106. }
  107. if _, err := Map(srcAddr1, dstIp1, portallocator.BeginPortRange); err == nil {
  108. t.Fatalf("Port %d should be bound but is not", portallocator.BeginPortRange)
  109. }
  110. for _, val := range hosts {
  111. if err := Unmap(val); err != nil {
  112. t.Fatal(err)
  113. }
  114. }
  115. hosts = []net.Addr{}
  116. }
  117. }