X-Git-Url: https://git.whamcloud.com/?a=blobdiff_plain;f=lustre%2Finclude%2Flustre_nrs_tbf.h;h=0a407197c36f65aea05d9fe74e761ae64b3b94c3;hb=2124e07ddde7789ebf0e8ec5600f2aff31788123;hp=3d52f7822c02bbc5189ff74cb8eb53e0b9102e37;hpb=d40b3f9663c7627f0f08c4e3b5efa6cd206ca0f0;p=fs%2Flustre-release.git

diff --git a/lustre/include/lustre_nrs_tbf.h b/lustre/include/lustre_nrs_tbf.h
index 3d52f78..0a40719 100644
--- a/lustre/include/lustre_nrs_tbf.h
+++ b/lustre/include/lustre_nrs_tbf.h
@@ -42,11 +42,40 @@
 struct nrs_tbf_head;
 struct nrs_tbf_cmd;
 
+#define NRS_TBF_MATCH_FULL	0x0000001
+#define NRS_TBF_MATCH_WILDCARD	0x0000002
+
 struct nrs_tbf_jobid {
 	char		*tj_id;
+	__u32		 tj_match_flag;
 	struct list_head tj_linkage;
 };
 
+#define MAX_U32_STR_LEN	10
+#define NRS_TBF_KEY_LEN	(LNET_NIDSTR_SIZE + LUSTRE_JOBID_SIZE + \
+			 MAX_U32_STR_LEN + MAX_U32_STR_LEN + 3 + 2)
+
+enum nrs_tbf_flag {
+	NRS_TBF_FLAG_INVALID    = 0x0000000,
+	NRS_TBF_FLAG_JOBID      = 0x0000001,
+	NRS_TBF_FLAG_NID        = 0x0000002,
+	NRS_TBF_FLAG_OPCODE     = 0x0000004,
+	NRS_TBF_FLAG_GENERIC    = 0x0000008,
+	NRS_TBF_FLAG_UID        = 0x0000010,
+	NRS_TBF_FLAG_GID        = 0x0000020,
+};
+
+struct tbf_id {
+	enum nrs_tbf_flag	ti_type;
+	u32			ti_uid;
+	u32			ti_gid;
+};
+
+struct nrs_tbf_id {
+	struct tbf_id		nti_id;
+	struct list_head	nti_linkage;
+};
+
 struct nrs_tbf_client {
 	/** Resource object for policy instance. */
 	struct ptlrpc_nrs_resource	 tc_res;
@@ -56,6 +85,12 @@ struct nrs_tbf_client {
 	lnet_nid_t			 tc_nid;
 	/** Jobid of the client. */
 	char				 tc_jobid[LUSTRE_JOBID_SIZE];
+	/** opcode of the client. */
+	__u32				 tc_opcode;
+	/** gid or uid of the client. */
+	struct tbf_id			tc_id;
+	/** Hash key of the client. */
+	char				 tc_key[NRS_TBF_KEY_LEN];
 	/** Reference number of the client. */
 	atomic_t			 tc_ref;
 	/** Lock to protect rule and linkage. */
@@ -76,6 +111,13 @@ struct nrs_tbf_client {
 	__u64				 tc_depth;
 	/** Time check-point. */
 	__u64				 tc_check_time;
+	/** Deadline of a class */
+	__u64				 tc_deadline;
+	/**
+	 * Time residue: the remainder of elapsed time
+	 * divided by nsecs when dequeue a request.
+	 */
+	__u64				 tc_nsecs_resid;
 	/** List of queued requests. */
 	struct list_head		 tc_list;
 	/** Node in binary heap. */
@@ -93,8 +135,11 @@ struct nrs_tbf_client {
 
 #define MAX_TBF_NAME (16)
 
-#define NTRS_STOPPING	0x0000001
-#define NTRS_DEFAULT	0x0000002
+enum nrs_rule_flags {
+	NTRS_STOPPING	= 0x00000001,
+	NTRS_DEFAULT	= 0x00000002,
+	NTRS_REALTIME	= 0x00000004,
+};
 
 struct nrs_tbf_rule {
 	/** Name of the rule. */
@@ -111,6 +156,18 @@ struct nrs_tbf_rule {
 	struct list_head		 tr_jobids;
 	/** Jobid list string of the rule.*/
 	char				*tr_jobids_str;
+	/** uid/gid list of the rule. */
+	struct list_head		tr_ids;
+	/** uid/gid list string of the rule. */
+	char				*tr_ids_str;
+	/** Opcode bitmap of the rule. */
+	struct cfs_bitmap		*tr_opcodes;
+	/** Opcode list string of the rule.*/
+	char				*tr_opcodes_str;
+	/** Condition list of the rule.*/
+	struct list_head		tr_conds;
+	/** Generic condition string of the rule. */
+	char				*tr_conds_str;
 	/** RPC/s limit. */
 	__u64				 tr_rpc_rate;
 	/** Time to wait for next token. */
@@ -122,7 +179,7 @@ struct nrs_tbf_rule {
 	/** List of client. */
 	struct list_head		 tr_cli_list;
 	/** Flags of the rule. */
-	__u32				 tr_flags;
+	enum nrs_rule_flags		 tr_flags;
 	/** Usage Reference count taken on the rule. */
 	atomic_t			 tr_ref;
 	/** Generation of the rule. */
@@ -149,9 +206,17 @@ struct nrs_tbf_ops {
 
 #define NRS_TBF_TYPE_JOBID	"jobid"
 #define NRS_TBF_TYPE_NID	"nid"
+#define NRS_TBF_TYPE_OPCODE	"opcode"
+#define NRS_TBF_TYPE_GENERIC	"generic"
+#define NRS_TBF_TYPE_UID	"uid"
+#define NRS_TBF_TYPE_GID	"gid"
 #define NRS_TBF_TYPE_MAX_LEN	20
-#define NRS_TBF_FLAG_JOBID	0x0000001
-#define NRS_TBF_FLAG_NID	0x0000002
+
+struct nrs_tbf_type {
+	const char		*ntt_name;
+	enum nrs_tbf_flag	 ntt_flag;
+	struct nrs_tbf_ops	*ntt_ops;
+};
 
 struct nrs_tbf_bucket {
 	/**
@@ -226,19 +291,61 @@ struct nrs_tbf_head {
 enum nrs_tbf_cmd_type {
 	NRS_CTL_TBF_START_RULE = 0,
 	NRS_CTL_TBF_STOP_RULE,
-	NRS_CTL_TBF_CHANGE_RATE,
+	NRS_CTL_TBF_CHANGE_RULE,
 };
 
 struct nrs_tbf_cmd {
-	enum nrs_tbf_cmd_type	 tc_cmd;
-	char			*tc_name;
-	__u64			 tc_rpc_rate;
-	struct list_head	 tc_nids;
-	char			*tc_nids_str;
-	struct list_head	 tc_jobids;
-	char			*tc_jobids_str;
-	__u32			 tc_valid_types;
-	__u32			 tc_rule_flags;
+	enum nrs_tbf_cmd_type			 tc_cmd;
+	char					*tc_name;
+	union {
+		struct nrs_tbf_cmd_start {
+			__u64			 ts_rpc_rate;
+			struct list_head	 ts_nids;
+			char			*ts_nids_str;
+			struct list_head	 ts_jobids;
+			char			*ts_jobids_str;
+			struct list_head	 ts_ids;
+			char			*ts_ids_str;
+			struct cfs_bitmap	*ts_opcodes;
+			char			*ts_opcodes_str;
+			struct list_head	 ts_conds;
+			char			*ts_conds_str;
+			__u32			 ts_valid_type;
+			enum nrs_rule_flags	 ts_rule_flags;
+			char			*ts_next_name;
+		} tc_start;
+		struct nrs_tbf_cmd_change {
+			__u64			 tc_rpc_rate;
+			char			*tc_next_name;
+		} tc_change;
+	} u;
+};
+
+enum nrs_tbf_field {
+	NRS_TBF_FIELD_NID,
+	NRS_TBF_FIELD_JOBID,
+	NRS_TBF_FIELD_OPCODE,
+	NRS_TBF_FIELD_UID,
+	NRS_TBF_FIELD_GID,
+	NRS_TBF_FIELD_MAX
+};
+
+struct nrs_tbf_expression {
+	enum nrs_tbf_field	 te_field;
+	struct list_head	 te_cond;
+	struct cfs_bitmap	*te_opcodes;
+	struct list_head	 te_linkage;
+};
+
+struct nrs_tbf_conjunction {
+	/**
+	 * link to disjunction.
+	 */
+	struct list_head tc_linkage;
+	/**
+	 * list of logical conjunction
+	 */
+	struct list_head tc_expressions;
 };
 
 struct nrs_tbf_req {
@@ -264,6 +371,10 @@ enum nrs_ctl_tbf {
 	 * Write the the data of a TBF policy.
 	 */
 	NRS_CTL_TBF_WR_RULE,
+	/**
+	 * Read the TBF policy type preset by proc entry "nrs_policies".
+	 */
+	NRS_CTL_TBF_RD_TYPE_FLAG,
 };
 
 /** @} tbf */