From 814dace37b5f546439df53b51c3a08afabfd10aa Mon Sep 17 00:00:00 2001 From: Sven Verdoolaege Date: Fri, 8 Sep 2017 12:36:52 +0200 Subject: [PATCH] isl_mat_drop_cols: check for valid column range Signed-off-by: Sven Verdoolaege --- isl_mat.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/isl_mat.c b/isl_mat.c index b0ea0e0c..d4445762 100644 --- a/isl_mat.c +++ b/isl_mat.c @@ -1401,8 +1401,8 @@ struct isl_mat *isl_mat_drop_cols(struct isl_mat *mat, unsigned col, unsigned n) return mat; mat = isl_mat_cow(mat); - if (!mat) - return NULL; + if (check_col_range(mat, col, n) < 0) + return isl_mat_free(mat); if (col != mat->n_col-n) { for (r = 0; r < mat->n_row; ++r) -- 2.11.4.GIT