#define WCN36XX_HAL_OFFLOAD_ENABLE                          1
 #define WCN36XX_HAL_OFFLOAD_BCAST_FILTER_ENABLE             0x2
 #define WCN36XX_HAL_OFFLOAD_ARP_AND_BCAST_FILTER_ENABLE        \
-       (HAL_OFFLOAD_ENABLE|HAL_OFFLOAD_BCAST_FILTER_ENABLE)
+       (WCN36XX_HAL_OFFLOAD_ENABLE | WCN36XX_HAL_OFFLOAD_BCAST_FILTER_ENABLE)
 
 struct wcn36xx_hal_ns_offload_params {
        u8 src_ipv6_addr[WCN36XX_HAL_IPV6_ADDR_LEN];
        /* slot index for this offload */
        u32 slot_index;
        u8 bss_index;
-};
+} __packed;
 
 struct wcn36xx_hal_host_offload_req {
-       u8 offload_Type;
+       u8 offload_type;
 
        /* enable or disable */
        u8 enable;
                u8 host_ipv4_addr[4];
                u8 host_ipv6_addr[WCN36XX_HAL_IPV6_ADDR_LEN];
        } u;
-};
+} __packed;
 
 struct wcn36xx_hal_host_offload_req_msg {
        struct wcn36xx_hal_msg_header header;
        struct wcn36xx_hal_host_offload_req host_offload_params;
        struct wcn36xx_hal_ns_offload_params ns_offload_params;
-};
+} __packed;
 
 /* Packet Types. */
 #define WCN36XX_HAL_KEEP_ALIVE_NULL_PKT              1
 
 
        flush_workqueue(wcn->hal_ind_wq);
        mutex_lock(&wcn->conf_mutex);
+
        vif = wcn36xx_get_first_assoc_vif(wcn);
-       if (vif)
+       if (vif) {
+               ret = wcn36xx_smd_arp_offload(wcn, vif, true);
+               if (ret)
+                       goto out;
                ret = wcn36xx_smd_set_power_params(wcn, true);
+       }
+out:
        mutex_unlock(&wcn->conf_mutex);
-
        return ret;
 }
 
        flush_workqueue(wcn->hal_ind_wq);
        mutex_lock(&wcn->conf_mutex);
        vif = wcn36xx_get_first_assoc_vif(wcn);
-       if (vif)
+       if (vif) {
                wcn36xx_smd_set_power_params(wcn, false);
+               wcn36xx_smd_arp_offload(wcn, vif, false);
+       }
        mutex_unlock(&wcn->conf_mutex);
 
        return 0;
 
        return ret;
 }
 
+int wcn36xx_smd_arp_offload(struct wcn36xx *wcn, struct ieee80211_vif *vif,
+                           bool enable)
+{
+       struct wcn36xx_vif *vif_priv = wcn36xx_vif_to_priv(vif);
+       struct wcn36xx_hal_host_offload_req_msg msg_body;
+       int ret;
+
+       mutex_lock(&wcn->hal_mutex);
+
+       INIT_HAL_MSG(msg_body, WCN36XX_HAL_HOST_OFFLOAD_REQ);
+       msg_body.host_offload_params.offload_type =
+               WCN36XX_HAL_IPV4_ARP_REPLY_OFFLOAD;
+       if (enable) {
+               msg_body.host_offload_params.enable =
+                       WCN36XX_HAL_OFFLOAD_ARP_AND_BCAST_FILTER_ENABLE;
+               memcpy(&msg_body.host_offload_params.u,
+                      &vif->bss_conf.arp_addr_list[0], sizeof(__be32));
+       }
+       msg_body.ns_offload_params.bss_index = vif_priv->bss_index;
+
+       PREPARE_HAL_BUF(wcn->hal_buf, msg_body);
+
+       ret = wcn36xx_smd_send_and_wait(wcn, msg_body.header.len);
+       if (ret) {
+               wcn36xx_err("Sending host_offload_arp failed\n");
+               goto out;
+       }
+       ret = wcn36xx_smd_rsp_status_check(wcn->hal_buf, wcn->hal_rsp_len);
+       if (ret) {
+               wcn36xx_err("host_offload_arp failed err=%d\n", ret);
+               goto out;
+       }
+out:
+       mutex_unlock(&wcn->hal_mutex);
+       return ret;
+}
+
 int wcn36xx_smd_rsp_process(struct rpmsg_device *rpdev,
                            void *buf, int len, void *priv, u32 addr)
 {
        case WCN36XX_HAL_8023_MULTICAST_LIST_RSP:
        case WCN36XX_HAL_START_SCAN_OFFLOAD_RSP:
        case WCN36XX_HAL_STOP_SCAN_OFFLOAD_RSP:
+       case WCN36XX_HAL_HOST_OFFLOAD_RSP:
                memcpy(wcn->hal_buf, buf, len);
                wcn->hal_rsp_len = len;
                complete(&wcn->hal_rsp_compl);