relocation.h

#ifndef RELOCATION_H
#define RELOCATION_H
 
#include "../common.h"
#include "location.h"
#include "target.h"
 
namespace Linker
{
    class Resolution;
 
    class Relocation
    {
    public:
        enum reference_kind
        {
            Direct,
            SegmentAddress,
            SegmentIndex,
            SelectorIndex, // TODO: unimplemented
            GOTOffset, // TODO: unimplemented
            PLTOffset, // TODO: unimplemented
            SizeOf, // TODO: unimplemented
        }
        kind;
        size_t size;
        Location source;
        Target target;
        Target reference;
        uint64_t addend;
        EndianType endiantype;
        int shift;
        uint64_t mask;
        bool subtract;
 
        Relocation(reference_kind kind, size_t size, Location source, Target target, Target reference, uint64_t addend, EndianType endiantype)
            : kind(kind), size(size), source(source), target(target), reference(reference), addend(addend), endiantype(endiantype),
                shift(0), mask(-1), subtract(false)
        {
        }
 
        static Relocation Empty();
 
        static Relocation Absolute(size_t size, Location source, Target target, uint64_t addend, EndianType endiantype);
 
        static Relocation Absolute(size_t size, Location source, Target target, uint64_t addend = 0);
 
        static Relocation Offset(size_t size, Location source, Target target, uint64_t addend, EndianType endiantype);
 
        static Relocation Offset(size_t size, Location source, Target target, uint64_t addend = 0);
 
        static Relocation OffsetFrom(size_t size, Location source, Target target, Target reference, uint64_t addend, EndianType endiantype);
 
        static Relocation OffsetFrom(size_t size, Location source, Target target, Target reference, uint64_t addend = 0);
 
        static Relocation Relative(size_t size, Location source, Target target, uint64_t addend, EndianType endiantype);
 
        static Relocation Relative(size_t size, Location source, Target target, uint64_t addend = 0);
 
        static Relocation Paragraph(Location source, Target target, uint64_t addend = 0);
 
        static Relocation Selector(Location source, Target target, uint64_t addend = 0);
 
        static Relocation Segment(size_t size, Location source, Target target, uint64_t addend = 0);
 
        static Relocation ParagraphDifference(Location source, Target target, Target reference, uint64_t addend = 0);
 
        Relocation& SetMask(uint64_t new_mask);
 
        Relocation& SetShift(int new_shift);
 
        Relocation& SetSubtract(bool to_subtract = true);
 
        bool Displace(const Displacement& displacement);
 
        bool Resolve(Module& object, Resolution& resolution);
 
        uint64_t ReadUnsigned();
 
        int64_t ReadSigned();
 
        void WriteWord(uint64_t value);
 
        void AddCurrentValue();
 
        bool IsRelative() const;
    };
 
    std::ostream& operator<<(std::ostream& out, const Relocation& relocation);
}
 
#endif /* RELOCATION_H */