arch/riscv/include/asm/asm.h

   1 /* SPDX-License-Identifier: GPL-2.0-only */
   2 /*
   3  * Copyright (C) 2015 Regents of the University of California
   4  */
   5
   6 #ifndef _ASM_RISCV_ASM_H
   7 #define _ASM_RISCV_ASM_H
   8
   9 #ifdef __ASSEMBLY__
  10 #define __ASM_STR(x)    x
  11 #else
  12 #define __ASM_STR(x)    #x
  13 #endif
  14
  15 #if __riscv_xlen == 64
  16 #define __REG_SEL(a, b) __ASM_STR(a)
  17 #elif __riscv_xlen == 32
  18 #define __REG_SEL(a, b) __ASM_STR(b)
  19 #else
  20 #error "Unexpected __riscv_xlen"
  21 #endif
  22
  23 #define REG_L           __REG_SEL(ld, lw)
  24 #define REG_S           __REG_SEL(sd, sw)
  25 #define REG_SC          __REG_SEL(sc.d, sc.w)
  26 #define REG_AMOSWAP_AQ  __REG_SEL(amoswap.d.aq, amoswap.w.aq)
  27 #define REG_ASM         __REG_SEL(.dword, .word)
  28 #define SZREG           __REG_SEL(8, 4)
  29 #define LGREG           __REG_SEL(3, 2)
  30
  31 #if __SIZEOF_POINTER__ == 8
  32 #ifdef __ASSEMBLY__
  33 #define RISCV_PTR               .dword
  34 #define RISCV_SZPTR             8
  35 #define RISCV_LGPTR             3
  36 #else
  37 #define RISCV_PTR               ".dword"
  38 #define RISCV_SZPTR             "8"
  39 #define RISCV_LGPTR             "3"
  40 #endif
  41 #elif __SIZEOF_POINTER__ == 4
  42 #ifdef __ASSEMBLY__
  43 #define RISCV_PTR               .word
  44 #define RISCV_SZPTR             4
  45 #define RISCV_LGPTR             2
  46 #else
  47 #define RISCV_PTR               ".word"
  48 #define RISCV_SZPTR             "4"
  49 #define RISCV_LGPTR             "2"
  50 #endif
  51 #else
  52 #error "Unexpected __SIZEOF_POINTER__"
  53 #endif
  54
  55 #if (__SIZEOF_INT__ == 4)
  56 #define RISCV_INT               __ASM_STR(.word)
  57 #define RISCV_SZINT             __ASM_STR(4)
  58 #define RISCV_LGINT             __ASM_STR(2)
  59 #else
  60 #error "Unexpected __SIZEOF_INT__"
  61 #endif
  62
  63 #if (__SIZEOF_SHORT__ == 2)
  64 #define RISCV_SHORT             __ASM_STR(.half)
  65 #define RISCV_SZSHORT           __ASM_STR(2)
  66 #define RISCV_LGSHORT           __ASM_STR(1)
  67 #else
  68 #error "Unexpected __SIZEOF_SHORT__"
  69 #endif
  70
  71 #ifdef __ASSEMBLY__
  72 #include <asm/asm-offsets.h>
  73
  74 /* Common assembly source macros */
  75
  76 /*
  77  * NOP sequence
  78  */
  79 .macro  nops, num
  80         .rept   \num
  81         nop
  82         .endr
  83 .endm
  84
  85         /* save all GPs except x1 ~ x5 */
  86         .macro save_from_x6_to_x31
  87         REG_S x6,  PT_T1(sp)
  88         REG_S x7,  PT_T2(sp)
  89         REG_S x8,  PT_S0(sp)
  90         REG_S x9,  PT_S1(sp)
  91         REG_S x10, PT_A0(sp)
  92         REG_S x11, PT_A1(sp)
  93         REG_S x12, PT_A2(sp)
  94         REG_S x13, PT_A3(sp)
  95         REG_S x14, PT_A4(sp)
  96         REG_S x15, PT_A5(sp)
  97         REG_S x16, PT_A6(sp)
  98         REG_S x17, PT_A7(sp)
  99         REG_S x18, PT_S2(sp)
 100         REG_S x19, PT_S3(sp)
 101         REG_S x20, PT_S4(sp)
 102         REG_S x21, PT_S5(sp)
 103         REG_S x22, PT_S6(sp)
 104         REG_S x23, PT_S7(sp)
 105         REG_S x24, PT_S8(sp)
 106         REG_S x25, PT_S9(sp)
 107         REG_S x26, PT_S10(sp)
 108         REG_S x27, PT_S11(sp)
 109         REG_S x28, PT_T3(sp)
 110         REG_S x29, PT_T4(sp)
 111         REG_S x30, PT_T5(sp)
 112         REG_S x31, PT_T6(sp)
 113         .endm
 114
 115         /* restore all GPs except x1 ~ x5 */
 116         .macro restore_from_x6_to_x31
 117         REG_L x6,  PT_T1(sp)
 118         REG_L x7,  PT_T2(sp)
 119         REG_L x8,  PT_S0(sp)
 120         REG_L x9,  PT_S1(sp)
 121         REG_L x10, PT_A0(sp)
 122         REG_L x11, PT_A1(sp)
 123         REG_L x12, PT_A2(sp)
 124         REG_L x13, PT_A3(sp)
 125         REG_L x14, PT_A4(sp)
 126         REG_L x15, PT_A5(sp)
 127         REG_L x16, PT_A6(sp)
 128         REG_L x17, PT_A7(sp)
 129         REG_L x18, PT_S2(sp)
 130         REG_L x19, PT_S3(sp)
 131         REG_L x20, PT_S4(sp)
 132         REG_L x21, PT_S5(sp)
 133         REG_L x22, PT_S6(sp)
 134         REG_L x23, PT_S7(sp)
 135         REG_L x24, PT_S8(sp)
 136         REG_L x25, PT_S9(sp)
 137         REG_L x26, PT_S10(sp)
 138         REG_L x27, PT_S11(sp)
 139         REG_L x28, PT_T3(sp)
 140         REG_L x29, PT_T4(sp)
 141         REG_L x30, PT_T5(sp)
 142         REG_L x31, PT_T6(sp)
 143         .endm
 144
 145 #endif /* __ASSEMBLY__ */
 146
 147 #endif /* _ASM_RISCV_ASM_H */