pfrd: fix mod by zero error when all branches are filtered

trapexit · Jun 25, 2021 · c06db9c · c06db9c
1 parent bcd5fde
commit c06db9c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/src/policy_eppfrd.cpp b/src/policy_eppfrd.cpp
@@ -204,6 +204,9 @@ namespace eppfrd
     uint64_t idx;
     uint64_t threshold;
 
+    if(sum_ == 0)
+      return NULL;
+
     idx = 0;
     threshold = RND::rand64(sum_);
     for(size_t i = 0; i < branchinfo_.size(); i++)

diff --git a/src/policy_msppfrd.cpp b/src/policy_msppfrd.cpp
@@ -130,6 +130,9 @@ namespace msppfrd
     uint64_t idx;
     uint64_t threshold;
 
+    if(sum_ == 0)
+      return NULL;
+
     idx = 0;
     threshold = RND::rand64(sum_);
     for(size_t i = 0; i < branchinfo_.size(); i++)

diff --git a/src/policy_pfrd.cpp b/src/policy_pfrd.cpp
@@ -98,6 +98,9 @@ namespace pfrd
     uint64_t idx;
     uint64_t threshold;
 
+    if(sum_ == 0)
+      return NULL;
+
     idx = 0;
     threshold = RND::rand64(sum_);
     for(size_t i = 0; i < branchinfo_.size(); i++)