From 6073dc491f0fdd56f83bcfd2beb74a07591badad Mon Sep 17 00:00:00 2001
From: Alexandre Pujol <alexandre@pujol.io>
Date: Fri, 19 Jul 2024 19:23:48 +0100
Subject: [PATCH] feat(profile): add nvidia-smi.

---
 apparmor.d/profiles-m-r/nvidia-smi | 34 ++++++++++++++++++++++++++++++
 dists/flags/main.flags             |  1 +
 2 files changed, 35 insertions(+)
 create mode 100644 apparmor.d/profiles-m-r/nvidia-smi

diff --git a/apparmor.d/profiles-m-r/nvidia-smi b/apparmor.d/profiles-m-r/nvidia-smi
new file mode 100644
index 00000000..571ab331
--- /dev/null
+++ b/apparmor.d/profiles-m-r/nvidia-smi
@@ -0,0 +1,34 @@
+# apparmor.d - Full set of apparmor profiles
+# Copyright (C) 2024 Alexandre Pujol <alexandre@pujol.io>
+# SPDX-License-Identifier: GPL-2.0-only
+
+abi <abi/3.0>,
+
+include <tunables/global>
+
+@{exec_path} = @{bin}/nvidia-smi
+profile nvidia-smi @{exec_path} {
+  include <abstractions/base>
+  include <abstractions/consoles>
+  include <abstractions/nvidia-strict>
+
+  @{exec_path} mr,
+
+  @{sys}/devices/system/node/ r,
+  @{sys}/devices/system/node/node@{int}/cpumap r,
+
+        @{PROC}/devices r,
+        @{PROC}/driver/nvidia/capabilities/mig/config r,
+        @{PROC}/driver/nvidia/capabilities/mig/monitor r,
+  owner @{PROC}/@{pid}/cmdline r,
+
+  /dev/char/@{dynamic}:@{int} w,          # For dynamic assignment range 234 to 254, 384 to 511
+  /dev/nvidia-caps/ rw,
+  /dev/nvidia-caps/nvidia-cap@{int} r,
+  /dev/nvidia-uvm rw,
+  /dev/nvidia-uvm-tools r,
+
+  include if exists <local/nvidia-smi>
+}
+
+# vim:syntax=apparmor
diff --git a/dists/flags/main.flags b/dists/flags/main.flags
index 737531b4..57862b8c 100644
--- a/dists/flags/main.flags
+++ b/dists/flags/main.flags
@@ -255,6 +255,7 @@ nmap complain
 nmcli complain
 nvidia-detector complain
 nvidia-persistenced complain
+nvidia-smi complain
 okular complain
 ollama attach_disconnected,complain
 os-prober attach_disconnected,complain