Add dial timeout env var

vms/registry/vm_getter.go

@@ -7,6 +7,7 @@ import (
 	"errors"
 	"fmt"
 	"path/filepath"
+	"strings"
 
 	"github.com/ava-labs/avalanchego/ids"
 	"github.com/ava-labs/avalanchego/utils/filesystem"
@@ -20,6 +21,7 @@ var (
 	_ VMGetter = (*vmGetter)(nil)
 
 	errInvalidVMID = errors.New("invalid vmID")
+	podSuffix      = "_pod"
 )
 
 // VMGetter defines functionality to get the plugins on the node.
@@ -66,16 +68,27 @@ func (getter *vmGetter) Get() (map[ids.ID]vms.Factory, map[ids.ID]vms.Factory, e
 			continue
 		}
 
+		isPod := false
 		nameWithExtension := file.Name()
+		fmt.Printf("nameWithExtension: %s\n", nameWithExtension)
+
+		// check the name of the file if it includes _pod.yml then we create a podman VM factory
+		// otherwise create the currently used factory
+
 		// Strip any extension from the file. This is to support windows .exe
 		// files.
 		name := nameWithExtension[:len(nameWithExtension)-len(filepath.Ext(nameWithExtension))]
+		fmt.Printf("name: %s\n", name)
 
 		// Skip hidden files.
 		if len(name) == 0 {
 			continue
 		}
 
+		if strings.HasSuffix(name, podSuffix) {
+			isPod = true
+			name = strings.TrimSuffix(name, podSuffix)
+		}
 		vmID, err := getter.config.Manager.Lookup(name)
 		if err != nil {
 			// there is no alias with plugin name, try to use full vmID.
@@ -99,11 +112,16 @@ func (getter *vmGetter) Get() (map[ids.ID]vms.Factory, map[ids.ID]vms.Factory, e
 			return nil, nil, err
 		}
 
-		unregisteredVMs[vmID] = rpcchainvm.NewFactory(
-			filepath.Join(getter.config.PluginDirectory, file.Name()),
-			getter.config.CPUTracker,
-			getter.config.RuntimeTracker,
-		)
+		if !isPod {
+			unregisteredVMs[vmID] = rpcchainvm.NewFactory(
+				filepath.Join(getter.config.PluginDirectory, file.Name()),
+				getter.config.CPUTracker,
+				getter.config.RuntimeTracker,
+			)
+		} else {
+			// TODO: use the container implementation here
+			fmt.Printf("Attempt to use the container implementation to create a VM Factory for: %s\n", file.Name())
+		}
 	}
 	return registeredVMs, unregisteredVMs, nil
 }

vms/rpcchainvm/runtime/runtime.go

@@ -13,6 +13,9 @@ const (
 	// Address of the runtime engine server.
 	EngineAddressKey = "AVALANCHE_VM_RUNTIME_ENGINE_ADDR"
 
+	// EngineDialDimeout is the timeout for the engine to connect to the server before it shuts down.
+	EngineDialTimeoutKey = "AVALANCHE_ENGINE_TIMEOUT"
+
 	// Duration before handshake timeout during bootstrap.
 	DefaultHandshakeTimeout = 5 * time.Second
 

vms/rpcchainvm/vm.go

@@ -26,7 +26,7 @@ import (
 	runtimepb "github.com/ava-labs/avalanchego/proto/pb/vm/runtime"
 )
 
-const defaultRuntimeDialTimeout = 5 * time.Second
+const defaultDialTimeout = 5 * time.Second
 
 // The address of the Runtime server is expected to be passed via ENV `runtime.EngineAddressKey`.
 // This address is used by the Runtime client to send Initialize RPC to server.
@@ -78,7 +78,15 @@ func Serve(ctx context.Context, vm block.ChainVM, opts ...grpcutils.ServerOption
 		return fmt.Errorf("failed to create new listener: %w", err)
 	}
 
-	ctx, cancel := context.WithTimeout(ctx, defaultRuntimeDialTimeout)
+	dialTimeout := defaultDialTimeout
+	if dialTimeoutStr, exists := os.LookupEnv(runtime.EngineDialTimeoutKey); exists {
+		parsedDialTimeout, err := time.ParseDuration(dialTimeoutStr)
+		if err != nil {
+			return fmt.Errorf("failed to parse engine dial timeout: %w", err)
+		}
+		dialTimeout = parsedDialTimeout
+	}
+	ctx, cancel := context.WithTimeout(ctx, dialTimeout)
 	defer cancel()
 	err = client.Initialize(ctx, version.RPCChainVMProtocol, listener.Addr().String())
 	if err != nil {