RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-25 17:37:37 +00:00
Code Issues Activity Actions

Utilities: Add pkill

Browse source
This commit is contained in:
Maxwell Trussell 2022-10-12 00:14:50 -07:00 • committed by Andreas Kling
parent a982e0380e
commit f5e68fcc20
2 changed files with 73 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

71
Userland/Utilities/pkill.cpp Normal file
View file

@ -0,0 +1,71 @@
/*
* Copyright (c) 2022, Maxwell Trussell <maxtrussell@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <AK/Format.h>
#include <AK/Vector.h>
#include <LibCore/ArgsParser.h>
#include <LibCore/ProcessStatisticsReader.h>
#include <LibCore/System.h>
#include <LibMain/Main.h>
#include <LibRegex/Regex.h>
#include <LibRegex/RegexOptions.h>
#include <signal.h>
#include <sys/types.h>
ErrorOr<int> serenity_main(Main::Arguments args)
{
TRY(Core::System::pledge("stdio proc rpath"));
TRY(Core::System::unveil("/proc/all", "r"));
TRY(Core::System::unveil("/etc/passwd", "r"));
TRY(Core::System::unveil(nullptr, nullptr));
bool case_insensitive = false;
bool echo = false;
char const* pattern = nullptr;
int signal = SIGTERM;
Core::ArgsParser args_parser;
args_parser.add_option(case_insensitive, "Make matches case-insensitive", nullptr, 'i');
args_parser.add_option(echo, "Display what is killed", "echo", 'e');
args_parser.add_option(signal, "Signal number to send", "signal", 's', "number");
args_parser.add_positional_argument(pattern, "Process name to search for", "process-name");
args_parser.parse(args);
auto all_processes = Core::ProcessStatisticsReader::get_all();
if (!all_processes.has_value()) {
return 1;
}
PosixOptions options {};
if (case_insensitive) {
options |= PosixFlags::Insensitive;
}
Regex<PosixExtended> re(pattern, options);
if (re.parser_result.error != regex::Error::NoError) {
return 1;
}
Vector<Core::ProcessStatistics> matched_processes;
for (auto& process : all_processes.value().processes) {
auto result = re.match(process.name, PosixFlags::Global);
if (result.success) {
matched_processes.append(process);
}
}
if (matched_processes.is_empty()) {
return 1;
}
for (auto& process : matched_processes) {
if (echo) {
outln("{} killed (pid {})", process.name, process.pid);
}
kill(process.pid, signal);
}
return 0;
}