#ifndef _GUARDED_STORAGE_H
#define _GUARDED_STORAGE_H

#include <linux/types.h>

struct gs_cb {
	__u64 reserved;
	__u64 gsd;
	__u64 gssm;
	__u64 gs_epl_a;
};

struct gs_epl {
	__u8 pad1;
	union {
		__u8 gs_eam;
		struct {
			__u8	: 6;
			__u8 e	: 1;
			__u8 b	: 1;
		};
	};
	union {
		__u8 gs_eci;
		struct {
			__u8 tx	: 1;
			__u8 cx	: 1;
			__u8	: 5;
			__u8 in	: 1;
		};
	};
	union {
		__u8 gs_eai;
		struct {
			__u8	: 1;
			__u8 t	: 1;
			__u8 as	: 2;
			__u8 ar	: 4;
		};
	};
	__u32 pad2;
	__u64 gs_eha;
	__u64 gs_eia;
	__u64 gs_eoa;
	__u64 gs_eir;
	__u64 gs_era;
};

#define GS_ENABLE	0
#define	GS_DISABLE	1
#define GS_SET_BC_CB	2
#define GS_CLEAR_BC_CB	3
#define GS_BROADCAST	4

static inline void load_gs_cb(struct gs_cb *gs_cb)
{
	asm volatile(".insn rxy,0xe3000000004d,0,%0" : : "Q" (*gs_cb));
}

static inline void store_gs_cb(struct gs_cb *gs_cb)
{
	asm volatile(".insn rxy,0xe30000000049,0,%0" : : "Q" (*gs_cb));
}

static inline void save_gs_cb(struct gs_cb *gs_cb)
{
	if (gs_cb)
		store_gs_cb(gs_cb);
}

static inline void restore_gs_cb(struct gs_cb *gs_cb)
{
	if (gs_cb)
		load_gs_cb(gs_cb);
}

#endif /* _GUARDED_STORAGE_H */
