segment.h

#ifndef SEGMENT_H
#define SEGMENT_H
 
#include <iostream>
#include <memory>
#include <string>
#include <vector>
#include "../common.h"
#include "writable.h"
#include "writer.h"
 
namespace Linker
{
    class Section;
    class Writable;
 
    class Segment : public Writable, public std::enable_shared_from_this<Segment>
    {
    public:
        std::string name;
        std::vector<std::shared_ptr<Section>> sections;
        offset_t base_address;
        offset_t align = 1;
        offset_t data_size = 0;
        offset_t zero_fill = 0;
        offset_t optional_extra = 0;
 
        Segment(std::string name, offset_t base_address = 0)
            : name(name), base_address(base_address)
        {
        }
 
        bool IsMissing();
        void Fill();
        void RealignEnd(offset_t align);
        void Append(std::shared_ptr<Section> section);
        offset_t WriteFile(std::ostream& out, offset_t size, offset_t offset = 0);
        offset_t WriteFile(std::ostream& out);
        offset_t WriteFile(Writer& wr, offset_t count, offset_t offset = 0) override;
        offset_t WriteFile(Writer& wr);
        int GetByte(offset_t offset) override;
        offset_t TotalSize();
        offset_t ActualDataSize() override;
        offset_t GetStartAddress();
        offset_t GetEndAddress();
        void SetEndAddress(offset_t address);
        void AlignEndAddress(offset_t align);
    protected:
        void ShiftAddress(int64_t amount);
    public:
        void SetStartAddress(offset_t address);
    };
 
    std::ostream& operator<<(std::ostream& out, const Segment& segment);
}
 
#endif /* SEGMENT_H */