1
0
Fork 0
forked from wry/wry

config: add exe client criteria

This commit is contained in:
Julian Orth 2025-05-03 13:09:13 +02:00
parent cc734a135c
commit a6257910bb
13 changed files with 85 additions and 2 deletions

View file

@ -93,4 +93,5 @@ pub enum ClientCriterionStringField {
SandboxAppId, SandboxAppId,
SandboxInstanceId, SandboxInstanceId,
Comm, Comm,
Exe,
} }

View file

@ -1547,6 +1547,8 @@ impl ConfigClient {
ClientCriterion::IsXwayland => ClientCriterionIpc::IsXwayland, ClientCriterion::IsXwayland => ClientCriterionIpc::IsXwayland,
ClientCriterion::Comm(t) => string!(t, Comm, false), ClientCriterion::Comm(t) => string!(t, Comm, false),
ClientCriterion::CommRegex(t) => string!(t, Comm, true), ClientCriterion::CommRegex(t) => string!(t, Comm, true),
ClientCriterion::Exe(t) => string!(t, Exe, false),
ClientCriterion::ExeRegex(t) => string!(t, Exe, true),
}; };
let res = self.send_with_response(&ClientMessage::CreateClientMatcher { criterion }); let res = self.send_with_response(&ClientMessage::CreateClientMatcher { criterion });
get_response!( get_response!(

View file

@ -87,6 +87,10 @@ pub enum ClientCriterion<'a> {
Comm(&'a str), Comm(&'a str),
/// Matches the `/proc/pid/comm` of the client with a regular expression. /// Matches the `/proc/pid/comm` of the client with a regular expression.
CommRegex(&'a str), CommRegex(&'a str),
/// Matches the `/proc/pid/exe` of the client verbatim.
Exe(&'a str),
/// Matches the `/proc/pid/exe` of the client with a regular expression.
ExeRegex(&'a str),
} }
impl ClientCriterion<'_> { impl ClientCriterion<'_> {

View file

@ -1883,6 +1883,7 @@ impl ConfigProxyHandler {
mgr.sandbox_instance_id(needle) mgr.sandbox_instance_id(needle)
} }
ClientCriterionStringField::Comm => mgr.comm(needle), ClientCriterionStringField::Comm => mgr.comm(needle),
ClientCriterionStringField::Exe => mgr.exe(needle),
} }
} }
ClientCriterionIpc::Sandboxed => mgr.sandboxed(), ClientCriterionIpc::Sandboxed => mgr.sandboxed(),

View file

@ -11,7 +11,7 @@ use {
clmm_pid::ClmMatchPid, clmm_pid::ClmMatchPid,
clmm_sandboxed::ClmMatchSandboxed, clmm_sandboxed::ClmMatchSandboxed,
clmm_string::{ clmm_string::{
ClmMatchComm, ClmMatchSandboxAppId, ClmMatchSandboxEngine, ClmMatchComm, ClmMatchExe, ClmMatchSandboxAppId, ClmMatchSandboxEngine,
ClmMatchSandboxInstanceId, ClmMatchSandboxInstanceId,
}, },
clmm_uid::ClmMatchUid, clmm_uid::ClmMatchUid,
@ -60,6 +60,7 @@ pub struct RootMatchers {
uid: ClmRootMatcherMap<ClmMatchUid>, uid: ClmRootMatcherMap<ClmMatchUid>,
pid: ClmRootMatcherMap<ClmMatchPid>, pid: ClmRootMatcherMap<ClmMatchPid>,
comm: ClmRootMatcherMap<ClmMatchComm>, comm: ClmRootMatcherMap<ClmMatchComm>,
exe: ClmRootMatcherMap<ClmMatchExe>,
} }
pub async fn handle_cl_changes(state: Rc<State>) { pub async fn handle_cl_changes(state: Rc<State>) {
@ -163,6 +164,7 @@ impl ClMatcherManager {
unconditional!(uid); unconditional!(uid);
unconditional!(pid); unconditional!(pid);
unconditional!(comm); unconditional!(comm);
unconditional!(exe);
fixed!(sandboxed); fixed!(sandboxed);
fixed!(is_xwayland); fixed!(is_xwayland);
self.constant[true].handle(data); self.constant[true].handle(data);
@ -200,6 +202,10 @@ impl ClMatcherManager {
pub fn comm(&self, string: CritLiteralOrRegex) -> Rc<ClmUpstreamNode> { pub fn comm(&self, string: CritLiteralOrRegex) -> Rc<ClmUpstreamNode> {
self.root(ClmMatchComm::new(string)) self.root(ClmMatchComm::new(string))
} }
pub fn exe(&self, string: CritLiteralOrRegex) -> Rc<ClmUpstreamNode> {
self.root(ClmMatchExe::new(string))
}
} }
impl CritTarget for Rc<Client> { impl CritTarget for Rc<Client> {

View file

@ -16,9 +16,11 @@ pub type ClmMatchSandboxEngine = ClmMatchString<AcceptorMetadataAccess<SandboxEn
pub type ClmMatchSandboxAppId = ClmMatchString<AcceptorMetadataAccess<SandboxAppIdField>>; pub type ClmMatchSandboxAppId = ClmMatchString<AcceptorMetadataAccess<SandboxAppIdField>>;
pub type ClmMatchSandboxInstanceId = ClmMatchString<AcceptorMetadataAccess<SandboxInstanceIdField>>; pub type ClmMatchSandboxInstanceId = ClmMatchString<AcceptorMetadataAccess<SandboxInstanceIdField>>;
pub type ClmMatchComm = ClmMatchString<CommAccess>; pub type ClmMatchComm = ClmMatchString<CommAccess>;
pub type ClmMatchExe = ClmMatchString<ExeAccess>;
pub struct AcceptorMetadataAccess<T>(PhantomData<T>); pub struct AcceptorMetadataAccess<T>(PhantomData<T>);
pub struct CommAccess; pub struct CommAccess;
pub struct ExeAccess;
trait SandboxField: Sized + 'static { trait SandboxField: Sized + 'static {
fn field(meta: &AcceptorMetadata) -> &Option<String>; fn field(meta: &AcceptorMetadata) -> &Option<String>;
@ -89,3 +91,13 @@ impl StringAccess<Rc<Client>> for CommAccess {
&roots.comm &roots.comm
} }
} }
impl StringAccess<Rc<Client>> for ExeAccess {
fn with_string(data: &Rc<Client>, f: impl FnOnce(&str) -> bool) -> bool {
f(&data.pid_info.exe)
}
fn nodes(roots: &RootMatchers) -> &ClmRootMatcherMap<ClmMatchString<Self>> {
&roots.exe
}
}

View file

@ -1,6 +1,7 @@
use { use {
crate::utils::{errorfmt::ErrorFmt, oserror::OsError}, crate::utils::{errorfmt::ErrorFmt, oserror::OsError},
bstr::ByteSlice, bstr::ByteSlice,
std::os::unix::ffi::OsStrExt,
uapi::{OwnedFd, c}, uapi::{OwnedFd, c},
}; };
@ -8,6 +9,7 @@ pub struct PidInfo {
pub uid: c::uid_t, pub uid: c::uid_t,
pub pid: c::pid_t, pub pid: c::pid_t,
pub comm: String, pub comm: String,
pub exe: String,
} }
pub fn get_pid_info(uid: c::uid_t, pid: c::pid_t) -> PidInfo { pub fn get_pid_info(uid: c::uid_t, pid: c::pid_t) -> PidInfo {
@ -18,7 +20,24 @@ pub fn get_pid_info(uid: c::uid_t, pid: c::pid_t) -> PidInfo {
"Unknown".to_string() "Unknown".to_string()
} }
}; };
PidInfo { uid, pid, comm } let exe = match std::fs::read_link(format!("/proc/{}/exe", pid)) {
Ok(name) => name
.as_os_str()
.as_bytes()
.trim_ascii_end()
.as_bstr()
.to_string(),
Err(e) => {
log::warn!("Could not read `exe` of pid {}: {}", pid, ErrorFmt(e));
"Unknown".to_string()
}
};
PidInfo {
uid,
pid,
comm,
exe,
}
} }
pub fn get_socket_creds(socket: &OwnedFd) -> Option<(c::uid_t, c::pid_t)> { pub fn get_socket_creds(socket: &OwnedFd) -> Option<(c::uid_t, c::pid_t)> {

View file

@ -237,6 +237,8 @@ pub struct ClientMatch {
pub is_xwayland: Option<bool>, pub is_xwayland: Option<bool>,
pub comm: Option<String>, pub comm: Option<String>,
pub comm_regex: Option<String>, pub comm_regex: Option<String>,
pub exe: Option<String>,
pub exe_regex: Option<String>,
} }
#[derive(Debug, Clone)] #[derive(Debug, Clone)]

View file

@ -57,6 +57,8 @@ impl Parser for ClientMatchParser<'_> {
is_xwayland, is_xwayland,
comm, comm,
comm_regex, comm_regex,
exe,
exe_regex,
), ),
) = ext.extract(( ) = ext.extract((
( (
@ -79,6 +81,8 @@ impl Parser for ClientMatchParser<'_> {
opt(bol("is-xwayland")), opt(bol("is-xwayland")),
opt(str("comm")), opt(str("comm")),
opt(str("comm-regex")), opt(str("comm-regex")),
opt(str("exe")),
opt(str("exe-regex")),
), ),
))?; ))?;
let mut not = None; let mut not = None;
@ -124,6 +128,8 @@ impl Parser for ClientMatchParser<'_> {
is_xwayland: is_xwayland.despan(), is_xwayland: is_xwayland.despan(),
comm: comm.despan_into(), comm: comm.despan_into(),
comm_regex: comm_regex.despan_into(), comm_regex: comm_regex.despan_into(),
exe: exe.despan_into(),
exe_regex: exe_regex.despan_into(),
}) })
} }
} }

View file

@ -122,6 +122,8 @@ impl Rule for ClientRule {
value_ref!(SandboxInstanceIdRegex, sandbox_instance_id_regex); value_ref!(SandboxInstanceIdRegex, sandbox_instance_id_regex);
value_ref!(Comm, comm); value_ref!(Comm, comm);
value_ref!(CommRegex, comm_regex); value_ref!(CommRegex, comm_regex);
value_ref!(Exe, exe);
value_ref!(ExeRegex, exe_regex);
value!(Uid, uid); value!(Uid, uid);
value!(Pid, pid); value!(Pid, pid);
bool!(Sandboxed, sandboxed); bool!(Sandboxed, sandboxed);

View file

@ -579,6 +579,14 @@
"comm-regex": { "comm-regex": {
"type": "string", "type": "string",
"description": "Matches the `/proc/pid/comm` of the client with a regular expression." "description": "Matches the `/proc/pid/comm` of the client with a regular expression."
},
"exe": {
"type": "string",
"description": "Matches the `/proc/pid/exe` of the client verbatim."
},
"exe-regex": {
"type": "string",
"description": "Matches the `/proc/pid/exe` of the client with a regular expression."
} }
}, },
"required": [] "required": []

View file

@ -905,6 +905,18 @@ The table has the following fields:
The value of this field should be a string. The value of this field should be a string.
- `exe` (optional):
Matches the `/proc/pid/exe` of the client verbatim.
The value of this field should be a string.
- `exe-regex` (optional):
Matches the `/proc/pid/exe` of the client with a regular expression.
The value of this field should be a string.
<a name="types-ClientMatchExactly"></a> <a name="types-ClientMatchExactly"></a>
### `ClientMatchExactly` ### `ClientMatchExactly`

View file

@ -3259,6 +3259,14 @@ ClientMatch:
kind: string kind: string
required: false required: false
description: Matches the `/proc/pid/comm` of the client with a regular expression. description: Matches the `/proc/pid/comm` of the client with a regular expression.
exe:
kind: string
required: false
description: Matches the `/proc/pid/exe` of the client verbatim.
exe-regex:
kind: string
required: false
description: Matches the `/proc/pid/exe` of the client with a regular expression.
ClientMatchExactly: ClientMatchExactly: