ladybird/Userland/Utilities/kill.cpp
itskarudo 16aeb8b51d kill: Parse CLI arguments using Arguments.strings
The previous implementation used argv and LibC to parse arguments,
it now uses the Arguments.strings interface to do so.
2022-02-12 10:53:07 -05:00

79 lines
1.8 KiB
C++

/*
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <AK/Optional.h>
#include <AK/String.h>
#include <LibCore/System.h>
#include <LibMain/Main.h>
#include <ctype.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
static void print_usage_and_exit()
{
warnln("usage: kill [-signal] <PID>");
exit(1);
}
ErrorOr<int> serenity_main(Main::Arguments arguments)
{
TRY(Core::System::pledge("stdio proc"));
int argc = arguments.argc;
auto strings = arguments.strings;
if (argc == 2 && strings[1] == "-l") {
for (size_t i = 0; i < NSIG; ++i) {
if (i && !(i % 5))
outln("");
out("{:2}) {:10}", i, getsignalname(i));
}
outln("");
return 0;
}
if (argc != 2 && argc != 3)
print_usage_and_exit();
unsigned signum = SIGTERM;
int pid_argi = 1;
if (argc == 3) {
pid_argi = 2;
if (strings[1][0] != '-')
print_usage_and_exit();
Optional<unsigned> number;
if (isalpha(strings[1][1])) {
int value = getsignalbyname(&strings[1][1]);
if (value >= 0 && value < NSIG)
number = value;
}
if (!number.has_value())
number = strings[1].substring_view(1, 1).to_uint();
if (!number.has_value()) {
warnln("'{}' is not a valid signal name or number", &strings[1][1]);
return 2;
}
signum = number.value();
}
auto pid_opt = strings[pid_argi].to_int();
if (!pid_opt.has_value()) {
warnln("'{}' is not a valid PID", strings[pid_argi]);
return 3;
}
pid_t pid = pid_opt.value();
int rc = kill(pid, signum);
if (rc < 0)
perror("kill");
return 0;
}