1 // Copyright 2014 PDFium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "public/fpdf_dataavail.h"
8
9 #include <memory>
10 #include <utility>
11
12 #include "core/fpdfapi/parser/cpdf_data_avail.h"
13 #include "core/fpdfapi/parser/cpdf_document.h"
14 #include "core/fxcrt/cfx_retain_ptr.h"
15 #include "fpdfsdk/fsdk_define.h"
16 #include "public/fpdf_formfill.h"
17 #include "third_party/base/ptr_util.h"
18
19 // These checks are here because core/ and public/ cannot depend on each other.
20 static_assert(CPDF_DataAvail::DataError == PDF_DATA_ERROR,
21 "CPDF_DataAvail::DataError value mismatch");
22 static_assert(CPDF_DataAvail::DataNotAvailable == PDF_DATA_NOTAVAIL,
23 "CPDF_DataAvail::DataNotAvailable value mismatch");
24 static_assert(CPDF_DataAvail::DataAvailable == PDF_DATA_AVAIL,
25 "CPDF_DataAvail::DataAvailable value mismatch");
26
27 static_assert(CPDF_DataAvail::LinearizationUnknown == PDF_LINEARIZATION_UNKNOWN,
28 "CPDF_DataAvail::LinearizationUnknown value mismatch");
29 static_assert(CPDF_DataAvail::NotLinearized == PDF_NOT_LINEARIZED,
30 "CPDF_DataAvail::NotLinearized value mismatch");
31 static_assert(CPDF_DataAvail::Linearized == PDF_LINEARIZED,
32 "CPDF_DataAvail::Linearized value mismatch");
33
34 static_assert(CPDF_DataAvail::FormError == PDF_FORM_ERROR,
35 "CPDF_DataAvail::FormError value mismatch");
36 static_assert(CPDF_DataAvail::FormNotAvailable == PDF_FORM_NOTAVAIL,
37 "CPDF_DataAvail::FormNotAvailable value mismatch");
38 static_assert(CPDF_DataAvail::FormAvailable == PDF_FORM_AVAIL,
39 "CPDF_DataAvail::FormAvailable value mismatch");
40 static_assert(CPDF_DataAvail::FormNotExist == PDF_FORM_NOTEXIST,
41 "CPDF_DataAvail::FormNotExist value mismatch");
42
43 namespace {
44
45 class CFPDF_FileAvailWrap : public CPDF_DataAvail::FileAvail {
46 public:
CFPDF_FileAvailWrap()47 CFPDF_FileAvailWrap() : m_pfileAvail(nullptr) {}
~CFPDF_FileAvailWrap()48 ~CFPDF_FileAvailWrap() override {}
49
Set(FX_FILEAVAIL * pfileAvail)50 void Set(FX_FILEAVAIL* pfileAvail) { m_pfileAvail = pfileAvail; }
51
52 // CPDF_DataAvail::FileAvail:
IsDataAvail(FX_FILESIZE offset,uint32_t size)53 bool IsDataAvail(FX_FILESIZE offset, uint32_t size) override {
54 return !!m_pfileAvail->IsDataAvail(m_pfileAvail, offset, size);
55 }
56
57 private:
58 FX_FILEAVAIL* m_pfileAvail;
59 };
60
61 class CFPDF_FileAccessWrap : public IFX_SeekableReadStream {
62 public:
Create()63 static CFX_RetainPtr<CFPDF_FileAccessWrap> Create() {
64 return CFX_RetainPtr<CFPDF_FileAccessWrap>(new CFPDF_FileAccessWrap());
65 }
~CFPDF_FileAccessWrap()66 ~CFPDF_FileAccessWrap() override {}
67
Set(FPDF_FILEACCESS * pFile)68 void Set(FPDF_FILEACCESS* pFile) { m_pFileAccess = pFile; }
69
70 // IFX_SeekableReadStream
GetSize()71 FX_FILESIZE GetSize() override { return m_pFileAccess->m_FileLen; }
72
ReadBlock(void * buffer,FX_FILESIZE offset,size_t size)73 bool ReadBlock(void* buffer, FX_FILESIZE offset, size_t size) override {
74 return !!m_pFileAccess->m_GetBlock(m_pFileAccess->m_Param, offset,
75 (uint8_t*)buffer, size);
76 }
77
78 private:
CFPDF_FileAccessWrap()79 CFPDF_FileAccessWrap() : m_pFileAccess(nullptr) {}
80
81 FPDF_FILEACCESS* m_pFileAccess;
82 };
83
84 class CFPDF_DownloadHintsWrap : public CPDF_DataAvail::DownloadHints {
85 public:
CFPDF_DownloadHintsWrap(FX_DOWNLOADHINTS * pDownloadHints)86 explicit CFPDF_DownloadHintsWrap(FX_DOWNLOADHINTS* pDownloadHints) {
87 m_pDownloadHints = pDownloadHints;
88 }
~CFPDF_DownloadHintsWrap()89 ~CFPDF_DownloadHintsWrap() override {}
90
91 public:
92 // IFX_DownloadHints
AddSegment(FX_FILESIZE offset,uint32_t size)93 void AddSegment(FX_FILESIZE offset, uint32_t size) override {
94 m_pDownloadHints->AddSegment(m_pDownloadHints, offset, size);
95 }
96
97 private:
98 FX_DOWNLOADHINTS* m_pDownloadHints;
99 };
100
101 class CFPDF_DataAvail {
102 public:
CFPDF_DataAvail()103 CFPDF_DataAvail()
104 : m_FileAvail(new CFPDF_FileAvailWrap),
105 m_FileRead(CFPDF_FileAccessWrap::Create()) {}
~CFPDF_DataAvail()106 ~CFPDF_DataAvail() {}
107
108 std::unique_ptr<CPDF_DataAvail> m_pDataAvail;
109 std::unique_ptr<CFPDF_FileAvailWrap> m_FileAvail;
110 CFX_RetainPtr<CFPDF_FileAccessWrap> m_FileRead;
111 };
112
CFPDFDataAvailFromFPDFAvail(FPDF_AVAIL avail)113 CFPDF_DataAvail* CFPDFDataAvailFromFPDFAvail(FPDF_AVAIL avail) {
114 return static_cast<CFPDF_DataAvail*>(avail);
115 }
116
117 } // namespace
118
FPDFAvail_Create(FX_FILEAVAIL * file_avail,FPDF_FILEACCESS * file)119 DLLEXPORT FPDF_AVAIL STDCALL FPDFAvail_Create(FX_FILEAVAIL* file_avail,
120 FPDF_FILEACCESS* file) {
121 CFPDF_DataAvail* pAvail = new CFPDF_DataAvail;
122 pAvail->m_FileAvail->Set(file_avail);
123 pAvail->m_FileRead->Set(file);
124 pAvail->m_pDataAvail = pdfium::MakeUnique<CPDF_DataAvail>(
125 pAvail->m_FileAvail.get(), pAvail->m_FileRead, true);
126 return pAvail;
127 }
128
FPDFAvail_Destroy(FPDF_AVAIL avail)129 DLLEXPORT void STDCALL FPDFAvail_Destroy(FPDF_AVAIL avail) {
130 delete (CFPDF_DataAvail*)avail;
131 }
132
FPDFAvail_IsDocAvail(FPDF_AVAIL avail,FX_DOWNLOADHINTS * hints)133 DLLEXPORT int STDCALL FPDFAvail_IsDocAvail(FPDF_AVAIL avail,
134 FX_DOWNLOADHINTS* hints) {
135 if (!avail || !hints)
136 return PDF_DATA_ERROR;
137 CFPDF_DownloadHintsWrap hints_wrap(hints);
138 return CFPDFDataAvailFromFPDFAvail(avail)->m_pDataAvail->IsDocAvail(
139 &hints_wrap);
140 }
141
142 DLLEXPORT FPDF_DOCUMENT STDCALL
FPDFAvail_GetDocument(FPDF_AVAIL avail,FPDF_BYTESTRING password)143 FPDFAvail_GetDocument(FPDF_AVAIL avail, FPDF_BYTESTRING password) {
144 CFPDF_DataAvail* pDataAvail = static_cast<CFPDF_DataAvail*>(avail);
145 if (!pDataAvail)
146 return nullptr;
147
148 std::unique_ptr<CPDF_Parser> pParser(new CPDF_Parser);
149 pParser->SetPassword(password);
150
151 std::unique_ptr<CPDF_Document> pDocument(
152 new CPDF_Document(std::move(pParser)));
153 CPDF_Parser::Error error = pDocument->GetParser()->StartLinearizedParse(
154 pDataAvail->m_pDataAvail->GetFileRead(), pDocument.get());
155 if (error != CPDF_Parser::SUCCESS) {
156 ProcessParseError(error);
157 return nullptr;
158 }
159 pDataAvail->m_pDataAvail->SetDocument(pDocument.get());
160 CheckUnSupportError(pDocument.get(), FPDF_ERR_SUCCESS);
161 return FPDFDocumentFromCPDFDocument(pDocument.release());
162 }
163
FPDFAvail_GetFirstPageNum(FPDF_DOCUMENT doc)164 DLLEXPORT int STDCALL FPDFAvail_GetFirstPageNum(FPDF_DOCUMENT doc) {
165 CPDF_Document* pDoc = CPDFDocumentFromFPDFDocument(doc);
166 return pDoc ? pDoc->GetParser()->GetFirstPageNo() : 0;
167 }
168
FPDFAvail_IsPageAvail(FPDF_AVAIL avail,int page_index,FX_DOWNLOADHINTS * hints)169 DLLEXPORT int STDCALL FPDFAvail_IsPageAvail(FPDF_AVAIL avail,
170 int page_index,
171 FX_DOWNLOADHINTS* hints) {
172 if (!avail || !hints)
173 return PDF_DATA_ERROR;
174 if (page_index < 0)
175 return PDF_DATA_NOTAVAIL;
176 CFPDF_DownloadHintsWrap hints_wrap(hints);
177 return CFPDFDataAvailFromFPDFAvail(avail)->m_pDataAvail->IsPageAvail(
178 page_index, &hints_wrap);
179 }
180
FPDFAvail_IsFormAvail(FPDF_AVAIL avail,FX_DOWNLOADHINTS * hints)181 DLLEXPORT int STDCALL FPDFAvail_IsFormAvail(FPDF_AVAIL avail,
182 FX_DOWNLOADHINTS* hints) {
183 if (!avail || !hints)
184 return PDF_FORM_ERROR;
185 CFPDF_DownloadHintsWrap hints_wrap(hints);
186 return CFPDFDataAvailFromFPDFAvail(avail)->m_pDataAvail->IsFormAvail(
187 &hints_wrap);
188 }
189
FPDFAvail_IsLinearized(FPDF_AVAIL avail)190 DLLEXPORT int STDCALL FPDFAvail_IsLinearized(FPDF_AVAIL avail) {
191 if (!avail)
192 return PDF_LINEARIZATION_UNKNOWN;
193 return CFPDFDataAvailFromFPDFAvail(avail)->m_pDataAvail->IsLinearizedPDF();
194 }
195