mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-11-26 17:40:27 +00:00
a9b387a1bf
Use unveil to allow access only to required paths. Switch to new pledge format.
60 lines
1.7 KiB
C++
60 lines
1.7 KiB
C++
/*
|
|
* Copyright (c) 2021, Aziz Berkay Yesilyurt <abyesilyurt@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/QuickSort.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>
|
|
|
|
ErrorOr<int> serenity_main(Main::Arguments args)
|
|
{
|
|
TRY(Core::System::pledge("stdio 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 invert_match = false;
|
|
char const* pattern = nullptr;
|
|
|
|
Core::ArgsParser args_parser;
|
|
args_parser.add_option(case_insensitive, "Make matches case-insensitive", nullptr, 'i');
|
|
args_parser.add_option(invert_match, "Select non-matching lines", "invert-match", 'v');
|
|
args_parser.add_positional_argument(pattern, "Process name to search for", "process-name");
|
|
args_parser.parse(args);
|
|
|
|
PosixOptions options {};
|
|
if (case_insensitive)
|
|
options |= PosixFlags::Insensitive;
|
|
|
|
Regex<PosixExtended> re(pattern, options);
|
|
if (re.parser_result.error != regex::Error::NoError) {
|
|
return 1;
|
|
}
|
|
|
|
auto all_processes = Core::ProcessStatisticsReader::get_all();
|
|
if (!all_processes.has_value())
|
|
return 1;
|
|
|
|
Vector<pid_t> matches;
|
|
for (auto& it : all_processes.value().processes) {
|
|
auto result = re.match(it.name, PosixFlags::Global);
|
|
if (result.success ^ invert_match) {
|
|
matches.append(it.pid);
|
|
}
|
|
}
|
|
|
|
quick_sort(matches);
|
|
|
|
for (auto& match : matches) {
|
|
outln("{}", match);
|
|
}
|
|
|
|
return 0;
|
|
}
|