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>
This commit is contained in:
Jay Dunkelberger
2017-03-09 13:55:15 -08:00
committed by Gabe Rosenhouse
parent 9b8a0e0c64
commit b07d768f12
2 changed files with 36 additions and 14 deletions

View File

@ -98,30 +98,49 @@ func RenameLink(curName, newName string) error {
return err 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. // SetupVeth sets up a virtual ethernet link.
// Should be in container netns, and will switch back to hostNS to set the host // Should be in container netns, and will switch back to hostNS to set the host
// veth end up. // veth end up.
func SetupVeth(contVethName string, mtu int, hostNS ns.NetNS) (hostVeth, contVeth netlink.Link, err error) { func SetupVeth(contVethName string, mtu int, hostNS ns.NetNS) (Link, Link, error) {
var hostVethName string hostVethName, contVeth, err := makeVeth(contVethName, mtu)
hostVethName, contVeth, err = makeVeth(contVethName, mtu)
if err != nil { if err != nil {
return return nil, nil, err
} }
if err = netlink.LinkSetUp(contVeth); err != nil { if err = netlink.LinkSetUp(contVeth); err != nil {
err = fmt.Errorf("failed to set %q up: %v", contVethName, err) return nil, nil, fmt.Errorf("failed to set %q up: %v", contVethName, err)
return
} }
hostVeth, err = netlink.LinkByName(hostVethName) hostVeth, err := netlink.LinkByName(hostVethName)
if err != nil { if err != nil {
err = fmt.Errorf("failed to lookup %q: %v", hostVethName, err) return nil, nil, fmt.Errorf("failed to lookup %q: %v", hostVethName, err)
return
} }
if err = netlink.LinkSetNsFd(hostVeth, int(hostNS.Fd())); err != nil { if err = netlink.LinkSetNsFd(hostVeth, int(hostNS.Fd())); err != nil {
err = fmt.Errorf("failed to move veth to host netns: %v", err) return nil, nil, fmt.Errorf("failed to move veth to host netns: %v", err)
return
} }
err = hostNS.Do(func(_ ns.NetNS) error { 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 nil
}) })
return if err != nil {
return nil, nil, err
}
return &link{hostVeth}, &link{contVeth}, nil
} }
// DelLinkByName removes an interface link. // DelLinkByName removes an interface link.

View File

@ -46,8 +46,8 @@ var _ = Describe("Link", func() {
hostNetNS ns.NetNS hostNetNS ns.NetNS
containerNetNS ns.NetNS containerNetNS ns.NetNS
ifaceCounter int = 0 ifaceCounter int = 0
hostVeth netlink.Link hostVeth ip.Link
containerVeth netlink.Link containerVeth ip.Link
hostVethName string hostVethName string
containerVethName string containerVethName string