fixed memory leak

valvor · Jul 5, 2014 · 73ef5e2 · 73ef5e2
1 parent 49035ef
commit 73ef5e2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/TunnelEndpoint.cpp b/TunnelEndpoint.cpp
@@ -9,6 +9,12 @@ namespace i2p
 {
 namespace tunnel
 {
+	TunnelEndpoint::~TunnelEndpoint ()
+	{
+		for (auto it: m_IncompleteMessages)
+			i2p::DeleteI2NPMessage (it.second.data);
+	}	
+
 	void TunnelEndpoint::HandleDecryptedTunnelDataMsg (I2NPMessage * msg)
 	{
 		m_NumReceivedBytes += TUNNEL_DATA_MSG_SIZE;

diff --git a/TunnelEndpoint.h b/TunnelEndpoint.h
@@ -21,6 +21,7 @@ namespace tunnel
 		public:
 
 			TunnelEndpoint (): m_NumReceivedBytes (0) {};
+			~TunnelEndpoint ();
 			size_t GetNumReceivedBytes () const { return m_NumReceivedBytes; };
 
 			void HandleDecryptedTunnelDataMsg (I2NPMessage * msg);