drivers: mbox: nrf_vevif_task_rx: Fix indexing in rx_set_enabled

Fixes VEVIF task receiver driver to work correctly on both nRF7120 and
nRF54h20 by correcting the mapping between Task IDs and Interrupts.

Signed-off-by: Erdem Simsek <[email protected]>

Author: erdemsimsek

drivers/mbox/mbox_nrf_vevif_task_rx.c

@@ -40,12 +40,13 @@ static const uint8_t vevif_irqs[VEVIF_TASKS_NUM] = {
 static void vevif_task_rx_isr(const void *parameter)
 {
 	uint8_t channel = *(uint8_t *)parameter;
-	uint8_t idx = channel - TASKS_IDX_MIN;
+	uint8_t task_id = channel;
+	uint8_t idx = task_id - TASKS_IDX_MIN;
 
-	nrf_vpr_csr_vevif_tasks_clear(BIT(channel));
+	nrf_vpr_csr_vevif_tasks_clear(BIT(task_id));
 
 	if (cbs.cb[idx] != NULL) {
-		cbs.cb[idx](DEVICE_DT_INST_GET(0), channel, cbs.user_data[idx], NULL);
+		cbs.cb[idx](DEVICE_DT_INST_GET(0), task_id, cbs.user_data[idx], NULL);
 	}
 }
 
@@ -65,12 +66,13 @@ static int vevif_task_rx_register_callback(const struct device *dev, uint32_t id
 					   mbox_callback_t cb, void *user_data)
 {
 	ARG_UNUSED(dev);
-	uint8_t idx = id - TASKS_IDX_MIN;
 
 	if (!vevif_task_rx_is_task_valid(id)) {
 		return -EINVAL;
 	}
 
+	uint8_t idx = id - TASKS_IDX_MIN;
+
 	cbs.cb[idx] = cb;
 	cbs.user_data[idx] = user_data;
 
@@ -80,12 +82,13 @@ static int vevif_task_rx_register_callback(const struct device *dev, uint32_t id
 static int vevif_task_rx_set_enabled(const struct device *dev, uint32_t id, bool enable)
 {
 	ARG_UNUSED(dev);
-	uint8_t idx = id - TASKS_IDX_MIN;
 
 	if (!vevif_task_rx_is_task_valid(id)) {
 		return -EINVAL;
 	}
 
+	uint8_t idx = id - vevif_irqs[0];
+
 	if (enable) {
 		if ((cbs.enabled_mask & BIT(id)) != 0U) {
 			return -EALREADY;