define MAX_SPDM_MSG_SIZE in test, after it is removed in libspdm.

Signed-off-by: Jiewen Yao <[email protected]>
DMTF · Apr 12, 2023 · 3a31939 · 3a31939
1 parent a62ddd9
commit 3a31939
Show file tree

Hide file tree

Showing 6 changed files with 34 additions and 2 deletions.
diff --git a/unit_test/fuzzing/spdm_unit_fuzzing_common/common.c b/unit_test/fuzzing/spdm_unit_fuzzing_common/common.c
@@ -75,7 +75,7 @@ size_t libspdm_unit_test_group_setup(void **State)
 
     libspdm_init_context(spdm_context);
     spdm_test_context->scratch_buffer_size =
-        libspdm_get_sizeof_required_scratch_buffer(spdm_context);
+        libspdm_get_sizeof_required_scratch_buffer(spdm_context, LIBSPDM_MAX_SPDM_MSG_SIZE);
     spdm_test_context->scratch_buffer = (void *)malloc(spdm_test_context->scratch_buffer_size);
 
     libspdm_register_device_io_func(spdm_context,

diff --git a/unit_test/fuzzing/spdm_unit_fuzzing_common/spdm_unit_fuzzing.h b/unit_test/fuzzing/spdm_unit_fuzzing_common/spdm_unit_fuzzing.h
@@ -90,6 +90,14 @@ typedef struct {
 #define LIBSPDM_MAX_SENDER_RECEIVER_BUFFER_SIZE LIBSPDM_RECEIVER_BUFFER_SIZE
 #endif
 
+/* Maximum size of a large SPDM message.
+ * If chunk is unsupported, it must be same as DATA_TRANSFER_SIZE.
+ * If chunk is supported, it must be larger than DATA_TRANSFER_SIZE.
+ * It matches MaxSPDMmsgSize in SPDM specification. */
+#ifndef LIBSPDM_MAX_SPDM_MSG_SIZE
+#define LIBSPDM_MAX_SPDM_MSG_SIZE 0x1200
+#endif
+
 extern uint8_t m_libspdm_use_measurement_spec;
 extern uint32_t m_libspdm_use_measurement_hash_algo;
 extern uint32_t m_libspdm_use_hash_algo;

diff --git a/unit_test/spdm_unit_test_common/common.c b/unit_test/spdm_unit_test_common/common.c
@@ -85,7 +85,7 @@ int libspdm_unit_test_group_setup(void **state)
 
     libspdm_init_context(spdm_context);
     spdm_test_context->scratch_buffer_size =
-        libspdm_get_sizeof_required_scratch_buffer(spdm_context);
+        libspdm_get_sizeof_required_scratch_buffer(spdm_context, LIBSPDM_MAX_SPDM_MSG_SIZE);
     spdm_test_context->scratch_buffer = (void *)malloc(spdm_test_context->scratch_buffer_size);
 
     libspdm_register_device_io_func(spdm_context,

diff --git a/unit_test/spdm_unit_test_common/spdm_unit_test.h b/unit_test/spdm_unit_test_common/spdm_unit_test.h
@@ -92,6 +92,14 @@ typedef struct {
 #define LIBSPDM_MAX_SENDER_RECEIVER_BUFFER_SIZE LIBSPDM_RECEIVER_BUFFER_SIZE
 #endif
 
+/* Maximum size of a large SPDM message.
+ * If chunk is unsupported, it must be same as DATA_TRANSFER_SIZE.
+ * If chunk is supported, it must be larger than DATA_TRANSFER_SIZE.
+ * It matches MaxSPDMmsgSize in SPDM specification. */
+#ifndef LIBSPDM_MAX_SPDM_MSG_SIZE
+#define LIBSPDM_MAX_SPDM_MSG_SIZE 0x1200
+#endif
+
 extern uint8_t m_libspdm_use_measurement_spec;
 extern uint32_t m_libspdm_use_measurement_hash_algo;
 extern uint32_t m_libspdm_use_hash_algo;

diff --git a/unit_test/test_size/test_size_of_spdm_requester/spdm_requester_init.c b/unit_test/test_size/test_size_of_spdm_requester/spdm_requester_init.c
@@ -29,6 +29,14 @@
 #define LIBSPDM_MAX_SENDER_RECEIVER_BUFFER_SIZE LIBSPDM_RECEIVER_BUFFER_SIZE
 #endif
 
+/* Maximum size of a large SPDM message.
+ * If chunk is unsupported, it must be same as DATA_TRANSFER_SIZE.
+ * If chunk is supported, it must be larger than DATA_TRANSFER_SIZE.
+ * It matches MaxSPDMmsgSize in SPDM specification. */
+#ifndef LIBSPDM_MAX_SPDM_MSG_SIZE
+#define LIBSPDM_MAX_SPDM_MSG_SIZE 0x1200
+#endif
+
 #if LIBSPDM_ENABLE_CAPABILITY_CHUNK_CAP
 #define LIBSPDM_SCRATCH_BUFFER_SECURE_MESSAGE_CAPACITY (LIBSPDM_MAX_SPDM_MSG_SIZE + \
                                                         LIBSPDM_TRANSPORT_ADDITIONAL_SIZE)

diff --git a/unit_test/test_size/test_size_of_spdm_responder/spdm_responder_init.c b/unit_test/test_size/test_size_of_spdm_responder/spdm_responder_init.c
@@ -29,6 +29,14 @@
 #define LIBSPDM_MAX_SENDER_RECEIVER_BUFFER_SIZE LIBSPDM_RECEIVER_BUFFER_SIZE
 #endif
 
+/* Maximum size of a large SPDM message.
+ * If chunk is unsupported, it must be same as DATA_TRANSFER_SIZE.
+ * If chunk is supported, it must be larger than DATA_TRANSFER_SIZE.
+ * It matches MaxSPDMmsgSize in SPDM specification. */
+#ifndef LIBSPDM_MAX_SPDM_MSG_SIZE
+#define LIBSPDM_MAX_SPDM_MSG_SIZE 0x1200
+#endif
+
 #if LIBSPDM_ENABLE_CAPABILITY_CHUNK_CAP
 #define LIBSPDM_SCRATCH_BUFFER_SECURE_MESSAGE_CAPACITY (LIBSPDM_MAX_SPDM_MSG_SIZE + \
                                                         LIBSPDM_TRANSPORT_ADDITIONAL_SIZE)