mirror of
https://github.com/zeek/zeek.git
synced 2025-10-07 17:18:20 +00:00
f5a76c1aed
This largely copies over Spicy's `.clang-format` configuration file. The one place where we deviate is header include order since Zeek depends on headers being included in a certain order.
143 lines
3.4 KiB
C++
143 lines
3.4 KiB
C++
// See the file "COPYING" in the main distribution directory for copyright.
|
|
|
|
#include "zeek/Pipe.h"
|
|
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <cerrno>
|
|
#include <cstdio>
|
|
|
|
#include "zeek/Reporter.h"
|
|
|
|
namespace zeek::detail {
|
|
|
|
static void pipe_fail(int eno) {
|
|
char tmp[256];
|
|
zeek::util::zeek_strerror_r(eno, tmp, sizeof(tmp));
|
|
|
|
if ( reporter )
|
|
reporter->FatalError("Pipe failure: %s", tmp);
|
|
else
|
|
fprintf(stderr, "Pipe failure: %s", tmp);
|
|
}
|
|
|
|
static int set_flags(int fd, int flags) {
|
|
auto rval = fcntl(fd, F_GETFD);
|
|
|
|
if ( rval == -1 )
|
|
pipe_fail(errno);
|
|
|
|
if ( flags ) {
|
|
rval |= flags;
|
|
|
|
if ( fcntl(fd, F_SETFD, rval) == -1 )
|
|
pipe_fail(errno);
|
|
}
|
|
|
|
return rval;
|
|
}
|
|
|
|
static int unset_flags(int fd, int flags) {
|
|
auto rval = fcntl(fd, F_GETFD);
|
|
|
|
if ( rval == -1 )
|
|
pipe_fail(errno);
|
|
|
|
if ( flags ) {
|
|
rval &= ~flags;
|
|
|
|
if ( fcntl(fd, F_SETFD, rval) == -1 )
|
|
pipe_fail(errno);
|
|
}
|
|
|
|
return rval;
|
|
}
|
|
|
|
static int set_status_flags(int fd, int flags) {
|
|
auto rval = fcntl(fd, F_GETFL);
|
|
|
|
if ( rval == -1 )
|
|
pipe_fail(errno);
|
|
|
|
if ( flags ) {
|
|
rval |= flags;
|
|
|
|
if ( fcntl(fd, F_SETFL, rval) == -1 )
|
|
pipe_fail(errno);
|
|
}
|
|
|
|
return rval;
|
|
}
|
|
|
|
static int dup_or_fail(int fd, int flags, int status_flags) {
|
|
int rval = dup(fd);
|
|
|
|
if ( rval < 0 )
|
|
pipe_fail(errno);
|
|
|
|
set_flags(fd, flags);
|
|
set_status_flags(fd, status_flags);
|
|
return rval;
|
|
}
|
|
|
|
Pipe::Pipe(int flags0, int flags1, int status_flags0, int status_flags1, int* arg_fds) {
|
|
if ( arg_fds ) {
|
|
fds[0] = arg_fds[0];
|
|
fds[1] = arg_fds[1];
|
|
}
|
|
else {
|
|
// pipe2 can set flags atomically, but not yet available everywhere.
|
|
if ( ::pipe(fds) )
|
|
pipe_fail(errno);
|
|
}
|
|
|
|
flags[0] = set_flags(fds[0], flags0);
|
|
flags[1] = set_flags(fds[1], flags1);
|
|
status_flags[0] = set_status_flags(fds[0], status_flags0);
|
|
status_flags[1] = set_status_flags(fds[1], status_flags1);
|
|
}
|
|
|
|
void Pipe::SetFlags(int arg_flags) {
|
|
flags[0] = set_flags(fds[0], arg_flags);
|
|
flags[1] = set_flags(fds[1], arg_flags);
|
|
}
|
|
|
|
void Pipe::UnsetFlags(int arg_flags) {
|
|
flags[0] = unset_flags(fds[0], arg_flags);
|
|
flags[1] = unset_flags(fds[1], arg_flags);
|
|
}
|
|
|
|
Pipe::~Pipe() {
|
|
close(fds[0]);
|
|
close(fds[1]);
|
|
}
|
|
|
|
Pipe::Pipe(const Pipe& other) {
|
|
fds[0] = dup_or_fail(other.fds[0], other.flags[0], other.status_flags[0]);
|
|
fds[1] = dup_or_fail(other.fds[1], other.flags[1], other.status_flags[1]);
|
|
flags[0] = other.flags[0];
|
|
flags[1] = other.flags[1];
|
|
status_flags[0] = other.status_flags[0];
|
|
status_flags[1] = other.status_flags[1];
|
|
}
|
|
|
|
Pipe& Pipe::operator=(const Pipe& other) {
|
|
if ( this == &other )
|
|
return *this;
|
|
|
|
close(fds[0]);
|
|
close(fds[1]);
|
|
fds[0] = dup_or_fail(other.fds[0], other.flags[0], other.status_flags[0]);
|
|
fds[1] = dup_or_fail(other.fds[1], other.flags[1], other.status_flags[1]);
|
|
flags[0] = other.flags[0];
|
|
flags[1] = other.flags[1];
|
|
status_flags[0] = other.status_flags[0];
|
|
status_flags[1] = other.status_flags[1];
|
|
return *this;
|
|
}
|
|
|
|
PipePair::PipePair(int flags, int status_flags, int* fds)
|
|
: pipes{Pipe(flags, flags, status_flags, status_flags, fds ? fds + 0 : nullptr),
|
|
Pipe(flags, flags, status_flags, status_flags, fds ? fds + 2 : nullptr)} {}
|
|
|
|
} // namespace zeek::detail
|