sys/riscv/riscv/copyinout.S

   1 /*-
   2  * Copyright (c) 2015-2018 Ruslan Bukin <br@bsdpad.com>
   3  * All rights reserved.
   4  *
   5  * Portions of this software were developed by SRI International and the
   6  * University of Cambridge Computer Laboratory under DARPA/AFRL contract
   7  * FA8750-10-C-0237 ("CTSRD"), as part of the DARPA CRASH research programme.
   8  *
   9  * Portions of this software were developed by the University of Cambridge
  10  * Computer Laboratory as part of the CTSRD Project, with support from the
  11  * UK Higher Education Innovation Fund (HEIF).
  12  *
  13  * Redistribution and use in source and binary forms, with or without
  14  * modification, are permitted provided that the following conditions
  15  * are met:
  16  * 1. Redistributions of source code must retain the above copyright
  17  *    notice, this list of conditions and the following disclaimer.
  18  * 2. Redistributions in binary form must reproduce the above copyright
  19  *    notice, this list of conditions and the following disclaimer in the
  20  *    documentation and/or other materials provided with the distribution.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  */
  34
  35 #include <machine/asm.h>
  36 __FBSDID("$FreeBSD$");
  37
  38 #include <machine/riscvreg.h>
  39 #include <sys/errno.h>
  40
  41 #include "assym.inc"
  42
  43 /*
  44  * Fault handler for the copy{in,out} functions below.
  45  */
  46 ENTRY(copyio_fault)
  47         SET_FAULT_HANDLER(x0, a1) /* Clear the handler */
  48         EXIT_USER_ACCESS(a1)
  49 copyio_fault_nopcb:
  50         li      a0, EFAULT
  51         ret
  52 END(copyio_fault)
  53
  54 /*
  55  * Copies from a kernel to user address
  56  *
  57  * int copyout(const void *kaddr, void *udaddr, size_t len)
  58  */
  59 ENTRY(copyout)
  60         beqz    a2, 2f          /* If len == 0 then skip loop */
  61         add     a3, a1, a2
  62         li      a4, VM_MAXUSER_ADDRESS
  63         bgt     a3, a4, copyio_fault_nopcb
  64
  65         la      a6, copyio_fault /* Get the handler address */
  66         SET_FAULT_HANDLER(a6, a7) /* Set the handler */
  67         ENTER_USER_ACCESS(a7)
  68
  69 1:      lb      a4, 0(a0)       /* Load from kaddr */
  70         addi    a0, a0, 1
  71         sb      a4, 0(a1)       /* Store in uaddr */
  72         addi    a1, a1, 1
  73         addi    a2, a2, -1      /* len-- */
  74         bnez    a2, 1b
  75
  76         EXIT_USER_ACCESS(a7)
  77         SET_FAULT_HANDLER(x0, a7) /* Clear the handler */
  78
  79 2:      li      a0, 0           /* return 0 */
  80         ret
  81 END(copyout)
  82
  83 /*
  84  * Copies from a user to kernel address
  85  *
  86  * int copyin(const void *uaddr, void *kdaddr, size_t len)
  87  */
  88 ENTRY(copyin)
  89         beqz    a2, 2f          /* If len == 0 then skip loop */
  90         add     a3, a0, a2
  91         li      a4, VM_MAXUSER_ADDRESS
  92         bgt     a3, a4, copyio_fault_nopcb
  93
  94         la      a6, copyio_fault /* Get the handler address */
  95         SET_FAULT_HANDLER(a6, a7) /* Set the handler */
  96         ENTER_USER_ACCESS(a7)
  97
  98 1:      lb      a4, 0(a0)       /* Load from uaddr */
  99         addi    a0, a0, 1
 100         sb      a4, 0(a1)       /* Store in kaddr */
 101         addi    a1, a1, 1
 102         addi    a2, a2, -1      /* len-- */
 103         bnez    a2, 1b
 104
 105         EXIT_USER_ACCESS(a7)
 106         SET_FAULT_HANDLER(x0, a7) /* Clear the handler */
 107
 108 2:      li      a0, 0           /* return 0 */
 109         ret
 110 END(copyin)
 111
 112 /*
 113  * Copies a string from a user to kernel address
 114  *
 115  * int copyinstr(const void *udaddr, void *kaddr, size_t len, size_t *done)
 116  */
 117 ENTRY(copyinstr)
 118         mv      a5, x0          /* count = 0 */
 119         beqz    a2, 3f          /* If len == 0 then skip loop */
 120
 121         la      a6, copyio_fault /* Get the handler address */
 122         SET_FAULT_HANDLER(a6, a7) /* Set the handler */
 123         ENTER_USER_ACCESS(a7)
 124
 125         li      a7, VM_MAXUSER_ADDRESS
 126 1:      bgt     a0, a7, copyio_fault
 127         lb      a4, 0(a0)       /* Load from uaddr */
 128         addi    a0, a0, 1
 129         sb      a4, 0(a1)       /* Store in kaddr */
 130         addi    a1, a1, 1
 131         beqz    a4, 2f
 132         addi    a2, a2, -1      /* len-- */
 133         addi    a5, a5, 1       /* count++ */
 134         bnez    a2, 1b
 135
 136 2:      EXIT_USER_ACCESS(a7)
 137         SET_FAULT_HANDLER(x0, a7) /* Clear the handler */
 138
 139 3:      beqz    a3, 4f          /* Check if done != NULL */
 140         addi    a5, a5, 1       /* count++ */
 141         sd      a5, 0(a3)       /* done = count */
 142
 143 4:      mv      a0, x0          /* return 0 */
 144         beqz    a4, 5f
 145         li      a0, ENAMETOOLONG
 146 5:
 147         ret
 148 END(copyinstr)