platform_drv: account session capabilities with a Cap_quota_guard

The platform driver is a critical component and must not allow sessions
to deplete its own resource quotas.

Fix #2576
This commit is contained in:
Emery Hemingway 2017-11-12 19:40:13 -06:00 committed by Christian Helmuth
parent 0b60b8954b
commit de5c0603f1
2 changed files with 44 additions and 38 deletions

View File

@ -15,7 +15,7 @@
/* base */ /* base */
#include <base/env.h> #include <base/env.h>
#include <os/ram_session_guard.h> #include <base/quota_guard.h>
#include <region_map/client.h> #include <region_map/client.h>
#include <pd_session/connection.h> #include <pd_session/connection.h>
@ -41,16 +41,18 @@ class Platform::Device_pd
*/ */
struct Expanding_region_map_client : Genode::Region_map_client struct Expanding_region_map_client : Genode::Region_map_client
{ {
Genode::Env &_env; Genode::Env &_env;
Genode::Pd_connection &_pd; Genode::Pd_connection &_pd;
Genode::Ram_session_guard &ram_guard; Genode::Ram_quota_guard &_ram_guard;
Genode::Cap_quota_guard &_cap_guard;
Expanding_region_map_client(Genode::Env &env, Expanding_region_map_client(Genode::Env &env,
Genode::Pd_connection &pd, Genode::Pd_connection &pd,
Genode::Ram_session_guard &ram_guard) Genode::Ram_quota_guard &ram_guard,
Genode::Cap_quota_guard &cap_guard)
: :
Region_map_client(pd.address_space()), _env(env), _pd(pd), Region_map_client(pd.address_space()), _env(env), _pd(pd),
ram_guard(ram_guard) _ram_guard(ram_guard), _cap_guard(cap_guard)
{ } { }
Local_addr attach(Genode::Dataspace_capability ds, Local_addr attach(Genode::Dataspace_capability ds,
@ -69,22 +71,17 @@ class Platform::Device_pd
executable); }, executable); },
[&] () { [&] () {
enum { UPGRADE_CAP_QUOTA = 2 }; enum { UPGRADE_CAP_QUOTA = 2 };
/* XXX actually we would need here a ram_cap Genode::Cap_quota const caps { UPGRADE_CAP_QUOTA };
* guard per session, not the process global _cap_guard.withdraw(caps);
* cap env to check for enough caps */ _env.pd().transfer_quota(_pd, caps);
if (_env.pd().avail_caps().value < UPGRADE_CAP_QUOTA)
throw;
_env.pd().transfer_quota(_pd, Genode::Cap_quota{UPGRADE_CAP_QUOTA});
} }
); );
}, },
[&] () { [&] () {
enum { UPGRADE_RAM_QUOTA = 4096 }; enum { UPGRADE_RAM_QUOTA = 4096 };
if (!ram_guard.withdraw(UPGRADE_RAM_QUOTA)) Genode::Ram_quota const ram { UPGRADE_RAM_QUOTA };
throw; _ram_guard.withdraw(ram);
_env.pd().transfer_quota(_pd, ram);
_env.pd().transfer_quota(_pd, Genode::Ram_quota{UPGRADE_RAM_QUOTA});
} }
); );
} }
@ -101,11 +98,12 @@ class Platform::Device_pd
Device_pd(Genode::Env &env, Device_pd(Genode::Env &env,
Genode::Session_label const &label, Genode::Session_label const &label,
Genode::Ram_session_guard &ram_guard) Genode::Ram_quota_guard &ram_guard,
Genode::Cap_quota_guard &cap_guard)
: :
_pd(env, label.string(), Genode::Pd_connection::Virt_space::UNCONSTRAIN), _pd(env, label.string(), Genode::Pd_connection::Virt_space::UNCONSTRAIN),
_label(label), _label(label),
_address_space(env, _pd, ram_guard) _address_space(env, _pd, ram_guard, cap_guard)
{ {
_pd.ref_account(env.pd_session_cap()); _pd.ref_account(env.pd_session_cap());
} }

View File

@ -28,10 +28,10 @@
/* os */ /* os */
#include <io_mem_session/connection.h> #include <io_mem_session/connection.h>
#include <os/ram_session_guard.h>
#include <os/reporter.h> #include <os/reporter.h>
#include <os/session_policy.h> #include <os/session_policy.h>
#include <platform_session/platform_session.h> #include <platform_session/platform_session.h>
#include <base/allocator_guard.h>
/* local */ /* local */
#include "pci_device_component.h" #include "pci_device_component.h"
@ -202,16 +202,19 @@ class Platform::Session_component : public Genode::Rpc_object<Session>
{ {
private: private:
Genode::Env &_env; Genode::Env &_env;
Genode::Attached_rom_dataspace &_config; Genode::Attached_rom_dataspace &_config;
Genode::Ram_session_guard _env_ram; Genode::Ram_quota_guard _ram_guard;
Genode::Heap _md_alloc; Genode::Cap_quota_guard _cap_guard;
Genode::Session_label const _label; Genode::Constrained_ram_allocator _env_ram {
Genode::Session_policy const _policy { _label, _config.xml() }; _env.pd(), _ram_guard, _cap_guard };
Genode::List<Device_component> _device_list; Genode::Heap _md_alloc;
Platform::Pci_buses &_pci_bus; Genode::Session_label const _label;
Genode::Heap &_global_heap; Genode::Session_policy const _policy { _label, _config.xml() };
bool _no_device_pd = false; Genode::List<Device_component> _device_list;
Platform::Pci_buses &_pci_bus;
Genode::Heap &_global_heap;
bool _no_device_pd = false;
/** /**
* Registry of RAM dataspaces allocated by the session * Registry of RAM dataspaces allocated by the session
@ -236,7 +239,7 @@ class Platform::Session_component : public Genode::Rpc_object<Session>
return false; return false;
} }
Platform::Device_pd _device_pd { _env, _label, _env_ram }; Platform::Device_pd _device_pd { _env, _label, _ram_guard, _cap_guard };
enum { MAX_PCI_DEVICES = Device_config::MAX_BUSES * enum { MAX_PCI_DEVICES = Device_config::MAX_BUSES *
Device_config::MAX_DEVICES * Device_config::MAX_DEVICES *
@ -457,11 +460,15 @@ class Platform::Session_component : public Genode::Rpc_object<Session>
: :
_env(env), _env(env),
_config(config), _config(config),
_env_ram(env.ram(), Genode::ram_quota_from_args(args).value), _ram_guard(Genode::ram_quota_from_args(args)),
_cap_guard(Genode::cap_quota_from_args(args)),
_md_alloc(_env_ram, env.rm()), _md_alloc(_env_ram, env.rm()),
_label(Genode::label_from_args(args)), _label(Genode::label_from_args(args)),
_pci_bus(buses), _global_heap(global_heap) _pci_bus(buses), _global_heap(global_heap)
{ {
/* subtract the RPC session and session dataspace capabilities */
_cap_guard.withdraw(Genode::Cap_quota{2});
/* non-pci devices */ /* non-pci devices */
_policy.for_each_sub_node("device", [&] (Genode::Xml_node device_node) { _policy.for_each_sub_node("device", [&] (Genode::Xml_node device_node) {
try { try {
@ -578,7 +585,11 @@ class Platform::Session_component : public Genode::Rpc_object<Session>
} }
void upgrade_ram_quota(long quota) { _env_ram.upgrade(quota); } void upgrade_resources(Genode::Session::Resources resources)
{
_ram_guard.upgrade(resources.ram_quota);
_cap_guard.upgrade(resources.cap_quota);
}
static void add_config_space(Genode::uint32_t bdf_start, static void add_config_space(Genode::uint32_t bdf_start,
@ -981,11 +992,8 @@ class Platform::Root : public Genode::Root_component<Session_component>
} }
void _upgrade_session(Session_component *s, const char *args) override void _upgrade_session(Session_component *s, const char *args) override {
{ s->upgrade_resources(Genode::session_resources_from_args(args)); }
long ram_quota = Genode::Arg_string::find_arg(args, "ram_quota").long_value(0);
s->upgrade_ram_quota(ram_quota);
}
public: public: