diff options
Diffstat (limited to 'runtime/ops/worker_host.rs')
-rw-r--r-- | runtime/ops/worker_host.rs | 43 |
1 files changed, 40 insertions, 3 deletions
diff --git a/runtime/ops/worker_host.rs b/runtime/ops/worker_host.rs index d8e60171e..2f297fb08 100644 --- a/runtime/ops/worker_host.rs +++ b/runtime/ops/worker_host.rs @@ -5,6 +5,7 @@ use crate::permissions::NetDescriptor; use crate::permissions::PermissionState; use crate::permissions::Permissions; use crate::permissions::ReadDescriptor; +use crate::permissions::RunDescriptor; use crate::permissions::UnaryPermission; use crate::permissions::UnitPermission; use crate::permissions::WriteDescriptor; @@ -189,6 +190,26 @@ fn merge_write_permission( Ok(main) } +fn merge_run_permission( + mut main: UnaryPermission<RunDescriptor>, + worker: Option<UnaryPermission<RunDescriptor>>, +) -> Result<UnaryPermission<RunDescriptor>, AnyError> { + if let Some(worker) = worker { + if (worker.global_state < main.global_state) + || !worker.granted_list.iter().all(|x| main.check(&x.0).is_ok()) + { + return Err(custom_error( + "PermissionDenied", + "Can't escalate parent thread permissions", + )); + } else { + main.global_state = worker.global_state; + main.granted_list = worker.granted_list; + } + } + Ok(main) +} + fn create_worker_permissions( main_perms: Permissions, worker_perms: PermissionsArg, @@ -199,7 +220,7 @@ fn create_worker_permissions( net: merge_net_permission(main_perms.net, worker_perms.net)?, plugin: merge_boolean_permission(main_perms.plugin, worker_perms.plugin)?, read: merge_read_permission(main_perms.read, worker_perms.read)?, - run: merge_boolean_permission(main_perms.run, worker_perms.run)?, + run: merge_run_permission(main_perms.run, worker_perms.run)?, write: merge_write_permission(main_perms.write, worker_perms.write)?, }) } @@ -216,8 +237,8 @@ struct PermissionsArg { plugin: Option<PermissionState>, #[serde(default, deserialize_with = "as_unary_read_permission")] read: Option<UnaryPermission<ReadDescriptor>>, - #[serde(default, deserialize_with = "as_permission_state")] - run: Option<PermissionState>, + #[serde(default, deserialize_with = "as_unary_run_permission")] + run: Option<UnaryPermission<RunDescriptor>>, #[serde(default, deserialize_with = "as_unary_write_permission")] write: Option<UnaryPermission<WriteDescriptor>>, } @@ -349,6 +370,22 @@ where })) } +fn as_unary_run_permission<'de, D>( + deserializer: D, +) -> Result<Option<UnaryPermission<RunDescriptor>>, D::Error> +where + D: Deserializer<'de>, +{ + let value: UnaryPermissionBase = + deserializer.deserialize_any(ParseBooleanOrStringVec)?; + + Ok(Some(UnaryPermission::<RunDescriptor> { + global_state: value.global_state, + granted_list: value.paths.into_iter().map(RunDescriptor).collect(), + ..Default::default() + })) +} + #[derive(Deserialize)] #[serde(rename_all = "camelCase")] pub struct CreateWorkerArgs { |