b26d2af10afc5c9be716908e67d8f23002a84091
[pdfium.git] / fpdfsdk / src / fpdf_dataavail.cpp
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 #include "../include/fsdk_define.h"
9
10 extern void ProcessParseError(FX_DWORD err_code);
11 class CFPDF_FileAvailWrap : public IFX_FileAvail
12 {
13 public:
14         CFPDF_FileAvailWrap()
15         {
16                 m_pfileAvail = NULL;
17         }
18
19         void Set(FX_FILEAVAIL* pfileAvail)
20         {
21                 m_pfileAvail = pfileAvail;
22         }
23
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:
36         CFPDF_FileAccessWrap()
37         {
38                 m_pFileAccess = NULL;
39         }
40
41         void Set(FPDF_FILEACCESS* pFile)
42         {
43                 m_pFileAccess = pFile;
44         }
45
46         virtual FX_FILESIZE             GetSize()
47         {
48                 return m_pFileAccess->m_FileLen;
49         }
50
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, (uint8_t*)buffer, size);
54         }
55
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:
67         CFPDF_DownloadHintsWrap(FX_DOWNLOADHINTS* pDownloadHints)
68         {
69                 m_pDownloadHints = pDownloadHints;
70         }
71 public:
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
81 {
82 public:
83         CFPDF_DataAvail()
84         {
85                 m_pDataAvail = NULL;
86         }
87
88         ~CFPDF_DataAvail()
89         {
90             delete m_pDataAvail;
91         }
92
93         IPDF_DataAvail*                 m_pDataAvail;
94         CFPDF_FileAvailWrap             m_FileAvail;
95         CFPDF_FileAccessWrap    m_FileRead;
96 };
97
98 DLLEXPORT FPDF_AVAIL STDCALL FPDFAvail_Create(FX_FILEAVAIL* file_avail, FPDF_FILEACCESS* file)
99 {
100         CFPDF_DataAvail* pAvail = new CFPDF_DataAvail;
101         pAvail->m_FileAvail.Set(file_avail);
102         pAvail->m_FileRead.Set(file);
103         pAvail->m_pDataAvail = IPDF_DataAvail::Create(&pAvail->m_FileAvail, &pAvail->m_FileRead);
104         return pAvail;
105 }
106
107 DLLEXPORT void STDCALL FPDFAvail_Destroy(FPDF_AVAIL avail)
108 {
109     delete (CFPDF_DataAvail*)avail;
110 }
111
112 DLLEXPORT int STDCALL FPDFAvail_IsDocAvail(FPDF_AVAIL avail, FX_DOWNLOADHINTS* hints)
113 {
114         if (avail == NULL || hints == NULL) return 0;
115         CFPDF_DownloadHintsWrap hints_wrap(hints);
116         return ((CFPDF_DataAvail*)avail)->m_pDataAvail->IsDocAvail(&hints_wrap);
117 }
118
119 extern void CheckUnSupportError(CPDF_Document * pDoc, FX_DWORD err_code);
120
121 DLLEXPORT FPDF_DOCUMENT STDCALL FPDFAvail_GetDocument(FPDF_AVAIL avail, FPDF_BYTESTRING password)
122 {
123         if (avail == NULL) return NULL;
124         CPDF_Parser* pParser = new CPDF_Parser;
125         pParser->SetPassword(password);
126
127         FX_DWORD err_code = pParser->StartAsynParse(((CFPDF_DataAvail*)avail)->m_pDataAvail->GetFileRead());
128         if (err_code) {
129                 delete pParser;
130                 ProcessParseError(err_code);
131                 return NULL;
132         }
133         ((CFPDF_DataAvail*)avail)->m_pDataAvail->SetDocument(pParser->GetDocument());
134         CheckUnSupportError(pParser->GetDocument(), FPDF_ERR_SUCCESS);
135         return pParser->GetDocument();
136 }
137
138 DLLEXPORT int STDCALL FPDFAvail_GetFirstPageNum(FPDF_DOCUMENT doc)
139 {
140         if (doc == NULL) return 0;
141         CPDF_Document* pDoc = (CPDF_Document*)doc;
142         return ((CPDF_Parser*)pDoc->GetParser())->GetFirstPageNo();
143 }
144
145 DLLEXPORT int STDCALL FPDFAvail_IsPageAvail(FPDF_AVAIL avail, int page_index, FX_DOWNLOADHINTS* hints)
146 {
147         if (avail == NULL || hints == NULL) return 0;
148         CFPDF_DownloadHintsWrap hints_wrap(hints);
149         return ((CFPDF_DataAvail*)avail)->m_pDataAvail->IsPageAvail(page_index, &hints_wrap);
150 }
151
152 DLLEXPORT int STDCALL FPDFAvail_IsFormAvail(FPDF_AVAIL avail, FX_DOWNLOADHINTS* hints)
153 {
154         if (avail == NULL || hints == NULL) return -1;
155         CFPDF_DownloadHintsWrap hints_wrap(hints);
156         return ((CFPDF_DataAvail*)avail)->m_pDataAvail->IsFormAvail(&hints_wrap);
157 }
158
159 DLLEXPORT FPDF_BOOL STDCALL FPDFAvail_IsLinearized(FPDF_AVAIL avail)
160 {
161                 if (avail == NULL) return -1;
162         return ((CFPDF_DataAvail*)avail)->m_pDataAvail->IsLinearizedPDF();
163
164 }