Add all option to disable all hooks

Signed-off-by: Carlos Eduardo Arango Gutierrez <[email protected]>

Author: ArangoGutierrez

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

@@ -287,7 +287,15 @@ func (m command) generateSpec(opts *options) (spec.Interface, error) {
 
 	if len(opts.disableHooks.Value()) > 0 {
 		for _, hook := range opts.disableHooks.Value() {
-			initOpts = append(initOpts, nvcdi.WithDisabledHook(nvcdi.HookName(hook)))
+			if hook == "all" {
+				initOpts = append(initOpts, nvcdi.WithDisabledHook(nvcdi.HookEnableCudaCompat))
+				initOpts = append(initOpts, nvcdi.WithDisabledHook(nvcdi.HookCreateSymlinks))
+				initOpts = append(initOpts, nvcdi.WithDisabledHook(nvcdi.HookUpdateLDCache))
+				break
+			}
+			if nvcdi.IsValidHook(hook) {
+				initOpts = append(initOpts, nvcdi.WithDisabledHook(nvcdi.HookName(hook)))
+			}
 		}
 	}
 

pkg/nvcdi/api.go

@@ -44,4 +44,19 @@ const (
 	// HookEnableCudaCompat refers to the hook used to enable CUDA Forward Compatibility.
 	// This was added with v1.17.5 of the NVIDIA Container Toolkit.
 	HookEnableCudaCompat = HookName("enable-cuda-compat")
+	// HookCreateSymlinks refers to the hook used create symlinks inside the
+	// directory path to be mounted into a container.
+	HookCreateSymlinks = HookName("create-symlinks")
+	// HookUpdateLDCache refers to the hook used to  Update the dynamic linker
+	// cache inside the directory path to be mounted into a container.
+	HookUpdateLDCache = HookName("update-ldcache")
 )
+
+func IsValidHook(hook string) bool {
+	switch HookName(hook) {
+	case HookEnableCudaCompat, HookCreateSymlinks, HookUpdateLDCache:
+		return true
+	default:
+		return false
+	}
+}