arch.h

#ifndef ARCH_H
#define ARCH_H
 
#include <array>
#include "../common.h"
#include "../dumper/dumper.h"
#include "../linker/buffer.h"
#include "../linker/format.h"
#include "../linker/reader.h"
#include "../linker/writer.h"
 
/* TODO: not fully implemented */
 
namespace Archive
{
    class ArchiveFormat : public virtual Linker::InputFormat, public virtual Linker::OutputFormat
    {
    public:
        class FileReader
        {
        public:
            virtual ~FileReader();
            virtual std::shared_ptr<Linker::Contents> ReadFile(const std::shared_ptr<Linker::Reader>& rd, offset_t size) = 0;
        };
 
        offset_t file_offset = 0;
        offset_t file_size = offset_t(-1);
        std::shared_ptr<FileReader> file_reader = nullptr;
 
        typedef std::shared_ptr<Linker::Contents> file_reader_type(const std::shared_ptr<Linker::Reader>& rd, offset_t size);
        typedef std::shared_ptr<Linker::Contents> file_reader1_type(const std::shared_ptr<Linker::Reader>& rd);
 
        void SetFileReader(std::shared_ptr<FileReader> file_reader);
        void SetFileReader(std::shared_ptr<Linker::Contents> (* file_reader)(const std::shared_ptr<Linker::Reader>& rd, offset_t size));
        void SetFileReader(std::shared_ptr<Linker::Contents> (* file_reader)(const std::shared_ptr<Linker::Reader>& rd));
 
        ArchiveFormat(std::shared_ptr<FileReader> file_reader = nullptr)
            : file_reader(file_reader)
        {
        }
 
        ArchiveFormat(std::shared_ptr<Linker::Contents> (* file_reader)(const std::shared_ptr<Linker::Reader>& rd, offset_t size));
        ArchiveFormat(std::shared_ptr<Linker::Contents> (* file_reader)(const std::shared_ptr<Linker::Reader>& rd));
 
        class File
        {
        public:
            std::string name;
            bool sysv_filename = false;
            offset_t extended_name_offset = 0;
            uint64_t modification = 0;
            uint32_t owner_id = 0, group_id = 0;
            uint32_t mode = 0;
            uint64_t size = 0;
            std::shared_ptr<Linker::Contents> contents;
        };
 
        std::vector<File> files;
 
        void ReadFile(const std::shared_ptr<Linker::Reader>& rd) override;
        using Linker::Format::WriteFile;
        offset_t WriteFile(const std::shared_ptr<Linker::Writer>& wr) const override;
        offset_t ImageSize() const override;
        void Dump(Dumper::Dumper& dump) const override;
        using Linker::InputFormat::GenerateModule;
        void GenerateModule(Linker::ModuleCollector& linker, std::string file_name, bool is_library = false) const override;
        void CalculateValues() override;
        // TODO
    };
}
 
#endif /* ARCH_H */