ladybird/Userland/Utilities/ps.cpp
Brian Gianforcaro 1682f0b760 Everything: Move to SPDX license identifiers in all files.
SPDX License Identifiers are a more compact / standardized
way of representing file license information.

See: https://spdx.dev/resources/use/#identifiers

This was done with the `ambr` search and replace tool.

 ambr --no-parent-ignore --key-from-file --rep-from-file key.txt rep.txt *
2021-04-22 11:22:27 +02:00

140 lines
3.8 KiB
C++

/*
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <LibCore/ArgsParser.h>
#include <LibCore/File.h>
#include <LibCore/ProcessStatisticsReader.h>
#include <fcntl.h>
#include <stdio.h>
#include <unistd.h>
int main(int argc, char** argv)
{
if (pledge("stdio rpath tty", nullptr) < 0) {
perror("pledge");
return 1;
}
String this_tty = ttyname(STDIN_FILENO);
if (pledge("stdio rpath", nullptr) < 0) {
perror("pledge");
return 1;
}
if (unveil("/proc/all", "r") < 0) {
perror("unveil");
return 1;
}
if (unveil("/etc/passwd", "r") < 0) {
perror("unveil");
return 1;
}
unveil(nullptr, nullptr);
enum class Alignment {
Left,
Right,
};
struct Column {
String title;
Alignment alignment { Alignment::Left };
int width { 0 };
String buffer;
};
bool every_process_flag = false;
bool full_format_flag = false;
Core::ArgsParser args_parser;
args_parser.add_option(every_process_flag, "Show every process", nullptr, 'e');
args_parser.add_option(full_format_flag, "Full format", nullptr, 'f');
args_parser.parse(argc, argv);
Vector<Column> columns;
int uid_column = -1;
int pid_column = -1;
int ppid_column = -1;
int state_column = -1;
int tty_column = -1;
int cmd_column = -1;
auto add_column = [&](auto title, auto alignment, auto width) {
columns.append({ title, alignment, width, {} });
return columns.size() - 1;
};
if (full_format_flag) {
uid_column = add_column("UID", Alignment::Left, 9);
pid_column = add_column("PID", Alignment::Right, 5);
ppid_column = add_column("PPID", Alignment::Right, 5);
state_column = add_column("STATE", Alignment::Left, 12);
tty_column = add_column("TTY", Alignment::Left, 6);
cmd_column = add_column("CMD", Alignment::Left, 0);
} else {
pid_column = add_column("PID", Alignment::Right, 5);
tty_column = add_column("TTY", Alignment::Left, 6);
cmd_column = add_column("CMD", Alignment::Left, 0);
}
auto print_column = [](auto& column, auto& string) {
if (!column.width) {
printf("%s", string.characters());
return;
}
if (column.alignment == Alignment::Right)
printf("%*s ", column.width, string.characters());
else
printf("%-*s ", column.width, string.characters());
};
for (auto& column : columns)
print_column(column, column.title);
printf("\n");
auto all_processes = Core::ProcessStatisticsReader::get_all();
if (!all_processes.has_value())
return 1;
for (const auto& it : all_processes.value()) {
const auto& proc = it.value;
auto tty = proc.tty;
if (!every_process_flag && tty != this_tty)
continue;
if (tty.starts_with("/dev/"))
tty = tty.characters() + 5;
else
tty = "n/a";
auto* state = proc.threads.is_empty() ? "Zombie" : proc.threads.first().state.characters();
if (uid_column != -1)
columns[uid_column].buffer = proc.username;
if (pid_column != -1)
columns[pid_column].buffer = String::number(proc.pid);
if (ppid_column != -1)
columns[ppid_column].buffer = String::number(proc.ppid);
if (tty_column != -1)
columns[tty_column].buffer = tty;
if (state_column != -1)
columns[state_column].buffer = state;
if (cmd_column != -1)
columns[cmd_column].buffer = proc.name;
for (auto& column : columns)
print_column(column, column.buffer);
printf("\n");
}
return 0;
}