shithub: aubio

Info  •  Files  •  Log  •  Branches

python/ext/py-fft.c: check input has correct size

--- a/python/ext/py-fft.c

+++ b/python/ext/py-fft.c

@@ -85,6 +85,13 @@

     return NULL;

   }

+  if (self->vecin.length != self->win_s) {

+    PyErr_Format(PyExc_ValueError,

+                 "input array has length %d, but fft has size %d",

+                 self->vecin.length, self->win_s);

+    return NULL;

+  }

+

   cvec_t c_out;

   Py_INCREF(self->doout);

   if (!PyAubio_PyCvecToCCvec(self->doout, &c_out)) {