Added memory leak detection to lzmadec.c.

1 files changed, 3 insertions, 0 deletions

diff --git a/src/lzmadec/lzmadec.c b/src/lzmadec/lzmadec.c
index f0ae3568..10d5cc60 100644
--- a/src/lzmadec/lzmadec.c
+++ b/src/lzmadec/lzmadec.c
@@ -484,6 +484,8 @@ main(int argc, char **argv)
 		exit(ERROR);
 	}
 
+	assert(lzma_memlimit_count(mem_limitter) == 0);
+
 	allocator.opaque = mem_limitter;
 	strm.allocator = &allocator;
 
@@ -517,6 +519,7 @@ main(int argc, char **argv)
 	// Free the memory only when debugging. Freeing wastes some time,
 	// but allows detecting possible memory leaks with Valgrind.
 	lzma_end(&strm);
+	assert(lzma_memlimit_count(mem_limitter) == 0);
 	lzma_memlimit_end(mem_limitter, false);
 #endif