Merge pull request #230 from steveeJ/netns-optional-on-del

plugins: don't require CNI_NETNS for DEL command
This commit is contained in:
Stefan Junker 2016-05-27 13:49:05 +02:00
commit 131ecc4055
6 changed files with 79 additions and 10 deletions

View File

@ -36,6 +36,8 @@ type CmdArgs struct {
StdinData []byte StdinData []byte
} }
type reqForCmdEntry map[string]bool
// PluginMain is the "main" for a plugin. It accepts // PluginMain is the "main" for a plugin. It accepts
// two callback functions for add and del commands. // two callback functions for add and del commands.
func PluginMain(cmdAdd, cmdDel func(_ *CmdArgs) error) { func PluginMain(cmdAdd, cmdDel func(_ *CmdArgs) error) {
@ -44,20 +46,62 @@ func PluginMain(cmdAdd, cmdDel func(_ *CmdArgs) error) {
vars := []struct { vars := []struct {
name string name string
val *string val *string
req bool reqForCmd reqForCmdEntry
}{ }{
{"CNI_COMMAND", &cmd, true}, {
{"CNI_CONTAINERID", &contID, false}, "CNI_COMMAND",
{"CNI_NETNS", &netns, true}, &cmd,
{"CNI_IFNAME", &ifName, true}, reqForCmdEntry{
{"CNI_ARGS", &args, false}, "ADD": true,
{"CNI_PATH", &path, true}, "DEL": true,
},
},
{
"CNI_CONTAINERID",
&contID,
reqForCmdEntry{
"ADD": false,
"DEL": false,
},
},
{
"CNI_NETNS",
&netns,
reqForCmdEntry{
"ADD": true,
"DEL": false,
},
},
{
"CNI_IFNAME",
&ifName,
reqForCmdEntry{
"ADD": true,
"DEL": true,
},
},
{
"CNI_ARGS",
&args,
reqForCmdEntry{
"ADD": false,
"DEL": false,
},
},
{
"CNI_PATH",
&path,
reqForCmdEntry{
"ADD": true,
"DEL": true,
},
},
} }
argsMissing := false argsMissing := false
for _, v := range vars { for _, v := range vars {
*v.val = os.Getenv(v.name) *v.val = os.Getenv(v.name)
if v.req && *v.val == "" { if v.reqForCmd[cmd] && *v.val == "" {
log.Printf("%v env variable missing", v.name) log.Printf("%v env variable missing", v.name)
argsMissing = true argsMissing = true
} }

View File

@ -71,5 +71,14 @@ var _ = Describe("Skel", func() {
// Expect(err).NotTo(HaveOccurred()) // Expect(err).NotTo(HaveOccurred())
// PluginMain(fErr, nil) // PluginMain(fErr, nil)
// }) // })
It("should not fail with DEL and no NETNS and noop callback", func() {
err := os.Setenv("CNI_COMMAND", "DEL")
Expect(err).NotTo(HaveOccurred())
err = os.Unsetenv("CNI_NETNS")
Expect(err).NotTo(HaveOccurred())
PluginMain(nil, fNoop)
})
}) })
}) })

View File

@ -289,6 +289,10 @@ func cmdDel(args *skel.CmdArgs) error {
return err return err
} }
if args.Netns == "" {
return nil
}
var ipn *net.IPNet var ipn *net.IPNet
err = ns.WithNetNSPath(args.Netns, func(_ ns.NetNS) error { err = ns.WithNetNSPath(args.Netns, func(_ ns.NetNS) error {
var err error var err error

View File

@ -152,6 +152,10 @@ func cmdDel(args *skel.CmdArgs) error {
return err return err
} }
if args.Netns == "" {
return nil
}
return ns.WithNetNSPath(args.Netns, func(_ ns.NetNS) error { return ns.WithNetNSPath(args.Netns, func(_ ns.NetNS) error {
return ip.DelLinkByName(args.IfName) return ip.DelLinkByName(args.IfName)
}) })

View File

@ -170,6 +170,10 @@ func cmdDel(args *skel.CmdArgs) error {
return err return err
} }
if args.Netns == "" {
return nil
}
return ns.WithNetNSPath(args.Netns, func(_ ns.NetNS) error { return ns.WithNetNSPath(args.Netns, func(_ ns.NetNS) error {
return ip.DelLinkByName(args.IfName) return ip.DelLinkByName(args.IfName)
}) })

View File

@ -199,6 +199,10 @@ func cmdDel(args *skel.CmdArgs) error {
return err return err
} }
if args.Netns == "" {
return nil
}
var ipn *net.IPNet var ipn *net.IPNet
err := ns.WithNetNSPath(args.Netns, func(_ ns.NetNS) error { err := ns.WithNetNSPath(args.Netns, func(_ ns.NetNS) error {
var err error var err error