pcos.h

#ifndef PCOS_H
#define PCOS_H
 
#include <array>
#include <memory>
#include <vector>
#include "../common.h"
#include "../linker/reader.h"
#include "../linker/segment_manager.h"
#include "../linker/writer.h"
#include "../dumper/dumper.h"
 
/* TODO: untested */
 
namespace PCOS
{
    class CMDFormat : public virtual Linker::SegmentManager
    {
    public:
        /* * * General members * * */
 
        class MemoryBlock
        {
        public:
            enum block_type : uint8_t
            {
                TYPE_LOAD = 3,
                TYPE_OFFSET_RELOCATION = 4,
                TYPE_SEGMENT_RELOCATION = 5,
                TYPE_END = 6,
            };
            block_type type;
 
            virtual ~MemoryBlock();
 
            virtual uint16_t GetLength() const;
            virtual void ReadFile(const std::shared_ptr<Linker::Reader>& rd, uint16_t length);
            virtual void WriteFile(const std::shared_ptr<Linker::Writer>& wr) const;
            virtual int GetDisplayOptions() const;
            virtual std::unique_ptr<Dumper::Region> MakeRegion(std::string name, offset_t offset, unsigned display_width) const;
            virtual void AddFields(Dumper::Region& region, const CMDFormat& module) const;
            virtual void DumpContents(Dumper::Dumper& dump, offset_t file_offset, const CMDFormat& module) const;
            void Dump(Dumper::Dumper& dump, offset_t file_offset, const CMDFormat& module) const;
 
            static std::unique_ptr<MemoryBlock> ReadFile(const std::shared_ptr<Linker::Reader>& rd);
 
            explicit MemoryBlock(int type)
                : type(block_type(type))
            {
            }
        };
 
        class LoadBlock : public MemoryBlock
        {
        public:
            LoadBlock()
                : MemoryBlock(TYPE_LOAD)
            {
            }
 
            uint32_t block_id;
            std::shared_ptr<Linker::Contents> image;
 
            uint16_t GetLength() const override;
            void ReadFile(const std::shared_ptr<Linker::Reader>& rd, uint16_t length) override;
            void WriteFile(const std::shared_ptr<Linker::Writer>& wr) const override;
            int GetDisplayOptions() const override;
            std::unique_ptr<Dumper::Region> MakeRegion(std::string name, offset_t offset, unsigned display_width) const override;
            void AddFields(Dumper::Region& region, const CMDFormat& module) const override;
        };
 
        class RelocationBlock : public MemoryBlock
        {
        public:
            uint8_t source = 0;
            uint8_t target = 0;
            std::vector<uint16_t> offsets;
 
            RelocationBlock(int type)
                : MemoryBlock(type)
            {
            }
 
            uint16_t GetLength() const override;
            void ReadFile(const std::shared_ptr<Linker::Reader>& rd, uint16_t length) override;
            void WriteFile(const std::shared_ptr<Linker::Writer>& wr) const override;
            int GetDisplayOptions() const override;
            void AddFields(Dumper::Region& region, const CMDFormat& module) const override;
            void DumpContents(Dumper::Dumper& dump, offset_t file_offset, const CMDFormat& module) const override;
        };
 
        class UnknownBlock : public MemoryBlock
        {
        public:
            UnknownBlock(int type)
                : MemoryBlock(type)
            {
            }
 
            std::shared_ptr<Linker::Contents> image;
 
            uint16_t GetLength() const override;
            void ReadFile(const std::shared_ptr<Linker::Reader>& rd, uint16_t length) override;
            void WriteFile(const std::shared_ptr<Linker::Writer>& wr) const override;
            std::unique_ptr<Dumper::Region> MakeRegion(std::string name, offset_t offset, unsigned display_width) const override;
        };
 
        uint16_t file_header_size = 0x40;
        std::array<char, 6> linker_version;
        enum file_type : uint8_t
        {
            TYPE_CMD = 1,
            TYPE_SAV = 9,
        };
        file_type type = TYPE_CMD;
        uint32_t entry_point;
        uint16_t stack_size;
        // TODO: exact meaning
        uint16_t allocation_length;
        std::vector<std::unique_ptr<MemoryBlock>> blocks;
 
        CMDFormat(int type = TYPE_CMD)
            : type(file_type(type))
        {
        }
 
        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;
 
        void Dump(Dumper::Dumper& dump) const override;
 
        void CalculateValues() override;
 
        using Linker::OutputFormat::GetDefaultExtension;
        std::string GetDefaultExtension(Linker::Module& module, std::string filename) const override;
 
        const LoadBlock * GetLoadBlockById(uint32_t block_id) const;
        LoadBlock * GetLoadBlockById(uint32_t block_id);
    };
}
 
#endif /* PCOS_H */