Do some customized settings to cover more code logical branch

Signed-off-by: Xiaohanjlll <[email protected]>
Wenxing-hou · Mar 21, 2024 · 2586f39 · 2586f39
1 parent e8a76ab
commit 2586f39
Show file tree

Hide file tree

Showing 15 changed files with 117 additions and 24 deletions.
diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_algorithms/algorithms.c b/unit_test/fuzzing/test_responder/test_spdm_responder_algorithms/algorithms.c
@@ -454,8 +454,15 @@ libspdm_test_context_t libspdm_test_responder_context = {
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&libspdm_test_responder_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_NEGOTIATE_ALGORITHMS) {
+        spdm_request_header->request_response_code = SPDM_NEGOTIATE_ALGORITHMS;
+    }
+
     libspdm_test_responder_context.test_buffer = test_buffer;
     libspdm_test_responder_context.test_buffer_size = test_buffer_size;
 

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_capabilities/capabilities.c b/unit_test/fuzzing/test_responder/test_spdm_responder_capabilities/capabilities.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -83,8 +83,15 @@ libspdm_test_context_t libspdm_test_responder_context = {
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&libspdm_test_responder_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_GET_CAPABILITIES) {
+        spdm_request_header->request_response_code = SPDM_GET_CAPABILITIES;
+    }
+
     libspdm_test_responder_context.test_buffer = test_buffer;
     libspdm_test_responder_context.test_buffer_size = test_buffer_size;
 

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_certificate/certificate.c b/unit_test/fuzzing/test_responder/test_spdm_responder_certificate/certificate.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -188,9 +188,15 @@ void libspdm_test_responder_certificate_case5(void **State)
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_certificate_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_GET_CERTIFICATE) {
+        spdm_request_header->request_response_code = SPDM_GET_CERTIFICATE;
+    }
+
     m_libspdm_responder_certificate_test_context.test_buffer = test_buffer;
     m_libspdm_responder_certificate_test_context.test_buffer_size =
         test_buffer_size;

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_challenge_auth/challenge_auth.c b/unit_test/fuzzing/test_responder/test_spdm_responder_challenge_auth/challenge_auth.c
@@ -358,9 +358,15 @@ void libspdm_test_responder_challenge_case8(void **State)
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_challenge_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_CHALLENGE) {
+        spdm_request_header->request_response_code = SPDM_CHALLENGE;
+    }
+
     m_libspdm_responder_challenge_test_context.test_buffer = test_buffer;
     m_libspdm_responder_challenge_test_context.test_buffer_size = test_buffer_size;
 

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_csr/csr.c b/unit_test/fuzzing/test_responder/test_spdm_responder_csr/csr.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -94,9 +94,15 @@ libspdm_test_context_t m_libspdm_responder_csr_test_context = {
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_csr_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_GET_CSR) {
+        spdm_request_header->request_response_code = SPDM_GET_CSR;
+    }
+
     m_libspdm_responder_csr_test_context.test_buffer = test_buffer;
     m_libspdm_responder_csr_test_context.test_buffer_size =
         test_buffer_size;

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_digests/digests.c b/unit_test/fuzzing/test_responder/test_spdm_responder_digests/digests.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -133,9 +133,15 @@ libspdm_test_context_t m_libspdm_responder_digests_test_context = {
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_digests_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_GET_DIGESTS) {
+        spdm_request_header->request_response_code = SPDM_GET_DIGESTS;
+    }
+
     m_libspdm_responder_digests_test_context.test_buffer = test_buffer;
     m_libspdm_responder_digests_test_context.test_buffer_size =
         test_buffer_size;

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_finish_rsp/finish_rsp.c b/unit_test/fuzzing/test_responder/test_spdm_responder_finish_rsp/finish_rsp.c
@@ -506,9 +506,15 @@ void libspdm_test_responder_finish_case8(void **State)
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_finish_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_FINISH) {
+        spdm_request_header->request_response_code = SPDM_FINISH;
+    }
+
     m_libspdm_responder_finish_test_context.test_buffer = (void *)test_buffer;
     m_libspdm_responder_finish_test_context.test_buffer_size = test_buffer_size;
 

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_heartbeat_ack/heartbeat_ack.c b/unit_test/fuzzing/test_responder/test_spdm_responder_heartbeat_ack/heartbeat_ack.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -206,9 +206,15 @@ void libspdm_test_responder_heartbeat_case4(void **State)
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_heartbeat_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_HEARTBEAT) {
+        spdm_request_header->request_response_code = SPDM_HEARTBEAT;
+    }
+
     m_libspdm_responder_heartbeat_test_context.test_buffer = (void *)test_buffer;
     m_libspdm_responder_heartbeat_test_context.test_buffer_size =
         test_buffer_size;

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_key_exchange/key_exchange.c b/unit_test/fuzzing/test_responder/test_spdm_responder_key_exchange/key_exchange.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -614,9 +614,15 @@ void libspdm_test_responder_key_exchange_case8(void **State)
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_key_exchange_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_KEY_EXCHANGE) {
+        spdm_request_header->request_response_code = SPDM_KEY_EXCHANGE;
+    }
+
     m_libspdm_responder_key_exchange_test_context.test_buffer = test_buffer;
     m_libspdm_responder_key_exchange_test_context.test_buffer_size = test_buffer_size;
 

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_key_update/key_update.c b/unit_test/fuzzing/test_responder/test_spdm_responder_key_update/key_update.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -158,9 +158,15 @@ void libspdm_test_responder_key_update(void **State)
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_key_update_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_KEY_UPDATE) {
+        spdm_request_header->request_response_code = SPDM_KEY_UPDATE;
+    }
+
     m_libspdm_responder_key_update_test_context.test_buffer = test_buffer;
     m_libspdm_responder_key_update_test_context.test_buffer_size =
         test_buffer_size;

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_measurements/measurements.c b/unit_test/fuzzing/test_responder/test_spdm_responder_measurements/measurements.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -244,9 +244,15 @@ libspdm_test_context_t m_libspdm_responder_measurements_test_context = {
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_measurements_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_GET_MEASUREMENTS) {
+        spdm_request_header->request_response_code = SPDM_GET_MEASUREMENTS;
+    }
+
     m_libspdm_responder_measurements_test_context.test_buffer = test_buffer;
     m_libspdm_responder_measurements_test_context.test_buffer_size = test_buffer_size;
 

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_psk_exchange_rsp/psk_exchange_rsp.c b/unit_test/fuzzing/test_responder/test_spdm_responder_psk_exchange_rsp/psk_exchange_rsp.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -546,9 +546,15 @@ void libspdm_test_responder_psk_exchange_case7(void **State)
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_psk_exchange_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_PSK_EXCHANGE) {
+        spdm_request_header->request_response_code = SPDM_PSK_EXCHANGE;
+    }
+
     m_libspdm_responder_psk_exchange_test_context.test_buffer = (void *)test_buffer;
     m_libspdm_responder_psk_exchange_test_context.test_buffer_size = test_buffer_size;
 

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_psk_finish_rsp/psk_finish_rsp.c b/unit_test/fuzzing/test_responder/test_spdm_responder_psk_finish_rsp/psk_finish_rsp.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -281,9 +281,15 @@ void libspdm_test_responder_psk_finish_rsp_case3(void **State)
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_psk_finish_rsp_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_PSK_FINISH) {
+        spdm_request_header->request_response_code = SPDM_PSK_FINISH;
+    }
+
     m_libspdm_responder_psk_finish_rsp_test_context.test_buffer = test_buffer;
     m_libspdm_responder_psk_finish_rsp_test_context.test_buffer_size = test_buffer_size;
 

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_set_certificate/set_certificate_rsp.c b/unit_test/fuzzing/test_responder/test_spdm_responder_set_certificate/set_certificate_rsp.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -97,8 +97,15 @@ void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
     void *State;
     size_t buffer_size;
 
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_set_certificate_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_SET_CERTIFICATE) {
+        spdm_request_header->request_response_code = SPDM_SET_CERTIFICATE;
+    }
+
     buffer_size = test_buffer_size;
     if(buffer_size > LIBSPDM_MAX_CERT_CHAIN_SIZE) {
         buffer_size = LIBSPDM_MAX_CERT_CHAIN_SIZE;

diff --git a/unit_test/fuzzing/test_responder/test_spdm_responder_version/version.c b/unit_test/fuzzing/test_responder/test_spdm_responder_version/version.c
@@ -1,6 +1,6 @@
 /**
  *  Copyright Notice:
- *  Copyright 2021-2022 DMTF. All rights reserved.
+ *  Copyright 2021-2024 DMTF. All rights reserved.
  *  License: BSD 3-Clause License. For full text see link: https://github.com/DMTF/libspdm/blob/main/LICENSE.md
  **/
 
@@ -38,9 +38,15 @@ libspdm_test_context_t m_libspdm_responder_version_test_context = {
 void libspdm_run_test_harness(void *test_buffer, size_t test_buffer_size)
 {
     void *State;
-
+    spdm_message_header_t *spdm_request_header;
     libspdm_setup_test_context(&m_libspdm_responder_version_test_context);
 
+    spdm_request_header = (spdm_message_header_t*)test_buffer;
+
+    if (spdm_request_header->request_response_code != SPDM_GET_VERSION) {
+        spdm_request_header->request_response_code = SPDM_GET_VERSION;
+    }
+
     m_libspdm_responder_version_test_context.test_buffer = test_buffer;
     m_libspdm_responder_version_test_context.test_buffer_size =
         test_buffer_size;