9a0c72b2d1
I believe this was failing because 'nc' wouldn't show the data it received sometimes. So intead of looking for that data we now look for the output of the echo that comes after the nc command successfully runs Signed-off-by: Doug Davis <dug@us.ibm.com> Upstream-commit: 769df832a39e2507fb46cb8896666a1c3197ff0b Component: engine
117 lines
3.4 KiB
Go
117 lines
3.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/go-check/check"
|
|
)
|
|
|
|
func startServerContainer(c *check.C, proto string, port int) string {
|
|
pStr := fmt.Sprintf("%d:%d", port, port)
|
|
bCmd := fmt.Sprintf("nc -lp %d && echo bye", port)
|
|
cmd := []string{"-d", "-p", pStr, "busybox", "sh", "-c", bCmd}
|
|
if proto == "udp" {
|
|
cmd = append(cmd, "-u")
|
|
}
|
|
|
|
name := "server"
|
|
if err := waitForContainer(name, cmd...); err != nil {
|
|
c.Fatalf("Failed to launch server container: %v", err)
|
|
}
|
|
return name
|
|
}
|
|
|
|
func getExternalAddress(c *check.C) net.IP {
|
|
iface, err := net.InterfaceByName("eth0")
|
|
if err != nil {
|
|
c.Skip(fmt.Sprintf("Test not running with `make test`. Interface eth0 not found: %v", err))
|
|
}
|
|
|
|
ifaceAddrs, err := iface.Addrs()
|
|
if err != nil || len(ifaceAddrs) == 0 {
|
|
c.Fatalf("Error retrieving addresses for eth0: %v (%d addresses)", err, len(ifaceAddrs))
|
|
}
|
|
|
|
ifaceIP, _, err := net.ParseCIDR(ifaceAddrs[0].String())
|
|
if err != nil {
|
|
c.Fatalf("Error retrieving the up for eth0: %s", err)
|
|
}
|
|
|
|
return ifaceIP
|
|
}
|
|
|
|
func getContainerLogs(c *check.C, containerID string) string {
|
|
runCmd := exec.Command(dockerBinary, "logs", containerID)
|
|
out, _, err := runCommandWithOutput(runCmd)
|
|
if err != nil {
|
|
c.Fatal(out, err)
|
|
}
|
|
return strings.Trim(out, "\r\n")
|
|
}
|
|
|
|
func getContainerStatus(c *check.C, containerID string) string {
|
|
runCmd := exec.Command(dockerBinary, "inspect", "-f", "{{.State.Running}}", containerID)
|
|
out, _, err := runCommandWithOutput(runCmd)
|
|
if err != nil {
|
|
c.Fatal(out, err)
|
|
}
|
|
return strings.Trim(out, "\r\n")
|
|
}
|
|
|
|
func (s *DockerSuite) TestNetworkNat(c *check.C) {
|
|
testRequires(c, SameHostDaemon, NativeExecDriver)
|
|
defer deleteAllContainers()
|
|
|
|
srv := startServerContainer(c, "tcp", 8080)
|
|
|
|
// Spawn a new container which connects to the server through the
|
|
// interface address.
|
|
endpoint := getExternalAddress(c)
|
|
runCmd := exec.Command(dockerBinary, "run", "busybox", "sh", "-c", fmt.Sprintf("echo hello world | nc -w 30 %s 8080", endpoint))
|
|
if out, _, err := runCommandWithOutput(runCmd); err != nil {
|
|
c.Fatalf("Failed to connect to server: %v (output: %q)", err, string(out))
|
|
}
|
|
|
|
result := getContainerLogs(c, srv)
|
|
|
|
// Ideally we'd like to check for "hello world" but sometimes
|
|
// nc doesn't show the data it received so instead let's look for
|
|
// the output of the 'echo bye' that should be printed once
|
|
// the nc command gets a connection
|
|
expected := "bye"
|
|
if !strings.Contains(result, expected) {
|
|
c.Fatalf("Unexpected output. Expected: %q, received: %q", expected, result)
|
|
}
|
|
}
|
|
|
|
func (s *DockerSuite) TestNetworkLocalhostTCPNat(c *check.C) {
|
|
testRequires(c, SameHostDaemon, NativeExecDriver)
|
|
defer deleteAllContainers()
|
|
|
|
srv := startServerContainer(c, "tcp", 8081)
|
|
|
|
// Attempt to connect from the host to the listening container.
|
|
conn, err := net.Dial("tcp", "localhost:8081")
|
|
if err != nil {
|
|
c.Fatalf("Failed to connect to container (%v)", err)
|
|
}
|
|
if _, err := conn.Write([]byte("hello world\n")); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
conn.Close()
|
|
|
|
result := getContainerLogs(c, srv)
|
|
|
|
// Ideally we'd like to check for "hello world" but sometimes
|
|
// nc doesn't show the data it received so instead let's look for
|
|
// the output of the 'echo bye' that should be printed once
|
|
// the nc command gets a connection
|
|
expected := "bye"
|
|
if !strings.Contains(result, expected) {
|
|
c.Fatalf("Unexpected output. Expected: %q, received: %q", expected, result)
|
|
}
|
|
}
|