xref: /kvm-unit-tests/lib/s390x/asm/arch_def.h (revision b43c912f811a6d55455515fc60fb570edb4b6429)
16c9f99dfSJanosch Frank /* SPDX-License-Identifier: GPL-2.0-only */
2cfb204f9SDavid Hildenbrand /*
3cfb204f9SDavid Hildenbrand  * Copyright (c) 2017 Red Hat Inc
4cfb204f9SDavid Hildenbrand  *
5cfb204f9SDavid Hildenbrand  * Authors:
6cfb204f9SDavid Hildenbrand  *  David Hildenbrand <david@redhat.com>
7cfb204f9SDavid Hildenbrand  */
8cfb204f9SDavid Hildenbrand #ifndef _ASM_S390X_ARCH_DEF_H_
9cfb204f9SDavid Hildenbrand #define _ASM_S390X_ARCH_DEF_H_
10cfb204f9SDavid Hildenbrand 
113ae7f80fSJanosch Frank struct stack_frame {
1236cfc0b7SJanosch Frank 	struct stack_frame *back_chain;
1336cfc0b7SJanosch Frank 	uint64_t reserved;
1436cfc0b7SJanosch Frank 	/* GRs 2 - 5 */
1536cfc0b7SJanosch Frank 	uint64_t argument_area[4];
1636cfc0b7SJanosch Frank 	/* GRs 6 - 15 */
1736cfc0b7SJanosch Frank 	uint64_t grs[10];
1836cfc0b7SJanosch Frank 	/* FPRs 0, 2, 4, 6 */
1936cfc0b7SJanosch Frank 	int64_t  fprs[4];
2036cfc0b7SJanosch Frank };
2136cfc0b7SJanosch Frank 
2236cfc0b7SJanosch Frank struct stack_frame_int {
2336cfc0b7SJanosch Frank 	struct stack_frame *back_chain;
2436cfc0b7SJanosch Frank 	uint64_t reserved;
2536cfc0b7SJanosch Frank 	/*
2636cfc0b7SJanosch Frank 	 * The GRs are offset compatible with struct stack_frame so we
2736cfc0b7SJanosch Frank 	 * can easily fetch GR14 for backtraces.
2836cfc0b7SJanosch Frank 	 */
2936cfc0b7SJanosch Frank 	/* GRs 2 - 15 */
3036cfc0b7SJanosch Frank 	uint64_t grs0[14];
3136cfc0b7SJanosch Frank 	/* GRs 0 and 1 */
3236cfc0b7SJanosch Frank 	uint64_t grs1[2];
3336cfc0b7SJanosch Frank 	uint32_t reserved1;
3436cfc0b7SJanosch Frank 	uint32_t fpc;
3536cfc0b7SJanosch Frank 	uint64_t fprs[16];
3636cfc0b7SJanosch Frank 	uint64_t crs[16];
373ae7f80fSJanosch Frank };
383ae7f80fSJanosch Frank 
39cfb204f9SDavid Hildenbrand struct psw {
40cfb204f9SDavid Hildenbrand 	uint64_t	mask;
41cfb204f9SDavid Hildenbrand 	uint64_t	addr;
42cfb204f9SDavid Hildenbrand };
43cfb204f9SDavid Hildenbrand 
4465dc6515SJanosch Frank #define PSW_MASK_EXT			0x0100000000000000UL
458cb729e4SPierre Morel #define PSW_MASK_IO			0x0200000000000000UL
46c08c320bSDavid Hildenbrand #define PSW_MASK_DAT			0x0400000000000000UL
47f73b4b9eSPierre Morel #define PSW_MASK_WAIT			0x0002000000000000UL
484ef6f57dSJanosch Frank #define PSW_MASK_PSTATE			0x0001000000000000UL
49c08c320bSDavid Hildenbrand 
50f7df2911SJanosch Frank #define CR0_EXTM_SCLP			0x0000000000000200UL
51f7df2911SJanosch Frank #define CR0_EXTM_EXTC			0x0000000000002000UL
52f7df2911SJanosch Frank #define CR0_EXTM_EMGC			0x0000000000004000UL
53f7df2911SJanosch Frank #define CR0_EXTM_MASK			0x0000000000006200UL
54df121a0cSJanosch Frank 
55cfb204f9SDavid Hildenbrand struct lowcore {
56cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x0000[0x0080 - 0x0000];	/* 0x0000 */
57cfb204f9SDavid Hildenbrand 	uint32_t	ext_int_param;			/* 0x0080 */
58cfb204f9SDavid Hildenbrand 	uint16_t	cpu_addr;			/* 0x0084 */
59cfb204f9SDavid Hildenbrand 	uint16_t	ext_int_code;			/* 0x0086 */
60cfb204f9SDavid Hildenbrand 	uint16_t	svc_int_id;			/* 0x0088 */
61cfb204f9SDavid Hildenbrand 	uint16_t	svc_int_code;			/* 0x008a */
62cfb204f9SDavid Hildenbrand 	uint16_t	pgm_int_id;			/* 0x008c */
63cfb204f9SDavid Hildenbrand 	uint16_t	pgm_int_code;			/* 0x008e */
64cfb204f9SDavid Hildenbrand 	uint32_t	dxc_vxc;			/* 0x0090 */
65cfb204f9SDavid Hildenbrand 	uint16_t	mon_class_nb;			/* 0x0094 */
66cfb204f9SDavid Hildenbrand 	uint8_t		per_code;			/* 0x0096 */
67cfb204f9SDavid Hildenbrand 	uint8_t		per_atmid;			/* 0x0097 */
68cfb204f9SDavid Hildenbrand 	uint64_t	per_addr;			/* 0x0098 */
69cfb204f9SDavid Hildenbrand 	uint8_t		exc_acc_id;			/* 0x00a0 */
70cfb204f9SDavid Hildenbrand 	uint8_t		per_acc_id;			/* 0x00a1 */
71cfb204f9SDavid Hildenbrand 	uint8_t		op_acc_id;			/* 0x00a2 */
72cfb204f9SDavid Hildenbrand 	uint8_t		arch_mode_id;			/* 0x00a3 */
73cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x00a4[0x00a8 - 0x00a4];	/* 0x00a4 */
74cfb204f9SDavid Hildenbrand 	uint64_t	trans_exc_id;			/* 0x00a8 */
75cfb204f9SDavid Hildenbrand 	uint64_t	mon_code;			/* 0x00b0 */
76cfb204f9SDavid Hildenbrand 	uint32_t	subsys_id_word;			/* 0x00b8 */
77cfb204f9SDavid Hildenbrand 	uint32_t	io_int_param;			/* 0x00bc */
78cfb204f9SDavid Hildenbrand 	uint32_t	io_int_word;			/* 0x00c0 */
79cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x00c4[0x00c8 - 0x00c4];	/* 0x00c4 */
80cfb204f9SDavid Hildenbrand 	uint32_t	stfl;				/* 0x00c8 */
81cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x00cc[0x00e8 - 0x00cc];	/* 0x00cc */
82cfb204f9SDavid Hildenbrand 	uint64_t	mcck_int_code;			/* 0x00e8 */
83cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x00f0[0x00f4 - 0x00f0];	/* 0x00f0 */
84cfb204f9SDavid Hildenbrand 	uint32_t	ext_damage_code;		/* 0x00f4 */
85cfb204f9SDavid Hildenbrand 	uint64_t	failing_storage_addr;		/* 0x00f8 */
86cfb204f9SDavid Hildenbrand 	uint64_t	emon_ca_origin;			/* 0x0100 */
87cfb204f9SDavid Hildenbrand 	uint32_t	emon_ca_size;			/* 0x0108 */
88cfb204f9SDavid Hildenbrand 	uint32_t	emon_exc_count;			/* 0x010c */
89cfb204f9SDavid Hildenbrand 	uint64_t	breaking_event_addr;		/* 0x0110 */
90cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x0118[0x0120 - 0x0118];	/* 0x0118 */
91cfb204f9SDavid Hildenbrand 	struct psw	restart_old_psw;		/* 0x0120 */
92cfb204f9SDavid Hildenbrand 	struct psw	ext_old_psw;			/* 0x0130 */
93cfb204f9SDavid Hildenbrand 	struct psw	svc_old_psw;			/* 0x0140 */
94cfb204f9SDavid Hildenbrand 	struct psw	pgm_old_psw;			/* 0x0150 */
95cfb204f9SDavid Hildenbrand 	struct psw	mcck_old_psw;			/* 0x0160 */
96cfb204f9SDavid Hildenbrand 	struct psw	io_old_psw;			/* 0x0170 */
97cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x0180[0x01a0 - 0x0180];	/* 0x0180 */
98cfb204f9SDavid Hildenbrand 	struct psw	restart_new_psw;		/* 0x01a0 */
99cfb204f9SDavid Hildenbrand 	struct psw	ext_new_psw;			/* 0x01b0 */
100cfb204f9SDavid Hildenbrand 	struct psw	svc_new_psw;			/* 0x01c0 */
101cfb204f9SDavid Hildenbrand 	struct psw	pgm_new_psw;			/* 0x01d0 */
102cfb204f9SDavid Hildenbrand 	struct psw	mcck_new_psw;			/* 0x01e0 */
103cfb204f9SDavid Hildenbrand 	struct psw	io_new_psw;			/* 0x01f0 */
1044da93626SDavid Hildenbrand 	/* sw definition: save area for registers in interrupt handlers */
1054da93626SDavid Hildenbrand 	uint64_t	sw_int_grs[16];			/* 0x0200 */
1063a92a013SJanosch Frank 	uint8_t		pad_0x0280[0x0308 - 0x0280];	/* 0x0280 */
107736b9295SJanosch Frank 	uint64_t	sw_int_crs[16];			/* 0x0308 */
108da6ce270SJanosch Frank 	struct psw	sw_int_psw;			/* 0x0388 */
109da6ce270SJanosch Frank 	uint8_t		pad_0x0310[0x11b0 - 0x0398];	/* 0x0398 */
110cfb204f9SDavid Hildenbrand 	uint64_t	mcck_ext_sa_addr;		/* 0x11b0 */
111cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x11b8[0x1200 - 0x11b8];	/* 0x11b8 */
112cfb204f9SDavid Hildenbrand 	uint64_t	fprs_sa[16];			/* 0x1200 */
113cfb204f9SDavid Hildenbrand 	uint64_t	grs_sa[16];			/* 0x1280 */
114cfb204f9SDavid Hildenbrand 	struct psw	psw_sa;				/* 0x1300 */
115cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x1310[0x1318 - 0x1310];	/* 0x1310 */
116cfb204f9SDavid Hildenbrand 	uint32_t	prefix_sa;			/* 0x1318 */
117cfb204f9SDavid Hildenbrand 	uint32_t	fpc_sa;				/* 0x131c */
118cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x1320[0x1324 - 0x1320];	/* 0x1320 */
119cfb204f9SDavid Hildenbrand 	uint32_t	tod_pr_sa;			/* 0x1324 */
120cfb204f9SDavid Hildenbrand 	uint64_t	cputm_sa;			/* 0x1328 */
121cfb204f9SDavid Hildenbrand 	uint64_t	cc_sa;				/* 0x1330 */
122cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x1338[0x1340 - 0x1338];	/* 0x1338 */
123cfb204f9SDavid Hildenbrand 	uint32_t	ars_sa[16];			/* 0x1340 */
124cfb204f9SDavid Hildenbrand 	uint64_t	crs_sa[16];			/* 0x1380 */
125cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x1400[0x1800 - 0x1400];	/* 0x1400 */
126cfb204f9SDavid Hildenbrand 	uint8_t		pgm_int_tdb[0x1900 - 0x1800];	/* 0x1800 */
127cfb204f9SDavid Hildenbrand } __attribute__ ((__packed__));
128da6ce270SJanosch Frank _Static_assert(sizeof(struct lowcore) == 0x1900, "Lowcore size");
129cfb204f9SDavid Hildenbrand 
1304da93626SDavid Hildenbrand #define PGM_INT_CODE_OPERATION			0x01
1314da93626SDavid Hildenbrand #define PGM_INT_CODE_PRIVILEGED_OPERATION	0x02
1324da93626SDavid Hildenbrand #define PGM_INT_CODE_EXECUTE			0x03
1334da93626SDavid Hildenbrand #define PGM_INT_CODE_PROTECTION			0x04
1344da93626SDavid Hildenbrand #define PGM_INT_CODE_ADDRESSING			0x05
1354da93626SDavid Hildenbrand #define PGM_INT_CODE_SPECIFICATION		0x06
1364da93626SDavid Hildenbrand #define PGM_INT_CODE_DATA			0x07
1374da93626SDavid Hildenbrand #define PGM_INT_CODE_FIXED_POINT_OVERFLOW	0x08
1384da93626SDavid Hildenbrand #define PGM_INT_CODE_FIXED_POINT_DIVIDE		0x09
1394da93626SDavid Hildenbrand #define PGM_INT_CODE_DECIMAL_OVERFLOW		0x0a
1404da93626SDavid Hildenbrand #define PGM_INT_CODE_DECIMAL_DIVIDE		0x0b
1414da93626SDavid Hildenbrand #define PGM_INT_CODE_HFP_EXPONENT_OVERFLOW	0x0c
1424da93626SDavid Hildenbrand #define PGM_INT_CODE_HFP_EXPONENT_UNDERFLOW	0x0d
1434da93626SDavid Hildenbrand #define PGM_INT_CODE_HFP_SIGNIFICANCE		0x0e
1444da93626SDavid Hildenbrand #define PGM_INT_CODE_HFP_DIVIDE			0x0f
1454da93626SDavid Hildenbrand #define PGM_INT_CODE_SEGMENT_TRANSLATION	0x10
1464da93626SDavid Hildenbrand #define PGM_INT_CODE_PAGE_TRANSLATION		0x11
1474da93626SDavid Hildenbrand #define PGM_INT_CODE_TRANSLATION_SPEC		0x12
1484da93626SDavid Hildenbrand #define PGM_INT_CODE_SPECIAL_OPERATION		0x13
1494da93626SDavid Hildenbrand #define PGM_INT_CODE_OPERAND			0x15
1504da93626SDavid Hildenbrand #define PGM_INT_CODE_TRACE_TABLE		0x16
1514da93626SDavid Hildenbrand #define PGM_INT_CODE_VECTOR_PROCESSING		0x1b
1524da93626SDavid Hildenbrand #define PGM_INT_CODE_SPACE_SWITCH_EVENT		0x1c
1534da93626SDavid Hildenbrand #define PGM_INT_CODE_HFP_SQUARE_ROOT		0x1d
1544da93626SDavid Hildenbrand #define PGM_INT_CODE_PC_TRANSLATION_SPEC	0x1f
1554da93626SDavid Hildenbrand #define PGM_INT_CODE_AFX_TRANSLATION		0x20
1564da93626SDavid Hildenbrand #define PGM_INT_CODE_ASX_TRANSLATION		0x21
1574da93626SDavid Hildenbrand #define PGM_INT_CODE_LX_TRANSLATION		0x22
1584da93626SDavid Hildenbrand #define PGM_INT_CODE_EX_TRANSLATION		0x23
1594da93626SDavid Hildenbrand #define PGM_INT_CODE_PRIMARY_AUTHORITY		0x24
1604da93626SDavid Hildenbrand #define PGM_INT_CODE_SECONDARY_AUTHORITY	0x25
1614da93626SDavid Hildenbrand #define PGM_INT_CODE_LFX_TRANSLATION		0x26
1624da93626SDavid Hildenbrand #define PGM_INT_CODE_LSX_TRANSLATION		0x27
1634da93626SDavid Hildenbrand #define PGM_INT_CODE_ALET_SPECIFICATION		0x28
1644da93626SDavid Hildenbrand #define PGM_INT_CODE_ALEN_TRANSLATION		0x29
1654da93626SDavid Hildenbrand #define PGM_INT_CODE_ALE_SEQUENCE		0x2a
1664da93626SDavid Hildenbrand #define PGM_INT_CODE_ASTE_VALIDITY		0x2b
1674da93626SDavid Hildenbrand #define PGM_INT_CODE_ASTE_SEQUENCE		0x2c
1684da93626SDavid Hildenbrand #define PGM_INT_CODE_EXTENDED_AUTHORITY		0x2d
1694da93626SDavid Hildenbrand #define PGM_INT_CODE_LSTE_SEQUENCE		0x2e
1704da93626SDavid Hildenbrand #define PGM_INT_CODE_ASTE_INSTANCE		0x2f
1714da93626SDavid Hildenbrand #define PGM_INT_CODE_STACK_FULL			0x30
1724da93626SDavid Hildenbrand #define PGM_INT_CODE_STACK_EMPTY		0x31
1734da93626SDavid Hildenbrand #define PGM_INT_CODE_STACK_SPECIFICATION	0x32
1744da93626SDavid Hildenbrand #define PGM_INT_CODE_STACK_TYPE			0x33
1754da93626SDavid Hildenbrand #define PGM_INT_CODE_STACK_OPERATION		0x34
1764da93626SDavid Hildenbrand #define PGM_INT_CODE_ASCE_TYPE			0x38
1774da93626SDavid Hildenbrand #define PGM_INT_CODE_REGION_FIRST_TRANS		0x39
1784da93626SDavid Hildenbrand #define PGM_INT_CODE_REGION_SECOND_TRANS	0x3a
1794da93626SDavid Hildenbrand #define PGM_INT_CODE_REGION_THIRD_TRANS		0x3b
1804da93626SDavid Hildenbrand #define PGM_INT_CODE_MONITOR_EVENT		0x40
1814da93626SDavid Hildenbrand #define PGM_INT_CODE_PER			0x80
1824da93626SDavid Hildenbrand #define PGM_INT_CODE_CRYPTO_OPERATION		0x119
1834da93626SDavid Hildenbrand #define PGM_INT_CODE_TX_ABORTED_EVENT		0x200
1844da93626SDavid Hildenbrand 
185484a3a57SDavid Hildenbrand struct cpuid {
186484a3a57SDavid Hildenbrand 	uint64_t version : 8;
187484a3a57SDavid Hildenbrand 	uint64_t id : 24;
188484a3a57SDavid Hildenbrand 	uint64_t type : 16;
189484a3a57SDavid Hildenbrand 	uint64_t format : 1;
190484a3a57SDavid Hildenbrand 	uint64_t reserved : 15;
191484a3a57SDavid Hildenbrand };
192484a3a57SDavid Hildenbrand 
193*b43c912fSClaudio Imbrenda #define SVC_LEAVE_PSTATE 1
194*b43c912fSClaudio Imbrenda 
195f77c0515SJanosch Frank static inline unsigned short stap(void)
196f77c0515SJanosch Frank {
197f77c0515SJanosch Frank 	unsigned short cpu_address;
198f77c0515SJanosch Frank 
199f77c0515SJanosch Frank 	asm volatile("stap %0" : "=Q" (cpu_address));
200f77c0515SJanosch Frank 	return cpu_address;
201f77c0515SJanosch Frank }
202f77c0515SJanosch Frank 
2033db880b6SDavid Hildenbrand static inline int tprot(unsigned long addr)
2043db880b6SDavid Hildenbrand {
2053db880b6SDavid Hildenbrand 	int cc;
2063db880b6SDavid Hildenbrand 
2073db880b6SDavid Hildenbrand 	asm volatile(
2083db880b6SDavid Hildenbrand 		"	tprot	0(%1),0\n"
2093db880b6SDavid Hildenbrand 		"	ipm	%0\n"
2103db880b6SDavid Hildenbrand 		"	srl	%0,28\n"
2113db880b6SDavid Hildenbrand 		: "=d" (cc) : "a" (addr) : "cc");
2123db880b6SDavid Hildenbrand 	return cc;
2133db880b6SDavid Hildenbrand }
2143db880b6SDavid Hildenbrand 
215c08c320bSDavid Hildenbrand static inline void lctlg(int cr, uint64_t value)
216c08c320bSDavid Hildenbrand {
217c08c320bSDavid Hildenbrand 	asm volatile(
218c08c320bSDavid Hildenbrand 		"	lctlg	%1,%1,%0\n"
219c08c320bSDavid Hildenbrand 		: : "Q" (value), "i" (cr));
220c08c320bSDavid Hildenbrand }
221c08c320bSDavid Hildenbrand 
222c08c320bSDavid Hildenbrand static inline uint64_t stctg(int cr)
223c08c320bSDavid Hildenbrand {
224c08c320bSDavid Hildenbrand 	uint64_t value;
225c08c320bSDavid Hildenbrand 
226c08c320bSDavid Hildenbrand 	asm volatile(
227c08c320bSDavid Hildenbrand 		"	stctg	%1,%1,%0\n"
228c08c320bSDavid Hildenbrand 		: "=Q" (value) : "i" (cr) : "memory");
229c08c320bSDavid Hildenbrand 	return value;
230c08c320bSDavid Hildenbrand }
231c08c320bSDavid Hildenbrand 
23243868475SJanosch Frank static inline void ctl_set_bit(int cr, unsigned int bit)
23343868475SJanosch Frank {
23443868475SJanosch Frank         uint64_t reg;
23543868475SJanosch Frank 
23643868475SJanosch Frank 	reg = stctg(cr);
23743868475SJanosch Frank 	reg |= 1UL << bit;
23843868475SJanosch Frank 	lctlg(cr, reg);
23943868475SJanosch Frank }
24043868475SJanosch Frank 
24143868475SJanosch Frank static inline void ctl_clear_bit(int cr, unsigned int bit)
24243868475SJanosch Frank {
24343868475SJanosch Frank         uint64_t reg;
24443868475SJanosch Frank 
24543868475SJanosch Frank 	reg = stctg(cr);
24643868475SJanosch Frank 	reg &= ~(1UL << bit);
24743868475SJanosch Frank 	lctlg(cr, reg);
24843868475SJanosch Frank }
24943868475SJanosch Frank 
250c08c320bSDavid Hildenbrand static inline uint64_t extract_psw_mask(void)
251c08c320bSDavid Hildenbrand {
252c08c320bSDavid Hildenbrand 	uint32_t mask_upper = 0, mask_lower = 0;
253c08c320bSDavid Hildenbrand 
254c08c320bSDavid Hildenbrand 	asm volatile(
255c08c320bSDavid Hildenbrand 		"	epsw	%0,%1\n"
256c08c320bSDavid Hildenbrand 		: "+r" (mask_upper), "+r" (mask_lower) : : );
257c08c320bSDavid Hildenbrand 
258c08c320bSDavid Hildenbrand 	return (uint64_t) mask_upper << 32 | mask_lower;
259c08c320bSDavid Hildenbrand }
260c08c320bSDavid Hildenbrand 
261c08c320bSDavid Hildenbrand static inline void load_psw_mask(uint64_t mask)
262c08c320bSDavid Hildenbrand {
263c08c320bSDavid Hildenbrand 	struct psw psw = {
264c08c320bSDavid Hildenbrand 		.mask = mask,
265c08c320bSDavid Hildenbrand 		.addr = 0,
266c08c320bSDavid Hildenbrand 	};
267c08c320bSDavid Hildenbrand 	uint64_t tmp = 0;
268c08c320bSDavid Hildenbrand 
269c08c320bSDavid Hildenbrand 	asm volatile(
270c08c320bSDavid Hildenbrand 		"	larl	%0,0f\n"
271c08c320bSDavid Hildenbrand 		"	stg	%0,8(%1)\n"
272c08c320bSDavid Hildenbrand 		"	lpswe	0(%1)\n"
273c08c320bSDavid Hildenbrand 		"0:\n"
274c08c320bSDavid Hildenbrand 		: "+r" (tmp) :  "a" (&psw) : "memory", "cc" );
275c08c320bSDavid Hildenbrand }
276c08c320bSDavid Hildenbrand 
277f73b4b9eSPierre Morel static inline void wait_for_interrupt(uint64_t irq_mask)
278f73b4b9eSPierre Morel {
279f73b4b9eSPierre Morel 	uint64_t psw_mask = extract_psw_mask();
280f73b4b9eSPierre Morel 
281f73b4b9eSPierre Morel 	load_psw_mask(psw_mask | irq_mask | PSW_MASK_WAIT);
282f73b4b9eSPierre Morel 	/*
283f73b4b9eSPierre Morel 	 * After being woken and having processed the interrupt, let's restore
284f73b4b9eSPierre Morel 	 * the PSW mask.
285f73b4b9eSPierre Morel 	 */
286f73b4b9eSPierre Morel 	load_psw_mask(psw_mask);
287f73b4b9eSPierre Morel }
288f73b4b9eSPierre Morel 
2894ef6f57dSJanosch Frank static inline void enter_pstate(void)
2904ef6f57dSJanosch Frank {
2914ef6f57dSJanosch Frank 	uint64_t mask;
2924ef6f57dSJanosch Frank 
2934ef6f57dSJanosch Frank 	mask = extract_psw_mask();
2944ef6f57dSJanosch Frank 	mask |= PSW_MASK_PSTATE;
2954ef6f57dSJanosch Frank 	load_psw_mask(mask);
2964ef6f57dSJanosch Frank }
2974ef6f57dSJanosch Frank 
298*b43c912fSClaudio Imbrenda static inline void leave_pstate(void)
299*b43c912fSClaudio Imbrenda {
300*b43c912fSClaudio Imbrenda 	asm volatile("	svc %0\n" : : "i" (SVC_LEAVE_PSTATE));
301*b43c912fSClaudio Imbrenda }
302*b43c912fSClaudio Imbrenda 
303c132c9e2SJanosch Frank static inline int stsi(void *addr, int fc, int sel1, int sel2)
304c132c9e2SJanosch Frank {
305c132c9e2SJanosch Frank 	register int r0 asm("0") = (fc << 28) | sel1;
306c132c9e2SJanosch Frank 	register int r1 asm("1") = sel2;
307c132c9e2SJanosch Frank 	int cc;
308c132c9e2SJanosch Frank 
309c132c9e2SJanosch Frank 	asm volatile(
310c132c9e2SJanosch Frank 		"stsi	0(%3)\n"
311c132c9e2SJanosch Frank 		"ipm	%[cc]\n"
312c132c9e2SJanosch Frank 		"srl	%[cc],28\n"
313c132c9e2SJanosch Frank 		: "+d" (r0), [cc] "=d" (cc)
314c132c9e2SJanosch Frank 		: "d" (r1), "a" (addr)
315c132c9e2SJanosch Frank 		: "cc", "memory");
316c132c9e2SJanosch Frank 	return cc;
317c132c9e2SJanosch Frank }
318c132c9e2SJanosch Frank 
319f9395bfeSClaudio Imbrenda static inline int servc(uint32_t command, unsigned long sccb)
320f9395bfeSClaudio Imbrenda {
321f9395bfeSClaudio Imbrenda 	int cc;
322f9395bfeSClaudio Imbrenda 
323f9395bfeSClaudio Imbrenda 	asm volatile(
324f9395bfeSClaudio Imbrenda 		"       .insn   rre,0xb2200000,%1,%2\n"  /* servc %1,%2 */
325f9395bfeSClaudio Imbrenda 		"       ipm     %0\n"
326f9395bfeSClaudio Imbrenda 		"       srl     %0,28"
327f9395bfeSClaudio Imbrenda 		: "=&d" (cc) : "d" (command), "a" (sccb)
328f9395bfeSClaudio Imbrenda 		: "cc", "memory");
329f9395bfeSClaudio Imbrenda 	return cc;
330f9395bfeSClaudio Imbrenda }
331f9395bfeSClaudio Imbrenda 
3326b3278c9SClaudio Imbrenda static inline void set_prefix(uint32_t new_prefix)
3336b3278c9SClaudio Imbrenda {
3346b3278c9SClaudio Imbrenda 	asm volatile("	spx %0" : : "Q" (new_prefix) : "memory");
3356b3278c9SClaudio Imbrenda }
3366b3278c9SClaudio Imbrenda 
3376b3278c9SClaudio Imbrenda static inline uint32_t get_prefix(void)
3386b3278c9SClaudio Imbrenda {
3396b3278c9SClaudio Imbrenda 	uint32_t current_prefix;
3406b3278c9SClaudio Imbrenda 
3416b3278c9SClaudio Imbrenda 	asm volatile("	stpx %0" : "=Q" (current_prefix));
3426b3278c9SClaudio Imbrenda 	return current_prefix;
3436b3278c9SClaudio Imbrenda }
3446b3278c9SClaudio Imbrenda 
345cfb204f9SDavid Hildenbrand #endif
346