summaryrefslogtreecommitdiff
path: root/Userland/Utilities
diff options
context:
space:
mode:
Diffstat (limited to 'Userland/Utilities')
-rw-r--r--Userland/Utilities/CMakeLists.txt1
-rw-r--r--Userland/Utilities/w.cpp52
2 files changed, 15 insertions, 38 deletions
diff --git a/Userland/Utilities/CMakeLists.txt b/Userland/Utilities/CMakeLists.txt
index 3c23ac8c34..612063e393 100644
--- a/Userland/Utilities/CMakeLists.txt
+++ b/Userland/Utilities/CMakeLists.txt
@@ -102,5 +102,6 @@ target_link_libraries(zip LibArchive LibCompress LibCrypto)
target_link_libraries(cpp-lexer LibCpp)
target_link_libraries(cpp-parser LibCpp LibGUI)
target_link_libraries(cpp-preprocessor LibCpp LibGUI)
+target_link_libraries(w LibMain)
target_link_libraries(wasm LibWasm LibLine)
target_link_libraries(wsctl LibGUI)
diff --git a/Userland/Utilities/w.cpp b/Userland/Utilities/w.cpp
index ad30396efd..4dd68aa98a 100644
--- a/Userland/Utilities/w.cpp
+++ b/Userland/Utilities/w.cpp
@@ -9,48 +9,24 @@
#include <LibCore/DateTime.h>
#include <LibCore/File.h>
#include <LibCore/ProcessStatisticsReader.h>
+#include <LibMain/Main.h>
+#include <LibSystem/Wrappers.h>
#include <pwd.h>
-#include <stdio.h>
#include <sys/stat.h>
#include <time.h>
-int main()
+ErrorOr<int> serenity_main(Main::Arguments)
{
- if (pledge("stdio rpath", nullptr) < 0) {
- perror("pledge");
- return 1;
- }
-
- if (unveil("/dev", "r") < 0) {
- perror("unveil");
- return 1;
- }
-
- if (unveil("/etc/passwd", "r") < 0) {
- perror("unveil");
- return 1;
- }
-
- if (unveil("/var/run/utmp", "r") < 0) {
- perror("unveil");
- return 1;
- }
-
- if (unveil("/proc", "r") < 0) {
- perror("unveil");
- return 1;
- }
-
- unveil(nullptr, nullptr);
-
- auto file_or_error = Core::File::open("/var/run/utmp", Core::OpenMode::ReadOnly);
- if (file_or_error.is_error()) {
- warnln("Error: {}", file_or_error.error());
- return 1;
- }
- auto& file = *file_or_error.value();
- auto json = JsonValue::from_string(file.read_all());
- if (json.is_error() || !json.value().is_object()) {
+ TRY(System::pledge("stdio rpath", nullptr));
+ TRY(System::unveil("/dev", "r"));
+ TRY(System::unveil("/etc/passwd", "r"));
+ TRY(System::unveil("/var/run/utmp", "r"));
+ TRY(System::unveil("/proc", "r"));
+ TRY(System::unveil(nullptr, nullptr));
+
+ auto file = TRY(Core::File::open("/var/run/utmp", Core::OpenMode::ReadOnly));
+ auto json = TRY(JsonValue::from_string(file->read_all()));
+ if (!json.is_object()) {
warnln("Error: Could not parse /var/run/utmp");
return 1;
}
@@ -64,7 +40,7 @@ int main()
auto now = time(nullptr);
outln("\033[1m{:10} {:12} {:16} {:6} {}\033[0m", "USER", "TTY", "LOGIN@", "IDLE", "WHAT");
- json.value().as_object().for_each_member([&](auto& tty, auto& value) {
+ json.as_object().for_each_member([&](auto& tty, auto& value) {
const JsonObject& entry = value.as_object();
auto uid = entry.get("uid").to_u32();
[[maybe_unused]] auto pid = entry.get("pid").to_i32();