virtcontainers: fix the issue of missing watchConsole

When do the reloading sandbox in shimv2, it's needed to
rewatch the hypervisor's console when debug enabled.

Fixes:kata-containers#2091

Signed-off-by: lifupan <lifupan@gmail.com>

virtcontainers/agent.go

@@ -126,6 +126,9 @@ type agent interface {
 	// check will check the agent liveness
 	check() error
 
+	// tell whether the agent is long  live connected or not
+	longLiveConn() bool
+
 	// disconnect will disconnect the connection to the agent
 	disconnect() error
 

virtcontainers/api.go

@@ -187,9 +187,9 @@ func FetchSandbox(ctx context.Context, sandboxID string) (VCSandbox, error) {
 		return nil, err
 	}
 
-	// If the proxy is KataBuiltInProxyType type, it needs to restart the proxy to watch the
-	// guest console if it hadn't been watched.
-	if isProxyBuiltIn(s.config.ProxyType) {
+	// If the agent is long live connection, it needs to restart the proxy to
+	// watch the guest console if it hadn't been watched.
+	if s.agent.longLiveConn() {
 		err = s.startProxy()
 		if err != nil {
 			s.Release()

virtcontainers/kata_agent.go

@@ -190,6 +190,7 @@ type kataAgent struct {
 	state          KataAgentState
 	keepConn       bool
 	proxyBuiltIn   bool
+	useVsock       bool
 	dynamicTracing bool
 	dead           bool
 	kmodules       []string
@@ -224,6 +225,10 @@ func (k *kataAgent) getSharePath(id string) string {
 	return filepath.Join(kataHostSharedDir(), id)
 }
 
+func (k *kataAgent) longLiveConn() bool {
+	return k.keepConn
+}
+
 // KataAgentSetDefaultTraceConfigOptions validates agent trace options and
 // sets defaults.
 func KataAgentSetDefaultTraceConfigOptions(config *KataAgentConfig) error {
@@ -297,6 +302,7 @@ func (k *kataAgent) init(ctx context.Context, sandbox *Sandbox, config interface
 		disableVMShutdown = k.handleTraceSettings(c)
 		k.keepConn = c.LongLiveConn
 		k.kmodules = c.KernelModules
+		k.useVsock = c.UseVSock
 	default:
 		return false, vcTypes.ErrInvalidConfigType
 	}
@@ -350,6 +356,7 @@ func (k *kataAgent) internalConfigure(h hypervisor, id, sharePath string, builti
 	if config != nil {
 		switch c := config.(type) {
 		case KataAgentConfig:
+			k.useVsock = c.UseVSock
 			if k.vmSocket, err = h.generateSocket(id, c.UseVSock); err != nil {
 				return err
 			}
@@ -617,6 +624,7 @@ func (k *kataAgent) startProxy(sandbox *Sandbox) error {
 	defer span.Finish()
 
 	var err error
+	var agentURL string
 
 	if k.proxy == nil {
 		return errorMissingProxy
@@ -627,18 +635,25 @@ func (k *kataAgent) startProxy(sandbox *Sandbox) error {
 	}
 
 	if k.state.URL != "" {
-		k.Logger().WithFields(logrus.Fields{
-			"sandbox":   sandbox.id,
-			"proxy-pid": k.state.ProxyPid,
-			"proxy-url": k.state.URL,
-		}).Infof("proxy already started")
-		return nil
-	}
-
-	// Get agent socket path to provide it to the proxy.
-	agentURL, err := k.agentURL()
-	if err != nil {
-		return err
+		// For keepConn case, when k.state.URL isn't nil, it means shimv2 had disconnected from
+		// sandbox and try to relaunch sandbox again. Here it needs to start proxy again to watch
+		// the hypervisor console.
+		if k.keepConn {
+			agentURL = k.state.URL
+		} else {
+			k.Logger().WithFields(logrus.Fields{
+				"sandbox":   sandbox.id,
+				"proxy-pid": k.state.ProxyPid,
+				"proxy-url": k.state.URL,
+			}).Infof("proxy already started")
+			return nil
+		}
+	} else {
+		// Get agent socket path to provide it to the proxy.
+		agentURL, err = k.agentURL()
+		if err != nil {
+			return err
+		}
 	}
 
 	consoleURL, err := sandbox.hypervisor.getSandboxConsole(sandbox.id)
@@ -1713,7 +1728,7 @@ func (k *kataAgent) connect() error {
 	}
 
 	k.Logger().WithField("url", k.state.URL).WithField("proxy", k.state.ProxyPid).Info("New client")
-	client, err := kataclient.NewAgentClient(k.ctx, k.state.URL, k.proxyBuiltIn)
+	client, err := kataclient.NewAgentClient(k.ctx, k.state.URL, !k.useVsock)
 	if err != nil {
 		k.dead = true
 		return err

virtcontainers/kata_agent_test.go

@@ -290,6 +290,7 @@ func TestKataAgentSendReq(t *testing.T) {
 		state: KataAgentState{
 			URL: testKataProxyURL,
 		},
+		useVsock: true,
 	}
 
 	for _, req := range reqList {
@@ -790,6 +791,7 @@ func TestAgentNetworkOperation(t *testing.T) {
 		state: KataAgentState{
 			URL: testKataProxyURL,
 		},
+		useVsock: true,
 	}
 
 	_, err = k.updateInterface(nil)

virtcontainers/noop_agent.go

@@ -32,6 +32,10 @@ func (n *noopAgent) init(ctx context.Context, sandbox *Sandbox, config interface
 	return false, nil
 }
 
+func (n *noopAgent) longLiveConn() bool {
+	return false
+}
+
 // createSandbox is the Noop agent sandbox creation implementation. It does nothing.
 func (n *noopAgent) createSandbox(sandbox *Sandbox) error {
 	return nil