1 files changed, 86 insertions, 0 deletions
diff --git a/xlators/mgmt/glusterd/src/glusterd-etcd.c b/xlators/mgmt/glusterd/src/glusterd-etcd.c
new file mode 100644
index 000000000..3382e20ae
--- /dev/null
+++ b/xlators/mgmt/glusterd/src/glusterd-etcd.c
@@ -0,0 +1,86 @@
+/*
+   Copyright (c) 2013 Red Hat, Inc. <http://www.redhat.com>
+   This file is part of GlusterFS.
+
+   This file is licensed to you under your choice of the GNU Lesser
+   General Public License, version 3 or any later version (LGPLv3 or
+   later), or the GNU General Public License, version 2 (GPLv2), in all
+   cases as published by the Free Software Foundation.
+*/
+
+#include <errno.h>
+#include <signal.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+#include "glusterfs.h"
+#include "run.h"
+#include "glusterd-etcd.h"
+
+#define GLUSTERD_ETCD_DIR       "/var/lib/glusterd/etcd"
+#define GLUSTERD_ETCD_CMD       "/root/etcd/etcd"
+
+pid_t
+start_etcd (char *this_host, char *other_host)
+{
+        runner_t        runner;
+        char            me[256];
+
+        if (gethostname(me,sizeof(me)-1) != 0) {
+                gf_log (__func__, GF_LOG_ERROR, "gethostname failed?!?");
+                return -1;
+        }
+        me[sizeof(me)-1] = '\0';
+
+        if ((mkdir(GLUSTERD_ETCD_DIR,0700) < 0) && (errno != EEXIST)) {
+                gf_log (__func__, GF_LOG_ERROR,
+                        "failed to create %s", GLUSTERD_ETCD_DIR);
+                return -1;
+        }
+
+        runinit (&runner);
+        runner_add_args (&runner, GLUSTERD_ETCD_CMD,
+                                  "-n", this_host, "-d", GLUSTERD_ETCD_DIR,
+                                  "-c", NULL);
+        runner_argprintf( &runner, "%s:4001", me);
+        runner_add_arg (&runner, "-s");
+        runner_argprintf (&runner, "%s:7001", me);
+        if (other_host) {
+                runner_add_arg (&runner, "-C");
+                runner_argprintf (&runner, "%s:7001", other_host);
+                gf_log (__func__, GF_LOG_INFO, "starting etcd  via %s", other_host);
+        } else {
+                gf_log (__func__, GF_LOG_INFO, "starting etcd standalone");
+        }
+
+        /*
+         * Runner_run would wait for it.  Runner_run_nowait would not wait,
+         * but would detach it so thoroughly that it won't die when we do.
+         * Also, runner->chpid would be the PID of the transient middle
+         * process, not the one we might actually need to kill later.  This
+         * seems to do exactly what we need.
+         */
+        if (runner_start(&runner) != 0) {
+                gf_log (__func__, GF_LOG_ERROR,
+                        "failed to start %s", GLUSTERD_ETCD_CMD);
+                return -1;
+        }
+
+        return runner.chpid;
+}
+
+void
+stop_etcd (pid_t pid)
+{
+        if (pid > 0) {
+                gf_log (__func__, GF_LOG_INFO, "killing etcd %d", pid);
+                (void)kill(pid,SIGKILL);
+                (void)waitpid(pid,NULL,0);
+        }
+}
+
+void
+nuke_etcd_dir (void)
+{
+        (void)runcmd("rm","-rf",GLUSTERD_ETCD_DIR,NULL);
+}