diff options
Diffstat (limited to 'vespalog/src/logfmt/logfilter.c')
-rw-r--r-- | vespalog/src/logfmt/logfilter.c | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/vespalog/src/logfmt/logfilter.c b/vespalog/src/logfmt/logfilter.c new file mode 100644 index 00000000000..709ce2c016e --- /dev/null +++ b/vespalog/src/logfmt/logfilter.c @@ -0,0 +1,78 @@ +// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +#include <stdio.h> +#include <stdlib.h> +#include <sys/types.h> +#include <sys/wait.h> +#include <unistd.h> +#include <signal.h> + +int main(int argc, char *argv[]) +{ + int lfpipe[2]; + pid_t lfmtpid; + pid_t progpid; + int wstat; + + if (argc < 3) { + fprintf(stderr, "Usage: %s logfmt prog [...]\n", argv[0]); + exit(1); + } + + pipe(lfpipe); + lfmtpid = fork(); + if (lfmtpid == 0) { + close(lfpipe[1]); + if (lfpipe[0] != 0) { + close(0); + dup(lfpipe[0]); + close(lfpipe[0]); + } + execlp(argv[1], argv[1], "-", (const char *)NULL); + perror("exec logfmt failed"); + exit(1); + } + if (lfmtpid < 0) { + perror("fork failed"); + exit(1); + } + close(lfpipe[0]); + + progpid = fork(); + if (progpid == 0) { + char buf[20]; + int i=2; + + sprintf(buf, "fd:%d", lfpipe[1]); + setenv("VESPA_LOG_TARGET", buf, 1); + for (; i<argc; i++) { + argv[i-2] = argv[i]; + } + argv[i-2] = NULL; + execvp(argv[0], argv); + perror("exec program failed"); + exit(1); + } + if (progpid < 0) { + perror("fork failed"); + exit(1); + } + close(lfpipe[1]); + + if (waitpid(lfmtpid, &wstat, 0) != lfmtpid) { + perror("bad waitpid for logfmt"); + } + if (waitpid(progpid, &wstat, 0) != progpid) { + perror("bad waitpid for program"); + } + + if (WIFEXITED(wstat)) { + return WEXITSTATUS(wstat); + } + if (WIFSIGNALED(wstat)) { + int sig = WTERMSIG(wstat); + signal(sig, SIG_DFL); + kill(getpid(), sig); + } + /* should not get here */ + exit(1); +} |