cb8abc3ae5
72f7ccd94 oaknut: 1.1.3 0b5745e4e oaknut: Add Windows on Arm support (#1) 5de40335d oaknut: 1.1.2 2952b759f oaknut: Correct MOV (UMOV alias) c90eb31ca oaknut: 1.1.1 7c777a28f oaknut: Fix ADR and ADRP 7470c7611 oaknut: Add ARMv8.2 instructions 9eb7cca88 oaknut: Update README 3fe32849a oaknut: 1.1.0 542128b51 oaknut: Add ARMv8.1 instructions 9acafdcdd oaknut: fpsimd MOV and UMOV corrections 636f91bac oaknut: MOV: Fix MOVN case 9cb332621 oaknut: Implement arranged accessors from DReg and QReg ba2dc2afe oaknut: dx 94bf56b08 oaknut: align aa7a3519f oaknut: Add dw 898f666ec oaknut: Add common system registers git-subtree-dir: externals/oaknut git-subtree-split: 72f7ccd9409dadf6a4ab98bad1fb11fbf0ca4d74
130 lines
3.6 KiB
C++
130 lines
3.6 KiB
C++
// SPDX-FileCopyrightText: Copyright (c) 2022 merryhime <https://mary.rs>
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#pragma once
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <variant>
|
|
|
|
namespace oaknut {
|
|
|
|
struct Label;
|
|
|
|
namespace detail {
|
|
|
|
constexpr std::uint64_t inverse_mask_from_size(std::size_t size)
|
|
{
|
|
return (~std::uint64_t{0}) << size;
|
|
}
|
|
|
|
constexpr std::uint64_t mask_from_size(std::size_t size)
|
|
{
|
|
return (~std::uint64_t{0}) >> (64 - size);
|
|
}
|
|
|
|
template<std::size_t bit_count>
|
|
constexpr std::uint64_t sign_extend(std::uint64_t value)
|
|
{
|
|
static_assert(bit_count != 0, "cannot sign-extend zero-sized value");
|
|
constexpr size_t shift_amount = 64 - bit_count;
|
|
return static_cast<std::uint64_t>(static_cast<std::int64_t>(value << shift_amount) >> shift_amount);
|
|
}
|
|
|
|
} // namespace detail
|
|
|
|
template<std::size_t bitsize, std::size_t alignment>
|
|
struct AddrOffset {
|
|
AddrOffset(std::ptrdiff_t diff)
|
|
: m_payload(encode(diff))
|
|
{}
|
|
|
|
AddrOffset(Label& label)
|
|
: m_payload(&label)
|
|
{}
|
|
|
|
AddrOffset(void* ptr)
|
|
: m_payload(ptr)
|
|
{}
|
|
|
|
static std::uint32_t encode(std::ptrdiff_t diff)
|
|
{
|
|
const std::uint64_t diff_u64 = static_cast<std::uint64_t>(diff);
|
|
if (detail::sign_extend<bitsize>(diff_u64) != diff_u64)
|
|
throw "out of range";
|
|
if (diff_u64 != (diff_u64 & detail::inverse_mask_from_size(alignment)))
|
|
throw "misalignment";
|
|
|
|
return static_cast<std::uint32_t>((diff_u64 & detail::mask_from_size(bitsize)) >> alignment);
|
|
}
|
|
|
|
private:
|
|
template<typename Policy>
|
|
friend class BasicCodeGenerator;
|
|
std::variant<std::uint32_t, Label*, void*> m_payload;
|
|
};
|
|
|
|
template<std::size_t bitsize, std::size_t shift_amount>
|
|
struct PageOffset {
|
|
PageOffset(void* ptr)
|
|
: m_payload(ptr)
|
|
{}
|
|
|
|
PageOffset(Label& label)
|
|
: m_payload(&label)
|
|
{}
|
|
|
|
static std::uint32_t encode(std::uintptr_t current_addr, std::uintptr_t target)
|
|
{
|
|
std::uint64_t diff = (static_cast<std::uint64_t>(target) >> shift_amount) - (static_cast<std::uint64_t>(current_addr) >> shift_amount);
|
|
if (detail::sign_extend<bitsize>(diff) != diff)
|
|
throw "out of range";
|
|
diff &= detail::mask_from_size(bitsize);
|
|
return static_cast<std::uint32_t>(((diff & 3) << (bitsize - 2)) | (diff >> 2));
|
|
}
|
|
|
|
private:
|
|
template<typename Policy>
|
|
friend class BasicCodeGenerator;
|
|
std::variant<Label*, void*> m_payload;
|
|
};
|
|
|
|
template<std::size_t bitsize, std::size_t alignment>
|
|
struct SOffset {
|
|
SOffset(std::int64_t offset)
|
|
{
|
|
const std::uint64_t diff_u64 = static_cast<std::uint64_t>(offset);
|
|
if (detail::sign_extend<bitsize>(diff_u64) != diff_u64)
|
|
throw "out of range";
|
|
if (diff_u64 != (diff_u64 & detail::inverse_mask_from_size(alignment)))
|
|
throw "misalignment";
|
|
|
|
m_encoded = static_cast<std::uint32_t>((diff_u64 & detail::mask_from_size(bitsize)) >> alignment);
|
|
}
|
|
|
|
private:
|
|
template<typename Policy>
|
|
friend class BasicCodeGenerator;
|
|
std::uint32_t m_encoded;
|
|
};
|
|
|
|
template<std::size_t bitsize, std::size_t alignment>
|
|
struct POffset {
|
|
POffset(std::int64_t offset)
|
|
{
|
|
const std::uint64_t diff_u64 = static_cast<std::uint64_t>(offset);
|
|
if (diff_u64 > detail::mask_from_size(bitsize))
|
|
throw "out of range";
|
|
if (diff_u64 != (diff_u64 & detail::inverse_mask_from_size(alignment)))
|
|
throw "misalignment";
|
|
|
|
m_encoded = static_cast<std::uint32_t>((diff_u64 & detail::mask_from_size(bitsize)) >> alignment);
|
|
}
|
|
|
|
private:
|
|
template<typename Policy>
|
|
friend class BasicCodeGenerator;
|
|
std::uint32_t m_encoded;
|
|
};
|
|
|
|
} // namespace oaknut
|