1 files changed, 73 insertions, 0 deletions
diff --git a/logforwarder/src/apps/vespa-logforwarder-start/child-handler.cpp b/logforwarder/src/apps/vespa-logforwarder-start/child-handler.cpp
new file mode 100644
index 00000000000..d7c7a923ddb
--- /dev/null
+++ b/logforwarder/src/apps/vespa-logforwarder-start/child-handler.cpp
@@ -0,0 +1,73 @@
+// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#include "child-handler.h"
+
+#include <stdio.h>
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+
+#include <vespa/log/log.h>
+LOG_SETUP(".child-handler");
+
+ChildHandler::ChildHandler() : _childRunning(false) {}
+
+namespace {
+
+void
+runSplunk(const vespalib::string &prefix, const char *a1, const char *a2 = 0)
+{
+    const char *argv[] = { 0, a1, a2, 0 };
+    vespalib::string path = prefix + "/bin/splunk";
+    argv[0] = path.c_str();
+    fprintf(stdout, "starting splunk forwarder with command: '%s' '%s'\n",
+            argv[0], argv[1]);
+    pid_t child = fork();
+    if (child == -1) {
+        perror("fork()");
+        return;
+    }
+    if (child == 0) {
+        vespalib::string env = "SPLUNK_HOME=" + prefix;
+        char *cenv = const_cast<char *>(env.c_str()); // safe cast
+        putenv(cenv);
+        char **cargv = const_cast<char **>(argv); // safe cast
+        execv(argv[0], cargv);
+        // if execv fails:
+        perror(argv[0]);
+        exit(1);
+    }
+    int waitStatus = 0;
+    if (waitpid(child, &waitStatus, 0) == -1) {
+        perror("waitpid()");
+        return;
+    }
+    if (WIFEXITED(waitStatus) && WEXITSTATUS(waitStatus) == 0) {
+        // all OK
+        return;
+    }
+    if (WIFEXITED(waitStatus)) {
+        fprintf(stderr, "failed starting splunk forwarder (exit status %d)\n",
+                WEXITSTATUS(waitStatus));
+    } else if (WIFSIGNALED(waitStatus)) {
+        fprintf(stderr, "failed starting splunk forwarder (exit on signal %d)\n",
+                WTERMSIG(waitStatus));
+    } else {
+        fprintf(stderr, "failed starting splunk forwarder (abnormal exit status %d)\n",
+                waitStatus);
+    }
+}
+
+} // namespace <unnamed>
+
+
+void
+ChildHandler::startChild(const vespalib::string &prefix)
+{
+    if (_childRunning) {
+        runSplunk(prefix, "restart");
+    } else {
+        runSplunk(prefix, "start", "--accept-license");
+        _childRunning = true;
+    }
+}