fixed memory leak

valvor · May 4, 2014 · 44bba19 · 44bba19
1 parent 4f429b3
commit 44bba19
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/SSU.cpp b/SSU.cpp
@@ -859,6 +859,7 @@ namespace ssu
 				len = 0;
 			fragmentNum++;
 		}	
+		DeleteI2NPMessage (msg);
 	}		
 
 	void SSUSession::Send (uint8_t type, const uint8_t * payload, size_t len)

diff --git a/SSUData.cpp b/SSUData.cpp
@@ -90,6 +90,8 @@ namespace ssu
 
 					if (isLast)
 					{
+						if (!msg)
+							DeleteI2NPMessage (it->second->msg);
 						delete it->second;
 						m_IncomleteMessages.erase (it);
 					}