moby--moby/libnetwork/drivers/bridge/setup_ip_forwarding_test.go

package bridge

import (
	"bytes"
	"io/ioutil"
	"testing"
)

func TestSetupIPForwarding(t *testing.T) {
	// Read current setting and ensure the original value gets restored
	procSetting := readCurrentIPForwardingSetting(t)
	defer reconcileIPForwardingSetting(t, procSetting)

	// Disable IP Forwarding if enabled
	if bytes.Compare(procSetting, []byte("1\n")) == 0 {
		writeIPForwardingSetting(t, []byte{'0', '\n'})
	}

	// Create test interface with ip forwarding setting enabled
	config := &Configuration{
		BridgeName:         DefaultBridgeName,
		EnableIPForwarding: true}
	br := &bridgeInterface{}

	// Set IP Forwarding
	if err := setupIPForwarding(config, br); err != nil {
		t.Fatalf("Failed to setup IP forwarding: %v", err)
	}

	// Read new setting
	procSetting = readCurrentIPForwardingSetting(t)
	if bytes.Compare(procSetting, []byte("1\n")) != 0 {
		t.Fatalf("Failed to effectively setup IP forwarding")
	}
}

func TestUnexpectedSetupIPForwarding(t *testing.T) {
	// Read current setting and ensure the original value gets restored
	procSetting := readCurrentIPForwardingSetting(t)
	defer reconcileIPForwardingSetting(t, procSetting)

	// Create test interface without ip forwarding setting enabled
	config := &Configuration{
		BridgeName:         DefaultBridgeName,
		EnableIPForwarding: false}
	br := &bridgeInterface{}

	// Attempt Set IP Forwarding
	err := setupIPForwarding(config, br)
	if err == nil {
		t.Fatal("Setup IP forwarding was expected to fail")
	}

	if _, ok := err.(*ipForwardCfgError); !ok {
		t.Fatalf("Setup IP forwarding failed with unexpected error: %v", err)
	}
}

func readCurrentIPForwardingSetting(t *testing.T) []byte {
	procSetting, err := ioutil.ReadFile(ipv4ForwardConf)
	if err != nil {
		t.Fatalf("Can't execute test: Failed to read current IP forwarding setting: %v", err)
	}
	return procSetting
}

func writeIPForwardingSetting(t *testing.T, chars []byte) {
	err := ioutil.WriteFile(ipv4ForwardConf, chars, ipv4ForwardConfPerm)
	if err != nil {
		t.Fatalf("Can't execute or cleanup after test: Failed to reset IP forwarding: %v", err)
	}
}

func reconcileIPForwardingSetting(t *testing.T, original []byte) {
	current := readCurrentIPForwardingSetting(t)
	if bytes.Compare(original, current) != 0 {
		writeIPForwardingSetting(t, original)
	}
}