1 files changed, 29 insertions, 0 deletions

diff --git a/sys/security/mac_test/mac_test.c b/sys/security/mac_test/mac_test.c
index 7a6a76ce23cc..890b8328055e 100644
--- a/sys/security/mac_test/mac_test.c
+++ b/sys/security/mac_test/mac_test.c
@@ -257,6 +257,32 @@ test_cred_check_setauid(struct ucred *cred, uid_t auid)
 	return (0);
 }
 
+COUNTER_DECL(cred_setcred_enter);
+static void
+test_cred_setcred_enter(void)
+{
+	COUNTER_INC(cred_setcred_enter);
+}
+
+COUNTER_DECL(cred_check_setcred);
+static int
+test_cred_check_setcred(u_int flags, const struct ucred *old_cred,
+    struct ucred *new_cred)
+{
+	LABEL_CHECK(old_cred->cr_label, MAGIC_CRED);
+	LABEL_CHECK(new_cred->cr_label, MAGIC_CRED);
+	COUNTER_INC(cred_check_setcred);
+
+	return (0);
+}
+
+COUNTER_DECL(cred_setcred_exit);
+static void
+test_cred_setcred_exit(void)
+{
+	COUNTER_INC(cred_setcred_exit);
+}
+
 COUNTER_DECL(cred_check_setegid);
 static int
 test_cred_check_setegid(struct ucred *cred, gid_t egid)
@@ -3033,6 +3059,9 @@ static struct mac_policy_ops test_ops =
 	.mpo_cred_check_setaudit = test_cred_check_setaudit,
 	.mpo_cred_check_setaudit_addr = test_cred_check_setaudit_addr,
 	.mpo_cred_check_setauid = test_cred_check_setauid,
+	.mpo_cred_setcred_enter = test_cred_setcred_enter,
+	.mpo_cred_check_setcred = test_cred_check_setcred,
+	.mpo_cred_setcred_exit = test_cred_setcred_exit,
 	.mpo_cred_check_seteuid = test_cred_check_seteuid,
 	.mpo_cred_check_setegid = test_cred_check_setegid,
 	.mpo_cred_check_setgid = test_cred_check_setgid,