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:

committed by
Gabe Rosenhouse

parent
9b8a0e0c64
commit
b07d768f12
46
ip/link.go
46
ip/link.go
@ -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.
|
||||||
|
@ -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
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user