Add pci class code for SATA & AHCI, and replace some magic numbers.

Signed-off-by: Conke Hu <conke.hu@amd.com>
Signed-off-by: Jeff Garzik <jeff@garzik.org>

2 files changed, 5 insertions, 5 deletions

diff --git a/drivers/ata/ahci.c b/drivers/ata/ahci.c
index e2796fb40eb7..32cfaa89dc7c 100644
--- a/drivers/ata/ahci.c
+++ b/drivers/ata/ahci.c
@@ -431,7 +431,7 @@ static const struct pci_device_id ahci_pci_tbl[] = {
 
         /* Generic, PCI class code for AHCI */
         { PCI_ANY_ID, PCI_ANY_ID, PCI_ANY_ID, PCI_ANY_ID,
-          0x010601, 0xffffff, board_ahci },
+          PCI_CLASS_STORAGE_SATA_AHCI, 0xffffff, board_ahci },
 
         { }     /* terminate list */
 };
@@ -1619,11 +1619,11 @@ static void ahci_print_info(struct ata_probe_ent *probe_ent)
                 speed_s = "?";
 
         pci_read_config_word(pdev, 0x0a, &cc);
-        if (cc == 0x0101)
+        if (cc == PCI_CLASS_STORAGE_IDE)
                 scc_s = "IDE";
-        else if (cc == 0x0106)
+        else if (cc == PCI_CLASS_STORAGE_SATA)
                 scc_s = "SATA";
-        else if (cc == 0x0104)
+        else if (cc == PCI_CLASS_STORAGE_RAID)
                 scc_s = "RAID";
         else
                 scc_s = "unknown";
diff --git a/drivers/pci/quirks.c b/drivers/pci/quirks.c
index 11217bda4b9e..1e6eda25c0d8 100644
--- a/drivers/pci/quirks.c
+++ b/drivers/pci/quirks.c
@@ -871,7 +871,7 @@ static void __devinit quirk_sb600_sata(struct pci_dev *pdev)
                 pci_write_config_byte(pdev, 0xa, 6);
                 pci_write_config_byte(pdev, 0x40, tmp);
 
-                pdev->class = 0x010601;
+                pdev->class = PCI_CLASS_STORAGE_SATA_AHCI;
         }
 }
 DECLARE_PCI_FIXUP_HEADER(PCI_VENDOR_ID_ATI, PCI_DEVICE_ID_ATI_IXP600_SATA, quirk_sb600_sata);