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>

diff --git a/ip/link.go b/ip/link.go
index 6431bb4..8900f65 100644 (file)
--- 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 3df9ab8..2c9ae40 100644 (file)
--- 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