page_table_x64.h

Go to the documentation of this file.

//
// Bareflank Hypervisor
//
// Copyright (C) 2015 Assured Information Security, Inc.
// Author: Rian Quinn        <quinnr@ainfosec.com>
// Author: Brendan Kerrigan  <kerriganb@ainfosec.com>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

#ifndef PAGE_TABLE_X64_H
#define PAGE_TABLE_X64_H

#include <gsl/gsl>

#include <vector>
#include <memory>

#include <memory.h>
#include <memory_manager/page_table_entry_x64.h>

class page_table_x64
{
public:

    using pointer = uintptr_t *;
    using integer_pointer = uintptr_t;
    using size_type = std::size_t;
    using memory_descriptor_list = std::vector<memory_descriptor>;

    page_table_x64(gsl::not_null<pointer> pte);

    ~page_table_x64() = default;

    page_table_entry_x64 add_page_1g(integer_pointer addr)
    { return add_page(addr, x64::page_table::pml4::from, x64::page_table::pdpt::from); }

    page_table_entry_x64 add_page_2m(integer_pointer addr)
    { return add_page(addr, x64::page_table::pml4::from, x64::page_table::pd::from); }

    page_table_entry_x64 add_page_4k(integer_pointer addr)
    { return add_page(addr, x64::page_table::pml4::from, x64::page_table::pt::from); }

    void remove_page(integer_pointer addr)
    { remove_page(addr, x64::page_table::pml4::from); }

    page_table_entry_x64 virt_to_pte(integer_pointer addr) const
    { return virt_to_pte(addr, x64::page_table::pml4::from); }

    memory_descriptor_list pt_to_mdl() const
    { memory_descriptor_list mdl; return pt_to_mdl(mdl); }

private:

    page_table_entry_x64 add_page(integer_pointer addr, integer_pointer bits, integer_pointer end);
    void remove_page(integer_pointer addr, integer_pointer bits);
    page_table_entry_x64 virt_to_pte(integer_pointer addr, integer_pointer bits) const;
    memory_descriptor_list pt_to_mdl(memory_descriptor_list &mdl) const;

    bool empty() const noexcept;
    size_type global_size() const noexcept;
    size_type global_capacity() const noexcept;

private:

    friend class memory_manager_ut;

    std::unique_ptr<integer_pointer[]> m_pt;
    std::vector<std::unique_ptr<page_table_x64>> m_pts;

public:

    page_table_x64(page_table_x64 &&) noexcept = default;
    page_table_x64 &operator=(page_table_x64 &&) noexcept = default;

    page_table_x64(const page_table_x64 &) = delete;
    page_table_x64 &operator=(const page_table_x64 &) = delete;
};

#endif