nat_test.go 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145
  1. package container // import "github.com/docker/docker/integration/container"
  2. import (
  3. "bytes"
  4. "context"
  5. "fmt"
  6. "io"
  7. "net"
  8. "strings"
  9. "testing"
  10. "time"
  11. "github.com/docker/docker/api/types"
  12. "github.com/docker/docker/integration/internal/container"
  13. "github.com/docker/go-connections/nat"
  14. "gotest.tools/v3/assert"
  15. is "gotest.tools/v3/assert/cmp"
  16. "gotest.tools/v3/poll"
  17. "gotest.tools/v3/skip"
  18. )
  19. func TestNetworkNat(t *testing.T) {
  20. skip.If(t, testEnv.DaemonInfo.OSType == "windows", "FIXME")
  21. skip.If(t, testEnv.IsRemoteDaemon)
  22. defer setupTest(t)()
  23. msg := "it works"
  24. startServerContainer(t, msg, 8080)
  25. endpoint := getExternalAddress(t)
  26. conn, err := net.Dial("tcp", net.JoinHostPort(endpoint.String(), "8080"))
  27. assert.NilError(t, err)
  28. defer conn.Close()
  29. data, err := io.ReadAll(conn)
  30. assert.NilError(t, err)
  31. assert.Check(t, is.Equal(msg, strings.TrimSpace(string(data))))
  32. }
  33. func TestNetworkLocalhostTCPNat(t *testing.T) {
  34. skip.If(t, testEnv.IsRemoteDaemon)
  35. defer setupTest(t)()
  36. msg := "hi yall"
  37. startServerContainer(t, msg, 8081)
  38. conn, err := net.Dial("tcp", "localhost:8081")
  39. assert.NilError(t, err)
  40. defer conn.Close()
  41. data, err := io.ReadAll(conn)
  42. assert.NilError(t, err)
  43. assert.Check(t, is.Equal(msg, strings.TrimSpace(string(data))))
  44. }
  45. func TestNetworkLoopbackNat(t *testing.T) {
  46. skip.If(t, testEnv.GitHubActions, "FIXME: https://github.com/moby/moby/issues/41561")
  47. skip.If(t, testEnv.DaemonInfo.OSType == "windows", "FIXME")
  48. skip.If(t, testEnv.IsRemoteDaemon)
  49. defer setupTest(t)()
  50. msg := "it works"
  51. serverContainerID := startServerContainer(t, msg, 8080)
  52. endpoint := getExternalAddress(t)
  53. apiClient := testEnv.APIClient()
  54. ctx := context.Background()
  55. cID := container.Run(ctx, t, apiClient,
  56. container.WithCmd("sh", "-c", fmt.Sprintf("stty raw && nc -w 1 %s 8080", endpoint.String())),
  57. container.WithTty(true),
  58. container.WithNetworkMode("container:"+serverContainerID),
  59. )
  60. poll.WaitOn(t, container.IsStopped(ctx, apiClient, cID), poll.WithDelay(100*time.Millisecond))
  61. body, err := apiClient.ContainerLogs(ctx, cID, types.ContainerLogsOptions{
  62. ShowStdout: true,
  63. })
  64. assert.NilError(t, err)
  65. defer body.Close()
  66. var b bytes.Buffer
  67. _, err = io.Copy(&b, body)
  68. assert.NilError(t, err)
  69. assert.Check(t, is.Equal(msg, strings.TrimSpace(b.String())))
  70. }
  71. func startServerContainer(t *testing.T, msg string, port int) string {
  72. t.Helper()
  73. apiClient := testEnv.APIClient()
  74. ctx := context.Background()
  75. cID := container.Run(ctx, t, apiClient,
  76. container.WithName("server-"+t.Name()),
  77. container.WithCmd("sh", "-c", fmt.Sprintf("echo %q | nc -lp %d", msg, port)),
  78. container.WithExposedPorts(fmt.Sprintf("%d/tcp", port)),
  79. func(c *container.TestContainerConfig) {
  80. c.HostConfig.PortBindings = nat.PortMap{
  81. nat.Port(fmt.Sprintf("%d/tcp", port)): []nat.PortBinding{
  82. {
  83. HostPort: fmt.Sprintf("%d", port),
  84. },
  85. },
  86. }
  87. })
  88. poll.WaitOn(t, container.IsInState(ctx, apiClient, cID, "running"), poll.WithDelay(100*time.Millisecond))
  89. return cID
  90. }
  91. // getExternalAddress() returns the external IP-address from eth0. If eth0 has
  92. // multiple IP-addresses, it returns the first IPv4 IP-address; if no IPv4
  93. // address is present, it returns the first IP-address found.
  94. func getExternalAddress(t *testing.T) net.IP {
  95. t.Helper()
  96. iface, err := net.InterfaceByName("eth0")
  97. skip.If(t, err != nil, "Test not running with `make test-integration`. Interface eth0 not found: %s", err)
  98. ifaceAddrs, err := iface.Addrs()
  99. assert.NilError(t, err)
  100. assert.Check(t, 0 != len(ifaceAddrs))
  101. if len(ifaceAddrs) > 1 {
  102. // Prefer IPv4 address if multiple addresses found, as rootlesskit
  103. // does not handle IPv6 currently https://github.com/moby/moby/pull/41908#issuecomment-774200001
  104. for _, a := range ifaceAddrs {
  105. ifaceIP, _, err := net.ParseCIDR(a.String())
  106. assert.NilError(t, err)
  107. if ifaceIP.To4() != nil {
  108. return ifaceIP
  109. }
  110. }
  111. }
  112. ifaceIP, _, err := net.ParseCIDR(ifaceAddrs[0].String())
  113. assert.NilError(t, err)
  114. return ifaceIP
  115. }