support websocket for exec and attach

Signed-off-by: haozi007 <duguhaotian@gmail.com>

Author: duguhaotian

cmd/crictl/attach.go

@@ -43,6 +43,12 @@ var runtimeAttachCommand = &cli.Command{
 			Aliases: []string{"i"},
 			Usage:   "Keep STDIN open",
 		},
+		&cli.StringFlag{
+			Name:    "transport",
+			Aliases: []string{"p"},
+			Usage:   "transport protocol, One of: spdy|websocket",
+			Value:   "spdy",
+		},
 	},
 	Action: func(c *cli.Context) error {
 		id := c.Args().First()
@@ -63,9 +69,10 @@ var runtimeAttachCommand = &cli.Command{
 		defer cancel()
 
 		var opts = attachOptions{
-			id:    id,
-			tty:   c.Bool("tty"),
-			stdin: c.Bool("stdin"),
+			id:        id,
+			tty:       c.Bool("tty"),
+			stdin:     c.Bool("stdin"),
+			transport: c.String("transport"),
 		}
 		if err = Attach(ctx, runtimeClient, opts); err != nil {
 			return fmt.Errorf("attaching running container failed: %w", err)
@@ -109,5 +116,5 @@ func Attach(ctx context.Context, client internalapi.RuntimeService, opts attachO
 	}
 
 	logrus.Debugf("Attach URL: %v", URL)
-	return stream(ctx, opts.stdin, opts.tty, URL)
+	return stream(ctx, opts.stdin, opts.tty, opts.transport, URL)
 }

cmd/crictl/exec.go

@@ -26,6 +26,7 @@ import (
 	mobyterm "github.com/moby/term"
 	"github.com/sirupsen/logrus"
 	"github.com/urfave/cli/v2"
+	"k8s.io/apimachinery/pkg/util/httpstream"
 	restclient "k8s.io/client-go/rest"
 	remoteclient "k8s.io/client-go/tools/remotecommand"
 	internalapi "k8s.io/cri-api/pkg/apis"
@@ -67,6 +68,12 @@ var runtimeExecCommand = &cli.Command{
 			Aliases: []string{"i"},
 			Usage:   "Keep STDIN open",
 		},
+		&cli.StringFlag{
+			Name:    "transport",
+			Aliases: []string{"p"},
+			Usage:   "transport protocol, One of: spdy|websocket",
+			Value:   "spdy",
+		},
 	},
 	Action: func(c *cli.Context) error {
 		if c.NArg() < 2 {
@@ -79,11 +86,12 @@ var runtimeExecCommand = &cli.Command{
 		}
 
 		var opts = execOptions{
-			id:      c.Args().First(),
-			timeout: c.Int64("timeout"),
-			tty:     c.Bool("tty"),
-			stdin:   c.Bool("interactive"),
-			cmd:     c.Args().Slice()[1:],
+			id:        c.Args().First(),
+			timeout:   c.Int64("timeout"),
+			tty:       c.Bool("tty"),
+			stdin:     c.Bool("interactive"),
+			transport: c.String("transport"),
+			cmd:       c.Args().Slice()[1:],
 		}
 		if c.Bool("sync") {
 			exitCode, err := ExecSync(runtimeClient, opts)
@@ -160,11 +168,35 @@ func Exec(ctx context.Context, client internalapi.RuntimeService, opts execOptio
 	}
 
 	logrus.Debugf("Exec URL: %v", URL)
-	return stream(ctx, opts.stdin, opts.tty, URL)
+	return stream(ctx, opts.stdin, opts.tty, opts.transport, URL)
+}
+
+func getExecutor(protocol string, url *url.URL) (exec remoteclient.Executor, err error) {
+	config := &restclient.Config{TLSClientConfig: restclient.TLSClientConfig{Insecure: true}}
+
+	switch protocol {
+	case "spdy":
+		exec, err = remoteclient.NewSPDYExecutor(config, "POST", url)
+	case "websocket":
+		exec, err = remoteclient.NewWebSocketExecutor(config, "GET", url.String())
+	default:
+		exec, err = remoteclient.NewSPDYExecutor(config, "POST", url)
+		if err != nil {
+			break
+		}
+		var wexec remoteclient.Executor
+		wexec, err = remoteclient.NewWebSocketExecutor(config, "GET", url.String())
+		if err != nil {
+			break
+		}
+		exec, err = remoteclient.NewFallbackExecutor(wexec, exec, httpstream.IsUpgradeFailure)
+	}
+
+	return
 }
 
-func stream(ctx context.Context, in, tty bool, url *url.URL) error {
-	executor, err := remoteclient.NewSPDYExecutor(&restclient.Config{TLSClientConfig: restclient.TLSClientConfig{Insecure: true}}, "POST", url)
+func stream(ctx context.Context, in, tty bool, protocol string, url *url.URL) error {
+	executor, err := getExecutor(protocol, url)
 	if err != nil {
 		return err
 	}

cmd/crictl/util.go

@@ -112,6 +112,8 @@ type execOptions struct {
 	tty bool
 	// Whether to stream stdin
 	stdin bool
+	// protocol of transport
+	transport string
 	// Command to exec
 	cmd []string
 }
@@ -122,6 +124,8 @@ type attachOptions struct {
 	tty bool
 	// Whether pass Stdin to container
 	stdin bool
+	// protocol of transport
+	transport string
 }
 
 type portforwardOptions struct {