shithub: openh264

Info  •  Files  •  Log  •  Branches

avoid encoder crash with random pixel input

--- a/codec/encoder/core/src/encoder_ext.cpp

+++ b/codec/encoder/core/src/encoder_ext.cpp

@@ -2965,6 +2965,9 @@

  * \return	EFrameType (videoFrameTypeIDR/videoFrameTypeI/videoFrameTypeP)

  */

 int32_t WelsEncoderEncodeExt (sWelsEncCtx* pCtx, SFrameBSInfo* pFbi, const SSourcePicture* pSrcPic) {

+  if( pCtx == NULL ) {

+    return ENC_RETURN_MEMALLOCERR;

+  }

   SLayerBSInfo* pLayerBsInfo					= &pFbi->sLayerInfo[0];

   SWelsSvcCodingParam* pSvcParam	= pCtx->pSvcParam;

   SSpatialPicIndex* pSpatialIndexMap = &pCtx->sSpatialIndexMap[0];