From b07d768f12b80671aafd48ff5f560d3c2a32dc7a Mon Sep 17 00:00:00 2001
From: Jay Dunkelberger <ldunkelberger@pivotal.io>
Date: Thu, 9 Mar 2017 13:55:15 -0800
Subject: [PATCH] pkg/ip: do not leak types from vendored netlink package

The exported function SetupVeth now returns a package-defined type.

Signed-off-by: Gabe Rosenhouse <grosenhouse@pivotal.io>
---
 ip/link.go      | 46 ++++++++++++++++++++++++++++++++++------------
 ip/link_test.go |  4 ++--
 2 files changed, 36 insertions(+), 14 deletions(-)

diff --git a/ip/link.go b/ip/link.go
index 6431bb41..8900f653 100644
--- a/ip/link.go
+++ b/ip/link.go
@@ -98,30 +98,49 @@ func RenameLink(curName, newName string) error {
 	return err
 }
 
+type LinkAttrs struct {
+	Name         string
+	HardwareAddr net.HardwareAddr
+	Index        int
+}
+
+type link struct {
+	netlink.Link
+}
+
+func (l *link) Attrs() LinkAttrs {
+	a := l.Link.Attrs()
+	return LinkAttrs{
+		Name:         a.Name,
+		HardwareAddr: a.HardwareAddr,
+		Index:        a.Index,
+	}
+}
+
+type Link interface {
+	Attrs() LinkAttrs
+}
+
 // SetupVeth sets up a virtual ethernet link.
 // Should be in container netns, and will switch back to hostNS to set the host
 // veth end up.
-func SetupVeth(contVethName string, mtu int, hostNS ns.NetNS) (hostVeth, contVeth netlink.Link, err error) {
-	var hostVethName string
-	hostVethName, contVeth, err = makeVeth(contVethName, mtu)
+func SetupVeth(contVethName string, mtu int, hostNS ns.NetNS) (Link, Link, error) {
+	hostVethName, contVeth, err := makeVeth(contVethName, mtu)
 	if err != nil {
-		return
+		return nil, nil, err
 	}
 
 	if err = netlink.LinkSetUp(contVeth); err != nil {
-		err = fmt.Errorf("failed to set %q up: %v", contVethName, err)
-		return
+		return nil, nil, fmt.Errorf("failed to set %q up: %v", contVethName, err)
 	}
 
-	hostVeth, err = netlink.LinkByName(hostVethName)
+	hostVeth, err := netlink.LinkByName(hostVethName)
 	if err != nil {
-		err = fmt.Errorf("failed to lookup %q: %v", hostVethName, err)
-		return
+		return nil, nil, fmt.Errorf("failed to lookup %q: %v", hostVethName, err)
 	}
 
 	if err = netlink.LinkSetNsFd(hostVeth, int(hostNS.Fd())); err != nil {
-		err = fmt.Errorf("failed to move veth to host netns: %v", err)
-		return
+		return nil, nil, fmt.Errorf("failed to move veth to host netns: %v", err)
 	}
 
 	err = hostNS.Do(func(_ ns.NetNS) error {
@@ -135,7 +154,10 @@ func SetupVeth(contVethName string, mtu int, hostNS ns.NetNS) (hostVeth, contVet
 		}
 		return nil
 	})
-	return
+	if err != nil {
+		return nil, nil, err
+	}
+	return &link{hostVeth}, &link{contVeth}, nil
 }
 
 // DelLinkByName removes an interface link.
diff --git a/ip/link_test.go b/ip/link_test.go
index 3df9ab8f..2c9ae400 100644
--- a/ip/link_test.go
+++ b/ip/link_test.go
@@ -46,8 +46,8 @@ var _ = Describe("Link", func() {
 		hostNetNS         ns.NetNS
 		containerNetNS    ns.NetNS
 		ifaceCounter      int = 0
-		hostVeth          netlink.Link
-		containerVeth     netlink.Link
+		hostVeth          ip.Link
+		containerVeth     ip.Link
 		hostVethName      string
 		containerVethName string