/*
 * \brief   Protection-domain facility
 * \author  Martin Stein
 * \author  Stefan Kalkowski
 * \author  Sebastian Sumpf
 * \date    2012-02-12
 */

/*
 * Copyright (C) 2012-2017 Genode Labs GmbH
 *
 * This file is part of the Genode OS framework, which is distributed
 * under the terms of the GNU Affero General Public License version 3.
 */

/* Genode includes */
#include <root/root.h>

/* core includes */
#include <hw/assert.h>
#include <platform_pd.h>
#include <platform_thread.h>

using namespace Genode;
using Hw::Page_table;


/**************************************
 ** Hw::Address_space implementation **
 **************************************/

Core_mem_allocator * Hw::Address_space::_cma() {
	return static_cast<Core_mem_allocator*>(platform()->core_mem_alloc()); }


void * Hw::Address_space::_table_alloc()
{
	void * ret;
	if (!_cma()->alloc_aligned(sizeof(Page_table), (void**)&ret,
	                           Page_table::ALIGNM_LOG2).ok())
		throw Insufficient_ram_quota();
	return ret;
}


bool Hw::Address_space::insert_translation(addr_t virt, addr_t phys,
                                           size_t size, Page_flags flags)
{
	try {
		for (;;) {
			try {
				Lock::Guard guard(_lock);
				_tt.insert_translation(virt, phys, size, flags, _tt_alloc);
				return true;
			} catch(Hw::Out_of_tables &) {
				flush(platform()->vm_start(), platform()->vm_size());
			}
		}
	} catch(...) {
		error("invalid mapping ", Hex(phys), " -> ", Hex(virt), " (", size, ")");
	}
	return false;
}


void Hw::Address_space::flush(addr_t virt, size_t size)
{
	Lock::Guard guard(_lock);

	try {
		_tt.remove_translation(virt, size, _tt_alloc);

		/* update translation caches */
		Kernel::update_pd(&_kernel_pd);
	} catch(...) {
		error("tried to remove invalid region!");
	}
}


Hw::Address_space::Address_space(Kernel::Pd & pd, Page_table & tt,
                                 Page_table::Allocator & tt_alloc)
: _tt(tt), _tt_phys(Platform::core_phys_addr((addr_t)&tt)),
  _tt_alloc(tt_alloc), _kernel_pd(pd) {
	Kernel::mtc()->map(_tt, _tt_alloc); }


Hw::Address_space::Address_space(Kernel::Pd & pd)
: _tt(*construct_at<Page_table>(_table_alloc())),
  _tt_phys((addr_t)_cma()->phys_addr(&_tt)),
  _tt_array(new (_cma()) Array([this] (void * virt) {
    return (addr_t)_cma()->phys_addr(virt);})),
  _tt_alloc(_tt_array->alloc()),
  _kernel_pd(pd)
{
	Lock::Guard guard(_lock);
	Kernel::mtc()->map(_tt, _tt_alloc);
}


Hw::Address_space::~Address_space()
{
	flush(platform()->vm_start(), platform()->vm_size());
	destroy(_cma(), _tt_array);
	destroy(_cma(), &_tt);
}


/******************************
 ** Cap_space implementation **
 ******************************/

Cap_space::Cap_space() : _slab(nullptr, &_initial_sb) { }


void Cap_space::upgrade_slab(Allocator &alloc)
{
	void * block = nullptr;
	if (!alloc.alloc(SLAB_SIZE, &block))
		throw Out_of_ram();
	_slab.insert_sb(block);
}


/********************************
 ** Platform_pd implementation **
 ********************************/

bool Platform_pd::bind_thread(Platform_thread * t)
{
	/* is this the first and therefore main thread in this PD? */
	bool main_thread = !_thread_associated;
	_thread_associated = true;
	t->join_pd(this, main_thread, Address_space::weak_ptr());
	return true;
}


void Platform_pd::unbind_thread(Platform_thread *t) {
	t->join_pd(nullptr, false, Address_space::weak_ptr()); }


void Platform_pd::assign_parent(Native_capability parent)
{
	if (!_parent.valid() && parent.valid())
		_parent = parent;
}


Platform_pd::Platform_pd(Page_table & tt,
                         Page_table::Allocator & alloc)
: Hw::Address_space(*kernel_object(), tt, alloc),
  Kernel_object<Kernel::Pd>(false, (Page_table*)translation_table_phys(), this),
  _label("core") { }


Platform_pd::Platform_pd(Allocator * md_alloc, char const *label)
: Hw::Address_space(*kernel_object()),
  Kernel_object<Kernel::Pd>(true, (Page_table*)translation_table_phys(), this),
  _label(label)
{
	if (!_cap.valid()) {
		error("failed to create kernel object");
		throw Service_denied();
	}
}

Platform_pd::~Platform_pd()
{
	/* invalidate weak pointers to this object */
	Address_space::lock_for_destruction();
}


/*************************************
 ** Core_platform_pd implementation **
 *************************************/

extern int _mt_master_context_begin;

Core_platform_pd::Core_platform_pd()
: Platform_pd(Platform::core_page_table(),
              Platform::core_page_table_allocator())
{
	Genode::construct_at<Kernel::Cpu_context>(&_mt_master_context_begin,
	                                          (Page_table*)translation_table_phys());
}