1 files changed, 11 insertions, 6 deletions

diff --git a/drivers/iommu/intel-iommu.c b/drivers/iommu/intel-iommu.c
index a8f7ae0eb7a4..238ad3447712 100644
--- a/drivers/iommu/intel-iommu.c
+++ b/drivers/iommu/intel-iommu.c
@@ -4730,11 +4730,16 @@ static int intel_iommu_cpu_dead(unsigned int cpu)
         return 0;
 }
 
+static inline struct intel_iommu *dev_to_intel_iommu(struct device *dev)
+{
+        return container_of(dev, struct intel_iommu, iommu.dev);
+}
+
 static ssize_t intel_iommu_show_version(struct device *dev,
                                         struct device_attribute *attr,
                                         char *buf)
 {
-        struct intel_iommu *iommu = dev_get_drvdata(dev);
+        struct intel_iommu *iommu = dev_to_intel_iommu(dev);
         u32 ver = readl(iommu->reg + DMAR_VER_REG);
         return sprintf(buf, "%d:%d\n",
                        DMAR_VER_MAJOR(ver), DMAR_VER_MINOR(ver));
@@ -4745,7 +4750,7 @@ static ssize_t intel_iommu_show_address(struct device *dev,
                                         struct device_attribute *attr,
                                         char *buf)
 {
-        struct intel_iommu *iommu = dev_get_drvdata(dev);
+        struct intel_iommu *iommu = dev_to_intel_iommu(dev);
         return sprintf(buf, "%llx\n", iommu->reg_phys);
 }
 static DEVICE_ATTR(address, S_IRUGO, intel_iommu_show_address, NULL);
@@ -4754,7 +4759,7 @@ static ssize_t intel_iommu_show_cap(struct device *dev,
                                     struct device_attribute *attr,
                                     char *buf)
 {
-        struct intel_iommu *iommu = dev_get_drvdata(dev);
+        struct intel_iommu *iommu = dev_to_intel_iommu(dev);
         return sprintf(buf, "%llx\n", iommu->cap);
 }
 static DEVICE_ATTR(cap, S_IRUGO, intel_iommu_show_cap, NULL);
@@ -4763,7 +4768,7 @@ static ssize_t intel_iommu_show_ecap(struct device *dev,
                                     struct device_attribute *attr,
                                     char *buf)
 {
-        struct intel_iommu *iommu = dev_get_drvdata(dev);
+        struct intel_iommu *iommu = dev_to_intel_iommu(dev);
         return sprintf(buf, "%llx\n", iommu->ecap);
 }
 static DEVICE_ATTR(ecap, S_IRUGO, intel_iommu_show_ecap, NULL);
@@ -4772,7 +4777,7 @@ static ssize_t intel_iommu_show_ndoms(struct device *dev,
                                       struct device_attribute *attr,
                                       char *buf)
 {
-        struct intel_iommu *iommu = dev_get_drvdata(dev);
+        struct intel_iommu *iommu = dev_to_intel_iommu(dev);
         return sprintf(buf, "%ld\n", cap_ndoms(iommu->cap));
 }
 static DEVICE_ATTR(domains_supported, S_IRUGO, intel_iommu_show_ndoms, NULL);
@@ -4781,7 +4786,7 @@ static ssize_t intel_iommu_show_ndoms_used(struct device *dev,
                                            struct device_attribute *attr,
                                            char *buf)
 {
-        struct intel_iommu *iommu = dev_get_drvdata(dev);
+        struct intel_iommu *iommu = dev_to_intel_iommu(dev);
         return sprintf(buf, "%d\n", bitmap_weight(iommu->domain_ids,
                                                   cap_ndoms(iommu->cap)));
 }