constant traffic generator

1 files changed, 11 insertions, 0 deletions

diff --git a/src/ats-tests/ats-testing.h b/src/ats-tests/ats-testing.h
index 59a9145d7..22c3b1419 100644
--- a/src/ats-tests/ats-testing.h
+++ b/src/ats-tests/ats-testing.h
@@ -56,6 +56,14 @@ struct TrafficGenerator;
 
 struct LoggingHandle;
 
+enum TrafficGeneratorType
+{
+  GNUNET_ATS_TEST_LINEAR,
+  GNUNET_ATS_TEST_TG_CONSTANT,
+  GNUNET_ATS_TEST_RANDOM,
+  GNUNET_ATS_TEST_SINUS
+};
+
 
 /**
  * Callback to call when topology setup is completed
@@ -215,6 +223,8 @@ struct TrafficGenerator
   struct TrafficGenerator *prev;
   struct TrafficGenerator *next;
 
+  enum TrafficGeneratorType type;
+
   struct BenchmarkPeer *src;
   struct BenchmarkPartner *dest;
   unsigned int rate;
@@ -529,6 +539,7 @@ GNUNET_ATS_TEST_traffic_handle_pong (struct BenchmarkPartner *p);
 struct TrafficGenerator *
 GNUNET_ATS_TEST_generate_traffic_start (struct BenchmarkPeer *src,
     struct BenchmarkPartner *dest,
+    enum TrafficGeneratorType type,
     unsigned int rate,
     struct GNUNET_TIME_Relative duration);