75 lines
1.8 KiB
C++
75 lines
1.8 KiB
C++
/*
|
|
* \brief Utility for tracking the allocation of dataspaces by the malloc heap
|
|
* \author Norman Feske
|
|
* \date 2019-08-20
|
|
*/
|
|
|
|
/*
|
|
* Copyright (C) 2016-2019 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.
|
|
*/
|
|
|
|
#ifndef _LIBC__INTERNAL__MALLOC_RAM_ALLOCATOR_H_
|
|
#define _LIBC__INTERNAL__MALLOC_RAM_ALLOCATOR_H_
|
|
|
|
/* Genode includes */
|
|
#include <base/registry.h>
|
|
#include <base/allocator.h>
|
|
#include <base/ram_allocator.h>
|
|
|
|
namespace Libc { struct Malloc_ram_allocator; }
|
|
|
|
|
|
struct Libc::Malloc_ram_allocator : Genode::Ram_allocator
|
|
{
|
|
Genode::Allocator &_md_alloc;
|
|
Genode::Ram_allocator &_ram;
|
|
|
|
struct Dataspace
|
|
{
|
|
Genode::Ram_dataspace_capability cap;
|
|
Dataspace(Genode::Ram_dataspace_capability cap) : cap(cap) { }
|
|
virtual ~Dataspace() { }
|
|
};
|
|
|
|
Genode::Registry<Genode::Registered<Dataspace> > _dataspaces { };
|
|
|
|
void _release(Genode::Registered<Dataspace> &ds)
|
|
{
|
|
_ram.free(ds.cap);
|
|
destroy(_md_alloc, &ds);
|
|
}
|
|
|
|
Malloc_ram_allocator(Allocator &md_alloc, Ram_allocator &ram)
|
|
: _md_alloc(md_alloc), _ram(ram) { }
|
|
|
|
~Malloc_ram_allocator()
|
|
{
|
|
_dataspaces.for_each([&] (Registered<Dataspace> &ds) {
|
|
_release(ds); });
|
|
}
|
|
|
|
Ram_dataspace_capability alloc(size_t size, Cache_attribute cached) override
|
|
{
|
|
Ram_dataspace_capability cap = _ram.alloc(size, cached);
|
|
new (_md_alloc) Registered<Dataspace>(_dataspaces, cap);
|
|
return cap;
|
|
}
|
|
|
|
void free(Ram_dataspace_capability ds_cap) override
|
|
{
|
|
_dataspaces.for_each([&] (Registered<Dataspace> &ds) {
|
|
if (ds_cap == ds.cap)
|
|
_release(ds); });
|
|
}
|
|
|
|
size_t dataspace_size(Ram_dataspace_capability ds_cap) const override
|
|
{
|
|
return _ram.dataspace_size(ds_cap);
|
|
}
|
|
};
|
|
|
|
#endif /* _LIBC__INTERNAL__MALLOC_RAM_ALLOCATOR_H_ */
|