PCI: Cleanup PCI_REBAR_CTRL_BAR_SHIFT handling

Cleanup PCI_REBAR_CTRL_BAR_SHIFT handling.  That was hard coded instead of
properly defined in the header for some reason.

Signed-off-by: Christian König <christian.koenig@amd.com>
Signed-off-by: Bjorn Helgaas <bhelgaas@google.com>

2 files changed, 5 insertions, 4 deletions

diff --git a/drivers/pci/pci.c b/drivers/pci/pci.c
index 44ccfb31363e..1b20c4392f09 100644
--- a/drivers/pci/pci.c
+++ b/drivers/pci/pci.c
@@ -1193,7 +1193,7 @@ static void pci_restore_rebar_state(struct pci_dev *pdev)
                 res = pdev->resource + bar_idx;
                 size = order_base_2((resource_size(res) >> 20) | 1) - 1;
                 ctrl &= ~PCI_REBAR_CTRL_BAR_SIZE;
-                ctrl |= size << 8;
+                ctrl |= size << PCI_REBAR_CTRL_BAR_SHIFT;
                 pci_write_config_dword(pdev, pos + PCI_REBAR_CTRL, ctrl);
         }
 }
@@ -3098,7 +3098,7 @@ int pci_rebar_get_current_size(struct pci_dev *pdev, int bar)
                 return pos;
 
         pci_read_config_dword(pdev, pos + PCI_REBAR_CTRL, &ctrl);
-        return (ctrl & PCI_REBAR_CTRL_BAR_SIZE) >> 8;
+        return (ctrl & PCI_REBAR_CTRL_BAR_SIZE) >> PCI_REBAR_CTRL_BAR_SHIFT;
 }
 
 /**
@@ -3121,7 +3121,7 @@ int pci_rebar_set_size(struct pci_dev *pdev, int bar, int size)
 
         pci_read_config_dword(pdev, pos + PCI_REBAR_CTRL, &ctrl);
         ctrl &= ~PCI_REBAR_CTRL_BAR_SIZE;
-        ctrl |= size << 8;
+        ctrl |= size << PCI_REBAR_CTRL_BAR_SHIFT;
         pci_write_config_dword(pdev, pos + PCI_REBAR_CTRL, ctrl);
         return 0;
 }
diff --git a/include/uapi/linux/pci_regs.h b/include/uapi/linux/pci_regs.h
index 4da87e2ef8a8..82e6b361204e 100644
--- a/include/uapi/linux/pci_regs.h
+++ b/include/uapi/linux/pci_regs.h
@@ -960,8 +960,9 @@
 #define PCI_REBAR_CTRL          8       /* control register */
 #define  PCI_REBAR_CTRL_BAR_IDX         0x00000007  /* BAR index */
 #define  PCI_REBAR_CTRL_NBAR_MASK       0x000000E0  /* # of resizable BARs */
-#define  PCI_REBAR_CTRL_NBAR_SHIFT      5           /* shift for # of BARs */
+#define  PCI_REBAR_CTRL_NBAR_SHIFT      5           /* shift for # of BARs */
 #define  PCI_REBAR_CTRL_BAR_SIZE        0x00001F00  /* BAR size */
+#define  PCI_REBAR_CTRL_BAR_SHIFT       8           /* shift for BAR size */
 
 /* Dynamic Power Allocation */
 #define PCI_DPA_CAP             4       /* capability register */