Switch to side-by-side view

--- a
+++ b/src/policies/DIF/RMT/MaxQueue/ReadRateReducer/ReadRateReducer.cc
@@ -0,0 +1,64 @@
+//
+// 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
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public License
+// along with this program.  If not, see http://www.gnu.org/licenses/.
+//
+
+#include <ReadRateReducer.h>
+
+Define_Module(ReadRateReducer);
+
+void ReadRateReducer::onPolicyInit()
+{
+    fwd = check_and_cast<IntPDUForwarding*>
+        (getModuleByPath("^.pduForwardingPolicy"));
+}
+
+bool ReadRateReducer::run(RMTQueue* queue)
+{
+    if (queue->getLength() >= queue->getMaxLength())
+    {
+        if (queue->getType() == RMTQueue::OUTPUT)
+        {
+            const PDU* pdu = dynamic_cast<const PDU*>(queue->getLastPDU());
+
+            // this isn't optimal, but there doesn't seem to be a better way to
+            // get output->input port mapping in connection-less forwarding
+            std::vector<RMTPort*> inputPorts = fwd->lookup(
+                    pdu->getSrcAddr(),
+                    pdu->getConnId().getQoSId());
+
+            if (!inputPorts.front()->hasBlockedInput())
+            { // block read from the input port
+                inputPorts.front()->blockInput();
+            }
+
+        }
+    }
+
+    return false;
+}
+
+void ReadRateReducer::onQueueLengthDrop(RMTQueue* queue)
+{
+    const PDU* pdu = dynamic_cast<const PDU*>(queue->getLastPDU());
+    std::vector<RMTPort*> inputPorts = fwd->lookup(
+                        pdu->getSrcAddr(),
+                        pdu->getConnId().getQoSId());
+
+    if (inputPorts.front()->hasBlockedInput())
+    { // the output buffers are keeping up again, continue receiving on input
+        inputPorts.front()->unblockInput();
+    }
+}