| Index: fpdfsdk/fpdftext.cpp
|
| diff --git a/fpdfsdk/fpdftext.cpp b/fpdfsdk/fpdftext.cpp
|
| index 007e18274e48cc4c2c2b2e6d5c7390b51ba01ffb..2a841314224ea704aacc2580ebdd0a8550597c1a 100644
|
| --- a/fpdfsdk/fpdftext.cpp
|
| +++ b/fpdfsdk/fpdftext.cpp
|
| @@ -12,6 +12,7 @@
|
| #include "core/fpdftext/include/cpdf_textpage.h"
|
| #include "core/fpdftext/include/cpdf_textpagefind.h"
|
| #include "fpdfsdk/include/fsdk_define.h"
|
| +#include "third_party/base/numerics/safe_conversions.h"
|
|
|
| #ifdef PDF_ENABLE_XFA
|
| #include "fpdfsdk/fpdfxfa/include/fpdfxfa_doc.h"
|
| @@ -273,8 +274,9 @@ DLLEXPORT FPDF_PAGELINK STDCALL FPDFLink_LoadWebLinks(FPDF_TEXTPAGE text_page) {
|
| if (!text_page)
|
| return nullptr;
|
|
|
| - CPDF_LinkExtract* pageLink = new CPDF_LinkExtract;
|
| - pageLink->ExtractLinks(CPDFTextPageFromFPDFTextPage(text_page));
|
| + CPDF_LinkExtract* pageLink =
|
| + new CPDF_LinkExtract(CPDFTextPageFromFPDFTextPage(text_page));
|
| + pageLink->ExtractLinks();
|
| return pageLink;
|
| }
|
|
|
| @@ -283,42 +285,40 @@ DLLEXPORT int STDCALL FPDFLink_CountWebLinks(FPDF_PAGELINK link_page) {
|
| return 0;
|
|
|
| CPDF_LinkExtract* pageLink = CPDFLinkExtractFromFPDFPageLink(link_page);
|
| - return pageLink->CountLinks();
|
| + return pdfium::base::checked_cast<int>(pageLink->CountLinks());
|
| }
|
|
|
| DLLEXPORT int STDCALL FPDFLink_GetURL(FPDF_PAGELINK link_page,
|
| int link_index,
|
| unsigned short* buffer,
|
| int buflen) {
|
| - if (!link_page)
|
| - return 0;
|
| -
|
| - CPDF_LinkExtract* pageLink = CPDFLinkExtractFromFPDFPageLink(link_page);
|
| - CFX_WideString url = pageLink->GetURL(link_index);
|
| -
|
| - CFX_ByteString cbUTF16URL = url.UTF16LE_Encode();
|
| - int len = cbUTF16URL.GetLength() / sizeof(unsigned short);
|
| + CFX_WideString wsUrl(L"");
|
| + if (link_page && link_index >= 0) {
|
| + CPDF_LinkExtract* pageLink = CPDFLinkExtractFromFPDFPageLink(link_page);
|
| + wsUrl = pageLink->GetURL(link_index);
|
| + }
|
| + CFX_ByteString cbUTF16URL = wsUrl.UTF16LE_Encode();
|
| + int required = cbUTF16URL.GetLength() / sizeof(unsigned short);
|
| if (!buffer || buflen <= 0)
|
| - return len;
|
| + return required;
|
|
|
| - int size = len < buflen ? len : buflen;
|
| + int size = std::min(required, buflen);
|
| if (size > 0) {
|
| int buf_size = size * sizeof(unsigned short);
|
| FXSYS_memcpy(buffer, cbUTF16URL.GetBuffer(buf_size), buf_size);
|
| - cbUTF16URL.ReleaseBuffer(buf_size);
|
| }
|
| return size;
|
| }
|
|
|
| DLLEXPORT int STDCALL FPDFLink_CountRects(FPDF_PAGELINK link_page,
|
| int link_index) {
|
| - if (!link_page)
|
| + if (!link_page || link_index < 0)
|
| return 0;
|
|
|
| + CFX_RectArray rects;
|
| CPDF_LinkExtract* pageLink = CPDFLinkExtractFromFPDFPageLink(link_page);
|
| - CFX_RectArray rectArray;
|
| - pageLink->GetRects(link_index, rectArray);
|
| - return rectArray.GetSize();
|
| + pageLink->GetRects(link_index, &rects);
|
| + return rects.GetSize();
|
| }
|
|
|
| DLLEXPORT void STDCALL FPDFLink_GetRect(FPDF_PAGELINK link_page,
|
| @@ -328,20 +328,22 @@ DLLEXPORT void STDCALL FPDFLink_GetRect(FPDF_PAGELINK link_page,
|
| double* top,
|
| double* right,
|
| double* bottom) {
|
| - if (!link_page)
|
| + if (!link_page || link_index < 0 || rect_index < 0)
|
| return;
|
|
|
| - CPDF_LinkExtract* pageLink = CPDFLinkExtractFromFPDFPageLink(link_page);
|
| CFX_RectArray rectArray;
|
| - pageLink->GetRects(link_index, rectArray);
|
| - if (rect_index >= 0 && rect_index < rectArray.GetSize()) {
|
| - CFX_FloatRect rect = rectArray.GetAt(rect_index);
|
| - *left = rect.left;
|
| - *right = rect.right;
|
| - *top = rect.top;
|
| - *bottom = rect.bottom;
|
| - }
|
| + CPDF_LinkExtract* pageLink = CPDFLinkExtractFromFPDFPageLink(link_page);
|
| + pageLink->GetRects(link_index, &rectArray);
|
| + if (rect_index >= rectArray.GetSize())
|
| + return;
|
| +
|
| + CFX_FloatRect rect = rectArray.GetAt(rect_index);
|
| + *left = rect.left;
|
| + *right = rect.right;
|
| + *top = rect.top;
|
| + *bottom = rect.bottom;
|
| }
|
| +
|
| DLLEXPORT void STDCALL FPDFLink_CloseWebLinks(FPDF_PAGELINK link_page) {
|
| delete CPDFLinkExtractFromFPDFPageLink(link_page);
|
| }
|
|
|