fixed potential bug

mochi-hpc · Jan 15, 2025 · 9c970c1 · 9c970c1
1 parent e64b13b
commit 9c970c1
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/MofkaConsumer.cpp b/src/MofkaConsumer.cpp
@@ -187,12 +187,16 @@ void MofkaConsumer::recvBatch(const tl::request& req,
                         batch->m_meta_sizes[i]}};
                 serializer.deserialize(metadata_archive, metadata);
                 // deserialize the data descriptors
-                BufferWrapperInputArchive descriptors_archive{
-                    std::string_view{
-                        batch->m_data_desc_buffer.data() + data_desc_offset,
-                        batch->m_data_desc_sizes[i]}};
                 DataDescriptor descriptor;
-                descriptor.load(descriptors_archive);
+                if(batch->m_data_desc_sizes[i] > 0) {
+                    BufferWrapperInputArchive descriptors_archive{
+                        std::string_view{
+                            batch->m_data_desc_buffer.data() + data_desc_offset,
+                                batch->m_data_desc_sizes[i]}};
+                    descriptor.load(descriptors_archive);
+                } else {
+                    descriptor = DataDescriptor::Null();
+                }
                 // request Data associated with the event
                 auto data = requestData(
                     partition, metadata,