#include <mach_assert.h>
#include <sys/errno.h>
#include <i386/param.h>
#include <i386/misc_protos.h>
#include <i386/cpu_data.h>
#include <i386/machine_routines.h>
#include <i386/cpuid.h>
#include <i386/vmx.h>
#include <vm/pmap.h>
#include <vm/vm_map.h>
#include <vm/vm_kern.h>
#include <vm/vm_fault.h>
#include <san/kasan.h>
#include <sys/kdebug.h>
#include <kern/copyout_shim.h>
static int copyio(int, user_addr_t, char *, vm_size_t, vm_size_t *, int);
static int copyio_phys(addr64_t, addr64_t, vm_size_t, int);
const int copysize_limit_panic = (64 * MB);
extern int _bcopy(const void *, void *, vm_size_t);
extern int _bcopystr(const void *, void *, vm_size_t, vm_size_t *);
extern int _copyin_word(const char *src, uint64_t *dst, vm_size_t len);
#define COPYIN 0
#define COPYOUT 1
#define COPYINSTR 2
#define COPYINPHYS 3
#define COPYOUTPHYS 4
#define COPYINWORD 5
#if ENABLE_SMAPLOG
typedef struct {
uint64_t timestamp;
thread_t thread;
uintptr_t cr4;
uint8_t cpuid;
uint8_t smap_state;
uint8_t copyio_active;
} smaplog_entry_t;
#define SMAPLOG_BUFFER_SIZE (50)
static smaplog_entry_t smaplog_cbuf[SMAPLOG_BUFFER_SIZE];
static uint32_t smaplog_head = 0;
static void
smaplog_add_entry(boolean_t enabling)
{
uint32_t index = 0;
thread_t thread = current_thread();
do {
index = smaplog_head;
} while (!OSCompareAndSwap(index, (index + 1) % SMAPLOG_BUFFER_SIZE, &smaplog_head));
assert(index < SMAPLOG_BUFFER_SIZE);
assert(smaplog_head < SMAPLOG_BUFFER_SIZE);
assert(thread);
smaplog_cbuf[index].timestamp = mach_absolute_time();
smaplog_cbuf[index].thread = thread;
smaplog_cbuf[index].cpuid = cpu_number();
smaplog_cbuf[index].cr4 = get_cr4();
smaplog_cbuf[index].smap_state = enabling;
smaplog_cbuf[index].copyio_active = (thread->machine.specFlags & CopyIOActive) ? 1 : 0;
}
#endif
extern boolean_t pmap_smap_enabled;
static inline void user_access_enable(void) {
if (pmap_smap_enabled) {
stac();
#if ENABLE_SMAPLOG
smaplog_add_entry(TRUE);
#endif
}
}
static inline void user_access_disable(void) {
if (pmap_smap_enabled) {
clac();
#if ENABLE_SMAPLOG
smaplog_add_entry(FALSE);
#endif
}
}
#if COPYIO_TRACE_ENABLED
#define COPYIO_TRACE(x, a, b, c, d, e) KERNEL_DEBUG_CONSTANT(x, a, b, c, d, e)
#else
#define COPYIO_TRACE(x, a, b, c, d, e) do { } while(0)
#endif
static int
copyio(int copy_type, user_addr_t user_addr, char *kernel_addr,
vm_size_t nbytes, vm_size_t *lencopied, int use_kernel_map)
{
thread_t thread = current_thread();
pmap_t pmap;
vm_size_t bytes_copied;
int error = 0;
boolean_t istate = FALSE;
boolean_t recursive_CopyIOActive;
#if COPYIO_TRACE_ENABLED
int debug_type = 0xeff70010;
debug_type += (copy_type << 2);
#endif
if (__improbable(nbytes > copysize_limit_panic))
panic("%s(%p, %p, %lu) - transfer too large", __func__,
(void *)user_addr, (void *)kernel_addr, nbytes);
COPYIO_TRACE(debug_type | DBG_FUNC_START,
user_addr, kernel_addr, nbytes, use_kernel_map, 0);
if (__improbable(nbytes == 0))
goto out;
pmap = thread->map->pmap;
boolean_t nopagezero = pmap->pagezero_accessible;
if (__improbable((copy_type != COPYINPHYS) && (copy_type != COPYOUTPHYS) && ((vm_offset_t)kernel_addr < VM_MIN_KERNEL_AND_KEXT_ADDRESS))) {
panic("Invalid copy parameter, copy type: %d, kernel address: %p", copy_type, kernel_addr);
}
if (__improbable(((pmap != kernel_pmap) && (use_kernel_map == 0)) &&
((nbytes && (user_addr+nbytes <= user_addr)) || ((user_addr + nbytes) > vm_map_max(thread->map))))) {
error = EFAULT;
goto out;
}
#if KASAN
if (copy_type == COPYIN || copy_type == COPYINSTR || copy_type == COPYINWORD) {
__asan_storeN((uptr)kernel_addr, nbytes);
} else if (copy_type == COPYOUT) {
__asan_loadN((uptr)kernel_addr, nbytes);
}
#endif
recursive_CopyIOActive = thread->machine.specFlags & CopyIOActive;
boolean_t pdswitch = no_shared_cr3 || nopagezero;
if (__improbable(pdswitch)) {
istate = ml_set_interrupts_enabled(FALSE);
if (nopagezero && pmap_pcid_ncpus) {
pmap_pcid_activate(pmap, cpu_number(), TRUE, TRUE);
} else if (get_cr3_base() != pmap->pm_cr3) {
set_cr3_raw(pmap->pm_cr3);
}
thread->machine.specFlags |= CopyIOActive;
} else {
thread->machine.specFlags |= CopyIOActive;
}
user_access_enable();
#if DEVELOPMENT || DEBUG
if ((pmap != kernel_pmap) && !use_kernel_map &&
(get_cr3_base() != pmap->pm_cr3)) {
panic("copyio(%d,%p,%p,%ld,%p,%d) cr3 is %p expects %p",
copy_type, (void *)user_addr, kernel_addr, nbytes, lencopied, use_kernel_map,
(void *) get_cr3_raw(), (void *) pmap->pm_cr3);
}
#endif
if (__improbable(pdswitch)) {
(void) ml_set_interrupts_enabled(istate);
}
COPYIO_TRACE(0xeff70044 | DBG_FUNC_NONE, user_addr,
kernel_addr, nbytes, 0, 0);
switch (copy_type) {
case COPYIN:
error = _bcopy((const void *) user_addr,
kernel_addr,
nbytes);
break;
case COPYOUT:
error = _bcopy(kernel_addr,
(void *) user_addr,
nbytes);
break;
case COPYINPHYS:
error = _bcopy((const void *) user_addr,
PHYSMAP_PTOV(kernel_addr),
nbytes);
break;
case COPYOUTPHYS:
error = _bcopy((const void *) PHYSMAP_PTOV(kernel_addr),
(void *) user_addr,
nbytes);
break;
case COPYINWORD:
error = _copyin_word((const void *) user_addr,
(void *) kernel_addr,
nbytes);
break;
case COPYINSTR:
error = _bcopystr((const void *) user_addr,
kernel_addr,
(int) nbytes,
&bytes_copied);
if (error != EFAULT)
*lencopied = bytes_copied;
if (error) {
#if KDEBUG
nbytes = *lencopied;
#endif
break;
}
if (*(kernel_addr + bytes_copied - 1) == 0) {
#if KDEBUG
nbytes = *lencopied;
#endif
break;
} else {
#if KDEBUG
nbytes = *lencopied;
#endif
error = ENAMETOOLONG;
break;
}
}
user_access_disable();
if (__improbable(pdswitch)) {
istate = ml_set_interrupts_enabled(FALSE);
if (!recursive_CopyIOActive && (get_cr3_raw() != kernel_pmap->pm_cr3)) {
if (nopagezero && pmap_pcid_ncpus) {
pmap_pcid_activate(pmap, cpu_number(), TRUE, FALSE);
} else {
set_cr3_raw(kernel_pmap->pm_cr3);
}
}
if (!recursive_CopyIOActive) {
thread->machine.specFlags &= ~CopyIOActive;
}
(void) ml_set_interrupts_enabled(istate);
} else if (!recursive_CopyIOActive) {
thread->machine.specFlags &= ~CopyIOActive;
}
out:
COPYIO_TRACE(debug_type | DBG_FUNC_END, user_addr, kernel_addr, nbytes, error, 0);
return (error);
}
static int
copyio_phys(addr64_t source, addr64_t sink, vm_size_t csize, int which)
{
char *paddr;
user_addr_t vaddr;
int ctype;
if (which & cppvPsnk) {
paddr = (char *)sink;
vaddr = (user_addr_t)source;
ctype = COPYINPHYS;
} else {
paddr = (char *)source;
vaddr = (user_addr_t)sink;
ctype = COPYOUTPHYS;
CALL_COPYOUT_SHIM_PHYS((void *)PHYSMAP_PTOV(source),sink,csize)
}
return copyio(ctype, vaddr, paddr, csize, NULL, which & cppvKmap);
}
int
copyinmsg(const user_addr_t user_addr, char *kernel_addr, mach_msg_size_t nbytes)
{
return copyio(COPYIN, user_addr, kernel_addr, nbytes, NULL, 0);
}
int
copyin(const user_addr_t user_addr, char *kernel_addr, vm_size_t nbytes)
{
return copyio(COPYIN, user_addr, kernel_addr, nbytes, NULL, 0);
}
int
copyin_word(const user_addr_t user_addr, uint64_t *kernel_addr, vm_size_t nbytes)
{
if ((nbytes != 4) && (nbytes != 8))
return EINVAL;
if (user_addr & (nbytes - 1))
return EINVAL;
return copyio(COPYINWORD, user_addr, (char *)(uintptr_t)kernel_addr, nbytes, NULL, 0);
}
int
copyinstr(const user_addr_t user_addr, char *kernel_addr, vm_size_t nbytes, vm_size_t *lencopied)
{
*lencopied = 0;
return copyio(COPYINSTR, user_addr, kernel_addr, nbytes, lencopied, 0);
}
int
copyoutmsg(const char *kernel_addr, user_addr_t user_addr, mach_msg_size_t nbytes)
{
CALL_COPYOUT_SHIM_MSG(kernel_addr,user_addr,(vm_size_t)nbytes)
return copyio(COPYOUT, user_addr, (char *)(uintptr_t)kernel_addr, nbytes, NULL, 0);
}
int
copyout(const void *kernel_addr, user_addr_t user_addr, vm_size_t nbytes)
{
CALL_COPYOUT_SHIM_NRML(kernel_addr,user_addr,nbytes)
return copyio(COPYOUT, user_addr, (char *)(uintptr_t)kernel_addr, nbytes, NULL, 0);
}
kern_return_t
copypv(addr64_t src64, addr64_t snk64, unsigned int size, int which)
{
unsigned int lop, csize;
int bothphys = 0;
KERNEL_DEBUG(0xeff7004c | DBG_FUNC_START, (unsigned)src64,
(unsigned)snk64, size, which, 0);
if ((which & (cppvPsrc | cppvPsnk)) == 0 )
panic("copypv: no more than 1 parameter may be virtual\n");
if ((which & (cppvPsrc | cppvPsnk)) == (cppvPsrc | cppvPsnk))
bothphys = 1;
while (size) {
if (bothphys) {
lop = (unsigned int)(PAGE_SIZE - (snk64 & (PAGE_SIZE - 1)));
if (lop > (unsigned int)(PAGE_SIZE - (src64 & (PAGE_SIZE - 1))))
lop = (unsigned int)(PAGE_SIZE - (src64 & (PAGE_SIZE - 1)));
} else {
if (which & cppvPsrc)
lop = (unsigned int)(PAGE_SIZE - (src64 & (PAGE_SIZE - 1)));
else
lop = (unsigned int)(PAGE_SIZE - (snk64 & (PAGE_SIZE - 1)));
}
csize = size;
if (lop < size)
csize = lop;
#if 0
if (which & cppvFsrc)
flush_dcache64(src64, csize, 1);
if (which & cppvFsnk)
flush_dcache64(snk64, csize, 1);
#endif
if (bothphys)
bcopy_phys(src64, snk64, csize);
else {
if (copyio_phys(src64, snk64, csize, which))
return (KERN_FAILURE);
}
#if 0
if (which & cppvFsrc)
flush_dcache64(src64, csize, 1);
if (which & cppvFsnk)
flush_dcache64(snk64, csize, 1);
#endif
size -= csize;
snk64 += csize;
src64 += csize;
}
KERNEL_DEBUG(0xeff7004c | DBG_FUNC_END, (unsigned)src64,
(unsigned)snk64, size, which, 0);
return KERN_SUCCESS;
}