--- a/src/policies/DIF/RMT/MaxQueue/RMTMaxQBase.h
+++ b/src/policies/DIF/RMT/MaxQueue/RMTMaxQBase.h
@@ -1,5 +1,5 @@
//
-// Copyright © 2014 PRISTINE Consortium (http://ict-pristine.eu)
+// Copyright © 2014 - 2015 PRISTINE Consortium (http://ict-pristine.eu)
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
@@ -20,8 +20,11 @@
#include <omnetpp.h>
+#include "RINASignals.h"
#include "RMTQueue.h"
#include "RMTQMonitorBase.h"
+#include "AddressComparatorBase.h"
+#include "RMTModuleAllocator.h"
/**
@@ -54,9 +57,54 @@
virtual void handleMessage(cMessage* msg);
/**
+ * Send CDAP congestion notification to the PDU sender via RIBd.
+ *
+ * @param pdu pointer to PDU
+ */
+ void notifySenderOfCongestion(const cPacket* pdu);
+
+ /**
+ * Stops receiving data from PDU sender's arrival port.
+ *
+ * @param pointer to PDU
+ */
+ void disableSenderPortDrain(const cPacket* pdu);
+
+ /**
+ * Starts receiving data from PDU sender's arrival port.
+ *
+ * @param pointer to PDU
+ */
+ void enableSenderPortDrain(const cPacket* pdu);
+
+ /**
+ * Increases the rate of receiving data from PDU sender's arrival port at a faster rate.
+ *
+ * @param pointer to PDU
+ */
+ void slowDownSenderPortDrain(const cPacket* pdu);
+
+ /**
+ * Stops receiving data from PDU sender's arrival port.
+ *
+ * @param pointer to PDU
+ */
+ void speedUpSenderPortDrain(const cPacket* pdu);
+
+ /**
* Pointer to the monitoring policy module.
*/
RMTQMonitorBase* qMonPolicy;
+
+ /**
+ * Pointer to an address comparator module.
+ */
+ AddressComparatorBase* addrComparator;
+
+ /**
+ * Pointer to the RMT allocator module (also providing queue<->port mappings).
+ */
+ RMTModuleAllocator* rmtAllocator;
private:
@@ -65,6 +113,31 @@
* Inherited policies should be using onPolicyInit() instead.
*/
void initialize();
+
+ /**
+ * Congestion notifier.
+ */
+ simsignal_t sigRMTSDReq;
+
+ /**
+ * Signal used for disabling an incoming PDU's arrival port.
+ */
+ simsignal_t sigRMTPortDrainDisable;
+
+ /**
+ * Signal used for enabling an incoming PDU's arrival port.
+ */
+ simsignal_t sigRMTPortDrainEnable;
+
+ /**
+ * Signal used for speeding up an incoming PDU's arrival port drain rate.
+ */
+ simsignal_t sigRMTPortDrainSpeedUp;
+
+ /**
+ * Signal used for slowing down an incoming PDU's arrival port drain rate.
+ */
+ simsignal_t sigRMTPortDrainSlowDown;
};
#endif /* RMTMAXQBASE_H_ */