2 files changed, 7 insertions, 1 deletions

diff --git a/nvdebug_entry.c b/nvdebug_entry.c
index 14ad6e9..0854b8b 100644
--- a/nvdebug_entry.c
+++ b/nvdebug_entry.c
@@ -6,6 +6,7 @@
  * - Add sysfs trigger for a preemption
  */
 
+#include <linux/device.h>  // For struct device, bus_find_device*(), struct bus_type
 #include <linux/kernel.h>
 #include <linux/module.h>
 #include <linux/proc_fs.h> // So we can set up entries in /proc
diff --git a/runlist.c b/runlist.c
index 03528af..c8ff99f 100644
--- a/runlist.c
+++ b/runlist.c
@@ -1,4 +1,4 @@
-#include <linux/device.h>  // For struct device, bus_find_device*()
+#include <linux/device.h>  // For struct device, bus_find_device*(), struct bus_type
 //#include <linux/iommu.h>  // For struct iommu_domain
 #include <linux/kernel.h>  // Kernel types
 #include <asm/io.h>
@@ -68,6 +68,11 @@ int get_runlist_iter(struct runlist_iter *rl_iter) {
         runlist_iova = ((u64)rl_base.ptr) << 12;
         printk(KERN_INFO "[nvdebug] Runlist ptr: %x, type: %d, raw: %x, IOVA: %px\n",
                 rl_base.ptr, rl_base.type, rl_base.raw, (void*)runlist_iova);
+        // TODO: Support reading video memory
+        if (rl_base.type == TARGET_VID_MEM) {
+                printk(KERN_ERR "[nvdebug] Runlist is located in video memory. Access to video memory is unimplemented.");
+                return -ENOTSUPP;
+        }
         // Segfaults
         //u32 attempted_read = ioread32(runlist_iova);
         //printk(KERN_INFO "[nvdebug] first word of runlist: %0x\n", attempted_read);