shithub: aubio

Info  •  Files  •  Log  •  Branches

src/pitch/pitchyinfft.c: return NULL if fft creation failed

--- a/src/pitch/pitchyinfft.c

+++ b/src/pitch/pitchyinfft.c

@@ -62,6 +62,7 @@

   aubio_pitchyinfft_t *p = AUBIO_NEW (aubio_pitchyinfft_t);

   p->winput = new_fvec (bufsize);

   p->fft = new_aubio_fft (bufsize);

+  if (!p->fft) goto beach;

   p->fftout = new_fvec (bufsize);

   p->sqrmag = new_fvec (bufsize);

   p->yinfft = new_fvec (bufsize / 2 + 1);

@@ -95,6 +96,11 @@

   // check for octave errors above 1300 Hz

   p->short_period = (uint_t)ROUND(samplerate / 1300.);

   return p;

+

+beach:

+  if (p->winput) del_fvec(p->winput);

+  AUBIO_FREE(p);

+  return NULL;

 }

 void