beenull/ladybird
1
0
Fork 0
mirror of https://github.com/LadybirdBrowser/ladybird.git synced 2024-09-30 08:41:15 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Userland: Use Core::ArgsParser for 'su'

Browse source
This commit is contained in:
Linus Groh 2020-08-05 20:26:32 +02:00 committed by Andreas Kling
parent d320f7b9d2
commit 3298b0318c
Notes: sideshowbarker 2024-07-19 04:12:15 +09:00 
Author: https://github.com/linusg
Commit: https://github.com/SerenityOS/serenity/commit/3298b0318c7
Pull-request: https://github.com/SerenityOS/serenity/pull/3019
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
Userland/su.cpp
View file

@ -25,6 +25,7 @@
*/
#include <AK/Vector.h>
#include <LibCore/ArgsParser.h>
#include <LibCore/GetPassword.h>
#include <alloca.h>
#include <grp.h>
@ -37,17 +38,22 @@ extern "C" int main(int, char**);
int main(int argc, char** argv)
{
if (geteuid() != 0) {
const char* user = nullptr;
Core::ArgsParser args_parser;
args_parser.add_positional_argument(user, "User to switch to (defaults to user with UID 0)", "user", Core::ArgsParser::Required::No);
args_parser.parse(argc, argv);
if (geteuid() != 0)
fprintf(stderr, "Not running as root :(\n");
}
uid_t uid = 0;
gid_t gid = 0;
struct passwd* pwd = nullptr;
if (argc > 1) {
pwd = getpwnam(argv[1]);
if (user) {
pwd = getpwnam(user);
if (!pwd) {
fprintf(stderr, "No such user: %s\n", argv[1]);
fprintf(stderr, "No such user: %s\n", user);
return 1;
}
uid = pwd->pw_uid;