ladybird/Userland/Utilities/killall.cpp
Shannon Booth e2e7c4d574 Everywhere: Use to_number<T> instead of to_{int,uint,float,double}
In a bunch of cases, this actually ends up simplifying the code as
to_number will handle something such as:

```
Optional<I> opt;
if constexpr (IsSigned<I>)
    opt = view.to_int<I>();
else
    opt = view.to_uint<I>();
```

For us.

The main goal here however is to have a single generic number conversion
API between all of the String classes.
2023-12-23 20:41:07 +01:00

69 lines
1.8 KiB
C++

/*
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
* Copyright (c) 2022, Zachary Penn <zack@sysdevs.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <AK/CharacterTypes.h>
#include <LibCore/ProcessStatisticsReader.h>
#include <LibCore/System.h>
#include <LibMain/Main.h>
#include <ctype.h>
#include <signal.h>
#include <stdlib.h>
static void print_usage_and_exit()
{
warnln("usage: killall [-signal] process_name");
exit(1);
}
static ErrorOr<int> kill_all(StringView process_name, unsigned const signum)
{
auto all_processes = TRY(Core::ProcessStatisticsReader::get_all());
for (auto& process : all_processes.processes) {
if (process.name == process_name) {
if (auto maybe_error = Core::System::kill(process.pid, signum); maybe_error.is_error())
warnln("{}", maybe_error.release_error());
}
}
return 0;
}
ErrorOr<int> serenity_main(Main::Arguments arguments)
{
unsigned signum = SIGTERM;
int name_argi = 1;
if (arguments.argc != 2 && arguments.argc != 3)
print_usage_and_exit();
if (arguments.argc == 3) {
name_argi = 2;
if (arguments.argv[1][0] != '-')
print_usage_and_exit();
Optional<unsigned> number;
if (is_ascii_alpha(arguments.argv[1][1])) {
int value = getsignalbyname(&arguments.argv[1][1]);
if (value > 0 && value < NSIG)
number = value;
}
if (!number.has_value())
number = arguments.strings[1].substring_view(1).to_number<unsigned>();
if (!number.has_value()) {
warnln("'{}' is not a valid signal name or number", &arguments.argv[1][1]);
return 2;
}
signum = number.value();
}
return kill_all(arguments.strings[name_argi], signum);
}