summaryrefslogtreecommitdiff
path: root/fpdfsdk/fpdftext.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'fpdfsdk/fpdftext.cpp')
-rw-r--r--fpdfsdk/fpdftext.cpp21
1 files changed, 9 insertions, 12 deletions
diff --git a/fpdfsdk/fpdftext.cpp b/fpdfsdk/fpdftext.cpp
index ae6841febc..9d7d56311f 100644
--- a/fpdfsdk/fpdftext.cpp
+++ b/fpdfsdk/fpdftext.cpp
@@ -164,29 +164,26 @@ FPDF_EXPORT int FPDF_CALLCONV FPDFText_GetText(FPDF_TEXTPAGE text_page,
int start,
int count,
unsigned short* result) {
- if (start < 0 || count < 1 || !result || !text_page)
+ if (start < 0 || count < 0 || !result || !text_page)
return 0;
CPDF_TextPage* textpage = CPDFTextPageFromFPDFTextPage(text_page);
if (start >= textpage->CountChars())
return 0;
- WideString str = textpage->GetPageText(start, count - 1);
- if (str.GetLength() <= 0)
- return 0;
-
+ WideString str = textpage->GetPageText(start, count);
if (str.GetLength() > static_cast<size_t>(count))
str = str.Left(static_cast<size_t>(count));
// UFT16LE_Encode doesn't handle surrogate pairs properly, so it is expected
// the number of items to stay the same.
- ByteString cbUTF16str = str.UTF16LE_Encode();
- ASSERT(cbUTF16str.GetLength() / kBytesPerCharacter <=
- static_cast<size_t>(count));
- memcpy(result, cbUTF16str.GetBuffer(cbUTF16str.GetLength()),
- cbUTF16str.GetLength());
-
- return cbUTF16str.GetLength() / kBytesPerCharacter;
+ ByteString byte_str = str.UTF16LE_Encode();
+ ASSERT((byte_str.GetLength()) / kBytesPerCharacter <=
+ static_cast<size_t>(count + 1)); // +1 to account for the string
+ // terminator
+ memcpy(result, byte_str.GetBuffer(byte_str.GetLength()),
+ byte_str.GetLength());
+ return (byte_str.GetLength() / kBytesPerCharacter);
}
FPDF_EXPORT int FPDF_CALLCONV FPDFText_CountRects(FPDF_TEXTPAGE text_page,