Fix memory leak introduced in bb6814a

skiz · Apr 2, 2019 · 0338e55 · 0338e55
1 parent 25c1363
commit 0338e55
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/modules/mono/glue/arguments_vector.h b/modules/mono/glue/arguments_vector.h
@@ -39,6 +39,7 @@ struct ArgumentsVector {
 private:
 	T pool[POOL_SIZE];
 	T *_ptr;
+	int size;
 
 	ArgumentsVector();
 	ArgumentsVector(const ArgumentsVector &);
@@ -48,11 +49,18 @@ struct ArgumentsVector {
 	T &get(int p_idx) { return _ptr[p_idx]; }
 	void set(int p_idx, const T &p_value) { _ptr[p_idx] = p_value; }
 
-	explicit ArgumentsVector(int size) {
-		if (size <= POOL_SIZE) {
+	explicit ArgumentsVector(int p_size) :
+			size(p_size) {
+		if (p_size <= POOL_SIZE) {
 			_ptr = pool;
 		} else {
-			_ptr = memnew_arr(T, size);
+			_ptr = memnew_arr(T, p_size);
+		}
+	}
+
+	~ArgumentsVector() {
+		if (size > POOL_SIZE) {
+			memdelete_arr(_ptr);
 		}
 	}
 };