// SPDX-License-Identifier: GPL-2.0
/*
 * Generic PCI resource mmap helper
 *
 * Copyright © 2017 Amazon.com, Inc. or its affiliates.
 *
 * Author: David Woodhouse <dwmw2@infradead.org>
 */

#include <linux/kernel.h>
#include <linux/mm.h>
#include <linux/pci.h>

#include "pci.h"

#ifdef ARCH_GENERIC_PCI_MMAP_RESOURCE

static const struct vm_operations_struct pci_phys_vm_ops = {
#ifdef CONFIG_HAVE_IOREMAP_PROT
 .access = generic_access_phys,
#endif
};

int pci_mmap_resource_range(struct pci_dev *pdev, int bar,
       struct vm_area_struct *vma,
       enum pci_mmap_state mmap_state, int write_combine)
{
 unsigned long size;
 int ret;

 size = ((pci_resource_len(pdev, bar) - 1) >> PAGE_SHIFT) + 1;
 if (vma->vm_pgoff + vma_pages(vma) > size)
  return -EINVAL;

 if (write_combine)
  vma->vm_page_prot = pgprot_writecombine(vma->vm_page_prot);
 else
  vma->vm_page_prot = pgprot_device(vma->vm_page_prot);

 if (mmap_state == pci_mmap_io) {
  ret = pci_iobar_pfn(pdev, bar, vma);
  if (ret)
   return ret;
 } else
  vma->vm_pgoff += (pci_resource_start(pdev, bar) >> PAGE_SHIFT);

 vma->vm_ops = &pci_phys_vm_ops;

 return io_remap_pfn_range(vma, vma->vm_start, vma->vm_pgoff,
      vma->vm_end - vma->vm_start,
      vma->vm_page_prot);
}

#endif

#if (defined(CONFIG_SYSFS) || defined(CONFIG_PROC_FS)) && \
    (defined(HAVE_PCI_MMAP) || defined(ARCH_GENERIC_PCI_MMAP_RESOURCE))

int pci_mmap_fits(struct pci_dev *pdev, int resno, struct vm_area_struct *vma,
    enum pci_mmap_api mmap_api)
{
 resource_size_t pci_start = 0, pci_end;
 unsigned long nr, start, size;

 if (pci_resource_len(pdev, resno) == 0)
  return 0;
 nr = vma_pages(vma);
 start = vma->vm_pgoff;
 size = ((pci_resource_len(pdev, resno) - 1) >> PAGE_SHIFT) + 1;
 if (mmap_api == PCI_MMAP_PROCFS) {
  pci_resource_to_user(pdev, resno, &pdev->resource[resno],
         &pci_start, &pci_end);
  pci_start >>= PAGE_SHIFT;
 }
 if (start >= pci_start && start < pci_start + size &&
     start + nr <= pci_start + size)
  return 1;
 return 0;
}

#endif