/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _LINUX_IOMMU_HELPER_H
#define _LINUX_IOMMU_HELPER_H

#include <linux/bug.h>
#include <linux/log2.h>
#include <linux/math.h>
#include <linux/types.h>

static inline unsigned long iommu_device_max_index(unsigned long size,
                           unsigned long offset,
                           u64 dma_mask)
{
    if (size + offset > dma_mask)
        return dma_mask - offset + 1;
    else
        return size;
}

static inline int iommu_is_span_boundary(unsigned int index, unsigned int nr,
        unsigned long shift, unsigned long boundary_size)
{
    BUG_ON(!is_power_of_2(boundary_size));

    shift = (shift + index) & (boundary_size - 1);
    return shift + nr > boundary_size;
}

extern unsigned long iommu_area_alloc(unsigned long *map, unsigned long size,
                      unsigned long start, unsigned int nr,
                      unsigned long shift,
                      unsigned long boundary_size,
                      unsigned long align_mask);

static inline unsigned long iommu_num_pages(unsigned long addr,
                        unsigned long len,
                        unsigned long io_page_size)
{
    unsigned long size = (addr & (io_page_size - 1)) + len;

    return DIV_ROUND_UP(size, io_page_size);
}

#endif