RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-05-14 08:44:58 +00:00
Code Issues Activity Actions

usermod: Add -a and -r options to append or remove extra groups

Browse source
This commit is contained in:
Tim Ledbetter 2023-06-16 20:59:24 +01:00 • committed by Andreas Kling
parent caf55c0b2d
commit b17aa47769
2 changed files with 33 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

31
Userland/Utilities/usermod.cpp
View file

@ -49,8 +49,10 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
TRY(Core::System::unveil("/etc", "rwc"));
uid_t uid = 0;
bool append_extra_gids = false;
Optional<gid_t> gid;
bool lock = false;
bool remove_extra_gids = false;
bool unlock = false;
StringView new_home_directory;
bool move_home = false;
@ -61,6 +63,7 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
auto args_parser = Core::ArgsParser();
args_parser.set_general_help("Modify a user account");
args_parser.add_option(append_extra_gids, "Append the supplementary groups specified with the -G option to the user", "append", 'a');
args_parser.add_option(uid, "The new numerical value of the user's ID", "uid", 'u', "uid");
args_parser.add_option(Core::ArgsParser::Option {
.argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
@ -91,6 +94,7 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
},
});
args_parser.add_option(lock, "Lock password", "lock", 'L');
args_parser.add_option(remove_extra_gids, "Remove the supplementary groups specified with the -G option from the user", "remove", 'r');
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');
@ -100,6 +104,18 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
args_parser.parse(arguments);
if (extra_gids.is_empty() && (append_extra_gids || remove_extra_gids)) {
warnln("The -a and -r options can only be used with the -G option");
args_parser.print_usage(stderr, arguments.strings[0]);
return 1;
}
if (append_extra_gids && remove_extra_gids) {
warnln("The -a and -r options are mutually exclusive");
args_parser.print_usage(stderr, arguments.strings[0]);
return 1;
}
if (lock && unlock) {
warnln("The -L and -U options are mutually exclusive");
args_parser.print_usage(stderr, arguments.strings[0]);
@ -178,7 +194,20 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
target_account.set_gecos(gecos);
}
if (!extra_gids.is_empty()) {
if (append_extra_gids) {
for (auto gid : target_account.extra_gids())
extra_gids.append(gid);
}
if (remove_extra_gids) {
Vector<gid_t> current_extra_gids = target_account.extra_gids();
for (auto gid : extra_gids)
current_extra_gids.remove_all_matching([gid](auto current_gid) { return current_gid == gid; });
extra_gids = move(current_extra_gids);
}
if (!extra_gids.is_empty() || remove_extra_gids) {
target_account.set_extra_gids(extra_gids);
}