package network

import (
	"context"
	"fmt"
	"testing"
	"time"

	"github.com/docker/docker/api/types"
	"github.com/docker/docker/api/types/network"
	"github.com/docker/docker/client"
	"github.com/docker/docker/integration-cli/daemon"
	"github.com/docker/docker/integration/internal/container"
	"github.com/docker/docker/pkg/parsers/kernel"
	"github.com/gotestyourself/gotestyourself/assert"
	"github.com/gotestyourself/gotestyourself/assert/cmp"
	"github.com/gotestyourself/gotestyourself/icmd"
	"github.com/gotestyourself/gotestyourself/skip"
)

func TestDockerNetworkMacvlanPersistance(t *testing.T) {
	// verify the driver automatically provisions the 802.1q link (dm-dummy0.60)
	skip.If(t, testEnv.DaemonInfo.OSType != "linux")
	skip.If(t, testEnv.IsRemoteDaemon())
	skip.If(t, !macvlanKernelSupport(), "Kernel doesn't support macvlan")

	d := daemon.New(t, "", "dockerd", daemon.Config{})
	d.StartWithBusybox(t)
	defer d.Stop(t)

	master := "dm-dummy0"
	createMasterDummy(t, master)
	defer deleteInterface(t, master)

	client, err := d.NewClient()
	assert.NilError(t, err)

	_, err = client.NetworkCreate(context.Background(), "dm-persist", types.NetworkCreate{
		Driver: "macvlan",
		Options: map[string]string{
			"parent": "dm-dummy0.60",
		},
	})
	assert.NilError(t, err)
	assert.Check(t, isNetworkAvailable(client, "dm-persist"))
	d.Restart(t)
	assert.Check(t, isNetworkAvailable(client, "dm-persist"))
}

func TestDockerNetworkMacvlanOverlapParent(t *testing.T) {
	// verify the same parent interface cannot be used if already in use by an existing network
	skip.If(t, testEnv.DaemonInfo.OSType != "linux")
	skip.If(t, testEnv.IsRemoteDaemon())
	skip.If(t, !macvlanKernelSupport(), "Kernel doesn't support macvlan")

	d := daemon.New(t, "", "dockerd", daemon.Config{})
	d.StartWithBusybox(t)
	defer d.Stop(t)

	master := "dm-dummy0"
	createMasterDummy(t, master)
	defer deleteInterface(t, master)

	client, err := d.NewClient()
	assert.NilError(t, err)

	_, err = client.NetworkCreate(context.Background(), "dm-subinterface", types.NetworkCreate{
		Driver: "macvlan",
		Options: map[string]string{
			"parent": "dm-dummy0.40",
		},
	})
	assert.NilError(t, err)
	assert.Check(t, isNetworkAvailable(client, "dm-subinterface"))

	_, err = client.NetworkCreate(context.Background(), "dm-parent-net-overlap", types.NetworkCreate{
		Driver: "macvlan",
		Options: map[string]string{
			"parent": "dm-dummy0.40",
		},
	})
	assert.Check(t, err != nil)
	// delete the network while preserving the parent link
	err = client.NetworkRemove(context.Background(), "dm-subinterface")
	assert.NilError(t, err)

	assert.Check(t, isNetworkNotAvailable(client, "dm-subinterface"))
	// verify the network delete did not delete the predefined link
	linkExists(t, "dm-dummy0")
}

func TestDockerNetworkMacvlanSubinterface(t *testing.T) {
	// verify the same parent interface cannot be used if already in use by an existing network
	skip.If(t, testEnv.DaemonInfo.OSType != "linux")
	skip.If(t, testEnv.IsRemoteDaemon())
	skip.If(t, !macvlanKernelSupport(), "Kernel doesn't support macvlan")

	d := daemon.New(t, "", "dockerd", daemon.Config{})
	d.StartWithBusybox(t)
	defer d.Stop(t)

	master := "dm-dummy0"
	createMasterDummy(t, master)
	defer deleteInterface(t, master)
	createVlanInterface(t, master, "dm-dummy0.20", "20")

	client, err := d.NewClient()
	assert.NilError(t, err)

	_, err = client.NetworkCreate(context.Background(), "dm-subinterface", types.NetworkCreate{
		Driver: "macvlan",
		Options: map[string]string{
			"parent": "dm-dummy0.20",
		},
	})
	assert.NilError(t, err)
	assert.Check(t, isNetworkAvailable(client, "dm-subinterface"))

	// delete the network while preserving the parent link
	err = client.NetworkRemove(context.Background(), "dm-subinterface")
	assert.NilError(t, err)

	assert.Check(t, isNetworkNotAvailable(client, "dm-subinterface"))
	// verify the network delete did not delete the predefined link
	linkExists(t, "dm-dummy0.20")
}

func TestDockerNetworkMacvlanBridgeNilParent(t *testing.T) {
	// macvlan bridge mode - dummy parent interface is provisioned dynamically
	skip.If(t, testEnv.DaemonInfo.OSType != "linux")
	skip.If(t, testEnv.IsRemoteDaemon())
	skip.If(t, !macvlanKernelSupport(), "Kernel doesn't support macvlan")

	d := daemon.New(t, "", "dockerd", daemon.Config{})
	d.StartWithBusybox(t)
	defer d.Stop(t)
	client, err := d.NewClient()
	assert.NilError(t, err)

	_, err = client.NetworkCreate(context.Background(), "dm-nil-parent", types.NetworkCreate{
		Driver: "macvlan",
	})
	assert.NilError(t, err)
	assert.Check(t, isNetworkAvailable(client, "dm-nil-parent"))

	ctx := context.Background()
	container.Run(t, ctx, client, container.WithNetworkMode("dm-nil-parent"), container.WithName(t.Name()+"first"))
	id2 := container.Run(t, ctx, client, container.WithNetworkMode("dm-nil-parent"), container.WithName(t.Name()+"second"))

	_, err = container.Exec(ctx, client, id2, []string{"ping", "-c", "1", t.Name() + "first"})
	assert.Check(t, err == nil)
}

func TestDockerNetworkMacvlanBridgeInternal(t *testing.T) {
	// macvlan bridge mode - dummy parent interface is provisioned dynamically
	skip.If(t, testEnv.DaemonInfo.OSType != "linux")
	skip.If(t, testEnv.IsRemoteDaemon())
	skip.If(t, !macvlanKernelSupport(), "Kernel doesn't support macvlan")

	d := daemon.New(t, "", "dockerd", daemon.Config{})
	d.StartWithBusybox(t)
	defer d.Stop(t)
	client, err := d.NewClient()
	assert.NilError(t, err)

	_, err = client.NetworkCreate(context.Background(), "dm-internal", types.NetworkCreate{
		Driver:   "macvlan",
		Internal: true,
	})
	assert.NilError(t, err)
	assert.Check(t, isNetworkAvailable(client, "dm-internal"))

	ctx := context.Background()
	id1 := container.Run(t, ctx, client, container.WithNetworkMode("dm-internal"), container.WithName(t.Name()+"first"))
	id2 := container.Run(t, ctx, client, container.WithNetworkMode("dm-internal"), container.WithName(t.Name()+"second"))

	timeoutCtx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
	defer cancel()
	_, err = container.Exec(timeoutCtx, client, id1, []string{"ping", "-c", "1", "-w", "1", "8.8.8.8"})
	// FIXME(vdemeester) check the time of error ?
	assert.Check(t, err != nil)
	assert.Check(t, timeoutCtx.Err() == context.DeadlineExceeded)

	_, err = container.Exec(ctx, client, id2, []string{"ping", "-c", "1", t.Name() + "first"})
	assert.Check(t, err == nil)
}

func TestDockerNetworkMacvlanMultiSubnet(t *testing.T) {
	skip.If(t, testEnv.DaemonInfo.OSType != "linux")
	skip.If(t, testEnv.IsRemoteDaemon())
	skip.If(t, !macvlanKernelSupport(), "Kernel doesn't support macvlan")
	t.Skip("Temporarily skipping while investigating sporadic v6 CI issues")

	d := daemon.New(t, "", "dockerd", daemon.Config{})
	d.StartWithBusybox(t)
	defer d.Stop(t)
	client, err := d.NewClient()
	assert.NilError(t, err)

	_, err = client.NetworkCreate(context.Background(), "dualstackbridge", types.NetworkCreate{
		Driver:     "macvlan",
		EnableIPv6: true,
		IPAM: &network.IPAM{
			Config: []network.IPAMConfig{
				{
					Subnet:     "172.28.100.0/24",
					AuxAddress: map[string]string{},
				},
				{
					Subnet:     "172.28.102.0/24",
					Gateway:    "172.28.102.54",
					AuxAddress: map[string]string{},
				},
				{
					Subnet:     "2001:db8:abc2::/64",
					AuxAddress: map[string]string{},
				},
				{
					Subnet:     "2001:db8:abc4::/64",
					Gateway:    "2001:db8:abc4::254",
					AuxAddress: map[string]string{},
				},
			},
		},
	})
	assert.NilError(t, err)
	assert.Check(t, isNetworkAvailable(client, "dualstackbridge"))

	// start dual stack containers and verify the user specified --ip and --ip6 addresses on subnets 172.28.100.0/24 and 2001:db8:abc2::/64
	ctx := context.Background()
	id1 := container.Run(t, ctx, client,
		container.WithNetworkMode("dualstackbridge"),
		container.WithName(t.Name()+"first"),
		container.WithIPv4("dualstackbridge", "172.28.100.20"),
		container.WithIPv6("dualstackbridge", "2001:db8:abc2::20"),
	)
	id2 := container.Run(t, ctx, client,
		container.WithNetworkMode("dualstackbridge"),
		container.WithName(t.Name()+"second"),
		container.WithIPv4("dualstackbridge", "172.28.100.21"),
		container.WithIPv6("dualstackbridge", "2001:db8:abc2::21"),
	)
	c1, err := client.ContainerInspect(ctx, id1)
	assert.NilError(t, err)

	// verify ipv4 connectivity to the explicit --ipv address second to first
	_, err = container.Exec(ctx, client, id2, []string{"ping", "-c", "1", c1.NetworkSettings.Networks["dualstackbridge"].IPAddress})
	assert.NilError(t, err)
	// verify ipv6 connectivity to the explicit --ipv6 address second to first
	_, err = container.Exec(ctx, client, id2, []string{"ping6", "-c", "1", c1.NetworkSettings.Networks["dualstackbridge"].GlobalIPv6Address})
	assert.NilError(t, err)

	// start dual stack containers and verify the user specified --ip and --ip6 addresses on subnets 172.28.102.0/24 and 2001:db8:abc4::/64
	id3 := container.Run(t, ctx, client,
		container.WithNetworkMode("dualstackbridge"),
		container.WithName(t.Name()+"third"),
		container.WithIPv4("dualstackbridge", "172.28.102.20"),
		container.WithIPv6("dualstackbridge", "2001:db8:abc4::20"),
	)
	id4 := container.Run(t, ctx, client,
		container.WithNetworkMode("dualstackbridge"),
		container.WithName(t.Name()+"fourth"),
		container.WithIPv4("dualstackbridge", "172.28.102.21"),
		container.WithIPv6("dualstackbridge", "2001:db8:abc4::21"),
	)
	c3, err := client.ContainerInspect(ctx, id3)
	assert.NilError(t, err)

	// verify ipv4 connectivity to the explicit --ipv address from third to fourth
	_, err = container.Exec(ctx, client, id4, []string{"ping", "-c", "1", c3.NetworkSettings.Networks["dualstackbridge"].IPAddress})
	assert.NilError(t, err)
	// verify ipv6 connectivity to the explicit --ipv6 address from third to fourth
	_, err = container.Exec(ctx, client, id4, []string{"ping6", "-c", "1", c3.NetworkSettings.Networks["dualstackbridge"].GlobalIPv6Address})
	assert.NilError(t, err)

	// Inspect the v4 gateway to ensure the proper default GW was assigned
	assert.Equal(t, c1.NetworkSettings.Networks["dualstackbridge"].Gateway, "172.28.100.1")
	// Inspect the v6 gateway to ensure the proper default GW was assigned
	assert.Equal(t, c1.NetworkSettings.Networks["dualstackbridge"].IPv6Gateway, "2001:db8:abc2::1")
	// Inspect the v4 gateway to ensure the proper explicitly assigned default GW was assigned
	assert.Equal(t, c3.NetworkSettings.Networks["dualstackbridge"].Gateway, "172.28.102.254")
	// Inspect the v6 gateway to ensure the proper explicitly assigned default GW was assigned
	assert.Equal(t, c3.NetworkSettings.Networks["dualstackbridge"].IPv6Gateway, "2001:db8.abc4::254")
}

func isNetworkAvailable(c client.NetworkAPIClient, name string) cmp.Comparison {
	return func() cmp.Result {
		networks, err := c.NetworkList(context.Background(), types.NetworkListOptions{})
		if err != nil {
			return cmp.ResultFromError(err)
		}
		for _, network := range networks {
			if network.Name == name {
				return cmp.ResultSuccess
			}
		}
		return cmp.ResultFailure(fmt.Sprintf("could not find network %s", name))
	}
}

func isNetworkNotAvailable(c client.NetworkAPIClient, name string) cmp.Comparison {
	return func() cmp.Result {
		networks, err := c.NetworkList(context.Background(), types.NetworkListOptions{})
		if err != nil {
			return cmp.ResultFromError(err)
		}
		for _, network := range networks {
			if network.Name == name {
				return cmp.ResultFailure(fmt.Sprintf("network %s is still present", name))
			}
		}
		return cmp.ResultSuccess
	}
}

func createMasterDummy(t *testing.T, master string) {
	// ip link add <dummy_name> type dummy
	icmd.RunCommand("ip", "link", "add", master, "type", "dummy").Assert(t, icmd.Success)
	icmd.RunCommand("ip", "link", "set", master, "up").Assert(t, icmd.Success)
}

func createVlanInterface(t *testing.T, master, slave, id string) {
	// ip link add link <master> name <master>.<VID> type vlan id <VID>
	icmd.RunCommand("ip", "link", "add", "link", master, "name", slave, "type", "vlan", "id", id).Assert(t, icmd.Success)
	// ip link set <sub_interface_name> up
	icmd.RunCommand("ip", "link", "set", slave, "up").Assert(t, icmd.Success)
}

func deleteInterface(t *testing.T, ifName string) {
	icmd.RunCommand("ip", "link", "delete", ifName).Assert(t, icmd.Success)
	icmd.RunCommand("iptables", "-t", "nat", "--flush").Assert(t, icmd.Success)
	icmd.RunCommand("iptables", "--flush").Assert(t, icmd.Success)
}

func linkExists(t *testing.T, master string) {
	// verify the specified link exists, ip link show <link_name>
	icmd.RunCommand("ip", "link", "show", master).Assert(t, icmd.Success)
}

// ensure Kernel version is >= v3.9 for macvlan support
func macvlanKernelSupport() bool {
	return checkKernelMajorVersionGreaterOrEqualThen(3, 9)
}

func checkKernelMajorVersionGreaterOrEqualThen(kernelVersion int, majorVersion int) bool {
	kv, err := kernel.GetKernelVersion()
	if err != nil {
		return false
	}
	if kv.Kernel < kernelVersion || (kv.Kernel == kernelVersion && kv.Major < majorVersion) {
		return false
	}
	return true
}