segment.h

#ifndef SEGMENT_H
#define SEGMENT_H
 
#include <iostream>
#include <memory>
#include <string>
#include <vector>
#include "../common.h"
#include "image.h"
 
namespace Linker
{
    class Section;
    class Writer;
 
    class Segment : public ActualImage
    {
    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) const;
        offset_t WriteFile(std::ostream& out) const;
        offset_t WriteFile(Writer& wr, offset_t count, offset_t offset = 0) const override;
        offset_t WriteFile(Writer& wr) const override;
 
        size_t ReadData(size_t bytes, offset_t offset, void * buffer) const override;
 
        offset_t TotalSize();
        offset_t ImageSize() const 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);
    protected:
        std::shared_ptr<Segment> shared_from_this();
    };
 
    std::ostream& operator<<(std::ostream& out, const Segment& segment);
}
 
#endif /* SEGMENT_H */