Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 18fedd0f authored by Wei Jia's avatar Wei Jia Committed by android-build-merger
Browse files

SoftAMR: check input buffer size to avoid overflow. am: 55e5218f am: 4a9d762d... 

SoftAMR: check input buffer size to avoid overflow. am: 55e5218f am: 4a9d762d am: d82ca561 am: 7840f059 am: adb7d9c7 am: 9eedbf45
am: e00b888d

* commit 'e00b888d':
  SoftAMR: check input buffer size to avoid overflow.
parents 5de7130b e00b888d

media/libstagefright/codecs/amrnb/dec/SoftAMR.cpp

+24 −1
Original line number Diff line number Diff line
@@ -306,6 +306,13 @@ void SoftAMR::onQueueFilled(OMX_U32 /* portIndex */) { 
        BufferInfo *inInfo = *inQueue.begin();

        OMX_BUFFERHEADERTYPE *inHeader = inInfo->mHeader;



        if (inHeader->nFilledLen == 0) {

            inInfo->mOwnedByUs = false;

            inQueue.erase(inQueue.begin());

            notifyEmptyBufferDone(inHeader);

            continue;

        }



        BufferInfo *outInfo = *outQueue.begin();

        OMX_BUFFERHEADERTYPE *outHeader = outInfo->mHeader;
@@ -341,6 +348,17 @@ void SoftAMR::onQueueFilled(OMX_U32 /* portIndex */) { 
                return;

            }



            int16 mode = ((inputPtr[0] >> 3) & 0x0f);

            // for WMF since MIME_IETF is used when calling AMRDecode.

            size_t frameSize = WmfDecBytesPerFrame[mode] + 1;



            if (inHeader->nFilledLen < frameSize) {

                ALOGE("b/27662364: expected %zu bytes vs %u", frameSize, inHeader->nFilledLen);

                notify(OMX_EventError, OMX_ErrorStreamCorrupt, 0, NULL);

                mSignalledError = true;

                return;

            }



            numBytesRead =

                AMRDecode(mState,

                  (Frame_Type_3GPP)((inputPtr[0] >> 3) & 0x0f),
@@ -390,7 +408,12 @@ void SoftAMR::onQueueFilled(OMX_U32 /* portIndex */) { 
            }



            size_t frameSize = getFrameSize(mode);

            CHECK_GE(inHeader->nFilledLen, frameSize);

            if (inHeader->nFilledLen < frameSize) {

                ALOGE("b/27662364: expected %zu bytes vs %u", frameSize, inHeader->nFilledLen);

                notify(OMX_EventError, OMX_ErrorStreamCorrupt, 0, NULL);

                mSignalledError = true;

                return;

            }



            int16_t *outPtr = (int16_t *)outHeader->pBuffer;

media/libstagefright/codecs/amrnb/dec/src/amrdecode.h

+0 −1
Original line number Diff line number Diff line
@@ -104,7 +104,6 @@ terms listed above has been obtained from the copyright holder. 
; INCLUDES

----------------------------------------------------------------------------*/

#include    "typedef.h"

#include    "mode.h"

#include    "frame_type_3gpp.h"



/*--------------------------------------------------------------------------*/

media/libstagefright/codecs/amrnb/dec/src/gsmamr_dec.h

+1 −13
Original line number Diff line number Diff line
@@ -87,6 +87,7 @@ terms listed above has been obtained from the copyright holder. 


#include "gsm_amr_typedefs.h"

#include "frame_type_3gpp.h"

#include "amrdecode.h"



/*--------------------------------------------------------------------------*/

#ifdef __cplusplus
@@ -135,19 +136,6 @@ extern "C" 
    Word16 GSMInitDecode(void **state_data,

                         Word8 *id);



    /*

     * AMRDecode steps into the part of the library that decodes the raw data

     * speech bits for the decoding process. It returns the address offset of

     * the next frame to be decoded.

     */

    Word16 AMRDecode(

        void                      *state_data,

        enum Frame_Type_3GPP      frame_type,

        UWord8                    *speech_bits_ptr,

        Word16                    *raw_pcm_buffer,

        Word16                    input_format

    );



    /*

     * This function resets the state memory used by the GSM AMR decoder. This

     * function returns zero. It will return negative one if there is an error.