From 0d7e7a54f60fa229b96e149734abae6afd873fbf Mon Sep 17 00:00:00 2001
From: "arseny.kapoulkine"
 <arseny.kapoulkine@99668b35-9821-0410-8761-19e4c4f06640>
Date: Fri, 8 Dec 2006 14:38:34 +0000
Subject: Fixed memory leak, correct behavior for several parse() on one parser

git-svn-id: http://pugixml.googlecode.com/svn/trunk@15 99668b35-9821-0410-8761-19e4c4f06640
---
 src/pugixml.cpp | 18 +++++++++++++++++-
 src/pugixml.hpp |  2 ++
 2 files changed, 19 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/pugixml.cpp b/src/pugixml.cpp
index b77a21d..d0d581b 100644
--- a/src/pugixml.cpp
+++ b/src/pugixml.cpp
@@ -1870,6 +1870,14 @@ namespace pugi
 
 	xml_parser::~xml_parser()
 	{
+		free();
+	}
+
+	void xml_parser::free()
+	{
+		delete _buffer;
+		_buffer = 0;
+
 		xml_memory_block* current = _memory.next;
 
 		while (current)
@@ -1878,6 +1886,9 @@ namespace pugi
 			delete current;
 			current = next;
 		}
+		
+		_memory.next = 0;
+		_memory.size = 0;
 	}
 
 	xml_parser::operator xml_node() const
@@ -1905,6 +1916,8 @@ namespace pugi
 #ifndef PUGIXML_NO_STL
 	void xml_parser::parse(std::istream& stream,unsigned int optmsk)
 	{
+		free();
+
 		int length = 0, pos = stream.tellg();
 		stream.seekg(0, std::ios_base::end);
 		length = stream.tellg();
@@ -1920,6 +1933,8 @@ namespace pugi
 
 	char* xml_parser::parse(char* xmlstr,unsigned int optmsk)
 	{
+		free();
+
 		if(!xmlstr) return 0;
 
 		xml_allocator alloc(&_memory);
@@ -1935,9 +1950,10 @@ namespace pugi
 	
 	char* xml_parser::parse(const transfer_ownership_tag&, char* xmlstr,unsigned int optmsk)
 	{
+		free();
+
 		if(!xmlstr) return 0;
 
-		delete[] _buffer;
 		_buffer = xmlstr;
 
 		xml_allocator alloc(&_memory);
diff --git a/src/pugixml.hpp b/src/pugixml.hpp
index 739caaa..42a7950 100644
--- a/src/pugixml.hpp
+++ b/src/pugixml.hpp
@@ -507,6 +507,8 @@ namespace pugi
 		xml_parser(const xml_parser&);
 		const xml_parser& operator=(const xml_parser&);
 
+		void free();	///< free memory
+
 	public:
 		/// Constructor.
 		/// \param optmsk - Options mask.
-- 
cgit v1.2.3