return 0;
 }
 
-static void qtnf_pcie_control_rx_callback(void *arg, const u8 *buf, size_t len)
+static void qtnf_pcie_control_rx_callback(void *arg, const u8 __iomem *buf,
+                                         size_t len)
 {
        struct qtnf_pcie_bus_priv *priv = arg;
        struct qtnf_bus *bus = pci_get_drvdata(priv->pdev);
                return;
        }
 
-       skb_put_data(skb, buf, len);
+       memcpy_fromio(skb_put(skb, len), buf, len);
 
        qtnf_trans_handle_rx_ctl_packet(bus, skb);
 }
 
        if (unlikely(size == 0 || size > QTN_IPC_MAX_DATA_SZ)) {
                pr_err("wrong rx packet size: %zu\n", size);
                rx_buff_ok = false;
-       } else {
-               memcpy_fromio(ipc->rx_data, ipc->shm_region->data, size);
+       }
+
+       if (likely(rx_buff_ok)) {
+               ipc->rx_packet_count++;
+               ipc->rx_callback.fn(ipc->rx_callback.arg,
+                                   ipc->shm_region->data, size);
        }
 
        writel(QTNF_SHM_IPC_ACK, &shm_reg_hdr->flags);
        readl(&shm_reg_hdr->flags); /* flush PCIe write */
 
        ipc->interrupt.fn(ipc->interrupt.arg);
-
-       if (likely(rx_buff_ok)) {
-               ipc->rx_packet_count++;
-               ipc->rx_callback.fn(ipc->rx_callback.arg, ipc->rx_data, size);
-       }
 }
 
 static void qtnf_shm_ipc_irq_work(struct work_struct *work)
 
 };
 
 struct qtnf_shm_ipc_rx_callback {
-       void (*fn)(void *arg, const u8 *buf, size_t len);
+       void (*fn)(void *arg, const u8 __iomem *buf, size_t len);
        void *arg;
 };
 
 
        u8 waiting_for_ack;
 
-       u8 rx_data[QTN_IPC_MAX_DATA_SZ] __aligned(sizeof(u32));
-
        struct qtnf_shm_ipc_int interrupt;
        struct qtnf_shm_ipc_rx_callback rx_callback;