54 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_dev_add()  local
56 	bal_dev_ctx = nrf_wifi_osal_mem_zalloc(sizeof(*bal_dev_ctx));  in nrf_wifi_bal_dev_add()
58 	if (!bal_dev_ctx) {  in nrf_wifi_bal_dev_add()
63 	bal_dev_ctx->bpriv = bpriv;  in nrf_wifi_bal_dev_add()
64 	bal_dev_ctx->hal_dev_ctx = hal_dev_ctx;  in nrf_wifi_bal_dev_add()
66 	bal_dev_ctx->bus_dev_ctx = bpriv->ops->dev_add(bpriv->bus_priv,  in nrf_wifi_bal_dev_add()
67 						       bal_dev_ctx);  in nrf_wifi_bal_dev_add()
69 	if (!bal_dev_ctx->bus_dev_ctx) {  in nrf_wifi_bal_dev_add()
77 		if (bal_dev_ctx) {  in nrf_wifi_bal_dev_add()
78 			nrf_wifi_osal_mem_free(bal_dev_ctx);  in nrf_wifi_bal_dev_add()
79 			bal_dev_ctx = NULL;  in nrf_wifi_bal_dev_add()
83 	return bal_dev_ctx;  in nrf_wifi_bal_dev_add()
87 void nrf_wifi_bal_dev_rem(struct nrf_wifi_bal_dev_ctx *bal_dev_ctx)  in nrf_wifi_bal_dev_rem()  argument
89 	bal_dev_ctx->bpriv->ops->dev_rem(bal_dev_ctx->bus_dev_ctx);  in nrf_wifi_bal_dev_rem()
91 	nrf_wifi_osal_mem_free(bal_dev_ctx);  in nrf_wifi_bal_dev_rem()
95 enum nrf_wifi_status nrf_wifi_bal_dev_init(struct nrf_wifi_bal_dev_ctx *bal_dev_ctx)  in nrf_wifi_bal_dev_init()  argument
100 	bal_dev_ctx->rpu_fw_booted = true;  in nrf_wifi_bal_dev_init()
103 	status = bal_dev_ctx->bpriv->ops->dev_init(bal_dev_ctx->bus_dev_ctx);  in nrf_wifi_bal_dev_init()
114 void nrf_wifi_bal_dev_deinit(struct nrf_wifi_bal_dev_ctx *bal_dev_ctx)  in nrf_wifi_bal_dev_deinit()  argument
116 	bal_dev_ctx->bpriv->ops->dev_deinit(bal_dev_ctx->bus_dev_ctx);  in nrf_wifi_bal_dev_deinit()
122 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_isr()  local
125 	bal_dev_ctx = (struct nrf_wifi_bal_dev_ctx *)ctx;  in nrf_wifi_bal_isr()
127 	status = bal_dev_ctx->bpriv->intr_callbk_fn(bal_dev_ctx->hal_dev_ctx);  in nrf_wifi_bal_isr()
181 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_read_word()  local
184 	bal_dev_ctx = (struct nrf_wifi_bal_dev_ctx *)ctx;  in nrf_wifi_bal_read_word()
188 	nrf_wifi_rpu_bal_sleep_chk(bal_dev_ctx,  in nrf_wifi_bal_read_word()
193 	val = bal_dev_ctx->bpriv->ops->read_word(bal_dev_ctx->bus_dev_ctx,  in nrf_wifi_bal_read_word()
204 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_write_word()  local
206 	bal_dev_ctx = (struct nrf_wifi_bal_dev_ctx *)ctx;  in nrf_wifi_bal_write_word()
210 	nrf_wifi_rpu_bal_sleep_chk(bal_dev_ctx,  in nrf_wifi_bal_write_word()
215 	bal_dev_ctx->bpriv->ops->write_word(bal_dev_ctx->bus_dev_ctx,  in nrf_wifi_bal_write_word()
226 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_read_block()  local
228 	bal_dev_ctx = (struct nrf_wifi_bal_dev_ctx *)ctx;  in nrf_wifi_bal_read_block()
232 	nrf_wifi_rpu_bal_sleep_chk(bal_dev_ctx,  in nrf_wifi_bal_read_block()
237 	bal_dev_ctx->bpriv->ops->read_block(bal_dev_ctx->bus_dev_ctx,  in nrf_wifi_bal_read_block()
249 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_write_block()  local
251 	bal_dev_ctx = (struct nrf_wifi_bal_dev_ctx *)ctx;  in nrf_wifi_bal_write_block()
255 	nrf_wifi_rpu_bal_sleep_chk(bal_dev_ctx,  in nrf_wifi_bal_write_block()
260 	bal_dev_ctx->bpriv->ops->write_block(bal_dev_ctx->bus_dev_ctx,  in nrf_wifi_bal_write_block()
272 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_dma_map()  local
275 	bal_dev_ctx = (struct nrf_wifi_bal_dev_ctx *)ctx;  in nrf_wifi_bal_dma_map()
277 	phy_addr = bal_dev_ctx->bpriv->ops->dma_map(bal_dev_ctx->bus_dev_ctx,  in nrf_wifi_bal_dma_map()
291 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_dma_unmap()  local
294 	bal_dev_ctx = (struct nrf_wifi_bal_dev_ctx *)ctx;  in nrf_wifi_bal_dma_unmap()
296 	virt_addr = bal_dev_ctx->bpriv->ops->dma_unmap(bal_dev_ctx->bus_dev_ctx,  in nrf_wifi_bal_dma_unmap()
308 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_rpu_ps_sleep()  local
310 	bal_dev_ctx = (struct nrf_wifi_bal_dev_ctx *)ctx;  in nrf_wifi_bal_rpu_ps_sleep()
312 	bal_dev_ctx->bpriv->ops->rpu_ps_sleep(bal_dev_ctx->bus_dev_ctx);  in nrf_wifi_bal_rpu_ps_sleep()
318 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_rpu_ps_wake()  local
320 	bal_dev_ctx = (struct nrf_wifi_bal_dev_ctx *)ctx;  in nrf_wifi_bal_rpu_ps_wake()
322 	bal_dev_ctx->bpriv->ops->rpu_ps_wake(bal_dev_ctx->bus_dev_ctx);  in nrf_wifi_bal_rpu_ps_wake()
328 	struct nrf_wifi_bal_dev_ctx *bal_dev_ctx = NULL;  in nrf_wifi_bal_rpu_ps_status()  local
330 	bal_dev_ctx = (struct nrf_wifi_bal_dev_ctx *)ctx;  in nrf_wifi_bal_rpu_ps_status()
332 	return bal_dev_ctx->bpriv->ops->rpu_ps_status(bal_dev_ctx->bus_dev_ctx);  in nrf_wifi_bal_rpu_ps_status()