fix TcpSocketBase::RecvFrom()

2010-12-30 11:16:35 -08:00
parent ef2941417d
commit 2eb811f9f1
1 changed files with 5 additions and 1 deletions
--- a/src/internet-stack/tcp-socket-base.cc
+++ b/src/internet-stack/tcp-socket-base.cc
@@ -412,7 +412,11 @@ TcpSocketBase::RecvFrom (uint32_t maxSize, uint32_t flags, Address &fromAddress)
  // Null packet means no data to read, and an empty packet indicates EOF
  if (packet != 0 && packet->GetSize () != 0)
    {
-      GetSockName (fromAddress);
+      SocketAddressTag tag;
+      bool found;
+      found = packet->PeekPacketTag (tag);
+      NS_ASSERT (found);
+      fromAddress = tag.GetAddress ();
    }
  return packet;
 }