WILC_WFI_p2p_rx(g_linux_wlan->vif[1].ndev, buff, size);
 }
 
+void wl_wlan_cleanup(void)
+{
+       int i = 0;
+       perInterface_wlan_t *nic[NUM_CONCURRENT_IFC];
+
+       if (g_linux_wlan &&
+          (g_linux_wlan->vif[0].ndev || g_linux_wlan->vif[1].ndev)) {
+               unregister_inetaddr_notifier(&g_dev_notifier);
+
+               for (i = 0; i < NUM_CONCURRENT_IFC; i++)
+                       nic[i] = netdev_priv(g_linux_wlan->vif[i].ndev);
+       }
+
+       if (g_linux_wlan && g_linux_wlan->wilc_firmware)
+               release_firmware(g_linux_wlan->wilc_firmware);
+
+       if (g_linux_wlan &&
+          (g_linux_wlan->vif[0].ndev || g_linux_wlan->vif[1].ndev)) {
+               linux_wlan_lock_timeout(&close_exit_sync, 12 * 1000);
+
+               for (i = 0; i < NUM_CONCURRENT_IFC; i++)
+                       if (g_linux_wlan->vif[i].ndev)
+                               if (nic[i]->mac_opened)
+                                       mac_close(g_linux_wlan->vif[i].ndev);
+
+               for (i = 0; i < NUM_CONCURRENT_IFC; i++) {
+                       unregister_netdev(g_linux_wlan->vif[i].ndev);
+                       wilc_free_wiphy(g_linux_wlan->vif[i].ndev);
+                       free_netdev(g_linux_wlan->vif[i].ndev);
+               }
+       }
+
+       kfree(g_linux_wlan);
+
+#if defined(WILC_DEBUGFS)
+       wilc_debugfs_remove();
+#endif
+       linux_wlan_device_detection(0);
+       linux_wlan_device_power(0);
+}
+
 int wilc_netdev_init(void)
 {
 
 
 static void __exit exit_wilc_driver(void)
 {
-       int i = 0;
-       perInterface_wlan_t *nic[NUM_CONCURRENT_IFC] = {NULL,};
-       #define CLOSE_TIMEOUT (12 * 1000)
-
-       if ((g_linux_wlan != NULL) && (((g_linux_wlan->vif[0].ndev) != NULL)
-                                      || ((g_linux_wlan->vif[1].ndev) != NULL))) {
-               unregister_inetaddr_notifier(&g_dev_notifier);
-
-               for (i = 0; i < NUM_CONCURRENT_IFC; i++)
-                       nic[i] = netdev_priv(g_linux_wlan->vif[i].ndev);
-       }
-
-       if ((g_linux_wlan != NULL) && g_linux_wlan->wilc_firmware != NULL) {
-               release_firmware(g_linux_wlan->wilc_firmware);
-               g_linux_wlan->wilc_firmware = NULL;
-       }
-
-       if ((g_linux_wlan != NULL) && (((g_linux_wlan->vif[0].ndev) != NULL)
-                                      || ((g_linux_wlan->vif[1].ndev) != NULL))) {
-               PRINT_D(INIT_DBG, "Waiting for mac_close ....\n");
-
-               if (linux_wlan_lock_timeout(&close_exit_sync, CLOSE_TIMEOUT) < 0)
-                       PRINT_D(INIT_DBG, "Closed TimedOUT\n");
-               else
-                       PRINT_D(INIT_DBG, "mac_closed\n");
-
-               for (i = 0; i < NUM_CONCURRENT_IFC; i++) {
-                       /* close all opened interfaces */
-                       if (g_linux_wlan->vif[i].ndev != NULL) {
-                               if (nic[i]->mac_opened)
-                                       mac_close(g_linux_wlan->vif[i].ndev);
-                       }
-               }
-               for (i = 0; i < NUM_CONCURRENT_IFC; i++) {
-                       PRINT_D(INIT_DBG, "Unregistering netdev %p\n", g_linux_wlan->vif[i].ndev);
-                       unregister_netdev(g_linux_wlan->vif[i].ndev);
-                       PRINT_D(INIT_DBG, "Freeing Wiphy...\n");
-                       wilc_free_wiphy(g_linux_wlan->vif[i].ndev);
-                       PRINT_D(INIT_DBG, "Freeing netdev...\n");
-                       free_netdev(g_linux_wlan->vif[i].ndev);
-               }
-       }
-
 #ifndef WILC_SDIO
        PRINT_D(INIT_DBG, "SPI unregsiter...\n");
        spi_unregister_driver(&wilc_bus);
        PRINT_D(INIT_DBG, "SDIO unregsiter...\n");
        sdio_unregister_driver(&wilc_bus);
 #endif
-
-       kfree(g_linux_wlan);
-       g_linux_wlan = NULL;
-       printk("Module_exit Done.\n");
-
-#if defined(WILC_DEBUGFS)
-       wilc_debugfs_remove();
-#endif
-
-       linux_wlan_device_detection(0);
-       linux_wlan_device_power(0);
 }
 module_exit(exit_wilc_driver);