xref: /kvm-unit-tests/lib/s390x/asm/arch_def.h (revision 36cfc0b79d04932df55736a16a37179576df794a)
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 {
12*36cfc0b7SJanosch Frank 	struct stack_frame *back_chain;
13*36cfc0b7SJanosch Frank 	uint64_t reserved;
14*36cfc0b7SJanosch Frank 	/* GRs 2 - 5 */
15*36cfc0b7SJanosch Frank 	uint64_t argument_area[4];
16*36cfc0b7SJanosch Frank 	/* GRs 6 - 15 */
17*36cfc0b7SJanosch Frank 	uint64_t grs[10];
18*36cfc0b7SJanosch Frank 	/* FPRs 0, 2, 4, 6 */
19*36cfc0b7SJanosch Frank 	int64_t  fprs[4];
20*36cfc0b7SJanosch Frank };
21*36cfc0b7SJanosch Frank 
22*36cfc0b7SJanosch Frank struct stack_frame_int {
23*36cfc0b7SJanosch Frank 	struct stack_frame *back_chain;
24*36cfc0b7SJanosch Frank 	uint64_t reserved;
25*36cfc0b7SJanosch Frank 	/*
26*36cfc0b7SJanosch Frank 	 * The GRs are offset compatible with struct stack_frame so we
27*36cfc0b7SJanosch Frank 	 * can easily fetch GR14 for backtraces.
28*36cfc0b7SJanosch Frank 	 */
29*36cfc0b7SJanosch Frank 	/* GRs 2 - 15 */
30*36cfc0b7SJanosch Frank 	uint64_t grs0[14];
31*36cfc0b7SJanosch Frank 	/* GRs 0 and 1 */
32*36cfc0b7SJanosch Frank 	uint64_t grs1[2];
33*36cfc0b7SJanosch Frank 	uint32_t reserved1;
34*36cfc0b7SJanosch Frank 	uint32_t fpc;
35*36cfc0b7SJanosch Frank 	uint64_t fprs[16];
36*36cfc0b7SJanosch 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 */
1064da93626SDavid Hildenbrand 	uint64_t	sw_int_fprs[16];		/* 0x0280 */
1074da93626SDavid Hildenbrand 	uint32_t	sw_int_fpc;			/* 0x0300 */
1088caf933eSDavid Hildenbrand 	uint8_t		pad_0x0304[0x0308 - 0x0304];	/* 0x0304 */
109736b9295SJanosch Frank 	uint64_t	sw_int_crs[16];			/* 0x0308 */
110da6ce270SJanosch Frank 	struct psw	sw_int_psw;			/* 0x0388 */
111da6ce270SJanosch Frank 	uint8_t		pad_0x0310[0x11b0 - 0x0398];	/* 0x0398 */
112cfb204f9SDavid Hildenbrand 	uint64_t	mcck_ext_sa_addr;		/* 0x11b0 */
113cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x11b8[0x1200 - 0x11b8];	/* 0x11b8 */
114cfb204f9SDavid Hildenbrand 	uint64_t	fprs_sa[16];			/* 0x1200 */
115cfb204f9SDavid Hildenbrand 	uint64_t	grs_sa[16];			/* 0x1280 */
116cfb204f9SDavid Hildenbrand 	struct psw	psw_sa;				/* 0x1300 */
117cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x1310[0x1318 - 0x1310];	/* 0x1310 */
118cfb204f9SDavid Hildenbrand 	uint32_t	prefix_sa;			/* 0x1318 */
119cfb204f9SDavid Hildenbrand 	uint32_t	fpc_sa;				/* 0x131c */
120cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x1320[0x1324 - 0x1320];	/* 0x1320 */
121cfb204f9SDavid Hildenbrand 	uint32_t	tod_pr_sa;			/* 0x1324 */
122cfb204f9SDavid Hildenbrand 	uint64_t	cputm_sa;			/* 0x1328 */
123cfb204f9SDavid Hildenbrand 	uint64_t	cc_sa;				/* 0x1330 */
124cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x1338[0x1340 - 0x1338];	/* 0x1338 */
125cfb204f9SDavid Hildenbrand 	uint32_t	ars_sa[16];			/* 0x1340 */
126cfb204f9SDavid Hildenbrand 	uint64_t	crs_sa[16];			/* 0x1380 */
127cfb204f9SDavid Hildenbrand 	uint8_t		pad_0x1400[0x1800 - 0x1400];	/* 0x1400 */
128cfb204f9SDavid Hildenbrand 	uint8_t		pgm_int_tdb[0x1900 - 0x1800];	/* 0x1800 */
129cfb204f9SDavid Hildenbrand } __attribute__ ((__packed__));
130da6ce270SJanosch Frank _Static_assert(sizeof(struct lowcore) == 0x1900, "Lowcore size");
131cfb204f9SDavid Hildenbrand 
1324da93626SDavid Hildenbrand #define PGM_INT_CODE_OPERATION			0x01
1334da93626SDavid Hildenbrand #define PGM_INT_CODE_PRIVILEGED_OPERATION	0x02
1344da93626SDavid Hildenbrand #define PGM_INT_CODE_EXECUTE			0x03
1354da93626SDavid Hildenbrand #define PGM_INT_CODE_PROTECTION			0x04
1364da93626SDavid Hildenbrand #define PGM_INT_CODE_ADDRESSING			0x05
1374da93626SDavid Hildenbrand #define PGM_INT_CODE_SPECIFICATION		0x06
1384da93626SDavid Hildenbrand #define PGM_INT_CODE_DATA			0x07
1394da93626SDavid Hildenbrand #define PGM_INT_CODE_FIXED_POINT_OVERFLOW	0x08
1404da93626SDavid Hildenbrand #define PGM_INT_CODE_FIXED_POINT_DIVIDE		0x09
1414da93626SDavid Hildenbrand #define PGM_INT_CODE_DECIMAL_OVERFLOW		0x0a
1424da93626SDavid Hildenbrand #define PGM_INT_CODE_DECIMAL_DIVIDE		0x0b
1434da93626SDavid Hildenbrand #define PGM_INT_CODE_HFP_EXPONENT_OVERFLOW	0x0c
1444da93626SDavid Hildenbrand #define PGM_INT_CODE_HFP_EXPONENT_UNDERFLOW	0x0d
1454da93626SDavid Hildenbrand #define PGM_INT_CODE_HFP_SIGNIFICANCE		0x0e
1464da93626SDavid Hildenbrand #define PGM_INT_CODE_HFP_DIVIDE			0x0f
1474da93626SDavid Hildenbrand #define PGM_INT_CODE_SEGMENT_TRANSLATION	0x10
1484da93626SDavid Hildenbrand #define PGM_INT_CODE_PAGE_TRANSLATION		0x11
1494da93626SDavid Hildenbrand #define PGM_INT_CODE_TRANSLATION_SPEC		0x12
1504da93626SDavid Hildenbrand #define PGM_INT_CODE_SPECIAL_OPERATION		0x13
1514da93626SDavid Hildenbrand #define PGM_INT_CODE_OPERAND			0x15
1524da93626SDavid Hildenbrand #define PGM_INT_CODE_TRACE_TABLE		0x16
1534da93626SDavid Hildenbrand #define PGM_INT_CODE_VECTOR_PROCESSING		0x1b
1544da93626SDavid Hildenbrand #define PGM_INT_CODE_SPACE_SWITCH_EVENT		0x1c
1554da93626SDavid Hildenbrand #define PGM_INT_CODE_HFP_SQUARE_ROOT		0x1d
1564da93626SDavid Hildenbrand #define PGM_INT_CODE_PC_TRANSLATION_SPEC	0x1f
1574da93626SDavid Hildenbrand #define PGM_INT_CODE_AFX_TRANSLATION		0x20
1584da93626SDavid Hildenbrand #define PGM_INT_CODE_ASX_TRANSLATION		0x21
1594da93626SDavid Hildenbrand #define PGM_INT_CODE_LX_TRANSLATION		0x22
1604da93626SDavid Hildenbrand #define PGM_INT_CODE_EX_TRANSLATION		0x23
1614da93626SDavid Hildenbrand #define PGM_INT_CODE_PRIMARY_AUTHORITY		0x24
1624da93626SDavid Hildenbrand #define PGM_INT_CODE_SECONDARY_AUTHORITY	0x25
1634da93626SDavid Hildenbrand #define PGM_INT_CODE_LFX_TRANSLATION		0x26
1644da93626SDavid Hildenbrand #define PGM_INT_CODE_LSX_TRANSLATION		0x27
1654da93626SDavid Hildenbrand #define PGM_INT_CODE_ALET_SPECIFICATION		0x28
1664da93626SDavid Hildenbrand #define PGM_INT_CODE_ALEN_TRANSLATION		0x29
1674da93626SDavid Hildenbrand #define PGM_INT_CODE_ALE_SEQUENCE		0x2a
1684da93626SDavid Hildenbrand #define PGM_INT_CODE_ASTE_VALIDITY		0x2b
1694da93626SDavid Hildenbrand #define PGM_INT_CODE_ASTE_SEQUENCE		0x2c
1704da93626SDavid Hildenbrand #define PGM_INT_CODE_EXTENDED_AUTHORITY		0x2d
1714da93626SDavid Hildenbrand #define PGM_INT_CODE_LSTE_SEQUENCE		0x2e
1724da93626SDavid Hildenbrand #define PGM_INT_CODE_ASTE_INSTANCE		0x2f
1734da93626SDavid Hildenbrand #define PGM_INT_CODE_STACK_FULL			0x30
1744da93626SDavid Hildenbrand #define PGM_INT_CODE_STACK_EMPTY		0x31
1754da93626SDavid Hildenbrand #define PGM_INT_CODE_STACK_SPECIFICATION	0x32
1764da93626SDavid Hildenbrand #define PGM_INT_CODE_STACK_TYPE			0x33
1774da93626SDavid Hildenbrand #define PGM_INT_CODE_STACK_OPERATION		0x34
1784da93626SDavid Hildenbrand #define PGM_INT_CODE_ASCE_TYPE			0x38
1794da93626SDavid Hildenbrand #define PGM_INT_CODE_REGION_FIRST_TRANS		0x39
1804da93626SDavid Hildenbrand #define PGM_INT_CODE_REGION_SECOND_TRANS	0x3a
1814da93626SDavid Hildenbrand #define PGM_INT_CODE_REGION_THIRD_TRANS		0x3b
1824da93626SDavid Hildenbrand #define PGM_INT_CODE_MONITOR_EVENT		0x40
1834da93626SDavid Hildenbrand #define PGM_INT_CODE_PER			0x80
1844da93626SDavid Hildenbrand #define PGM_INT_CODE_CRYPTO_OPERATION		0x119
1854da93626SDavid Hildenbrand #define PGM_INT_CODE_TX_ABORTED_EVENT		0x200
1864da93626SDavid Hildenbrand 
187484a3a57SDavid Hildenbrand struct cpuid {
188484a3a57SDavid Hildenbrand 	uint64_t version : 8;
189484a3a57SDavid Hildenbrand 	uint64_t id : 24;
190484a3a57SDavid Hildenbrand 	uint64_t type : 16;
191484a3a57SDavid Hildenbrand 	uint64_t format : 1;
192484a3a57SDavid Hildenbrand 	uint64_t reserved : 15;
193484a3a57SDavid Hildenbrand };
194484a3a57SDavid Hildenbrand 
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 
298c132c9e2SJanosch Frank static inline int stsi(void *addr, int fc, int sel1, int sel2)
299c132c9e2SJanosch Frank {
300c132c9e2SJanosch Frank 	register int r0 asm("0") = (fc << 28) | sel1;
301c132c9e2SJanosch Frank 	register int r1 asm("1") = sel2;
302c132c9e2SJanosch Frank 	int cc;
303c132c9e2SJanosch Frank 
304c132c9e2SJanosch Frank 	asm volatile(
305c132c9e2SJanosch Frank 		"stsi	0(%3)\n"
306c132c9e2SJanosch Frank 		"ipm	%[cc]\n"
307c132c9e2SJanosch Frank 		"srl	%[cc],28\n"
308c132c9e2SJanosch Frank 		: "+d" (r0), [cc] "=d" (cc)
309c132c9e2SJanosch Frank 		: "d" (r1), "a" (addr)
310c132c9e2SJanosch Frank 		: "cc", "memory");
311c132c9e2SJanosch Frank 	return cc;
312c132c9e2SJanosch Frank }
313c132c9e2SJanosch Frank 
314f9395bfeSClaudio Imbrenda static inline int servc(uint32_t command, unsigned long sccb)
315f9395bfeSClaudio Imbrenda {
316f9395bfeSClaudio Imbrenda 	int cc;
317f9395bfeSClaudio Imbrenda 
318f9395bfeSClaudio Imbrenda 	asm volatile(
319f9395bfeSClaudio Imbrenda 		"       .insn   rre,0xb2200000,%1,%2\n"  /* servc %1,%2 */
320f9395bfeSClaudio Imbrenda 		"       ipm     %0\n"
321f9395bfeSClaudio Imbrenda 		"       srl     %0,28"
322f9395bfeSClaudio Imbrenda 		: "=&d" (cc) : "d" (command), "a" (sccb)
323f9395bfeSClaudio Imbrenda 		: "cc", "memory");
324f9395bfeSClaudio Imbrenda 	return cc;
325f9395bfeSClaudio Imbrenda }
326f9395bfeSClaudio Imbrenda 
3276b3278c9SClaudio Imbrenda static inline void set_prefix(uint32_t new_prefix)
3286b3278c9SClaudio Imbrenda {
3296b3278c9SClaudio Imbrenda 	asm volatile("	spx %0" : : "Q" (new_prefix) : "memory");
3306b3278c9SClaudio Imbrenda }
3316b3278c9SClaudio Imbrenda 
3326b3278c9SClaudio Imbrenda static inline uint32_t get_prefix(void)
3336b3278c9SClaudio Imbrenda {
3346b3278c9SClaudio Imbrenda 	uint32_t current_prefix;
3356b3278c9SClaudio Imbrenda 
3366b3278c9SClaudio Imbrenda 	asm volatile("	stpx %0" : "=Q" (current_prefix));
3376b3278c9SClaudio Imbrenda 	return current_prefix;
3386b3278c9SClaudio Imbrenda }
3396b3278c9SClaudio Imbrenda 
340cfb204f9SDavid Hildenbrand #endif
341