Add snapshot save/restore functionality for vrnetlab nodes (#2904)

Adds snapshot functionality for vrnetlab-based VMs:

- `clab tools snapshot save` - Save VM snapshots
- `clab deploy --restore-all` - Restore from snapshots directory
- `clab deploy --restore node=path` - Restore specific nodes

Snapshots capture full VM state for fast restoration. Non-vrnetlab nodes
are automatically skipped.

Vrnetlab part of this PR - srl-labs/vrnetlab#403

---------

Co-authored-by: Roman Dodin <[email protected]>

Author: torbbang

.github/workflows/build-containerlab.yml

@@ -12,7 +12,7 @@ jobs:
     runs-on: ubuntu-24.04
     steps:
       - uses: actions/checkout@v5
-      - uses: WillAbides/setup-go[email protected]
+      - uses: actions/setup-go@v6
         with:
           go-version: ${{ inputs.go_ver }}
 

.github/workflows/golangci-lint.yml

@@ -17,7 +17,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v5
-      - uses: WillAbides/setup-go[email protected]
+      - uses: actions/setup-go@v6
         with:
           go-version: ${{ inputs.go_ver }}
 

cmd/deploy.go

@@ -129,6 +129,23 @@ func deployCmd(o *Options) (*cobra.Command, error) { //nolint: funlen
 		"lab owner name (only for users in clab_admins group)",
 	)
 
+	c.Flags().StringVar(
+		&o.Deploy.RestoreAll,
+		"restore-all",
+		"",
+		"restore all nodes that have snapshots in this directory (default: ./snapshots)",
+	)
+	// Allow flag without value to default to ./snapshots
+	c.Flags().Lookup("restore-all").NoOptDefVal = "./snapshots"
+
+	c.Flags().StringArrayVar(
+		&o.Deploy.RestoreNodeSnapshots,
+		"restore",
+		nil,
+		"restore specific node from snapshot file (format: node=path/to/snapshot.tar). "+
+			"Can be specified multiple times. Overrides --restore-all for specified nodes.",
+	)
+
 	return c, nil
 }
 
@@ -157,7 +174,9 @@ func deployFn(cobraCmd *cobra.Command, o *Options) error {
 		SetReconfigure(o.Deploy.Reconfigure).
 		SetGraph(o.Deploy.GenerateGraph).
 		SetSkipPostDeploy(o.Deploy.SkipPostDeploy).
-		SetSkipLabDirFileACLs(o.Deploy.SkipLabDirectoryFileACLs)
+		SetSkipLabDirFileACLs(o.Deploy.SkipLabDirectoryFileACLs).
+		SetRestoreAll(o.Deploy.RestoreAll).
+		SetRestoreNodeSnapshots(o.Deploy.RestoreNodeSnapshots)
 
 	containers, err := c.Deploy(cobraCmd.Context(), deploymentOptions)
 	if err != nil {

cmd/options.go

@@ -107,6 +107,12 @@ func GetOptions() *Options {
 				SrcPort:        0,
 				DeletionPrefix: "vx-",
 			},
+			ToolsSnapshot: &ToolsSnapshotOptions{
+				OutputDir:     "./snapshots",
+				Format:        "table",
+				Timeout:       5 * time.Minute,
+				MaxConcurrent: 0,
+			},
 			Version: &VersionOptions{
 				Short: false,
 				JSON:  false,
@@ -135,6 +141,7 @@ type Options struct {
 	ToolsSSHX      *ToolsSSHXOptions
 	ToolsVeth      *ToolsVethOptions
 	ToolsVxlan     *ToolsVxlanOptions
+	ToolsSnapshot  *ToolsSnapshotOptions
 	Version        *VersionOptions
 }
 
@@ -282,6 +289,8 @@ type DeployOptions struct {
 	SkipLabDirectoryFileACLs bool
 	ExportTemplate           string
 	LabOwner                 string
+	RestoreAll               string
+	RestoreNodeSnapshots     []string
 }
 
 func (o *DeployOptions) toClabOptions() []clabcore.ClabOption {
@@ -456,6 +465,13 @@ type ToolsVxlanOptions struct {
 	DeletionPrefix string
 }
 
+type ToolsSnapshotOptions struct {
+	OutputDir     string
+	Format        string
+	Timeout       time.Duration
+	MaxConcurrent int
+}
+
 type VersionOptions struct {
 	Short bool
 	JSON  bool

cmd/tools.go

@@ -19,6 +19,7 @@ func toolsSubcommandRegisterFuncs() []func(*Options) (*cobra.Command, error) {
 		disableTxOffloadCmd,
 		gottyCmd,
 		netemCmd,
+		snapshotCmd,
 		sshxCmd,
 		vethCmd,
 		vxlanCmd,