Allow inclusion of sockets in CDI specification

This change adds with-fabricmanager and with-persistenced
flags to the nvidia-ctk cdi generate command that ensures that
a generated specification includes these sockets if present on
the host.

Note that for mangement mode, these sockets are always included
if detected.

Signed-off-by: Evan Lezar <elezar@nvidia.com>

Author: elezar

cmd/nvidia-ctk/cdi/generate/generate.go

@@ -60,6 +60,9 @@ type options struct {
 		files          cli.StringSlice
 		ignorePatterns cli.StringSlice
 	}
+
+	withFabricmanager bool
+	withPersistenced  bool
 }
 
 // NewCommand constructs a generate-cdi command with the specified logger
@@ -169,6 +172,16 @@ func (m command) build() *cli.Command {
 			Usage:       "Specify a pattern the CSV mount specifications.",
 			Destination: &opts.csv.ignorePatterns,
 		},
+		&cli.BoolFlag{
+			Name:        "with-fabricmanager",
+			Usage:       "Include the nvidia-fabricmanager socket in the generated CDI specification.",
+			Destination: &opts.withFabricmanager,
+		},
+		&cli.BoolFlag{
+			Name:        "with-persistenced",
+			Usage:       "Include the nvidia-persistenced socket in the generated CDI specification.",
+			Destination: &opts.withPersistenced,
+		},
 	}
 
 	return &c
@@ -273,6 +286,8 @@ func (m command) generateSpec(opts *options) (spec.Interface, error) {
 		nvcdi.WithLibrarySearchPaths(opts.librarySearchPaths.Value()),
 		nvcdi.WithCSVFiles(opts.csv.files.Value()),
 		nvcdi.WithCSVIgnorePatterns(opts.csv.ignorePatterns.Value()),
+		nvcdi.WithOptInFeature("allow-fabricmanager", opts.withFabricmanager),
+		nvcdi.WithOptInFeature("allow-persistenced", opts.withPersistenced),
 	)
 	if err != nil {
 		return nil, fmt.Errorf("failed to create CDI library: %v", err)

pkg/nvcdi/lib.go

@@ -132,6 +132,10 @@ func New(opts ...Option) (Interface, error) {
 		if l.vendor == "" {
 			l.vendor = "management.nvidia.com"
 		}
+		// For management specifications we always allow the fabricmanager and
+		// persistenced sockets.
+		WithOptInFeature("allow-fabricmanager", true)(l)
+		WithOptInFeature("allow-persistenced", true)(l)
 		lib = (*managementlib)(l)
 	case ModeNvml:
 		lib = (*nvmllib)(l)