mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-11 13:36:13 +03:00
chown: Implement recursion
This commit is contained in:
parent
7b7c9a2194
commit
25e6ad8d3e
Notes:
sideshowbarker
2024-07-17 08:34:38 +09:00
Author: https://github.com/timschumi Commit: https://github.com/SerenityOS/serenity/commit/25e6ad8d3e Pull-request: https://github.com/SerenityOS/serenity/pull/14689 Reviewed-by: https://github.com/linusg
@ -7,6 +7,7 @@
|
||||
#include <AK/String.h>
|
||||
#include <AK/Vector.h>
|
||||
#include <LibCore/ArgsParser.h>
|
||||
#include <LibCore/DirIterator.h>
|
||||
#include <LibCore/System.h>
|
||||
#include <LibMain/Main.h>
|
||||
#include <grp.h>
|
||||
@ -22,11 +23,15 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
|
||||
|
||||
String spec;
|
||||
Vector<StringView> paths;
|
||||
bool dont_follow_symlinks = false;
|
||||
bool no_dereference = false;
|
||||
bool recursive = false;
|
||||
bool follow_symlinks = false;
|
||||
|
||||
Core::ArgsParser args_parser;
|
||||
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(paths, "Paths to files", "PATH");
|
||||
args_parser.parse(arguments);
|
||||
@ -66,12 +71,30 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
|
||||
}
|
||||
}
|
||||
|
||||
for (auto path : paths) {
|
||||
if (dont_follow_symlinks) {
|
||||
Function<ErrorOr<void>(StringView)> update_path_owner = [&](StringView path) -> ErrorOr<void> {
|
||||
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));
|
||||
} else {
|
||||
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;
|
||||
|
Loading…
Reference in New Issue
Block a user