Nmi branch test

include/linux/bpf.h

@@ -1793,6 +1793,18 @@ struct rex_saved_states{
 	struct bpf_link *link;
 };
 
+struct rex_mem {
+	void *mem;
+	u32 total_page;
+};
+
+struct rex_saved_states{
+	int cpu_id;
+	int loader_pid;
+	u64 unwinder_insn_off;
+	struct bpf_link *link;
+};
+
 struct bpf_prog {
 	u16			pages;		/* Number of allocated pages */
 	u16			jited:1,	/* Is our filter JIT'ed? */

include/uapi/linux/bpf.h

@@ -992,9 +992,11 @@ enum bpf_cmd {
 	BPF_PROG_BIND_MAP,
 	BPF_TOKEN_CREATE,
 	BPF_PROG_STREAM_READ_BY_FD,
-	BPF_PROG_ASSOC_STRUCT_OPS,
 	BPF_PROG_LOAD_REX_BASE,
 	BPF_PROG_LOAD_REX,
+	BPF_SCHED_EXT_ATTACH_REX,
+	BPF_SCHED_EXT_DETACH_REX,
+	BPF_PROG_ASSOC_STRUCT_OPS,
 	BPF_PROG_TERMINATE,
 	__MAX_BPF_CMD,
 };
@@ -1528,6 +1530,11 @@ struct rex_text_sym {
 	const char __user	*symbol;
 };
 
+struct rex_sched_ops_sym {
+	const char __user	*name;
+	__u64			offset;
+};
+
 union bpf_attr {
 	struct { /* anonymous struct used by BPF_MAP_CREATE command */
 		__u32	map_type;	/* one of enum bpf_map_type */
@@ -1955,6 +1962,17 @@ union bpf_attr {
 		__u32		prog_fd;
 	} prog_stream_read;
 
+	struct { /* BPF_SCHED_EXT_ATTACH_REX */
+		__u32		base_prog_fd;
+		__aligned_u64	sched_ops_syms;		/* ptr to rex_sched_ops_sym array */
+		__u32		nr_sched_ops_syms;
+		__u32		timeout_ms;		/* ops.timeout_ms, 0 = default */
+		__u32		exit_dump_len;		/* ops.exit_dump_len, 0 = default */
+		__u32		pad;
+		__aligned_u64	ops_flags;		/* SCX_OPS_* flags */
+		char		name[128];		/* ops.name; empty string = use base->aux->name */
+	} sched_ext_attach;
+
 	struct {
 		__u32		map_fd;
 		__u32		prog_fd;