#include "mt7921.h"
 #include "../mt76_connac2_mac.h"
 #include "../sdio.h"
+#include <linux/mmc/host.h>
+#include <linux/kallsyms.h>
 
 static void mt7921s_enable_irq(struct mt76_dev *dev)
 {
        struct mt76_sdio *sdio = &dev->mt76.sdio;
        u32 val, status;
 
+       if (atomic_read(&dev->mt76.bus_hung))
+               return 0;
+
        mt7921s_mcu_drv_pmctrl(dev);
 
        sdio_claim_host(sdio->func);
        return 0;
 }
 
+static struct mt76_sdio *msdio;
+static void mt7921s_card_reset(struct work_struct *work)
+{
+       struct mmc_host *sdio_host = msdio->func->card->host;
+
+       sdio_claim_host(msdio->func);
+       sdio_release_irq(msdio->func);
+       sdio_release_host(msdio->func);
+
+       mmc_remove_host(sdio_host);
+       msleep(50);
+       mmc_add_host(sdio_host);
+}
+
+static DECLARE_WORK(sdio_reset_work, mt7921s_card_reset);
+static int mt7921s_check_bus(struct mt76_dev *dev)
+{
+       struct mt76_sdio *sdio = &dev->sdio;
+       int err;
+
+       sdio_claim_host(sdio->func);
+       sdio_readl(dev->sdio.func, MCR_WHCR, &err);
+       sdio_release_host(sdio->func);
+
+       return err;
+}
+
+static int mt7921s_host_reset(struct mt792x_dev *dev)
+{
+       struct mt76_dev *mdev = &dev->mt76;
+       int err = -1;
+
+       if (!atomic_read(&mdev->bus_hung))
+               err = mt7921s_check_bus(&dev->mt76);
+
+       if (err) {
+               atomic_set(&mdev->bus_hung, true);
+               msdio = &dev->mt76.sdio;
+               dev_err(mdev->dev, "SDIO bus problem detected(%d), resetting card!!\n", err);
+               schedule_work(&sdio_reset_work);
+               return err;
+       }
+
+       atomic_set(&mdev->bus_hung, false);
+
+       return 0;
+}
+
 int mt7921s_mac_reset(struct mt792x_dev *dev)
 {
        int err;
 
        mt76_connac_free_pending_tx_skbs(&dev->pm, NULL);
+
+       mt7921s_host_reset(dev);
+       if (atomic_read(&dev->mt76.bus_hung))
+               return 0;
+
        mt76_txq_schedule_all(&dev->mphy);
        mt76_worker_disable(&dev->mt76.tx_worker);
        set_bit(MT76_MCU_RESET, &dev->mphy.state);
 
 
        if (err < 0) {
                dev_err(dev->dev, "sdio read data failed:%d\n", err);
+               atomic_set(&dev->bus_hung, true);
                put_page(page);
                return err;
        }
        err = sdio_writesb(sdio->func, MCR_WTDR1, data, len);
        sdio_release_host(sdio->func);
 
-       if (err)
+       if (err) {
                dev_err(dev->dev, "sdio write failed: %d\n", err);
-
+               atomic_set(&dev->bus_hung, true);
+       }
        return err;
 }