--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/dsr/model/dsr-errorbuff.cc	Sun May 06 20:52:24 2012 -0700
@@ -0,0 +1,193 @@
+/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
+/*
+ * Copyright (c) 2011 Yufei Cheng
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation;
+ *
+ * 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ *
+ * Author: Yufei Cheng   <yfcheng@ittc.ku.edu>
+ *
+ * James P.G. Sterbenz <jpgs@ittc.ku.edu>, director
+ * ResiliNets Research Group  http://wiki.ittc.ku.edu/resilinets
+ * Information and Telecommunication Technology Center (ITTC)
+ * and Department of Electrical Engineering and Computer Science
+ * The University of Kansas Lawrence, KS USA.
+ *
+ * Work supported in part by NSF FIND (Future Internet Design) Program
+ * under grant CNS-0626918 (Postmodern Internet Architecture),
+ * NSF grant CNS-1050226 (Multilayer Network Resilience Analysis and Experimentation on GENI),
+ * US Department of Defense (DoD), and ITTC at The University of Kansas.
+ */
+
+#include "dsr-errorbuff.h"
+#include <algorithm>
+#include <functional>
+#include "ns3/ipv4-route.h"
+#include "ns3/socket.h"
+#include "ns3/log.h"
+
+NS_LOG_COMPONENT_DEFINE ("DsrErrorBuffer");
+
+namespace ns3 {
+namespace dsr {
+
+uint32_t
+ErrorBuffer::GetSize ()
+{
+  Purge ();
+  return m_errorBuffer.size ();
+}
+
+bool
+ErrorBuffer::Enqueue (ErrorBuffEntry & entry)
+{
+  Purge ();
+  for (std::vector<ErrorBuffEntry>::const_iterator i = m_errorBuffer.begin (); i
+       != m_errorBuffer.end (); ++i)
+    {
+      NS_LOG_INFO ("packet id " << i->GetPacket ()->GetUid () << " " << entry.GetPacket ()->GetUid () << " source " << i->GetSource () << " " << entry.GetSource ()
+                   << " next hop " << i->GetNextHop () << " " << entry.GetNextHop () << " dst " << i->GetDestination () << " " << entry.GetDestination ());
+
+      if ((i->GetPacket ()->GetUid () == entry.GetPacket ()->GetUid ()) && (i->GetSource () == entry.GetSource ()) && (i->GetNextHop () == entry.GetSource ())
+          && (i->GetDestination () == entry.GetDestination ()))
+        {
+          return false;
+        }
+    }
+
+  entry.SetExpireTime (m_errorBufferTimeout);     // Initialize the send buffer timeout
+  /*
+   * Drop the most aged packet when buffer reaches to max
+   */
+  if (m_errorBuffer.size () >= m_maxLen)
+    {
+      Drop (m_errorBuffer.front (), "Drop the most aged packet");         // Drop the most aged packet
+      m_errorBuffer.erase (m_errorBuffer.begin ());
+    }
+  // enqueue the entry
+  m_errorBuffer.push_back (entry);
+  return true;
+}
+
+void
+ErrorBuffer::DropPacketForErrLink (Ipv4Address source, Ipv4Address nextHop)
+{
+  NS_LOG_FUNCTION (this << source << nextHop);
+  Purge ();
+  std::vector<Ipv4Address> list;
+  list.push_back (source);
+  list.push_back (nextHop);
+  const std::vector<Ipv4Address> link = list;
+  /*
+   * Drop the packet with the error link source----------nextHop
+   */
+  for (std::vector<ErrorBuffEntry>::iterator i = m_errorBuffer.begin (); i
+       != m_errorBuffer.end (); ++i)
+    {
+      if (LinkEqual (*i, link))
+        {
+          DropLink (*i, "DropPacketForErrLink");
+        }
+    }
+  m_errorBuffer.erase (std::remove_if (m_errorBuffer.begin (), m_errorBuffer.end (),
+                                      std::bind2nd (std::ptr_fun (ErrorBuffer::LinkEqual), link)), m_errorBuffer.end ());
+}
+
+bool
+ErrorBuffer::Dequeue (Ipv4Address dst, ErrorBuffEntry & entry)
+{
+  Purge ();
+  /*
+   * Dequeue the entry with destination address dst
+   */
+  for (std::vector<ErrorBuffEntry>::iterator i = m_errorBuffer.begin (); i != m_errorBuffer.end (); ++i)
+    {
+      if (i->GetDestination () == dst)
+        {
+          entry = *i;
+          m_errorBuffer.erase (i);
+          NS_LOG_DEBUG ("Packet size while dequeuing " << entry.GetPacket ()->GetSize ());
+          return true;
+        }
+    }
+  return false;
+}
+
+bool
+ErrorBuffer::Find (Ipv4Address dst)
+{
+  /*
+   * Make sure if the send buffer contains entry with certain dst
+   */
+  for (std::vector<ErrorBuffEntry>::const_iterator i = m_errorBuffer.begin (); i
+       != m_errorBuffer.end (); ++i)
+    {
+      if (i->GetDestination () == dst)
+        {
+          NS_LOG_DEBUG ("Found the packet");
+          return true;
+        }
+    }
+  return false;
+}
+
+struct IsExpired
+{
+  bool
+  operator() (ErrorBuffEntry const & e) const
+  {
+    // NS_LOG_DEBUG("Expire time for packet in req queue: "<<e.GetExpireTime ());
+    return (e.GetExpireTime () < Seconds (0));
+  }
+};
+
+void
+ErrorBuffer::Purge ()
+{
+  /*
+   * Purge the buffer to eliminate expired entries
+   */
+  NS_LOG_DEBUG ("The error buffer size " << m_errorBuffer.size ());
+  IsExpired pred;
+  for (std::vector<ErrorBuffEntry>::iterator i = m_errorBuffer.begin (); i
+       != m_errorBuffer.end (); ++i)
+    {
+      if (pred (*i))
+        {
+          NS_LOG_DEBUG ("Dropping Queue Packets");
+          Drop (*i, "Drop out-dated packet ");
+        }
+    }
+  m_errorBuffer.erase (std::remove_if (m_errorBuffer.begin (), m_errorBuffer.end (), pred),
+                      m_errorBuffer.end ());
+}
+
+void
+ErrorBuffer::Drop (ErrorBuffEntry en, std::string reason)
+{
+  NS_LOG_LOGIC (reason << en.GetPacket ()->GetUid () << " " << en.GetDestination ());
+//  en.GetErrorCallback () (en.GetPacket (), en.GetDestination (),
+//     Socket::ERROR_NOROUTETOHOST);
+  return;
+}
+
+void
+ErrorBuffer::DropLink (ErrorBuffEntry en, std::string reason)
+{
+  NS_LOG_LOGIC (reason << en.GetPacket ()->GetUid () << " " << en.GetSource () << " " << en.GetNextHop ());
+//  en.GetErrorCallback () (en.GetPacket (), en.GetDestination (),
+//     Socket::ERROR_NOROUTETOHOST);
+  return;
+}
+}  // namespace dsr
+}  // namespace ns3