[nrf fromtree] modules: nrf_wifi: Use spinlocks

Zephyr provides spinlock APIs which can be locked even during the
interrupt context.

Signed-off-by: Chaitanya Tata <[email protected]>
(cherry picked from commit 3e9dffb)

Author: krish2718

modules/nrf_wifi/os/shim.c

@@ -202,53 +202,66 @@ static void zep_shim_qspi_cpy_to(void *priv, unsigned long addr, const void *src
 }
 #endif /* !CONFIG_NRF71_ON_IPC */
 
+struct zep_shim_spinlock {
+	struct k_spinlock lock;
+	k_spinlock_key_t key;
+};
+
 static void *zep_shim_spinlock_alloc(void)
 {
-	struct k_mutex *lock = NULL;
-
-	lock = zep_shim_mem_zalloc(sizeof(*lock));
+	struct zep_shim_spinlock *slock = NULL;
 
-	if (!lock) {
+	slock = k_calloc(sizeof(*slock), sizeof(char));
+	if (!slock) {
 		LOG_ERR("%s: Unable to allocate memory for spinlock", __func__);
 	}
 
-	return lock;
+	return slock;
 }
 
 static void zep_shim_spinlock_free(void *lock)
 {
 #if defined(CONFIG_NRF_WIFI_GLOBAL_HEAP)
 	k_free(lock);
-#else /* CONFIG_NRF_WIFI_GLOBAL_HEAP */
+#else
 	k_heap_free(&wifi_drv_ctrl_mem_pool, lock);
-#endif /* CONFIG_NRF_WIFI_GLOBAL_HEAP */
+#endif
 }
 
 static void zep_shim_spinlock_init(void *lock)
 {
-	k_mutex_init(lock);
+	/* No explicit initialization needed for k_spinlock_t */
+	ARG_UNUSED(lock);
 }
 
 static void zep_shim_spinlock_take(void *lock)
 {
-	k_mutex_lock(lock, K_FOREVER);
+	struct zep_shim_spinlock *slock = (struct zep_shim_spinlock *)lock;
+
+	slock->key = k_spin_lock(&slock->lock);
 }
 
 static void zep_shim_spinlock_rel(void *lock)
 {
-	k_mutex_unlock(lock);
+	struct zep_shim_spinlock *slock = (struct zep_shim_spinlock *)lock;
+
+	k_spin_unlock(&slock->lock, slock->key);
 }
 
 static void zep_shim_spinlock_irq_take(void *lock, unsigned long *flags)
 {
+	struct zep_shim_spinlock *slock = (struct zep_shim_spinlock *)lock;
+
 	ARG_UNUSED(flags);
-	k_mutex_lock(lock, K_FOREVER);
+	slock->key = k_spin_lock(&slock->lock);
 }
 
 static void zep_shim_spinlock_irq_rel(void *lock, unsigned long *flags)
 {
+	struct zep_shim_spinlock *slock = (struct zep_shim_spinlock *)lock;
+
 	ARG_UNUSED(flags);
-	k_mutex_unlock(lock);
+	k_spin_unlock(&slock->lock, slock->key);
 }
 
 static int zep_shim_pr_dbg(const char *fmt, va_list args)