mirror of
https://github.com/SerenityOS/serenity
synced 2024-10-07 00:19:27 +00:00
58 lines
2.3 KiB
C++
58 lines
2.3 KiB
C++
/*
|
|
* Copyright (c) 2022, Ashley N. <dev-serenity@ne0ndrag0n.com>
|
|
* Copyright (c) 2022, the SerenityOS developers.
|
|
* Copyright (c) 2023, Fabian Dellwing <fabian@dellwing.net>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "EscalatorWindow.h"
|
|
#include <AK/ByteString.h>
|
|
#include <LibCore/Account.h>
|
|
#include <LibCore/ArgsParser.h>
|
|
#include <LibCore/System.h>
|
|
#include <LibGUI/Application.h>
|
|
#include <LibGUI/Desktop.h>
|
|
#include <LibGUI/MessageBox.h>
|
|
#include <LibMain/Main.h>
|
|
|
|
ErrorOr<int> serenity_main(Main::Arguments arguments)
|
|
{
|
|
Vector<StringView> command;
|
|
Core::ArgsParser args_parser;
|
|
StringView description;
|
|
bool preserve_env = false;
|
|
bool forward_stdin = false;
|
|
bool forward_stdout = false;
|
|
args_parser.set_general_help("Escalate privilege to root for a given command using a GUI prompt.");
|
|
args_parser.set_stop_on_first_non_option(true);
|
|
args_parser.add_option(description, "Custom prompt to use for dialog", "prompt", 'P', "prompt");
|
|
args_parser.add_option(preserve_env, "Preserve user environment when running command", "preserve-env", 'E');
|
|
args_parser.add_option(forward_stdin, "Forward stdin to targets stdin", "forward-stdin", 'I');
|
|
args_parser.add_option(forward_stdout, "Forward targets stdout to stdout", "forward-stdout", 'O');
|
|
args_parser.add_positional_argument(command, "Command to run at elevated privilege level", "command");
|
|
args_parser.parse(arguments);
|
|
|
|
TRY(Core::System::pledge("stdio recvfd sendfd thread cpath rpath wpath unix proc exec id"));
|
|
|
|
auto app = TRY(GUI::Application::create(arguments));
|
|
|
|
auto executable_path = Core::System::resolve_executable_from_environment(command[0], AT_EACCESS);
|
|
if (executable_path.is_error()) {
|
|
GUI::MessageBox::show_error(nullptr, ByteString::formatted("Could not execute command {}: Command not found.", command[0]));
|
|
return 127;
|
|
}
|
|
|
|
auto current_user = TRY(Core::Account::self());
|
|
auto window = TRY(Escalator::EscalatorWindow::try_create(executable_path.value(), command, Escalator::EscalatorWindow::Options { description, current_user, preserve_env, forward_stdin, forward_stdout }));
|
|
|
|
if (current_user.uid() != 0) {
|
|
window->show();
|
|
return app->exec();
|
|
} else {
|
|
// Run directly as root if already root uid.
|
|
TRY(window->execute_command());
|
|
return 0;
|
|
}
|
|
}
|