diff --git a/Userland/Utilities/CMakeLists.txt b/Userland/Utilities/CMakeLists.txt index 5639dd0125..507e918277 100644 --- a/Userland/Utilities/CMakeLists.txt +++ b/Userland/Utilities/CMakeLists.txt @@ -141,6 +141,7 @@ target_link_libraries(netstat LibMain) target_link_libraries(nl LibMain) target_link_libraries(notify LibGUI LibMain) target_link_libraries(nproc LibMain) +target_link_libraries(ntpquery LibMain) target_link_libraries(open LibDesktop) target_link_libraries(pape LibGUI) target_link_libraries(passwd LibCrypt LibMain) diff --git a/Userland/Utilities/ntpquery.cpp b/Userland/Utilities/ntpquery.cpp index e764853354..26dce2a1a2 100644 --- a/Userland/Utilities/ntpquery.cpp +++ b/Userland/Utilities/ntpquery.cpp @@ -10,6 +10,8 @@ #include #include #include +#include +#include #include #include #include @@ -88,14 +90,14 @@ static String format_ntp_timestamp(NtpTimestamp ntp_timestamp) buffer[written] = '\0'; return buffer; } - +#ifdef __serenity__ +ErrorOr serenity_main(Main::Arguments arguments) +#else int main(int argc, char** argv) +#endif { #ifdef __serenity__ - if (pledge("stdio inet unix settime", nullptr) < 0) { - perror("pledge"); - return 1; - } + TRY(Core::System::pledge("stdio inet unix settime")); #endif bool adjust_time = false; @@ -117,7 +119,11 @@ int main(int argc, char** argv) args_parser.add_option(set_time, "Immediately set system time (requires root)", "set", 's'); args_parser.add_option(verbose, "Verbose output", "verbose", 'v'); args_parser.add_positional_argument(host, "NTP server", "host", Core::ArgsParser::Required::No); +#ifdef __serenity__ + args_parser.parse(arguments); +#else args_parser.parse(argc, argv); +#endif if (adjust_time && set_time) { warnln("-a and -s are mutually exclusive"); @@ -126,10 +132,7 @@ int main(int argc, char** argv) #ifdef __serenity__ if (!adjust_time && !set_time) { - if (pledge("stdio inet unix", nullptr) < 0) { - perror("pledge"); - return 1; - } + TRY(Core::System::pledge("stdio inet unix")); } #endif @@ -140,11 +143,8 @@ int main(int argc, char** argv) } #ifdef __serenity__ - if (pledge((adjust_time || set_time) ? "stdio inet settime" : "stdio inet", nullptr) < 0) { - perror("pledge"); - return 1; - } - unveil(nullptr, nullptr); + TRY(Core::System::pledge((adjust_time || set_time) ? "stdio inet settime" : "stdio inet")); + TRY(Core::System::unveil(nullptr, nullptr)); #endif int fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); @@ -322,4 +322,6 @@ int main(int argc, char** argv) return 1; } } + + return 0; }