Runtime v2

Signed-off-by: Michael Crosby <crosbymichael@gmail.com>

cmd/ctr/commands/tasks/checkpoint.go

@@ -18,9 +18,12 @@ package tasks
 
 import (
 	"fmt"
+	"runtime"
 
 	"github.com/containerd/containerd"
 	"github.com/containerd/containerd/cmd/ctr/commands"
+	"github.com/containerd/containerd/runtime/linux/runctypes"
+	"github.com/containerd/containerd/runtime/v2/runc/options"
 	"github.com/pkg/errors"
 	"github.com/urfave/cli"
 )
@@ -34,6 +37,11 @@ var checkpointCommand = cli.Command{
 			Name:  "exit",
 			Usage: "stop the container after the checkpoint",
 		},
+		cli.StringFlag{
+			Name:  "runtime",
+			Usage: "runtime name",
+			Value: fmt.Sprintf("io.containerd.runtime.v1.%s", runtime.GOOS),
+		},
 	},
 	Action: func(context *cli.Context) error {
 		id := context.Args().First()
@@ -55,7 +63,7 @@ var checkpointCommand = cli.Command{
 		}
 		var opts []containerd.CheckpointTaskOpts
 		if context.Bool("exit") {
-			opts = append(opts, containerd.WithExit)
+			opts = append(opts, withExit(context.String("runtime")))
 		}
 		checkpoint, err := task.Checkpoint(ctx, opts...)
 		if err != nil {
@@ -65,3 +73,19 @@ var checkpointCommand = cli.Command{
 		return nil
 	},
 }
+
+func withExit(rt string) containerd.CheckpointTaskOpts {
+	return func(r *containerd.CheckpointTaskInfo) error {
+		switch rt {
+		case "io.containerd.runc.v1":
+			r.Options = &options.CheckpointOptions{
+				Exit: true,
+			}
+		default:
+			r.Options = &runctypes.CheckpointOptions{
+				Exit: true,
+			}
+		}
+		return nil
+	}
+}