diff --git a/Userland/Utilities/CMakeLists.txt b/Userland/Utilities/CMakeLists.txt index 5af8531716..a704faad54 100644 --- a/Userland/Utilities/CMakeLists.txt +++ b/Userland/Utilities/CMakeLists.txt @@ -51,6 +51,7 @@ foreach(CMD_SRC ${CMD_SOURCES}) endif() endforeach() +target_link_libraries(allocate LibMain) target_link_libraries(aplay LibAudio) target_link_libraries(arp LibMain) target_link_libraries(asctl LibAudio) diff --git a/Userland/Utilities/allocate.cpp b/Userland/Utilities/allocate.cpp index 903f99f676..3bdcb7a4ff 100644 --- a/Userland/Utilities/allocate.cpp +++ b/Userland/Utilities/allocate.cpp @@ -7,7 +7,7 @@ #include #include #include -#include +#include #include static void usage() @@ -22,25 +22,25 @@ enum class Unit { MiB, }; -int main(int argc, char** argv) +ErrorOr serenity_main(Main::Arguments arguments) { int count = 50; auto unit = Unit::MiB; - if (argc >= 2) { - auto number = String(argv[1]).to_uint(); + if (arguments.argc >= 2) { + auto number = arguments.strings[1].to_uint(); if (!number.has_value()) { usage(); } count = number.value(); } - if (argc >= 3) { - if (strcmp(argv[2], "B") == 0) + if (arguments.argc >= 3) { + if (arguments.strings[2] == "B") unit = Unit::Bytes; - else if (strcmp(argv[2], "KiB") == 0) + else if (arguments.strings[2] == "KiB") unit = Unit::KiB; - else if (strcmp(argv[2], "MiB") == 0) + else if (arguments.strings[2] == "MiB") unit = Unit::MiB; else usage();