pgrep/pkill/pidwait: Support --cgroup option

Author: dezgeg

src/uu/pgrep/src/process_matcher.rs

@@ -44,6 +44,7 @@ pub struct Settings {
     pub gid: Option<HashSet<u32>>,
     pub pgroup: Option<HashSet<u64>>,
     pub session: Option<HashSet<u64>>,
+    pub cgroup: Option<HashSet<String>>,
     pub threads: bool,
 }
 
@@ -101,6 +102,9 @@ pub fn get_match_settings(matches: &ArgMatches) -> UResult<Settings> {
             })
             .collect()
         }),
+        cgroup: matches
+            .get_many::<String>("cgroup")
+            .map(|groups| groups.cloned().collect()),
         threads: false,
     };
 
@@ -115,6 +119,7 @@ pub fn get_match_settings(matches: &ArgMatches) -> UResult<Settings> {
         && settings.gid.is_none()
         && settings.pgroup.is_none()
         && settings.session.is_none()
+        && settings.cgroup.is_none()
         && !settings.require_handler
         && pattern.is_empty()
     {
@@ -235,6 +240,10 @@ fn collect_matched_pids(settings: &Settings) -> Vec<ProcessInformation> {
             let parent_matched = any_matches(&settings.parent, pid.ppid().unwrap());
             let pgroup_matched = any_matches(&settings.pgroup, pid.pgid().unwrap());
             let session_matched = any_matches(&settings.session, pid.sid().unwrap());
+            let cgroup_matched = any_matches(
+                &settings.cgroup,
+                pid.cgroup_v2_path().unwrap_or("/".to_string()),
+            );
 
             let ids_matched = any_matches(&settings.uid, pid.uid().unwrap())
                 && any_matches(&settings.euid, pid.euid().unwrap())
@@ -265,6 +274,7 @@ fn collect_matched_pids(settings: &Settings) -> Vec<ProcessInformation> {
                 && parent_matched
                 && pgroup_matched
                 && session_matched
+                && cgroup_matched
                 && ids_matched
                 && handler_matched)
                 ^ settings.inverse
@@ -413,8 +423,7 @@ pub fn clap_args(pattern_help: &'static str, enable_v_flag: bool) -> Vec<Arg> {
         // arg!(-L --logpidfile           "fail if PID file is not locked"),
         arg!(-r --runstates <state>    "match runstates [D,S,Z,...]"),
         // arg!(-A --"ignore-ancestors"   "exclude our ancestors from results"),
-        // arg!(--cgroup <grp>            "match by cgroup v2 names")
-        //     .value_delimiter(','),
+        arg!(--cgroup <grp>            "match by cgroup v2 names").value_delimiter(','),
         // arg!(--ns <PID>                "match the processes that belong to the same namespace as <pid>"),
         // arg!(--nslist <ns>             "list which namespaces will be considered for the --ns option.")
         //     .value_delimiter(',')

tests/by-util/test_pgrep.rs

@@ -501,6 +501,29 @@ fn test_nonexisting_session() {
     new_ucmd!().arg("--session=9999999999").fails();
 }
 
+#[test]
+#[cfg(target_os = "linux")]
+fn test_nonexisting_cgroup() {
+    new_ucmd!()
+        .arg("--cgroup")
+        .arg("NONEXISTING")
+        .fails()
+        .stdout_is("");
+}
+
+#[test]
+#[cfg(target_os = "linux")]
+fn test_cgroup() {
+    let init_is_systemd = new_ucmd!().arg("systemd").run().code() == 0;
+    if init_is_systemd {
+        new_ucmd!()
+            .arg("--cgroup")
+            .arg("/init.scope")
+            .succeeds()
+            .stdout_is("1\n");
+    }
+}
+
 #[test]
 #[cfg(target_os = "linux")]
 fn test_threads() {