cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

NGbzero.S (3465B)


      1/* SPDX-License-Identifier: GPL-2.0 */
      2/* NGbzero.S: Niagara optimized memset/clear_user.
      3 *
      4 * Copyright (C) 2006 David S. Miller (davem@davemloft.net)
      5 */
      6#include <asm/asi.h>
      7
      8#define EX_ST(x,y)		\
      998:	x,y;			\
     10	.section __ex_table,"a";\
     11	.align 4;		\
     12	.word 98b, __retl_o1_asi;\
     13	.text;			\
     14	.align 4;
     15
     16	.text
     17
     18	.globl		NGmemset
     19	.type		NGmemset, #function
     20NGmemset:		/* %o0=buf, %o1=pat, %o2=len */
     21	and		%o1, 0xff, %o3
     22	mov		%o2, %o1
     23	sllx		%o3, 8, %g1
     24	or		%g1, %o3, %o2
     25	sllx		%o2, 16, %g1
     26	or		%g1, %o2, %o2
     27	sllx		%o2, 32, %g1
     28	ba,pt		%xcc, 1f
     29	 or		%g1, %o2, %o2
     30
     31	.globl		NGbzero
     32	.type		NGbzero, #function
     33NGbzero:
     34	clr		%o2
     351:	brz,pn		%o1, NGbzero_return
     36	 mov		%o0, %o3
     37
     38	/* %o5: saved %asi, restored at NGbzero_done
     39	 * %g7: store-init %asi to use
     40	 * %o4:	non-store-init %asi to use
     41	 */
     42	rd		%asi, %o5
     43	mov		ASI_BLK_INIT_QUAD_LDD_P, %g7
     44	mov		ASI_P, %o4
     45	wr		%o4, 0x0, %asi
     46
     47NGbzero_from_clear_user:
     48	cmp		%o1, 15
     49	bl,pn		%icc, NGbzero_tiny
     50	 andcc		%o0, 0x7, %g1
     51	be,pt		%xcc, 2f
     52	 mov		8, %g2
     53	sub		%g2, %g1, %g1
     54	sub		%o1, %g1, %o1
     551:	EX_ST(stba %o2, [%o0 + 0x00] %asi)
     56	subcc		%g1, 1, %g1
     57	bne,pt		%xcc, 1b
     58	 add		%o0, 1, %o0
     592:	cmp		%o1, 128
     60	bl,pn		%icc, NGbzero_medium
     61	 andcc		%o0, (64 - 1), %g1
     62	be,pt		%xcc, NGbzero_pre_loop
     63	 mov		64, %g2
     64	sub		%g2, %g1, %g1
     65	sub		%o1, %g1, %o1
     661:	EX_ST(stxa %o2, [%o0 + 0x00] %asi)
     67	subcc		%g1, 8, %g1
     68	bne,pt		%xcc, 1b
     69	 add		%o0, 8, %o0
     70
     71NGbzero_pre_loop:
     72	wr		%g7, 0x0, %asi
     73	andn		%o1, (64 - 1), %g1
     74	sub		%o1, %g1, %o1
     75NGbzero_loop:
     76	EX_ST(stxa %o2, [%o0 + 0x00] %asi)
     77	EX_ST(stxa %o2, [%o0 + 0x08] %asi)
     78	EX_ST(stxa %o2, [%o0 + 0x10] %asi)
     79	EX_ST(stxa %o2, [%o0 + 0x18] %asi)
     80	EX_ST(stxa %o2, [%o0 + 0x20] %asi)
     81	EX_ST(stxa %o2, [%o0 + 0x28] %asi)
     82	EX_ST(stxa %o2, [%o0 + 0x30] %asi)
     83	EX_ST(stxa %o2, [%o0 + 0x38] %asi)
     84	subcc		%g1, 64, %g1
     85	bne,pt		%xcc, NGbzero_loop
     86	 add		%o0, 64, %o0
     87
     88	membar		#Sync
     89	wr		%o4, 0x0, %asi
     90	brz,pn		%o1, NGbzero_done
     91NGbzero_medium:
     92	 andncc		%o1, 0x7, %g1
     93	be,pn		%xcc, 2f
     94	 sub		%o1, %g1, %o1
     951:	EX_ST(stxa %o2, [%o0 + 0x00] %asi)
     96	subcc		%g1, 8, %g1
     97	bne,pt		%xcc, 1b
     98	 add		%o0, 8, %o0
     992:	brz,pt		%o1, NGbzero_done
    100	 nop
    101
    102NGbzero_tiny:
    1031:	EX_ST(stba %o2, [%o0 + 0x00] %asi)
    104	subcc		%o1, 1, %o1
    105	bne,pt		%icc, 1b
    106	 add		%o0, 1, %o0
    107
    108	/* fallthrough */
    109
    110NGbzero_done:
    111	wr		%o5, 0x0, %asi
    112
    113NGbzero_return:
    114	retl
    115	 mov		%o3, %o0
    116	.size		NGbzero, .-NGbzero
    117	.size		NGmemset, .-NGmemset
    118
    119	.globl		NGclear_user
    120	.type		NGclear_user, #function
    121NGclear_user:		/* %o0=buf, %o1=len */
    122	rd		%asi, %o5
    123	brz,pn		%o1, NGbzero_done
    124	 clr		%o3
    125	cmp		%o5, ASI_AIUS
    126	bne,pn		%icc, NGbzero
    127	 clr		%o2
    128	mov		ASI_BLK_INIT_QUAD_LDD_AIUS, %g7
    129	ba,pt		%xcc, NGbzero_from_clear_user
    130	 mov		ASI_AIUS, %o4
    131	.size		NGclear_user, .-NGclear_user
    132
    133#define BRANCH_ALWAYS	0x10680000
    134#define NOP		0x01000000
    135#define NG_DO_PATCH(OLD, NEW)	\
    136	sethi	%hi(NEW), %g1; \
    137	or	%g1, %lo(NEW), %g1; \
    138	sethi	%hi(OLD), %g2; \
    139	or	%g2, %lo(OLD), %g2; \
    140	sub	%g1, %g2, %g1; \
    141	sethi	%hi(BRANCH_ALWAYS), %g3; \
    142	sll	%g1, 11, %g1; \
    143	srl	%g1, 11 + 2, %g1; \
    144	or	%g3, %lo(BRANCH_ALWAYS), %g3; \
    145	or	%g3, %g1, %g3; \
    146	stw	%g3, [%g2]; \
    147	sethi	%hi(NOP), %g3; \
    148	or	%g3, %lo(NOP), %g3; \
    149	stw	%g3, [%g2 + 0x4]; \
    150	flush	%g2;
    151
    152	.globl	niagara_patch_bzero
    153	.type	niagara_patch_bzero,#function
    154niagara_patch_bzero:
    155	NG_DO_PATCH(memset, NGmemset)
    156	NG_DO_PATCH(__bzero, NGbzero)
    157	NG_DO_PATCH(__clear_user, NGclear_user)
    158	NG_DO_PATCH(tsb_init, NGtsb_init)
    159	retl
    160	 nop
    161	.size	niagara_patch_bzero,.-niagara_patch_bzero