fix: ctr run --cni get failed

when   user  executes ctr run --cni to start a container,it will call cni plugin to create network .But when user kills it,the network won’t be removed. if we run a container with same namespace and name again will trigger a bug. we should remove the network when user kills task if it enables cni plugin.

Fix:#6604

Signed-off-by: SongJiang Han <songjiang.dark@gmail.com>
This commit is contained in:
songjiang han 2022-03-11 23:50:20 +08:00
parent d4641e1ce1
commit 2a0b2ee9eb
4 changed files with 95 additions and 13 deletions

51
cmd/ctr/commands/cni.go Normal file
View File

@ -0,0 +1,51 @@
/*
Copyright The containerd Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package commands
import (
"context"
"fmt"
"github.com/containerd/containerd"
"github.com/containerd/containerd/namespaces"
"github.com/containerd/typeurl"
)
func init() {
typeurl.Register(&NetworkMetaData{},
"github.com/containerd/containerd/cmd/ctr/commands", "NetworkMetaData")
}
const (
// CtrCniMetadataExtension is an extension name that identify metadata of container in CreateContainerRequest
CtrCniMetadataExtension = "ctr.cni-containerd.metadata"
)
//ctr pass cni network metadata to containerd if ctr run use option of --cni
type NetworkMetaData struct {
EnableCni bool
}
func FullID(ctx context.Context, c containerd.Container) string {
id := c.ID()
ns, ok := namespaces.Namespace(ctx)
if !ok {
return id
}
return fmt.Sprintf("%s-%s", ns, id)
}

View File

@ -17,7 +17,6 @@
package run package run
import ( import (
"context"
gocontext "context" gocontext "context"
"encoding/csv" "encoding/csv"
"errors" "errors"
@ -31,7 +30,6 @@ import (
"github.com/containerd/containerd/cmd/ctr/commands/tasks" "github.com/containerd/containerd/cmd/ctr/commands/tasks"
"github.com/containerd/containerd/containers" "github.com/containerd/containerd/containers"
clabels "github.com/containerd/containerd/labels" clabels "github.com/containerd/containerd/labels"
"github.com/containerd/containerd/namespaces"
"github.com/containerd/containerd/oci" "github.com/containerd/containerd/oci"
gocni "github.com/containerd/go-cni" gocni "github.com/containerd/go-cni"
specs "github.com/opencontainers/runtime-spec/specs-go" specs "github.com/opencontainers/runtime-spec/specs-go"
@ -196,7 +194,7 @@ var Command = cli.Command{
if !detach { if !detach {
defer func() { defer func() {
if enableCNI { if enableCNI {
if err := network.Remove(ctx, fullID(ctx, container), ""); err != nil { if err := network.Remove(ctx, commands.FullID(ctx, container), ""); err != nil {
logrus.WithError(err).Error("network review") logrus.WithError(err).Error("network review")
} }
} }
@ -218,7 +216,7 @@ var Command = cli.Command{
return err return err
} }
if _, err := network.Setup(ctx, fullID(ctx, container), netNsPath); err != nil { if _, err := network.Setup(ctx, commands.FullID(ctx, container), netNsPath); err != nil {
return err return err
} }
} }
@ -251,15 +249,6 @@ var Command = cli.Command{
}, },
} }
func fullID(ctx context.Context, c containerd.Container) string {
id := c.ID()
ns, ok := namespaces.Namespace(ctx)
if !ok {
return id
}
return fmt.Sprintf("%s-%s", ns, id)
}
// buildLabel builds the labels from command line labels and the image labels // buildLabel builds the labels from command line labels and the image labels
func buildLabels(cmdLabels, imageLabels map[string]string) map[string]string { func buildLabels(cmdLabels, imageLabels map[string]string) map[string]string {
labels := make(map[string]string) labels := make(map[string]string)

View File

@ -214,6 +214,10 @@ func NewContainer(ctx gocontext.Context, client *containerd.Client, context *cli
) )
} }
if context.Bool("cni") {
cniMeta := &commands.NetworkMetaData{EnableCni: true}
cOpts = append(cOpts, containerd.WithContainerExtension(commands.CtrCniMetadataExtension, cniMeta))
}
if caps := context.StringSlice("cap-add"); len(caps) > 0 { if caps := context.StringSlice("cap-add"); len(caps) > 0 {
for _, cap := range caps { for _, cap := range caps {
if !strings.HasPrefix(cap, "CAP_") { if !strings.HasPrefix(cap, "CAP_") {

View File

@ -17,16 +17,50 @@
package tasks package tasks
import ( import (
"context"
"errors" "errors"
"fmt"
"github.com/containerd/containerd" "github.com/containerd/containerd"
"github.com/containerd/containerd/cmd/ctr/commands" "github.com/containerd/containerd/cmd/ctr/commands"
gocni "github.com/containerd/go-cni"
"github.com/containerd/typeurl"
"github.com/moby/sys/signal" "github.com/moby/sys/signal"
"github.com/sirupsen/logrus"
"github.com/urfave/cli" "github.com/urfave/cli"
) )
const defaultSignal = "SIGTERM" const defaultSignal = "SIGTERM"
func RemoveCniNetworkIfExist(ctx context.Context, container containerd.Container) error {
exts, err := container.Extensions(ctx)
if err != nil {
return err
}
networkMeta, ok := exts[commands.CtrCniMetadataExtension]
if !ok {
return nil
}
data, err := typeurl.UnmarshalAny(&networkMeta)
if err != nil {
return fmt.Errorf("failed to unmarshal cni metadata extension %s", commands.CtrCniMetadataExtension)
}
networkMetaData := data.(*commands.NetworkMetaData)
var network gocni.CNI
if networkMetaData.EnableCni {
if network, err = gocni.New(gocni.WithDefaultConf); err != nil {
return err
}
if err := network.Remove(ctx, commands.FullID(ctx, container), ""); err != nil {
logrus.WithError(err).Error("network remove error")
return err
}
}
return nil
}
var killCommand = cli.Command{ var killCommand = cli.Command{
Name: "kill", Name: "kill",
Usage: "signal a container (default: SIGTERM)", Usage: "signal a container (default: SIGTERM)",
@ -93,6 +127,10 @@ var killCommand = cli.Command{
if err != nil { if err != nil {
return err return err
} }
err = RemoveCniNetworkIfExist(ctx, container)
if err != nil {
return err
}
return task.Kill(ctx, sig, opts...) return task.Kill(ctx, sig, opts...)
}, },
} }