Add check for KD tree size to BLI_kdtree_insert()

Makes sure we're not trying to insert more nodes than we've
been expecting to.

1 files changed, 6 insertions, 0 deletions

diff --git a/source/blender/blenlib/intern/BLI_kdtree.c b/source/blender/blenlib/intern/BLI_kdtree.c
index c0e7c21420a..844d8502ff0 100644
--- a/source/blender/blenlib/intern/BLI_kdtree.c
+++ b/source/blender/blenlib/intern/BLI_kdtree.c
@@ -46,6 +46,7 @@ struct KDTree {
 	KDTreeNode *root;
 #ifdef DEBUG
 	bool is_balanced;  /* ensure we call balance first */
+	unsigned int maxsize;   /* max size of the tree */
 #endif
 };
 
@@ -67,6 +68,7 @@ KDTree *BLI_kdtree_new(unsigned int maxsize)
 
 #ifdef DEBUG
 	tree->is_balanced = false;
+	tree->maxsize = maxsize;
 #endif
 
 	return tree;
@@ -87,6 +89,10 @@ void BLI_kdtree_insert(KDTree *tree, int index, const float co[3], const float n
 {
 	KDTreeNode *node = &tree->nodes[tree->totnode++];
 
+#ifdef DEBUG
+	BLI_assert(tree->totnode < tree->maxsize);
+#endif
+
 	/* note, array isn't calloc'd,
 	 * need to initialize all struct members */