Function createDevCharSymlinks

cmd/nvidia-validator/main.go:943–992

createDevCharSymlinks creates symlinks in /host-dev-char that point to all possible NVIDIA devices nodes.

(driverInfo driverInfo, disableDevCharSymlinkCreation bool)

// createDevCharSymlinks creates symlinks in /host-dev-char that point to all possible NVIDIA devices nodes.
func createDevCharSymlinks(driverInfo driverInfo, disableDevCharSymlinkCreation bool) error {
	if disableDevCharSymlinkCreation {
		log.WithField("disableDevCharSymlinkCreation", true).
			Info("skipping the creation of symlinks under /dev/char that correspond to NVIDIA character devices")
		return nil
	}

	log.Info("creating symlinks under /dev/char that correspond to NVIDIA character devices")

	// Check if NVIDIA module is already loaded in kernel memory.
	// If it is, we don't need to run modprobe (which would fail if modules aren't in /lib/modules/).
	// This handles the case where the driver container performed a userspace-only install
	// after detecting that module was already loaded from a previous boot.
	moduleAlreadyLoaded := isNvidiaModuleLoaded()

	// Only attempt to load NVIDIA kernel modules when:
	// 1. Module is not already loaded in kernel memory, AND
	// 2. We can chroot into driverRoot to run modprobe
	loadKernelModules := !moduleAlreadyLoaded && (driverInfo.isHostDriver || (driverInfo.devRoot == driverInfo.driverRoot))

	// driverRootCtrPath is the path of the driver install dir in the container. This will either be
	// driverInstallDirCtrPathFlag or '/host'.
	// Note, if we always mounted the driver install dir to '/driver-root' in the validation container
	// instead, then we could simplify to always use driverInfo.driverRootCtrPath -- which would be
	// either '/host' or '/driver-root', both paths would exist in the validation container.
	driverRootCtrPath := driverInstallDirCtrPathFlag
	if driverInfo.isHostDriver {
		driverRootCtrPath = "/host"
	}

	// We now create the symlinks in /dev/char.
	creator, err := devchar.NewSymlinkCreator(
		devchar.WithDriverRoot(driverRootCtrPath),
		devchar.WithDevRoot(driverInfo.devRoot),
		devchar.WithDevCharPath(hostDevCharPath),
		devchar.WithCreateAll(true),
		devchar.WithCreateDeviceNodes(true),
		devchar.WithLoadKernelModules(loadKernelModules),
	)
	if err != nil {
		return fmt.Errorf("error creating symlink creator: %w", err)
	}

	err = creator.CreateLinks()
	if err != nil {
		return fmt.Errorf("error creating symlinks: %w", err)
	}

	return nil
}

func createStatusFile(statusFile string) error {
	_, err := os.Create(statusFile)