RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-27 10:57:35 +00:00
Code Issues Activity Actions

chown: Implement recursion

Browse source
This commit is contained in:
Tim Schumacher 2022-07-25 15:04:39 +02:00 • committed by Linus Groh
parent 7b7c9a2194
commit 25e6ad8d3e
1 changed files with 27 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

31
Userland/Utilities/chown.cpp
View file

@ -7,6 +7,7 @@
#include <AK/String.h> #include <AK/String.h>
#include <AK/Vector.h> #include <AK/Vector.h>
#include <LibCore/ArgsParser.h> #include <LibCore/ArgsParser.h>
#include <LibCore/DirIterator.h>
#include <LibCore/System.h> #include <LibCore/System.h>
#include <LibMain/Main.h> #include <LibMain/Main.h>
#include <grp.h> #include <grp.h>
@ -22,11 +23,15 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
String spec; String spec;
Vector<StringView> paths; Vector<StringView> paths;
bool dont_follow_symlinks = false; bool no_dereference = false;
bool recursive = false;
bool follow_symlinks = false;
Core::ArgsParser args_parser; Core::ArgsParser args_parser;
args_parser.set_general_help("Change the ownership of a file or directory."); args_parser.set_general_help("Change the ownership of a file or directory.");
args_parser.add_option(dont_follow_symlinks, "Don't follow symlinks", "no-dereference", 'h'); args_parser.add_option(no_dereference, "Don't follow symlinks", "no-dereference", 'h');
args_parser.add_option(recursive, "Change file ownership recursively", "recursive", 'R');
args_parser.add_option(follow_symlinks, "Follow symlinks while recursing into directories", nullptr, 'L');
args_parser.add_positional_argument(spec, "User and group IDs", "USER[:GROUP]"); args_parser.add_positional_argument(spec, "User and group IDs", "USER[:GROUP]");
args_parser.add_positional_argument(paths, "Paths to files", "PATH"); args_parser.add_positional_argument(paths, "Paths to files", "PATH");
args_parser.parse(arguments); args_parser.parse(arguments);
@ -66,12 +71,30 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
} }
} }
for (auto path : paths) { Function<ErrorOr<void>(StringView)> update_path_owner = [&](StringView path) -> ErrorOr<void> {
if (dont_follow_symlinks) { auto stat = TRY(Core::System::lstat(path));
if (S_ISLNK(stat.st_mode) && !follow_symlinks && !paths.contains_slow(path))
return {};
if (no_dereference) {
TRY(Core::System::lchown(path, new_uid, new_gid)); TRY(Core::System::lchown(path, new_uid, new_gid));
} else { } else {
TRY(Core::System::chown(path, new_uid, new_gid)); TRY(Core::System::chown(path, new_uid, new_gid));
} }
if (recursive && S_ISDIR(stat.st_mode)) {
Core::DirIterator it(path, Core::DirIterator::Flags::SkipParentAndBaseDir);
while (it.has_next())
TRY(update_path_owner(it.next_full_path()));
}
return {};
};
for (auto path : paths) {
TRY(update_path_owner(path));
} }
return 0; return 0;