mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-21 10:19:03 +03:00
c70f45ff44
This commit moves the length calculations out to be directly on the StringView users. This is an important step towards the goal of removing StringView(char const*), as it moves the responsibility of calculating the size of the string to the user of the StringView (which will prevent naive uses causing OOB access).
142 lines
4.5 KiB
C++
142 lines
4.5 KiB
C++
/*
|
|
* Copyright (c) 2021, Brandon Pruitt <brapru@pm.me>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <LibCore/Account.h>
|
|
#include <LibCore/ArgsParser.h>
|
|
#include <LibCore/File.h>
|
|
#include <LibCore/System.h>
|
|
#include <LibMain/Main.h>
|
|
#include <pwd.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
ErrorOr<int> serenity_main(Main::Arguments arguments)
|
|
{
|
|
if (geteuid() != 0) {
|
|
warnln("Not running as root :^(");
|
|
return 1;
|
|
}
|
|
|
|
TRY(Core::System::setegid(0));
|
|
|
|
TRY(Core::System::pledge("stdio wpath rpath cpath fattr tty"));
|
|
TRY(Core::System::unveil("/etc", "rwc"));
|
|
|
|
int uid = 0;
|
|
int gid = 0;
|
|
bool lock = false;
|
|
bool unlock = false;
|
|
char const* new_home_directory = nullptr;
|
|
bool move_home = false;
|
|
char const* shell = nullptr;
|
|
char const* gecos = nullptr;
|
|
char const* username = nullptr;
|
|
|
|
auto args_parser = Core::ArgsParser();
|
|
args_parser.set_general_help("Modify a user account");
|
|
args_parser.add_option(uid, "The new numerical value of the user's ID", "uid", 'u', "uid");
|
|
args_parser.add_option(gid, "The group number of the user's new initial login group", "gid", 'g', "gid");
|
|
args_parser.add_option(lock, "Lock password", "lock", 'L');
|
|
args_parser.add_option(unlock, "Unlock password", "unlock", 'U');
|
|
args_parser.add_option(new_home_directory, "The user's new login directory", "home", 'd', "new-home");
|
|
args_parser.add_option(move_home, "Move the content of the user's home directory to the new location", "move", 'm');
|
|
args_parser.add_option(shell, "The name of the user's new login shell", "shell", 's', "path-to-shell");
|
|
args_parser.add_option(gecos, "Change the GECOS field of the user", "gecos", 'n', "general-info");
|
|
args_parser.add_positional_argument(username, "Username of the account to modify", "username");
|
|
|
|
args_parser.parse(arguments);
|
|
|
|
auto account_or_error = Core::Account::from_name(username);
|
|
|
|
if (account_or_error.is_error()) {
|
|
warnln("Core::Account::from_name: {}", account_or_error.error());
|
|
return 1;
|
|
}
|
|
|
|
// target_account is the account we are modifying.
|
|
auto& target_account = account_or_error.value();
|
|
|
|
if (move_home) {
|
|
TRY(Core::System::unveil(target_account.home_directory().characters(), "c"));
|
|
TRY(Core::System::unveil({ new_home_directory, strlen(new_home_directory) }, "wc"));
|
|
}
|
|
|
|
unveil(nullptr, nullptr);
|
|
|
|
if (uid) {
|
|
if (uid < 0) {
|
|
warnln("invalid uid {}", uid);
|
|
return 1;
|
|
}
|
|
|
|
if (getpwuid(static_cast<uid_t>(uid))) {
|
|
warnln("uid {} already exists", uid);
|
|
return 1;
|
|
}
|
|
|
|
target_account.set_uid(uid);
|
|
}
|
|
|
|
if (gid) {
|
|
if (gid < 0) {
|
|
warnln("invalid gid {}", gid);
|
|
return 1;
|
|
}
|
|
|
|
target_account.set_gid(gid);
|
|
}
|
|
|
|
if (lock) {
|
|
target_account.set_password_enabled(false);
|
|
}
|
|
|
|
if (unlock) {
|
|
target_account.set_password_enabled(true);
|
|
}
|
|
|
|
if (new_home_directory) {
|
|
if (move_home) {
|
|
int rc = rename(target_account.home_directory().characters(), new_home_directory);
|
|
if (rc < 0) {
|
|
if (errno == EXDEV) {
|
|
auto result = Core::File::copy_file_or_directory(
|
|
new_home_directory, target_account.home_directory().characters(),
|
|
Core::File::RecursionMode::Allowed,
|
|
Core::File::LinkMode::Disallowed,
|
|
Core::File::AddDuplicateFileMarker::No);
|
|
|
|
if (result.is_error()) {
|
|
warnln("usermod: could not move directory {} : {}", target_account.home_directory().characters(), static_cast<Error const&>(result.error()));
|
|
return 1;
|
|
}
|
|
rc = unlink(target_account.home_directory().characters());
|
|
if (rc < 0)
|
|
warnln("usermod: unlink {} : {}", target_account.home_directory().characters(), strerror(errno));
|
|
} else {
|
|
warnln("usermod: could not move directory {} : {}", target_account.home_directory().characters(), strerror(errno));
|
|
}
|
|
}
|
|
}
|
|
|
|
target_account.set_home_directory(new_home_directory);
|
|
}
|
|
|
|
if (shell) {
|
|
target_account.set_shell(shell);
|
|
}
|
|
|
|
if (gecos) {
|
|
target_account.set_gecos(gecos);
|
|
}
|
|
|
|
TRY(Core::System::pledge("stdio wpath rpath cpath fattr"));
|
|
|
|
TRY(target_account.sync());
|
|
|
|
return 0;
|
|
}
|