/* quirk table based on DMI matching */
 static const struct dmi_system_id mwifiex_quirk_table[] = {
+       {
+               .ident = "Surface Pro 4",
+               .matches = {
+                       DMI_EXACT_MATCH(DMI_SYS_VENDOR, "Microsoft Corporation"),
+                       DMI_EXACT_MATCH(DMI_PRODUCT_NAME, "Surface Pro 4"),
+               },
+               .driver_data = (void *)QUIRK_FW_RST_D3COLD,
+       },
+       {
+               .ident = "Surface Pro 5",
+               .matches = {
+                       /* match for SKU here due to generic product name "Surface Pro" */
+                       DMI_EXACT_MATCH(DMI_SYS_VENDOR, "Microsoft Corporation"),
+                       DMI_EXACT_MATCH(DMI_PRODUCT_SKU, "Surface_Pro_1796"),
+               },
+               .driver_data = (void *)QUIRK_FW_RST_D3COLD,
+       },
+       {
+               .ident = "Surface Pro 5 (LTE)",
+               .matches = {
+                       /* match for SKU here due to generic product name "Surface Pro" */
+                       DMI_EXACT_MATCH(DMI_SYS_VENDOR, "Microsoft Corporation"),
+                       DMI_EXACT_MATCH(DMI_PRODUCT_SKU, "Surface_Pro_1807"),
+               },
+               .driver_data = (void *)QUIRK_FW_RST_D3COLD,
+       },
+       {
+               .ident = "Surface Pro 6",
+               .matches = {
+                       DMI_EXACT_MATCH(DMI_SYS_VENDOR, "Microsoft Corporation"),
+                       DMI_EXACT_MATCH(DMI_PRODUCT_NAME, "Surface Pro 6"),
+               },
+               .driver_data = (void *)QUIRK_FW_RST_D3COLD,
+       },
+       {
+               .ident = "Surface Book 1",
+               .matches = {
+                       DMI_EXACT_MATCH(DMI_SYS_VENDOR, "Microsoft Corporation"),
+                       DMI_EXACT_MATCH(DMI_PRODUCT_NAME, "Surface Book"),
+               },
+               .driver_data = (void *)QUIRK_FW_RST_D3COLD,
+       },
+       {
+               .ident = "Surface Book 2",
+               .matches = {
+                       DMI_EXACT_MATCH(DMI_SYS_VENDOR, "Microsoft Corporation"),
+                       DMI_EXACT_MATCH(DMI_PRODUCT_NAME, "Surface Book 2"),
+               },
+               .driver_data = (void *)QUIRK_FW_RST_D3COLD,
+       },
+       {
+               .ident = "Surface Laptop 1",
+               .matches = {
+                       DMI_EXACT_MATCH(DMI_SYS_VENDOR, "Microsoft Corporation"),
+                       DMI_EXACT_MATCH(DMI_PRODUCT_NAME, "Surface Laptop"),
+               },
+               .driver_data = (void *)QUIRK_FW_RST_D3COLD,
+       },
+       {
+               .ident = "Surface Laptop 2",
+               .matches = {
+                       DMI_EXACT_MATCH(DMI_SYS_VENDOR, "Microsoft Corporation"),
+                       DMI_EXACT_MATCH(DMI_PRODUCT_NAME, "Surface Laptop 2"),
+               },
+               .driver_data = (void *)QUIRK_FW_RST_D3COLD,
+       },
        {}
 };
 
 
        if (!card->quirks)
                dev_info(&pdev->dev, "no quirks enabled\n");
+       if (card->quirks & QUIRK_FW_RST_D3COLD)
+               dev_info(&pdev->dev, "quirk reset_d3cold enabled\n");
+}
+
+static void mwifiex_pcie_set_power_d3cold(struct pci_dev *pdev)
+{
+       dev_info(&pdev->dev, "putting into D3cold...\n");
+
+       pci_save_state(pdev);
+       if (pci_is_enabled(pdev))
+               pci_disable_device(pdev);
+       pci_set_power_state(pdev, PCI_D3cold);
+}
+
+static int mwifiex_pcie_set_power_d0(struct pci_dev *pdev)
+{
+       int ret;
+
+       dev_info(&pdev->dev, "putting into D0...\n");
+
+       pci_set_power_state(pdev, PCI_D0);
+       ret = pci_enable_device(pdev);
+       if (ret) {
+               dev_err(&pdev->dev, "pci_enable_device failed\n");
+               return ret;
+       }
+       pci_restore_state(pdev);
+
+       return 0;
+}
+
+int mwifiex_pcie_reset_d3cold_quirk(struct pci_dev *pdev)
+{
+       struct pci_dev *parent_pdev = pci_upstream_bridge(pdev);
+       int ret;
+
+       /* Power-cycle (put into D3cold then D0) */
+       dev_info(&pdev->dev, "Using reset_d3cold quirk to perform FW reset\n");
+
+       /* We need to perform power-cycle also for bridge of wifi because
+        * on some devices (e.g. Surface Book 1), the OS for some reasons
+        * can't know the real power state of the bridge.
+        * When tried to power-cycle only wifi, the reset failed with the
+        * following dmesg log:
+        * "Cannot transition to power state D0 for parent in D3hot".
+        */
+       mwifiex_pcie_set_power_d3cold(pdev);
+       mwifiex_pcie_set_power_d3cold(parent_pdev);
+
+       ret = mwifiex_pcie_set_power_d0(parent_pdev);
+       if (ret)
+               return ret;
+       ret = mwifiex_pcie_set_power_d0(pdev);
+       if (ret)
+               return ret;
+
+       return 0;
 }