Extern in .cpp file is a code smell, part 2.
[pdfium.git] / core / include / fpdfapi / fpdf_parser.h
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 #ifndef CORE_INCLUDE_FPDFAPI_FPDF_PARSER_H_
8 #define CORE_INCLUDE_FPDFAPI_FPDF_PARSER_H_
9
10 #include "../fxcrt/fx_system.h"
11 #include "fpdf_objects.h"
12
13 class CPDF_Document;
14 class CPDF_Parser;
15 class CPDF_SecurityHandler;
16 class CPDF_StandardSecurityHandler;
17 class CPDF_CryptoHandler;
18 class CPDF_Object;
19 class IFX_FileRead;
20 class CFDF_Document;
21 class CFDF_Parser;
22 class CFX_Font;
23 class CFX_AffineMatrix;
24 class CFX_FloatRect;
25 class CPDF_Point;
26 class CPDF_DocPageData;
27 class CPDF_DocRenderData;
28 class CPDF_ModuleMgr;
29 class CFX_DIBSource;
30 class CPDF_Font;
31 class CPDF_Image;
32 class CPDF_ColorSpace;
33 class CPDF_Pattern;
34 class CPDF_FontEncoding;
35 class CPDF_IccProfile;
36 class CFX_PrivateData;
37 #define FPDFPERM_PRINT 0x0004
38 #define FPDFPERM_MODIFY 0x0008
39 #define FPDFPERM_EXTRACT 0x0010
40 #define FPDFPERM_ANNOT_FORM 0x0020
41 #define FPDFPERM_FILL_FORM 0x0100
42 #define FPDFPERM_EXTRACT_ACCESS 0x0200
43 #define FPDFPERM_ASSEMBLE 0x0400
44 #define FPDFPERM_PRINT_HIGH 0x0800
45 #define FPDF_PAGE_MAX_NUM 0xFFFFF
46
47 // Indexed by 8-bit character code, contains either:
48 //   'W' - for whitespace: NUL, TAB, CR, LF, FF, 0x80, 0xff
49 //   'N' - for numeric: 0123456789+-.
50 //   'D' - for delimiter: %()/<>[]{}
51 //   'R' - otherwise.
52 extern const char PDF_CharType[256];
53
54 // Indexed by 8-bit char code, contains unicode code points.
55 extern const FX_WORD PDFDocEncoding[256];
56
57 class CPDF_Document : public CFX_PrivateData, public CPDF_IndirectObjects {
58  public:
59   CPDF_Document();
60   explicit CPDF_Document(CPDF_Parser* pParser);
61
62   ~CPDF_Document();
63
64   CPDF_Parser* GetParser() const { return m_pParser; }
65
66   CPDF_Dictionary* GetRoot() const { return m_pRootDict; }
67
68   CPDF_Dictionary* GetInfo() const { return m_pInfoDict; }
69
70   void GetID(CFX_ByteString& id1, CFX_ByteString& id2) const {
71     id1 = m_ID1;
72     id2 = m_ID2;
73   }
74
75   int GetPageCount() const;
76
77   CPDF_Dictionary* GetPage(int iPage);
78
79   int GetPageIndex(FX_DWORD objnum);
80
81   FX_DWORD GetUserPermissions(FX_BOOL bCheckRevision = FALSE) const;
82
83   FX_BOOL IsOwner() const;
84
85   CPDF_DocPageData* GetPageData() { return GetValidatePageData(); }
86
87   void ClearPageData();
88
89   void RemoveColorSpaceFromPageData(CPDF_Object* pObject);
90
91   CPDF_DocRenderData* GetRenderData() { return GetValidateRenderData(); }
92
93   void ClearRenderData();
94
95   void ClearRenderFont();
96
97   FX_BOOL IsFormStream(FX_DWORD objnum, FX_BOOL& bForm) const;
98
99   // |pFontDict| must not be null.
100   CPDF_Font* LoadFont(CPDF_Dictionary* pFontDict);
101
102   CPDF_ColorSpace* LoadColorSpace(CPDF_Object* pCSObj,
103                                   CPDF_Dictionary* pResources = NULL);
104
105   CPDF_Pattern* LoadPattern(CPDF_Object* pObj,
106                             FX_BOOL bShading,
107                             const CFX_AffineMatrix* matrix = NULL);
108
109   CPDF_Image* LoadImageF(CPDF_Object* pObj);
110
111   CPDF_StreamAcc* LoadFontFile(CPDF_Stream* pStream);
112
113   CPDF_IccProfile* LoadIccProfile(CPDF_Stream* pStream);
114
115 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_
116
117   CPDF_Font* AddWindowsFont(LOGFONTA* pLogFont,
118                             FX_BOOL bVert,
119                             FX_BOOL bTranslateName = FALSE);
120   CPDF_Font* AddWindowsFont(LOGFONTW* pLogFont,
121                             FX_BOOL bVert,
122                             FX_BOOL bTranslateName = FALSE);
123 #endif
124 #if _FXM_PLATFORM_ == _FXM_PLATFORM_APPLE_
125   CPDF_Font* AddMacFont(CTFontRef pFont,
126                         FX_BOOL bVert,
127                         FX_BOOL bTranslateName = FALSE);
128 #endif
129
130   CPDF_Font* AddStandardFont(const FX_CHAR* font, CPDF_FontEncoding* pEncoding);
131
132   CPDF_Font* AddFont(CFX_Font* pFont, int charset, FX_BOOL bVert);
133
134   void CreateNewDoc();
135
136   CPDF_Dictionary* CreateNewPage(int iPage);
137
138   void DeletePage(int iPage);
139
140   void LoadDoc();
141   void LoadAsynDoc(CPDF_Dictionary* pLinearized);
142   void LoadPages();
143
144  protected:
145   CPDF_Dictionary* m_pRootDict;
146
147   CPDF_Dictionary* m_pInfoDict;
148
149   CFX_ByteString m_ID1;
150
151   CFX_ByteString m_ID2;
152
153   FX_BOOL m_bLinearized;
154
155   FX_DWORD m_dwFirstPageNo;
156
157   FX_DWORD m_dwFirstPageObjNum;
158
159   CFX_DWordArray m_PageList;
160
161   int _GetPageCount() const;
162   CPDF_Dictionary* _FindPDFPage(CPDF_Dictionary* pPages,
163                                 int iPage,
164                                 int nPagesToGo,
165                                 int level);
166   int _FindPageIndex(CPDF_Dictionary* pNode,
167                      FX_DWORD& skip_count,
168                      FX_DWORD objnum,
169                      int& index,
170                      int level = 0);
171   FX_BOOL IsContentUsedElsewhere(FX_DWORD objnum, CPDF_Dictionary* pPageDict);
172   FX_BOOL CheckOCGVisible(CPDF_Dictionary* pOCG, FX_BOOL bPrinting);
173   CPDF_DocPageData* GetValidatePageData();
174   CPDF_DocRenderData* GetValidateRenderData();
175   friend class CPDF_Creator;
176   friend class CPDF_Parser;
177   friend class CPDF_DataAvail;
178   friend class CPDF_OCContext;
179
180   CPDF_DocPageData* m_pDocPage;
181
182   CPDF_DocRenderData* m_pDocRender;
183 };
184
185 #define PDFWORD_EOF 0
186 #define PDFWORD_NUMBER 1
187 #define PDFWORD_TEXT 2
188 #define PDFWORD_DELIMITER 3
189 #define PDFWORD_NAME 4
190 class CPDF_SimpleParser {
191  public:
192   CPDF_SimpleParser(const uint8_t* pData, FX_DWORD dwSize);
193
194   CPDF_SimpleParser(const CFX_ByteStringC& str);
195
196   CFX_ByteStringC GetWord();
197
198   FX_BOOL SearchToken(const CFX_ByteStringC& token);
199
200   FX_BOOL SkipWord(const CFX_ByteStringC& token);
201
202   FX_BOOL FindTagPair(const CFX_ByteStringC& start_token,
203                       const CFX_ByteStringC& end_token,
204                       FX_DWORD& start_pos,
205                       FX_DWORD& end_pos);
206
207   FX_BOOL FindTagParam(const CFX_ByteStringC& token, int nParams);
208
209   FX_DWORD GetPos() { return m_dwCurPos; }
210
211   void SetPos(FX_DWORD pos) {
212     ASSERT(pos <= m_dwSize);
213     m_dwCurPos = pos;
214   }
215
216  private:
217   void ParseWord(const uint8_t*& pStart, FX_DWORD& dwSize, int& type);
218
219   const uint8_t* m_pData;
220
221   FX_DWORD m_dwSize;
222
223   FX_DWORD m_dwCurPos;
224 };
225 class CPDF_SyntaxParser {
226  public:
227   CPDF_SyntaxParser();
228   virtual ~CPDF_SyntaxParser();
229
230   void InitParser(IFX_FileRead* pFileAccess, FX_DWORD HeaderOffset);
231
232   FX_FILESIZE SavePos() { return m_Pos; }
233
234   void RestorePos(FX_FILESIZE pos) { m_Pos = pos; }
235
236   CPDF_Object* GetObject(CPDF_IndirectObjects* pObjList,
237                          FX_DWORD objnum,
238                          FX_DWORD gennum,
239                          struct PARSE_CONTEXT* pContext = NULL,
240                          FX_BOOL bDecrypt = TRUE);
241
242   CPDF_Object* GetObjectByStrict(CPDF_IndirectObjects* pObjList,
243                                  FX_DWORD objnum,
244                                  FX_DWORD gennum,
245                                  struct PARSE_CONTEXT* pContext = NULL);
246
247   int GetDirectNum();
248
249   CFX_ByteString GetString(FX_DWORD objnum, FX_DWORD gennum);
250
251   CFX_ByteString GetName();
252
253   CFX_ByteString GetKeyword();
254
255   void GetBinary(uint8_t* buffer, FX_DWORD size);
256
257   void ToNextLine();
258
259   void ToNextWord();
260
261   FX_BOOL SearchWord(const CFX_ByteStringC& word,
262                      FX_BOOL bWholeWord,
263                      FX_BOOL bForward,
264                      FX_FILESIZE limit);
265
266   int SearchMultiWord(const CFX_ByteStringC& words,
267                       FX_BOOL bWholeWord,
268                       FX_FILESIZE limit);
269
270   FX_FILESIZE FindTag(const CFX_ByteStringC& tag, FX_FILESIZE limit);
271
272   void SetEncrypt(CPDF_CryptoHandler* pCryptoHandler) {
273     m_pCryptoHandler = pCryptoHandler;
274   }
275
276   FX_BOOL IsEncrypted() { return m_pCryptoHandler != NULL; }
277
278   FX_BOOL GetCharAt(FX_FILESIZE pos, uint8_t& ch);
279
280   FX_BOOL ReadBlock(uint8_t* pBuf, FX_DWORD size);
281
282   CFX_ByteString GetNextWord(FX_BOOL& bIsNumber);
283
284  protected:
285   static const int kParserMaxRecursionDepth = 64;
286   static int s_CurrentRecursionDepth;
287
288   virtual FX_BOOL GetNextChar(uint8_t& ch);
289
290   FX_BOOL GetCharAtBackward(FX_FILESIZE pos, uint8_t& ch);
291
292   void GetNextWord();
293
294   FX_BOOL IsWholeWord(FX_FILESIZE startpos,
295                       FX_FILESIZE limit,
296                       const uint8_t* tag,
297                       FX_DWORD taglen);
298
299   CFX_ByteString ReadString();
300
301   CFX_ByteString ReadHexString();
302
303   unsigned int ReadEOLMarkers(FX_FILESIZE pos);
304
305   CPDF_Stream* ReadStream(CPDF_Dictionary* pDict,
306                           PARSE_CONTEXT* pContext,
307                           FX_DWORD objnum,
308                           FX_DWORD gennum);
309
310   FX_FILESIZE m_Pos;
311
312   FX_BOOL m_bFileStream;
313
314   int m_MetadataObjnum;
315
316   IFX_FileRead* m_pFileAccess;
317
318   FX_DWORD m_HeaderOffset;
319
320   FX_FILESIZE m_FileLen;
321
322   uint8_t* m_pFileBuf;
323
324   FX_DWORD m_BufSize;
325
326   FX_FILESIZE m_BufOffset;
327
328   CPDF_CryptoHandler* m_pCryptoHandler;
329
330   uint8_t m_WordBuffer[257];
331
332   FX_DWORD m_WordSize;
333
334   FX_BOOL m_bIsNumber;
335
336   FX_FILESIZE m_dwWordPos;
337   friend class CPDF_Parser;
338   friend class CPDF_DataAvail;
339 };
340
341 #define PDFPARSE_TYPEONLY 1
342 #define PDFPARSE_NOSTREAM 2
343 struct PARSE_CONTEXT {
344   uint32_t m_Flags;
345   FX_FILESIZE m_DictStart;
346   FX_FILESIZE m_DictEnd;
347   FX_FILESIZE m_DataStart;
348   FX_FILESIZE m_DataEnd;
349 };
350
351 #define PDFPARSE_ERROR_SUCCESS 0
352 #define PDFPARSE_ERROR_FILE 1
353 #define PDFPARSE_ERROR_FORMAT 2
354 #define PDFPARSE_ERROR_PASSWORD 3
355 #define PDFPARSE_ERROR_HANDLER 4
356 #define PDFPARSE_ERROR_CERT 5
357
358 class CPDF_Parser {
359  public:
360   CPDF_Parser();
361   ~CPDF_Parser();
362
363   FX_DWORD StartParse(IFX_FileRead* pFile,
364                       FX_BOOL bReParse = FALSE,
365                       FX_BOOL bOwnFileRead = TRUE);
366
367   void CloseParser(FX_BOOL bReParse = FALSE);
368
369   FX_DWORD GetPermissions(FX_BOOL bCheckRevision = FALSE);
370
371   FX_BOOL IsOwner();
372
373   void SetPassword(const FX_CHAR* password) { m_Password = password; }
374
375   CFX_ByteString GetPassword() { return m_Password; }
376
377   CPDF_SecurityHandler* GetSecurityHandler() { return m_pSecurityHandler; }
378
379   CPDF_CryptoHandler* GetCryptoHandler() { return m_Syntax.m_pCryptoHandler; }
380
381   void SetSecurityHandler(CPDF_SecurityHandler* pSecurityHandler,
382                           FX_BOOL bForced = FALSE);
383
384   CFX_ByteString GetRecipient() { return m_bsRecipient; }
385
386   CPDF_Dictionary* GetTrailer() { return m_pTrailer; }
387
388   FX_FILESIZE GetLastXRefOffset() { return m_LastXRefOffset; }
389
390   CPDF_Document* GetDocument() { return m_pDocument; }
391
392   CFX_ArrayTemplate<CPDF_Dictionary*>* GetOtherTrailers() {
393     return &m_Trailers;
394   }
395
396   FX_DWORD GetRootObjNum();
397   FX_DWORD GetInfoObjNum();
398   CPDF_Array* GetIDArray();
399
400   CPDF_Dictionary* GetEncryptDict() { return m_pEncryptDict; }
401
402   FX_BOOL IsEncrypted() { return GetEncryptDict() != NULL; }
403
404   CPDF_Object* ParseIndirectObject(CPDF_IndirectObjects* pObjList,
405                                    FX_DWORD objnum,
406                                    PARSE_CONTEXT* pContext = NULL);
407   FX_DWORD GetLastObjNum();
408   FX_BOOL IsFormStream(FX_DWORD objnum, FX_BOOL& bForm);
409
410   FX_FILESIZE GetObjectOffset(FX_DWORD objnum);
411
412   FX_FILESIZE GetObjectSize(FX_DWORD objnum);
413
414   int GetObjectVersion(FX_DWORD objnum) { return m_ObjVersion[objnum]; }
415
416   void GetIndirectBinary(FX_DWORD objnum, uint8_t*& pBuffer, FX_DWORD& size);
417
418   FX_BOOL GetFileStreamOption() { return m_Syntax.m_bFileStream; }
419
420   void SetFileStreamOption(FX_BOOL b) { m_Syntax.m_bFileStream = b; }
421
422   IFX_FileRead* GetFileAccess() const { return m_Syntax.m_pFileAccess; }
423
424   int GetFileVersion() const { return m_FileVersion; }
425
426   FX_BOOL IsXRefStream() const { return m_bXRefStream; }
427   CPDF_Object* ParseIndirectObjectAt(CPDF_IndirectObjects* pObjList,
428                                      FX_FILESIZE pos,
429                                      FX_DWORD objnum,
430                                      struct PARSE_CONTEXT* pContext);
431
432   CPDF_Object* ParseIndirectObjectAtByStrict(CPDF_IndirectObjects* pObjList,
433                                              FX_FILESIZE pos,
434                                              FX_DWORD objnum,
435                                              struct PARSE_CONTEXT* pContext,
436                                              FX_FILESIZE* pResultPos);
437
438   FX_DWORD StartAsynParse(IFX_FileRead* pFile,
439                           FX_BOOL bReParse = FALSE,
440                           FX_BOOL bOwnFileRead = TRUE);
441
442   FX_DWORD GetFirstPageNo() { return m_dwFirstPageNo; }
443
444  protected:
445   CPDF_Document* m_pDocument;
446
447   CPDF_SyntaxParser m_Syntax;
448   FX_BOOL m_bOwnFileRead;
449   CPDF_Object* ParseDirect(CPDF_Object* pObj);
450
451   FX_BOOL LoadAllCrossRefV4(FX_FILESIZE pos);
452
453   FX_BOOL LoadAllCrossRefV5(FX_FILESIZE pos);
454
455   FX_BOOL LoadCrossRefV4(FX_FILESIZE pos,
456                          FX_FILESIZE streampos,
457                          FX_BOOL bSkip,
458                          FX_BOOL bFirst);
459
460   FX_BOOL LoadCrossRefV5(FX_FILESIZE pos, FX_FILESIZE& prev, FX_BOOL bMainXRef);
461
462   CPDF_Dictionary* LoadTrailerV4();
463
464   FX_BOOL RebuildCrossRef();
465
466   FX_DWORD SetEncryptHandler();
467
468   void ReleaseEncryptHandler();
469
470   FX_BOOL LoadLinearizedAllCrossRefV4(FX_FILESIZE pos, FX_DWORD dwObjCount);
471
472   FX_BOOL LoadLinearizedCrossRefV4(FX_FILESIZE pos, FX_DWORD dwObjCount);
473
474   FX_BOOL LoadLinearizedAllCrossRefV5(FX_FILESIZE pos);
475
476   FX_DWORD LoadLinearizedMainXRefTable();
477
478   CFX_MapPtrToPtr m_ObjectStreamMap;
479
480   CPDF_StreamAcc* GetObjectStream(FX_DWORD number);
481
482   FX_BOOL IsLinearizedFile(IFX_FileRead* pFileAccess, FX_DWORD offset);
483
484   int m_FileVersion;
485
486   CPDF_Dictionary* m_pTrailer;
487
488   CPDF_Dictionary* m_pEncryptDict;
489   void SetEncryptDictionary(CPDF_Dictionary* pDict);
490
491   FX_FILESIZE m_LastXRefOffset;
492
493   FX_BOOL m_bXRefStream;
494
495   CPDF_SecurityHandler* m_pSecurityHandler;
496
497   FX_BOOL m_bForceUseSecurityHandler;
498
499   CFX_ByteString m_bsRecipient;
500
501   CFX_ByteString m_FilePath;
502
503   CFX_ByteString m_Password;
504
505   CFX_FileSizeArray m_CrossRef;
506
507   CFX_ByteArray m_V5Type;
508
509   CFX_FileSizeArray m_SortedOffset;
510
511   CFX_WordArray m_ObjVersion;
512   CFX_ArrayTemplate<CPDF_Dictionary*> m_Trailers;
513
514   FX_BOOL m_bVersionUpdated;
515
516   CPDF_Object* m_pLinearized;
517
518   FX_DWORD m_dwFirstPageNo;
519
520   FX_DWORD m_dwXrefStartObjNum;
521   friend class CPDF_Creator;
522   friend class CPDF_DataAvail;
523 };
524 #define FXCIPHER_NONE 0
525 #define FXCIPHER_RC4 1
526 #define FXCIPHER_AES 2
527 #define FXCIPHER_AES2 3
528 class CPDF_SecurityHandler {
529  public:
530   virtual ~CPDF_SecurityHandler() {}
531
532   virtual FX_BOOL OnInit(CPDF_Parser* pParser,
533                          CPDF_Dictionary* pEncryptDict) = 0;
534
535   virtual FX_DWORD GetPermissions() = 0;
536
537   virtual FX_BOOL IsOwner() = 0;
538
539   virtual FX_BOOL GetCryptInfo(int& cipher,
540                                const uint8_t*& buffer,
541                                int& keylen) = 0;
542
543   virtual FX_BOOL IsMetadataEncrypted() { return TRUE; }
544
545   virtual CPDF_CryptoHandler* CreateCryptoHandler() = 0;
546
547   virtual CPDF_StandardSecurityHandler* GetStandardHandler() { return NULL; }
548 };
549 #define PDF_ENCRYPT_CONTENT 0
550 class CPDF_StandardSecurityHandler : public CPDF_SecurityHandler {
551  public:
552   CPDF_StandardSecurityHandler();
553   ~CPDF_StandardSecurityHandler() override;
554
555   // CPDF_SecurityHandler
556   FX_BOOL OnInit(CPDF_Parser* pParser, CPDF_Dictionary* pEncryptDict) override;
557   FX_DWORD GetPermissions() override;
558   FX_BOOL IsOwner() override { return m_bOwner; }
559   FX_BOOL GetCryptInfo(int& cipher,
560                        const uint8_t*& buffer,
561                        int& keylen) override;
562   FX_BOOL IsMetadataEncrypted() override;
563   CPDF_CryptoHandler* CreateCryptoHandler() override;
564   CPDF_StandardSecurityHandler* GetStandardHandler() override { return this; }
565
566   void OnCreate(CPDF_Dictionary* pEncryptDict,
567                 CPDF_Array* pIdArray,
568                 const uint8_t* user_pass,
569                 FX_DWORD user_size,
570                 const uint8_t* owner_pass,
571                 FX_DWORD owner_size,
572                 FX_DWORD type = PDF_ENCRYPT_CONTENT);
573
574   void OnCreate(CPDF_Dictionary* pEncryptDict,
575                 CPDF_Array* pIdArray,
576                 const uint8_t* user_pass,
577                 FX_DWORD user_size,
578                 FX_DWORD type = PDF_ENCRYPT_CONTENT);
579
580   CFX_ByteString GetUserPassword(const uint8_t* owner_pass, FX_DWORD pass_size);
581   CFX_ByteString GetUserPassword(const uint8_t* owner_pass,
582                                  FX_DWORD pass_size,
583                                  int32_t key_len);
584   int GetVersion() { return m_Version; }
585   int GetRevision() { return m_Revision; }
586
587   int CheckPassword(const uint8_t* password,
588                     FX_DWORD pass_size,
589                     FX_BOOL bOwner,
590                     uint8_t* key);
591   int CheckPassword(const uint8_t* password,
592                     FX_DWORD pass_size,
593                     FX_BOOL bOwner,
594                     uint8_t* key,
595                     int key_len);
596
597  private:
598   int m_Version;
599
600   int m_Revision;
601
602   CPDF_Parser* m_pParser;
603
604   CPDF_Dictionary* m_pEncryptDict;
605
606   FX_BOOL LoadDict(CPDF_Dictionary* pEncryptDict);
607   FX_BOOL LoadDict(CPDF_Dictionary* pEncryptDict,
608                    FX_DWORD type,
609                    int& cipher,
610                    int& key_len);
611
612   FX_BOOL CheckUserPassword(const uint8_t* password,
613                             FX_DWORD pass_size,
614                             FX_BOOL bIgnoreEncryptMeta,
615                             uint8_t* key,
616                             int32_t key_len);
617
618   FX_BOOL CheckOwnerPassword(const uint8_t* password,
619                              FX_DWORD pass_size,
620                              uint8_t* key,
621                              int32_t key_len);
622   FX_BOOL AES256_CheckPassword(const uint8_t* password,
623                                FX_DWORD size,
624                                FX_BOOL bOwner,
625                                uint8_t* key);
626   void AES256_SetPassword(CPDF_Dictionary* pEncryptDict,
627                           const uint8_t* password,
628                           FX_DWORD size,
629                           FX_BOOL bOwner,
630                           const uint8_t* key);
631   void AES256_SetPerms(CPDF_Dictionary* pEncryptDict,
632                        FX_DWORD permission,
633                        FX_BOOL bEncryptMetadata,
634                        const uint8_t* key);
635   void OnCreate(CPDF_Dictionary* pEncryptDict,
636                 CPDF_Array* pIdArray,
637                 const uint8_t* user_pass,
638                 FX_DWORD user_size,
639                 const uint8_t* owner_pass,
640                 FX_DWORD owner_size,
641                 FX_BOOL bDefault,
642                 FX_DWORD type);
643   FX_BOOL CheckSecurity(int32_t key_len);
644
645   FX_BOOL m_bOwner;
646
647   FX_DWORD m_Permissions;
648
649   int m_Cipher;
650
651   uint8_t m_EncryptKey[32];
652
653   int m_KeyLen;
654 };
655 class CPDF_CryptoHandler {
656  public:
657   virtual ~CPDF_CryptoHandler() {}
658
659   virtual FX_BOOL Init(CPDF_Dictionary* pEncryptDict,
660                        CPDF_SecurityHandler* pSecurityHandler) = 0;
661
662   virtual FX_DWORD DecryptGetSize(FX_DWORD src_size) = 0;
663
664   virtual void* DecryptStart(FX_DWORD objnum, FX_DWORD gennum) = 0;
665
666   virtual FX_BOOL DecryptStream(void* context,
667                                 const uint8_t* src_buf,
668                                 FX_DWORD src_size,
669                                 CFX_BinaryBuf& dest_buf) = 0;
670
671   virtual FX_BOOL DecryptFinish(void* context, CFX_BinaryBuf& dest_buf) = 0;
672
673   virtual FX_DWORD EncryptGetSize(FX_DWORD objnum,
674                                   FX_DWORD version,
675                                   const uint8_t* src_buf,
676                                   FX_DWORD src_size) = 0;
677
678   virtual FX_BOOL EncryptContent(FX_DWORD objnum,
679                                  FX_DWORD version,
680                                  const uint8_t* src_buf,
681                                  FX_DWORD src_size,
682                                  uint8_t* dest_buf,
683                                  FX_DWORD& dest_size) = 0;
684
685   void Decrypt(FX_DWORD objnum, FX_DWORD version, CFX_ByteString& str);
686 };
687 class CPDF_StandardCryptoHandler : public CPDF_CryptoHandler {
688  public:
689   CPDF_StandardCryptoHandler();
690   ~CPDF_StandardCryptoHandler() override;
691
692   // CPDF_CryptoHandler
693   FX_BOOL Init(CPDF_Dictionary* pEncryptDict,
694                CPDF_SecurityHandler* pSecurityHandler) override;
695   FX_DWORD DecryptGetSize(FX_DWORD src_size) override;
696   void* DecryptStart(FX_DWORD objnum, FX_DWORD gennum) override;
697   FX_BOOL DecryptStream(void* context,
698                         const uint8_t* src_buf,
699                         FX_DWORD src_size,
700                         CFX_BinaryBuf& dest_buf) override;
701   FX_BOOL DecryptFinish(void* context, CFX_BinaryBuf& dest_buf) override;
702   FX_DWORD EncryptGetSize(FX_DWORD objnum,
703                           FX_DWORD version,
704                           const uint8_t* src_buf,
705                           FX_DWORD src_size) override;
706   FX_BOOL EncryptContent(FX_DWORD objnum,
707                          FX_DWORD version,
708                          const uint8_t* src_buf,
709                          FX_DWORD src_size,
710                          uint8_t* dest_buf,
711                          FX_DWORD& dest_size) override;
712
713   FX_BOOL Init(int cipher, const uint8_t* key, int keylen);
714
715  protected:
716   virtual void CryptBlock(FX_BOOL bEncrypt,
717                           FX_DWORD objnum,
718                           FX_DWORD gennum,
719                           const uint8_t* src_buf,
720                           FX_DWORD src_size,
721                           uint8_t* dest_buf,
722                           FX_DWORD& dest_size);
723   virtual void* CryptStart(FX_DWORD objnum, FX_DWORD gennum, FX_BOOL bEncrypt);
724   virtual FX_BOOL CryptStream(void* context,
725                               const uint8_t* src_buf,
726                               FX_DWORD src_size,
727                               CFX_BinaryBuf& dest_buf,
728                               FX_BOOL bEncrypt);
729   virtual FX_BOOL CryptFinish(void* context,
730                               CFX_BinaryBuf& dest_buf,
731                               FX_BOOL bEncrypt);
732
733   uint8_t m_EncryptKey[32];
734
735   int m_KeyLen;
736
737   int m_Cipher;
738
739   uint8_t* m_pAESContext;
740 };
741 class CPDF_Point {
742  public:
743   CPDF_Point(FX_FLOAT xx, FX_FLOAT yy) {
744     x = xx;
745     y = yy;
746   }
747
748   FX_FLOAT x;
749
750   FX_FLOAT y;
751 };
752
753 #define CPDF_Rect CFX_FloatRect
754 #define CPDF_Matrix CFX_AffineMatrix
755 CFX_ByteString PDF_NameDecode(const CFX_ByteStringC& orig);
756 CFX_ByteString PDF_NameDecode(const CFX_ByteString& orig);
757 CFX_ByteString PDF_NameEncode(const CFX_ByteString& orig);
758 CFX_ByteString PDF_EncodeString(const CFX_ByteString& src,
759                                 FX_BOOL bHex = FALSE);
760 CFX_WideString PDF_DecodeText(const uint8_t* pData,
761                               FX_DWORD size,
762                               CFX_CharMap* pCharMap = NULL);
763 inline CFX_WideString PDF_DecodeText(const CFX_ByteString& bstr,
764                                      CFX_CharMap* pCharMap = NULL) {
765   return PDF_DecodeText((const uint8_t*)bstr.c_str(), bstr.GetLength(),
766                         pCharMap);
767 }
768 CFX_ByteString PDF_EncodeText(const FX_WCHAR* pString,
769                               int len = -1,
770                               CFX_CharMap* pCharMap = NULL);
771 inline CFX_ByteString PDF_EncodeText(const CFX_WideString& str,
772                                      CFX_CharMap* pCharMap = NULL) {
773   return PDF_EncodeText(str.c_str(), str.GetLength(), pCharMap);
774 }
775 FX_FLOAT PDF_ClipFloat(FX_FLOAT f);
776 class CFDF_Document : public CPDF_IndirectObjects {
777  public:
778   static CFDF_Document* CreateNewDoc();
779   static CFDF_Document* ParseFile(IFX_FileRead* pFile,
780                                   FX_BOOL bOwnFile = FALSE);
781   static CFDF_Document* ParseMemory(const uint8_t* pData, FX_DWORD size);
782
783   ~CFDF_Document();
784
785   FX_BOOL WriteBuf(CFX_ByteTextBuf& buf) const;
786
787   CPDF_Dictionary* GetRoot() const { return m_pRootDict; }
788
789   CFX_WideString GetWin32Path() const;
790
791  protected:
792   CFDF_Document();
793   void ParseStream(IFX_FileRead* pFile, FX_BOOL bOwnFile);
794   CPDF_Dictionary* m_pRootDict;
795   IFX_FileRead* m_pFile;
796   FX_BOOL m_bOwnFile;
797 };
798
799 CFX_WideString FPDF_FileSpec_GetWin32Path(const CPDF_Object* pFileSpec);
800 void FPDF_FileSpec_SetWin32Path(CPDF_Object* pFileSpec,
801                                 const CFX_WideString& fullpath);
802
803 void FlateEncode(const uint8_t* src_buf,
804                  FX_DWORD src_size,
805                  uint8_t*& dest_buf,
806                  FX_DWORD& dest_size);
807 void FlateEncode(const uint8_t* src_buf,
808                  FX_DWORD src_size,
809                  int predictor,
810                  int Colors,
811                  int BitsPerComponent,
812                  int Columns,
813                  uint8_t*& dest_buf,
814                  FX_DWORD& dest_size);
815 FX_DWORD FlateDecode(const uint8_t* src_buf,
816                      FX_DWORD src_size,
817                      uint8_t*& dest_buf,
818                      FX_DWORD& dest_size);
819 FX_DWORD RunLengthDecode(const uint8_t* src_buf,
820                          FX_DWORD src_size,
821                          uint8_t*& dest_buf,
822                          FX_DWORD& dest_size);
823 FX_BOOL IsSignatureDict(const CPDF_Dictionary* pDict);
824
825 class CPDF_NumberTree {
826  public:
827   CPDF_NumberTree(CPDF_Dictionary* pRoot) { m_pRoot = pRoot; }
828
829   CPDF_Object* LookupValue(int num);
830
831  protected:
832   CPDF_Dictionary* m_pRoot;
833 };
834
835 class IFX_FileAvail {
836  public:
837   virtual ~IFX_FileAvail() {}
838   virtual FX_BOOL IsDataAvail(FX_FILESIZE offset, FX_DWORD size) = 0;
839 };
840 class IFX_DownloadHints {
841  public:
842   virtual ~IFX_DownloadHints() {}
843   virtual void AddSegment(FX_FILESIZE offset, FX_DWORD size) = 0;
844 };
845 #define PDF_IS_LINEARIZED 1
846 #define PDF_NOT_LINEARIZED 0
847 #define PDF_UNKNOW_LINEARIZED -1
848 #define PDFFORM_NOTAVAIL 0
849 #define PDFFORM_AVAIL 1
850 #define PDFFORM_NOTEXIST 2
851 class IPDF_DataAvail {
852  public:
853   static IPDF_DataAvail* Create(IFX_FileAvail* pFileAvail,
854                                 IFX_FileRead* pFileRead);
855   virtual ~IPDF_DataAvail() {}
856
857   IFX_FileAvail* GetFileAvail() const { return m_pFileAvail; }
858   IFX_FileRead* GetFileRead() const { return m_pFileRead; }
859
860   virtual FX_BOOL IsDocAvail(IFX_DownloadHints* pHints) = 0;
861   virtual void SetDocument(CPDF_Document* pDoc) = 0;
862   virtual FX_BOOL IsPageAvail(int iPage, IFX_DownloadHints* pHints) = 0;
863   virtual FX_BOOL IsLinearized() = 0;
864   virtual int32_t IsFormAvail(IFX_DownloadHints* pHints) = 0;
865   virtual int32_t IsLinearizedPDF() = 0;
866   virtual void GetLinearizedMainXRefInfo(FX_FILESIZE* pPos,
867                                          FX_DWORD* pSize) = 0;
868
869  protected:
870   IPDF_DataAvail(IFX_FileAvail* pFileAvail, IFX_FileRead* pFileRead);
871
872   IFX_FileAvail* m_pFileAvail;
873   IFX_FileRead* m_pFileRead;
874 };
875 class CPDF_SortObjNumArray {
876  public:
877   void AddObjNum(FX_DWORD dwObjNum);
878
879   FX_BOOL Find(FX_DWORD dwObjNum);
880
881   void RemoveAll() { m_number_array.RemoveAll(); }
882
883  protected:
884   FX_BOOL BinarySearch(FX_DWORD value, int& iNext);
885
886  protected:
887   CFX_DWordArray m_number_array;
888 };
889 enum PDF_PAGENODE_TYPE {
890   PDF_PAGENODE_UNKOWN = 0,
891   PDF_PAGENODE_PAGE,
892   PDF_PAGENODE_PAGES,
893   PDF_PAGENODE_ARRAY,
894 };
895 class CPDF_PageNode {
896  public:
897   CPDF_PageNode() : m_type(PDF_PAGENODE_UNKOWN) {}
898   ~CPDF_PageNode();
899   PDF_PAGENODE_TYPE m_type;
900   FX_DWORD m_dwPageNo;
901   CFX_PtrArray m_childNode;
902 };
903 enum PDF_DATAAVAIL_STATUS {
904   PDF_DATAAVAIL_HEADER = 0,
905   PDF_DATAAVAIL_FIRSTPAGE,
906   PDF_DATAAVAIL_FIRSTPAGE_PREPARE,
907   PDF_DATAAVAIL_END,
908   PDF_DATAAVAIL_CROSSREF,
909   PDF_DATAAVAIL_CROSSREF_ITEM,
910   PDF_DATAAVAIL_CROSSREF_STREAM,
911   PDF_DATAAVAIL_TRAILER,
912   PDF_DATAAVAIL_LOADALLCRSOSSREF,
913   PDF_DATAAVAIL_ROOT,
914   PDF_DATAAVAIL_INFO,
915   PDF_DATAAVAIL_ACROFORM,
916   PDF_DATAAVAIL_ACROFORM_SUBOBJECT,
917   PDF_DATAAVAIL_PAGETREE,
918   PDF_DATAAVAIL_PAGE,
919   PDF_DATAAVAIL_PAGE_LATERLOAD,
920   PDF_DATAAVAIL_RESOURCES,
921   PDF_DATAAVAIL_DONE,
922   PDF_DATAAVAIL_ERROR,
923   PDF_DATAAVAIL_LOADALLFILE,
924   PDF_DATAAVAIL_TRAILER_APPEND
925 };
926
927 FX_BOOL PDF_DataDecode(const uint8_t* src_buf,
928                        FX_DWORD src_size,
929                        const CPDF_Dictionary* pDict,
930                        uint8_t*& dest_buf,
931                        FX_DWORD& dest_size,
932                        CFX_ByteString& ImageEncoding,
933                        CPDF_Dictionary*& pImageParms,
934                        FX_DWORD estimated_size,
935                        FX_BOOL bImageAcc);
936
937 #endif  // CORE_INCLUDE_FPDFAPI_FPDF_PARSER_H_