1 files changed, 3 insertions, 6 deletions

diff --git a/drivers/iommu/iommu.c b/drivers/iommu/iommu.c
index 6e6b6a11b3ce..e61a9bad6df3 100644
--- a/drivers/iommu/iommu.c
+++ b/drivers/iommu/iommu.c
@@ -16,6 +16,7 @@
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  */
 
+#include <linux/kernel.h>
 #include <linux/bug.h>
 #include <linux/types.h>
 #include <linux/module.h>
@@ -97,13 +98,11 @@ EXPORT_SYMBOL_GPL(iommu_domain_has_cap);
 int iommu_map(struct iommu_domain *domain, unsigned long iova,
               phys_addr_t paddr, int gfp_order, int prot)
 {
-        unsigned long invalid_mask;
         size_t size;
 
         size         = 0x1000UL << gfp_order;
-        invalid_mask = size - 1;
 
-        BUG_ON((iova | paddr) & invalid_mask);
+        BUG_ON(!IS_ALIGNED(iova | paddr, size));
 
         return iommu_ops->map(domain, iova, paddr, gfp_order, prot);
 }
@@ -111,13 +110,11 @@ EXPORT_SYMBOL_GPL(iommu_map);
 
 int iommu_unmap(struct iommu_domain *domain, unsigned long iova, int gfp_order)
 {
-        unsigned long invalid_mask;
         size_t size;
 
         size         = 0x1000UL << gfp_order;
-        invalid_mask = size - 1;
 
-        BUG_ON(iova & invalid_mask);
+        BUG_ON(!IS_ALIGNED(iova, size));
 
         return iommu_ops->unmap(domain, iova, gfp_order);
 }