From 2708bdf2f5e61f9d440fbaea0268c9689ffbcd5f Mon Sep 17 00:00:00 2001
From: Zachary Gershman <zgershman@pivotal.io>
Date: Fri, 12 Feb 2016 09:30:10 -0800
Subject: [PATCH] Create a plugin for up'ing a lo device

- Believe we need sudo to create netns
- Use syscall instead of relying on ip netns
- Add sudo to .travis.yml
- Needs more -E
- Revert Godeps GoVersion to 1.4.2
- in travis, test command is run with all necessary env vars
- Loopback plugin only works on 'lo' interface
- Update README, add loopback plugin config
- note script dependency on jq

Signed-off-by: Gabe Rosenhouse <grosenhouse@pivotal.io>
---
 .travis.yml                                  |  6 +-
 Godeps/Godeps.json                           |  2 +-
 README.md                                    |  9 ++-
 plugins/main/loopback/loopback.go            | 40 +++++++++++
 plugins/main/loopback/loopback_suite_test.go | 58 ++++++++++++++++
 plugins/main/loopback/loopback_test.go       | 70 ++++++++++++++++++++
 scripts/priv-net-run.sh                      |  1 -
 7 files changed, 182 insertions(+), 4 deletions(-)
 create mode 100644 plugins/main/loopback/loopback.go
 create mode 100644 plugins/main/loopback/loopback_suite_test.go
 create mode 100644 plugins/main/loopback/loopback_test.go

diff --git a/.travis.yml b/.travis.yml
index 34249a22..2c796907 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,4 +1,5 @@
 language: go
+sudo: required
 
 go:
   - 1.4
@@ -12,4 +13,7 @@ install:
  - go get ${TOOLS_CMD}/vet
 
 script:
- - ./test
+ - sudo -E /bin/bash -c 'PATH=$GOROOT/bin:$PATH ./test'
+
+notifications:
+  email: false
diff --git a/Godeps/Godeps.json b/Godeps/Godeps.json
index ae04ed03..69a1c72b 100644
--- a/Godeps/Godeps.json
+++ b/Godeps/Godeps.json
@@ -1,6 +1,6 @@
 {
 	"ImportPath": "github.com/appc/cni",
-	"GoVersion": "go1.5.3",
+	"GoVersion": "go1.4.2",
 	"Packages": [
 		"./..."
 	],
diff --git a/README.md b/README.md
index 933165ea..5c1c25b4 100644
--- a/README.md
+++ b/README.md
@@ -24,7 +24,9 @@ This repository includes a number of common plugins that can be found in plugins
 Please see Documentation/ folder for documentation about particular plugins.
 
 ## Running the plugins
-The scripts/ directory contains two scripts, priv-net-run.sh and docker-run.sh, that can be used to exercise the plugins.
+The scripts/ directory contains two scripts, `priv-net-run.sh` and `docker-run.sh`, that can be used to exercise the plugins.
+
+**note - priv-net-run.sh depends on `jq`**
 
 Start out by creating a netconf file to describe a network:
 
@@ -46,6 +48,11 @@ $ cat >/etc/cni/net.d/10-mynet.conf <<EOF
 	}
 }
 EOF
+$ cat >/etc/cni/net.d/99-loopback.conf <<EOF
+{
+	"type": "loopback"
+}
+EOF
 ```
 
 The directory `/etc/cni/net.d` is the default location in which the scripts will look for net configurations.
diff --git a/plugins/main/loopback/loopback.go b/plugins/main/loopback/loopback.go
new file mode 100644
index 00000000..d0e96462
--- /dev/null
+++ b/plugins/main/loopback/loopback.go
@@ -0,0 +1,40 @@
+package main
+
+import (
+	"os"
+
+	"github.com/appc/cni/pkg/ns"
+	"github.com/appc/cni/pkg/skel"
+	"github.com/vishvananda/netlink"
+)
+
+func cmdAdd(args *skel.CmdArgs) error {
+	args.IfName = "lo" // ignore config, this only works for loopback
+	err := ns.WithNetNSPath(args.Netns, false, func(hostNS *os.File) error {
+		link, err := netlink.LinkByName(args.IfName)
+		if err != nil {
+			return err // not tested
+		}
+
+		err = netlink.LinkSetUp(link)
+		if err != nil {
+			return err // not tested
+		}
+
+		return nil
+	})
+	if err != nil {
+		return err // not tested
+	}
+
+	return nil
+}
+
+func cmdDel(args *skel.CmdArgs) error {
+	// del does nothing, we're going to destroy the device anyway
+	return nil
+}
+
+func main() {
+	skel.PluginMain(cmdAdd, cmdDel)
+}
diff --git a/plugins/main/loopback/loopback_suite_test.go b/plugins/main/loopback/loopback_suite_test.go
new file mode 100644
index 00000000..c9c3438e
--- /dev/null
+++ b/plugins/main/loopback/loopback_suite_test.go
@@ -0,0 +1,58 @@
+package main_test
+
+import (
+	"os"
+
+	"github.com/onsi/gomega/gexec"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+
+	"testing"
+
+	"golang.org/x/sys/unix"
+)
+
+var pathToLoPlugin string
+
+func TestLoopback(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "Loopback Suite")
+}
+
+var _ = BeforeSuite(func() {
+	var err error
+	pathToLoPlugin, err = gexec.Build("github.com/appc/cni/plugins/main/loopback")
+	Expect(err).NotTo(HaveOccurred())
+})
+
+var _ = AfterSuite(func() {
+	gexec.CleanupBuildArtifacts()
+})
+
+func makeNetworkNS(containerID string) string {
+	namespace := "/var/run/netns/" + containerID
+
+	err := os.MkdirAll("/var/run/netns", 0600)
+	Expect(err).NotTo(HaveOccurred())
+
+	err = unix.Unshare(unix.CLONE_NEWNET)
+	Expect(err).NotTo(HaveOccurred())
+
+	fd, err := os.Create(namespace)
+	Expect(err).NotTo(HaveOccurred())
+	defer fd.Close()
+
+	err = unix.Mount("/proc/self/ns/net", namespace, "none", unix.MS_BIND, "")
+	Expect(err).NotTo(HaveOccurred())
+
+	Expect(namespace).To(BeAnExistingFile())
+	return namespace
+}
+
+func removeNetworkNS(networkNS string) error {
+	err := unix.Unmount(networkNS, unix.MNT_DETACH)
+
+	err = os.RemoveAll(networkNS)
+	return err
+}
diff --git a/plugins/main/loopback/loopback_test.go b/plugins/main/loopback/loopback_test.go
new file mode 100644
index 00000000..5ed22159
--- /dev/null
+++ b/plugins/main/loopback/loopback_test.go
@@ -0,0 +1,70 @@
+package main_test
+
+import (
+	"fmt"
+	"net"
+	"os"
+	"os/exec"
+	"strings"
+
+	"github.com/appc/cni/pkg/ns"
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+	"github.com/onsi/gomega/gexec"
+)
+
+var _ = Describe("Loopback", func() {
+	var (
+		networkNS   string
+		containerID string
+		command     *exec.Cmd
+	)
+
+	BeforeEach(func() {
+		command = exec.Command(pathToLoPlugin)
+
+		environ := os.Environ()
+
+		containerID = "some-container-id"
+		networkNS = makeNetworkNS(containerID)
+
+		cniEnvVars := []string{
+			fmt.Sprintf("CNI_COMMAND=%s", "ADD"),
+			fmt.Sprintf("CNI_CONTAINERID=%s", containerID),
+			fmt.Sprintf("CNI_NETNS=%s", networkNS),
+			fmt.Sprintf("CNI_IFNAME=%s", "this is ignored"),
+			fmt.Sprintf("CNI_ARGS=%s", "none"),
+			fmt.Sprintf("CNI_PATH=%s", "/some/test/path"),
+		}
+
+		for _, v := range cniEnvVars {
+			environ = append(environ, v)
+		}
+
+		command.Env = environ
+		command.Stdin = strings.NewReader("this doesn't matter")
+	})
+
+	AfterEach(func() {
+		Expect(removeNetworkNS(networkNS)).To(Succeed())
+	})
+
+	Context("when given a network namespace", func() {
+		It("sets the lo device to UP", func() {
+			session, err := gexec.Start(command, GinkgoWriter, GinkgoWriter)
+			Expect(err).NotTo(HaveOccurred())
+
+			Eventually(session).Should(gexec.Exit(0))
+
+			var lo *net.Interface
+			err = ns.WithNetNSPath(networkNS, false, func(hostNS *os.File) error {
+				var err error
+				lo, err = net.InterfaceByName("lo")
+				return err
+			})
+			Expect(err).NotTo(HaveOccurred())
+
+			Expect(lo.Flags & net.FlagUp).To(Equal(net.FlagUp))
+		})
+	})
+})
diff --git a/scripts/priv-net-run.sh b/scripts/priv-net-run.sh
index 5a27a78d..607a3524 100755
--- a/scripts/priv-net-run.sh
+++ b/scripts/priv-net-run.sh
@@ -8,7 +8,6 @@ contid=$(printf '%x%x%x%x' $RANDOM $RANDOM $RANDOM $RANDOM)
 netnspath=/var/run/netns/$contid
 
 ip netns add $contid
-ip netns exec $contid ip link set lo up
 ./exec-plugins.sh add $contid $netnspath