1 files changed, 8 insertions, 2 deletions

diff --git a/runlist.c b/runlist.c
index 91fca82..2e9577d 100644
--- a/runlist.c
+++ b/runlist.c
@@ -8,8 +8,10 @@
 #include "nvdebug.h"
 
 // Uncomment to, upon BAR2 access failure, return a PRAMIN-based runlist pointer
-// **If enabled, PRAMIN may not be otherwise used while walking the runlist!**
-// Runlists can only be printed on the Jetson TX2 if this is enabled.
+// in get_runlist_iter(). In order for this pointer to remain valid, PRAMIN
+// **must** not be moved during runlist traversal.
+// The Jetson TX2 has no BAR2, and stores the runlist in VID_MEM, so this must
+// be enabled to print the runlist on the TX2.
 //#define FALLBACK_TO_PRAMIN
 
 /* Get runlist head and info (incl. length)
@@ -142,12 +144,16 @@ int preempt_runlist(struct nvdebug_state *g, uint32_t rl_id) {
 int resubmit_runlist(struct nvdebug_state *g, uint32_t rl_id) {
         if (g->chip_id < NV_CHIP_ID_TURING) {
                 eng_runlist_gf100_t rl;
+                if (rl_id > MAX_RUNLISTS_GF100)
+                        return -EINVAL;
                 if ((rl.raw = nvdebug_readq(g, NV_PFIFO_ENG_RUNLIST_BASE_GF100(rl_id))) == -1)
                         return -EIO;
                 rl.id = rl_id;
                 nvdebug_writeq(g, NV_PFIFO_RUNLIST_BASE_GF100, rl.raw);
         } else if (g->chip_id < NV_CHIP_ID_AMPERE) {
                 runlist_submit_tu102_t submit;
+                if (rl_id > MAX_RUNLISTS_TU102)
+                        return -EINVAL;
                 if ((submit.raw = nvdebug_readq(g, NV_PFIFO_RUNLIST_SUBMIT_TU102(rl_id))) == -1)
                         return -EIO;
                 nvdebug_writeq(g, NV_PFIFO_RUNLIST_SUBMIT_TU102(rl_id), submit.raw);