Avoid possible double-freeing memory

shogun-toolbox · Aug 17, 2012 · 6f08f28 · 6f08f28
1 parent d0ac385
commit 6f08f28
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/shogun/structure/libp3bm.cpp b/src/shogun/structure/libp3bm.cpp
@@ -746,6 +746,7 @@ bmrm_return_value_T svm_p3bm_solver(
 		cp_ptr2=cp_ptr;
 		cp_ptr=cp_ptr->next;
 		LIBBMRM_FREE(cp_ptr2);
+		cp_ptr2 = NULL;
 	}
 
 cleanup:
@@ -773,9 +774,11 @@ bmrm_return_value_T svm_p3bm_solver(
 	LIBBMRM_FREE(H2);
 	LIBBMRM_FREE(C);
 	LIBBMRM_FREE(S);
-	LIBBMRM_FREE(cp_list);
 	LIBBMRM_FREE(Rt);
 
+	if (cp_list)
+		LIBBMRM_FREE(cp_list);
+
 	for (uint32_t p=0; p<cp_models; ++p)
 	{
 		LIBBMRM_FREE(subgrad_t[p]);