From f85ac0c4bbc4dfa6bcf05a1a2d8b047bbd20bb9f Mon Sep 17 00:00:00 2001 From: Damien Caliste Date: Fri, 2 Feb 2018 09:51:58 +0100 Subject: [PATCH] Add assertion on index when accessing array elements. --- src/pspio_pspdata.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/pspio_pspdata.c b/src/pspio_pspdata.c index 5253fdc..9c29d58 100644 --- a/src/pspio_pspdata.c +++ b/src/pspio_pspdata.c @@ -632,6 +632,7 @@ int pspio_pspdata_get_n_states(const pspio_pspdata_t *pspdata) const pspio_state_t * pspio_pspdata_get_state(const pspio_pspdata_t *pspdata, int index) { assert(pspdata != NULL); + assert(index >= 0 && index < pspdata->n_states); return pspdata->states[index]; } @@ -653,6 +654,7 @@ int pspio_pspdata_get_n_potentials(const pspio_pspdata_t *pspdata) const pspio_potential_t * pspio_pspdata_get_potential(const pspio_pspdata_t *pspdata, int index) { assert(pspdata != NULL); + assert(index >= 0 && index < pspdata->n_potentials); return pspdata->potentials[index]; } @@ -674,6 +676,7 @@ int * pspio_pspdata_get_n_projectors_per_l(const pspio_pspdata_t *pspdata) const pspio_projector_t * pspio_pspdata_get_projector(const pspio_pspdata_t *pspdata, int index) { assert(pspdata != NULL); + assert(index >= 0 && index < pspdata->n_projectors); return pspdata->projectors[index]; } -- GitLab