mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-11-22 23:50:19 +00:00
968 lines
28 KiB
C++
968 lines
28 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 {
|
|
// Full path to the file; either absolute or relative to cwd.
|
|
LexicalPath full_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 };
|
|
|
|
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.string().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.string(), 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.full_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.string();
|
|
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.string(), 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.string());
|
|
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:
|
|
ExecCommand(Vector<char*>&& argv)
|
|
: m_argv(move(argv))
|
|
{
|
|
}
|
|
|
|
private:
|
|
virtual bool evaluate(FileData& file_data) const override
|
|
{
|
|
pid_t pid = fork();
|
|
|
|
if (pid < 0) {
|
|
perror("fork");
|
|
g_there_was_an_error = true;
|
|
return false;
|
|
} else if (pid == 0) {
|
|
// Replace any occurrences of "{}" with the path. Since we're in the
|
|
// child and going to exec real soon, let's just const_cast away the
|
|
// constness.
|
|
auto argv = const_cast<Vector<char*>&>(m_argv);
|
|
for (auto& arg : argv) {
|
|
if (StringView { arg, strlen(arg) } == "{}")
|
|
arg = const_cast<char*>(file_data.full_path.string().characters());
|
|
}
|
|
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;
|
|
};
|
|
|
|
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") {
|
|
if (args.is_empty())
|
|
fatal_error("-exec: requires additional arguments");
|
|
g_have_seen_action_command = true;
|
|
Vector<char*> command_argv;
|
|
while (!args.is_empty()) {
|
|
char* next = args.take_first();
|
|
if (next[0] == ';')
|
|
break;
|
|
command_argv.append(next);
|
|
}
|
|
return make<ExecCommand>(move(command_argv));
|
|
} 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.string().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.full_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.string().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<LexicalPath> 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(LexicalPath(arg));
|
|
}
|
|
}
|
|
|
|
g_print_hyperlinks = TRY(Core::System::isatty(STDOUT_FILENO));
|
|
|
|
if (!command)
|
|
command = make<PrintCommand>();
|
|
|
|
if (paths.is_empty())
|
|
paths.append(LexicalPath("."));
|
|
|
|
for (auto& path : paths) {
|
|
DeprecatedString dirname = path.dirname();
|
|
DeprecatedString basename = path.basename();
|
|
|
|
int dirfd = TRY(Core::System::open(dirname, O_RDONLY | O_DIRECTORY | O_CLOEXEC));
|
|
|
|
FileData file_data {
|
|
path,
|
|
dirfd,
|
|
basename.characters(),
|
|
(struct stat) {},
|
|
false,
|
|
DT_UNKNOWN,
|
|
};
|
|
walk_tree(file_data, *command);
|
|
close(dirfd);
|
|
}
|
|
|
|
return g_there_was_an_error ? 1 : 0;
|
|
}
|