blob: b7bcecdb9457affbdcde1426cf445199dea3c694 [file] [log] [blame]
// Copyright 2016 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
#ifndef CORE_FPDFAPI_PARSER_CPDF_SYNTAX_PARSER_H_
#define CORE_FPDFAPI_PARSER_CPDF_SYNTAX_PARSER_H_
#include <stdint.h>
#include <memory>
#include <vector>
#include "core/fpdfapi/parser/cpdf_stream.h"
#include "core/fxcrt/fx_memory_wrappers.h"
#include "core/fxcrt/fx_types.h"
#include "core/fxcrt/retain_ptr.h"
#include "core/fxcrt/string_pool_template.h"
#include "core/fxcrt/unowned_ptr.h"
#include "core/fxcrt/weak_ptr.h"
class CPDF_Dictionary;
class CPDF_IndirectObjectHolder;
class CPDF_Object;
class CPDF_ReadValidator;
class CPDF_Stream;
class IFX_SeekableReadStream;
class CPDF_SyntaxParser {
public:
enum class ParseType : bool { kStrict, kLoose };
struct WordResult {
ByteString word;
bool is_number;
};
static std::unique_ptr<CPDF_SyntaxParser> CreateForTesting(
const RetainPtr<IFX_SeekableReadStream>& pFileAccess,
FX_FILESIZE HeaderOffset);
explicit CPDF_SyntaxParser(
const RetainPtr<IFX_SeekableReadStream>& pFileAccess);
CPDF_SyntaxParser(const RetainPtr<CPDF_ReadValidator>& pValidator,
FX_FILESIZE HeaderOffset);
~CPDF_SyntaxParser();
void SetReadBufferSize(uint32_t read_buffer_size) {
m_ReadBufferSize = read_buffer_size;
}
FX_FILESIZE GetPos() const { return m_Pos; }
void SetPos(FX_FILESIZE pos);
RetainPtr<CPDF_Object> GetObjectBody(CPDF_IndirectObjectHolder* pObjList);
RetainPtr<CPDF_Object> GetIndirectObject(CPDF_IndirectObjectHolder* pObjList,
ParseType parse_type);
ByteString GetKeyword();
void ToNextLine();
void ToNextWord();
void RecordingToNextWord();
bool BackwardsSearchToWord(ByteStringView word, FX_FILESIZE limit);
FX_FILESIZE FindTag(ByteStringView tag);
bool ReadBlock(uint8_t* pBuf, uint32_t size);
bool GetCharAt(FX_FILESIZE pos, uint8_t& ch);
WordResult GetNextWord();
ByteString PeekNextWord();
const RetainPtr<CPDF_ReadValidator>& GetValidator() const {
return m_pFileAccess;
}
uint32_t GetDirectNum();
bool GetNextChar(uint8_t& ch);
// The document size may be smaller than the file size.
// The syntax parser use position relative to document
// offset (|m_HeaderOffset|).
// The document size will be FileSize - "Header offset".
// All offsets was readed from document, should not be great than document
// size. Use it for checks instead of real file size.
FX_FILESIZE GetDocumentSize() const;
ByteString ReadString();
ByteString ReadHexString();
void SetTrailerEnds(std::vector<unsigned int>* trailer_ends) {
m_TrailerEnds = trailer_ends;
}
private:
enum class WordType : bool { kWord, kNumber };
friend class CPDF_DataAvail;
friend class cpdf_syntax_parser_ReadHexString_Test;
static constexpr int kParserMaxRecursionDepth = 64;
static int s_CurrentRecursionDepth;
bool ReadBlockAt(FX_FILESIZE read_pos);
bool GetCharAtBackward(FX_FILESIZE pos, uint8_t* ch);
WordType GetNextWordInternal();
bool IsWholeWord(FX_FILESIZE startpos,
FX_FILESIZE limit,
ByteStringView tag,
bool checkKeyword);
unsigned int ReadEOLMarkers(FX_FILESIZE pos);
FX_FILESIZE FindWordPos(ByteStringView word);
FX_FILESIZE FindStreamEndPos();
RetainPtr<CPDF_Stream> ReadStream(RetainPtr<CPDF_Dictionary> pDict);
bool IsPositionRead(FX_FILESIZE pos) const;
RetainPtr<CPDF_Object> GetObjectBodyInternal(
CPDF_IndirectObjectHolder* pObjList,
ParseType parse_type);
RetainPtr<CPDF_ReadValidator> m_pFileAccess;
// The syntax parser use position relative to header offset.
// The header contains at file start, and can follow after some stuff. We
// ignore this stuff.
const FX_FILESIZE m_HeaderOffset;
const FX_FILESIZE m_FileLen;
FX_FILESIZE m_Pos = 0;
WeakPtr<ByteStringPool> m_pPool;
std::vector<uint8_t, FxAllocAllocator<uint8_t>> m_pFileBuf;
FX_FILESIZE m_BufOffset = 0;
uint32_t m_WordSize = 0;
uint8_t m_WordBuffer[257];
uint32_t m_ReadBufferSize = CPDF_Stream::kFileBufSize;
// The syntax parser records traversed trailer end byte offsets here.
UnownedPtr<std::vector<unsigned int>> m_TrailerEnds;
};
#endif // CORE_FPDFAPI_PARSER_CPDF_SYNTAX_PARSER_H_