56 lines
1.6 KiB
C++
56 lines
1.6 KiB
C++
#pragma once
|
|
|
|
#include "capability/capability.h"
|
|
#include "lib/linked_list.h"
|
|
#include "lib/mutex.h"
|
|
#include "lib/ref_ptr.h"
|
|
|
|
class CapabilityTable {
|
|
public:
|
|
CapabilityTable();
|
|
|
|
CapabilityTable(CapabilityTable&) = delete;
|
|
CapabilityTable& operator=(CapabilityTable&) = delete;
|
|
|
|
template <typename T>
|
|
uint64_t AddNewCapability(const RefPtr<T>& object, uint64_t permissions);
|
|
uint64_t AddExistingCapability(const RefPtr<Capability>& cap);
|
|
|
|
// FIXME: Remove reliance on this.
|
|
template <typename T>
|
|
void AddNewCapabilityWithId(uint64_t id, const RefPtr<T>& object,
|
|
uint64_t permissions);
|
|
|
|
RefPtr<Capability> GetCapability(uint64_t id);
|
|
RefPtr<Capability> ReleaseCapability(uint64_t id);
|
|
|
|
private:
|
|
Mutex lock_{"cap table"};
|
|
// FIXME: store this id here rather than in the capability.
|
|
uint64_t next_cap_id_ = 0x100;
|
|
// FIXME: use a map data structure.
|
|
struct CapEntry {
|
|
uint64_t id;
|
|
RefPtr<Capability> cap;
|
|
};
|
|
LinkedList<CapEntry> capabilities_;
|
|
};
|
|
|
|
template <typename T>
|
|
uint64_t CapabilityTable::AddNewCapability(const RefPtr<T>& object,
|
|
uint64_t permissions) {
|
|
MutexHolder h(lock_);
|
|
uint64_t id = next_cap_id_++;
|
|
capabilities_.PushBack(
|
|
{.id = id, .cap = MakeRefCounted<Capability>(object, permissions)});
|
|
return id;
|
|
}
|
|
|
|
template <typename T>
|
|
void CapabilityTable::AddNewCapabilityWithId(uint64_t id,
|
|
const RefPtr<T>& object,
|
|
uint64_t permissions) {
|
|
capabilities_.PushBack(
|
|
{.id = id, .cap = MakeRefCounted<Capability>(object, permissions)});
|
|
}
|