2 #include <kunit/test.h>
6 static struct mptcp_subflow_request_sock *build_req_sock(struct kunit *test)  in build_req_sock()  argument
10 	req = kunit_kzalloc(test, sizeof(struct mptcp_subflow_request_sock),  in build_req_sock()
12 	KUNIT_EXPECT_NOT_ERR_OR_NULL(test, req);  in build_req_sock()
17 static void mptcp_token_test_req_basic(struct kunit *test)  in mptcp_token_test_req_basic()  argument
19 	struct mptcp_subflow_request_sock *req = build_req_sock(test);  in mptcp_token_test_req_basic()
22 	KUNIT_ASSERT_EQ(test, 0,  in mptcp_token_test_req_basic()
24 	KUNIT_EXPECT_NE(test, 0, (int)req->token);  in mptcp_token_test_req_basic()
25 	KUNIT_EXPECT_PTR_EQ(test, null_msk, mptcp_token_get_sock(req->token));  in mptcp_token_test_req_basic()
31 static struct inet_connection_sock *build_icsk(struct kunit *test)  in build_icsk()  argument
35 	icsk = kunit_kzalloc(test, sizeof(struct inet_connection_sock),  in build_icsk()
37 	KUNIT_EXPECT_NOT_ERR_OR_NULL(test, icsk);  in build_icsk()
41 static struct mptcp_subflow_context *build_ctx(struct kunit *test)  in build_ctx()  argument
45 	ctx = kunit_kzalloc(test, sizeof(struct mptcp_subflow_context),  in build_ctx()
47 	KUNIT_EXPECT_NOT_ERR_OR_NULL(test, ctx);  in build_ctx()
51 static struct mptcp_sock *build_msk(struct kunit *test)  in build_msk()  argument
55 	msk = kunit_kzalloc(test, sizeof(struct mptcp_sock), GFP_USER);  in build_msk()
56 	KUNIT_EXPECT_NOT_ERR_OR_NULL(test, msk);  in build_msk()
61 static void mptcp_token_test_msk_basic(struct kunit *test)  in mptcp_token_test_msk_basic()  argument
63 	struct inet_connection_sock *icsk = build_icsk(test);  in mptcp_token_test_msk_basic()
64 	struct mptcp_subflow_context *ctx = build_ctx(test);  in mptcp_token_test_msk_basic()
65 	struct mptcp_sock *msk = build_msk(test);  in mptcp_token_test_msk_basic()
73 	KUNIT_ASSERT_EQ(test, 0,  in mptcp_token_test_msk_basic()
75 	KUNIT_EXPECT_NE(test, 0, (int)ctx->token);  in mptcp_token_test_msk_basic()
76 	KUNIT_EXPECT_EQ(test, ctx->token, msk->token);  in mptcp_token_test_msk_basic()
77 	KUNIT_EXPECT_PTR_EQ(test, msk, mptcp_token_get_sock(ctx->token));  in mptcp_token_test_msk_basic()
78 	KUNIT_EXPECT_EQ(test, 2, (int)refcount_read(&sk->sk_refcnt));  in mptcp_token_test_msk_basic()
81 	KUNIT_EXPECT_PTR_EQ(test, null_msk, mptcp_token_get_sock(ctx->token));  in mptcp_token_test_msk_basic()
84 static void mptcp_token_test_accept(struct kunit *test)  in mptcp_token_test_accept()  argument
86 	struct mptcp_subflow_request_sock *req = build_req_sock(test);  in mptcp_token_test_accept()
87 	struct mptcp_sock *msk = build_msk(test);  in mptcp_token_test_accept()
89 	KUNIT_ASSERT_EQ(test, 0,  in mptcp_token_test_accept()
93 	KUNIT_EXPECT_PTR_EQ(test, msk, mptcp_token_get_sock(msk->token));  in mptcp_token_test_accept()
97 	KUNIT_EXPECT_PTR_EQ(test, msk, mptcp_token_get_sock(msk->token));  in mptcp_token_test_accept()
103 static void mptcp_token_test_destroyed(struct kunit *test)  in mptcp_token_test_destroyed()  argument
105 	struct mptcp_subflow_request_sock *req = build_req_sock(test);  in mptcp_token_test_destroyed()
106 	struct mptcp_sock *msk = build_msk(test);  in mptcp_token_test_destroyed()
112 	KUNIT_ASSERT_EQ(test, 0,  in mptcp_token_test_destroyed()
119 	KUNIT_EXPECT_PTR_EQ(test, null_msk, mptcp_token_get_sock(msk->token));  in mptcp_token_test_destroyed()