Line data Source code
1 : /**
2 : * Copyright Notice:
3 : * Copyright 2025 DMTF. All rights reserved.
4 : * License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
5 : **/
6 :
7 : #include "spdm_unit_test.h"
8 : #include "internal/libspdm_responder_lib.h"
9 :
10 : #if (LIBSPDM_ENABLE_CAPABILITY_ENCAP_CAP) && (LIBSPDM_ENABLE_CAPABILITY_EVENT_CAP)
11 :
12 : extern uint32_t g_event_count;
13 :
14 : static uint8_t m_send_buffer[LIBSPDM_MAX_SPDM_MSG_SIZE];
15 : static uint8_t m_receive_buffer[LIBSPDM_MAX_SPDM_MSG_SIZE];
16 : static uint32_t m_session_id = 0xFFFFFFFF;
17 :
18 2 : static void set_standard_state(libspdm_context_t *spdm_context)
19 : {
20 : libspdm_session_info_t *session_info;
21 :
22 2 : spdm_context->connection_info.version = SPDM_MESSAGE_VERSION_13 <<
23 : SPDM_VERSION_NUMBER_SHIFT_BIT;
24 2 : spdm_context->connection_info.connection_state = LIBSPDM_CONNECTION_STATE_NEGOTIATED;
25 :
26 2 : spdm_context->connection_info.capability.flags |=
27 : SPDM_GET_CAPABILITIES_REQUEST_FLAGS_ENCRYPT_CAP;
28 2 : spdm_context->connection_info.capability.flags |=
29 : SPDM_GET_CAPABILITIES_REQUEST_FLAGS_MAC_CAP;
30 2 : spdm_context->connection_info.capability.flags |=
31 : SPDM_GET_CAPABILITIES_REQUEST_FLAGS_KEY_EX_CAP;
32 2 : spdm_context->connection_info.capability.flags |=
33 : SPDM_GET_CAPABILITIES_REQUEST_FLAGS_ENCAP_CAP;
34 :
35 2 : spdm_context->local_context.capability.flags |=
36 : SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_EVENT_CAP;
37 2 : spdm_context->local_context.capability.flags |=
38 : SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_ENCRYPT_CAP;
39 2 : spdm_context->local_context.capability.flags |=
40 : SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_MAC_CAP;
41 2 : spdm_context->local_context.capability.flags |=
42 : SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_KEY_EX_CAP;
43 2 : spdm_context->local_context.capability.flags |=
44 : SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_ENCAP_CAP;
45 :
46 2 : spdm_context->connection_info.algorithm.base_hash_algo = m_libspdm_use_hash_algo;
47 2 : spdm_context->connection_info.algorithm.base_asym_algo = m_libspdm_use_asym_algo;
48 2 : spdm_context->connection_info.algorithm.dhe_named_group = m_libspdm_use_dhe_algo;
49 2 : spdm_context->connection_info.algorithm.aead_cipher_suite = m_libspdm_use_aead_algo;
50 :
51 2 : spdm_context->latest_session_id = m_session_id;
52 2 : spdm_context->last_spdm_request_session_id_valid = true;
53 2 : spdm_context->last_spdm_request_session_id = m_session_id;
54 2 : session_info = &spdm_context->session_info[0];
55 2 : libspdm_session_info_init(spdm_context, session_info, m_session_id,
56 : SECURED_SPDM_VERSION_11 << SPDM_VERSION_NUMBER_SHIFT_BIT, true);
57 2 : libspdm_secured_message_set_session_state(
58 : session_info->secured_message_context,
59 : LIBSPDM_SESSION_STATE_ESTABLISHED);
60 2 : }
61 :
62 : /**
63 : * Test 1: Responder forms the expected SEND_EVENT request message with one event.
64 : **/
65 1 : static void rsp_encap_send_event_case1(void **state)
66 : {
67 : libspdm_return_t status;
68 : libspdm_test_context_t *spdm_test_context;
69 : libspdm_context_t *spdm_context;
70 1 : size_t request_buffer_size = sizeof(m_send_buffer);
71 : spdm_send_event_request_t *spdm_request;
72 :
73 1 : spdm_test_context = *state;
74 1 : spdm_context = spdm_test_context->spdm_context;
75 1 : spdm_test_context->case_id = 0x01;
76 :
77 1 : set_standard_state(spdm_context);
78 :
79 1 : g_event_count = 1;
80 :
81 1 : status = libspdm_get_encap_request_send_event(spdm_context, &request_buffer_size,
82 : m_send_buffer);
83 :
84 1 : assert_int_equal(status, LIBSPDM_STATUS_SUCCESS);
85 :
86 1 : spdm_request = (spdm_send_event_request_t *)m_send_buffer;
87 :
88 1 : assert_int_equal(spdm_request->header.spdm_version, SPDM_MESSAGE_VERSION_13);
89 1 : assert_int_equal(spdm_request->header.request_response_code, SPDM_SEND_EVENT);
90 1 : assert_int_equal(spdm_request->header.param1, 0);
91 1 : assert_int_equal(spdm_request->header.param1, 0);
92 1 : assert_int_equal(spdm_request->event_count, g_event_count);
93 :
94 1 : for (unsigned int index = 0;
95 4601 : index < request_buffer_size - sizeof(spdm_send_event_request_t);
96 4600 : index++) {
97 4600 : assert_int_equal((uint8_t)index, ((uint8_t *)(spdm_request + 1))[index]);
98 : }
99 1 : }
100 :
101 : /**
102 : * Test 2: Responder processes the encapsulated EVENT_ACK response.
103 : **/
104 1 : static void rsp_encap_send_event_case2(void **state)
105 : {
106 : libspdm_return_t status;
107 : libspdm_test_context_t *spdm_test_context;
108 : libspdm_context_t *spdm_context;
109 : size_t response_size;
110 : spdm_event_ack_response_t *spdm_response;
111 : bool need_continue;
112 :
113 1 : spdm_test_context = *state;
114 1 : spdm_context = spdm_test_context->spdm_context;
115 1 : spdm_test_context->case_id = 0x02;
116 :
117 1 : set_standard_state(spdm_context);
118 :
119 1 : spdm_response = (spdm_event_ack_response_t *)m_receive_buffer;
120 1 : response_size = sizeof(spdm_event_ack_response_t);
121 :
122 1 : spdm_response->header.spdm_version = SPDM_MESSAGE_VERSION_13;
123 1 : spdm_response->header.request_response_code = SPDM_EVENT_ACK;
124 1 : spdm_response->header.param1 = 0;
125 1 : spdm_response->header.param2 = 0;
126 :
127 1 : status = libspdm_process_encap_response_event_ack(spdm_context, response_size, spdm_response,
128 : &need_continue);
129 :
130 1 : assert_int_equal(status, LIBSPDM_STATUS_SUCCESS);
131 1 : assert_false(need_continue);
132 1 : }
133 :
134 1 : int libspdm_rsp_encap_send_event_test(void)
135 : {
136 1 : const struct CMUnitTest test_cases[] = {
137 : cmocka_unit_test(rsp_encap_send_event_case1),
138 : cmocka_unit_test(rsp_encap_send_event_case2),
139 : };
140 :
141 1 : libspdm_test_context_t test_context = {
142 : LIBSPDM_TEST_CONTEXT_VERSION,
143 : false,
144 : };
145 :
146 1 : libspdm_setup_test_context(&test_context);
147 :
148 1 : return cmocka_run_group_tests(test_cases,
149 : libspdm_unit_test_group_setup,
150 : libspdm_unit_test_group_teardown);
151 : }
152 :
153 : #endif /* (LIBSPDM_ENABLE_CAPABILITY_ENCAP_CAP) && (LIBSPDM_ENABLE_CAPABILITY_EVENT_CAP) */
|
