Index: core/src/fxcodec/jbig2/JBig2_Image.cpp |
diff --git a/core/src/fxcodec/jbig2/JBig2_Image.cpp b/core/src/fxcodec/jbig2/JBig2_Image.cpp |
index 5da1fc63d0c296c5197ca160488dfdaf2120bb81..03929b84c867d60ad02be257df78f304d4dcef42 100644 |
--- a/core/src/fxcodec/jbig2/JBig2_Image.cpp |
+++ b/core/src/fxcodec/jbig2/JBig2_Image.cpp |
@@ -4,10 +4,12 @@ |
// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
-#include "JBig2_Image.h" |
+#include <limits.h> |
#include "../../../include/fxcrt/fx_basic.h" |
#include "../../../include/fxcrt/fx_coordinates.h" |
-#include <limits.h> |
+#include "../../../src/fxcrt/fx_safe_types.h" |
+#include "JBig2_Image.h" |
+ |
CJBig2_Image::CJBig2_Image(FX_INT32 w, FX_INT32 h) |
{ |
m_nWidth = w; |
@@ -768,7 +770,12 @@ void CJBig2_Image::expand(FX_INT32 h, FX_BOOL v) |
if (!m_pData) { |
return; |
} |
- m_pData = (FX_BYTE*)m_pModule->JBig2_Realloc(m_pData, h * m_nStride); |
+ FX_SAFE_DWORD safeMemSize = pdfium::base::checked_cast<FX_DWORD>(h); |
+ safeMemSize *= pdfium::base::checked_cast<FX_DWORD>(m_nStride); |
+ if (!safeMemSize.IsValid()) { |
+ return; |
+ } |
+ m_pData = (FX_BYTE*)m_pModule->JBig2_Realloc(m_pData, safeMemSize.ValueOrDie()); |
if(h > m_nHeight) { |
JBIG2_memset(m_pData + m_nHeight * m_nStride, v ? 0xff : 0, (h - m_nHeight)*m_nStride); |
} |