// SPDX-License-Identifier: GPL-2.0-only

#include <linux/uaccess.h>
#include <linux/kernel.h>

#include <asm/vsyscall.h>

#ifdef CONFIG_X86_64
bool copy_from_kernel_nofault_allowed(const void *unsafe_src, size_t size)
{
 unsigned long vaddr = (unsigned long)unsafe_src;

 /*
 * Do not allow userspace addresses.  This disallows
 * normal userspace and the userspace guard page:
 */
 if (vaddr < TASK_SIZE_MAX + PAGE_SIZE)
  return false;

 /*
 * Reading from the vsyscall page may cause an unhandled fault in
 * certain cases.  Though it is at an address above TASK_SIZE_MAX, it is
 * usually considered as a user space address.
 */
 if (is_vsyscall_vaddr(vaddr))
  return false;

 /*
 * Allow everything during early boot before 'x86_virt_bits'
 * is initialized.  Needed for instruction decoding in early
 * exception handlers.
 */
 if (!boot_cpu_data.x86_virt_bits)
  return true;

 return __is_canonical_address(vaddr, boot_cpu_data.x86_virt_bits);
}
#else
bool copy_from_kernel_nofault_allowed(const void *unsafe_src, size_t size)
{
 return (unsigned long)unsafe_src >= TASK_SIZE_MAX;
}
#endif