567 lines
16 KiB
C++
567 lines
16 KiB
C++
#include "file.hh"
|
|
|
|
#include "assert.hh"
|
|
#include "buffer.hh"
|
|
#include "exception.hh"
|
|
#include "flags.hh"
|
|
#include "ranked_match.hh"
|
|
#include "regex.hh"
|
|
#include "string.hh"
|
|
#include "unicode.hh"
|
|
|
|
#include <cerrno>
|
|
#include <cstring>
|
|
#include <sys/mman.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <dirent.h>
|
|
#include <cstdlib>
|
|
#include <sys/select.h>
|
|
|
|
#if defined(__FreeBSD__)
|
|
#include <sys/sysctl.h>
|
|
#endif
|
|
|
|
#if defined(__APPLE__)
|
|
#include <mach-o/dyld.h>
|
|
#define st_mtim st_mtimespec
|
|
#endif
|
|
|
|
#if defined(__HAIKU__)
|
|
#include <app/Application.h>
|
|
#include <app/Roster.h>
|
|
#include <storage/Path.h>
|
|
#endif
|
|
|
|
namespace Kakoune
|
|
{
|
|
|
|
struct file_access_error : runtime_error
|
|
{
|
|
public:
|
|
file_access_error(StringView filename,
|
|
StringView error_desc)
|
|
: runtime_error(format("{}: {}", filename, error_desc)) {}
|
|
|
|
file_access_error(int fd, StringView error_desc)
|
|
: runtime_error(format("fd {}: {}", fd, error_desc)) {}
|
|
};
|
|
|
|
String parse_filename(StringView filename)
|
|
{
|
|
auto prefix = filename.substr(0_byte, 2_byte);
|
|
if (prefix == "~" or prefix == "~/")
|
|
return getenv("HOME") + filename.substr(1_byte);
|
|
return filename.str();
|
|
}
|
|
|
|
std::pair<StringView, StringView> split_path(StringView path)
|
|
{
|
|
auto it = find(path | reverse(), '/');
|
|
if (it == path.rend())
|
|
return { {}, path };
|
|
const char* slash = it.base()-1;
|
|
return { {path.begin(), slash+1}, {slash+1, path.end()} };
|
|
}
|
|
|
|
String real_path(StringView filename)
|
|
{
|
|
char buffer[PATH_MAX+1];
|
|
|
|
StringView existing = filename;
|
|
StringView non_existing{};
|
|
|
|
while (true)
|
|
{
|
|
char* res = realpath(existing.zstr(), buffer);
|
|
if (res)
|
|
{
|
|
if (non_existing.empty())
|
|
return res;
|
|
return format("{}/{}", res, non_existing);
|
|
}
|
|
|
|
auto it = find(existing.rbegin(), existing.rend(), '/');
|
|
if (it == existing.rend())
|
|
{
|
|
char cwd[1024];
|
|
return format("{}/{}", getcwd(cwd, 1024), filename);
|
|
}
|
|
|
|
existing = StringView{existing.begin(), it.base()-1};
|
|
non_existing = StringView{it.base(), filename.end()};
|
|
}
|
|
}
|
|
|
|
String compact_path(StringView filename)
|
|
{
|
|
String real_filename = real_path(filename);
|
|
|
|
char cwd[1024];
|
|
if (!::getcwd(cwd, 1024))
|
|
throw runtime_error(format("unable to get the current working directory (errno: {})", ::strerror(errno)));
|
|
|
|
String real_cwd = real_path(cwd) + "/";
|
|
if (prefix_match(real_filename, real_cwd))
|
|
return real_filename.substr(real_cwd.length()).str();
|
|
|
|
const char* home = getenv("HOME");
|
|
if (home)
|
|
{
|
|
ByteCount home_len = (int)strlen(home);
|
|
if (real_filename.substr(0, home_len) == home)
|
|
return "~" + real_filename.substr(home_len);
|
|
}
|
|
|
|
return filename.str();
|
|
}
|
|
|
|
StringView tmpdir()
|
|
{
|
|
StringView tmpdir = getenv("TMPDIR");
|
|
if (not tmpdir.empty())
|
|
return tmpdir.back() == '/' ? tmpdir.substr(0_byte, tmpdir.length()-1)
|
|
: tmpdir;
|
|
return "/tmp";
|
|
}
|
|
|
|
bool fd_readable(int fd)
|
|
{
|
|
fd_set rfds;
|
|
FD_ZERO(&rfds);
|
|
FD_SET(fd, &rfds);
|
|
|
|
timeval tv{0,0};
|
|
return select(fd+1, &rfds, nullptr, nullptr, &tv) == 1;
|
|
}
|
|
|
|
bool fd_writable(int fd)
|
|
{
|
|
fd_set wfds;
|
|
FD_ZERO(&wfds);
|
|
FD_SET(fd, &wfds);
|
|
|
|
timeval tv{0,0};
|
|
return select(fd+1, nullptr, &wfds, nullptr, &tv) == 1;
|
|
}
|
|
|
|
String read_fd(int fd, bool text)
|
|
{
|
|
String content;
|
|
constexpr size_t bufsize = 256;
|
|
char buf[bufsize];
|
|
while (ssize_t size = read(fd, buf, bufsize))
|
|
{
|
|
if (size == -1)
|
|
throw file_access_error{fd, strerror(errno)};
|
|
|
|
if (text)
|
|
{
|
|
ssize_t beg = 0;
|
|
for (ssize_t pos = 0; pos < size; ++pos)
|
|
{
|
|
if (buf[pos] == '\r')
|
|
{
|
|
content += StringView{buf + beg, buf + pos};
|
|
beg = pos + 1;
|
|
}
|
|
}
|
|
content += StringView{buf + beg, buf + size};
|
|
}
|
|
else
|
|
content += StringView{buf, buf + size};
|
|
}
|
|
return content;
|
|
}
|
|
|
|
String read_file(StringView filename, bool text)
|
|
{
|
|
int fd = open(filename.zstr(), O_RDONLY);
|
|
if (fd == -1)
|
|
throw file_access_error(filename, strerror(errno));
|
|
|
|
auto close_fd = on_scope_end([fd]{ close(fd); });
|
|
return read_fd(fd, text);
|
|
}
|
|
|
|
MappedFile::MappedFile(StringView filename)
|
|
: data{nullptr}
|
|
{
|
|
fd = open(filename.zstr(), O_RDONLY | O_NONBLOCK);
|
|
if (fd == -1)
|
|
throw file_access_error(filename, strerror(errno));
|
|
|
|
fstat(fd, &st);
|
|
if (S_ISDIR(st.st_mode))
|
|
throw file_access_error(filename, "is a directory");
|
|
|
|
if (st.st_size == 0)
|
|
return;
|
|
|
|
data = (const char*)mmap(nullptr, st.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
|
|
if (data == MAP_FAILED)
|
|
throw file_access_error{filename, strerror(errno)};
|
|
|
|
if (st.st_size > std::numeric_limits<int>::max())
|
|
throw runtime_error("file is too big");
|
|
}
|
|
|
|
MappedFile::~MappedFile()
|
|
{
|
|
if (fd != -1)
|
|
{
|
|
if (data != nullptr)
|
|
munmap((void*)data, st.st_size);
|
|
close(fd);
|
|
}
|
|
}
|
|
|
|
MappedFile::operator StringView() const
|
|
{
|
|
return { data, (int)st.st_size };
|
|
}
|
|
|
|
bool file_exists(StringView filename)
|
|
{
|
|
struct stat st;
|
|
return stat(filename.zstr(), &st) == 0;
|
|
}
|
|
|
|
void write(int fd, StringView data)
|
|
{
|
|
const char* ptr = data.data();
|
|
ssize_t count = (int)data.length();
|
|
|
|
while (count)
|
|
{
|
|
ssize_t written = ::write(fd, ptr, count);
|
|
ptr += written;
|
|
count -= written;
|
|
|
|
if (written == -1)
|
|
throw file_access_error(format("fd: {}", fd), strerror(errno));
|
|
}
|
|
}
|
|
|
|
void write_buffer_to_fd(Buffer& buffer, int fd)
|
|
{
|
|
auto eolformat = buffer.options()["eolformat"].get<EolFormat>();
|
|
StringView eoldata;
|
|
if (eolformat == EolFormat::Crlf)
|
|
eoldata = "\r\n";
|
|
else
|
|
eoldata = "\n";
|
|
|
|
if (buffer.options()["BOM"].get<ByteOrderMark>() == ByteOrderMark::Utf8)
|
|
if (::write(fd, "\xEF\xBB\xBF", 3) < 0)
|
|
throw runtime_error(format("unable to write data to the buffer (fd: {}; errno: {})", fd, ::strerror(errno)));
|
|
|
|
for (LineCount i = 0; i < buffer.line_count(); ++i)
|
|
{
|
|
// end of lines are written according to eolformat but always
|
|
// stored as \n
|
|
StringView linedata = buffer[i];
|
|
write(fd, linedata.substr(0, linedata.length()-1));
|
|
write(fd, eoldata);
|
|
}
|
|
}
|
|
|
|
void write_buffer_to_file(Buffer& buffer, StringView filename, bool force)
|
|
{
|
|
struct stat st;
|
|
auto zfilename = filename.zstr();
|
|
|
|
if (force)
|
|
{
|
|
if (::stat(zfilename, &st) == 0)
|
|
{
|
|
if (::chmod(zfilename, st.st_mode | S_IWUSR) < 0)
|
|
throw runtime_error("couldn't change file permissions");
|
|
}
|
|
else
|
|
force = false;
|
|
}
|
|
auto restore_mode = on_scope_end([&]{
|
|
if (force and ::chmod(zfilename, st.st_mode) < 0)
|
|
throw runtime_error("couldn't restore file permissions");
|
|
});
|
|
|
|
int fd = open(zfilename, O_CREAT | O_WRONLY | O_TRUNC, 0644);
|
|
if (fd == -1)
|
|
throw file_access_error(filename, strerror(errno));
|
|
|
|
{
|
|
auto close_fd = on_scope_end([fd]{ close(fd); });
|
|
write_buffer_to_fd(buffer, fd);
|
|
}
|
|
|
|
if ((buffer.flags() & Buffer::Flags::File) and
|
|
real_path(filename) == real_path(buffer.name()))
|
|
buffer.notify_saved();
|
|
}
|
|
|
|
void write_buffer_to_backup_file(Buffer& buffer)
|
|
{
|
|
String path = real_path(buffer.name());
|
|
StringView dir, file;
|
|
std::tie(dir,file) = split_path(path);
|
|
|
|
char pattern[PATH_MAX];
|
|
if (dir.empty())
|
|
format_to(pattern, ".{}.kak.XXXXXX", file);
|
|
else
|
|
format_to(pattern, "{}/.{}.kak.XXXXXX", dir, file);
|
|
|
|
int fd = mkstemp(pattern);
|
|
if (fd >= 0)
|
|
{
|
|
write_buffer_to_fd(buffer, fd);
|
|
close(fd);
|
|
}
|
|
}
|
|
|
|
String find_file(StringView filename, ConstArrayView<String> paths)
|
|
{
|
|
struct stat buf;
|
|
if (filename.substr(0_byte, 1_byte) == "/")
|
|
{
|
|
if (stat(filename.zstr(), &buf) == 0 and S_ISREG(buf.st_mode))
|
|
return filename.str();
|
|
return "";
|
|
}
|
|
if (filename.substr(0_byte, 2_byte) == "~/")
|
|
{
|
|
String candidate = getenv("HOME") + filename.substr(1_byte);
|
|
if (stat(candidate.c_str(), &buf) == 0 and S_ISREG(buf.st_mode))
|
|
return candidate;
|
|
return "";
|
|
}
|
|
|
|
for (auto candidate : paths | transform(parse_filename))
|
|
{
|
|
if (not candidate.empty() and candidate.back() != '/')
|
|
candidate += '/';
|
|
candidate += filename;
|
|
if (stat(candidate.c_str(), &buf) == 0 and S_ISREG(buf.st_mode))
|
|
return candidate;
|
|
}
|
|
return "";
|
|
}
|
|
|
|
void make_directory(StringView dir, mode_t mode)
|
|
{
|
|
auto it = dir.begin(), end = dir.end();
|
|
while(it != end)
|
|
{
|
|
it = std::find(it+1, end, '/');
|
|
struct stat st;
|
|
StringView dirname{dir.begin(), it};
|
|
if (stat(dirname.zstr(), &st) == 0)
|
|
{
|
|
if (not S_ISDIR(st.st_mode))
|
|
throw runtime_error(format("Cannot make directory, '{}' exists but is not a directory", dirname));
|
|
}
|
|
else
|
|
{
|
|
auto old_mask = umask(0);
|
|
auto restore_mask = on_scope_end([old_mask]() { umask(old_mask); });
|
|
|
|
if (mkdir(dirname.zstr(), mode) != 0)
|
|
throw runtime_error(format("mkdir failed for directory '{}' errno {}", dirname, errno));
|
|
}
|
|
}
|
|
}
|
|
|
|
template<typename Filter>
|
|
Vector<String> list_files(StringView dirname, Filter filter)
|
|
{
|
|
char buffer[PATH_MAX+1];
|
|
format_to(buffer, "{}", dirname);
|
|
DIR* dir = opendir(dirname.empty() ? "./" : buffer);
|
|
if (not dir)
|
|
return {};
|
|
|
|
auto close_dir = on_scope_end([dir]{ closedir(dir); });
|
|
|
|
Vector<String> result;
|
|
while (dirent* entry = readdir(dir))
|
|
{
|
|
StringView filename = entry->d_name;
|
|
if (filename.empty())
|
|
continue;
|
|
|
|
struct stat st;
|
|
auto fmt_str = (dirname.empty() or dirname.back() == '/') ? "{}{}" : "{}/{}";
|
|
format_to(buffer, fmt_str, dirname, filename);
|
|
if (stat(buffer, &st) != 0 or not filter(*entry, st))
|
|
continue;
|
|
|
|
if (S_ISDIR(st.st_mode))
|
|
filename = format_to(buffer, "{}/", filename);
|
|
result.push_back(filename.str());
|
|
}
|
|
return result;
|
|
}
|
|
|
|
Vector<String> list_files(StringView directory)
|
|
{
|
|
return list_files(directory, [](const dirent& entry, const struct stat&) {
|
|
return StringView{entry.d_name}.substr(0_byte, 1_byte) != ".";
|
|
});
|
|
}
|
|
|
|
static CandidateList candidates(ConstArrayView<RankedMatch> matches, StringView dirname)
|
|
{
|
|
CandidateList res;
|
|
res.reserve(matches.size());
|
|
for (auto& match : matches)
|
|
res.push_back(dirname + match.candidate());
|
|
return res;
|
|
}
|
|
|
|
CandidateList complete_filename(StringView prefix, const Regex& ignored_regex,
|
|
ByteCount cursor_pos, FilenameFlags flags)
|
|
{
|
|
prefix = prefix.substr(0, cursor_pos);
|
|
StringView dirname, fileprefix;
|
|
std::tie(dirname, fileprefix) = split_path(prefix);
|
|
auto parsed_dirname = parse_filename(dirname);
|
|
|
|
const bool check_ignored_regex = not ignored_regex.empty() and
|
|
not regex_match(fileprefix.begin(), fileprefix.end(), ignored_regex);
|
|
const bool only_dirs = (flags & FilenameFlags::OnlyDirectories);
|
|
|
|
auto filter = [&ignored_regex, check_ignored_regex, only_dirs](const dirent& entry, struct stat& st)
|
|
{
|
|
StringView name{entry.d_name};
|
|
return (not check_ignored_regex or not regex_match(name.begin(), name.end(), ignored_regex)) and
|
|
(not only_dirs or S_ISDIR(st.st_mode));
|
|
};
|
|
auto files = list_files(parsed_dirname, filter);
|
|
Vector<RankedMatch> matches;
|
|
for (auto& file : files)
|
|
{
|
|
if (RankedMatch match{file, fileprefix})
|
|
matches.push_back(match);
|
|
}
|
|
std::sort(matches.begin(), matches.end());
|
|
const bool expand = (flags & FilenameFlags::Expand);
|
|
return candidates(matches, expand ? parsed_dirname : dirname);
|
|
}
|
|
|
|
CandidateList complete_command(StringView prefix, ByteCount cursor_pos)
|
|
{
|
|
String real_prefix = parse_filename(prefix.substr(0, cursor_pos));
|
|
StringView dirname, fileprefix;
|
|
std::tie(dirname, fileprefix) = split_path(real_prefix);
|
|
|
|
if (not dirname.empty())
|
|
{
|
|
auto filter = [&dirname](const dirent& entry, const struct stat& st)
|
|
{
|
|
bool executable = (st.st_mode & S_IXUSR)
|
|
| (st.st_mode & S_IXGRP)
|
|
| (st.st_mode & S_IXOTH);
|
|
return S_ISDIR(st.st_mode) or (S_ISREG(st.st_mode) and executable);
|
|
};
|
|
auto files = list_files(dirname, filter);
|
|
Vector<RankedMatch> matches;
|
|
for (auto& file : files)
|
|
{
|
|
if (RankedMatch match{file, real_prefix})
|
|
matches.push_back(match);
|
|
}
|
|
std::sort(matches.begin(), matches.end());
|
|
return candidates(matches, dirname);
|
|
}
|
|
|
|
using TimeSpec = decltype(stat::st_mtim);
|
|
|
|
struct CommandCache
|
|
{
|
|
TimeSpec mtim = {};
|
|
Vector<String> commands;
|
|
};
|
|
static HashMap<String, CommandCache, MemoryDomain::Commands> command_cache;
|
|
|
|
Vector<RankedMatch> matches;
|
|
for (auto dir : StringView{getenv("PATH")} | split<StringView>(':'))
|
|
{
|
|
auto dirname = ((not dir.empty() and dir.back() == '/') ? dir.substr(0, dir.length()-1) : dir).str();
|
|
|
|
struct stat st;
|
|
if (stat(dirname.c_str(), &st))
|
|
continue;
|
|
|
|
auto& cache = command_cache[dirname];
|
|
if (memcmp(&cache.mtim, &st.st_mtim, sizeof(TimeSpec)) != 0)
|
|
{
|
|
auto filter = [&dirname](const dirent& entry, const struct stat& st) {
|
|
bool executable = (st.st_mode & S_IXUSR)
|
|
| (st.st_mode & S_IXGRP)
|
|
| (st.st_mode & S_IXOTH);
|
|
return S_ISREG(st.st_mode) and executable;
|
|
};
|
|
|
|
cache.commands = list_files(dirname, filter);
|
|
memcpy(&cache.mtim, &st.st_mtim, sizeof(TimeSpec));
|
|
}
|
|
for (auto& cmd : cache.commands)
|
|
{
|
|
if (RankedMatch match{cmd, fileprefix})
|
|
matches.push_back(match);
|
|
}
|
|
}
|
|
std::sort(matches.begin(), matches.end());
|
|
auto it = std::unique(matches.begin(), matches.end());
|
|
matches.erase(it, matches.end());
|
|
return candidates(matches, "");
|
|
}
|
|
|
|
timespec get_fs_timestamp(StringView filename)
|
|
{
|
|
struct stat st;
|
|
if (stat(filename.zstr(), &st) != 0)
|
|
return InvalidTime;
|
|
return st.st_mtim;
|
|
}
|
|
|
|
String get_kak_binary_path()
|
|
{
|
|
char buffer[2048];
|
|
#if defined(__linux__) or defined(__CYGWIN__)
|
|
ssize_t res = readlink("/proc/self/exe", buffer, 2048);
|
|
kak_assert(res != -1);
|
|
buffer[res] = '\0';
|
|
return buffer;
|
|
#elif defined(__FreeBSD__)
|
|
int mib[] = {CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, -1};
|
|
size_t res = sizeof(buffer);
|
|
sysctl(mib, 4, buffer, &res, NULL, 0);
|
|
return buffer;
|
|
#elif defined(__APPLE__)
|
|
uint32_t bufsize = 2048;
|
|
_NSGetExecutablePath(buffer, &bufsize);
|
|
char* canonical_path = realpath(buffer, nullptr);
|
|
String path = canonical_path;
|
|
free(canonical_path);
|
|
return path;
|
|
#elif defined(__HAIKU__)
|
|
BApplication app("application/x-vnd.kakoune");
|
|
app_info info;
|
|
status_t status = app.GetAppInfo(&info);
|
|
kak_assert(status == B_OK);
|
|
BPath path(&info.ref);
|
|
return path.Path();
|
|
#elif defined(__DragonFly__)
|
|
ssize_t res = readlink("/proc/curproc/file", buffer, 2048);
|
|
kak_assert(res != -1);
|
|
buffer[res] = '\0';
|
|
return buffer;
|
|
#else
|
|
# error "finding executable path is not implemented on this platform"
|
|
#endif
|
|
}
|
|
|
|
}
|