wil6210: fix RX buffers release and unmap

RX SKBs are released in both wil6210 rmmod and RX handle.
As there is no lock to protect the buffers DMA unmap,
the SKB pointer in buff_arr is used to check if the buffer
memory was already released.
Setting wil->rx_buff_mgmt.buff_arr[buff_id].skb to NULL before the DMA
memory unmap will prevent duplicate unmapping of the same memory.
Move the buffer ID to the free list also in case the SKB is NULL.

Signed-off-by: Maya Erez <merez@codeaurora.org>
Signed-off-by: Kalle Valo <kvalo@codeaurora.org>

1 files changed, 10 insertions, 5 deletions

diff --git a/drivers/net/wireless/ath/wil6210/txrx_edma.c b/drivers/net/wireless/ath/wil6210/txrx_edma.c
index bca61cb44c37..3e7fc2983cbb 100644
--- a/drivers/net/wireless/ath/wil6210/txrx_edma.c
+++ b/drivers/net/wireless/ath/wil6210/txrx_edma.c
@@ -279,9 +279,6 @@ static void wil_move_all_rx_buff_to_free_list(struct wil6210_priv *wil,
                 u16 buff_id;
 
                 *d = *_d;
-                pa = wil_rx_desc_get_addr_edma(&d->dma);
-                dmalen = le16_to_cpu(d->dma.length);
-                dma_unmap_single(dev, pa, dmalen, DMA_FROM_DEVICE);
 
                 /* Extract the SKB from the rx_buff management array */
                 buff_id = __le16_to_cpu(d->mac.buff_id);
@@ -291,10 +288,15 @@ static void wil_move_all_rx_buff_to_free_list(struct wil6210_priv *wil,
                 }
                 skb = wil->rx_buff_mgmt.buff_arr[buff_id].skb;
                 wil->rx_buff_mgmt.buff_arr[buff_id].skb = NULL;
-                if (unlikely(!skb))
+                if (unlikely(!skb)) {
                         wil_err(wil, "No Rx skb at buff_id %d\n", buff_id);
-                else
+                } else {
+                        pa = wil_rx_desc_get_addr_edma(&d->dma);
+                        dmalen = le16_to_cpu(d->dma.length);
+                        dma_unmap_single(dev, pa, dmalen, DMA_FROM_DEVICE);
+
                         kfree_skb(skb);
+                }
 
                 /* Move the buffer from the active to the free list */
                 list_move(&wil->rx_buff_mgmt.buff_arr[buff_id].list,
@@ -906,6 +908,9 @@ again:
         wil->rx_buff_mgmt.buff_arr[buff_id].skb = NULL;
         if (!skb) {
                 wil_err(wil, "No Rx skb at buff_id %d\n", buff_id);
+                /* Move the buffer from the active list to the free list */
+                list_move(&wil->rx_buff_mgmt.buff_arr[buff_id].list,
+                          &wil->rx_buff_mgmt.free);
                 goto again;
         }