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 "../include/fsdk_define.h"
8 #include "../include/fpdf_dataavail.h"
9
10 extern void ProcessParseError(FX_DWORD err_code);
11 class CFPDF_FileAvailWrap : public IFX_FileAvail
12 {
13 public:
CFPDF_FileAvailWrap()14 CFPDF_FileAvailWrap()
15 {
16 m_pfileAvail = NULL;
17 }
18
Set(FX_FILEAVAIL * pfileAvail)19 void Set(FX_FILEAVAIL* pfileAvail)
20 {
21 m_pfileAvail = pfileAvail;
22 }
23
IsDataAvail(FX_FILESIZE offset,FX_DWORD size)24 virtual FX_BOOL IsDataAvail( FX_FILESIZE offset, FX_DWORD size)
25 {
26 return m_pfileAvail->IsDataAvail(m_pfileAvail, offset, size);
27 }
28
29 private:
30 FX_FILEAVAIL* m_pfileAvail;
31 };
32
33 class CFPDF_FileAccessWrap : public IFX_FileRead
34 {
35 public:
CFPDF_FileAccessWrap()36 CFPDF_FileAccessWrap()
37 {
38 m_pFileAccess = NULL;
39 }
40
Set(FPDF_FILEACCESS * pFile)41 void Set(FPDF_FILEACCESS* pFile)
42 {
43 m_pFileAccess = pFile;
44 }
45
GetSize()46 virtual FX_FILESIZE GetSize()
47 {
48 return m_pFileAccess->m_FileLen;
49 }
50
ReadBlock(void * buffer,FX_FILESIZE offset,size_t size)51 virtual FX_BOOL ReadBlock(void* buffer, FX_FILESIZE offset, size_t size)
52 {
53 return m_pFileAccess->m_GetBlock(m_pFileAccess->m_Param, offset, (FX_LPBYTE)buffer, size);
54 }
55
Release()56 virtual void Release()
57 {
58 }
59
60 private:
61 FPDF_FILEACCESS* m_pFileAccess;
62 };
63
64 class CFPDF_DownloadHintsWrap : public IFX_DownloadHints
65 {
66 public:
CFPDF_DownloadHintsWrap(FX_DOWNLOADHINTS * pDownloadHints)67 CFPDF_DownloadHintsWrap(FX_DOWNLOADHINTS* pDownloadHints)
68 {
69 m_pDownloadHints = pDownloadHints;
70 }
71 public:
AddSegment(FX_FILESIZE offset,FX_DWORD size)72 virtual void AddSegment(FX_FILESIZE offset, FX_DWORD size)
73 {
74 m_pDownloadHints->AddSegment(m_pDownloadHints, offset, size);
75 }
76 private:
77 FX_DOWNLOADHINTS* m_pDownloadHints;
78 };
79
80 class CFPDF_DataAvail : public CFX_Object
81 {
82 public:
CFPDF_DataAvail()83 CFPDF_DataAvail()
84 {
85 m_pDataAvail = NULL;
86 }
87
~CFPDF_DataAvail()88 ~CFPDF_DataAvail()
89 {
90 if (m_pDataAvail) delete m_pDataAvail;
91 }
92
93 CPDF_DataAvail* m_pDataAvail;
94 CFPDF_FileAvailWrap m_FileAvail;
95 CFPDF_FileAccessWrap m_FileRead;
96 };
97
FPDFAvail_Create(FX_FILEAVAIL * file_avail,FPDF_FILEACCESS * file)98 DLLEXPORT FPDF_AVAIL STDCALL FPDFAvail_Create(FX_FILEAVAIL* file_avail, FPDF_FILEACCESS* file)
99 {
100 CFPDF_DataAvail* pAvail = FX_NEW CFPDF_DataAvail;
101 pAvail->m_FileAvail.Set(file_avail);
102 pAvail->m_FileRead.Set(file);
103 pAvail->m_pDataAvail = FX_NEW CPDF_DataAvail(&pAvail->m_FileAvail, &pAvail->m_FileRead);
104 return pAvail;
105 }
106
FPDFAvail_Destroy(FPDF_AVAIL avail)107 DLLEXPORT void STDCALL FPDFAvail_Destroy(FPDF_AVAIL avail)
108 {
109 if (avail == NULL) return;
110 delete (CFPDF_DataAvail*)avail;
111 }
112
FPDFAvail_IsDocAvail(FPDF_AVAIL avail,FX_DOWNLOADHINTS * hints)113 DLLEXPORT int STDCALL FPDFAvail_IsDocAvail(FPDF_AVAIL avail, FX_DOWNLOADHINTS* hints)
114 {
115 if (avail == NULL || hints == NULL) return 0;
116 CFPDF_DownloadHintsWrap hints_wrap(hints);
117 return ((CFPDF_DataAvail*)avail)->m_pDataAvail->IsDocAvail(&hints_wrap);
118 }
119
120 extern void CheckUnSupportError(CPDF_Document * pDoc, FX_DWORD err_code);
121
FPDFAvail_GetDocument(FPDF_AVAIL avail,FPDF_BYTESTRING password)122 DLLEXPORT FPDF_DOCUMENT STDCALL FPDFAvail_GetDocument(FPDF_AVAIL avail, FPDF_BYTESTRING password)
123 {
124 if (avail == NULL) return NULL;
125 CPDF_Parser* pParser = FX_NEW CPDF_Parser;
126 pParser->SetPassword(password);
127
128 FX_DWORD err_code = pParser->StartAsynParse(((CFPDF_DataAvail*)avail)->m_pDataAvail->GetFileRead());
129 if (err_code) {
130 delete pParser;
131 ProcessParseError(err_code);
132 return NULL;
133 }
134 ((CFPDF_DataAvail*)avail)->m_pDataAvail->SetDocument(pParser->GetDocument());
135 CheckUnSupportError(pParser->GetDocument(), FPDF_ERR_SUCCESS);
136 return pParser->GetDocument();
137 }
138
FPDFAvail_GetFirstPageNum(FPDF_DOCUMENT doc)139 DLLEXPORT int STDCALL FPDFAvail_GetFirstPageNum(FPDF_DOCUMENT doc)
140 {
141 if (doc == NULL) return 0;
142 CPDF_Document* pDoc = (CPDF_Document*)doc;
143 return ((CPDF_Parser*)pDoc->GetParser())->GetFirstPageNo();
144 }
145
FPDFAvail_IsPageAvail(FPDF_AVAIL avail,int page_index,FX_DOWNLOADHINTS * hints)146 DLLEXPORT int STDCALL FPDFAvail_IsPageAvail(FPDF_AVAIL avail, int page_index, FX_DOWNLOADHINTS* hints)
147 {
148 if (avail == NULL || hints == NULL) return 0;
149 CFPDF_DownloadHintsWrap hints_wrap(hints);
150 return ((CFPDF_DataAvail*)avail)->m_pDataAvail->IsPageAvail(page_index, &hints_wrap);
151 }
152
FPDFAvail_IsFormAvail(FPDF_AVAIL avail,FX_DOWNLOADHINTS * hints)153 DLLEXPORT int STDCALL FPDFAvail_IsFormAvail(FPDF_AVAIL avail, FX_DOWNLOADHINTS* hints)
154 {
155 if (avail == NULL || hints == NULL) return -1;
156 CFPDF_DownloadHintsWrap hints_wrap(hints);
157 return ((CFPDF_DataAvail*)avail)->m_pDataAvail->IsFormAvail(&hints_wrap);
158 }
159
FPDFAvail_IsLinearized(FPDF_AVAIL avail)160 DLLEXPORT FPDF_BOOL STDCALL FPDFAvail_IsLinearized(FPDF_AVAIL avail)
161 {
162 if (avail == NULL) return -1;
163 return ((CFPDF_DataAvail*)avail)->m_pDataAvail->IsLinearizedPDF();
164
165 }
166