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