Add responder unit_test for get_key_pair_info

Signed-off-by: Wenxing Hou <[email protected]>

unit_test/test_spdm_responder/CMakeLists.txt

@@ -47,6 +47,7 @@ target_sources(test_spdm_responder
         receive_send.c
         chunk_get.c
         chunk_send_ack.c
+        key_pair_info.c
         ${LIBSPDM_DIR}/unit_test/spdm_unit_test_common/common.c
         ${LIBSPDM_DIR}/unit_test/spdm_unit_test_common/algo.c
         ${LIBSPDM_DIR}/unit_test/spdm_unit_test_common/support.c

unit_test/test_spdm_responder/key_pair_info.c

@@ -0,0 +1,87 @@
+/**
+ *  Copyright Notice:
+ *  Copyright 2024 DMTF. All rights reserved.
+ *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
+ **/
+
+#include "spdm_unit_test.h"
+#include "internal/libspdm_responder_lib.h"
+#include "internal/libspdm_requester_lib.h"
+
+#if LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP
+
+spdm_get_key_pair_info_request_t m_libspdm_get_key_pair_info_request1 = {
+    { SPDM_MESSAGE_VERSION_13, SPDM_GET_KEY_PAIR_INFO, 0, 0 },
+    4
+};
+size_t m_libspdm_get_key_pair_info_request1_size = sizeof(m_libspdm_get_key_pair_info_request1);
+
+/**
+ * Test 1: Successful response to get key pair info with key pair id 4
+ * Expected Behavior: get a LIBSPDM_STATUS_SUCCESS return code, and correct response message size and fields
+ **/
+void libspdm_test_responder_key_pair_info_case1(void **state)
+{
+    libspdm_return_t status;
+    libspdm_test_context_t *spdm_test_context;
+    libspdm_context_t *spdm_context;
+    size_t response_size;
+    uint8_t response[LIBSPDM_MAX_SPDM_MSG_SIZE];
+    spdm_key_pair_info_response_t *spdm_response;
+    uint8_t key_pair_id;
+    uint16_t public_key_info_len;
+    uint8_t public_key_info_ecp256[] = {0x30, 0x13, 0x06, 0x07, 0x2A, 0x86, 0x48, 0xCE, 0x3D,
+                                        0x02, 0x01, 0x06, 0x08, 0x2A, 0x86, 0x48, 0xCE, 0x3D,
+                                        0x03, 0x01, 0x07};
+
+    spdm_test_context = *state;
+    spdm_context = spdm_test_context->spdm_context;
+    spdm_test_context->case_id = 0x1;
+    spdm_context->connection_info.version = SPDM_MESSAGE_VERSION_13 <<
+                                            SPDM_VERSION_NUMBER_SHIFT_BIT;
+    spdm_context->connection_info.connection_state =
+        LIBSPDM_CONNECTION_STATE_AUTHENTICATED;
+    spdm_context->connection_info.algorithm.base_asym_algo =
+        m_libspdm_use_asym_algo;
+    spdm_context->local_context.capability.flags |=
+        SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_GET_KEY_PAIR_INFO_CAP;
+    spdm_context->local_context.total_key_pairs = 16;
+
+    key_pair_id = 4;
+    public_key_info_len = sizeof(public_key_info_ecp256);
+
+    response_size = sizeof(response);
+
+    status = libspdm_get_response_key_pair_info(
+        spdm_context, m_libspdm_get_key_pair_info_request1_size,
+        &m_libspdm_get_key_pair_info_request1, &response_size, response);
+    assert_int_equal(status, LIBSPDM_STATUS_SUCCESS);
+    assert_int_equal(response_size,
+                     sizeof(spdm_key_pair_info_response_t) + public_key_info_len);
+    spdm_response = (void *)response;
+    assert_int_equal(spdm_response->header.request_response_code,
+                     SPDM_KEY_PAIR_INFO);
+    assert_int_equal(spdm_response->key_pair_id,
+                     key_pair_id);
+}
+
+libspdm_test_context_t m_libspdm_responder_key_pair_info_test_context = {
+    LIBSPDM_TEST_CONTEXT_VERSION,
+    false,
+};
+
+int libspdm_responder_key_pair_info_test_main(void)
+{
+    const struct CMUnitTest spdm_responder_key_pair_info_tests[] = {
+        /* Success Case to get key pair info*/
+        cmocka_unit_test(libspdm_test_responder_key_pair_info_case1),
+    };
+
+    libspdm_setup_test_context(&m_libspdm_responder_key_pair_info_test_context);
+
+    return cmocka_run_group_tests(spdm_responder_key_pair_info_tests,
+                                  libspdm_unit_test_group_setup,
+                                  libspdm_unit_test_group_teardown);
+}
+
+#endif /* LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP*/

unit_test/test_spdm_responder/test_spdm_responder.c

@@ -28,6 +28,10 @@ int libspdm_responder_measurements_test_main(void);
 int libspdm_responder_measurement_extension_log_test_main(void);
 #endif /* LIBSPDM_ENABLE_CAPABILITY_MEL_CAP*/
 
+#if LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP
+int libspdm_responder_key_pair_info_test_main(void);
+#endif /* LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP*/
+
 #if LIBSPDM_RESPOND_IF_READY_SUPPORT
 #if (LIBSPDM_ENABLE_CAPABILITY_CERT_CAP || LIBSPDM_ENABLE_CAPABILITY_CHAL_CAP || \
      LIBSPDM_ENABLE_CAPABILITY_MEAS_CAP || LIBSPDM_ENABLE_CAPABILITY_KEY_EX_CAP || \
@@ -129,6 +133,12 @@ int main(void)
     }
     #endif /* LIBSPDM_ENABLE_CAPABILITY_MEL_CAP*/
 
+    #if LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP
+    if (libspdm_responder_key_pair_info_test_main() != 0) {
+        return_value = 1;
+    }
+    #endif /* LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP*/
+
     #if LIBSPDM_RESPOND_IF_READY_SUPPORT
     #if (LIBSPDM_ENABLE_CAPABILITY_CERT_CAP || LIBSPDM_ENABLE_CAPABILITY_CHAL_CAP || \
          LIBSPDM_ENABLE_CAPABILITY_MEAS_CAP || LIBSPDM_ENABLE_CAPABILITY_KEY_EX_CAP || \