Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(2559)

Unified Diff: core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp

Issue 322333002: Fix the potential integer overflow from "offset + size" (Closed) Base URL: https://pdfium.googlesource.com/pdfium.git@master
Patch Set: Created 6 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « no previous file | core/src/fxcrt/extension.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp
diff --git a/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp b/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp
index 926117722f7d93eb63a1cbe33e9c4a13ce6b98f6..b27ddd21a4a8b4f7a7259e2fbc13410fa2e7b7af 100644
--- a/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp
+++ b/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser.cpp
@@ -2865,10 +2865,21 @@ FX_BOOL CPDF_DataAvail::IsObjectsAvail(CFX_PtrArray& obj_array, FX_BOOL bParsePa
FX_DWORD dwNum = pRef->GetRefObjNum();
FX_FILESIZE offset;
FX_DWORD size = GetObjectSize(pRef->GetRefObjNum(), offset);
- if (!size) {
+
+ if (size == 0 || offset < 0 || offset >= m_dwFileLen)
break;
- }
- size = (FX_DWORD)((FX_FILESIZE)(offset + size + 512) > m_dwFileLen ? m_dwFileLen - offset : size + 512);
+
+ base::CheckedNumeric<FX_DWORD> safe_size = size;
palmer 2014/07/09 18:29:14 You should check the value that you actually use,
jun_fang 2014/07/10 22:21:54 safe_size is a temporary variable which is used to
+ safe_size += offset;
+ safe_size += 512;
+ if (!safe_size.IsValid())
+ break;
+
+ if (safe_size.ValueOrDie() > m_dwFileLen)
+ size = m_dwFileLen - offset;
jun_fang 2014/07/10 22:21:54 size = m_dwFileLen - offset; |size| should be vali
+ else
+ size = size + 512;
jun_fang 2014/07/10 22:21:54 because safe_size is valid, we can make sure size
+
if (!m_pFileAvail->IsDataAvail(offset, size)) {
pHints->AddSegment(offset, size);
ret_array.Add(pObj);
@@ -3070,8 +3081,22 @@ CPDF_Object* CPDF_DataAvail::GetObject(FX_DWORD objnum, IFX_DownloadHints* pHint
*pExistInFile = FALSE;
return NULL;
}
+
FX_DWORD size = (FX_DWORD)m_parser.GetObjectSize(objnum);
- size = (FX_DWORD)(((FX_FILESIZE)(offset + size + 512)) > m_dwFileLen ? m_dwFileLen - offset : size + 512);
+ if (size == 0 || offset < 0 || offset >= m_dwFileLen)
+ return NULL;
+
+ base::CheckedNumeric<FX_DWORD> safe_size = size;
palmer 2014/07/09 18:29:14 Same as above.
+ safe_size += offset;
+ safe_size += 512;
+ if (!safe_size.IsValid())
+ return NULL;
+
+ if (safe_size.ValueOrDie() > m_dwFileLen)
+ size = m_dwFileLen - offset;
+ else
+ size = size + 512;
+
if (!m_pFileAvail->IsDataAvail(offset, size)) {
pHints->AddSegment(offset, size);
return NULL;
@@ -3084,7 +3109,20 @@ CPDF_Object* CPDF_DataAvail::GetObject(FX_DWORD objnum, IFX_DownloadHints* pHint
}
FX_FILESIZE offset = 0;
FX_DWORD size = GetObjectSize(objnum, offset);
- size = (FX_DWORD)((FX_FILESIZE)(offset + size + 512) > m_dwFileLen ? m_dwFileLen - offset : size + 512);
+ if (size == 0 || offset < 0 || offset >= m_dwFileLen)
+ return NULL;
+
+ base::CheckedNumeric<FX_DWORD> safe_size = size;
palmer 2014/07/09 18:29:14 Same as above.
+ safe_size += offset;
+ safe_size += 512;
+ if (!safe_size.IsValid())
+ return NULL;
+
+ if (safe_size.ValueOrDie() > m_dwFileLen)
+ size = m_dwFileLen - offset;
+ else
+ size = size + 512;
+
if (!m_pFileAvail->IsDataAvail(offset, size)) {
pHints->AddSegment(offset, size);
return NULL;
« no previous file with comments | « no previous file | core/src/fxcrt/extension.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698