17 static int check_vma(struct dev_dax *dev_dax, struct vm_area_struct *vma,  in check_vma()  argument
20 	struct device *dev = &dev_dax->dev;  in check_vma()
23 	if (!dax_alive(dev_dax->dax_dev))  in check_vma()
34 	mask = dev_dax->align - 1;  in check_vma()
54 __weak phys_addr_t dax_pgoff_to_phys(struct dev_dax *dev_dax, pgoff_t pgoff,  in dax_pgoff_to_phys()  argument
59 	for (i = 0; i < dev_dax->nr_range; i++) {  in dax_pgoff_to_phys()
60 		struct dev_dax_range *dax_range = &dev_dax->ranges[i];  in dax_pgoff_to_phys()
76 static vm_fault_t __dev_dax_pte_fault(struct dev_dax *dev_dax,  in __dev_dax_pte_fault()  argument
79 	struct device *dev = &dev_dax->dev;  in __dev_dax_pte_fault()
83 	if (check_vma(dev_dax, vmf->vma, __func__))  in __dev_dax_pte_fault()
86 	if (dev_dax->align > PAGE_SIZE) {  in __dev_dax_pte_fault()
88 			dev_dax->align, fault_size);  in __dev_dax_pte_fault()
92 	if (fault_size != dev_dax->align)  in __dev_dax_pte_fault()
95 	phys = dax_pgoff_to_phys(dev_dax, vmf->pgoff, PAGE_SIZE);  in __dev_dax_pte_fault()
106 static vm_fault_t __dev_dax_pmd_fault(struct dev_dax *dev_dax,  in __dev_dax_pmd_fault()  argument
110 	struct device *dev = &dev_dax->dev;  in __dev_dax_pmd_fault()
115 	if (check_vma(dev_dax, vmf->vma, __func__))  in __dev_dax_pmd_fault()
118 	if (dev_dax->align > PMD_SIZE) {  in __dev_dax_pmd_fault()
120 			dev_dax->align, fault_size);  in __dev_dax_pmd_fault()
124 	if (fault_size < dev_dax->align)  in __dev_dax_pmd_fault()
126 	else if (fault_size > dev_dax->align)  in __dev_dax_pmd_fault()
135 	phys = dax_pgoff_to_phys(dev_dax, pgoff, PMD_SIZE);  in __dev_dax_pmd_fault()
147 static vm_fault_t __dev_dax_pud_fault(struct dev_dax *dev_dax,  in __dev_dax_pud_fault()  argument
151 	struct device *dev = &dev_dax->dev;  in __dev_dax_pud_fault()
157 	if (check_vma(dev_dax, vmf->vma, __func__))  in __dev_dax_pud_fault()
160 	if (dev_dax->align > PUD_SIZE) {  in __dev_dax_pud_fault()
162 			dev_dax->align, fault_size);  in __dev_dax_pud_fault()
166 	if (fault_size < dev_dax->align)  in __dev_dax_pud_fault()
168 	else if (fault_size > dev_dax->align)  in __dev_dax_pud_fault()
177 	phys = dax_pgoff_to_phys(dev_dax, pgoff, PUD_SIZE);  in __dev_dax_pud_fault()
188 static vm_fault_t __dev_dax_pud_fault(struct dev_dax *dev_dax,  in __dev_dax_pud_fault()  argument
203 	struct dev_dax *dev_dax = filp->private_data;  in dev_dax_huge_fault()  local
205 	dev_dbg(&dev_dax->dev, "%s: %s (%#lx - %#lx) size = %d\n", current->comm,  in dev_dax_huge_fault()
213 		rc = __dev_dax_pte_fault(dev_dax, vmf, &pfn);  in dev_dax_huge_fault()
217 		rc = __dev_dax_pmd_fault(dev_dax, vmf, &pfn);  in dev_dax_huge_fault()
221 		rc = __dev_dax_pud_fault(dev_dax, vmf, &pfn);  in dev_dax_huge_fault()
262 	struct dev_dax *dev_dax = filp->private_data;  in dev_dax_split()  local
264 	if (!IS_ALIGNED(addr, dev_dax->align))  in dev_dax_split()
272 	struct dev_dax *dev_dax = filp->private_data;  in dev_dax_pagesize()  local
274 	return dev_dax->align;  in dev_dax_pagesize()
286 	struct dev_dax *dev_dax = filp->private_data;  in dax_mmap()  local
289 	dev_dbg(&dev_dax->dev, "trace\n");  in dax_mmap()
296 	rc = check_vma(dev_dax, vma, __func__);  in dax_mmap()
312 	struct dev_dax *dev_dax = filp ? filp->private_data : NULL;  in dax_get_unmapped_area()  local
314 	if (!dev_dax || addr)  in dax_get_unmapped_area()
317 	align = dev_dax->align;  in dax_get_unmapped_area()
348 	struct dev_dax *dev_dax = dax_get_private(dax_dev);  in dax_open()  local
350 	dev_dbg(&dev_dax->dev, "trace\n");  in dax_open()
357 	filp->private_data = dev_dax;  in dax_open()
365 	struct dev_dax *dev_dax = filp->private_data;  in dax_release()  local
367 	dev_dbg(&dev_dax->dev, "trace\n");  in dax_release()
386 static void dev_dax_kill(void *dev_dax)  in dev_dax_kill()  argument
388 	kill_dev_dax(dev_dax);  in dev_dax_kill()
391 int dev_dax_probe(struct dev_dax *dev_dax)  in dev_dax_probe()  argument
393 	struct dax_device *dax_dev = dev_dax->dax_dev;  in dev_dax_probe()
394 	struct device *dev = &dev_dax->dev;  in dev_dax_probe()
401 	pgmap = dev_dax->pgmap;  in dev_dax_probe()
402 	if (dev_WARN_ONCE(dev, pgmap && dev_dax->nr_range > 1,  in dev_dax_probe()
408 				* (dev_dax->nr_range - 1), GFP_KERNEL);  in dev_dax_probe()
411 		pgmap->nr_range = dev_dax->nr_range;  in dev_dax_probe()
414 	for (i = 0; i < dev_dax->nr_range; i++) {  in dev_dax_probe()
415 		struct range *range = &dev_dax->ranges[i].range;  in dev_dax_probe()
424 		if (!dev_dax->pgmap)  in dev_dax_probe()
451 	return devm_add_action_or_reset(dev, dev_dax_kill, dev_dax);  in dev_dax_probe()
455 static int dev_dax_remove(struct dev_dax *dev_dax)  in dev_dax_remove()  argument