serenity/Userland/Utilities/find.cpp
Tim Ledbetter 6f1f0710f8 find: Don't canonicalize path operands
The POSIX specification for `find` says that: "Each path operand shall
be evaluated unaltered as it was provided, including all trailing
<slash> characters". This also matches the behavior of `find` on
FreeBSD and Linux.
2023-10-07 11:37:53 +02:00

1033 lines
30 KiB
C++

/*
* Copyright (c) 2020-2021, Sergey Bugaev <bugaevc@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <AK/Assertions.h>
#include <AK/CheckedFormatString.h>
#include <AK/LexicalPath.h>
#include <AK/NonnullOwnPtr.h>
#include <AK/OwnPtr.h>
#include <AK/Time.h>
#include <AK/URL.h>
#include <AK/Vector.h>
#include <LibCore/DirIterator.h>
#include <LibCore/System.h>
#include <LibFileSystem/FileSystem.h>
#include <LibMain/Main.h>
#include <LibRegex/Regex.h>
#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <grp.h>
#include <pwd.h>
#include <stdio.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>
bool g_follow_symlinks = false;
bool g_there_was_an_error = false;
bool g_have_seen_action_command = false;
bool g_print_hyperlinks = false;
Optional<u32> g_max_depth = {};
Optional<u32> g_min_depth = {};
template<typename... Parameters>
[[noreturn]] static void fatal_error(CheckedFormatString<Parameters...>&& fmtstr, Parameters const&... parameters)
{
warn("\033[31m");
warn(move(fmtstr), parameters...);
warn("\033[0m");
warnln();
exit(1);
}
struct FileData {
// The current path specified on the command line
StringView root_path;
// Path to the file relative to the current root path
LexicalPath relative_path;
// The parent directory of the file.
int dirfd { -1 };
// The file's basename, relative to the directory.
char const* basename { nullptr };
// Optionally, cached information as returned by stat/lstat/fstatat.
struct stat stat {
};
bool stat_is_valid : 1 { false };
// File type as returned from readdir(), or DT_UNKNOWN.
unsigned char d_type { DT_UNKNOWN };
DeprecatedString full_path() const
{
if (root_path.is_empty())
return relative_path.string();
if (relative_path.string() == ".")
return root_path;
// POSIX says that a single slash should be added between the root path and the relative portion if the root path doesn't end in one
// any additional trailing slashes should be left unaltered.
if (root_path.ends_with('/'))
return DeprecatedString::formatted("{}{}", root_path, relative_path.string());
return DeprecatedString::formatted("{}/{}", root_path, relative_path.string());
}
const struct stat* ensure_stat()
{
if (stat_is_valid)
return &stat;
int flags = g_follow_symlinks ? 0 : AT_SYMLINK_NOFOLLOW;
int rc = fstatat(dirfd, basename, &stat, flags);
if (rc < 0) {
perror(full_path().characters());
g_there_was_an_error = true;
return nullptr;
}
stat_is_valid = true;
if (S_ISREG(stat.st_mode))
d_type = DT_REG;
else if (S_ISDIR(stat.st_mode))
d_type = DT_DIR;
else if (S_ISCHR(stat.st_mode))
d_type = DT_CHR;
else if (S_ISBLK(stat.st_mode))
d_type = DT_BLK;
else if (S_ISFIFO(stat.st_mode))
d_type = DT_FIFO;
else if (S_ISLNK(stat.st_mode))
d_type = DT_LNK;
else if (S_ISSOCK(stat.st_mode))
d_type = DT_SOCK;
else
VERIFY_NOT_REACHED();
return &stat;
}
};
template<Unsigned T>
class NumericRange {
public:
NumericRange() = default;
static ErrorOr<NumericRange<T>> parse(StringView arg)
{
if (arg.is_empty())
return Error::from_errno(EINVAL);
auto comparison_type = ComparisonType::Equal;
if (arg[0] == '-') {
comparison_type = ComparisonType::LessThan;
arg = arg.substring_view(1);
} else if (arg[0] == '+') {
comparison_type = ComparisonType::GreaterThan;
arg = arg.substring_view(1);
}
auto maybe_value = arg.to_uint<T>();
if (!maybe_value.has_value())
return Error::from_errno(EINVAL);
return NumericRange<T>(maybe_value.release_value(), comparison_type);
}
bool contains(T other) const
{
if (m_comparison_type == ComparisonType::LessThan)
return other < m_value;
if (m_comparison_type == ComparisonType::GreaterThan)
return other > m_value;
return other == m_value;
}
private:
enum class ComparisonType {
Equal,
LessThan,
GreaterThan
};
NumericRange(T value, ComparisonType comparison_type)
: m_value(value)
, m_comparison_type(comparison_type)
{
}
T m_value {};
ComparisonType m_comparison_type { ComparisonType::Equal };
};
class Command {
public:
virtual ~Command() = default;
virtual bool evaluate(FileData& file_data) const = 0;
};
class StatCommand : public Command {
public:
virtual bool evaluate(const struct stat&) const = 0;
private:
virtual bool evaluate(FileData& file_data) const override
{
const struct stat* stat = file_data.ensure_stat();
if (!stat)
return false;
return evaluate(*stat);
}
};
class MaxDepthCommand : public Command {
public:
MaxDepthCommand(char const* arg)
{
auto max_depth_string = StringView { arg, strlen(arg) };
auto maybe_max_depth = max_depth_string.to_uint<u32>();
if (!maybe_max_depth.has_value())
fatal_error("-maxdepth: '{}' is not a valid non-negative integer", arg);
g_max_depth = maybe_max_depth.value();
}
virtual bool evaluate(FileData&) const override
{
return true;
}
};
class MinDepthCommand : public Command {
public:
MinDepthCommand(char const* arg)
{
auto min_depth_string = StringView { arg, strlen(arg) };
auto maybe_min_depth = min_depth_string.to_uint<u32>();
if (!maybe_min_depth.has_value())
fatal_error("-mindepth: '{}' is not a valid non-negative integer", arg);
g_min_depth = maybe_min_depth.value();
}
virtual bool evaluate(FileData&) const override
{
return true;
}
};
class TypeCommand final : public Command {
public:
TypeCommand(char const* arg)
{
StringView type { arg, strlen(arg) };
if (type.length() != 1 || !"bcdlpfs"sv.contains(type[0]))
fatal_error("Invalid mode: \033[1m{}", arg);
m_type = type[0];
}
private:
virtual bool evaluate(FileData& file_data) const override
{
// First, make sure we have a type, but avoid calling
// sys$stat() unless we need to.
if (file_data.d_type == DT_UNKNOWN) {
if (file_data.ensure_stat() == nullptr)
return false;
}
auto type = file_data.d_type;
switch (m_type) {
case 'b':
return type == DT_BLK;
case 'c':
return type == DT_CHR;
case 'd':
return type == DT_DIR;
case 'l':
return type == DT_LNK;
case 'p':
return type == DT_FIFO;
case 'f':
return type == DT_REG;
case 's':
return type == DT_SOCK;
default:
// We've verified this is a correct character before.
VERIFY_NOT_REACHED();
}
}
char m_type { 0 };
};
class LinksCommand final : public StatCommand {
public:
LinksCommand(char const* arg)
{
StringView arg_string { arg, strlen(arg) };
auto links_or_error = NumericRange<nlink_t>::parse(arg_string);
if (links_or_error.is_error())
fatal_error("Invalid number: \033[1m{}", arg);
m_links = links_or_error.release_value();
}
private:
virtual bool evaluate(const struct stat& stat) const override
{
return m_links.contains(stat.st_nlink);
}
NumericRange<nlink_t> m_links {};
};
class UserCommand final : public StatCommand {
public:
UserCommand(char const* arg)
{
if (struct passwd* passwd = getpwnam(arg)) {
m_uid = passwd->pw_uid;
} else {
// Attempt to parse it as decimal UID.
auto number = StringView { arg, strlen(arg) }.to_uint<uid_t>();
if (!number.has_value())
fatal_error("Invalid user: \033[1m{}", arg);
m_uid = number.value();
}
}
private:
virtual bool evaluate(const struct stat& stat) const override
{
return stat.st_uid == m_uid;
}
uid_t m_uid { 0 };
};
class GroupCommand final : public StatCommand {
public:
GroupCommand(char const* arg)
{
if (struct group* gr = getgrnam(arg)) {
m_gid = gr->gr_gid;
} else {
// Attempt to parse it as decimal GID.
auto number = StringView { arg, strlen(arg) }.to_uint<gid_t>();
if (!number.has_value())
fatal_error("Invalid group: \033[1m{}", arg);
m_gid = number.value();
}
}
private:
virtual bool evaluate(const struct stat& stat) const override
{
return stat.st_gid == m_gid;
}
gid_t m_gid { 0 };
};
class SizeCommand final : public StatCommand {
public:
SizeCommand(char const* arg)
{
StringView view { arg, strlen(arg) };
auto suffix = view[view.length() - 1];
if (!is_ascii_digit(suffix)) {
switch (suffix) {
case 'c':
m_unit_size = 1;
break;
case 'w':
m_unit_size = 2;
break;
case 'k':
m_unit_size = KiB;
break;
case 'M':
m_unit_size = MiB;
break;
case 'G':
m_unit_size = GiB;
break;
case 'b':
// The behavior of this suffix is the same as no suffix.
break;
default:
fatal_error("Invalid -size type '{}'", suffix);
}
view = view.substring_view(0, view.length() - 1);
}
auto number_or_error = NumericRange<u64>::parse(view);
if (number_or_error.is_error())
fatal_error("Invalid size: \033[1m{}", arg);
m_number_of_units = number_or_error.release_value();
}
private:
virtual bool evaluate(const struct stat& stat) const override
{
if (m_unit_size == 1)
return m_number_of_units.contains(stat.st_size);
auto size_divided_by_unit_rounded_up = (stat.st_size + m_unit_size - 1) / m_unit_size;
return m_number_of_units.contains(size_divided_by_unit_rounded_up);
}
NumericRange<u64> m_number_of_units {};
u64 m_unit_size { 512 };
};
class EmptyCommand final : public Command {
public:
EmptyCommand()
{
}
private:
virtual bool evaluate(FileData& file_data) const override
{
struct stat const* stat = file_data.ensure_stat();
if (!stat)
return false;
if (S_ISREG(stat->st_mode))
return stat->st_size == 0;
if (S_ISDIR(stat->st_mode)) {
auto dir_iterator = Core::DirIterator(file_data.full_path(), Core::DirIterator::SkipDots);
return !dir_iterator.has_next();
}
return false;
}
};
class NameCommand : public Command {
public:
NameCommand(char const* pattern, CaseSensitivity case_sensitivity)
: m_pattern(pattern, strlen(pattern))
, m_case_sensitivity(case_sensitivity)
{
}
private:
virtual bool evaluate(FileData& file_data) const override
{
return file_data.relative_path.basename().matches(m_pattern, m_case_sensitivity);
}
StringView m_pattern;
CaseSensitivity m_case_sensitivity { CaseSensitivity::CaseSensitive };
};
class RegexCommand : public Command {
public:
RegexCommand(Regex<PosixExtended>&& regex)
: m_regex(move(regex))
{
}
private:
virtual bool evaluate(FileData& file_data) const override
{
auto haystack = file_data.full_path();
auto match_result = m_regex.match(haystack);
return match_result.success;
}
Regex<PosixExtended> m_regex;
};
class AccessCommand final : public Command {
public:
AccessCommand(mode_t mode)
: m_mode(mode)
{
}
private:
virtual bool evaluate(FileData& file_data) const override
{
auto maybe_error = Core::System::access(file_data.full_path(), m_mode);
return !maybe_error.is_error();
}
mode_t m_mode { 0 };
};
class NewerCommand final : public StatCommand {
public:
enum class TimestampType {
LastAccess,
Creation,
LastModification
};
NewerCommand(char const* arg, TimestampType timestamp_type)
{
auto stat_function = g_follow_symlinks ? Core::System::stat : Core::System::lstat;
auto stat_or_error = stat_function({ arg, strlen(arg) });
if (stat_or_error.is_error())
fatal_error("find: '{}': {}", arg, strerror(stat_or_error.error().code()));
m_reference_file_stat = stat_or_error.release_value();
m_timestamp_type = timestamp_type;
}
private:
virtual bool evaluate(struct stat const& stat) const override
{
struct timespec current_file_timestamp;
struct timespec reference_file_timestamp;
switch (m_timestamp_type) {
case TimestampType::LastAccess:
current_file_timestamp = stat.st_atim;
reference_file_timestamp = m_reference_file_stat.st_atim;
break;
case TimestampType::Creation:
current_file_timestamp = stat.st_ctim;
reference_file_timestamp = m_reference_file_stat.st_ctim;
break;
case TimestampType::LastModification:
current_file_timestamp = stat.st_mtim;
reference_file_timestamp = m_reference_file_stat.st_mtim;
break;
}
return Duration::from_timespec(current_file_timestamp) > Duration::from_timespec(reference_file_timestamp);
}
struct stat m_reference_file_stat;
TimestampType m_timestamp_type { TimestampType::LastModification };
};
class GidCommand final : public StatCommand {
public:
GidCommand(char const* arg)
{
auto gid_or_error = NumericRange<gid_t>::parse({ arg, strlen(arg) });
if (gid_or_error.is_error())
fatal_error("find: Invalid argument '{}' to '-gid'", arg);
m_gid_range = gid_or_error.release_value();
}
private:
virtual bool evaluate(struct stat const& stat) const override
{
return m_gid_range.contains(stat.st_gid);
}
NumericRange<gid_t> m_gid_range {};
};
class UidCommand final : public StatCommand {
public:
UidCommand(char const* arg)
{
auto uid_or_error = NumericRange<uid_t>::parse({ arg, strlen(arg) });
if (uid_or_error.is_error())
fatal_error("find: Invalid argument '{}' to '-uid'", arg);
m_uid_range = uid_or_error.release_value();
}
private:
virtual bool evaluate(struct stat const& stat) const override
{
return m_uid_range.contains(stat.st_uid);
}
NumericRange<uid_t> m_uid_range {};
};
class PrintCommand final : public Command {
public:
PrintCommand(char terminator = '\n')
: m_terminator(terminator)
{
}
private:
virtual bool evaluate(FileData& file_data) const override
{
auto printed = false;
if (g_print_hyperlinks) {
auto full_path_or_error = FileSystem::real_path(file_data.full_path());
if (!full_path_or_error.is_error()) {
auto fullpath = full_path_or_error.release_value();
auto url = URL::create_with_file_scheme(fullpath.to_deprecated_string());
out("\033]8;;{}\033\\{}{}\033]8;;\033\\", url.serialize(), file_data.full_path(), m_terminator);
printed = true;
}
}
if (!printed)
out("{}{}", file_data.full_path(), m_terminator);
return true;
}
char m_terminator { '\n' };
};
class ExecCommand final : public Command {
public:
enum class AwaitConfirmation {
Yes,
No
};
ExecCommand(Vector<char*>&& argv, AwaitConfirmation await_confirmation = AwaitConfirmation::No)
: m_argv(move(argv))
, m_await_confirmation(await_confirmation)
{
}
private:
virtual bool evaluate(FileData& file_data) const override
{
// Replace any occurrences of "{}" with the path.
Vector<DeprecatedString> full_paths;
Vector<char*> argv = m_argv;
for (auto& arg : argv) {
if (StringView { arg, strlen(arg) } == "{}") {
auto full_path = file_data.full_path();
full_paths.append(full_path);
arg = const_cast<char*>(full_path.characters());
}
}
if (m_await_confirmation == AwaitConfirmation::Yes) {
out(stderr, "\"");
bool first = true;
for (auto const& arg : argv) {
if (!first)
out(stderr, " ");
out(stderr, "{}", arg);
first = false;
}
out(stderr, "\"? ");
fflush(stderr);
Array<u8, BUFSIZ> buffer;
auto nread_or_error = Core::System::read(STDIN_FILENO, buffer);
if (nread_or_error.is_error()) {
warn("Failed to read from stdin: {}", strerror(nread_or_error.error().code()));
return false;
}
auto nread = nread_or_error.release_value();
if (nread <= 0 || (buffer[0] != 'y' && buffer[0] != 'Y'))
return false;
}
pid_t pid = fork();
if (pid < 0) {
perror("fork");
g_there_was_an_error = true;
return false;
} else if (pid == 0) {
argv.append(nullptr);
execvp(m_argv[0], argv.data());
perror("execvp");
exit(1);
} else {
int status;
int rc = waitpid(pid, &status, 0);
if (rc < 0) {
perror("waitpid");
g_there_was_an_error = true;
return false;
}
return WIFEXITED(status) && WEXITSTATUS(status) == 0;
}
}
Vector<char*> m_argv;
AwaitConfirmation m_await_confirmation { AwaitConfirmation::No };
};
class AndCommand final : public Command {
public:
AndCommand(NonnullOwnPtr<Command>&& lhs, NonnullOwnPtr<Command>&& rhs)
: m_lhs(move(lhs))
, m_rhs(move(rhs))
{
}
private:
virtual bool evaluate(FileData& file_data) const override
{
return m_lhs->evaluate(file_data) && m_rhs->evaluate(file_data);
}
NonnullOwnPtr<Command> m_lhs;
NonnullOwnPtr<Command> m_rhs;
};
class OrCommand final : public Command {
public:
OrCommand(NonnullOwnPtr<Command>&& lhs, NonnullOwnPtr<Command>&& rhs)
: m_lhs(move(lhs))
, m_rhs(move(rhs))
{
}
private:
virtual bool evaluate(FileData& file_data) const override
{
return m_lhs->evaluate(file_data) || m_rhs->evaluate(file_data);
}
NonnullOwnPtr<Command> m_lhs;
NonnullOwnPtr<Command> m_rhs;
};
class NotCommand final : public Command {
public:
NotCommand(NonnullOwnPtr<Command>&& operand)
: m_operand(move(operand))
{
}
private:
virtual bool evaluate(FileData& file_data) const override
{
return !m_operand->evaluate(file_data);
}
NonnullOwnPtr<Command> m_operand;
};
static OwnPtr<Command> parse_complex_command(Vector<char*>& args);
// Parse a simple command starting at optind; leave optind at its the last
// argument. Return nullptr if we reach the end of arguments.
static OwnPtr<Command> parse_simple_command(Vector<char*>& args)
{
if (args.is_empty())
return {};
char* raw_arg = args.take_first();
StringView arg { raw_arg, strlen(raw_arg) };
if (arg == "(") {
auto command = parse_complex_command(args);
if (command && !args.is_empty() && StringView { args.first(), strlen(args.first()) } == ")")
return command;
fatal_error("Unmatched \033[1m(");
} else if (arg == "!") {
if (args.is_empty())
fatal_error("Expected an expression after '!'");
auto command = parse_simple_command(args).release_nonnull();
return make<NotCommand>(move(command));
} else if (arg == "-maxdepth"sv) {
if (args.is_empty())
fatal_error("-maxdepth: requires additional arguments");
return make<MaxDepthCommand>(args.take_first());
} else if (arg == "-mindepth"sv) {
if (args.is_empty())
fatal_error("-mindepth: requires additional arguments");
return make<MinDepthCommand>(args.take_first());
} else if (arg == "-type") {
if (args.is_empty())
fatal_error("-type: requires additional arguments");
return make<TypeCommand>(args.take_first());
} else if (arg == "-links") {
if (args.is_empty())
fatal_error("-links: requires additional arguments");
return make<LinksCommand>(args.take_first());
} else if (arg == "-user") {
if (args.is_empty())
fatal_error("-user: requires additional arguments");
return make<UserCommand>(args.take_first());
} else if (arg == "-group") {
if (args.is_empty())
fatal_error("-group: requires additional arguments");
return make<GroupCommand>(args.take_first());
} else if (arg == "-size") {
if (args.is_empty())
fatal_error("-size: requires additional arguments");
return make<SizeCommand>(args.take_first());
} else if (arg == "-empty") {
return make<EmptyCommand>();
} else if (arg == "-name") {
if (args.is_empty())
fatal_error("-name: requires additional arguments");
return make<NameCommand>(args.take_first(), CaseSensitivity::CaseSensitive);
} else if (arg == "-iname") {
if (args.is_empty())
fatal_error("-iname: requires additional arguments");
return make<NameCommand>(args.take_first(), CaseSensitivity::CaseInsensitive);
} else if (arg == "-regex") {
if (args.is_empty())
fatal_error("-regex: requires additional arguments");
Regex<PosixExtended> regex { args.take_first(), PosixFlags::Default };
if (regex.parser_result.error != regex::Error::NoError)
fatal_error("{}", regex.error_string());
return make<RegexCommand>(move(regex));
} else if (arg == "-iregex") {
if (args.is_empty())
fatal_error("-iregex: requires additional arguments");
Regex<PosixExtended> regex { args.take_first(), PosixFlags::Insensitive };
if (regex.parser_result.error != regex::Error::NoError)
fatal_error("{}", regex.error_string());
return make<RegexCommand>(move(regex));
} else if (arg == "-readable") {
return make<AccessCommand>(R_OK);
} else if (arg == "-writable") {
return make<AccessCommand>(W_OK);
} else if (arg == "-executable") {
return make<AccessCommand>(X_OK);
} else if (arg == "-newer") {
if (args.is_empty())
fatal_error("-newer: requires additional arguments");
return make<NewerCommand>(args.take_first(), NewerCommand::TimestampType::LastModification);
} else if (arg == "-anewer") {
if (args.is_empty())
fatal_error("-anewer: requires additional arguments");
return make<NewerCommand>(args.take_first(), NewerCommand::TimestampType::LastAccess);
} else if (arg == "-cnewer") {
if (args.is_empty())
fatal_error("-cnewer: requires additional arguments");
return make<NewerCommand>(args.take_first(), NewerCommand::TimestampType::Creation);
} else if (arg == "-gid") {
if (args.is_empty())
fatal_error("-gid: requires additional arguments");
return make<GidCommand>(args.take_first());
} else if (arg == "-uid") {
if (args.is_empty())
fatal_error("-uid: requires additional arguments");
return make<UidCommand>(args.take_first());
} else if (arg == "-print") {
g_have_seen_action_command = true;
return make<PrintCommand>();
} else if (arg == "-print0") {
g_have_seen_action_command = true;
return make<PrintCommand>(0);
} else if (arg == "-exec" || arg == "-ok") {
if (args.is_empty())
fatal_error("{}: requires additional arguments", arg);
g_have_seen_action_command = true;
Vector<char*> command_argv;
bool terminator_found = false;
while (!args.is_empty()) {
char* next = args.take_first();
if (next[0] == ';' && next[1] == '\0') {
terminator_found = true;
break;
}
command_argv.append(next);
}
if (!terminator_found)
fatal_error("{}: Terminating ';' not found", arg);
auto await_confirmation = (arg == "-ok") ? ExecCommand::AwaitConfirmation::Yes : ExecCommand::AwaitConfirmation::No;
return make<ExecCommand>(move(command_argv), await_confirmation);
} else {
fatal_error("Unsupported command \033[1m{}", arg);
}
}
static OwnPtr<Command> parse_complex_command(Vector<char*>& args)
{
auto command = parse_simple_command(args);
while (command && !args.is_empty()) {
char* raw_arg = args.take_first();
StringView arg { raw_arg, strlen(raw_arg) };
enum {
And,
Or,
} binary_operation { And };
if (arg == "-a") {
binary_operation = And;
} else if (arg == "-o") {
binary_operation = Or;
} else if (arg == ")") {
// Ooops, looked too far.
args.prepend(raw_arg);
return command;
} else {
// Juxtaposition is an And too, and there's nothing to skip.
args.prepend(raw_arg);
binary_operation = And;
}
auto rhs = parse_complex_command(args);
if (!rhs)
fatal_error("Missing right-hand side");
if (binary_operation == And)
command = make<AndCommand>(command.release_nonnull(), rhs.release_nonnull());
else
command = make<OrCommand>(command.release_nonnull(), rhs.release_nonnull());
}
return command;
}
static NonnullOwnPtr<Command> parse_all_commands(Vector<char*>& args)
{
auto command = parse_complex_command(args);
if (g_have_seen_action_command) {
VERIFY(command);
return command.release_nonnull();
}
if (!command) {
return make<PrintCommand>();
}
return make<AndCommand>(command.release_nonnull(), make<PrintCommand>());
}
static void walk_tree(FileData& root_data, Command& command, u32 depth = 0)
{
if (!g_min_depth.has_value() || g_min_depth.value() <= depth)
command.evaluate(root_data);
// We should try to read directory entries if either:
// * This is a directory.
// * This is a symlink (that could point to a directory),
// and we're following symlinks.
// * The type is unknown, so it could be a directory.
switch (root_data.d_type) {
case DT_DIR:
case DT_UNKNOWN:
break;
case DT_LNK:
if (g_follow_symlinks)
break;
return;
default:
return;
}
int dirfd = openat(root_data.dirfd, root_data.basename, O_RDONLY | O_DIRECTORY | O_CLOEXEC);
if (dirfd < 0) {
if (errno == ENOTDIR) {
// Above we decided to try to open this file because it could
// be a directory, but turns out it's not. This is fine though.
return;
}
perror(root_data.full_path().characters());
g_there_was_an_error = true;
return;
}
DIR* dir = fdopendir(dirfd);
while (true) {
errno = 0;
auto* dirent = readdir(dir);
if (!dirent)
break;
if (strcmp(dirent->d_name, ".") == 0 || strcmp(dirent->d_name, "..") == 0)
continue;
FileData file_data {
root_data.root_path,
root_data.relative_path.append({ dirent->d_name, strlen(dirent->d_name) }),
dirfd,
dirent->d_name,
(struct stat) {},
false,
dirent->d_type,
};
bool should_increase_depth = false;
if (g_max_depth.has_value() || g_min_depth.has_value()) {
if (g_max_depth.has_value() && depth >= g_max_depth.value())
return;
if (file_data.d_type == DT_UNKNOWN)
file_data.ensure_stat();
if (file_data.d_type == DT_DIR)
should_increase_depth = true;
}
walk_tree(file_data, command, should_increase_depth ? depth + 1 : depth);
}
if (errno != 0) {
perror(root_data.full_path().characters());
g_there_was_an_error = true;
}
closedir(dir);
}
ErrorOr<int> serenity_main(Main::Arguments arguments)
{
Vector<char*> args;
args.append(arguments.argv + 1, arguments.argc - 1);
OwnPtr<Command> command;
Vector<StringView> paths;
while (!args.is_empty()) {
char* raw_arg = args.take_first();
StringView arg { raw_arg, strlen(raw_arg) };
if (arg == "-L") {
g_follow_symlinks = true;
} else if (arg.starts_with('-') || arg == "!"sv) {
args.prepend(raw_arg);
command = parse_all_commands(args);
} else {
paths.append(arg);
}
}
g_print_hyperlinks = TRY(Core::System::isatty(STDOUT_FILENO));
if (!command)
command = make<PrintCommand>();
if (paths.is_empty())
paths.append("."sv);
for (auto& path : paths) {
LexicalPath lexical_path { path };
DeprecatedString dirname = lexical_path.dirname();
DeprecatedString basename = lexical_path.basename();
int dirfd = TRY(Core::System::open(dirname, O_RDONLY | O_DIRECTORY | O_CLOEXEC));
FileData file_data {
path,
LexicalPath { "" },
dirfd,
basename.characters(),
(struct stat) {},
false,
DT_UNKNOWN,
};
walk_tree(file_data, *command);
close(dirfd);
}
return g_there_was_an_error ? 1 : 0;
}