mirror of
https://github.com/denoland/deno.git
synced 2024-11-24 15:19:26 -05:00
01f3451869
A few small changes to avoid needing unsafe mem transmutes to instantiate the extensions --------- Signed-off-by: Richard Carson <Rscarson@rogers.com>
5014 lines
142 KiB
Rust
5014 lines
142 KiB
Rust
// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license.
|
||
|
||
use deno_core::parking_lot::Mutex;
|
||
use deno_core::serde::de;
|
||
use deno_core::serde::Deserialize;
|
||
use deno_core::serde::Deserializer;
|
||
use deno_core::serde::Serialize;
|
||
use deno_core::serde_json;
|
||
use deno_core::unsync::sync::AtomicFlag;
|
||
use deno_core::url::Url;
|
||
use deno_core::ModuleSpecifier;
|
||
use deno_path_util::normalize_path;
|
||
use deno_path_util::url_to_file_path;
|
||
use deno_terminal::colors;
|
||
use fqdn::FQDN;
|
||
use once_cell::sync::Lazy;
|
||
use std::borrow::Cow;
|
||
use std::collections::HashSet;
|
||
use std::ffi::OsStr;
|
||
use std::fmt;
|
||
use std::fmt::Debug;
|
||
use std::hash::Hash;
|
||
use std::net::IpAddr;
|
||
use std::net::Ipv6Addr;
|
||
use std::path::Path;
|
||
use std::path::PathBuf;
|
||
use std::string::ToString;
|
||
use std::sync::Arc;
|
||
|
||
pub mod prompter;
|
||
use prompter::permission_prompt;
|
||
use prompter::PERMISSION_EMOJI;
|
||
|
||
pub use prompter::set_prompt_callbacks;
|
||
pub use prompter::set_prompter;
|
||
pub use prompter::PermissionPrompter;
|
||
pub use prompter::PromptCallback;
|
||
pub use prompter::PromptResponse;
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
#[error("Requires {access}, {}", format_permission_error(.name))]
|
||
pub struct PermissionDeniedError {
|
||
pub access: String,
|
||
pub name: &'static str,
|
||
}
|
||
|
||
fn format_permission_error(name: &'static str) -> String {
|
||
if is_standalone() {
|
||
format!("specify the required permissions during compilation using `deno compile --allow-{name}`")
|
||
} else {
|
||
format!("run again with the --allow-{name} flag")
|
||
}
|
||
}
|
||
|
||
/// Fast exit from permission check routines if this permission
|
||
/// is in the "fully-granted" state.
|
||
macro_rules! skip_check_if_is_permission_fully_granted {
|
||
($this:ident) => {
|
||
if $this.is_allow_all() {
|
||
return Ok(());
|
||
}
|
||
};
|
||
}
|
||
|
||
#[inline]
|
||
fn resolve_from_known_cwd(path: &Path, cwd: &Path) -> PathBuf {
|
||
if path.is_absolute() {
|
||
normalize_path(path)
|
||
} else {
|
||
normalize_path(cwd.join(path))
|
||
}
|
||
}
|
||
|
||
static DEBUG_LOG_ENABLED: Lazy<bool> =
|
||
Lazy::new(|| log::log_enabled!(log::Level::Debug));
|
||
|
||
/// Quadri-state value for storing permission state
|
||
#[derive(
|
||
Eq, PartialEq, Default, Debug, Clone, Copy, Deserialize, PartialOrd,
|
||
)]
|
||
pub enum PermissionState {
|
||
Granted = 0,
|
||
GrantedPartial = 1,
|
||
#[default]
|
||
Prompt = 2,
|
||
Denied = 3,
|
||
}
|
||
|
||
/// `AllowPartial` prescribes how to treat a permission which is partially
|
||
/// denied due to a `--deny-*` flag affecting a subscope of the queried
|
||
/// permission.
|
||
///
|
||
/// `TreatAsGranted` is used in place of `TreatAsPartialGranted` when we don't
|
||
/// want to wastefully check for partial denials when, say, checking read
|
||
/// access for a file.
|
||
#[derive(Debug, Eq, PartialEq)]
|
||
#[allow(clippy::enum_variant_names)]
|
||
enum AllowPartial {
|
||
TreatAsGranted,
|
||
TreatAsDenied,
|
||
TreatAsPartialGranted,
|
||
}
|
||
|
||
impl From<bool> for AllowPartial {
|
||
fn from(value: bool) -> Self {
|
||
if value {
|
||
Self::TreatAsGranted
|
||
} else {
|
||
Self::TreatAsDenied
|
||
}
|
||
}
|
||
}
|
||
|
||
impl PermissionState {
|
||
#[inline(always)]
|
||
fn log_perm_access(
|
||
name: &'static str,
|
||
info: impl FnOnce() -> Option<String>,
|
||
) {
|
||
// Eliminates log overhead (when logging is disabled),
|
||
// log_enabled!(Debug) check in a hot path still has overhead
|
||
// TODO(AaronO): generalize or upstream this optimization
|
||
if *DEBUG_LOG_ENABLED {
|
||
log::debug!(
|
||
"{}",
|
||
colors::bold(&format!(
|
||
"{}️ Granted {}",
|
||
PERMISSION_EMOJI,
|
||
Self::fmt_access(name, info)
|
||
))
|
||
);
|
||
}
|
||
}
|
||
|
||
fn fmt_access(
|
||
name: &'static str,
|
||
info: impl FnOnce() -> Option<String>,
|
||
) -> String {
|
||
format!(
|
||
"{} access{}",
|
||
name,
|
||
info().map(|info| format!(" to {info}")).unwrap_or_default(),
|
||
)
|
||
}
|
||
|
||
fn error(
|
||
name: &'static str,
|
||
info: impl FnOnce() -> Option<String>,
|
||
) -> PermissionDeniedError {
|
||
PermissionDeniedError {
|
||
access: Self::fmt_access(name, info),
|
||
name,
|
||
}
|
||
}
|
||
|
||
/// Check the permission state. bool is whether a prompt was issued.
|
||
#[inline]
|
||
fn check(
|
||
self,
|
||
name: &'static str,
|
||
api_name: Option<&str>,
|
||
info: Option<&str>,
|
||
prompt: bool,
|
||
) -> (Result<(), PermissionDeniedError>, bool, bool) {
|
||
self.check2(name, api_name, || info.map(|s| s.to_string()), prompt)
|
||
}
|
||
|
||
#[inline]
|
||
fn check2(
|
||
self,
|
||
name: &'static str,
|
||
api_name: Option<&str>,
|
||
info: impl Fn() -> Option<String>,
|
||
prompt: bool,
|
||
) -> (Result<(), PermissionDeniedError>, bool, bool) {
|
||
match self {
|
||
PermissionState::Granted => {
|
||
Self::log_perm_access(name, info);
|
||
(Ok(()), false, false)
|
||
}
|
||
PermissionState::Prompt if prompt => {
|
||
let msg = format!(
|
||
"{} access{}",
|
||
name,
|
||
info()
|
||
.map(|info| { format!(" to {info}") })
|
||
.unwrap_or_default(),
|
||
);
|
||
match permission_prompt(&msg, name, api_name, true) {
|
||
PromptResponse::Allow => {
|
||
Self::log_perm_access(name, info);
|
||
(Ok(()), true, false)
|
||
}
|
||
PromptResponse::AllowAll => {
|
||
Self::log_perm_access(name, info);
|
||
(Ok(()), true, true)
|
||
}
|
||
PromptResponse::Deny => (Err(Self::error(name, info)), true, false),
|
||
}
|
||
}
|
||
_ => (Err(Self::error(name, info)), false, false),
|
||
}
|
||
}
|
||
}
|
||
|
||
impl fmt::Display for PermissionState {
|
||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||
match self {
|
||
PermissionState::Granted => f.pad("granted"),
|
||
PermissionState::GrantedPartial => f.pad("granted-partial"),
|
||
PermissionState::Prompt => f.pad("prompt"),
|
||
PermissionState::Denied => f.pad("denied"),
|
||
}
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Debug, Eq, PartialEq)]
|
||
pub struct UnitPermission {
|
||
pub name: &'static str,
|
||
pub description: &'static str,
|
||
pub state: PermissionState,
|
||
pub prompt: bool,
|
||
}
|
||
|
||
impl UnitPermission {
|
||
pub fn query(&self) -> PermissionState {
|
||
self.state
|
||
}
|
||
|
||
pub fn request(&mut self) -> PermissionState {
|
||
if self.state == PermissionState::Prompt {
|
||
if PromptResponse::Allow
|
||
== permission_prompt(
|
||
&format!("access to {}", self.description),
|
||
self.name,
|
||
Some("Deno.permissions.query()"),
|
||
false,
|
||
)
|
||
{
|
||
self.state = PermissionState::Granted;
|
||
} else {
|
||
self.state = PermissionState::Denied;
|
||
}
|
||
}
|
||
self.state
|
||
}
|
||
|
||
pub fn revoke(&mut self) -> PermissionState {
|
||
if self.state == PermissionState::Granted {
|
||
self.state = PermissionState::Prompt;
|
||
}
|
||
self.state
|
||
}
|
||
|
||
pub fn check(&mut self) -> Result<(), PermissionDeniedError> {
|
||
let (result, prompted, _is_allow_all) =
|
||
self.state.check(self.name, None, None, self.prompt);
|
||
if prompted {
|
||
if result.is_ok() {
|
||
self.state = PermissionState::Granted;
|
||
} else {
|
||
self.state = PermissionState::Denied;
|
||
}
|
||
}
|
||
result
|
||
}
|
||
|
||
fn create_child_permissions(
|
||
&mut self,
|
||
flag: ChildUnitPermissionArg,
|
||
) -> Result<Self, ChildPermissionError> {
|
||
let mut perm = self.clone();
|
||
match flag {
|
||
ChildUnitPermissionArg::Inherit => {
|
||
// copy
|
||
}
|
||
ChildUnitPermissionArg::Granted => {
|
||
if self.check().is_err() {
|
||
return Err(ChildPermissionError::Escalation);
|
||
}
|
||
perm.state = PermissionState::Granted;
|
||
}
|
||
ChildUnitPermissionArg::NotGranted => {
|
||
perm.state = PermissionState::Prompt;
|
||
}
|
||
}
|
||
if self.state == PermissionState::Denied {
|
||
perm.state = PermissionState::Denied;
|
||
}
|
||
Ok(perm)
|
||
}
|
||
}
|
||
|
||
/// A normalized environment variable name. On Windows this will
|
||
/// be uppercase and on other platforms it will stay as-is.
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
struct EnvVarName {
|
||
inner: String,
|
||
}
|
||
|
||
impl EnvVarName {
|
||
pub fn new(env: impl AsRef<str>) -> Self {
|
||
Self {
|
||
inner: if cfg!(windows) {
|
||
env.as_ref().to_uppercase()
|
||
} else {
|
||
env.as_ref().to_string()
|
||
},
|
||
}
|
||
}
|
||
}
|
||
|
||
impl AsRef<str> for EnvVarName {
|
||
fn as_ref(&self) -> &str {
|
||
self.inner.as_str()
|
||
}
|
||
}
|
||
|
||
pub trait QueryDescriptor: Debug {
|
||
type AllowDesc: Debug + Eq + Clone + Hash;
|
||
type DenyDesc: Debug + Eq + Clone + Hash;
|
||
|
||
fn flag_name() -> &'static str;
|
||
fn display_name(&self) -> Cow<str>;
|
||
|
||
fn from_allow(allow: &Self::AllowDesc) -> Self;
|
||
|
||
fn as_allow(&self) -> Option<Self::AllowDesc>;
|
||
fn as_deny(&self) -> Self::DenyDesc;
|
||
|
||
/// Generic check function to check this descriptor against a `UnaryPermission`.
|
||
fn check_in_permission(
|
||
&self,
|
||
perm: &mut UnaryPermission<Self>,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError>;
|
||
|
||
fn matches_allow(&self, other: &Self::AllowDesc) -> bool;
|
||
fn matches_deny(&self, other: &Self::DenyDesc) -> bool;
|
||
|
||
/// Gets if this query descriptor should revoke the provided allow descriptor.
|
||
fn revokes(&self, other: &Self::AllowDesc) -> bool;
|
||
fn stronger_than_deny(&self, other: &Self::DenyDesc) -> bool;
|
||
fn overlaps_deny(&self, other: &Self::DenyDesc) -> bool;
|
||
}
|
||
|
||
fn format_display_name(display_name: Cow<str>) -> String {
|
||
if display_name.starts_with('<') && display_name.ends_with('>') {
|
||
display_name.into_owned()
|
||
} else {
|
||
format!("\"{}\"", display_name)
|
||
}
|
||
}
|
||
|
||
#[derive(Debug, Eq, PartialEq)]
|
||
pub struct UnaryPermission<TQuery: QueryDescriptor + ?Sized> {
|
||
granted_global: bool,
|
||
granted_list: HashSet<TQuery::AllowDesc>,
|
||
flag_denied_global: bool,
|
||
flag_denied_list: HashSet<TQuery::DenyDesc>,
|
||
prompt_denied_global: bool,
|
||
prompt_denied_list: HashSet<TQuery::DenyDesc>,
|
||
prompt: bool,
|
||
}
|
||
|
||
impl<TQuery: QueryDescriptor> Default for UnaryPermission<TQuery> {
|
||
fn default() -> Self {
|
||
UnaryPermission {
|
||
granted_global: Default::default(),
|
||
granted_list: Default::default(),
|
||
flag_denied_global: Default::default(),
|
||
flag_denied_list: Default::default(),
|
||
prompt_denied_global: Default::default(),
|
||
prompt_denied_list: Default::default(),
|
||
prompt: Default::default(),
|
||
}
|
||
}
|
||
}
|
||
|
||
impl<TQuery: QueryDescriptor> Clone for UnaryPermission<TQuery> {
|
||
fn clone(&self) -> Self {
|
||
Self {
|
||
granted_global: self.granted_global,
|
||
granted_list: self.granted_list.clone(),
|
||
flag_denied_global: self.flag_denied_global,
|
||
flag_denied_list: self.flag_denied_list.clone(),
|
||
prompt_denied_global: self.prompt_denied_global,
|
||
prompt_denied_list: self.prompt_denied_list.clone(),
|
||
prompt: self.prompt,
|
||
}
|
||
}
|
||
}
|
||
|
||
impl<TQuery: QueryDescriptor> UnaryPermission<TQuery> {
|
||
pub fn allow_all() -> Self {
|
||
Self {
|
||
granted_global: true,
|
||
..Default::default()
|
||
}
|
||
}
|
||
|
||
pub fn is_allow_all(&self) -> bool {
|
||
self.granted_global
|
||
&& self.flag_denied_list.is_empty()
|
||
&& self.prompt_denied_list.is_empty()
|
||
}
|
||
|
||
pub fn check_all_api(
|
||
&mut self,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(None, false, api_name)
|
||
}
|
||
|
||
fn check_desc(
|
||
&mut self,
|
||
desc: Option<&TQuery>,
|
||
assert_non_partial: bool,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
let (result, prompted, is_allow_all) = self
|
||
.query_desc(desc, AllowPartial::from(!assert_non_partial))
|
||
.check2(
|
||
TQuery::flag_name(),
|
||
api_name,
|
||
|| desc.map(|d| format_display_name(d.display_name())),
|
||
self.prompt,
|
||
);
|
||
if prompted {
|
||
if result.is_ok() {
|
||
if is_allow_all {
|
||
self.insert_granted(None);
|
||
} else {
|
||
self.insert_granted(desc);
|
||
}
|
||
} else {
|
||
self.insert_prompt_denied(desc.map(|d| d.as_deny()));
|
||
}
|
||
}
|
||
result
|
||
}
|
||
|
||
fn query_desc(
|
||
&self,
|
||
desc: Option<&TQuery>,
|
||
allow_partial: AllowPartial,
|
||
) -> PermissionState {
|
||
if self.is_flag_denied(desc) || self.is_prompt_denied(desc) {
|
||
PermissionState::Denied
|
||
} else if self.is_granted(desc) {
|
||
match allow_partial {
|
||
AllowPartial::TreatAsGranted => PermissionState::Granted,
|
||
AllowPartial::TreatAsDenied => {
|
||
if self.is_partial_flag_denied(desc) {
|
||
PermissionState::Denied
|
||
} else {
|
||
PermissionState::Granted
|
||
}
|
||
}
|
||
AllowPartial::TreatAsPartialGranted => {
|
||
if self.is_partial_flag_denied(desc) {
|
||
PermissionState::GrantedPartial
|
||
} else {
|
||
PermissionState::Granted
|
||
}
|
||
}
|
||
}
|
||
} else if matches!(allow_partial, AllowPartial::TreatAsDenied)
|
||
&& self.is_partial_flag_denied(desc)
|
||
{
|
||
PermissionState::Denied
|
||
} else {
|
||
PermissionState::Prompt
|
||
}
|
||
}
|
||
|
||
fn request_desc(&mut self, desc: Option<&TQuery>) -> PermissionState {
|
||
let state = self.query_desc(desc, AllowPartial::TreatAsPartialGranted);
|
||
if state == PermissionState::Granted {
|
||
self.insert_granted(desc);
|
||
return state;
|
||
}
|
||
if state != PermissionState::Prompt {
|
||
return state;
|
||
}
|
||
if !self.prompt {
|
||
return PermissionState::Denied;
|
||
}
|
||
let mut message = String::with_capacity(40);
|
||
message.push_str(&format!("{} access", TQuery::flag_name()));
|
||
if let Some(desc) = desc {
|
||
message
|
||
.push_str(&format!(" to {}", format_display_name(desc.display_name())));
|
||
}
|
||
match permission_prompt(
|
||
&message,
|
||
TQuery::flag_name(),
|
||
Some("Deno.permissions.request()"),
|
||
true,
|
||
) {
|
||
PromptResponse::Allow => {
|
||
self.insert_granted(desc);
|
||
PermissionState::Granted
|
||
}
|
||
PromptResponse::Deny => {
|
||
self.insert_prompt_denied(desc.map(|d| d.as_deny()));
|
||
PermissionState::Denied
|
||
}
|
||
PromptResponse::AllowAll => {
|
||
self.insert_granted(None);
|
||
PermissionState::Granted
|
||
}
|
||
}
|
||
}
|
||
|
||
fn revoke_desc(&mut self, desc: Option<&TQuery>) -> PermissionState {
|
||
match desc {
|
||
Some(desc) => {
|
||
self.granted_list.retain(|v| !desc.revokes(v));
|
||
}
|
||
None => {
|
||
self.granted_global = false;
|
||
// Revoke global is a special case where the entire granted list is
|
||
// cleared. It's inconsistent with the granular case where only
|
||
// descriptors stronger than the revoked one are purged.
|
||
self.granted_list.clear();
|
||
}
|
||
}
|
||
self.query_desc(desc, AllowPartial::TreatAsPartialGranted)
|
||
}
|
||
|
||
fn is_granted(&self, query: Option<&TQuery>) -> bool {
|
||
match query {
|
||
Some(query) => {
|
||
self.granted_global
|
||
|| self.granted_list.iter().any(|v| query.matches_allow(v))
|
||
}
|
||
None => self.granted_global,
|
||
}
|
||
}
|
||
|
||
fn is_flag_denied(&self, query: Option<&TQuery>) -> bool {
|
||
match query {
|
||
Some(query) => {
|
||
self.flag_denied_global
|
||
|| self.flag_denied_list.iter().any(|v| query.matches_deny(v))
|
||
}
|
||
None => self.flag_denied_global,
|
||
}
|
||
}
|
||
|
||
fn is_prompt_denied(&self, query: Option<&TQuery>) -> bool {
|
||
match query {
|
||
Some(query) => self
|
||
.prompt_denied_list
|
||
.iter()
|
||
.any(|v| query.stronger_than_deny(v)),
|
||
None => self.prompt_denied_global || !self.prompt_denied_list.is_empty(),
|
||
}
|
||
}
|
||
|
||
fn is_partial_flag_denied(&self, query: Option<&TQuery>) -> bool {
|
||
match query {
|
||
None => !self.flag_denied_list.is_empty(),
|
||
Some(query) => {
|
||
self.flag_denied_list.iter().any(|v| query.overlaps_deny(v))
|
||
}
|
||
}
|
||
}
|
||
|
||
fn insert_granted(&mut self, query: Option<&TQuery>) -> bool {
|
||
let desc = match query.map(|q| q.as_allow()) {
|
||
Some(Some(allow_desc)) => Some(allow_desc),
|
||
Some(None) => {
|
||
// the user was prompted for this descriptor in order to not
|
||
// expose anything about the system to the program, but the
|
||
// descriptor wasn't valid so no permission was raised
|
||
return false;
|
||
}
|
||
None => None,
|
||
};
|
||
Self::list_insert(desc, &mut self.granted_global, &mut self.granted_list);
|
||
true
|
||
}
|
||
|
||
fn insert_prompt_denied(&mut self, desc: Option<TQuery::DenyDesc>) {
|
||
Self::list_insert(
|
||
desc,
|
||
&mut self.prompt_denied_global,
|
||
&mut self.prompt_denied_list,
|
||
);
|
||
}
|
||
|
||
fn list_insert<T: Hash + Eq>(
|
||
desc: Option<T>,
|
||
list_global: &mut bool,
|
||
list: &mut HashSet<T>,
|
||
) {
|
||
match desc {
|
||
Some(desc) => {
|
||
list.insert(desc);
|
||
}
|
||
None => *list_global = true,
|
||
}
|
||
}
|
||
|
||
fn create_child_permissions<E>(
|
||
&mut self,
|
||
flag: ChildUnaryPermissionArg,
|
||
parse: impl Fn(&str) -> Result<Option<TQuery::AllowDesc>, E>,
|
||
) -> Result<UnaryPermission<TQuery>, ChildPermissionError>
|
||
where
|
||
ChildPermissionError: From<E>,
|
||
{
|
||
let mut perms = Self::default();
|
||
|
||
match flag {
|
||
ChildUnaryPermissionArg::Inherit => {
|
||
perms.clone_from(self);
|
||
}
|
||
ChildUnaryPermissionArg::Granted => {
|
||
if self.check_all_api(None).is_err() {
|
||
return Err(ChildPermissionError::Escalation);
|
||
}
|
||
perms.granted_global = true;
|
||
}
|
||
ChildUnaryPermissionArg::NotGranted => {}
|
||
ChildUnaryPermissionArg::GrantedList(granted_list) => {
|
||
perms.granted_list = granted_list
|
||
.iter()
|
||
.filter_map(|i| parse(i).transpose())
|
||
.collect::<Result<_, E>>()?;
|
||
if !perms.granted_list.iter().all(|desc| {
|
||
TQuery::from_allow(desc)
|
||
.check_in_permission(self, None)
|
||
.is_ok()
|
||
}) {
|
||
return Err(ChildPermissionError::Escalation);
|
||
}
|
||
}
|
||
}
|
||
perms.flag_denied_global = self.flag_denied_global;
|
||
perms.prompt_denied_global = self.prompt_denied_global;
|
||
perms.prompt = self.prompt;
|
||
perms.flag_denied_list.clone_from(&self.flag_denied_list);
|
||
perms
|
||
.prompt_denied_list
|
||
.clone_from(&self.prompt_denied_list);
|
||
|
||
Ok(perms)
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct PathQueryDescriptor {
|
||
pub requested: String,
|
||
pub resolved: PathBuf,
|
||
}
|
||
|
||
impl PathQueryDescriptor {
|
||
pub fn into_ffi(self) -> FfiQueryDescriptor {
|
||
FfiQueryDescriptor(self)
|
||
}
|
||
|
||
pub fn into_read(self) -> ReadQueryDescriptor {
|
||
ReadQueryDescriptor(self)
|
||
}
|
||
|
||
pub fn into_write(self) -> WriteQueryDescriptor {
|
||
WriteQueryDescriptor(self)
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct ReadQueryDescriptor(pub PathQueryDescriptor);
|
||
|
||
impl QueryDescriptor for ReadQueryDescriptor {
|
||
type AllowDesc = ReadDescriptor;
|
||
type DenyDesc = ReadDescriptor;
|
||
|
||
fn flag_name() -> &'static str {
|
||
"read"
|
||
}
|
||
|
||
fn display_name(&self) -> Cow<str> {
|
||
Cow::Borrowed(self.0.requested.as_str())
|
||
}
|
||
|
||
fn from_allow(allow: &Self::AllowDesc) -> Self {
|
||
PathQueryDescriptor {
|
||
requested: allow.0.to_string_lossy().into_owned(),
|
||
resolved: allow.0.clone(),
|
||
}
|
||
.into_read()
|
||
}
|
||
|
||
fn as_allow(&self) -> Option<Self::AllowDesc> {
|
||
Some(ReadDescriptor(self.0.resolved.clone()))
|
||
}
|
||
|
||
fn as_deny(&self) -> Self::DenyDesc {
|
||
ReadDescriptor(self.0.resolved.clone())
|
||
}
|
||
|
||
fn check_in_permission(
|
||
&self,
|
||
perm: &mut UnaryPermission<Self>,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(perm);
|
||
perm.check_desc(Some(self), true, api_name)
|
||
}
|
||
|
||
fn matches_allow(&self, other: &Self::AllowDesc) -> bool {
|
||
self.0.resolved.starts_with(&other.0)
|
||
}
|
||
|
||
fn matches_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.0.resolved.starts_with(&other.0)
|
||
}
|
||
|
||
fn revokes(&self, other: &Self::AllowDesc) -> bool {
|
||
self.matches_allow(other)
|
||
}
|
||
|
||
fn stronger_than_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
other.0.starts_with(&self.0.resolved)
|
||
}
|
||
|
||
fn overlaps_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.stronger_than_deny(other)
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct ReadDescriptor(pub PathBuf);
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct WriteQueryDescriptor(pub PathQueryDescriptor);
|
||
|
||
impl QueryDescriptor for WriteQueryDescriptor {
|
||
type AllowDesc = WriteDescriptor;
|
||
type DenyDesc = WriteDescriptor;
|
||
|
||
fn flag_name() -> &'static str {
|
||
"write"
|
||
}
|
||
|
||
fn display_name(&self) -> Cow<str> {
|
||
Cow::Borrowed(&self.0.requested)
|
||
}
|
||
|
||
fn from_allow(allow: &Self::AllowDesc) -> Self {
|
||
WriteQueryDescriptor(PathQueryDescriptor {
|
||
requested: allow.0.to_string_lossy().into_owned(),
|
||
resolved: allow.0.clone(),
|
||
})
|
||
}
|
||
|
||
fn as_allow(&self) -> Option<Self::AllowDesc> {
|
||
Some(WriteDescriptor(self.0.resolved.clone()))
|
||
}
|
||
|
||
fn as_deny(&self) -> Self::DenyDesc {
|
||
WriteDescriptor(self.0.resolved.clone())
|
||
}
|
||
|
||
fn check_in_permission(
|
||
&self,
|
||
perm: &mut UnaryPermission<Self>,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(perm);
|
||
perm.check_desc(Some(self), true, api_name)
|
||
}
|
||
|
||
fn matches_allow(&self, other: &Self::AllowDesc) -> bool {
|
||
self.0.resolved.starts_with(&other.0)
|
||
}
|
||
|
||
fn matches_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.0.resolved.starts_with(&other.0)
|
||
}
|
||
|
||
fn revokes(&self, other: &Self::AllowDesc) -> bool {
|
||
self.matches_allow(other)
|
||
}
|
||
|
||
fn stronger_than_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
other.0.starts_with(&self.0.resolved)
|
||
}
|
||
|
||
fn overlaps_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.stronger_than_deny(other)
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct WriteDescriptor(pub PathBuf);
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub enum Host {
|
||
Fqdn(FQDN),
|
||
Ip(IpAddr),
|
||
}
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
pub enum HostParseError {
|
||
#[error("invalid IPv6 address: '{0}'")]
|
||
InvalidIpv6(String),
|
||
#[error("invalid host: '{0}'")]
|
||
InvalidHost(String),
|
||
#[error("invalid empty host: '{0}'")]
|
||
InvalidEmptyHost(String),
|
||
#[error("invalid host '{host}': {error}")]
|
||
Fqdn {
|
||
#[source]
|
||
error: fqdn::Error,
|
||
host: String,
|
||
},
|
||
}
|
||
|
||
impl Host {
|
||
fn parse(s: &str) -> Result<Self, HostParseError> {
|
||
if s.starts_with('[') && s.ends_with(']') {
|
||
let ip = s[1..s.len() - 1]
|
||
.parse::<Ipv6Addr>()
|
||
.map_err(|_| HostParseError::InvalidIpv6(s.to_string()))?;
|
||
return Ok(Host::Ip(IpAddr::V6(ip)));
|
||
}
|
||
let (without_trailing_dot, has_trailing_dot) =
|
||
s.strip_suffix('.').map_or((s, false), |s| (s, true));
|
||
if let Ok(ip) = without_trailing_dot.parse::<IpAddr>() {
|
||
if has_trailing_dot {
|
||
return Err(HostParseError::InvalidHost(
|
||
without_trailing_dot.to_string(),
|
||
));
|
||
}
|
||
Ok(Host::Ip(ip))
|
||
} else {
|
||
let lower = if s.chars().all(|c| c.is_ascii_lowercase()) {
|
||
Cow::Borrowed(s)
|
||
} else {
|
||
Cow::Owned(s.to_ascii_lowercase())
|
||
};
|
||
let fqdn = {
|
||
use std::str::FromStr;
|
||
FQDN::from_str(&lower).map_err(|e| HostParseError::Fqdn {
|
||
error: e,
|
||
host: s.to_string(),
|
||
})?
|
||
};
|
||
if fqdn.is_root() {
|
||
return Err(HostParseError::InvalidEmptyHost(s.to_string()));
|
||
}
|
||
Ok(Host::Fqdn(fqdn))
|
||
}
|
||
}
|
||
|
||
#[cfg(test)]
|
||
#[track_caller]
|
||
fn must_parse(s: &str) -> Self {
|
||
Self::parse(s).unwrap()
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct NetDescriptor(pub Host, pub Option<u16>);
|
||
|
||
impl QueryDescriptor for NetDescriptor {
|
||
type AllowDesc = NetDescriptor;
|
||
type DenyDesc = NetDescriptor;
|
||
|
||
fn flag_name() -> &'static str {
|
||
"net"
|
||
}
|
||
|
||
fn display_name(&self) -> Cow<str> {
|
||
Cow::from(format!("{}", self))
|
||
}
|
||
|
||
fn from_allow(allow: &Self::AllowDesc) -> Self {
|
||
allow.clone()
|
||
}
|
||
|
||
fn as_allow(&self) -> Option<Self::AllowDesc> {
|
||
Some(self.clone())
|
||
}
|
||
|
||
fn as_deny(&self) -> Self::DenyDesc {
|
||
self.clone()
|
||
}
|
||
|
||
fn check_in_permission(
|
||
&self,
|
||
perm: &mut UnaryPermission<Self>,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(perm);
|
||
perm.check_desc(Some(self), false, api_name)
|
||
}
|
||
|
||
fn matches_allow(&self, other: &Self::AllowDesc) -> bool {
|
||
self.0 == other.0 && (other.1.is_none() || self.1 == other.1)
|
||
}
|
||
|
||
fn matches_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.0 == other.0 && (other.1.is_none() || self.1 == other.1)
|
||
}
|
||
|
||
fn revokes(&self, other: &Self::AllowDesc) -> bool {
|
||
self.matches_allow(other)
|
||
}
|
||
|
||
fn stronger_than_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.matches_deny(other)
|
||
}
|
||
|
||
fn overlaps_deny(&self, _other: &Self::DenyDesc) -> bool {
|
||
false
|
||
}
|
||
}
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
pub enum NetDescriptorParseError {
|
||
#[error("invalid value '{0}': URLs are not supported, only domains and ips")]
|
||
Url(String),
|
||
#[error("invalid IPv6 address in '{hostname}': '{ip}'")]
|
||
InvalidIpv6 { hostname: String, ip: String },
|
||
#[error("invalid port in '{hostname}': '{port}'")]
|
||
InvalidPort { hostname: String, port: String },
|
||
#[error("invalid host: '{0}'")]
|
||
InvalidHost(String),
|
||
#[error("invalid empty port in '{0}'")]
|
||
EmptyPort(String),
|
||
#[error("ipv6 addresses must be enclosed in square brackets: '{0}'")]
|
||
Ipv6MissingSquareBrackets(String),
|
||
#[error("{0}")]
|
||
Host(#[from] HostParseError),
|
||
}
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
pub enum NetDescriptorFromUrlParseError {
|
||
#[error("Missing host in url: '{0}'")]
|
||
MissingHost(Url),
|
||
#[error("{0}")]
|
||
Host(#[from] HostParseError),
|
||
}
|
||
|
||
impl NetDescriptor {
|
||
pub fn parse(hostname: &str) -> Result<Self, NetDescriptorParseError> {
|
||
if hostname.starts_with("http://") || hostname.starts_with("https://") {
|
||
return Err(NetDescriptorParseError::Url(hostname.to_string()));
|
||
}
|
||
|
||
// If this is a IPv6 address enclosed in square brackets, parse it as such.
|
||
if hostname.starts_with('[') {
|
||
if let Some((ip, after)) = hostname.split_once(']') {
|
||
let ip = ip[1..].parse::<Ipv6Addr>().map_err(|_| {
|
||
NetDescriptorParseError::InvalidIpv6 {
|
||
hostname: hostname.to_string(),
|
||
ip: ip.to_string(),
|
||
}
|
||
})?;
|
||
let port = if let Some(port) = after.strip_prefix(':') {
|
||
let port = port.parse::<u16>().map_err(|_| {
|
||
NetDescriptorParseError::InvalidPort {
|
||
hostname: hostname.to_string(),
|
||
port: port.to_string(),
|
||
}
|
||
})?;
|
||
Some(port)
|
||
} else if after.is_empty() {
|
||
None
|
||
} else {
|
||
return Err(NetDescriptorParseError::InvalidHost(
|
||
hostname.to_string(),
|
||
));
|
||
};
|
||
return Ok(NetDescriptor(Host::Ip(IpAddr::V6(ip)), port));
|
||
} else {
|
||
return Err(NetDescriptorParseError::InvalidHost(hostname.to_string()));
|
||
}
|
||
}
|
||
|
||
// Otherwise it is an IPv4 address or a FQDN with an optional port.
|
||
let (host, port) = match hostname.split_once(':') {
|
||
Some((_, "")) => {
|
||
return Err(NetDescriptorParseError::EmptyPort(hostname.to_string()));
|
||
}
|
||
Some((host, port)) => (host, port),
|
||
None => (hostname, ""),
|
||
};
|
||
let host = Host::parse(host)?;
|
||
|
||
let port = if port.is_empty() {
|
||
None
|
||
} else {
|
||
let port = port.parse::<u16>().map_err(|_| {
|
||
// If the user forgot to enclose an IPv6 address in square brackets, we
|
||
// should give them a hint. There are always at least two colons in an
|
||
// IPv6 address, so this heuristic finds likely a bare IPv6 address.
|
||
if port.contains(':') {
|
||
NetDescriptorParseError::Ipv6MissingSquareBrackets(
|
||
hostname.to_string(),
|
||
)
|
||
} else {
|
||
NetDescriptorParseError::InvalidPort {
|
||
hostname: hostname.to_string(),
|
||
port: port.to_string(),
|
||
}
|
||
}
|
||
})?;
|
||
Some(port)
|
||
};
|
||
|
||
Ok(NetDescriptor(host, port))
|
||
}
|
||
|
||
pub fn from_url(url: &Url) -> Result<Self, NetDescriptorFromUrlParseError> {
|
||
let host = url.host_str().ok_or_else(|| {
|
||
NetDescriptorFromUrlParseError::MissingHost(url.clone())
|
||
})?;
|
||
let host = Host::parse(host)?;
|
||
let port = url.port_or_known_default();
|
||
Ok(NetDescriptor(host, port))
|
||
}
|
||
}
|
||
|
||
impl fmt::Display for NetDescriptor {
|
||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||
match &self.0 {
|
||
Host::Fqdn(fqdn) => write!(f, "{fqdn}"),
|
||
Host::Ip(IpAddr::V4(ip)) => write!(f, "{ip}"),
|
||
Host::Ip(IpAddr::V6(ip)) => write!(f, "[{ip}]"),
|
||
}?;
|
||
if let Some(port) = self.1 {
|
||
write!(f, ":{}", port)?;
|
||
}
|
||
Ok(())
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct ImportDescriptor(NetDescriptor);
|
||
|
||
impl QueryDescriptor for ImportDescriptor {
|
||
type AllowDesc = ImportDescriptor;
|
||
type DenyDesc = ImportDescriptor;
|
||
|
||
fn flag_name() -> &'static str {
|
||
"import"
|
||
}
|
||
|
||
fn display_name(&self) -> Cow<str> {
|
||
self.0.display_name()
|
||
}
|
||
|
||
fn from_allow(allow: &Self::AllowDesc) -> Self {
|
||
Self(NetDescriptor::from_allow(&allow.0))
|
||
}
|
||
|
||
fn as_allow(&self) -> Option<Self::AllowDesc> {
|
||
self.0.as_allow().map(ImportDescriptor)
|
||
}
|
||
|
||
fn as_deny(&self) -> Self::DenyDesc {
|
||
Self(self.0.as_deny())
|
||
}
|
||
|
||
fn check_in_permission(
|
||
&self,
|
||
perm: &mut UnaryPermission<Self>,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(perm);
|
||
perm.check_desc(Some(self), false, api_name)
|
||
}
|
||
|
||
fn matches_allow(&self, other: &Self::AllowDesc) -> bool {
|
||
self.0.matches_allow(&other.0)
|
||
}
|
||
|
||
fn matches_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.0.matches_deny(&other.0)
|
||
}
|
||
|
||
fn revokes(&self, other: &Self::AllowDesc) -> bool {
|
||
self.0.revokes(&other.0)
|
||
}
|
||
|
||
fn stronger_than_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.0.stronger_than_deny(&other.0)
|
||
}
|
||
|
||
fn overlaps_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.0.overlaps_deny(&other.0)
|
||
}
|
||
}
|
||
|
||
impl ImportDescriptor {
|
||
pub fn parse(specifier: &str) -> Result<Self, NetDescriptorParseError> {
|
||
Ok(ImportDescriptor(NetDescriptor::parse(specifier)?))
|
||
}
|
||
|
||
pub fn from_url(url: &Url) -> Result<Self, NetDescriptorFromUrlParseError> {
|
||
Ok(ImportDescriptor(NetDescriptor::from_url(url)?))
|
||
}
|
||
}
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
#[error("Empty env not allowed")]
|
||
pub struct EnvDescriptorParseError;
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct EnvDescriptor(EnvVarName);
|
||
|
||
impl EnvDescriptor {
|
||
pub fn new(env: impl AsRef<str>) -> Self {
|
||
Self(EnvVarName::new(env))
|
||
}
|
||
}
|
||
|
||
impl QueryDescriptor for EnvDescriptor {
|
||
type AllowDesc = EnvDescriptor;
|
||
type DenyDesc = EnvDescriptor;
|
||
|
||
fn flag_name() -> &'static str {
|
||
"env"
|
||
}
|
||
|
||
fn display_name(&self) -> Cow<str> {
|
||
Cow::from(self.0.as_ref())
|
||
}
|
||
|
||
fn from_allow(allow: &Self::AllowDesc) -> Self {
|
||
allow.clone()
|
||
}
|
||
|
||
fn as_allow(&self) -> Option<Self::AllowDesc> {
|
||
Some(self.clone())
|
||
}
|
||
|
||
fn as_deny(&self) -> Self::DenyDesc {
|
||
self.clone()
|
||
}
|
||
|
||
fn check_in_permission(
|
||
&self,
|
||
perm: &mut UnaryPermission<Self>,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(perm);
|
||
perm.check_desc(Some(self), false, api_name)
|
||
}
|
||
|
||
fn matches_allow(&self, other: &Self::AllowDesc) -> bool {
|
||
self == other
|
||
}
|
||
|
||
fn matches_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self == other
|
||
}
|
||
|
||
fn revokes(&self, other: &Self::AllowDesc) -> bool {
|
||
self == other
|
||
}
|
||
|
||
fn stronger_than_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self == other
|
||
}
|
||
|
||
fn overlaps_deny(&self, _other: &Self::DenyDesc) -> bool {
|
||
false
|
||
}
|
||
}
|
||
|
||
impl AsRef<str> for EnvDescriptor {
|
||
fn as_ref(&self) -> &str {
|
||
self.0.as_ref()
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug, Serialize, Deserialize)]
|
||
pub enum RunQueryDescriptor {
|
||
Path {
|
||
requested: String,
|
||
resolved: PathBuf,
|
||
},
|
||
/// This variant won't actually grant permissions because the path of
|
||
/// the executable is unresolved. It's mostly used so that prompts and
|
||
/// everything works the same way as when the command is resolved,
|
||
/// meaning that a script can't tell
|
||
/// if a command is resolved or not based on how long something
|
||
/// takes to ask for permissions.
|
||
Name(String),
|
||
}
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
pub enum PathResolveError {
|
||
#[error("failed resolving cwd: {0}")]
|
||
CwdResolve(#[source] std::io::Error),
|
||
#[error("Empty path is not allowed")]
|
||
EmptyPath,
|
||
}
|
||
|
||
impl RunQueryDescriptor {
|
||
pub fn parse(
|
||
requested: &str,
|
||
) -> Result<RunQueryDescriptor, PathResolveError> {
|
||
if is_path(requested) {
|
||
let path = PathBuf::from(requested);
|
||
let resolved = if path.is_absolute() {
|
||
normalize_path(path)
|
||
} else {
|
||
let cwd =
|
||
std::env::current_dir().map_err(PathResolveError::CwdResolve)?;
|
||
normalize_path(cwd.join(path))
|
||
};
|
||
Ok(RunQueryDescriptor::Path {
|
||
requested: requested.to_string(),
|
||
resolved,
|
||
})
|
||
} else {
|
||
match which::which(requested) {
|
||
Ok(resolved) => Ok(RunQueryDescriptor::Path {
|
||
requested: requested.to_string(),
|
||
resolved,
|
||
}),
|
||
Err(_) => Ok(RunQueryDescriptor::Name(requested.to_string())),
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
impl QueryDescriptor for RunQueryDescriptor {
|
||
type AllowDesc = AllowRunDescriptor;
|
||
type DenyDesc = DenyRunDescriptor;
|
||
|
||
fn flag_name() -> &'static str {
|
||
"run"
|
||
}
|
||
|
||
fn display_name(&self) -> Cow<str> {
|
||
match self {
|
||
RunQueryDescriptor::Path { requested, .. } => Cow::Borrowed(requested),
|
||
RunQueryDescriptor::Name(name) => Cow::Borrowed(name),
|
||
}
|
||
}
|
||
|
||
fn from_allow(allow: &Self::AllowDesc) -> Self {
|
||
RunQueryDescriptor::Path {
|
||
requested: allow.0.to_string_lossy().into_owned(),
|
||
resolved: allow.0.clone(),
|
||
}
|
||
}
|
||
|
||
fn as_allow(&self) -> Option<Self::AllowDesc> {
|
||
match self {
|
||
RunQueryDescriptor::Path { resolved, .. } => {
|
||
Some(AllowRunDescriptor(resolved.clone()))
|
||
}
|
||
RunQueryDescriptor::Name(_) => None,
|
||
}
|
||
}
|
||
|
||
fn as_deny(&self) -> Self::DenyDesc {
|
||
match self {
|
||
RunQueryDescriptor::Path {
|
||
resolved,
|
||
requested,
|
||
} => {
|
||
if requested.contains('/')
|
||
|| (cfg!(windows) && requested.contains("\\"))
|
||
{
|
||
DenyRunDescriptor::Path(resolved.clone())
|
||
} else {
|
||
DenyRunDescriptor::Name(requested.clone())
|
||
}
|
||
}
|
||
RunQueryDescriptor::Name(name) => DenyRunDescriptor::Name(name.clone()),
|
||
}
|
||
}
|
||
|
||
fn check_in_permission(
|
||
&self,
|
||
perm: &mut UnaryPermission<Self>,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(perm);
|
||
perm.check_desc(Some(self), false, api_name)
|
||
}
|
||
|
||
fn matches_allow(&self, other: &Self::AllowDesc) -> bool {
|
||
match self {
|
||
RunQueryDescriptor::Path { resolved, .. } => *resolved == other.0,
|
||
RunQueryDescriptor::Name(_) => false,
|
||
}
|
||
}
|
||
|
||
fn matches_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
match other {
|
||
DenyRunDescriptor::Name(deny_desc) => match self {
|
||
RunQueryDescriptor::Path { resolved, .. } => {
|
||
denies_run_name(deny_desc, resolved)
|
||
}
|
||
RunQueryDescriptor::Name(query) => query == deny_desc,
|
||
},
|
||
DenyRunDescriptor::Path(deny_desc) => match self {
|
||
RunQueryDescriptor::Path { resolved, .. } => {
|
||
resolved.starts_with(deny_desc)
|
||
}
|
||
RunQueryDescriptor::Name(query) => denies_run_name(query, deny_desc),
|
||
},
|
||
}
|
||
}
|
||
|
||
fn revokes(&self, other: &Self::AllowDesc) -> bool {
|
||
match self {
|
||
RunQueryDescriptor::Path {
|
||
resolved,
|
||
requested,
|
||
} => {
|
||
if *resolved == other.0 {
|
||
return true;
|
||
}
|
||
if is_path(requested) {
|
||
false
|
||
} else {
|
||
denies_run_name(requested, &other.0)
|
||
}
|
||
}
|
||
RunQueryDescriptor::Name(query) => denies_run_name(query, &other.0),
|
||
}
|
||
}
|
||
|
||
fn stronger_than_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.matches_deny(other)
|
||
}
|
||
|
||
fn overlaps_deny(&self, _other: &Self::DenyDesc) -> bool {
|
||
false
|
||
}
|
||
}
|
||
|
||
pub enum RunDescriptorArg {
|
||
Name(String),
|
||
Path(PathBuf),
|
||
}
|
||
|
||
pub enum AllowRunDescriptorParseResult {
|
||
/// An error occured getting the descriptor that should
|
||
/// be surfaced as a warning when launching deno, but should
|
||
/// be ignored when creating a worker.
|
||
Unresolved(Box<which::Error>),
|
||
Descriptor(AllowRunDescriptor),
|
||
}
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
pub enum RunDescriptorParseError {
|
||
#[error("{0}")]
|
||
Which(#[from] which::Error),
|
||
#[error("{0}")]
|
||
PathResolve(#[from] PathResolveError),
|
||
#[error("Empty run query is not allowed")]
|
||
EmptyRunQuery,
|
||
}
|
||
|
||
#[derive(Debug, Clone, Hash, Eq, PartialEq)]
|
||
pub struct AllowRunDescriptor(pub PathBuf);
|
||
|
||
impl AllowRunDescriptor {
|
||
pub fn parse(
|
||
text: &str,
|
||
cwd: &Path,
|
||
) -> Result<AllowRunDescriptorParseResult, which::Error> {
|
||
let is_path = is_path(text);
|
||
// todo(dsherret): canonicalize in #25458
|
||
let path = if is_path {
|
||
resolve_from_known_cwd(Path::new(text), cwd)
|
||
} else {
|
||
match which::which_in(text, std::env::var_os("PATH"), cwd) {
|
||
Ok(path) => path,
|
||
Err(err) => match err {
|
||
which::Error::BadAbsolutePath | which::Error::BadRelativePath => {
|
||
return Err(err);
|
||
}
|
||
which::Error::CannotFindBinaryPath
|
||
| which::Error::CannotGetCurrentDir
|
||
| which::Error::CannotCanonicalize => {
|
||
return Ok(AllowRunDescriptorParseResult::Unresolved(Box::new(err)))
|
||
}
|
||
},
|
||
}
|
||
};
|
||
Ok(AllowRunDescriptorParseResult::Descriptor(
|
||
AllowRunDescriptor(path),
|
||
))
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub enum DenyRunDescriptor {
|
||
/// Warning: You may want to construct with `RunDescriptor::from()` for case
|
||
/// handling.
|
||
Name(String),
|
||
/// Warning: You may want to construct with `RunDescriptor::from()` for case
|
||
/// handling.
|
||
Path(PathBuf),
|
||
}
|
||
|
||
impl DenyRunDescriptor {
|
||
pub fn parse(text: &str, cwd: &Path) -> Self {
|
||
if text.contains('/') || cfg!(windows) && text.contains('\\') {
|
||
let path = resolve_from_known_cwd(Path::new(&text), cwd);
|
||
DenyRunDescriptor::Path(path)
|
||
} else {
|
||
DenyRunDescriptor::Name(text.to_string())
|
||
}
|
||
}
|
||
}
|
||
|
||
fn is_path(text: &str) -> bool {
|
||
if cfg!(windows) {
|
||
text.contains('/') || text.contains('\\') || Path::new(text).is_absolute()
|
||
} else {
|
||
text.contains('/')
|
||
}
|
||
}
|
||
|
||
fn denies_run_name(name: &str, cmd_path: &Path) -> bool {
|
||
let Some(file_stem) = cmd_path.file_stem() else {
|
||
return false;
|
||
};
|
||
let Some(file_stem) = file_stem.to_str() else {
|
||
return false;
|
||
};
|
||
if file_stem.len() < name.len() {
|
||
return false;
|
||
}
|
||
let (prefix, suffix) = file_stem.split_at(name.len());
|
||
if !prefix.eq_ignore_ascii_case(name) {
|
||
return false;
|
||
}
|
||
// be broad and consider anything like `deno.something` as matching deny perms
|
||
suffix.is_empty() || suffix.starts_with('.')
|
||
}
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
pub enum SysDescriptorParseError {
|
||
#[error("unknown system info kind \"{0}\"")]
|
||
InvalidKind(String), // TypeError
|
||
#[error("Empty sys not allowed")]
|
||
Empty, // Error
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct SysDescriptor(String);
|
||
|
||
impl SysDescriptor {
|
||
pub fn parse(kind: String) -> Result<Self, SysDescriptorParseError> {
|
||
match kind.as_str() {
|
||
"hostname" | "inspector" | "osRelease" | "osUptime" | "loadavg"
|
||
| "networkInterfaces" | "systemMemoryInfo" | "uid" | "gid" | "cpus"
|
||
| "homedir" | "getegid" | "statfs" | "getPriority" | "setPriority"
|
||
| "userInfo" => Ok(Self(kind)),
|
||
|
||
// the underlying permission check changed to `userInfo` to better match the API,
|
||
// alias this to avoid breaking existing projects with `--allow-sys=username`
|
||
"username" => Ok(Self("userInfo".into())),
|
||
_ => Err(SysDescriptorParseError::InvalidKind(kind)),
|
||
}
|
||
}
|
||
|
||
pub fn into_string(self) -> String {
|
||
self.0
|
||
}
|
||
}
|
||
|
||
impl QueryDescriptor for SysDescriptor {
|
||
type AllowDesc = SysDescriptor;
|
||
type DenyDesc = SysDescriptor;
|
||
|
||
fn flag_name() -> &'static str {
|
||
"sys"
|
||
}
|
||
|
||
fn display_name(&self) -> Cow<str> {
|
||
Cow::from(self.0.to_string())
|
||
}
|
||
|
||
fn from_allow(allow: &Self::AllowDesc) -> Self {
|
||
allow.clone()
|
||
}
|
||
|
||
fn as_allow(&self) -> Option<Self::AllowDesc> {
|
||
Some(self.clone())
|
||
}
|
||
|
||
fn as_deny(&self) -> Self::DenyDesc {
|
||
self.clone()
|
||
}
|
||
|
||
fn check_in_permission(
|
||
&self,
|
||
perm: &mut UnaryPermission<Self>,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(perm);
|
||
perm.check_desc(Some(self), false, api_name)
|
||
}
|
||
|
||
fn matches_allow(&self, other: &Self::AllowDesc) -> bool {
|
||
self == other
|
||
}
|
||
|
||
fn matches_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self == other
|
||
}
|
||
|
||
fn revokes(&self, other: &Self::AllowDesc) -> bool {
|
||
self == other
|
||
}
|
||
|
||
fn stronger_than_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self == other
|
||
}
|
||
|
||
fn overlaps_deny(&self, _other: &Self::DenyDesc) -> bool {
|
||
false
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct FfiQueryDescriptor(pub PathQueryDescriptor);
|
||
|
||
impl QueryDescriptor for FfiQueryDescriptor {
|
||
type AllowDesc = FfiDescriptor;
|
||
type DenyDesc = FfiDescriptor;
|
||
|
||
fn flag_name() -> &'static str {
|
||
"ffi"
|
||
}
|
||
|
||
fn display_name(&self) -> Cow<str> {
|
||
Cow::Borrowed(&self.0.requested)
|
||
}
|
||
|
||
fn from_allow(allow: &Self::AllowDesc) -> Self {
|
||
PathQueryDescriptor {
|
||
requested: allow.0.to_string_lossy().into_owned(),
|
||
resolved: allow.0.clone(),
|
||
}
|
||
.into_ffi()
|
||
}
|
||
|
||
fn as_allow(&self) -> Option<Self::AllowDesc> {
|
||
Some(FfiDescriptor(self.0.resolved.clone()))
|
||
}
|
||
|
||
fn as_deny(&self) -> Self::DenyDesc {
|
||
FfiDescriptor(self.0.resolved.clone())
|
||
}
|
||
|
||
fn check_in_permission(
|
||
&self,
|
||
perm: &mut UnaryPermission<Self>,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(perm);
|
||
perm.check_desc(Some(self), true, api_name)
|
||
}
|
||
|
||
fn matches_allow(&self, other: &Self::AllowDesc) -> bool {
|
||
self.0.resolved.starts_with(&other.0)
|
||
}
|
||
|
||
fn matches_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.0.resolved.starts_with(&other.0)
|
||
}
|
||
|
||
fn revokes(&self, other: &Self::AllowDesc) -> bool {
|
||
self.matches_allow(other)
|
||
}
|
||
|
||
fn stronger_than_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
other.0.starts_with(&self.0.resolved)
|
||
}
|
||
|
||
fn overlaps_deny(&self, other: &Self::DenyDesc) -> bool {
|
||
self.stronger_than_deny(other)
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
|
||
pub struct FfiDescriptor(pub PathBuf);
|
||
|
||
impl UnaryPermission<ReadQueryDescriptor> {
|
||
pub fn query(&self, desc: Option<&ReadQueryDescriptor>) -> PermissionState {
|
||
self.query_desc(desc, AllowPartial::TreatAsPartialGranted)
|
||
}
|
||
|
||
pub fn request(
|
||
&mut self,
|
||
path: Option<&ReadQueryDescriptor>,
|
||
) -> PermissionState {
|
||
self.request_desc(path)
|
||
}
|
||
|
||
pub fn revoke(
|
||
&mut self,
|
||
desc: Option<&ReadQueryDescriptor>,
|
||
) -> PermissionState {
|
||
self.revoke_desc(desc)
|
||
}
|
||
|
||
pub fn check(
|
||
&mut self,
|
||
desc: &ReadQueryDescriptor,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(Some(desc), true, api_name)
|
||
}
|
||
|
||
#[inline]
|
||
pub fn check_partial(
|
||
&mut self,
|
||
desc: &ReadQueryDescriptor,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(Some(desc), false, api_name)
|
||
}
|
||
|
||
pub fn check_all(
|
||
&mut self,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(None, false, api_name)
|
||
}
|
||
}
|
||
|
||
impl UnaryPermission<WriteQueryDescriptor> {
|
||
pub fn query(&self, path: Option<&WriteQueryDescriptor>) -> PermissionState {
|
||
self.query_desc(path, AllowPartial::TreatAsPartialGranted)
|
||
}
|
||
|
||
pub fn request(
|
||
&mut self,
|
||
path: Option<&WriteQueryDescriptor>,
|
||
) -> PermissionState {
|
||
self.request_desc(path)
|
||
}
|
||
|
||
pub fn revoke(
|
||
&mut self,
|
||
path: Option<&WriteQueryDescriptor>,
|
||
) -> PermissionState {
|
||
self.revoke_desc(path)
|
||
}
|
||
|
||
pub fn check(
|
||
&mut self,
|
||
path: &WriteQueryDescriptor,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(Some(path), true, api_name)
|
||
}
|
||
|
||
#[inline]
|
||
pub fn check_partial(
|
||
&mut self,
|
||
path: &WriteQueryDescriptor,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(Some(path), false, api_name)
|
||
}
|
||
|
||
pub fn check_all(
|
||
&mut self,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(None, false, api_name)
|
||
}
|
||
}
|
||
|
||
impl UnaryPermission<NetDescriptor> {
|
||
pub fn query(&self, host: Option<&NetDescriptor>) -> PermissionState {
|
||
self.query_desc(host, AllowPartial::TreatAsPartialGranted)
|
||
}
|
||
|
||
pub fn request(&mut self, host: Option<&NetDescriptor>) -> PermissionState {
|
||
self.request_desc(host)
|
||
}
|
||
|
||
pub fn revoke(&mut self, host: Option<&NetDescriptor>) -> PermissionState {
|
||
self.revoke_desc(host)
|
||
}
|
||
|
||
pub fn check(
|
||
&mut self,
|
||
host: &NetDescriptor,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(Some(host), false, api_name)
|
||
}
|
||
|
||
pub fn check_all(&mut self) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(None, false, None)
|
||
}
|
||
}
|
||
|
||
impl UnaryPermission<ImportDescriptor> {
|
||
pub fn query(&self, host: Option<&ImportDescriptor>) -> PermissionState {
|
||
self.query_desc(host, AllowPartial::TreatAsPartialGranted)
|
||
}
|
||
|
||
pub fn request(
|
||
&mut self,
|
||
host: Option<&ImportDescriptor>,
|
||
) -> PermissionState {
|
||
self.request_desc(host)
|
||
}
|
||
|
||
pub fn revoke(&mut self, host: Option<&ImportDescriptor>) -> PermissionState {
|
||
self.revoke_desc(host)
|
||
}
|
||
|
||
pub fn check(
|
||
&mut self,
|
||
host: &ImportDescriptor,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(Some(host), false, api_name)
|
||
}
|
||
|
||
pub fn check_all(&mut self) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(None, false, None)
|
||
}
|
||
}
|
||
|
||
impl UnaryPermission<EnvDescriptor> {
|
||
pub fn query(&self, env: Option<&str>) -> PermissionState {
|
||
self.query_desc(
|
||
env.map(EnvDescriptor::new).as_ref(),
|
||
AllowPartial::TreatAsPartialGranted,
|
||
)
|
||
}
|
||
|
||
pub fn request(&mut self, env: Option<&str>) -> PermissionState {
|
||
self.request_desc(env.map(EnvDescriptor::new).as_ref())
|
||
}
|
||
|
||
pub fn revoke(&mut self, env: Option<&str>) -> PermissionState {
|
||
self.revoke_desc(env.map(EnvDescriptor::new).as_ref())
|
||
}
|
||
|
||
pub fn check(
|
||
&mut self,
|
||
env: &str,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(Some(&EnvDescriptor::new(env)), false, api_name)
|
||
}
|
||
|
||
pub fn check_all(&mut self) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(None, false, None)
|
||
}
|
||
}
|
||
|
||
impl UnaryPermission<SysDescriptor> {
|
||
pub fn query(&self, kind: Option<&SysDescriptor>) -> PermissionState {
|
||
self.query_desc(kind, AllowPartial::TreatAsPartialGranted)
|
||
}
|
||
|
||
pub fn request(&mut self, kind: Option<&SysDescriptor>) -> PermissionState {
|
||
self.request_desc(kind)
|
||
}
|
||
|
||
pub fn revoke(&mut self, kind: Option<&SysDescriptor>) -> PermissionState {
|
||
self.revoke_desc(kind)
|
||
}
|
||
|
||
pub fn check(
|
||
&mut self,
|
||
kind: &SysDescriptor,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(Some(kind), false, api_name)
|
||
}
|
||
|
||
pub fn check_all(&mut self) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(None, false, None)
|
||
}
|
||
}
|
||
|
||
impl UnaryPermission<RunQueryDescriptor> {
|
||
pub fn query(&self, cmd: Option<&RunQueryDescriptor>) -> PermissionState {
|
||
self.query_desc(cmd, AllowPartial::TreatAsPartialGranted)
|
||
}
|
||
|
||
pub fn request(
|
||
&mut self,
|
||
cmd: Option<&RunQueryDescriptor>,
|
||
) -> PermissionState {
|
||
self.request_desc(cmd)
|
||
}
|
||
|
||
pub fn revoke(
|
||
&mut self,
|
||
cmd: Option<&RunQueryDescriptor>,
|
||
) -> PermissionState {
|
||
self.revoke_desc(cmd)
|
||
}
|
||
|
||
pub fn check(
|
||
&mut self,
|
||
cmd: &RunQueryDescriptor,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
self.check_desc(Some(cmd), false, api_name)
|
||
}
|
||
|
||
pub fn check_all(
|
||
&mut self,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
self.check_desc(None, false, api_name)
|
||
}
|
||
|
||
/// Queries without prompting
|
||
pub fn query_all(&mut self, api_name: Option<&str>) -> bool {
|
||
if self.is_allow_all() {
|
||
return true;
|
||
}
|
||
let (result, _prompted, _is_allow_all) =
|
||
self.query_desc(None, AllowPartial::TreatAsDenied).check2(
|
||
RunQueryDescriptor::flag_name(),
|
||
api_name,
|
||
|| None,
|
||
/* prompt */ false,
|
||
);
|
||
result.is_ok()
|
||
}
|
||
}
|
||
|
||
impl UnaryPermission<FfiQueryDescriptor> {
|
||
pub fn query(&self, path: Option<&FfiQueryDescriptor>) -> PermissionState {
|
||
self.query_desc(path, AllowPartial::TreatAsPartialGranted)
|
||
}
|
||
|
||
pub fn request(
|
||
&mut self,
|
||
path: Option<&FfiQueryDescriptor>,
|
||
) -> PermissionState {
|
||
self.request_desc(path)
|
||
}
|
||
|
||
pub fn revoke(
|
||
&mut self,
|
||
path: Option<&FfiQueryDescriptor>,
|
||
) -> PermissionState {
|
||
self.revoke_desc(path)
|
||
}
|
||
|
||
pub fn check(
|
||
&mut self,
|
||
path: &FfiQueryDescriptor,
|
||
api_name: Option<&str>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(Some(path), true, api_name)
|
||
}
|
||
|
||
pub fn check_partial(
|
||
&mut self,
|
||
path: Option<&FfiQueryDescriptor>,
|
||
) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(path, false, None)
|
||
}
|
||
|
||
pub fn check_all(&mut self) -> Result<(), PermissionDeniedError> {
|
||
skip_check_if_is_permission_fully_granted!(self);
|
||
self.check_desc(None, false, Some("all"))
|
||
}
|
||
}
|
||
|
||
#[derive(Clone, Debug, Eq, PartialEq)]
|
||
pub struct Permissions {
|
||
pub read: UnaryPermission<ReadQueryDescriptor>,
|
||
pub write: UnaryPermission<WriteQueryDescriptor>,
|
||
pub net: UnaryPermission<NetDescriptor>,
|
||
pub env: UnaryPermission<EnvDescriptor>,
|
||
pub sys: UnaryPermission<SysDescriptor>,
|
||
pub run: UnaryPermission<RunQueryDescriptor>,
|
||
pub ffi: UnaryPermission<FfiQueryDescriptor>,
|
||
pub import: UnaryPermission<ImportDescriptor>,
|
||
pub all: UnitPermission,
|
||
}
|
||
|
||
#[derive(Clone, Debug, Eq, PartialEq, Default, Serialize, Deserialize)]
|
||
pub struct PermissionsOptions {
|
||
pub allow_all: bool,
|
||
pub allow_env: Option<Vec<String>>,
|
||
pub deny_env: Option<Vec<String>>,
|
||
pub allow_net: Option<Vec<String>>,
|
||
pub deny_net: Option<Vec<String>>,
|
||
pub allow_ffi: Option<Vec<String>>,
|
||
pub deny_ffi: Option<Vec<String>>,
|
||
pub allow_read: Option<Vec<String>>,
|
||
pub deny_read: Option<Vec<String>>,
|
||
pub allow_run: Option<Vec<String>>,
|
||
pub deny_run: Option<Vec<String>>,
|
||
pub allow_sys: Option<Vec<String>>,
|
||
pub deny_sys: Option<Vec<String>>,
|
||
pub allow_write: Option<Vec<String>>,
|
||
pub deny_write: Option<Vec<String>>,
|
||
pub allow_import: Option<Vec<String>>,
|
||
pub prompt: bool,
|
||
}
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
pub enum PermissionsFromOptionsError {
|
||
#[error("{0}")]
|
||
PathResolve(#[from] PathResolveError),
|
||
#[error("{0}")]
|
||
SysDescriptorParse(#[from] SysDescriptorParseError),
|
||
#[error("{0}")]
|
||
NetDescriptorParse(#[from] NetDescriptorParseError),
|
||
#[error("{0}")]
|
||
EnvDescriptorParse(#[from] EnvDescriptorParseError),
|
||
#[error("{0}")]
|
||
RunDescriptorParse(#[from] RunDescriptorParseError),
|
||
#[error("Empty command name not allowed in --allow-run=...")]
|
||
RunEmptyCommandName,
|
||
}
|
||
|
||
impl Permissions {
|
||
pub fn new_unary<TQuery>(
|
||
allow_list: Option<HashSet<TQuery::AllowDesc>>,
|
||
deny_list: Option<HashSet<TQuery::DenyDesc>>,
|
||
prompt: bool,
|
||
) -> UnaryPermission<TQuery>
|
||
where
|
||
TQuery: QueryDescriptor,
|
||
{
|
||
UnaryPermission::<TQuery> {
|
||
granted_global: global_from_option(allow_list.as_ref()),
|
||
granted_list: allow_list.unwrap_or_default(),
|
||
flag_denied_global: global_from_option(deny_list.as_ref()),
|
||
flag_denied_list: deny_list.unwrap_or_default(),
|
||
prompt,
|
||
..Default::default()
|
||
}
|
||
}
|
||
|
||
pub const fn new_all(allow_state: bool) -> UnitPermission {
|
||
unit_permission_from_flag_bools(
|
||
allow_state,
|
||
false,
|
||
"all",
|
||
"all",
|
||
false, // never prompt for all
|
||
)
|
||
}
|
||
|
||
pub fn from_options(
|
||
parser: &dyn PermissionDescriptorParser,
|
||
opts: &PermissionsOptions,
|
||
) -> Result<Self, PermissionsFromOptionsError> {
|
||
fn resolve_allow_run(
|
||
parser: &dyn PermissionDescriptorParser,
|
||
allow_run: &[String],
|
||
) -> Result<HashSet<AllowRunDescriptor>, PermissionsFromOptionsError> {
|
||
let mut new_allow_run = HashSet::with_capacity(allow_run.len());
|
||
for unresolved in allow_run {
|
||
if unresolved.is_empty() {
|
||
return Err(PermissionsFromOptionsError::RunEmptyCommandName);
|
||
}
|
||
match parser.parse_allow_run_descriptor(unresolved)? {
|
||
AllowRunDescriptorParseResult::Descriptor(descriptor) => {
|
||
new_allow_run.insert(descriptor);
|
||
}
|
||
AllowRunDescriptorParseResult::Unresolved(err) => {
|
||
log::info!(
|
||
"{} Failed to resolve '{}' for allow-run: {}",
|
||
colors::gray("Info"),
|
||
unresolved,
|
||
err
|
||
);
|
||
}
|
||
}
|
||
}
|
||
Ok(new_allow_run)
|
||
}
|
||
|
||
fn parse_maybe_vec<T: Eq + PartialEq + Hash, E>(
|
||
items: Option<&[String]>,
|
||
parse: impl Fn(&str) -> Result<T, E>,
|
||
) -> Result<Option<HashSet<T>>, PermissionsFromOptionsError>
|
||
where
|
||
PermissionsFromOptionsError: From<E>,
|
||
{
|
||
match items {
|
||
Some(items) => Ok(Some(
|
||
items
|
||
.iter()
|
||
.map(|item| parse(item))
|
||
.collect::<Result<HashSet<_>, _>>()?,
|
||
)),
|
||
None => Ok(None),
|
||
}
|
||
}
|
||
|
||
let mut deny_write = parse_maybe_vec(opts.deny_write.as_deref(), |item| {
|
||
parser.parse_write_descriptor(item)
|
||
})?;
|
||
let allow_run = opts
|
||
.allow_run
|
||
.as_ref()
|
||
.and_then(|raw_allow_run| {
|
||
match resolve_allow_run(parser, raw_allow_run) {
|
||
Ok(resolved_allow_run) => {
|
||
if resolved_allow_run.is_empty() && !raw_allow_run.is_empty() {
|
||
None // convert to no permissions if now empty
|
||
} else {
|
||
Some(Ok(resolved_allow_run))
|
||
}
|
||
}
|
||
Err(err) => Some(Err(err)),
|
||
}
|
||
})
|
||
.transpose()?;
|
||
// add the allow_run list to deny_write
|
||
if let Some(allow_run_vec) = &allow_run {
|
||
if !allow_run_vec.is_empty() {
|
||
let deny_write = deny_write.get_or_insert_with(Default::default);
|
||
deny_write.extend(
|
||
allow_run_vec
|
||
.iter()
|
||
.map(|item| WriteDescriptor(item.0.clone())),
|
||
);
|
||
}
|
||
}
|
||
|
||
Ok(Self {
|
||
read: Permissions::new_unary(
|
||
parse_maybe_vec(opts.allow_read.as_deref(), |item| {
|
||
parser.parse_read_descriptor(item)
|
||
})?,
|
||
parse_maybe_vec(opts.deny_read.as_deref(), |item| {
|
||
parser.parse_read_descriptor(item)
|
||
})?,
|
||
opts.prompt,
|
||
),
|
||
write: Permissions::new_unary(
|
||
parse_maybe_vec(opts.allow_write.as_deref(), |item| {
|
||
parser.parse_write_descriptor(item)
|
||
})?,
|
||
deny_write,
|
||
opts.prompt,
|
||
),
|
||
net: Permissions::new_unary(
|
||
parse_maybe_vec(opts.allow_net.as_deref(), |item| {
|
||
parser.parse_net_descriptor(item)
|
||
})?,
|
||
parse_maybe_vec(opts.deny_net.as_deref(), |item| {
|
||
parser.parse_net_descriptor(item)
|
||
})?,
|
||
opts.prompt,
|
||
),
|
||
env: Permissions::new_unary(
|
||
parse_maybe_vec(opts.allow_env.as_deref(), |item| {
|
||
parser.parse_env_descriptor(item)
|
||
})?,
|
||
parse_maybe_vec(opts.deny_env.as_deref(), |text| {
|
||
parser.parse_env_descriptor(text)
|
||
})?,
|
||
opts.prompt,
|
||
),
|
||
sys: Permissions::new_unary(
|
||
parse_maybe_vec(opts.allow_sys.as_deref(), |text| {
|
||
parser.parse_sys_descriptor(text)
|
||
})?,
|
||
parse_maybe_vec(opts.deny_sys.as_deref(), |text| {
|
||
parser.parse_sys_descriptor(text)
|
||
})?,
|
||
opts.prompt,
|
||
),
|
||
run: Permissions::new_unary(
|
||
allow_run,
|
||
parse_maybe_vec(opts.deny_run.as_deref(), |text| {
|
||
parser.parse_deny_run_descriptor(text)
|
||
})?,
|
||
opts.prompt,
|
||
),
|
||
ffi: Permissions::new_unary(
|
||
parse_maybe_vec(opts.allow_ffi.as_deref(), |text| {
|
||
parser.parse_ffi_descriptor(text)
|
||
})?,
|
||
parse_maybe_vec(opts.deny_ffi.as_deref(), |text| {
|
||
parser.parse_ffi_descriptor(text)
|
||
})?,
|
||
opts.prompt,
|
||
),
|
||
import: Permissions::new_unary(
|
||
parse_maybe_vec(opts.allow_import.as_deref(), |item| {
|
||
parser.parse_import_descriptor(item)
|
||
})?,
|
||
None,
|
||
opts.prompt,
|
||
),
|
||
all: Permissions::new_all(opts.allow_all),
|
||
})
|
||
}
|
||
|
||
/// Create a set of permissions that explicitly allow everything.
|
||
pub fn allow_all() -> Self {
|
||
Self {
|
||
read: UnaryPermission::allow_all(),
|
||
write: UnaryPermission::allow_all(),
|
||
net: UnaryPermission::allow_all(),
|
||
env: UnaryPermission::allow_all(),
|
||
sys: UnaryPermission::allow_all(),
|
||
run: UnaryPermission::allow_all(),
|
||
ffi: UnaryPermission::allow_all(),
|
||
import: UnaryPermission::allow_all(),
|
||
all: Permissions::new_all(true),
|
||
}
|
||
}
|
||
|
||
/// Create a set of permissions that enable nothing, but will allow prompting.
|
||
pub fn none_with_prompt() -> Self {
|
||
Self::none(true)
|
||
}
|
||
|
||
/// Create a set of permissions that enable nothing, and will not allow prompting.
|
||
pub fn none_without_prompt() -> Self {
|
||
Self::none(false)
|
||
}
|
||
|
||
fn none(prompt: bool) -> Self {
|
||
Self {
|
||
read: Permissions::new_unary(None, None, prompt),
|
||
write: Permissions::new_unary(None, None, prompt),
|
||
net: Permissions::new_unary(None, None, prompt),
|
||
env: Permissions::new_unary(None, None, prompt),
|
||
sys: Permissions::new_unary(None, None, prompt),
|
||
run: Permissions::new_unary(None, None, prompt),
|
||
ffi: Permissions::new_unary(None, None, prompt),
|
||
import: Permissions::new_unary(None, None, prompt),
|
||
all: Permissions::new_all(false),
|
||
}
|
||
}
|
||
}
|
||
|
||
#[derive(Debug, Clone, Copy, Eq, PartialEq)]
|
||
pub enum CheckSpecifierKind {
|
||
Static,
|
||
Dynamic,
|
||
}
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
pub enum ChildPermissionError {
|
||
#[error("Can't escalate parent thread permissions")]
|
||
Escalation,
|
||
#[error("{0}")]
|
||
PathResolve(#[from] PathResolveError),
|
||
#[error("{0}")]
|
||
NetDescriptorParse(#[from] NetDescriptorParseError),
|
||
#[error("{0}")]
|
||
EnvDescriptorParse(#[from] EnvDescriptorParseError),
|
||
#[error("{0}")]
|
||
SysDescriptorParse(#[from] SysDescriptorParseError),
|
||
#[error("{0}")]
|
||
RunDescriptorParse(#[from] RunDescriptorParseError),
|
||
}
|
||
|
||
#[derive(Debug, thiserror::Error)]
|
||
pub enum PermissionCheckError {
|
||
#[error(transparent)]
|
||
PermissionDenied(#[from] PermissionDeniedError),
|
||
#[error("Invalid file path.\n Specifier: {0}")]
|
||
InvalidFilePath(Url),
|
||
#[error(transparent)]
|
||
NetDescriptorForUrlParse(#[from] NetDescriptorFromUrlParseError),
|
||
#[error(transparent)]
|
||
SysDescriptorParse(#[from] SysDescriptorParseError),
|
||
#[error(transparent)]
|
||
PathResolve(#[from] PathResolveError),
|
||
#[error(transparent)]
|
||
HostParse(#[from] HostParseError),
|
||
}
|
||
|
||
/// Wrapper struct for `Permissions` that can be shared across threads.
|
||
///
|
||
/// We need a way to have internal mutability for permissions as they might get
|
||
/// passed to a future that will prompt the user for permission (and in such
|
||
/// case might need to be mutated). Also for the Web Worker API we need a way
|
||
/// to send permissions to a new thread.
|
||
#[derive(Clone, Debug)]
|
||
pub struct PermissionsContainer {
|
||
descriptor_parser: Arc<dyn PermissionDescriptorParser>,
|
||
inner: Arc<Mutex<Permissions>>,
|
||
}
|
||
|
||
impl PermissionsContainer {
|
||
pub fn new(
|
||
descriptor_parser: Arc<dyn PermissionDescriptorParser>,
|
||
perms: Permissions,
|
||
) -> Self {
|
||
Self {
|
||
descriptor_parser,
|
||
inner: Arc::new(Mutex::new(perms)),
|
||
}
|
||
}
|
||
|
||
pub fn allow_all(
|
||
descriptor_parser: Arc<dyn PermissionDescriptorParser>,
|
||
) -> Self {
|
||
Self::new(descriptor_parser, Permissions::allow_all())
|
||
}
|
||
|
||
pub fn create_child_permissions(
|
||
&self,
|
||
child_permissions_arg: ChildPermissionsArg,
|
||
) -> Result<PermissionsContainer, ChildPermissionError> {
|
||
fn is_granted_unary(arg: &ChildUnaryPermissionArg) -> bool {
|
||
match arg {
|
||
ChildUnaryPermissionArg::Inherit | ChildUnaryPermissionArg::Granted => {
|
||
true
|
||
}
|
||
ChildUnaryPermissionArg::NotGranted
|
||
| ChildUnaryPermissionArg::GrantedList(_) => false,
|
||
}
|
||
}
|
||
|
||
let mut worker_perms = Permissions::none_without_prompt();
|
||
|
||
let mut inner = self.inner.lock();
|
||
worker_perms.all = inner
|
||
.all
|
||
.create_child_permissions(ChildUnitPermissionArg::Inherit)?;
|
||
|
||
// downgrade the `worker_perms.all` based on the other values
|
||
if worker_perms.all.query() == PermissionState::Granted {
|
||
let unary_perms = [
|
||
&child_permissions_arg.read,
|
||
&child_permissions_arg.write,
|
||
&child_permissions_arg.net,
|
||
&child_permissions_arg.import,
|
||
&child_permissions_arg.env,
|
||
&child_permissions_arg.sys,
|
||
&child_permissions_arg.run,
|
||
&child_permissions_arg.ffi,
|
||
];
|
||
let allow_all = unary_perms.into_iter().all(is_granted_unary);
|
||
if !allow_all {
|
||
worker_perms.all.revoke();
|
||
}
|
||
}
|
||
|
||
// WARNING: When adding a permission here, ensure it is handled
|
||
// in the worker_perms.all block above
|
||
worker_perms.read = inner.read.create_child_permissions(
|
||
child_permissions_arg.read,
|
||
|text| {
|
||
Ok::<_, PathResolveError>(Some(
|
||
self.descriptor_parser.parse_read_descriptor(text)?,
|
||
))
|
||
},
|
||
)?;
|
||
worker_perms.write = inner.write.create_child_permissions(
|
||
child_permissions_arg.write,
|
||
|text| {
|
||
Ok::<_, PathResolveError>(Some(
|
||
self.descriptor_parser.parse_write_descriptor(text)?,
|
||
))
|
||
},
|
||
)?;
|
||
worker_perms.import = inner.import.create_child_permissions(
|
||
child_permissions_arg.import,
|
||
|text| {
|
||
Ok::<_, NetDescriptorParseError>(Some(
|
||
self.descriptor_parser.parse_import_descriptor(text)?,
|
||
))
|
||
},
|
||
)?;
|
||
worker_perms.net = inner.net.create_child_permissions(
|
||
child_permissions_arg.net,
|
||
|text| {
|
||
Ok::<_, NetDescriptorParseError>(Some(
|
||
self.descriptor_parser.parse_net_descriptor(text)?,
|
||
))
|
||
},
|
||
)?;
|
||
worker_perms.env = inner.env.create_child_permissions(
|
||
child_permissions_arg.env,
|
||
|text| {
|
||
Ok::<_, EnvDescriptorParseError>(Some(
|
||
self.descriptor_parser.parse_env_descriptor(text)?,
|
||
))
|
||
},
|
||
)?;
|
||
worker_perms.sys = inner.sys.create_child_permissions(
|
||
child_permissions_arg.sys,
|
||
|text| {
|
||
Ok::<_, SysDescriptorParseError>(Some(
|
||
self.descriptor_parser.parse_sys_descriptor(text)?,
|
||
))
|
||
},
|
||
)?;
|
||
worker_perms.run = inner.run.create_child_permissions(
|
||
child_permissions_arg.run,
|
||
|text| match self.descriptor_parser.parse_allow_run_descriptor(text)? {
|
||
AllowRunDescriptorParseResult::Unresolved(_) => {
|
||
Ok::<_, RunDescriptorParseError>(None)
|
||
}
|
||
AllowRunDescriptorParseResult::Descriptor(desc) => Ok(Some(desc)),
|
||
},
|
||
)?;
|
||
worker_perms.ffi = inner.ffi.create_child_permissions(
|
||
child_permissions_arg.ffi,
|
||
|text| {
|
||
Ok::<_, PathResolveError>(Some(
|
||
self.descriptor_parser.parse_ffi_descriptor(text)?,
|
||
))
|
||
},
|
||
)?;
|
||
|
||
Ok(PermissionsContainer::new(
|
||
self.descriptor_parser.clone(),
|
||
worker_perms,
|
||
))
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_specifier(
|
||
&self,
|
||
specifier: &ModuleSpecifier,
|
||
kind: CheckSpecifierKind,
|
||
) -> Result<(), PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
match specifier.scheme() {
|
||
"file" => {
|
||
if inner.read.is_allow_all() || kind == CheckSpecifierKind::Static {
|
||
return Ok(());
|
||
}
|
||
|
||
match url_to_file_path(specifier) {
|
||
Ok(path) => inner
|
||
.read
|
||
.check(
|
||
&PathQueryDescriptor {
|
||
requested: path.to_string_lossy().into_owned(),
|
||
resolved: path,
|
||
}
|
||
.into_read(),
|
||
Some("import()"),
|
||
)
|
||
.map_err(PermissionCheckError::PermissionDenied),
|
||
Err(_) => {
|
||
Err(PermissionCheckError::InvalidFilePath(specifier.clone()))
|
||
}
|
||
}
|
||
}
|
||
"data" => Ok(()),
|
||
"blob" => Ok(()),
|
||
_ => {
|
||
if inner.import.is_allow_all() {
|
||
return Ok(()); // avoid allocation below
|
||
}
|
||
|
||
let desc = self
|
||
.descriptor_parser
|
||
.parse_import_descriptor_from_url(specifier)?;
|
||
inner.import.check(&desc, Some("import()"))?;
|
||
Ok(())
|
||
}
|
||
}
|
||
}
|
||
|
||
#[must_use = "the resolved return value to mitigate time-of-check to time-of-use issues"]
|
||
#[inline(always)]
|
||
pub fn check_read(
|
||
&self,
|
||
path: &str,
|
||
api_name: &str,
|
||
) -> Result<PathBuf, PermissionCheckError> {
|
||
self.check_read_with_api_name(path, Some(api_name))
|
||
}
|
||
|
||
#[must_use = "the resolved return value to mitigate time-of-check to time-of-use issues"]
|
||
#[inline(always)]
|
||
pub fn check_read_with_api_name(
|
||
&self,
|
||
path: &str,
|
||
api_name: Option<&str>,
|
||
) -> Result<PathBuf, PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.read;
|
||
if inner.is_allow_all() {
|
||
Ok(PathBuf::from(path))
|
||
} else {
|
||
let desc = self.descriptor_parser.parse_path_query(path)?.into_read();
|
||
inner.check(&desc, api_name)?;
|
||
Ok(desc.0.resolved)
|
||
}
|
||
}
|
||
|
||
#[must_use = "the resolved return value to mitigate time-of-check to time-of-use issues"]
|
||
#[inline(always)]
|
||
pub fn check_read_path<'a>(
|
||
&self,
|
||
path: &'a Path,
|
||
api_name: Option<&str>,
|
||
) -> Result<Cow<'a, Path>, PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.read;
|
||
if inner.is_allow_all() {
|
||
Ok(Cow::Borrowed(path))
|
||
} else {
|
||
let desc = PathQueryDescriptor {
|
||
requested: path.to_string_lossy().into_owned(),
|
||
resolved: path.to_path_buf(),
|
||
}
|
||
.into_read();
|
||
inner.check(&desc, api_name)?;
|
||
Ok(Cow::Owned(desc.0.resolved))
|
||
}
|
||
}
|
||
|
||
/// As `check_read()`, but permission error messages will anonymize the path
|
||
/// by replacing it with the given `display`.
|
||
#[inline(always)]
|
||
pub fn check_read_blind(
|
||
&mut self,
|
||
path: &Path,
|
||
display: &str,
|
||
api_name: &str,
|
||
) -> Result<(), PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.read;
|
||
skip_check_if_is_permission_fully_granted!(inner);
|
||
inner.check(
|
||
&PathQueryDescriptor {
|
||
requested: format!("<{}>", display),
|
||
resolved: path.to_path_buf(),
|
||
}
|
||
.into_read(),
|
||
Some(api_name),
|
||
)?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_read_all(
|
||
&self,
|
||
api_name: &str,
|
||
) -> Result<(), PermissionCheckError> {
|
||
self.inner.lock().read.check_all(Some(api_name))?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn query_read_all(&self) -> bool {
|
||
self.inner.lock().read.query(None) == PermissionState::Granted
|
||
}
|
||
|
||
#[must_use = "the resolved return value to mitigate time-of-check to time-of-use issues"]
|
||
#[inline(always)]
|
||
pub fn check_write(
|
||
&self,
|
||
path: &str,
|
||
api_name: &str,
|
||
) -> Result<PathBuf, PermissionCheckError> {
|
||
self.check_write_with_api_name(path, Some(api_name))
|
||
}
|
||
|
||
#[must_use = "the resolved return value to mitigate time-of-check to time-of-use issues"]
|
||
#[inline(always)]
|
||
pub fn check_write_with_api_name(
|
||
&self,
|
||
path: &str,
|
||
api_name: Option<&str>,
|
||
) -> Result<PathBuf, PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.write;
|
||
if inner.is_allow_all() {
|
||
Ok(PathBuf::from(path))
|
||
} else {
|
||
let desc = self.descriptor_parser.parse_path_query(path)?.into_write();
|
||
inner.check(&desc, api_name)?;
|
||
Ok(desc.0.resolved)
|
||
}
|
||
}
|
||
|
||
#[must_use = "the resolved return value to mitigate time-of-check to time-of-use issues"]
|
||
#[inline(always)]
|
||
pub fn check_write_path<'a>(
|
||
&self,
|
||
path: &'a Path,
|
||
api_name: &str,
|
||
) -> Result<Cow<'a, Path>, PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.write;
|
||
if inner.is_allow_all() {
|
||
Ok(Cow::Borrowed(path))
|
||
} else {
|
||
let desc = PathQueryDescriptor {
|
||
requested: path.to_string_lossy().into_owned(),
|
||
resolved: path.to_path_buf(),
|
||
}
|
||
.into_write();
|
||
inner.check(&desc, Some(api_name))?;
|
||
Ok(Cow::Owned(desc.0.resolved))
|
||
}
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_write_all(
|
||
&self,
|
||
api_name: &str,
|
||
) -> Result<(), PermissionCheckError> {
|
||
self.inner.lock().write.check_all(Some(api_name))?;
|
||
Ok(())
|
||
}
|
||
|
||
/// As `check_write()`, but permission error messages will anonymize the path
|
||
/// by replacing it with the given `display`.
|
||
#[inline(always)]
|
||
pub fn check_write_blind(
|
||
&self,
|
||
path: &Path,
|
||
display: &str,
|
||
api_name: &str,
|
||
) -> Result<(), PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.write;
|
||
skip_check_if_is_permission_fully_granted!(inner);
|
||
inner.check(
|
||
&PathQueryDescriptor {
|
||
requested: format!("<{}>", display),
|
||
resolved: path.to_path_buf(),
|
||
}
|
||
.into_write(),
|
||
Some(api_name),
|
||
)?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_write_partial(
|
||
&mut self,
|
||
path: &str,
|
||
api_name: &str,
|
||
) -> Result<PathBuf, PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.write;
|
||
if inner.is_allow_all() {
|
||
Ok(PathBuf::from(path))
|
||
} else {
|
||
let desc = self.descriptor_parser.parse_path_query(path)?.into_write();
|
||
inner.check_partial(&desc, Some(api_name))?;
|
||
Ok(desc.0.resolved)
|
||
}
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_run(
|
||
&mut self,
|
||
cmd: &RunQueryDescriptor,
|
||
api_name: &str,
|
||
) -> Result<(), PermissionCheckError> {
|
||
self.inner.lock().run.check(cmd, Some(api_name))?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_run_all(
|
||
&mut self,
|
||
api_name: &str,
|
||
) -> Result<(), PermissionCheckError> {
|
||
self.inner.lock().run.check_all(Some(api_name))?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn query_run_all(&mut self, api_name: &str) -> bool {
|
||
self.inner.lock().run.query_all(Some(api_name))
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_sys(
|
||
&self,
|
||
kind: &str,
|
||
api_name: &str,
|
||
) -> Result<(), PermissionCheckError> {
|
||
self.inner.lock().sys.check(
|
||
&self.descriptor_parser.parse_sys_descriptor(kind)?,
|
||
Some(api_name),
|
||
)?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_env(&mut self, var: &str) -> Result<(), PermissionCheckError> {
|
||
self.inner.lock().env.check(var, None)?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_env_all(&mut self) -> Result<(), PermissionCheckError> {
|
||
self.inner.lock().env.check_all()?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_sys_all(&mut self) -> Result<(), PermissionCheckError> {
|
||
self.inner.lock().sys.check_all()?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_ffi_all(&mut self) -> Result<(), PermissionCheckError> {
|
||
self.inner.lock().ffi.check_all()?;
|
||
Ok(())
|
||
}
|
||
|
||
/// This checks to see if the allow-all flag was passed, not whether all
|
||
/// permissions are enabled!
|
||
#[inline(always)]
|
||
pub fn check_was_allow_all_flag_passed(
|
||
&mut self,
|
||
) -> Result<(), PermissionCheckError> {
|
||
self.inner.lock().all.check()?;
|
||
Ok(())
|
||
}
|
||
|
||
/// Checks special file access, returning the failed permission type if
|
||
/// not successful.
|
||
pub fn check_special_file(
|
||
&mut self,
|
||
path: &Path,
|
||
_api_name: &str,
|
||
) -> Result<(), &'static str> {
|
||
let error_all = |_| "all";
|
||
|
||
// Safe files with no major additional side-effects. While there's a small risk of someone
|
||
// draining system entropy by just reading one of these files constantly, that's not really
|
||
// something we worry about as they already have --allow-read to /dev.
|
||
if cfg!(unix)
|
||
&& (path == OsStr::new("/dev/random")
|
||
|| path == OsStr::new("/dev/urandom")
|
||
|| path == OsStr::new("/dev/zero")
|
||
|| path == OsStr::new("/dev/null"))
|
||
{
|
||
return Ok(());
|
||
}
|
||
|
||
/// We'll allow opening /proc/self/fd/{n} without additional permissions under the following conditions:
|
||
///
|
||
/// 1. n > 2. This allows for opening bash-style redirections, but not stdio
|
||
/// 2. the fd referred to by n is a pipe
|
||
#[cfg(unix)]
|
||
fn is_fd_file_is_pipe(path: &Path) -> bool {
|
||
if let Some(fd) = path.file_name() {
|
||
if let Ok(s) = std::str::from_utf8(fd.as_encoded_bytes()) {
|
||
if let Ok(n) = s.parse::<i32>() {
|
||
if n > 2 {
|
||
// SAFETY: This is proper use of the stat syscall
|
||
unsafe {
|
||
let mut stat = std::mem::zeroed::<libc::stat>();
|
||
if libc::fstat(n, &mut stat as _) == 0
|
||
&& ((stat.st_mode & libc::S_IFMT) & libc::S_IFIFO) != 0
|
||
{
|
||
return true;
|
||
}
|
||
};
|
||
}
|
||
}
|
||
}
|
||
}
|
||
false
|
||
}
|
||
|
||
// On unixy systems, we allow opening /dev/fd/XXX for valid FDs that
|
||
// are pipes.
|
||
#[cfg(unix)]
|
||
if path.starts_with("/dev/fd") && is_fd_file_is_pipe(path) {
|
||
return Ok(());
|
||
}
|
||
|
||
if cfg!(target_os = "linux") {
|
||
// On Linux, we also allow opening /proc/self/fd/XXX for valid FDs that
|
||
// are pipes.
|
||
#[cfg(unix)]
|
||
if path.starts_with("/proc/self/fd") && is_fd_file_is_pipe(path) {
|
||
return Ok(());
|
||
}
|
||
if path.starts_with("/dev")
|
||
|| path.starts_with("/proc")
|
||
|| path.starts_with("/sys")
|
||
{
|
||
if path.ends_with("/environ") {
|
||
self.check_env_all().map_err(|_| "env")?;
|
||
} else {
|
||
self.check_was_allow_all_flag_passed().map_err(error_all)?;
|
||
}
|
||
}
|
||
} else if cfg!(unix) {
|
||
if path.starts_with("/dev") {
|
||
self.check_was_allow_all_flag_passed().map_err(error_all)?;
|
||
}
|
||
} else if cfg!(target_os = "windows") {
|
||
// \\.\nul is allowed
|
||
let s = path.as_os_str().as_encoded_bytes();
|
||
if s.eq_ignore_ascii_case(br#"\\.\nul"#) {
|
||
return Ok(());
|
||
}
|
||
|
||
fn is_normalized_windows_drive_path(path: &Path) -> bool {
|
||
let s = path.as_os_str().as_encoded_bytes();
|
||
// \\?\X:\
|
||
if s.len() < 7 {
|
||
false
|
||
} else if s.starts_with(br#"\\?\"#) {
|
||
s[4].is_ascii_alphabetic() && s[5] == b':' && s[6] == b'\\'
|
||
} else {
|
||
false
|
||
}
|
||
}
|
||
|
||
// If this is a normalized drive path, accept it
|
||
if !is_normalized_windows_drive_path(path) {
|
||
self.check_was_allow_all_flag_passed().map_err(error_all)?;
|
||
}
|
||
} else {
|
||
unimplemented!()
|
||
}
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_net_url(
|
||
&mut self,
|
||
url: &Url,
|
||
api_name: &str,
|
||
) -> Result<(), PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
if inner.net.is_allow_all() {
|
||
return Ok(());
|
||
}
|
||
let desc = self.descriptor_parser.parse_net_descriptor_from_url(url)?;
|
||
inner.net.check(&desc, Some(api_name))?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_net<T: AsRef<str>>(
|
||
&mut self,
|
||
host: &(T, Option<u16>),
|
||
api_name: &str,
|
||
) -> Result<(), PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.net;
|
||
skip_check_if_is_permission_fully_granted!(inner);
|
||
let hostname = Host::parse(host.0.as_ref())?;
|
||
let descriptor = NetDescriptor(hostname, host.1);
|
||
inner.check(&descriptor, Some(api_name))?;
|
||
Ok(())
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn check_ffi(
|
||
&mut self,
|
||
path: &str,
|
||
) -> Result<PathBuf, PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.ffi;
|
||
if inner.is_allow_all() {
|
||
Ok(PathBuf::from(path))
|
||
} else {
|
||
let desc = self.descriptor_parser.parse_path_query(path)?.into_ffi();
|
||
inner.check(&desc, None)?;
|
||
Ok(desc.0.resolved)
|
||
}
|
||
}
|
||
|
||
#[must_use = "the resolved return value to mitigate time-of-check to time-of-use issues"]
|
||
#[inline(always)]
|
||
pub fn check_ffi_partial_no_path(
|
||
&mut self,
|
||
) -> Result<(), PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.ffi;
|
||
if !inner.is_allow_all() {
|
||
inner.check_partial(None)?;
|
||
}
|
||
Ok(())
|
||
}
|
||
|
||
#[must_use = "the resolved return value to mitigate time-of-check to time-of-use issues"]
|
||
#[inline(always)]
|
||
pub fn check_ffi_partial_with_path(
|
||
&mut self,
|
||
path: &str,
|
||
) -> Result<PathBuf, PermissionCheckError> {
|
||
let mut inner = self.inner.lock();
|
||
let inner = &mut inner.ffi;
|
||
if inner.is_allow_all() {
|
||
Ok(PathBuf::from(path))
|
||
} else {
|
||
let desc = self.descriptor_parser.parse_path_query(path)?.into_ffi();
|
||
inner.check_partial(Some(&desc))?;
|
||
Ok(desc.0.resolved)
|
||
}
|
||
}
|
||
|
||
// query
|
||
|
||
#[inline(always)]
|
||
pub fn query_read(
|
||
&self,
|
||
path: Option<&str>,
|
||
) -> Result<PermissionState, PathResolveError> {
|
||
let inner = self.inner.lock();
|
||
let permission = &inner.read;
|
||
if permission.is_allow_all() {
|
||
return Ok(PermissionState::Granted);
|
||
}
|
||
Ok(
|
||
permission.query(
|
||
path
|
||
.map(|path| {
|
||
Ok::<_, PathResolveError>(
|
||
self.descriptor_parser.parse_path_query(path)?.into_read(),
|
||
)
|
||
})
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn query_write(
|
||
&self,
|
||
path: Option<&str>,
|
||
) -> Result<PermissionState, PathResolveError> {
|
||
let inner = self.inner.lock();
|
||
let permission = &inner.write;
|
||
if permission.is_allow_all() {
|
||
return Ok(PermissionState::Granted);
|
||
}
|
||
Ok(
|
||
permission.query(
|
||
path
|
||
.map(|path| {
|
||
Ok::<_, PathResolveError>(
|
||
self.descriptor_parser.parse_path_query(path)?.into_write(),
|
||
)
|
||
})
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn query_net(
|
||
&self,
|
||
host: Option<&str>,
|
||
) -> Result<PermissionState, NetDescriptorParseError> {
|
||
let inner = self.inner.lock();
|
||
let permission = &inner.net;
|
||
if permission.is_allow_all() {
|
||
return Ok(PermissionState::Granted);
|
||
}
|
||
Ok(
|
||
permission.query(
|
||
match host {
|
||
None => None,
|
||
Some(h) => Some(self.descriptor_parser.parse_net_descriptor(h)?),
|
||
}
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn query_env(&self, var: Option<&str>) -> PermissionState {
|
||
let inner = self.inner.lock();
|
||
let permission = &inner.env;
|
||
if permission.is_allow_all() {
|
||
return PermissionState::Granted;
|
||
}
|
||
permission.query(var)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn query_sys(
|
||
&self,
|
||
kind: Option<&str>,
|
||
) -> Result<PermissionState, SysDescriptorParseError> {
|
||
let inner = self.inner.lock();
|
||
let permission = &inner.sys;
|
||
if permission.is_allow_all() {
|
||
return Ok(PermissionState::Granted);
|
||
}
|
||
Ok(
|
||
permission.query(
|
||
kind
|
||
.map(|kind| self.descriptor_parser.parse_sys_descriptor(kind))
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn query_run(
|
||
&self,
|
||
cmd: Option<&str>,
|
||
) -> Result<PermissionState, RunDescriptorParseError> {
|
||
let inner = self.inner.lock();
|
||
let permission = &inner.run;
|
||
if permission.is_allow_all() {
|
||
return Ok(PermissionState::Granted);
|
||
}
|
||
Ok(
|
||
permission.query(
|
||
cmd
|
||
.map(|request| self.descriptor_parser.parse_run_query(request))
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn query_ffi(
|
||
&self,
|
||
path: Option<&str>,
|
||
) -> Result<PermissionState, PathResolveError> {
|
||
let inner = self.inner.lock();
|
||
let permission = &inner.ffi;
|
||
if permission.is_allow_all() {
|
||
return Ok(PermissionState::Granted);
|
||
}
|
||
Ok(
|
||
permission.query(
|
||
path
|
||
.map(|path| {
|
||
Ok::<_, PathResolveError>(
|
||
self.descriptor_parser.parse_path_query(path)?.into_ffi(),
|
||
)
|
||
})
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
// revoke
|
||
|
||
#[inline(always)]
|
||
pub fn revoke_read(
|
||
&self,
|
||
path: Option<&str>,
|
||
) -> Result<PermissionState, PathResolveError> {
|
||
Ok(
|
||
self.inner.lock().read.revoke(
|
||
path
|
||
.map(|path| {
|
||
Ok::<_, PathResolveError>(
|
||
self.descriptor_parser.parse_path_query(path)?.into_read(),
|
||
)
|
||
})
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn revoke_write(
|
||
&self,
|
||
path: Option<&str>,
|
||
) -> Result<PermissionState, PathResolveError> {
|
||
Ok(
|
||
self.inner.lock().write.revoke(
|
||
path
|
||
.map(|path| {
|
||
Ok::<_, PathResolveError>(
|
||
self.descriptor_parser.parse_path_query(path)?.into_write(),
|
||
)
|
||
})
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn revoke_net(
|
||
&self,
|
||
host: Option<&str>,
|
||
) -> Result<PermissionState, NetDescriptorParseError> {
|
||
Ok(
|
||
self.inner.lock().net.revoke(
|
||
match host {
|
||
None => None,
|
||
Some(h) => Some(self.descriptor_parser.parse_net_descriptor(h)?),
|
||
}
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn revoke_env(&self, var: Option<&str>) -> PermissionState {
|
||
self.inner.lock().env.revoke(var)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn revoke_sys(
|
||
&self,
|
||
kind: Option<&str>,
|
||
) -> Result<PermissionState, SysDescriptorParseError> {
|
||
Ok(
|
||
self.inner.lock().sys.revoke(
|
||
kind
|
||
.map(|kind| self.descriptor_parser.parse_sys_descriptor(kind))
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn revoke_run(
|
||
&self,
|
||
cmd: Option<&str>,
|
||
) -> Result<PermissionState, RunDescriptorParseError> {
|
||
Ok(
|
||
self.inner.lock().run.revoke(
|
||
cmd
|
||
.map(|request| self.descriptor_parser.parse_run_query(request))
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn revoke_ffi(
|
||
&self,
|
||
path: Option<&str>,
|
||
) -> Result<PermissionState, PathResolveError> {
|
||
Ok(
|
||
self.inner.lock().ffi.revoke(
|
||
path
|
||
.map(|path| {
|
||
Ok::<_, PathResolveError>(
|
||
self.descriptor_parser.parse_path_query(path)?.into_ffi(),
|
||
)
|
||
})
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
// request
|
||
|
||
#[inline(always)]
|
||
pub fn request_read(
|
||
&self,
|
||
path: Option<&str>,
|
||
) -> Result<PermissionState, PathResolveError> {
|
||
Ok(
|
||
self.inner.lock().read.request(
|
||
path
|
||
.map(|path| {
|
||
Ok::<_, PathResolveError>(
|
||
self.descriptor_parser.parse_path_query(path)?.into_read(),
|
||
)
|
||
})
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn request_write(
|
||
&self,
|
||
path: Option<&str>,
|
||
) -> Result<PermissionState, PathResolveError> {
|
||
Ok(
|
||
self.inner.lock().write.request(
|
||
path
|
||
.map(|path| {
|
||
Ok::<_, PathResolveError>(
|
||
self.descriptor_parser.parse_path_query(path)?.into_write(),
|
||
)
|
||
})
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn request_net(
|
||
&self,
|
||
host: Option<&str>,
|
||
) -> Result<PermissionState, NetDescriptorParseError> {
|
||
Ok(
|
||
self.inner.lock().net.request(
|
||
match host {
|
||
None => None,
|
||
Some(h) => Some(self.descriptor_parser.parse_net_descriptor(h)?),
|
||
}
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn request_env(&self, var: Option<&str>) -> PermissionState {
|
||
self.inner.lock().env.request(var)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn request_sys(
|
||
&self,
|
||
kind: Option<&str>,
|
||
) -> Result<PermissionState, SysDescriptorParseError> {
|
||
Ok(
|
||
self.inner.lock().sys.request(
|
||
kind
|
||
.map(|kind| self.descriptor_parser.parse_sys_descriptor(kind))
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn request_run(
|
||
&self,
|
||
cmd: Option<&str>,
|
||
) -> Result<PermissionState, RunDescriptorParseError> {
|
||
Ok(
|
||
self.inner.lock().run.request(
|
||
cmd
|
||
.map(|request| self.descriptor_parser.parse_run_query(request))
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
|
||
#[inline(always)]
|
||
pub fn request_ffi(
|
||
&self,
|
||
path: Option<&str>,
|
||
) -> Result<PermissionState, PathResolveError> {
|
||
Ok(
|
||
self.inner.lock().ffi.request(
|
||
path
|
||
.map(|path| {
|
||
Ok::<_, PathResolveError>(
|
||
self.descriptor_parser.parse_path_query(path)?.into_ffi(),
|
||
)
|
||
})
|
||
.transpose()?
|
||
.as_ref(),
|
||
),
|
||
)
|
||
}
|
||
}
|
||
|
||
const fn unit_permission_from_flag_bools(
|
||
allow_flag: bool,
|
||
deny_flag: bool,
|
||
name: &'static str,
|
||
description: &'static str,
|
||
prompt: bool,
|
||
) -> UnitPermission {
|
||
UnitPermission {
|
||
name,
|
||
description,
|
||
state: if deny_flag {
|
||
PermissionState::Denied
|
||
} else if allow_flag {
|
||
PermissionState::Granted
|
||
} else {
|
||
PermissionState::Prompt
|
||
},
|
||
prompt,
|
||
}
|
||
}
|
||
|
||
fn global_from_option<T>(flag: Option<&HashSet<T>>) -> bool {
|
||
matches!(flag, Some(v) if v.is_empty())
|
||
}
|
||
|
||
#[derive(Debug, Eq, PartialEq)]
|
||
pub enum ChildUnitPermissionArg {
|
||
Inherit,
|
||
Granted,
|
||
NotGranted,
|
||
}
|
||
|
||
impl<'de> Deserialize<'de> for ChildUnitPermissionArg {
|
||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||
where
|
||
D: Deserializer<'de>,
|
||
{
|
||
struct ChildUnitPermissionArgVisitor;
|
||
impl<'de> de::Visitor<'de> for ChildUnitPermissionArgVisitor {
|
||
type Value = ChildUnitPermissionArg;
|
||
|
||
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
|
||
formatter.write_str("\"inherit\" or boolean")
|
||
}
|
||
|
||
fn visit_unit<E>(self) -> Result<ChildUnitPermissionArg, E>
|
||
where
|
||
E: de::Error,
|
||
{
|
||
Ok(ChildUnitPermissionArg::NotGranted)
|
||
}
|
||
|
||
fn visit_str<E>(self, v: &str) -> Result<ChildUnitPermissionArg, E>
|
||
where
|
||
E: de::Error,
|
||
{
|
||
if v == "inherit" {
|
||
Ok(ChildUnitPermissionArg::Inherit)
|
||
} else {
|
||
Err(de::Error::invalid_value(de::Unexpected::Str(v), &self))
|
||
}
|
||
}
|
||
|
||
fn visit_bool<E>(self, v: bool) -> Result<ChildUnitPermissionArg, E>
|
||
where
|
||
E: de::Error,
|
||
{
|
||
match v {
|
||
true => Ok(ChildUnitPermissionArg::Granted),
|
||
false => Ok(ChildUnitPermissionArg::NotGranted),
|
||
}
|
||
}
|
||
}
|
||
deserializer.deserialize_any(ChildUnitPermissionArgVisitor)
|
||
}
|
||
}
|
||
|
||
#[derive(Debug, Eq, PartialEq)]
|
||
pub enum ChildUnaryPermissionArg {
|
||
Inherit,
|
||
Granted,
|
||
NotGranted,
|
||
GrantedList(Vec<String>),
|
||
}
|
||
|
||
impl<'de> Deserialize<'de> for ChildUnaryPermissionArg {
|
||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||
where
|
||
D: Deserializer<'de>,
|
||
{
|
||
struct ChildUnaryPermissionArgVisitor;
|
||
impl<'de> de::Visitor<'de> for ChildUnaryPermissionArgVisitor {
|
||
type Value = ChildUnaryPermissionArg;
|
||
|
||
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
|
||
formatter.write_str("\"inherit\" or boolean or string[]")
|
||
}
|
||
|
||
fn visit_unit<E>(self) -> Result<ChildUnaryPermissionArg, E>
|
||
where
|
||
E: de::Error,
|
||
{
|
||
Ok(ChildUnaryPermissionArg::NotGranted)
|
||
}
|
||
|
||
fn visit_str<E>(self, v: &str) -> Result<ChildUnaryPermissionArg, E>
|
||
where
|
||
E: de::Error,
|
||
{
|
||
if v == "inherit" {
|
||
Ok(ChildUnaryPermissionArg::Inherit)
|
||
} else {
|
||
Err(de::Error::invalid_value(de::Unexpected::Str(v), &self))
|
||
}
|
||
}
|
||
|
||
fn visit_bool<E>(self, v: bool) -> Result<ChildUnaryPermissionArg, E>
|
||
where
|
||
E: de::Error,
|
||
{
|
||
match v {
|
||
true => Ok(ChildUnaryPermissionArg::Granted),
|
||
false => Ok(ChildUnaryPermissionArg::NotGranted),
|
||
}
|
||
}
|
||
|
||
fn visit_seq<V>(
|
||
self,
|
||
mut v: V,
|
||
) -> Result<ChildUnaryPermissionArg, V::Error>
|
||
where
|
||
V: de::SeqAccess<'de>,
|
||
{
|
||
let mut granted_list = vec![];
|
||
while let Some(value) = v.next_element::<String>()? {
|
||
granted_list.push(value);
|
||
}
|
||
Ok(ChildUnaryPermissionArg::GrantedList(granted_list))
|
||
}
|
||
}
|
||
deserializer.deserialize_any(ChildUnaryPermissionArgVisitor)
|
||
}
|
||
}
|
||
|
||
/// Directly deserializable from JS worker and test permission options.
|
||
#[derive(Debug, Eq, PartialEq)]
|
||
pub struct ChildPermissionsArg {
|
||
env: ChildUnaryPermissionArg,
|
||
net: ChildUnaryPermissionArg,
|
||
ffi: ChildUnaryPermissionArg,
|
||
import: ChildUnaryPermissionArg,
|
||
read: ChildUnaryPermissionArg,
|
||
run: ChildUnaryPermissionArg,
|
||
sys: ChildUnaryPermissionArg,
|
||
write: ChildUnaryPermissionArg,
|
||
}
|
||
|
||
impl ChildPermissionsArg {
|
||
pub fn inherit() -> Self {
|
||
ChildPermissionsArg {
|
||
env: ChildUnaryPermissionArg::Inherit,
|
||
net: ChildUnaryPermissionArg::Inherit,
|
||
ffi: ChildUnaryPermissionArg::Inherit,
|
||
import: ChildUnaryPermissionArg::Inherit,
|
||
read: ChildUnaryPermissionArg::Inherit,
|
||
run: ChildUnaryPermissionArg::Inherit,
|
||
sys: ChildUnaryPermissionArg::Inherit,
|
||
write: ChildUnaryPermissionArg::Inherit,
|
||
}
|
||
}
|
||
|
||
pub fn none() -> Self {
|
||
ChildPermissionsArg {
|
||
env: ChildUnaryPermissionArg::NotGranted,
|
||
net: ChildUnaryPermissionArg::NotGranted,
|
||
ffi: ChildUnaryPermissionArg::NotGranted,
|
||
import: ChildUnaryPermissionArg::NotGranted,
|
||
read: ChildUnaryPermissionArg::NotGranted,
|
||
run: ChildUnaryPermissionArg::NotGranted,
|
||
sys: ChildUnaryPermissionArg::NotGranted,
|
||
write: ChildUnaryPermissionArg::NotGranted,
|
||
}
|
||
}
|
||
}
|
||
|
||
impl<'de> Deserialize<'de> for ChildPermissionsArg {
|
||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||
where
|
||
D: Deserializer<'de>,
|
||
{
|
||
struct ChildPermissionsArgVisitor;
|
||
impl<'de> de::Visitor<'de> for ChildPermissionsArgVisitor {
|
||
type Value = ChildPermissionsArg;
|
||
|
||
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
|
||
formatter.write_str("\"inherit\" or \"none\" or object")
|
||
}
|
||
|
||
fn visit_unit<E>(self) -> Result<ChildPermissionsArg, E>
|
||
where
|
||
E: de::Error,
|
||
{
|
||
Ok(ChildPermissionsArg::inherit())
|
||
}
|
||
|
||
fn visit_str<E>(self, v: &str) -> Result<ChildPermissionsArg, E>
|
||
where
|
||
E: de::Error,
|
||
{
|
||
if v == "inherit" {
|
||
Ok(ChildPermissionsArg::inherit())
|
||
} else if v == "none" {
|
||
Ok(ChildPermissionsArg::none())
|
||
} else {
|
||
Err(de::Error::invalid_value(de::Unexpected::Str(v), &self))
|
||
}
|
||
}
|
||
|
||
fn visit_map<V>(self, mut v: V) -> Result<ChildPermissionsArg, V::Error>
|
||
where
|
||
V: de::MapAccess<'de>,
|
||
{
|
||
let mut child_permissions_arg = ChildPermissionsArg::none();
|
||
while let Some((key, value)) =
|
||
v.next_entry::<String, serde_json::Value>()?
|
||
{
|
||
if key == "env" {
|
||
let arg = serde_json::from_value::<ChildUnaryPermissionArg>(value);
|
||
child_permissions_arg.env = arg.map_err(|e| {
|
||
de::Error::custom(format!("(deno.permissions.env) {e}"))
|
||
})?;
|
||
} else if key == "net" {
|
||
let arg = serde_json::from_value::<ChildUnaryPermissionArg>(value);
|
||
child_permissions_arg.net = arg.map_err(|e| {
|
||
de::Error::custom(format!("(deno.permissions.net) {e}"))
|
||
})?;
|
||
} else if key == "ffi" {
|
||
let arg = serde_json::from_value::<ChildUnaryPermissionArg>(value);
|
||
child_permissions_arg.ffi = arg.map_err(|e| {
|
||
de::Error::custom(format!("(deno.permissions.ffi) {e}"))
|
||
})?;
|
||
} else if key == "import" {
|
||
let arg = serde_json::from_value::<ChildUnaryPermissionArg>(value);
|
||
child_permissions_arg.import = arg.map_err(|e| {
|
||
de::Error::custom(format!("(deno.permissions.import) {e}"))
|
||
})?;
|
||
} else if key == "read" {
|
||
let arg = serde_json::from_value::<ChildUnaryPermissionArg>(value);
|
||
child_permissions_arg.read = arg.map_err(|e| {
|
||
de::Error::custom(format!("(deno.permissions.read) {e}"))
|
||
})?;
|
||
} else if key == "run" {
|
||
let arg = serde_json::from_value::<ChildUnaryPermissionArg>(value);
|
||
child_permissions_arg.run = arg.map_err(|e| {
|
||
de::Error::custom(format!("(deno.permissions.run) {e}"))
|
||
})?;
|
||
} else if key == "sys" {
|
||
let arg = serde_json::from_value::<ChildUnaryPermissionArg>(value);
|
||
child_permissions_arg.sys = arg.map_err(|e| {
|
||
de::Error::custom(format!("(deno.permissions.sys) {e}"))
|
||
})?;
|
||
} else if key == "write" {
|
||
let arg = serde_json::from_value::<ChildUnaryPermissionArg>(value);
|
||
child_permissions_arg.write = arg.map_err(|e| {
|
||
de::Error::custom(format!("(deno.permissions.write) {e}"))
|
||
})?;
|
||
} else {
|
||
return Err(de::Error::custom("unknown permission name"));
|
||
}
|
||
}
|
||
Ok(child_permissions_arg)
|
||
}
|
||
}
|
||
deserializer.deserialize_any(ChildPermissionsArgVisitor)
|
||
}
|
||
}
|
||
|
||
/// Parses and normalizes permissions.
|
||
///
|
||
/// This trait is necessary because this crate doesn't have access
|
||
/// to the file system.
|
||
pub trait PermissionDescriptorParser: Debug + Send + Sync {
|
||
fn parse_read_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<ReadDescriptor, PathResolveError>;
|
||
|
||
fn parse_write_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<WriteDescriptor, PathResolveError>;
|
||
|
||
fn parse_net_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<NetDescriptor, NetDescriptorParseError>;
|
||
|
||
fn parse_net_descriptor_from_url(
|
||
&self,
|
||
url: &Url,
|
||
) -> Result<NetDescriptor, NetDescriptorFromUrlParseError> {
|
||
NetDescriptor::from_url(url)
|
||
}
|
||
|
||
fn parse_import_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<ImportDescriptor, NetDescriptorParseError>;
|
||
|
||
fn parse_import_descriptor_from_url(
|
||
&self,
|
||
url: &Url,
|
||
) -> Result<ImportDescriptor, NetDescriptorFromUrlParseError> {
|
||
ImportDescriptor::from_url(url)
|
||
}
|
||
|
||
fn parse_env_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<EnvDescriptor, EnvDescriptorParseError>;
|
||
|
||
fn parse_sys_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<SysDescriptor, SysDescriptorParseError>;
|
||
|
||
fn parse_allow_run_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<AllowRunDescriptorParseResult, RunDescriptorParseError>;
|
||
|
||
fn parse_deny_run_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<DenyRunDescriptor, PathResolveError>;
|
||
|
||
fn parse_ffi_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<FfiDescriptor, PathResolveError>;
|
||
|
||
// queries
|
||
|
||
fn parse_path_query(
|
||
&self,
|
||
path: &str,
|
||
) -> Result<PathQueryDescriptor, PathResolveError>;
|
||
|
||
fn parse_run_query(
|
||
&self,
|
||
requested: &str,
|
||
) -> Result<RunQueryDescriptor, RunDescriptorParseError>;
|
||
}
|
||
|
||
static IS_STANDALONE: AtomicFlag = AtomicFlag::lowered();
|
||
|
||
pub fn mark_standalone() {
|
||
IS_STANDALONE.raise();
|
||
}
|
||
|
||
pub fn is_standalone() -> bool {
|
||
IS_STANDALONE.is_raised()
|
||
}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
use super::*;
|
||
use deno_core::serde_json::json;
|
||
use fqdn::fqdn;
|
||
use prompter::tests::*;
|
||
use std::net::Ipv4Addr;
|
||
|
||
// Creates vector of strings, Vec<String>
|
||
macro_rules! svec {
|
||
($($x:expr),*) => (vec![$($x.to_string()),*]);
|
||
}
|
||
|
||
#[derive(Debug, Clone)]
|
||
struct TestPermissionDescriptorParser;
|
||
|
||
impl TestPermissionDescriptorParser {
|
||
fn join_path_with_root(&self, path: &str) -> PathBuf {
|
||
if path.starts_with("C:\\") {
|
||
PathBuf::from(path)
|
||
} else {
|
||
PathBuf::from("/").join(path)
|
||
}
|
||
}
|
||
}
|
||
|
||
impl PermissionDescriptorParser for TestPermissionDescriptorParser {
|
||
fn parse_read_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<ReadDescriptor, PathResolveError> {
|
||
Ok(ReadDescriptor(self.join_path_with_root(text)))
|
||
}
|
||
|
||
fn parse_write_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<WriteDescriptor, PathResolveError> {
|
||
Ok(WriteDescriptor(self.join_path_with_root(text)))
|
||
}
|
||
|
||
fn parse_net_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<NetDescriptor, NetDescriptorParseError> {
|
||
NetDescriptor::parse(text)
|
||
}
|
||
|
||
fn parse_import_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<ImportDescriptor, NetDescriptorParseError> {
|
||
ImportDescriptor::parse(text)
|
||
}
|
||
|
||
fn parse_env_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<EnvDescriptor, EnvDescriptorParseError> {
|
||
Ok(EnvDescriptor::new(text))
|
||
}
|
||
|
||
fn parse_sys_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<SysDescriptor, SysDescriptorParseError> {
|
||
SysDescriptor::parse(text.to_string())
|
||
}
|
||
|
||
fn parse_allow_run_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<AllowRunDescriptorParseResult, RunDescriptorParseError> {
|
||
Ok(AllowRunDescriptorParseResult::Descriptor(
|
||
AllowRunDescriptor(self.join_path_with_root(text)),
|
||
))
|
||
}
|
||
|
||
fn parse_deny_run_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<DenyRunDescriptor, PathResolveError> {
|
||
if text.contains("/") {
|
||
Ok(DenyRunDescriptor::Path(self.join_path_with_root(text)))
|
||
} else {
|
||
Ok(DenyRunDescriptor::Name(text.to_string()))
|
||
}
|
||
}
|
||
|
||
fn parse_ffi_descriptor(
|
||
&self,
|
||
text: &str,
|
||
) -> Result<FfiDescriptor, PathResolveError> {
|
||
Ok(FfiDescriptor(self.join_path_with_root(text)))
|
||
}
|
||
|
||
fn parse_path_query(
|
||
&self,
|
||
path: &str,
|
||
) -> Result<PathQueryDescriptor, PathResolveError> {
|
||
Ok(PathQueryDescriptor {
|
||
resolved: self.join_path_with_root(path),
|
||
requested: path.to_string(),
|
||
})
|
||
}
|
||
|
||
fn parse_run_query(
|
||
&self,
|
||
requested: &str,
|
||
) -> Result<RunQueryDescriptor, RunDescriptorParseError> {
|
||
RunQueryDescriptor::parse(requested).map_err(Into::into)
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn check_paths() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let allowlist = svec!["/a/specific/dir/name", "/a/specific", "/b/c"];
|
||
|
||
let parser = TestPermissionDescriptorParser;
|
||
let perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_read: Some(allowlist.clone()),
|
||
allow_write: Some(allowlist.clone()),
|
||
allow_ffi: Some(allowlist),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
let mut perms = PermissionsContainer::new(Arc::new(parser), perms);
|
||
|
||
let cases = [
|
||
// Inside of /a/specific and /a/specific/dir/name
|
||
("/a/specific/dir/name", true),
|
||
// Inside of /a/specific but outside of /a/specific/dir/name
|
||
("/a/specific/dir", true),
|
||
// Inside of /a/specific and /a/specific/dir/name
|
||
("/a/specific/dir/name/inner", true),
|
||
// Inside of /a/specific but outside of /a/specific/dir/name
|
||
("/a/specific/other/dir", true),
|
||
// Exact match with /b/c
|
||
("/b/c", true),
|
||
// Sub path within /b/c
|
||
("/b/c/sub/path", true),
|
||
// Sub path within /b/c, needs normalizing
|
||
("/b/c/sub/path/../path/.", true),
|
||
// Inside of /b but outside of /b/c
|
||
("/b/e", false),
|
||
// Inside of /a but outside of /a/specific
|
||
("/a/b", false),
|
||
];
|
||
|
||
for (path, is_ok) in cases {
|
||
assert_eq!(perms.check_read(path, "api").is_ok(), is_ok);
|
||
assert_eq!(perms.check_write(path, "api").is_ok(), is_ok);
|
||
assert_eq!(perms.check_ffi(path).is_ok(), is_ok);
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn test_check_net_with_values() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let parser = TestPermissionDescriptorParser;
|
||
let mut perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_net: Some(svec![
|
||
"localhost",
|
||
"deno.land",
|
||
"github.com:3000",
|
||
"127.0.0.1",
|
||
"172.16.0.2:8000",
|
||
"www.github.com:443",
|
||
"80.example.com:80",
|
||
"443.example.com:443"
|
||
]),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
|
||
let domain_tests = vec![
|
||
("localhost", 1234, true),
|
||
("deno.land", 0, true),
|
||
("deno.land", 3000, true),
|
||
("deno.lands", 0, false),
|
||
("deno.lands", 3000, false),
|
||
("github.com", 3000, true),
|
||
("github.com", 0, false),
|
||
("github.com", 2000, false),
|
||
("github.net", 3000, false),
|
||
("127.0.0.1", 0, true),
|
||
("127.0.0.1", 3000, true),
|
||
("127.0.0.2", 0, false),
|
||
("127.0.0.2", 3000, false),
|
||
("172.16.0.2", 8000, true),
|
||
("172.16.0.2", 0, false),
|
||
("172.16.0.2", 6000, false),
|
||
("172.16.0.1", 8000, false),
|
||
("443.example.com", 444, false),
|
||
("80.example.com", 81, false),
|
||
("80.example.com", 80, true),
|
||
// Just some random hosts that should err
|
||
("somedomain", 0, false),
|
||
("192.168.0.1", 0, false),
|
||
];
|
||
|
||
for (host, port, is_ok) in domain_tests {
|
||
let host = Host::parse(host).unwrap();
|
||
let descriptor = NetDescriptor(host, Some(port));
|
||
assert_eq!(
|
||
is_ok,
|
||
perms.net.check(&descriptor, None).is_ok(),
|
||
"{descriptor}",
|
||
);
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn test_check_net_only_flag() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let parser = TestPermissionDescriptorParser;
|
||
let mut perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_net: Some(svec![]), // this means `--allow-net` is present without values following `=` sign
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
|
||
let domain_tests = vec![
|
||
("localhost", 1234),
|
||
("deno.land", 0),
|
||
("deno.land", 3000),
|
||
("deno.lands", 0),
|
||
("deno.lands", 3000),
|
||
("github.com", 3000),
|
||
("github.com", 0),
|
||
("github.com", 2000),
|
||
("github.net", 3000),
|
||
("127.0.0.1", 0),
|
||
("127.0.0.1", 3000),
|
||
("127.0.0.2", 0),
|
||
("127.0.0.2", 3000),
|
||
("172.16.0.2", 8000),
|
||
("172.16.0.2", 0),
|
||
("172.16.0.2", 6000),
|
||
("172.16.0.1", 8000),
|
||
("somedomain", 0),
|
||
("192.168.0.1", 0),
|
||
];
|
||
|
||
for (host_str, port) in domain_tests {
|
||
let host = Host::parse(host_str).unwrap();
|
||
let descriptor = NetDescriptor(host, Some(port));
|
||
assert!(
|
||
perms.net.check(&descriptor, None).is_ok(),
|
||
"expected {host_str}:{port} to pass"
|
||
);
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn test_check_net_no_flag() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let parser = TestPermissionDescriptorParser;
|
||
let mut perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_net: None,
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
|
||
let domain_tests = vec![
|
||
("localhost", 1234),
|
||
("deno.land", 0),
|
||
("deno.land", 3000),
|
||
("deno.lands", 0),
|
||
("deno.lands", 3000),
|
||
("github.com", 3000),
|
||
("github.com", 0),
|
||
("github.com", 2000),
|
||
("github.net", 3000),
|
||
("127.0.0.1", 0),
|
||
("127.0.0.1", 3000),
|
||
("127.0.0.2", 0),
|
||
("127.0.0.2", 3000),
|
||
("172.16.0.2", 8000),
|
||
("172.16.0.2", 0),
|
||
("172.16.0.2", 6000),
|
||
("172.16.0.1", 8000),
|
||
("somedomain", 0),
|
||
("192.168.0.1", 0),
|
||
];
|
||
|
||
for (host_str, port) in domain_tests {
|
||
let host = Host::parse(host_str).unwrap();
|
||
let descriptor = NetDescriptor(host, Some(port));
|
||
assert!(
|
||
perms.net.check(&descriptor, None).is_err(),
|
||
"expected {host_str}:{port} to fail"
|
||
);
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn test_check_net_url() {
|
||
let parser = TestPermissionDescriptorParser;
|
||
let perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_net: Some(svec![
|
||
"localhost",
|
||
"deno.land",
|
||
"github.com:3000",
|
||
"127.0.0.1",
|
||
"172.16.0.2:8000",
|
||
"www.github.com:443"
|
||
]),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
let mut perms = PermissionsContainer::new(Arc::new(parser), perms);
|
||
|
||
let url_tests = vec![
|
||
// Any protocol + port for localhost should be ok, since we don't specify
|
||
("http://localhost", true),
|
||
("https://localhost", true),
|
||
("https://localhost:4443", true),
|
||
("tcp://localhost:5000", true),
|
||
("udp://localhost:6000", true),
|
||
// Correct domain + any port and protocol should be ok incorrect shouldn't
|
||
("https://deno.land/std/example/welcome.ts", true),
|
||
("https://deno.land:3000/std/example/welcome.ts", true),
|
||
("https://deno.lands/std/example/welcome.ts", false),
|
||
("https://deno.lands:3000/std/example/welcome.ts", false),
|
||
// Correct domain + port should be ok all other combinations should err
|
||
("https://github.com:3000/denoland/deno", true),
|
||
("https://github.com/denoland/deno", false),
|
||
("https://github.com:2000/denoland/deno", false),
|
||
("https://github.net:3000/denoland/deno", false),
|
||
// Correct ipv4 address + any port should be ok others should err
|
||
("tcp://127.0.0.1", true),
|
||
("https://127.0.0.1", true),
|
||
("tcp://127.0.0.1:3000", true),
|
||
("https://127.0.0.1:3000", true),
|
||
("tcp://127.0.0.2", false),
|
||
("https://127.0.0.2", false),
|
||
("tcp://127.0.0.2:3000", false),
|
||
("https://127.0.0.2:3000", false),
|
||
// Correct address + port should be ok all other combinations should err
|
||
("tcp://172.16.0.2:8000", true),
|
||
("https://172.16.0.2:8000", true),
|
||
("tcp://172.16.0.2", false),
|
||
("https://172.16.0.2", false),
|
||
("tcp://172.16.0.2:6000", false),
|
||
("https://172.16.0.2:6000", false),
|
||
("tcp://172.16.0.1:8000", false),
|
||
("https://172.16.0.1:8000", false),
|
||
// Testing issue #6531 (Network permissions check doesn't account for well-known default ports) so we dont regress
|
||
("https://www.github.com:443/robots.txt", true),
|
||
];
|
||
|
||
for (url_str, is_ok) in url_tests {
|
||
let u = Url::parse(url_str).unwrap();
|
||
assert_eq!(is_ok, perms.check_net_url(&u, "api()").is_ok(), "{}", u);
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn check_specifiers() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let read_allowlist = if cfg!(target_os = "windows") {
|
||
svec!["C:\\a"]
|
||
} else {
|
||
svec!["/a"]
|
||
};
|
||
let parser = TestPermissionDescriptorParser;
|
||
let perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_read: Some(read_allowlist),
|
||
allow_import: Some(svec!["localhost"]),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
let perms = PermissionsContainer::new(Arc::new(parser), perms);
|
||
|
||
let mut fixtures = vec![
|
||
(
|
||
ModuleSpecifier::parse("http://localhost:4545/mod.ts").unwrap(),
|
||
CheckSpecifierKind::Static,
|
||
true,
|
||
),
|
||
(
|
||
ModuleSpecifier::parse("http://localhost:4545/mod.ts").unwrap(),
|
||
CheckSpecifierKind::Dynamic,
|
||
true,
|
||
),
|
||
(
|
||
ModuleSpecifier::parse("http://deno.land/x/mod.ts").unwrap(),
|
||
CheckSpecifierKind::Dynamic,
|
||
false,
|
||
),
|
||
(
|
||
ModuleSpecifier::parse("data:text/plain,Hello%2C%20Deno!").unwrap(),
|
||
CheckSpecifierKind::Dynamic,
|
||
true,
|
||
),
|
||
];
|
||
|
||
if cfg!(target_os = "windows") {
|
||
fixtures.push((
|
||
ModuleSpecifier::parse("file:///C:/a/mod.ts").unwrap(),
|
||
CheckSpecifierKind::Dynamic,
|
||
true,
|
||
));
|
||
fixtures.push((
|
||
ModuleSpecifier::parse("file:///C:/b/mod.ts").unwrap(),
|
||
CheckSpecifierKind::Static,
|
||
true,
|
||
));
|
||
fixtures.push((
|
||
ModuleSpecifier::parse("file:///C:/b/mod.ts").unwrap(),
|
||
CheckSpecifierKind::Dynamic,
|
||
false,
|
||
));
|
||
} else {
|
||
fixtures.push((
|
||
ModuleSpecifier::parse("file:///a/mod.ts").unwrap(),
|
||
CheckSpecifierKind::Dynamic,
|
||
true,
|
||
));
|
||
fixtures.push((
|
||
ModuleSpecifier::parse("file:///b/mod.ts").unwrap(),
|
||
CheckSpecifierKind::Static,
|
||
true,
|
||
));
|
||
fixtures.push((
|
||
ModuleSpecifier::parse("file:///b/mod.ts").unwrap(),
|
||
CheckSpecifierKind::Dynamic,
|
||
false,
|
||
));
|
||
}
|
||
|
||
for (specifier, kind, expected) in fixtures {
|
||
assert_eq!(
|
||
perms.check_specifier(&specifier, kind).is_ok(),
|
||
expected,
|
||
"{}",
|
||
specifier,
|
||
);
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn test_query() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let parser = TestPermissionDescriptorParser;
|
||
let perms1 = Permissions::allow_all();
|
||
let perms2 = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_read: Some(svec!["/foo"]),
|
||
allow_write: Some(svec!["/foo"]),
|
||
allow_ffi: Some(svec!["/foo"]),
|
||
allow_net: Some(svec!["127.0.0.1:8000"]),
|
||
allow_env: Some(svec!["HOME"]),
|
||
allow_sys: Some(svec!["hostname"]),
|
||
allow_run: Some(svec!["/deno"]),
|
||
allow_all: false,
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
let perms3 = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
deny_read: Some(svec!["/foo"]),
|
||
deny_write: Some(svec!["/foo"]),
|
||
deny_ffi: Some(svec!["/foo"]),
|
||
deny_net: Some(svec!["127.0.0.1:8000"]),
|
||
deny_env: Some(svec!["HOME"]),
|
||
deny_sys: Some(svec!["hostname"]),
|
||
deny_run: Some(svec!["deno"]),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
let perms4 = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_read: Some(vec![]),
|
||
deny_read: Some(svec!["/foo"]),
|
||
allow_write: Some(vec![]),
|
||
deny_write: Some(svec!["/foo"]),
|
||
allow_ffi: Some(vec![]),
|
||
deny_ffi: Some(svec!["/foo"]),
|
||
allow_net: Some(vec![]),
|
||
deny_net: Some(svec!["127.0.0.1:8000"]),
|
||
allow_env: Some(vec![]),
|
||
deny_env: Some(svec!["HOME"]),
|
||
allow_sys: Some(vec![]),
|
||
deny_sys: Some(svec!["hostname"]),
|
||
allow_run: Some(vec![]),
|
||
deny_run: Some(svec!["deno"]),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
#[rustfmt::skip]
|
||
{
|
||
let read_query = |path: &str| parser.parse_path_query(path).unwrap().into_read();
|
||
let write_query = |path: &str| parser.parse_path_query(path).unwrap().into_write();
|
||
let ffi_query = |path: &str| parser.parse_path_query(path).unwrap().into_ffi();
|
||
assert_eq!(perms1.read.query(None), PermissionState::Granted);
|
||
assert_eq!(perms1.read.query(Some(&read_query("/foo"))), PermissionState::Granted);
|
||
assert_eq!(perms2.read.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms2.read.query(Some(&read_query("/foo"))), PermissionState::Granted);
|
||
assert_eq!(perms2.read.query(Some(&read_query("/foo/bar"))), PermissionState::Granted);
|
||
assert_eq!(perms3.read.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms3.read.query(Some(&read_query("/foo"))), PermissionState::Denied);
|
||
assert_eq!(perms3.read.query(Some(&read_query("/foo/bar"))), PermissionState::Denied);
|
||
assert_eq!(perms4.read.query(None), PermissionState::GrantedPartial);
|
||
assert_eq!(perms4.read.query(Some(&read_query("/foo"))), PermissionState::Denied);
|
||
assert_eq!(perms4.read.query(Some(&read_query("/foo/bar"))), PermissionState::Denied);
|
||
assert_eq!(perms4.read.query(Some(&read_query("/bar"))), PermissionState::Granted);
|
||
assert_eq!(perms1.write.query(None), PermissionState::Granted);
|
||
assert_eq!(perms1.write.query(Some(&write_query("/foo"))), PermissionState::Granted);
|
||
assert_eq!(perms2.write.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms2.write.query(Some(&write_query("/foo"))), PermissionState::Granted);
|
||
assert_eq!(perms2.write.query(Some(&write_query("/foo/bar"))), PermissionState::Granted);
|
||
assert_eq!(perms3.write.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms3.write.query(Some(&write_query("/foo"))), PermissionState::Denied);
|
||
assert_eq!(perms3.write.query(Some(&write_query("/foo/bar"))), PermissionState::Denied);
|
||
assert_eq!(perms4.write.query(None), PermissionState::GrantedPartial);
|
||
assert_eq!(perms4.write.query(Some(&write_query("/foo"))), PermissionState::Denied);
|
||
assert_eq!(perms4.write.query(Some(&write_query("/foo/bar"))), PermissionState::Denied);
|
||
assert_eq!(perms4.write.query(Some(&write_query("/bar"))), PermissionState::Granted);
|
||
assert_eq!(perms1.ffi.query(None), PermissionState::Granted);
|
||
assert_eq!(perms1.ffi.query(Some(&ffi_query("/foo"))), PermissionState::Granted);
|
||
assert_eq!(perms2.ffi.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms2.ffi.query(Some(&ffi_query("/foo"))), PermissionState::Granted);
|
||
assert_eq!(perms2.ffi.query(Some(&ffi_query("/foo/bar"))), PermissionState::Granted);
|
||
assert_eq!(perms3.ffi.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms3.ffi.query(Some(&ffi_query("/foo"))), PermissionState::Denied);
|
||
assert_eq!(perms3.ffi.query(Some(&ffi_query("/foo/bar"))), PermissionState::Denied);
|
||
assert_eq!(perms4.ffi.query(None), PermissionState::GrantedPartial);
|
||
assert_eq!(perms4.ffi.query(Some(&ffi_query("/foo"))), PermissionState::Denied);
|
||
assert_eq!(perms4.ffi.query(Some(&ffi_query("/foo/bar"))), PermissionState::Denied);
|
||
assert_eq!(perms4.ffi.query(Some(&ffi_query("/bar"))), PermissionState::Granted);
|
||
assert_eq!(perms1.net.query(None), PermissionState::Granted);
|
||
assert_eq!(perms1.net.query(Some(&NetDescriptor(Host::must_parse("127.0.0.1"), None))), PermissionState::Granted);
|
||
assert_eq!(perms2.net.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms2.net.query(Some(&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8000)))), PermissionState::Granted);
|
||
assert_eq!(perms3.net.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms3.net.query(Some(&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8000)))), PermissionState::Denied);
|
||
assert_eq!(perms4.net.query(None), PermissionState::GrantedPartial);
|
||
assert_eq!(perms4.net.query(Some(&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8000)))), PermissionState::Denied);
|
||
assert_eq!(perms4.net.query(Some(&NetDescriptor(Host::must_parse("192.168.0.1"), Some(8000)))), PermissionState::Granted);
|
||
assert_eq!(perms1.env.query(None), PermissionState::Granted);
|
||
assert_eq!(perms1.env.query(Some("HOME")), PermissionState::Granted);
|
||
assert_eq!(perms2.env.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms2.env.query(Some("HOME")), PermissionState::Granted);
|
||
assert_eq!(perms3.env.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms3.env.query(Some("HOME")), PermissionState::Denied);
|
||
assert_eq!(perms4.env.query(None), PermissionState::GrantedPartial);
|
||
assert_eq!(perms4.env.query(Some("HOME")), PermissionState::Denied);
|
||
assert_eq!(perms4.env.query(Some("AWAY")), PermissionState::Granted);
|
||
let sys_desc = |name: &str| SysDescriptor::parse(name.to_string()).unwrap();
|
||
assert_eq!(perms1.sys.query(None), PermissionState::Granted);
|
||
assert_eq!(perms1.sys.query(Some(&sys_desc("osRelease"))), PermissionState::Granted);
|
||
assert_eq!(perms2.sys.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms2.sys.query(Some(&sys_desc("hostname"))), PermissionState::Granted);
|
||
assert_eq!(perms3.sys.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms3.sys.query(Some(&sys_desc("hostname"))), PermissionState::Denied);
|
||
assert_eq!(perms4.sys.query(None), PermissionState::GrantedPartial);
|
||
assert_eq!(perms4.sys.query(Some(&sys_desc("hostname"))), PermissionState::Denied);
|
||
assert_eq!(perms4.sys.query(Some(&sys_desc("uid"))), PermissionState::Granted);
|
||
assert_eq!(perms1.run.query(None), PermissionState::Granted);
|
||
let deno_run_query = RunQueryDescriptor::Path {
|
||
requested: "deno".to_string(),
|
||
resolved: PathBuf::from("/deno"),
|
||
};
|
||
let node_run_query = RunQueryDescriptor::Path {
|
||
requested: "node".to_string(),
|
||
resolved: PathBuf::from("/node"),
|
||
};
|
||
assert_eq!(perms1.run.query(Some(&deno_run_query)), PermissionState::Granted);
|
||
assert_eq!(perms1.write.query(Some(&write_query("/deno"))), PermissionState::Granted);
|
||
assert_eq!(perms2.run.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms2.run.query(Some(&deno_run_query)), PermissionState::Granted);
|
||
assert_eq!(perms2.write.query(Some(&write_query("/deno"))), PermissionState::Denied);
|
||
assert_eq!(perms3.run.query(None), PermissionState::Prompt);
|
||
assert_eq!(perms3.run.query(Some(&deno_run_query)), PermissionState::Denied);
|
||
assert_eq!(perms4.run.query(None), PermissionState::GrantedPartial);
|
||
assert_eq!(perms4.run.query(Some(&deno_run_query)), PermissionState::Denied);
|
||
assert_eq!(perms4.run.query(Some(&node_run_query)), PermissionState::Granted);
|
||
};
|
||
}
|
||
|
||
#[test]
|
||
fn test_request() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let parser = TestPermissionDescriptorParser;
|
||
let mut perms: Permissions = Permissions::none_with_prompt();
|
||
let mut perms_no_prompt: Permissions = Permissions::none_without_prompt();
|
||
let read_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_read();
|
||
let write_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_write();
|
||
let ffi_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_ffi();
|
||
#[rustfmt::skip]
|
||
{
|
||
let prompt_value = PERMISSION_PROMPT_STUB_VALUE_SETTER.lock();
|
||
prompt_value.set(true);
|
||
assert_eq!(perms.read.request(Some(&read_query("/foo"))), PermissionState::Granted);
|
||
assert_eq!(perms.read.query(None), PermissionState::Prompt);
|
||
prompt_value.set(false);
|
||
assert_eq!(perms.read.request(Some(&read_query("/foo/bar"))), PermissionState::Granted);
|
||
prompt_value.set(false);
|
||
assert_eq!(perms.write.request(Some(&write_query("/foo"))), PermissionState::Denied);
|
||
assert_eq!(perms.write.query(Some(&write_query("/foo/bar"))), PermissionState::Prompt);
|
||
prompt_value.set(true);
|
||
assert_eq!(perms.write.request(None), PermissionState::Denied);
|
||
prompt_value.set(false);
|
||
assert_eq!(perms.ffi.request(Some(&ffi_query("/foo"))), PermissionState::Denied);
|
||
assert_eq!(perms.ffi.query(Some(&ffi_query("/foo/bar"))), PermissionState::Prompt);
|
||
prompt_value.set(true);
|
||
assert_eq!(perms.ffi.request(None), PermissionState::Denied);
|
||
prompt_value.set(true);
|
||
assert_eq!(perms.net.request(Some(&NetDescriptor(Host::must_parse("127.0.0.1"), None))), PermissionState::Granted);
|
||
prompt_value.set(false);
|
||
assert_eq!(perms.net.request(Some(&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8000)))), PermissionState::Granted);
|
||
prompt_value.set(true);
|
||
assert_eq!(perms.env.request(Some("HOME")), PermissionState::Granted);
|
||
assert_eq!(perms.env.query(None), PermissionState::Prompt);
|
||
prompt_value.set(false);
|
||
assert_eq!(perms.env.request(Some("HOME")), PermissionState::Granted);
|
||
prompt_value.set(true);
|
||
let sys_desc = |name: &str| SysDescriptor::parse(name.to_string()).unwrap();
|
||
assert_eq!(perms.sys.request(Some(&sys_desc("hostname"))), PermissionState::Granted);
|
||
assert_eq!(perms.sys.query(None), PermissionState::Prompt);
|
||
prompt_value.set(false);
|
||
assert_eq!(perms.sys.request(Some(&sys_desc("hostname"))), PermissionState::Granted);
|
||
prompt_value.set(true);
|
||
let run_query = RunQueryDescriptor::Path {
|
||
requested: "deno".to_string(),
|
||
resolved: PathBuf::from("/deno"),
|
||
};
|
||
assert_eq!(perms.run.request(Some(&run_query)), PermissionState::Granted);
|
||
assert_eq!(perms.run.query(None), PermissionState::Prompt);
|
||
prompt_value.set(false);
|
||
assert_eq!(perms.run.request(Some(&run_query)), PermissionState::Granted);
|
||
assert_eq!(perms_no_prompt.read.request(Some(&read_query("/foo"))), PermissionState::Denied);
|
||
};
|
||
}
|
||
|
||
#[test]
|
||
fn test_revoke() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let parser = TestPermissionDescriptorParser;
|
||
let mut perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_read: Some(svec!["/foo", "/foo/baz"]),
|
||
allow_write: Some(svec!["/foo", "/foo/baz"]),
|
||
allow_ffi: Some(svec!["/foo", "/foo/baz"]),
|
||
allow_net: Some(svec!["127.0.0.1", "127.0.0.1:8000"]),
|
||
allow_env: Some(svec!["HOME"]),
|
||
allow_sys: Some(svec!["hostname"]),
|
||
allow_run: Some(svec!["/deno"]),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
let read_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_read();
|
||
let write_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_write();
|
||
let ffi_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_ffi();
|
||
#[rustfmt::skip]
|
||
{
|
||
assert_eq!(perms.read.revoke(Some(&read_query("/foo/bar"))), PermissionState::Prompt);
|
||
assert_eq!(perms.read.query(Some(&read_query("/foo"))), PermissionState::Prompt);
|
||
assert_eq!(perms.read.query(Some(&read_query("/foo/baz"))), PermissionState::Granted);
|
||
assert_eq!(perms.write.revoke(Some(&write_query("/foo/bar"))), PermissionState::Prompt);
|
||
assert_eq!(perms.write.query(Some(&write_query("/foo"))), PermissionState::Prompt);
|
||
assert_eq!(perms.write.query(Some(&write_query("/foo/baz"))), PermissionState::Granted);
|
||
assert_eq!(perms.ffi.revoke(Some(&ffi_query("/foo/bar"))), PermissionState::Prompt);
|
||
assert_eq!(perms.ffi.query(Some(&ffi_query("/foo"))), PermissionState::Prompt);
|
||
assert_eq!(perms.ffi.query(Some(&ffi_query("/foo/baz"))), PermissionState::Granted);
|
||
assert_eq!(perms.net.revoke(Some(&NetDescriptor(Host::must_parse("127.0.0.1"), Some(9000)))), PermissionState::Prompt);
|
||
assert_eq!(perms.net.query(Some(&NetDescriptor(Host::must_parse("127.0.0.1"), None))), PermissionState::Prompt);
|
||
assert_eq!(perms.net.query(Some(&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8000)))), PermissionState::Granted);
|
||
assert_eq!(perms.env.revoke(Some("HOME")), PermissionState::Prompt);
|
||
assert_eq!(perms.env.revoke(Some("hostname")), PermissionState::Prompt);
|
||
let run_query = RunQueryDescriptor::Path {
|
||
requested: "deno".to_string(),
|
||
resolved: PathBuf::from("/deno"),
|
||
};
|
||
assert_eq!(perms.run.revoke(Some(&run_query)), PermissionState::Prompt);
|
||
};
|
||
}
|
||
|
||
#[test]
|
||
fn test_check() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let mut perms = Permissions::none_with_prompt();
|
||
let prompt_value = PERMISSION_PROMPT_STUB_VALUE_SETTER.lock();
|
||
let parser = TestPermissionDescriptorParser;
|
||
let read_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_read();
|
||
let write_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_write();
|
||
let ffi_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_ffi();
|
||
|
||
prompt_value.set(true);
|
||
assert!(perms.read.check(&read_query("/foo"), None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.read.check(&read_query("/foo"), None).is_ok());
|
||
assert!(perms.read.check(&read_query("/bar"), None).is_err());
|
||
|
||
prompt_value.set(true);
|
||
assert!(perms.write.check(&write_query("/foo"), None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.write.check(&write_query("/foo"), None).is_ok());
|
||
assert!(perms.write.check(&write_query("/bar"), None).is_err());
|
||
|
||
prompt_value.set(true);
|
||
assert!(perms.ffi.check(&ffi_query("/foo"), None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.ffi.check(&ffi_query("/foo"), None).is_ok());
|
||
assert!(perms.ffi.check(&ffi_query("/bar"), None).is_err());
|
||
|
||
prompt_value.set(true);
|
||
assert!(perms
|
||
.net
|
||
.check(
|
||
&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8000)),
|
||
None
|
||
)
|
||
.is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms
|
||
.net
|
||
.check(
|
||
&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8000)),
|
||
None
|
||
)
|
||
.is_ok());
|
||
assert!(perms
|
||
.net
|
||
.check(
|
||
&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8001)),
|
||
None
|
||
)
|
||
.is_err());
|
||
assert!(perms
|
||
.net
|
||
.check(&NetDescriptor(Host::must_parse("127.0.0.1"), None), None)
|
||
.is_err());
|
||
assert!(perms
|
||
.net
|
||
.check(
|
||
&NetDescriptor(Host::must_parse("deno.land"), Some(8000)),
|
||
None
|
||
)
|
||
.is_err());
|
||
assert!(perms
|
||
.net
|
||
.check(&NetDescriptor(Host::must_parse("deno.land"), None), None)
|
||
.is_err());
|
||
|
||
#[allow(clippy::disallowed_methods)]
|
||
let cwd = std::env::current_dir().unwrap();
|
||
prompt_value.set(true);
|
||
assert!(perms
|
||
.run
|
||
.check(
|
||
&RunQueryDescriptor::Path {
|
||
requested: "cat".to_string(),
|
||
resolved: cwd.join("cat")
|
||
},
|
||
None
|
||
)
|
||
.is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms
|
||
.run
|
||
.check(
|
||
&RunQueryDescriptor::Path {
|
||
requested: "cat".to_string(),
|
||
resolved: cwd.join("cat")
|
||
},
|
||
None
|
||
)
|
||
.is_ok());
|
||
assert!(perms
|
||
.run
|
||
.check(
|
||
&RunQueryDescriptor::Path {
|
||
requested: "ls".to_string(),
|
||
resolved: cwd.join("ls")
|
||
},
|
||
None
|
||
)
|
||
.is_err());
|
||
|
||
prompt_value.set(true);
|
||
assert!(perms.env.check("HOME", None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.env.check("HOME", None).is_ok());
|
||
assert!(perms.env.check("PATH", None).is_err());
|
||
|
||
prompt_value.set(true);
|
||
assert!(perms.env.check("hostname", None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.env.check("hostname", None).is_ok());
|
||
assert!(perms.env.check("osRelease", None).is_err());
|
||
}
|
||
|
||
#[test]
|
||
fn test_check_fail() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let mut perms = Permissions::none_with_prompt();
|
||
let prompt_value = PERMISSION_PROMPT_STUB_VALUE_SETTER.lock();
|
||
let parser = TestPermissionDescriptorParser;
|
||
let read_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_read();
|
||
let write_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_write();
|
||
let ffi_query =
|
||
|path: &str| parser.parse_path_query(path).unwrap().into_ffi();
|
||
|
||
prompt_value.set(false);
|
||
assert!(perms.read.check(&read_query("/foo"), None).is_err());
|
||
prompt_value.set(true);
|
||
assert!(perms.read.check(&read_query("/foo"), None).is_err());
|
||
assert!(perms.read.check(&read_query("/bar"), None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.read.check(&read_query("/bar"), None).is_ok());
|
||
|
||
prompt_value.set(false);
|
||
assert!(perms.write.check(&write_query("/foo"), None).is_err());
|
||
prompt_value.set(true);
|
||
assert!(perms.write.check(&write_query("/foo"), None).is_err());
|
||
assert!(perms.write.check(&write_query("/bar"), None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.write.check(&write_query("/bar"), None).is_ok());
|
||
|
||
prompt_value.set(false);
|
||
assert!(perms.ffi.check(&ffi_query("/foo"), None).is_err());
|
||
prompt_value.set(true);
|
||
assert!(perms.ffi.check(&ffi_query("/foo"), None).is_err());
|
||
assert!(perms.ffi.check(&ffi_query("/bar"), None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.ffi.check(&ffi_query("/bar"), None).is_ok());
|
||
|
||
prompt_value.set(false);
|
||
assert!(perms
|
||
.net
|
||
.check(
|
||
&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8000)),
|
||
None
|
||
)
|
||
.is_err());
|
||
prompt_value.set(true);
|
||
assert!(perms
|
||
.net
|
||
.check(
|
||
&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8000)),
|
||
None
|
||
)
|
||
.is_err());
|
||
assert!(perms
|
||
.net
|
||
.check(
|
||
&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8001)),
|
||
None
|
||
)
|
||
.is_ok());
|
||
assert!(perms
|
||
.net
|
||
.check(
|
||
&NetDescriptor(Host::must_parse("deno.land"), Some(8000)),
|
||
None
|
||
)
|
||
.is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms
|
||
.net
|
||
.check(
|
||
&NetDescriptor(Host::must_parse("127.0.0.1"), Some(8001)),
|
||
None
|
||
)
|
||
.is_ok());
|
||
assert!(perms
|
||
.net
|
||
.check(
|
||
&NetDescriptor(Host::must_parse("deno.land"), Some(8000)),
|
||
None
|
||
)
|
||
.is_ok());
|
||
|
||
prompt_value.set(false);
|
||
#[allow(clippy::disallowed_methods)]
|
||
let cwd = std::env::current_dir().unwrap();
|
||
assert!(perms
|
||
.run
|
||
.check(
|
||
&RunQueryDescriptor::Path {
|
||
requested: "cat".to_string(),
|
||
resolved: cwd.join("cat")
|
||
},
|
||
None
|
||
)
|
||
.is_err());
|
||
prompt_value.set(true);
|
||
assert!(perms
|
||
.run
|
||
.check(
|
||
&RunQueryDescriptor::Path {
|
||
requested: "cat".to_string(),
|
||
resolved: cwd.join("cat")
|
||
},
|
||
None
|
||
)
|
||
.is_err());
|
||
assert!(perms
|
||
.run
|
||
.check(
|
||
&RunQueryDescriptor::Path {
|
||
requested: "ls".to_string(),
|
||
resolved: cwd.join("ls")
|
||
},
|
||
None
|
||
)
|
||
.is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms
|
||
.run
|
||
.check(
|
||
&RunQueryDescriptor::Path {
|
||
requested: "ls".to_string(),
|
||
resolved: cwd.join("ls")
|
||
},
|
||
None
|
||
)
|
||
.is_ok());
|
||
|
||
prompt_value.set(false);
|
||
assert!(perms.env.check("HOME", None).is_err());
|
||
prompt_value.set(true);
|
||
assert!(perms.env.check("HOME", None).is_err());
|
||
assert!(perms.env.check("PATH", None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.env.check("PATH", None).is_ok());
|
||
|
||
prompt_value.set(false);
|
||
let sys_desc = |name: &str| SysDescriptor::parse(name.to_string()).unwrap();
|
||
assert!(perms.sys.check(&sys_desc("hostname"), None).is_err());
|
||
prompt_value.set(true);
|
||
assert!(perms.sys.check(&sys_desc("hostname"), None).is_err());
|
||
assert!(perms.sys.check(&sys_desc("osRelease"), None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.sys.check(&sys_desc("osRelease"), None).is_ok());
|
||
}
|
||
|
||
#[test]
|
||
#[cfg(windows)]
|
||
fn test_env_windows() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let prompt_value = PERMISSION_PROMPT_STUB_VALUE_SETTER.lock();
|
||
let mut perms = Permissions::allow_all();
|
||
perms.env = UnaryPermission {
|
||
granted_global: false,
|
||
..Permissions::new_unary(
|
||
Some(HashSet::from([EnvDescriptor::new("HOME")])),
|
||
None,
|
||
false,
|
||
)
|
||
};
|
||
|
||
prompt_value.set(true);
|
||
assert!(perms.env.check("HOME", None).is_ok());
|
||
prompt_value.set(false);
|
||
assert!(perms.env.check("HOME", None).is_ok());
|
||
assert!(perms.env.check("hOmE", None).is_ok());
|
||
|
||
assert_eq!(perms.env.revoke(Some("HomE")), PermissionState::Prompt);
|
||
}
|
||
|
||
#[test]
|
||
fn test_check_partial_denied() {
|
||
let parser = TestPermissionDescriptorParser;
|
||
let mut perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_read: Some(vec![]),
|
||
deny_read: Some(svec!["/foo/bar"]),
|
||
allow_write: Some(vec![]),
|
||
deny_write: Some(svec!["/foo/bar"]),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
|
||
let read_query = parser.parse_path_query("/foo").unwrap().into_read();
|
||
perms.read.check_partial(&read_query, None).unwrap();
|
||
assert!(perms.read.check(&read_query, None).is_err());
|
||
|
||
let write_query = parser.parse_path_query("/foo").unwrap().into_write();
|
||
perms.write.check_partial(&write_query, None).unwrap();
|
||
assert!(perms.write.check(&write_query, None).is_err());
|
||
}
|
||
|
||
#[test]
|
||
fn test_net_fully_qualified_domain_name() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let parser = TestPermissionDescriptorParser;
|
||
let perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_net: Some(svec!["allowed.domain", "1.1.1.1"]),
|
||
deny_net: Some(svec!["denied.domain", "2.2.2.2"]),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
let mut perms = PermissionsContainer::new(Arc::new(parser), perms);
|
||
let cases = [
|
||
("allowed.domain.", true),
|
||
("1.1.1.1", true),
|
||
("denied.domain.", false),
|
||
("2.2.2.2", false),
|
||
];
|
||
|
||
for (host, is_ok) in cases {
|
||
assert_eq!(perms.check_net(&(host, None), "api").is_ok(), is_ok);
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn test_deserialize_child_permissions_arg() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
assert_eq!(
|
||
ChildPermissionsArg::inherit(),
|
||
ChildPermissionsArg {
|
||
env: ChildUnaryPermissionArg::Inherit,
|
||
net: ChildUnaryPermissionArg::Inherit,
|
||
ffi: ChildUnaryPermissionArg::Inherit,
|
||
import: ChildUnaryPermissionArg::Inherit,
|
||
read: ChildUnaryPermissionArg::Inherit,
|
||
run: ChildUnaryPermissionArg::Inherit,
|
||
sys: ChildUnaryPermissionArg::Inherit,
|
||
write: ChildUnaryPermissionArg::Inherit,
|
||
}
|
||
);
|
||
assert_eq!(
|
||
ChildPermissionsArg::none(),
|
||
ChildPermissionsArg {
|
||
env: ChildUnaryPermissionArg::NotGranted,
|
||
net: ChildUnaryPermissionArg::NotGranted,
|
||
ffi: ChildUnaryPermissionArg::NotGranted,
|
||
import: ChildUnaryPermissionArg::NotGranted,
|
||
read: ChildUnaryPermissionArg::NotGranted,
|
||
run: ChildUnaryPermissionArg::NotGranted,
|
||
sys: ChildUnaryPermissionArg::NotGranted,
|
||
write: ChildUnaryPermissionArg::NotGranted,
|
||
}
|
||
);
|
||
assert_eq!(
|
||
serde_json::from_value::<ChildPermissionsArg>(json!("inherit")).unwrap(),
|
||
ChildPermissionsArg::inherit()
|
||
);
|
||
assert_eq!(
|
||
serde_json::from_value::<ChildPermissionsArg>(json!("none")).unwrap(),
|
||
ChildPermissionsArg::none()
|
||
);
|
||
assert_eq!(
|
||
serde_json::from_value::<ChildPermissionsArg>(json!({})).unwrap(),
|
||
ChildPermissionsArg::none()
|
||
);
|
||
assert_eq!(
|
||
serde_json::from_value::<ChildPermissionsArg>(json!({
|
||
"env": ["foo", "bar"],
|
||
}))
|
||
.unwrap(),
|
||
ChildPermissionsArg {
|
||
env: ChildUnaryPermissionArg::GrantedList(svec!["foo", "bar"]),
|
||
..ChildPermissionsArg::none()
|
||
}
|
||
);
|
||
assert_eq!(
|
||
serde_json::from_value::<ChildPermissionsArg>(json!({
|
||
"env": true,
|
||
"net": true,
|
||
"ffi": true,
|
||
"import": true,
|
||
"read": true,
|
||
"run": true,
|
||
"sys": true,
|
||
"write": true,
|
||
}))
|
||
.unwrap(),
|
||
ChildPermissionsArg {
|
||
env: ChildUnaryPermissionArg::Granted,
|
||
net: ChildUnaryPermissionArg::Granted,
|
||
ffi: ChildUnaryPermissionArg::Granted,
|
||
import: ChildUnaryPermissionArg::Granted,
|
||
read: ChildUnaryPermissionArg::Granted,
|
||
run: ChildUnaryPermissionArg::Granted,
|
||
sys: ChildUnaryPermissionArg::Granted,
|
||
write: ChildUnaryPermissionArg::Granted,
|
||
}
|
||
);
|
||
assert_eq!(
|
||
serde_json::from_value::<ChildPermissionsArg>(json!({
|
||
"env": false,
|
||
"net": false,
|
||
"ffi": false,
|
||
"import": false,
|
||
"read": false,
|
||
"run": false,
|
||
"sys": false,
|
||
"write": false,
|
||
}))
|
||
.unwrap(),
|
||
ChildPermissionsArg {
|
||
env: ChildUnaryPermissionArg::NotGranted,
|
||
net: ChildUnaryPermissionArg::NotGranted,
|
||
ffi: ChildUnaryPermissionArg::NotGranted,
|
||
import: ChildUnaryPermissionArg::NotGranted,
|
||
read: ChildUnaryPermissionArg::NotGranted,
|
||
run: ChildUnaryPermissionArg::NotGranted,
|
||
sys: ChildUnaryPermissionArg::NotGranted,
|
||
write: ChildUnaryPermissionArg::NotGranted,
|
||
}
|
||
);
|
||
assert_eq!(
|
||
serde_json::from_value::<ChildPermissionsArg>(json!({
|
||
"env": ["foo", "bar"],
|
||
"net": ["foo", "bar:8000"],
|
||
"ffi": ["foo", "file:///bar/baz"],
|
||
"import": ["example.com"],
|
||
"read": ["foo", "file:///bar/baz"],
|
||
"run": ["foo", "file:///bar/baz", "./qux"],
|
||
"sys": ["hostname", "osRelease"],
|
||
"write": ["foo", "file:///bar/baz"],
|
||
}))
|
||
.unwrap(),
|
||
ChildPermissionsArg {
|
||
env: ChildUnaryPermissionArg::GrantedList(svec!["foo", "bar"]),
|
||
net: ChildUnaryPermissionArg::GrantedList(svec!["foo", "bar:8000"]),
|
||
ffi: ChildUnaryPermissionArg::GrantedList(svec![
|
||
"foo",
|
||
"file:///bar/baz"
|
||
]),
|
||
import: ChildUnaryPermissionArg::GrantedList(svec!["example.com"]),
|
||
read: ChildUnaryPermissionArg::GrantedList(svec![
|
||
"foo",
|
||
"file:///bar/baz"
|
||
]),
|
||
run: ChildUnaryPermissionArg::GrantedList(svec![
|
||
"foo",
|
||
"file:///bar/baz",
|
||
"./qux"
|
||
]),
|
||
sys: ChildUnaryPermissionArg::GrantedList(svec![
|
||
"hostname",
|
||
"osRelease"
|
||
]),
|
||
write: ChildUnaryPermissionArg::GrantedList(svec![
|
||
"foo",
|
||
"file:///bar/baz"
|
||
]),
|
||
}
|
||
);
|
||
}
|
||
|
||
#[test]
|
||
fn test_create_child_permissions() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let parser = TestPermissionDescriptorParser;
|
||
let main_perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
allow_env: Some(vec![]),
|
||
allow_net: Some(svec!["foo", "bar"]),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
let main_perms = PermissionsContainer::new(Arc::new(parser), main_perms);
|
||
assert_eq!(
|
||
main_perms
|
||
.create_child_permissions(ChildPermissionsArg {
|
||
env: ChildUnaryPermissionArg::Inherit,
|
||
net: ChildUnaryPermissionArg::GrantedList(svec!["foo"]),
|
||
ffi: ChildUnaryPermissionArg::NotGranted,
|
||
..ChildPermissionsArg::none()
|
||
})
|
||
.unwrap()
|
||
.inner
|
||
.lock()
|
||
.clone(),
|
||
Permissions {
|
||
env: Permissions::new_unary(Some(HashSet::new()), None, false),
|
||
net: Permissions::new_unary(
|
||
Some(HashSet::from([NetDescriptor::parse("foo").unwrap()])),
|
||
None,
|
||
false
|
||
),
|
||
..Permissions::none_without_prompt()
|
||
}
|
||
);
|
||
assert!(main_perms
|
||
.create_child_permissions(ChildPermissionsArg {
|
||
net: ChildUnaryPermissionArg::Granted,
|
||
..ChildPermissionsArg::none()
|
||
})
|
||
.is_err());
|
||
assert!(main_perms
|
||
.create_child_permissions(ChildPermissionsArg {
|
||
net: ChildUnaryPermissionArg::GrantedList(svec!["foo", "bar", "baz"]),
|
||
..ChildPermissionsArg::none()
|
||
})
|
||
.is_err());
|
||
assert!(main_perms
|
||
.create_child_permissions(ChildPermissionsArg {
|
||
ffi: ChildUnaryPermissionArg::GrantedList(svec!["foo"]),
|
||
..ChildPermissionsArg::none()
|
||
})
|
||
.is_err());
|
||
}
|
||
|
||
#[test]
|
||
fn test_create_child_permissions_with_prompt() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let prompt_value = PERMISSION_PROMPT_STUB_VALUE_SETTER.lock();
|
||
let main_perms = Permissions::from_options(
|
||
&TestPermissionDescriptorParser,
|
||
&PermissionsOptions {
|
||
prompt: true,
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
let main_perms = PermissionsContainer::new(
|
||
Arc::new(TestPermissionDescriptorParser),
|
||
main_perms,
|
||
);
|
||
prompt_value.set(true);
|
||
let worker_perms = main_perms
|
||
.create_child_permissions(ChildPermissionsArg {
|
||
read: ChildUnaryPermissionArg::Granted,
|
||
run: ChildUnaryPermissionArg::GrantedList(svec!["foo", "bar"]),
|
||
..ChildPermissionsArg::none()
|
||
})
|
||
.unwrap();
|
||
assert_eq!(
|
||
main_perms.inner.lock().clone(),
|
||
worker_perms.inner.lock().clone()
|
||
);
|
||
assert_eq!(
|
||
main_perms.inner.lock().run.granted_list,
|
||
HashSet::from([
|
||
AllowRunDescriptor(PathBuf::from("/bar")),
|
||
AllowRunDescriptor(PathBuf::from("/foo")),
|
||
])
|
||
);
|
||
}
|
||
|
||
#[test]
|
||
fn test_create_child_permissions_with_inherited_denied_list() {
|
||
set_prompter(Box::new(TestPrompter));
|
||
let prompt_value = PERMISSION_PROMPT_STUB_VALUE_SETTER.lock();
|
||
let parser = TestPermissionDescriptorParser;
|
||
let main_perms = Permissions::from_options(
|
||
&parser,
|
||
&PermissionsOptions {
|
||
prompt: true,
|
||
..Default::default()
|
||
},
|
||
)
|
||
.unwrap();
|
||
let main_perms =
|
||
PermissionsContainer::new(Arc::new(parser.clone()), main_perms);
|
||
prompt_value.set(false);
|
||
assert!(main_perms
|
||
.inner
|
||
.lock()
|
||
.write
|
||
.check(&parser.parse_path_query("foo").unwrap().into_write(), None)
|
||
.is_err());
|
||
let worker_perms = main_perms
|
||
.create_child_permissions(ChildPermissionsArg::none())
|
||
.unwrap();
|
||
assert_eq!(
|
||
worker_perms.inner.lock().write.flag_denied_list.clone(),
|
||
main_perms.inner.lock().write.flag_denied_list
|
||
);
|
||
}
|
||
|
||
#[test]
|
||
fn test_host_parse() {
|
||
let hosts = &[
|
||
("deno.land", Some(Host::Fqdn(fqdn!("deno.land")))),
|
||
("DENO.land", Some(Host::Fqdn(fqdn!("deno.land")))),
|
||
("deno.land.", Some(Host::Fqdn(fqdn!("deno.land")))),
|
||
(
|
||
"1.1.1.1",
|
||
Some(Host::Ip(IpAddr::V4(Ipv4Addr::new(1, 1, 1, 1)))),
|
||
),
|
||
(
|
||
"::1",
|
||
Some(Host::Ip(IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1)))),
|
||
),
|
||
(
|
||
"[::1]",
|
||
Some(Host::Ip(IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1)))),
|
||
),
|
||
("[::1", None),
|
||
("::1]", None),
|
||
("deno. land", None),
|
||
("1. 1.1.1", None),
|
||
("1.1.1.1.", None),
|
||
("1::1.", None),
|
||
("deno.land.", Some(Host::Fqdn(fqdn!("deno.land")))),
|
||
(".deno.land", None),
|
||
(
|
||
"::ffff:1.1.1.1",
|
||
Some(Host::Ip(IpAddr::V6(Ipv6Addr::new(
|
||
0, 0, 0, 0, 0, 0xffff, 0x0101, 0x0101,
|
||
)))),
|
||
),
|
||
];
|
||
|
||
for (host_str, expected) in hosts {
|
||
assert_eq!(Host::parse(host_str).ok(), *expected, "{host_str}");
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn test_net_descriptor_parse() {
|
||
let cases = &[
|
||
(
|
||
"deno.land",
|
||
Some(NetDescriptor(Host::Fqdn(fqdn!("deno.land")), None)),
|
||
),
|
||
(
|
||
"DENO.land",
|
||
Some(NetDescriptor(Host::Fqdn(fqdn!("deno.land")), None)),
|
||
),
|
||
(
|
||
"deno.land:8000",
|
||
Some(NetDescriptor(Host::Fqdn(fqdn!("deno.land")), Some(8000))),
|
||
),
|
||
("deno.land:", None),
|
||
("deno.land:a", None),
|
||
("deno. land:a", None),
|
||
("deno.land.: a", None),
|
||
(
|
||
"1.1.1.1",
|
||
Some(NetDescriptor(
|
||
Host::Ip(IpAddr::V4(Ipv4Addr::new(1, 1, 1, 1))),
|
||
None,
|
||
)),
|
||
),
|
||
("1.1.1.1.", None),
|
||
("1.1.1.1..", None),
|
||
(
|
||
"1.1.1.1:8000",
|
||
Some(NetDescriptor(
|
||
Host::Ip(IpAddr::V4(Ipv4Addr::new(1, 1, 1, 1))),
|
||
Some(8000),
|
||
)),
|
||
),
|
||
("::", None),
|
||
(":::80", None),
|
||
("::80", None),
|
||
(
|
||
"[::]",
|
||
Some(NetDescriptor(
|
||
Host::Ip(IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 0))),
|
||
None,
|
||
)),
|
||
),
|
||
("[::1", None),
|
||
("::1]", None),
|
||
("::1]", None),
|
||
("[::1]:", None),
|
||
("[::1]:a", None),
|
||
(
|
||
"[::1]:443",
|
||
Some(NetDescriptor(
|
||
Host::Ip(IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1))),
|
||
Some(443),
|
||
)),
|
||
),
|
||
("", None),
|
||
("deno.land..", None),
|
||
];
|
||
|
||
for (input, expected) in cases {
|
||
assert_eq!(NetDescriptor::parse(input).ok(), *expected, "'{input}'");
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn test_denies_run_name() {
|
||
let cases = [
|
||
#[cfg(windows)]
|
||
("deno", "C:\\deno.exe", true),
|
||
#[cfg(windows)]
|
||
("deno", "C:\\sub\\deno.cmd", true),
|
||
#[cfg(windows)]
|
||
("deno", "C:\\sub\\DeNO.cmd", true),
|
||
#[cfg(windows)]
|
||
("DEno", "C:\\sub\\deno.cmd", true),
|
||
#[cfg(windows)]
|
||
("deno", "C:\\other\\sub\\deno.batch", true),
|
||
#[cfg(windows)]
|
||
("deno", "C:\\other\\sub\\deno", true),
|
||
#[cfg(windows)]
|
||
("denort", "C:\\other\\sub\\deno.exe", false),
|
||
("deno", "/home/test/deno", true),
|
||
("deno", "/home/test/denot", false),
|
||
];
|
||
for (name, cmd_path, denies) in cases {
|
||
assert_eq!(
|
||
denies_run_name(name, &PathBuf::from(cmd_path)),
|
||
denies,
|
||
"{} {}",
|
||
name,
|
||
cmd_path
|
||
);
|
||
}
|
||
}
|
||
}
|