Don't bother checking pointers before delete[] and FX_Free().
[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 { FX_Free(m_pBuffer); }
51
52   void Release() override { delete this; }
53   FX_BOOL IsEOF() override {
54     return m_nStart + (FX_FILESIZE)m_dwSize >= m_pFileRead->GetSize();
55   }
56   FX_FILESIZE GetPosition() override {
57     return m_nStart + (FX_FILESIZE)m_dwSize;
58   }
59   size_t ReadBlock(void* buffer, size_t size) override { return 0; }
60   FX_BOOL ReadNextBlock(FX_BOOL bRestart = FALSE) override {
61     if (bRestart) {
62       m_nStart = 0;
63     }
64     FX_FILESIZE nLength = m_pFileRead->GetSize();
65     m_nStart += (FX_FILESIZE)m_dwSize;
66     if (m_nStart >= nLength) {
67       return FALSE;
68     }
69     m_dwSize = (size_t)FX_MIN(FX_XMLDATASTREAM_BufferSize, nLength - m_nStart);
70     if (!m_pBuffer) {
71       m_pBuffer = FX_Alloc(uint8_t, m_dwSize);
72     }
73     return m_pFileRead->ReadBlock(m_pBuffer, m_nStart, m_dwSize);
74   }
75   const uint8_t* GetBlockBuffer() override { return (const uint8_t*)m_pBuffer; }
76   size_t GetBlockSize() override { return m_dwSize; }
77   FX_FILESIZE GetBlockOffset() override { return m_nStart; }
78
79  protected:
80   IFX_FileRead* m_pFileRead;
81   uint8_t* m_pBuffer;
82   FX_FILESIZE m_nStart;
83   size_t m_dwSize;
84 };
85
86 class CXML_Parser {
87  public:
88   ~CXML_Parser();
89   IFX_BufferRead* m_pDataAcc;
90   FX_BOOL m_bOwnedStream;
91   FX_FILESIZE m_nOffset;
92   FX_BOOL m_bSaveSpaceChars;
93   const uint8_t* m_pBuffer;
94   size_t m_dwBufferSize;
95   FX_FILESIZE m_nBufferOffset;
96   size_t m_dwIndex;
97   FX_BOOL Init(uint8_t* pBuffer, size_t size);
98   FX_BOOL Init(IFX_FileRead* pFileRead);
99   FX_BOOL Init(IFX_BufferRead* pBuffer);
100   FX_BOOL Init(FX_BOOL bOwndedStream);
101   FX_BOOL ReadNextBlock();
102   FX_BOOL IsEOF();
103   FX_BOOL HaveAvailData();
104   void SkipWhiteSpaces();
105   void GetName(CFX_ByteString& space, CFX_ByteString& name);
106   void GetAttrValue(CFX_WideString& value);
107   FX_DWORD GetCharRef();
108   void GetTagName(CFX_ByteString& space,
109                   CFX_ByteString& name,
110                   FX_BOOL& bEndTag,
111                   FX_BOOL bStartTag = FALSE);
112   void SkipLiterals(const CFX_ByteStringC& str);
113   CXML_Element* ParseElement(CXML_Element* pParent, FX_BOOL bStartTag = FALSE);
114   void InsertContentSegment(FX_BOOL bCDATA,
115                             const CFX_WideStringC& content,
116                             CXML_Element* pElement);
117   void InsertCDATASegment(CFX_UTF8Decoder& decoder, CXML_Element* pElement);
118 };
119
120 void FX_XML_SplitQualifiedName(const CFX_ByteStringC& bsFullName,
121                                CFX_ByteStringC& bsSpace,
122                                CFX_ByteStringC& bsName);
123
124 #endif  // CORE_SRC_FXCRT_XML_INT_H_