(svn r7237) -Fix: TGP landscape generation could leak memory if aborted during the generation of the heightmap.

1 files changed, 3 insertions, 0 deletions

diff --git a/tgp.c b/tgp.c
index 8c99856f6..d7f9722f9 100644
--- a/tgp.c
+++ b/tgp.c
@@ -798,6 +798,8 @@ void GenerateTerrainPerlin(void)
 	uint x, y;
 
 	if (!AllocHeightMap()) return;
+	GenerateWorldSetAbortCallback(FreeHeightMap);
+
 	HeightMapGenerate();
 
 	IncreaseGeneratingWorldProgress(GWP_LANDSCAPE);
@@ -823,4 +825,5 @@ void GenerateTerrainPerlin(void)
 	for (x = 0; x < _height_map.size_x;     x++) MakeVoid(_height_map.size_x * y + x);
 
 	FreeHeightMap();
+	GenerateWorldSetAbortCallback(NULL);
 }