kmod-6.1-nvidia-r580: Add grid-license-check

Add a unit that checks for the license to be valid for GRID. The NVIDIA
k8s device plugin requires this unit so if the license is not present,
then the node never offers gpu resources. This prevents a situation
where a node could fail to get a license, join the cluster, and then
later have workloads start to fail due to the unlicensed status.

Signed-off-by: Matthew Yeazel <yeazelm@amazon.com>

Author: yeazelm

packages/kmod-6.1-nvidia-r580/grid-license-check.service

@@ -0,0 +1,19 @@
+[Unit]
+Description=GRID License Check
+RefuseManualStart=true
+RefuseManualStop=true
+After=nvidia-gridd.service
+Requires=nvidia-gridd.service
+
+[Service]
+Type=oneshot
+ExecCondition=/usr/bin/ghostdog match-nvidia-driver grid
+ExecStart=-/usr/bin/truncate /tmp/.nvidia-gridd-license
+ExecStart=/usr/bin/nvidia-smi -q
+ExecStart=/usr/bin/grep -q "License Status.*: Licensed" /tmp/.nvidia-gridd-license
+ExecStart=/usr/bin/touch /etc/drivers/.grid-licensed
+ExecStart=/usr/bin/systemctl stop grid-license-check.timer --no-block
+StandardOutput=append:/tmp/.nvidia-gridd-license
+
+[Install]
+WantedBy=nvidia-k8s-device-plugin.service

packages/kmod-6.1-nvidia-r580/grid-license-check.timer

@@ -0,0 +1,12 @@
+[Unit]
+Description=GRID License Check Timer
+RefuseManualStart=true
+
+[Timer]
+Unit=grid-license-check.service
+OnBootSec=5s
+OnUnitActiveSec=2s
+AccuracySec=1s
+
+[Install]
+WantedBy=timers.target

packages/kmod-6.1-nvidia-r580/grid-license-file-check.conf

@@ -0,0 +1,2 @@
+[Service]
+ExecStartPre=/usr/bin/test -f /etc/drivers/.grid-licensed

packages/kmod-6.1-nvidia-r580/kmod-6.1-nvidia-r580.spec

@@ -49,6 +49,11 @@ Source206: nvidia-persistenced.service
 Source207: fabricmanager.env
 Source208: gridd.conf
 Source209: nvidia-gridd.service
+Source210: grid-license-check.service
+Source211: grid-license-check.timer
+Source212: open-gpu-license-fallback.service
+Source213: tesla-license-fallback.service
+Source214: grid-license-file-check.conf
 
 # NVIDIA tesla conf files from 300 to 399
 Source300: nvidia-tesla-tmpfiles.conf
@@ -394,7 +399,9 @@ install kernel-open/nvidia-drm.ko %{buildroot}%{_cross_datadir}/nvidia/grid/driv
 # Install nvidia-gridd and related files
 install -m 755 nvidia-gridd %{buildroot}%{_cross_bindir}/nvidia-gridd
 install -m 644 %{S:208} %{buildroot}%{_cross_factorydir}%{_cross_sysconfdir}/nvidia/gridd.conf
-install -p -m 0644 %{S:209} %{buildroot}%{_cross_unitdir}
+install -p -m 0644 %{S:209} %{S:210} %{S:211} %{S:212} %{S:213} %{buildroot}%{_cross_unitdir}
+install -d %{buildroot}%{_cross_unitdir}/nvidia-k8s-device-plugin.service.d
+install -p -m 0644 %{S:214} %{buildroot}%{_cross_unitdir}/nvidia-k8s-device-plugin.service.d
 popd
 # End GRID driver
 %endif
@@ -730,6 +737,11 @@ popd
 %{_cross_bindir}/nvidia-gridd
 %{_cross_factorydir}%{_cross_sysconfdir}/nvidia/gridd.conf
 %{_cross_unitdir}/nvidia-gridd.service
+%{_cross_unitdir}/grid-license-check.service
+%{_cross_unitdir}/grid-license-check.timer
+%{_cross_unitdir}/open-gpu-license-fallback.service
+%{_cross_unitdir}/tesla-license-fallback.service
+%{_cross_unitdir}/nvidia-k8s-device-plugin.service.d/grid-license-file-check.conf
 
 %{_cross_datadir}/nvidia/grid/drivers/nvidia.ko
 %{_cross_datadir}/nvidia/grid/drivers/nvidia-uvm.ko

packages/kmod-6.1-nvidia-r580/open-gpu-license-fallback.service

@@ -0,0 +1,14 @@
+[Unit]
+Description=Open GPU GRID License Check Fallback
+RefuseManualStart=true
+RefuseManualStop=true
+
+[Service]
+Type=oneshot
+ExecCondition=/usr/bin/ghostdog match-nvidia-driver open-gpu
+ExecStart=/usr/bin/touch /etc/drivers/.grid-licensed
+ExecStart=/usr/bin/systemctl stop grid-license-check.timer --no-block
+RemainAfterExit=true
+
+[Install]
+WantedBy=nvidia-k8s-device-plugin.service

packages/kmod-6.1-nvidia-r580/tesla-license-fallback.service

@@ -0,0 +1,14 @@
+[Unit]
+Description=Tesla GRID License Check Fallback
+RefuseManualStart=true
+RefuseManualStop=true
+
+[Service]
+Type=oneshot
+ExecCondition=/usr/bin/ghostdog match-nvidia-driver tesla
+ExecStart=/usr/bin/touch /etc/drivers/.grid-licensed
+ExecStart=/usr/bin/systemctl stop grid-license-check.timer --no-block
+RemainAfterExit=true
+
+[Install]
+WantedBy=nvidia-k8s-device-plugin.service