Merge pull request #17 from speedb-io/15-fault_injection_fs-avoid-div…

…ision-by-zero-in-droprandomunsynceddata #15 SPDB-720: fault_injection_fs: avoid division by zero in DropRandomUns…
speedb-io · Jun 23, 2022 · 7a5089d · 7a5089d
2 parents 1525822 + e225fd1
commit 7a5089d
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/utilities/fault_injection_fs.cc b/utilities/fault_injection_fs.cc
@@ -81,9 +81,11 @@ IOStatus FSFileState::DropUnsyncedData() {
 }
 
 IOStatus FSFileState::DropRandomUnsyncedData(Random* rand) {
-  int range = static_cast<int>(buffer_.size());
-  size_t truncated_size = static_cast<size_t>(rand->Uniform(range));
-  buffer_.resize(truncated_size);
+  const int range = static_cast<int>(buffer_.size());
+  if (range > 0) {
+    size_t truncated_size = static_cast<size_t>(rand->Uniform(range));
+    buffer_.resize(truncated_size);
+  }
   return IOStatus::OK();
 }