Skip to content

Commit

Permalink
Add requester unit_test for get_key_pair_info
Browse files Browse the repository at this point in the history 
Signed-off-by: Wenxing Hou <[email protected]>
  • Loading branch information
@Wenxing-hou
Wenxing-hou committed Jul 29, 2024
1 parent cd355b4 commit 2c8ab83
Show file tree
Hide file tree
Showing 3 changed files with 235 additions and 0 deletions.
1 change: 1 addition & 0 deletions unit_test/test_spdm_requester/CMakeLists.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ target_sources(test_spdm_requester
chunk_get.c
chunk_send.c
vendor_request.c
get_key_pair_info.c
error_test/get_version_err.c
error_test/get_capabilities_err.c
error_test/negotiate_algorithms_err.c
Expand Down
223 changes: 223 additions & 0 deletions unit_test/test_spdm_requester/get_key_pair_info.c
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,223 @@
/**
* 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_requester_lib.h"
#include "internal/libspdm_secured_message_lib.h"

#if LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP

libspdm_return_t libspdm_requester_get_key_pair_info_test_send_message(
void *spdm_context, size_t request_size, const void *request,
uint64_t timeout)
{
libspdm_test_context_t *spdm_test_context;

spdm_test_context = libspdm_get_test_context();
switch (spdm_test_context->case_id) {
case 0x1:
return LIBSPDM_STATUS_SEND_FAIL;
case 0x2:
return LIBSPDM_STATUS_SUCCESS;
default:
return LIBSPDM_STATUS_SEND_FAIL;
}
}

libspdm_return_t libspdm_requester_get_key_pair_info_test_receive_message(
void *spdm_context, size_t *response_size,
void **response, uint64_t timeout)
{
libspdm_test_context_t *spdm_test_context;

spdm_test_context = libspdm_get_test_context();
switch (spdm_test_context->case_id) {
case 0x1:
return LIBSPDM_STATUS_RECEIVE_FAIL;

case 0x2: {
spdm_key_pair_info_response_t *spdm_response;
size_t spdm_response_size;
size_t transport_header_size;

uint8_t total_key_pairs;
uint8_t key_pair_id;
uint16_t public_key_info_len;
uint16_t capabilities;
uint16_t key_usage_capabilities;
uint16_t current_key_usage;
uint32_t asym_algo_capabilities;
uint32_t current_asym_algo;
uint8_t assoc_cert_slot_mask;

uint8_t public_key_info_rsa2048[] = {0x30, 0x0D, 0x06, 0x09, 0x2A, 0x86, 0x48, 0x86, 0xF7,
0x0D, 0x01, 0x01, 0x01, 0x05, 0x00};

key_pair_id = 1;
total_key_pairs = 1;
public_key_info_len = (uint16_t)sizeof(public_key_info_rsa2048);

transport_header_size = LIBSPDM_TEST_TRANSPORT_HEADER_SIZE;
spdm_response = (void *)((uint8_t *)*response + transport_header_size);
spdm_response_size = sizeof(spdm_key_pair_info_response_t) + public_key_info_len;

capabilities = SPDM_KEY_PAIR_CAP_GEN_KEY_CAP;
key_usage_capabilities = SPDM_KEY_USAGE_BIT_MASK_KEY_EX_USE;
current_key_usage = SPDM_KEY_USAGE_BIT_MASK_KEY_EX_USE;
asym_algo_capabilities = SPDM_KEY_PAIR_ASYM_ALGO_CAP_RSA2048;
current_asym_algo = SPDM_KEY_PAIR_ASYM_ALGO_CAP_RSA2048;

/*association with slot 1*/
assoc_cert_slot_mask = 0x02;

spdm_response->header.spdm_version = SPDM_MESSAGE_VERSION_13;
spdm_response->header.request_response_code = SPDM_KEY_PAIR_INFO;
spdm_response->header.param1 = 0;
spdm_response->header.param2 = 0;
spdm_response->total_key_pairs = total_key_pairs;
spdm_response->key_pair_id = key_pair_id;
spdm_response->capabilities = capabilities;
spdm_response->key_usage_capabilities = key_usage_capabilities;
spdm_response->current_key_usage = current_key_usage;
spdm_response->asym_algo_capabilities = asym_algo_capabilities;
spdm_response->current_asym_algo = current_asym_algo;
spdm_response->public_key_info_len = public_key_info_len;
spdm_response->assoc_cert_slot_mask = assoc_cert_slot_mask;

libspdm_copy_mem((void*)(spdm_response + 1), public_key_info_len,
public_key_info_rsa2048, public_key_info_len);

libspdm_transport_test_encode_message(spdm_context, NULL, false,
false, spdm_response_size,
spdm_response, response_size,
response);
}
return LIBSPDM_STATUS_SUCCESS;
default:
return LIBSPDM_STATUS_RECEIVE_FAIL;
}
}

/**
* Test 1: message could not be sent
* Expected Behavior: get a RETURN_DEVICE_ERROR return code
**/
void libspdm_test_requester_get_key_pair_info_case1(void **state)
{
libspdm_return_t status;
libspdm_test_context_t *spdm_test_context;
libspdm_context_t *spdm_context;

uint8_t key_pair_id;
uint8_t total_key_pairs;
uint16_t capabilities;
uint16_t key_usage_capabilities;
uint16_t current_key_usage;
uint32_t asym_algo_capabilities;
uint32_t current_asym_algo;
uint16_t public_key_info_len;
uint8_t assoc_cert_slot_mask;
uint8_t public_key_info[SPDM_MAX_PUBLIC_KEY_INFO_LEN];

key_pair_id = 1;
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_NEGOTIATED;
spdm_context->connection_info.capability.flags |=
SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_GET_KEY_PAIR_INFO_CAP;

public_key_info_len = SPDM_MAX_PUBLIC_KEY_INFO_LEN;

status = libspdm_get_key_pair_info(spdm_context, NULL, key_pair_id, &total_key_pairs,
&capabilities, &key_usage_capabilities, &current_key_usage,
&asym_algo_capabilities, &current_asym_algo,
&public_key_info_len, &assoc_cert_slot_mask,
public_key_info);
assert_int_equal(status, LIBSPDM_STATUS_SEND_FAIL);
}

/**
* Test 2: Successful response to get csr
* Expected Behavior: get a RETURN_SUCCESS return code
**/
void libspdm_test_requester_get_key_pair_info_case2(void **state)
{
libspdm_return_t status;
libspdm_test_context_t *spdm_test_context;
libspdm_context_t *spdm_context;

uint8_t key_pair_id;
uint8_t associated_slot_id;
uint8_t total_key_pairs;
uint16_t capabilities;
uint16_t key_usage_capabilities;
uint16_t current_key_usage;
uint32_t asym_algo_capabilities;
uint32_t current_asym_algo;
uint16_t public_key_info_len;
uint8_t assoc_cert_slot_mask;
uint8_t public_key_info[SPDM_MAX_PUBLIC_KEY_INFO_LEN];

key_pair_id = 1;
associated_slot_id = 1;
spdm_test_context = *state;
spdm_context = spdm_test_context->spdm_context;
spdm_test_context->case_id = 0x2;
spdm_context->connection_info.version = SPDM_MESSAGE_VERSION_13 <<
SPDM_VERSION_NUMBER_SHIFT_BIT;

spdm_context->connection_info.connection_state =
LIBSPDM_CONNECTION_STATE_NEGOTIATED;
spdm_context->connection_info.capability.flags |=
SPDM_GET_CAPABILITIES_RESPONSE_FLAGS_GET_KEY_PAIR_INFO_CAP;

spdm_context->connection_info.peer_key_pair_id[associated_slot_id] = key_pair_id;
public_key_info_len = SPDM_MAX_PUBLIC_KEY_INFO_LEN;

status = libspdm_get_key_pair_info(spdm_context, NULL, key_pair_id, &total_key_pairs,
&capabilities, &key_usage_capabilities, &current_key_usage,
&asym_algo_capabilities, &current_asym_algo,
&public_key_info_len, &assoc_cert_slot_mask,
public_key_info);

assert_int_equal(status, LIBSPDM_STATUS_SUCCESS);
assert_int_equal(capabilities, SPDM_KEY_PAIR_CAP_GEN_KEY_CAP);
assert_int_equal(key_usage_capabilities, SPDM_KEY_USAGE_BIT_MASK_KEY_EX_USE);
assert_int_equal(current_key_usage, SPDM_KEY_USAGE_BIT_MASK_KEY_EX_USE);
assert_int_equal(asym_algo_capabilities, SPDM_KEY_PAIR_ASYM_ALGO_CAP_RSA2048);
assert_int_equal(current_asym_algo, SPDM_KEY_PAIR_ASYM_ALGO_CAP_RSA2048);
}

libspdm_test_context_t m_libspdm_requester_get_key_pair_info_test_context = {
LIBSPDM_TEST_CONTEXT_VERSION,
true,
libspdm_requester_get_key_pair_info_test_send_message,
libspdm_requester_get_key_pair_info_test_receive_message,
};

int libspdm_requester_get_key_pair_info_test_main(void)
{
const struct CMUnitTest spdm_requester_get_key_pair_info_tests[] = {
/* SendRequest failed*/
cmocka_unit_test(libspdm_test_requester_get_key_pair_info_case1),
/* Successful response to get key pair info, key_pair_id is 1*/
cmocka_unit_test(libspdm_test_requester_get_key_pair_info_case2),
};

libspdm_setup_test_context(
&m_libspdm_requester_get_key_pair_info_test_context);

return cmocka_run_group_tests(spdm_requester_get_key_pair_info_tests,
libspdm_unit_test_group_setup,
libspdm_unit_test_group_teardown);
}

#endif /*LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP*/
11 changes: 11 additions & 0 deletions unit_test/test_spdm_requester/test_spdm_requester.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -82,6 +82,11 @@ int libspdm_requester_vendor_cmds_test_main(void);
int libspdm_requester_vendor_cmds_error_test_main(void);
#endif /* LIBSPDM_ENABLE_VENDOR_DEFINED_MESSAGES */

#if LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP
int libspdm_requester_get_key_pair_info_test_main(void);
#endif /* LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP */


int main(void)
{
int return_value = 0;
Expand Down Expand Up @@ -232,6 +237,12 @@ int main(void)
}
#endif /* LIBSPDM_EVENT_RECIPIENT_SUPPORT */

#if LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP
if (libspdm_requester_get_key_pair_info_test_main() != 0) {
return_value = 1;
}
#endif /* LIBSPDM_ENABLE_CAPABILITY_GET_KEY_PAIR_INFO_CAP */

#if LIBSPDM_ENABLE_VENDOR_DEFINED_MESSAGES
if (libspdm_requester_vendor_cmds_test_main() != 0) {
return_value = 1;
Expand Down

0 comments on commit 2c8ab83

Please sign in to comment.