expose driver capabilities to volume_hook

client/allocrunner/taskrunner/volume_hook.go

@@ -121,6 +121,18 @@ func (h *volumeHook) prepareHostVolumes(req *interfaces.TaskPrestartRequest, vol
 		return nil, err
 	}
 
+	if len(hostVolumeMounts) > 0 {
+		caps, err := h.runner.DriverCapabilities()
+		if err != nil {
+			return nil, fmt.Errorf("could not validate task driver capabilities: %v", err)
+		}
+		switch caps.Volumes {
+		case drivers.VolumeSupportNone, drivers.VolumeSupportCSIOnly:
+			return nil, fmt.Errorf("task driver does not support host volumes")
+		default:
+		}
+	}
+
 	return hostVolumeMounts, nil
 }
 
@@ -167,6 +179,18 @@ func (h *volumeHook) prepareCSIVolumes(req *interfaces.TaskPrestartRequest, volu
 		}
 	}
 
+	if len(mounts) > 0 {
+		caps, err := h.runner.DriverCapabilities()
+		if err != nil {
+			return nil, fmt.Errorf("could not validate task driver capabilities: %v", err)
+		}
+		switch caps.Volumes {
+		case drivers.VolumeSupportNone, drivers.VolumeSupportHostOnly:
+			return nil, fmt.Errorf("task driver does not support CSI")
+		default:
+		}
+	}
+
 	return mounts, nil
 }