Initial commit

2021-08-21 18:22:58 +02:00
commit 39e2f6ecc9
10 changed files with 1561 additions and 0 deletions
--- a/include/uPDFParser.h
+++ b/include/uPDFParser.h
@@ -0,0 +1,109 @@
+/*
+  Copyright 2021 Grégory Soutadé
+
+  This file is part of uPDFParser.
+
+  uPDFParser is free software: you can redistribute it and/or modify
+  it under the terms of the GNU Lesser General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  uPDFParser is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU Lesser General Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public License
+  along with uPDFParser. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef _UPDFPARSER_HPP_
+#define _UPDFPARSER_HPP_
+
+#include <exception>
+#include <map>
+#include <vector>
+#include <string>
+#include <sstream>
+#include <iostream>
+#include <iomanip>
+#include <string.h>
+#include <unistd.h>
+
+#include "uPDFTypes.h"
+#include "uPDFObject.h"
+
+namespace uPDFParser
+{
+    /**
+     * @brief PDF Parser
+     */
+    class Parser
+    {
+    public:
+	Parser():
+	    fd(0)
+	{}
+
+	~Parser()
+	{
+	    if (fd) close(fd);
+	    
+	    std::vector<Object*>::iterator it;
+	    for(it=_objects.begin(); it!=_objects.end(); it++)
+		delete *it;
+	}
+
+	/**
+	 * @brief Parse a file
+	 */
+	void parse(const std::string& filename);
+
+	/**
+	 * @brief Write a PDF file with internal objects
+	 *
+	 * @param filename File path
+	 * @param update   Only append new objects if true
+	 *                 Write a new PDF file if false (not supported for now)
+	 */
+	void write(const std::string& filename, bool update=false);
+
+	/**
+	 * @brief Get internals (or parsed) objects
+	 */
+	std::vector<Object*>& objects() { return _objects; }
+
+	/**
+	 * @brief Add an object
+	 */
+	void addObject(Object* object) { _objects.push_back(object); }
+	
+    private:
+	void parseObject(std::string& token);
+	void parseXref();
+	void parseTrailer();
+
+	std::string nextToken(bool exceptionOnEOF=true);
+	
+	DataType* parseType(std::string& token, Object* object, std::map<std::string, DataType*>& dict);
+	void parseDictionary(Object* object, std::map<std::string, DataType*>& dict);
+	DataType* parseSignedNumber(std::string& token);
+	DataType* parseNumber(std::string& token);
+	DataType* parseNumberOrReference(std::string& token);
+	Array* parseArray(Object* object);
+	String* parseString();
+	HexaString* parseHexaString();
+	Stream* parseStream();
+	Name* parseName(std::string& token);
+
+	void writeUpdate(const std::string& filename);
+	
+	std::vector<Object*> _objects;
+	Object trailer;
+	off_t xrefOffset;
+	int fd;
+	off_t curOffset;
+    };
+}
+
+#endif