Skip to content

Commit

Permalink
Userland: Add groups program
Browse files Browse the repository at this point in the history
  • Loading branch information
setepenre authored and linusg committed May 5, 2021
1 parent 5601037 commit 580b335
Showing 1 changed file with 82 additions and 0 deletions.
82 changes: 82 additions & 0 deletions Userland/Utilities/groups.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,82 @@
/*
* Copyright (c) 2021, the SerenityOS developers.
*
* SPDX-License-Identifier: BSD-2-Clause
*/

#include <AK/Vector.h>
#include <LibCore/Account.h>
#include <LibCore/ArgsParser.h>
#include <grp.h>
#include <unistd.h>

static void print_account_gids(const Core::Account& account)
{
auto* gr = getgrgid(account.gid());
if (!gr) {
outln();
return;
}

out("{}", gr->gr_name);
for (auto& gid : account.extra_gids()) {
gr = getgrgid(gid);
out(" {}", gr->gr_name);
}
outln();
}

int main(int argc, char** argv)
{
if (unveil("/etc/passwd", "r") < 0) {
perror("unveil");
return 1;
}

if (unveil("/etc/shadow", "r") < 0) {
perror("unveil");
return 1;
}

if (unveil("/etc/group", "r") < 0) {
perror("unveil");
return 1;
}

if (unveil(nullptr, nullptr) < 0) {
perror("unveil");
return 1;
}

if (pledge("stdio rpath", nullptr) < 0) {
perror("pledge");
return 1;
}

Vector<const char*> usernames;

Core::ArgsParser args_parser;
args_parser.set_general_help("Print group memberships for each username or, if no username is specified, for the current process.");
args_parser.add_positional_argument(usernames, "Usernames to list group memberships for", "usernames", Core::ArgsParser::Required::No);
args_parser.parse(argc, argv);

if (usernames.is_empty()) {
auto result = Core::Account::from_uid(geteuid());
if (result.is_error()) {
warnln("{}", result.error());
return 1;
}
print_account_gids(result.value());
}

for (auto username : usernames) {
auto result = Core::Account::from_name(username);
if (result.is_error()) {
warnln("{} '{}'", result.error(), username);
continue;
}
out("{} : ", username);
print_account_gids(result.value());
}
return 0;
}

0 comments on commit 580b335

Please sign in to comment.