mirror of https://github.com/tstack/lnav
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.1 KiB
C++
66 lines
1.1 KiB
C++
|
|
#include <errno.h>
|
|
#include <paths.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <signal.h>
|
|
#include <unistd.h>
|
|
|
|
#include "piper_proc.hh"
|
|
|
|
using namespace std;
|
|
|
|
piper_proc::piper_proc(int pipefd)
|
|
: pp_fd(-1), pp_child(-1)
|
|
{
|
|
char piper_tmpname[PATH_MAX];
|
|
const char *tmpdir;
|
|
|
|
if ((tmpdir = getenv("TMPDIR")) == NULL) {
|
|
tmpdir = _PATH_VARTMP;
|
|
}
|
|
snprintf(piper_tmpname, sizeof(piper_tmpname),
|
|
"%s/lnav.piper.XXXXXX",
|
|
tmpdir);
|
|
if ((this->pp_fd = mkstemp(piper_tmpname)) == -1) {
|
|
throw error(errno);
|
|
}
|
|
|
|
unlink(piper_tmpname);
|
|
this->pp_child = fork();
|
|
switch (this->pp_child) {
|
|
case -1:
|
|
throw error(errno);
|
|
case 0:
|
|
{
|
|
char buffer[4096];
|
|
off_t off = 0;
|
|
int rc;
|
|
|
|
while ((rc = read(pipefd, buffer, sizeof(buffer))) > 0) {
|
|
int wrc = pwrite(this->pp_fd, buffer, rc, off);
|
|
|
|
off += wrc;
|
|
}
|
|
}
|
|
exit(0);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
piper_proc::~piper_proc()
|
|
{
|
|
if (this->pp_child > 0) {
|
|
int status;
|
|
|
|
kill(this->pp_child, SIGTERM);
|
|
while (waitpid(this->pp_child, &status, 0) < 0 && (errno == EINTR)) {
|
|
;
|
|
}
|
|
|
|
this->pp_child = -1;
|
|
}
|
|
}
|