/*
 * Copyright (c) 2021, Andreas Kling <kling@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <AK/String.h>
#include <LibJS/Bytecode/Block.h>
#include <LibJS/Bytecode/Op.h>
#include <sys/mman.h>

namespace JS::Bytecode {

NonnullOwnPtr<Block> Block::create()
{
    return adopt_own(*new Block);
}

Block::Block()
{
    // FIXME: This is not the smartest solution ever. Find something cleverer!
    // The main issue we're working around here is that we don't want pointers into the bytecode stream to become invalidated
    // during code generation due to dynamic buffer resizing. Otherwise we could just use a Vector.
    m_buffer_capacity = 64 * KiB;
    m_buffer = (u8*)mmap(nullptr, m_buffer_capacity, PROT_READ | PROT_WRITE, MAP_ANONYMOUS | MAP_PRIVATE, 0, 0);
    VERIFY(m_buffer != MAP_FAILED);
}

Block::~Block()
{
    unseal();
    Bytecode::InstructionStreamIterator it(instruction_stream());
    while (!it.at_end()) {
        auto& to_destroy = (*it);
        ++it;
        Instruction::destroy(const_cast<Instruction&>(to_destroy));
    }

    munmap(m_buffer, m_buffer_capacity);
}

void Block::seal() const
{
    if (mprotect(m_buffer, m_buffer_capacity, PROT_READ) < 0) {
        perror("ByteCode::Block::seal: mprotect");
        VERIFY_NOT_REACHED();
    }
}

void Block::unseal()
{
    if (mprotect(m_buffer, m_buffer_capacity, PROT_READ | PROT_WRITE) < 0) {
        perror("ByteCode::Block::unseal: mprotect");
        VERIFY_NOT_REACHED();
    }
}

void Block::dump() const
{
    Bytecode::InstructionStreamIterator it(instruction_stream());
    while (!it.at_end()) {
        warnln("[{:4x}] {}", it.offset(), (*it).to_string());
        ++it;
    }
}

void Block::grow(size_t additional_size)
{
    m_buffer_size += additional_size;
    VERIFY(m_buffer_size <= m_buffer_capacity);
}

void InstructionStreamIterator::operator++()
{
    m_offset += dereference().length();
}

}