#2302: Add more buffer boundary check

DARMA-tasking · Dec 12, 2024 · e057fb8 · e057fb8
1 parent a971233
commit e057fb8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/vt/metrics/perf_data.cc b/src/vt/metrics/perf_data.cc
@@ -146,6 +146,11 @@ std::unordered_map<std::string, uint64_t> PerfData::getTaskMeasurements() {
     uint64_t count = 0;
 
     if (event_fds_[i] != -1) {
+      if (sizeof(count) != sizeof(uint64_t)) {
+          vtAbort("Buffer size mismatch: expected " + std::to_string(sizeof(uint64_t)) +
+                  " bytes, but got " + std::to_string(sizeof(count)));
+      }
+
       ssize_t bytesRead = read(event_fds_[i], &count, sizeof(count));
 
       if (bytesRead == sizeof(count)) {