diff options
author | Lei Zhang <thestig@chromium.org> | 2018-02-09 21:58:59 +0000 |
---|---|---|
committer | Chromium commit bot <commit-bot@chromium.org> | 2018-02-09 21:58:59 +0000 |
commit | 2404eba504ac36ce2bfd2f7d6092f9e3898de9b0 (patch) | |
tree | f774e5ab5903e189a81d4ee4c698249d6f707d8b /fpdfsdk | |
parent | 107fa7b06773a2b3e83fc360125d54105574f9bf (diff) | |
download | pdfium-2404eba504ac36ce2bfd2f7d6092f9e3898de9b0.tar.xz |
Fix nits in ParsePageRangeString().
- Fix typos.
- Do not pass by value.
- Return early if the page range string is empty after removing spaces.
- Simplify and rearrange bits of the implementation.
Change-Id: Ia4c4f43c2ca18383ad1edc8233969a7013e34722
Reviewed-on: https://pdfium-review.googlesource.com/24290
Commit-Queue: Lei Zhang <thestig@chromium.org>
Reviewed-by: Henrique Nakashima <hnakashima@chromium.org>
Reviewed-by: Shirleen Lou <xlou@chromium.org>
Diffstat (limited to 'fpdfsdk')
-rw-r--r-- | fpdfsdk/fpdf_ppo.cpp | 54 |
1 files changed, 27 insertions, 27 deletions
diff --git a/fpdfsdk/fpdf_ppo.cpp b/fpdfsdk/fpdf_ppo.cpp index efec4b222b..85da592d94 100644 --- a/fpdfsdk/fpdf_ppo.cpp +++ b/fpdfsdk/fpdf_ppo.cpp @@ -214,48 +214,42 @@ bool CopyInheritable(CPDF_Dictionary* pCurPageDict, return true; } -bool ParserPageRangeString(ByteString rangstring, - uint32_t nCount, - std::vector<uint32_t>* pageArray) { - if (rangstring.IsEmpty()) +bool ParsePageRangeString(const ByteString& bsPageRange, + uint32_t nCount, + std::vector<uint32_t>* pageArray) { + ByteString bsStrippedPageRange = bsPageRange; + bsStrippedPageRange.Remove(' '); + size_t nLength = bsStrippedPageRange.GetLength(); + if (nLength == 0) return true; - rangstring.Remove(' '); - size_t nLength = rangstring.GetLength(); - ByteString cbCompareString("0123456789-,"); + static const ByteString cbCompareString("0123456789-,"); for (size_t i = 0; i < nLength; ++i) { - if (!cbCompareString.Contains(rangstring[i])) + if (!cbCompareString.Contains(bsStrippedPageRange[i])) return false; } ByteString cbMidRange; size_t nStringFrom = 0; - Optional<size_t> nStringTo = 0; + size_t nStringTo = 0; while (nStringTo < nLength) { - nStringTo = rangstring.Find(',', nStringFrom); - if (!nStringTo.has_value()) - nStringTo = nLength; - cbMidRange = rangstring.Mid(nStringFrom, nStringTo.value() - nStringFrom); - auto nMid = cbMidRange.Find('-'); - if (!nMid.has_value()) { - uint32_t pageNum = - pdfium::base::checked_cast<uint32_t>(atoi(cbMidRange.c_str())); - if (pageNum <= 0 || pageNum > nCount) - return false; - pageArray->push_back(pageNum); - } else { + nStringTo = bsStrippedPageRange.Find(',', nStringFrom).value_or(nLength); + cbMidRange = bsStrippedPageRange.Mid(nStringFrom, nStringTo - nStringFrom); + Optional<size_t> nDashPosition = cbMidRange.Find('-'); + if (nDashPosition) { + size_t nMid = nDashPosition.value(); uint32_t nStartPageNum = pdfium::base::checked_cast<uint32_t>( - atoi(cbMidRange.Left(nMid.value()).c_str())); + atoi(cbMidRange.Left(nMid).c_str())); if (nStartPageNum == 0) return false; - nMid = nMid.value() + 1; - size_t nEnd = cbMidRange.GetLength() - nMid.value(); + ++nMid; + size_t nEnd = cbMidRange.GetLength() - nMid; if (nEnd == 0) return false; uint32_t nEndPageNum = pdfium::base::checked_cast<uint32_t>( - atoi(cbMidRange.Mid(nMid.value(), nEnd).c_str())); + atoi(cbMidRange.Mid(nMid, nEnd).c_str())); if (nStartPageNum < 0 || nStartPageNum > nEndPageNum || nEndPageNum > nCount) { return false; @@ -263,8 +257,14 @@ bool ParserPageRangeString(ByteString rangstring, for (uint32_t i = nStartPageNum; i <= nEndPageNum; ++i) { pageArray->push_back(i); } + } else { + uint32_t nPageNum = + pdfium::base::checked_cast<uint32_t>(atoi(cbMidRange.c_str())); + if (nPageNum <= 0 || nPageNum > nCount) + return false; + pageArray->push_back(nPageNum); } - nStringFrom = nStringTo.value() + 1; + nStringFrom = nStringTo + 1; } return true; } @@ -277,7 +277,7 @@ std::vector<uint32_t> GetPageNumbers(const CPDF_Document& doc, for (uint32_t i = 1; i <= nCount; ++i) page_numbers.push_back(i); } else { - if (!ParserPageRangeString(bsPageRange, nCount, &page_numbers)) + if (!ParsePageRangeString(bsPageRange, nCount, &page_numbers)) page_numbers.clear(); } return page_numbers; |