6020f45819b89d737c3e8595bffd130d4bd4d965
[pdfium.git] / core / src / fxcrt / xml_int.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_SRC_FXCRT_XML_INT_H_
8 #define CORE_SRC_FXCRT_XML_INT_H_
9
10 #include "../../include/fxcrt/fx_stream.h"
11
12 class CXML_DataBufAcc : public IFX_BufferRead {
13  public:
14   CXML_DataBufAcc(const uint8_t* pBuffer, size_t size)
15       : m_pBuffer(pBuffer), m_dwSize(size), m_dwCurPos(0) {}
16   ~CXML_DataBufAcc() override {}
17
18   // IFX_BufferRead
19   void Release() override { delete this; }
20   FX_BOOL IsEOF() override { return m_dwCurPos >= m_dwSize; }
21   FX_FILESIZE GetPosition() override { return (FX_FILESIZE)m_dwCurPos; }
22   size_t ReadBlock(void* buffer, size_t size) override { return 0; }
23   FX_BOOL ReadNextBlock(FX_BOOL bRestart = FALSE) override {
24     if (bRestart) {
25       m_dwCurPos = 0;
26     }
27     if (m_dwCurPos < m_dwSize) {
28       m_dwCurPos = m_dwSize;
29       return TRUE;
30     }
31     return FALSE;
32   }
33   const uint8_t* GetBlockBuffer() override { return m_pBuffer; }
34   size_t GetBlockSize() override { return m_dwSize; }
35   FX_FILESIZE GetBlockOffset() override { return 0; }
36
37  protected:
38   const uint8_t* m_pBuffer;
39   size_t m_dwSize;
40   size_t m_dwCurPos;
41 };
42
43 #define FX_XMLDATASTREAM_BufferSize (32 * 1024)
44 class CXML_DataStmAcc : public IFX_BufferRead {
45  public:
46   CXML_DataStmAcc(IFX_FileRead* pFileRead)
47       : m_pFileRead(pFileRead), m_pBuffer(NULL), m_nStart(0), m_dwSize(0) {
48     FXSYS_assert(m_pFileRead != NULL);
49   }
50   ~CXML_DataStmAcc() override {
51     if (m_pBuffer) {
52       FX_Free(m_pBuffer);
53     }
54   }
55
56   void Release() override { delete this; }
57   FX_BOOL IsEOF() override {
58     return m_nStart + (FX_FILESIZE)m_dwSize >= m_pFileRead->GetSize();
59   }
60   FX_FILESIZE GetPosition() override {
61     return m_nStart + (FX_FILESIZE)m_dwSize;
62   }
63   size_t ReadBlock(void* buffer, size_t size) override { return 0; }
64   FX_BOOL ReadNextBlock(FX_BOOL bRestart = FALSE) override {
65     if (bRestart) {
66       m_nStart = 0;
67     }
68     FX_FILESIZE nLength = m_pFileRead->GetSize();
69     m_nStart += (FX_FILESIZE)m_dwSize;
70     if (m_nStart >= nLength) {
71       return FALSE;
72     }
73     m_dwSize = (size_t)FX_MIN(FX_XMLDATASTREAM_BufferSize, nLength - m_nStart);
74     if (!m_pBuffer) {
75       m_pBuffer = FX_Alloc(uint8_t, m_dwSize);
76     }
77     return m_pFileRead->ReadBlock(m_pBuffer, m_nStart, m_dwSize);
78   }
79   const uint8_t* GetBlockBuffer() override { return (const uint8_t*)m_pBuffer; }
80   size_t GetBlockSize() override { return m_dwSize; }
81   FX_FILESIZE GetBlockOffset() override { return m_nStart; }
82
83  protected:
84   IFX_FileRead* m_pFileRead;
85   uint8_t* m_pBuffer;
86   FX_FILESIZE m_nStart;
87   size_t m_dwSize;
88 };
89
90 class CXML_Parser {
91  public:
92   ~CXML_Parser();
93   IFX_BufferRead* m_pDataAcc;
94   FX_BOOL m_bOwnedStream;
95   FX_FILESIZE m_nOffset;
96   FX_BOOL m_bSaveSpaceChars;
97   const uint8_t* m_pBuffer;
98   size_t m_dwBufferSize;
99   FX_FILESIZE m_nBufferOffset;
100   size_t m_dwIndex;
101   FX_BOOL Init(uint8_t* pBuffer, size_t size);
102   FX_BOOL Init(IFX_FileRead* pFileRead);
103   FX_BOOL Init(IFX_BufferRead* pBuffer);
104   FX_BOOL Init(FX_BOOL bOwndedStream);
105   FX_BOOL ReadNextBlock();
106   FX_BOOL IsEOF();
107   FX_BOOL HaveAvailData();
108   void SkipWhiteSpaces();
109   void GetName(CFX_ByteString& space, CFX_ByteString& name);
110   void GetAttrValue(CFX_WideString& value);
111   FX_DWORD GetCharRef();
112   void GetTagName(CFX_ByteString& space,
113                   CFX_ByteString& name,
114                   FX_BOOL& bEndTag,
115                   FX_BOOL bStartTag = FALSE);
116   void SkipLiterals(const CFX_ByteStringC& str);
117   CXML_Element* ParseElement(CXML_Element* pParent, FX_BOOL bStartTag = FALSE);
118   void InsertContentSegment(FX_BOOL bCDATA,
119                             const CFX_WideStringC& content,
120                             CXML_Element* pElement);
121   void InsertCDATASegment(CFX_UTF8Decoder& decoder, CXML_Element* pElement);
122 };
123
124 void FX_XML_SplitQualifiedName(const CFX_ByteStringC& bsFullName,
125                                CFX_ByteStringC& bsSpace,
126                                CFX_ByteStringC& bsName);
127
128 #endif  // CORE_SRC_FXCRT_XML_INT_H_