KERNEL: Use pte offset kernel function for mapped pages

tzafrir-mellanox · Oct 2, 2023 · f642033 · f642033
1 parent e39bb4d
commit f642033
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kernel/xpmem_pfn.c b/kernel/xpmem_pfn.c
@@ -164,7 +164,7 @@ xpmem_vaddr_to_pte_offset(struct mm_struct *mm, u64 vaddr, u64 *offset)
 	}
 #endif
 
-	pte = pte_offset_map(pmd, vaddr);
+	pte = pte_offset_kernel(pmd, vaddr);
 	if (!pte_present(*pte))
 		return NULL;
 
@@ -218,7 +218,7 @@ xpmem_vaddr_to_pte_size(struct mm_struct *mm, u64 vaddr, u64 *size)
 		return NULL;
 	}
 
-	pte = pte_offset_map(pmd, vaddr);
+	pte = pte_offset_kernel(pmd, vaddr);
 	if (!pte_present(*pte)) {
 		*size = PAGE_SIZE;
 		return NULL;