Line data Source code
1 : /**
2 : * Copyright Notice:
3 : * Copyright 2025-2026 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 |= SPDM_GET_CAPABILITIES_REQUEST_FLAGS_MAC_CAP;
29 2 : spdm_context->connection_info.capability.flags |=
30 : SPDM_GET_CAPABILITIES_REQUEST_FLAGS_KEY_EX_CAP;
31 2 : spdm_context->connection_info.capability.flags |= SPDM_GET_CAPABILITIES_REQUEST_FLAGS_ENCAP_CAP;
32 :
33 2 : spdm_context->local_context.capability.flags |= SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_EVENT_CAP;
34 2 : spdm_context->local_context.capability.flags |=
35 : SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_ENCRYPT_CAP;
36 2 : spdm_context->local_context.capability.flags |= SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_MAC_CAP;
37 2 : spdm_context->local_context.capability.flags |= SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_KEY_EX_CAP;
38 2 : spdm_context->local_context.capability.flags |= SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_ENCAP_CAP;
39 :
40 2 : spdm_context->connection_info.algorithm.base_hash_algo = m_libspdm_use_hash_algo;
41 2 : spdm_context->connection_info.algorithm.base_asym_algo = m_libspdm_use_asym_algo;
42 2 : spdm_context->connection_info.algorithm.dhe_named_group = m_libspdm_use_dhe_algo;
43 2 : spdm_context->connection_info.algorithm.aead_cipher_suite = m_libspdm_use_aead_algo;
44 :
45 2 : spdm_context->latest_session_id = m_session_id;
46 2 : spdm_context->last_spdm_request_session_id_valid = true;
47 2 : spdm_context->last_spdm_request_session_id = m_session_id;
48 2 : session_info = &spdm_context->session_info[0];
49 2 : libspdm_session_info_init(spdm_context, session_info, m_session_id,
50 : SECURED_SPDM_VERSION_11 << SPDM_VERSION_NUMBER_SHIFT_BIT, true);
51 2 : libspdm_secured_message_set_session_state(
52 : session_info->secured_message_context,
53 : LIBSPDM_SESSION_STATE_ESTABLISHED);
54 2 : }
55 :
56 : /**
57 : * Test 1: Responder forms the expected SEND_EVENT request message with one event.
58 : **/
59 1 : static void rsp_encap_send_event_case1(void **state)
60 : {
61 : libspdm_return_t status;
62 : libspdm_test_context_t *spdm_test_context;
63 : libspdm_context_t *spdm_context;
64 1 : size_t request_buffer_size = sizeof(m_send_buffer);
65 : spdm_send_event_request_t *spdm_request;
66 :
67 1 : spdm_test_context = *state;
68 1 : spdm_context = spdm_test_context->spdm_context;
69 1 : spdm_test_context->case_id = 0x01;
70 :
71 1 : set_standard_state(spdm_context);
72 :
73 1 : g_event_count = 1;
74 :
75 1 : status = libspdm_get_encap_request_send_event(spdm_context, &request_buffer_size,
76 : m_send_buffer);
77 :
78 1 : assert_int_equal(status, LIBSPDM_STATUS_SUCCESS);
79 :
80 1 : spdm_request = (spdm_send_event_request_t *)m_send_buffer;
81 :
82 1 : assert_int_equal(spdm_request->header.spdm_version, SPDM_MESSAGE_VERSION_13);
83 1 : assert_int_equal(spdm_request->header.request_response_code, SPDM_SEND_EVENT);
84 1 : assert_int_equal(spdm_request->header.param1, 0);
85 1 : assert_int_equal(spdm_request->header.param1, 0);
86 1 : assert_int_equal(spdm_request->event_count, g_event_count);
87 :
88 1 : for (unsigned int index = 0;
89 4601 : index < request_buffer_size - sizeof(spdm_send_event_request_t);
90 4600 : index++) {
91 4600 : assert_int_equal((uint8_t)index, ((uint8_t *)(spdm_request + 1))[index]);
92 : }
93 1 : }
94 :
95 : /**
96 : * Test 2: Responder processes the encapsulated EVENT_ACK response.
97 : **/
98 1 : static void rsp_encap_send_event_case2(void **state)
99 : {
100 : libspdm_return_t status;
101 : libspdm_test_context_t *spdm_test_context;
102 : libspdm_context_t *spdm_context;
103 : size_t response_size;
104 : spdm_event_ack_response_t *spdm_response;
105 : bool need_continue;
106 :
107 1 : spdm_test_context = *state;
108 1 : spdm_context = spdm_test_context->spdm_context;
109 1 : spdm_test_context->case_id = 0x02;
110 :
111 1 : set_standard_state(spdm_context);
112 :
113 1 : spdm_response = (spdm_event_ack_response_t *)m_receive_buffer;
114 1 : response_size = sizeof(spdm_event_ack_response_t);
115 :
116 1 : spdm_response->header.spdm_version = SPDM_MESSAGE_VERSION_13;
117 1 : spdm_response->header.request_response_code = SPDM_EVENT_ACK;
118 1 : spdm_response->header.param1 = 0;
119 1 : spdm_response->header.param2 = 0;
120 :
121 1 : status = libspdm_process_encap_response_event_ack(spdm_context, response_size, spdm_response,
122 : &need_continue);
123 :
124 1 : assert_int_equal(status, LIBSPDM_STATUS_SUCCESS);
125 1 : assert_false(need_continue);
126 1 : }
127 :
128 1 : int libspdm_rsp_encap_send_event_test(void)
129 : {
130 1 : const struct CMUnitTest test_cases[] = {
131 : cmocka_unit_test(rsp_encap_send_event_case1),
132 : cmocka_unit_test(rsp_encap_send_event_case2),
133 : };
134 :
135 1 : libspdm_test_context_t test_context = {
136 : LIBSPDM_TEST_CONTEXT_VERSION,
137 : false,
138 : };
139 :
140 1 : libspdm_setup_test_context(&test_context);
141 :
142 1 : return cmocka_run_group_tests(test_cases,
143 : libspdm_unit_test_group_setup,
144 : libspdm_unit_test_group_teardown);
145 : }
146 :
147 : #endif /* (LIBSPDM_ENABLE_CAPABILITY_ENCAP_CAP) && (LIBSPDM_ENABLE_CAPABILITY_EVENT_CAP) */
|
