iLBC: Handle a case of bad input data

webrtc/modules/audio_coding/codecs/ilbc/cb_construct.c

 /*
  *  Copyright (c) 2011 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
 /******************************************************************
 
  iLBC Speech Coder ANSI-C Source Code
 
  WebRtcIlbcfix_CbConstruct.c
 
 ******************************************************************/
 
+#include "cb_construct.h"
+
 #include "defines.h"
 #include "gain_dequant.h"
 #include "get_cd_vec.h"
 
 /*----------------------------------------------------------------*
  *  Construct decoded vector from codebook and gains.
  *---------------------------------------------------------------*/
 
-void WebRtcIlbcfix_CbConstruct(
+bool WebRtcIlbcfix_CbConstruct(
     int16_t *decvector,  /* (o) Decoded vector */
     int16_t *index,   /* (i) Codebook indices */
     int16_t *gain_index,  /* (i) Gain quantization indices */
     int16_t *mem,   /* (i) Buffer for codevector construction */
     size_t lMem,   /* (i) Length of buffer */
     size_t veclen   /* (i) Length of vector */
                                ){
   size_t j;
   int16_t gain[CB_NSTAGES];
   /* Stack based */
   int16_t cbvec0[SUBL];
   int16_t cbvec1[SUBL];
   int16_t cbvec2[SUBL];
   int32_t a32;
   int16_t *gainPtr;
 
   /* gain de-quantization */
 
   gain[0] = WebRtcIlbcfix_GainDequant(gain_index[0], 16384, 0);
   gain[1] = WebRtcIlbcfix_GainDequant(gain_index[1], gain[0], 1);
   gain[2] = WebRtcIlbcfix_GainDequant(gain_index[2], gain[1], 2);
 
   /* codebook vector construction and construction of total vector */
 
   /* Stack based */
-  WebRtcIlbcfix_GetCbVec(cbvec0, mem, (size_t)index[0], lMem, veclen);
-  WebRtcIlbcfix_GetCbVec(cbvec1, mem, (size_t)index[1], lMem, veclen);
-  WebRtcIlbcfix_GetCbVec(cbvec2, mem, (size_t)index[2], lMem, veclen);
+  if (!WebRtcIlbcfix_GetCbVec(cbvec0, mem, (size_t)index[0], lMem, veclen))
+    return false;  // Failure.
+  if (!WebRtcIlbcfix_GetCbVec(cbvec1, mem, (size_t)index[1], lMem, veclen))
+    return false;  // Failure.
+  if (!WebRtcIlbcfix_GetCbVec(cbvec2, mem, (size_t)index[2], lMem, veclen))
+    return false;  // Failure.
 
   gainPtr = &gain[0];
   for (j=0;j<veclen;j++) {
     a32 = (*gainPtr++) * cbvec0[j];
     a32 += (*gainPtr++) * cbvec1[j];
     a32 += (*gainPtr) * cbvec2[j];
     gainPtr -= 2;
     decvector[j] = (int16_t)((a32 + 8192) >> 14);
   }
 
-  return;
+  return true;  // Success.
 }