safety checks

1 files changed, 11 insertions, 2 deletions

diff --git a/src/ats-tests/ats-testing-traffic.c b/src/ats-tests/ats-testing-traffic.c
index 1bd31cc26..0938f79f4 100644
--- a/src/ats-tests/ats-testing-traffic.c
+++ b/src/ats-tests/ats-testing-traffic.c
@@ -54,9 +54,13 @@ get_delay (struct TrafficGenerator *tg)
       time_delta = GNUNET_TIME_absolute_get_duration(tg->time_start);
       /* Calculate point of time in the current period */
       time_delta.rel_value_us = time_delta.rel_value_us % tg->duration_period.rel_value_us;
-
       delta_rate = ((double) time_delta.rel_value_us  / tg->duration_period.rel_value_us) *
           (tg->max_rate - tg->base_rate);
+      if ((tg->max_rate - tg->base_rate) > tg->base_rate)
+      {
+        /* This will cause an underflow */
+        GNUNET_break (0);
+      }
       cur_rate = tg->base_rate + delta_rate;
       break;
     case GNUNET_ATS_TEST_TG_RANDOM:
@@ -67,7 +71,7 @@ get_delay (struct TrafficGenerator *tg)
       time_delta = GNUNET_TIME_absolute_get_duration(tg->time_start);
       /* Calculate point of time in the current period */
       time_delta.rel_value_us = time_delta.rel_value_us % tg->duration_period.rel_value_us;
-      if ((tg->max_rate - tg->base_rate) < tg->base_rate)
+      if ((tg->max_rate - tg->base_rate) > tg->base_rate)
       {
         /* This will cause an underflow for second half of sinus period,
          * will be detected in general when experiments are loaded */
@@ -82,6 +86,11 @@ get_delay (struct TrafficGenerator *tg)
       break;
   }
 
+  if (cur_rate < 0)
+  {
+    cur_rate = 0;
+  }
+
   /* Calculate the delay for the next message based on the current delay  */
   delay.rel_value_us =  GNUNET_TIME_UNIT_SECONDS.rel_value_us * TEST_MESSAGE_SIZE / cur_rate;