Add image load.

Signed-off-by: Lantao Liu <lantaol@google.com>

cmd/cri-containerd/options/options.go

@@ -19,6 +19,7 @@ package options
 import (
 	"fmt"
 	"os"
+	"time"
 
 	"github.com/BurntSushi/toml"
 	"github.com/containerd/containerd"
@@ -30,6 +31,8 @@ const (
 	configFilePathArgName = "config"
 	// defaultConfigFilePath is the default config file path.
 	defaultConfigFilePath = "/etc/cri-containerd/config.toml"
+	// connectionTimeout is the grpc connection timeout.
+	connectionTimeout = 10 * time.Second
 )
 
 // ContainerdConfig contains config related to containerd
@@ -178,6 +181,13 @@ func PrintDefaultTomlConfig() {
 	}
 }
 
+// AddGRPCFlags add flags for grpc connection.
+func AddGRPCFlags(fs *pflag.FlagSet) (*string, *time.Duration) {
+	endpoint := fs.String("endpoint", defaultConfig().SocketPath, "cri-containerd endpoint.")
+	timeout := fs.Duration("timeout", connectionTimeout, "cri-containerd connection timeout.")
+	return endpoint, timeout
+}
+
 // defaultConfig returns default configurations of cri-containerd.
 func defaultConfig() Config {
 	return Config{