From f8764d581d69d9af389f13ef6d14b96684550bb3 Mon Sep 17 00:00:00 2001
From: Alex Forencich <alex@alexforencich.com>
Date: Thu, 5 Mar 2026 15:46:30 -0800
Subject: [PATCH] cndm: Check for queue allocation failures in the driver model

Signed-off-by: Alex Forencich <alex@alexforencich.com>
---
 src/cndm/tb/cndm.py | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

diff --git a/src/cndm/tb/cndm.py b/src/cndm/tb/cndm.py
index 5e75d99..a78088b 100644
--- a/src/cndm/tb/cndm.py
+++ b/src/cndm/tb/cndm.py
@@ -85,6 +85,8 @@ class Cq:
 
         self.cons_ptr = None
 
+        self.db_offset = None
+
         self.hw_regs = self.driver.hw_regs
 
     async def open(self, irqn, size):
@@ -128,6 +130,13 @@ class Cq:
         rsp_unpacked = struct.unpack("<HHLLLLLLLQQLLLL", rsp)
         print(rsp_unpacked)
         self.cqn = rsp_unpacked[4]
+        self.db_offset = rsp_unpacked[8]
+
+        if self.db_offset == 0:
+            self.cqn = None
+            self.db_offset = None
+            self.log.error("Failed to allocate CQ")
+            return
 
         self.log.info("Opened CQ %d", self.cqn)
 
@@ -242,7 +251,13 @@ class Sq:
         self.sqn = rsp_unpacked[4]
         self.db_offset = rsp_unpacked[8]
 
-        self.log.info("Opened SQ %d", self.sqn)
+        if self.db_offset == 0:
+            self.sqn = None
+            self.db_offset = None
+            self.log.error("Failed to allocate SQ")
+            return
+
+        self.log.info("Opened SQ %d (CQ %d)", self.sqn, cq.cqn)
 
         self.enabled = True
 
@@ -408,7 +423,13 @@ class Rq:
         self.rqn = rsp_unpacked[4]
         self.db_offset = rsp_unpacked[8]
 
-        self.log.info("Opened RQ %d", self.rqn)
+        if self.db_offset == 0:
+            self.rqn = None
+            self.db_offset = None
+            self.log.error("Failed to allocate RQ")
+            return
+
+        self.log.info("Opened RQ %d (CQ %d)", self.rqn, cq.cqn)
 
         self.enabled = True
 
@@ -531,7 +552,9 @@ class Port:
         self.rx_queue = Queue()
 
     async def init(self):
+        await self.open()
 
+    async def open(self):
         self.rxcq = Cq(self.driver, self)
         await self.rxcq.open(self.index, 256)