mirror of
https://github.com/fish-shell/fish-shell.git
synced 2026-06-20 22:21:16 -03:00
33aff87c10
Prior to this fix, a file redirection was turned into an io_file_t. This is annoying because every place where we want to apply the redirection, we might fail due to open() failing. Switch to opening the file at the point we resolve the redirection spec. This will simplify a lot of code.
101 lines
3.0 KiB
C++
101 lines
3.0 KiB
C++
#include "config.h" // IWYU pragma: keep
|
|
|
|
#include "redirection.h"
|
|
|
|
#include <fcntl.h>
|
|
|
|
#include "io.h"
|
|
#include "wutil.h"
|
|
|
|
#define NOCLOB_ERROR _(L"The file '%ls' already exists")
|
|
|
|
#define FILE_ERROR _(L"An error occurred while redirecting file '%ls'")
|
|
|
|
/// Base open mode to pass to calls to open.
|
|
#define OPEN_MASK 0666
|
|
|
|
dup2_list_t::~dup2_list_t() = default;
|
|
|
|
maybe_t<int> redirection_spec_t::get_target_as_fd() const {
|
|
errno = 0;
|
|
int result = fish_wcstoi(target.c_str());
|
|
if (errno || result < 0) return none();
|
|
return result;
|
|
}
|
|
|
|
int redirection_spec_t::oflags() const {
|
|
switch (mode) {
|
|
case redirection_mode_t::append:
|
|
return O_CREAT | O_APPEND | O_WRONLY;
|
|
case redirection_mode_t::overwrite:
|
|
return O_CREAT | O_WRONLY | O_TRUNC;
|
|
case redirection_mode_t::noclob:
|
|
return O_CREAT | O_EXCL | O_WRONLY;
|
|
case redirection_mode_t::input:
|
|
return O_RDONLY;
|
|
case redirection_mode_t::fd:
|
|
DIE("Not a file redirection");
|
|
}
|
|
}
|
|
|
|
maybe_t<dup2_list_t> dup2_list_t::resolve_chain(const io_chain_t &io_chain) {
|
|
ASSERT_IS_NOT_FORKED_CHILD();
|
|
dup2_list_t result;
|
|
for (const auto &io_ref : io_chain) {
|
|
switch (io_ref->io_mode) {
|
|
case io_mode_t::file: {
|
|
const io_file_t *io = static_cast<const io_file_t *>(io_ref.get());
|
|
result.add_dup2(io->file_fd(), io->fd);
|
|
break;
|
|
}
|
|
|
|
case io_mode_t::close: {
|
|
const io_close_t *io = static_cast<const io_close_t *>(io_ref.get());
|
|
result.add_close(io->fd);
|
|
break;
|
|
}
|
|
|
|
case io_mode_t::fd: {
|
|
const io_fd_t *io = static_cast<const io_fd_t *>(io_ref.get());
|
|
result.add_dup2(io->old_fd, io->fd);
|
|
break;
|
|
}
|
|
|
|
case io_mode_t::pipe: {
|
|
const io_pipe_t *io = static_cast<const io_pipe_t *>(io_ref.get());
|
|
result.add_dup2(io->pipe_fd(), io->fd);
|
|
result.add_close(io->pipe_fd());
|
|
break;
|
|
}
|
|
|
|
case io_mode_t::bufferfill: {
|
|
const io_bufferfill_t *io = static_cast<const io_bufferfill_t *>(io_ref.get());
|
|
result.add_dup2(io->write_fd(), io->fd);
|
|
result.add_close(io->write_fd());
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return {std::move(result)};
|
|
}
|
|
|
|
int dup2_list_t::fd_for_target_fd(int target) const {
|
|
// Paranoia.
|
|
if (target < 0) {
|
|
return target;
|
|
}
|
|
// Note we can simply walk our action list backwards, looking for src -> target dups.
|
|
int cursor = target;
|
|
for (auto iter = actions_.rbegin(); iter != actions_.rend(); ++iter) {
|
|
if (iter->target == cursor) {
|
|
// cursor is replaced by iter->src
|
|
cursor = iter->src;
|
|
} else if (iter->src == cursor && iter->target < 0) {
|
|
// cursor is closed.
|
|
cursor = -1;
|
|
break;
|
|
}
|
|
}
|
|
return cursor;
|
|
}
|