From 8d95066b58b79f813c91bc124d36af7017f2ffe8 Mon Sep 17 00:00:00 2001
From: Tommaso Pecorella <tommypec@gmail.com>
Date: Fri, 28 Feb 2020 08:11:57 +0000
Subject: [PATCH] Fix early socket close in tcp-large-transfer example

---
 examples/tcp/tcp-large-transfer.cc | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/examples/tcp/tcp-large-transfer.cc b/examples/tcp/tcp-large-transfer.cc
index 27deb300f..ee94f7448 100644
--- a/examples/tcp/tcp-large-transfer.cc
+++ b/examples/tcp/tcp-large-transfer.cc
@@ -212,5 +212,8 @@ void WriteUntilBufferFull (Ptr<Socket> localSocket, uint32_t txSpace)
         }
       currentTxBytes += amountSent;
     }
-  localSocket->Close ();
+  if (currentTxBytes >= totalTxBytes)
+    {
+      localSocket->Close ();
+    }
 }