Add RED protocole for ns3.

[simgrid.git] / src / surf / ns3 / ns3_simulator.h

diff --git a/src/surf/ns3/ns3_simulator.h b/src/surf/ns3/ns3_simulator.h
index 4f91f1a..503a7d5 100644 (file)
--- a/src/surf/ns3/ns3_simulator.h
+++ b/src/surf/ns3/ns3_simulator.h
@@ -10,6 +10,9 @@
 #ifdef __cplusplus
 
 #include "ns3/core-module.h"
+#ifdef _HAVE_NS3_RED
+#include "my-point-to-point-helper.h"
+#endif
 
 #ifdef _NS3_3_10
        /*NS3 3.10*/
@@ -29,6 +32,7 @@
        #include "ns3/point-to-point-helper.h"
        #include "ns3/packet-sink-helper.h"
        #include "ns3/inet-socket-address.h"
+       #include "ns3/tcp-socket-factory.h"
 #endif
 
 using namespace ns3;