diff options
Diffstat (limited to 'lib/band_width.c')
| -rw-r--r-- | lib/band_width.c | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/lib/band_width.c b/lib/band_width.c index 822d835..4a4d6cb 100644 --- a/lib/band_width.c +++ b/lib/band_width.c @@ -21,7 +21,7 @@ ts_parameter_t const ts_band_width_p = { .n_inputs = 1, .n_outputs = 1, .n_const uint_t ts_band_width_offset( ts_constant_t const *c ) { ( void )c; - return ( c ? ((ts_band_width_params_t *)c)->k - 1 : 0 ); + return ( c ? ((ts_band_width_params_t const *)c)->k > 0 ? ((ts_band_width_params_t const *)c)->k - 1 : 0 : 0 ); } int ts_band_width( @@ -32,9 +32,11 @@ int ts_band_width( { TS_PROLOGUE( ts_band_width_p ); if ( !c ) return -1; - uint_t k = ((ts_band_width_params_t *)c)->k; - fp_t std_mult = ((ts_band_width_params_t *)c)->std_mult; + TS_CHECK_INPUT(x, x_n, n); + uint_t k = ((ts_band_width_params_t const *)c)->k; + fp_t std_mult = ((ts_band_width_params_t const *)c)->std_mult; if (k == 0 || n < k) return -1; + if (std_mult < 0 || isnan(std_mult) || isinf(std_mult)) return -1; for (uint_t i = k - 1; i < n; ++i) { |