[Decode] fix kw issues(CL#777411)

fix kw issues Change-Id: Iee57216bb82824cafa75dd8b32117de16ee8300b
Sherry-Lin · Jun 29, 2018 · 4e6e44c · 4e6e44c
1 parent 8f2fdb9
commit 4e6e44c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/media_driver/agnostic/common/codec/hal/codechal_decode_mpeg2.cpp b/media_driver/agnostic/common/codec/hal/codechal_decode_mpeg2.cpp
@@ -427,6 +427,8 @@ MOS_STATUS CodechalDecodeMpeg2::AllocateResources ()
 
         CodechalResLock DummyBitstreamLock(m_osInterface, &m_resMpeg2DummyBistream);
         auto data = DummyBitstreamLock.Lock(CodechalResLock::writeOnly);
+
+        CODECHAL_DECODE_CHK_NULL_RETURN(data);
 
         MOS_ZeroMemory(data, size);
         CODECHAL_DECODE_CHK_STATUS_MESSAGE_RETURN(MOS_SecureMemcpy(
@@ -435,6 +437,7 @@ MOS_STATUS CodechalDecodeMpeg2::AllocateResources ()
             (void *)CODECHAL_DECODE_MPEG2_WaDummyBitstream,
             sizeof(CODECHAL_DECODE_MPEG2_WaDummyBitstream)),
             "Failed to copy memory.");
+
     }
 
     if (m_mode == CODECHAL_DECODE_MODE_MPEG2IDCT)

diff --git a/media_driver/agnostic/common/codec/hal/codechal_decode_vp8.cpp b/media_driver/agnostic/common/codec/hal/codechal_decode_vp8.cpp
@@ -804,6 +804,8 @@ MOS_STATUS CodechalDecodeVp8::ParseFrameHead(uint8_t* bitstreamBuffer, uint32_t
     CodechalResLock ResourceLock(m_osInterface, &m_resCoefProbBuffer);
     auto data = (uint8_t*)ResourceLock.Lock(CodechalResLock::writeOnly);
 
+    CODECHAL_DECODE_CHK_NULL_RETURN(data);
+
     MOS_SecureMemcpy(
         data,
         sizeof(m_vp8FrameHead.FrameContext.CoefProbs),