1 files changed, 314 insertions, 0 deletions
diff --git a/src/test/test_subprocess.cc b/src/test/test_subprocess.cc
new file mode 100644
index 000000000..7e7693e4c
--- /dev/null
+++ b/src/test/test_subprocess.cc
@@ -0,0 +1,314 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+/*
+ * Ceph distributed storage system
+ *
+ * Copyright (C) 2015 Mirantis Inc
+ *
+ * Author: Mykola Golub <mgolub@mirantis.com>
+ *
+ *  This library is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU Lesser General Public
+ *  License as published by the Free Software Foundation; either
+ *  version 2.1 of the License, or (at your option) any later version.
+ *
+ */
+
+#include <unistd.h>
+
+#include <iostream>
+
+#include "common/SubProcess.h"
+#include "common/safe_io.h"
+#include "gtest/gtest.h"
+#include "common/fork_function.h"
+
+#ifdef _WIN32
+// Some of the tests expect GNU binaries to be available. We'll just rely on
+// the ones provided by Msys (which also comes with Git for Windows).
+#define SHELL "bash.exe"
+#else
+#define SHELL "/bin/sh"
+#endif
+
+bool read_from_fd(int fd, std::string &out) {
+  out.clear();
+  char buf[1024];
+  ssize_t n = safe_read(fd, buf, sizeof(buf) - 1);
+  if (n < 0)
+    return false;
+  buf[n] = '\0';
+  out = buf;
+  return true;
+}
+
+TEST(SubProcess, True)
+{
+  SubProcess p("true");
+  ASSERT_EQ(p.spawn(), 0);
+  ASSERT_EQ(p.join(), 0);
+  ASSERT_TRUE(p.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcess, False)
+{
+  SubProcess p("false");
+  ASSERT_EQ(p.spawn(), 0);
+  ASSERT_EQ(p.join(), 1);
+  ASSERT_FALSE(p.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcess, NotFound)
+{
+  SubProcess p("NOTEXISTENTBINARY", SubProcess::CLOSE, SubProcess::CLOSE, SubProcess::PIPE);
+  #ifdef _WIN32
+  // Windows will error out early.
+  ASSERT_EQ(p.spawn(), -1);
+  #else
+  ASSERT_EQ(p.spawn(), 0);
+  std::string buf;
+  ASSERT_TRUE(read_from_fd(p.get_stderr(), buf));
+  std::cerr << "stderr: " << buf;
+  ASSERT_EQ(p.join(), 1);
+  std::cerr << "err: " << p.err() << std::endl;
+  ASSERT_FALSE(p.err().c_str()[0] == '\0');
+  #endif
+}
+
+TEST(SubProcess, Echo)
+{
+  SubProcess echo("echo", SubProcess::CLOSE, SubProcess::PIPE);
+  echo.add_cmd_args("1", "2", "3", NULL);
+
+  ASSERT_EQ(echo.spawn(), 0);
+  std::string buf;
+  ASSERT_TRUE(read_from_fd(echo.get_stdout(), buf));
+  std::cerr << "stdout: " << buf;
+  ASSERT_EQ(buf, "1 2 3\n");
+  ASSERT_EQ(echo.join(), 0);
+  ASSERT_TRUE(echo.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcess, Cat)
+{
+  SubProcess cat("cat", SubProcess::PIPE, SubProcess::PIPE, SubProcess::PIPE);
+
+  ASSERT_EQ(cat.spawn(), 0);
+  std::string msg("to my, trociny!");
+  int n = write(cat.get_stdin(), msg.c_str(), msg.size());
+  ASSERT_EQ(n, (int)msg.size());
+  cat.close_stdin();
+  std::string buf;
+  ASSERT_TRUE(read_from_fd(cat.get_stdout(), buf));
+  std::cerr << "stdout: " << buf << std::endl;
+  ASSERT_EQ(buf, msg);
+  ASSERT_TRUE(read_from_fd(cat.get_stderr(), buf));
+  ASSERT_EQ(buf, "");
+  ASSERT_EQ(cat.join(), 0);
+  ASSERT_TRUE(cat.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcess, CatDevNull)
+{
+  SubProcess cat("cat", SubProcess::PIPE, SubProcess::PIPE, SubProcess::PIPE);
+  cat.add_cmd_arg("/dev/null");
+
+  ASSERT_EQ(cat.spawn(), 0);
+  std::string buf;
+  ASSERT_TRUE(read_from_fd(cat.get_stdout(), buf));
+  ASSERT_EQ(buf, "");
+  ASSERT_TRUE(read_from_fd(cat.get_stderr(), buf));
+  ASSERT_EQ(buf, "");
+  ASSERT_EQ(cat.join(), 0);
+  ASSERT_TRUE(cat.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcess, Killed)
+{
+  SubProcessTimed cat("cat", SubProcess::PIPE, SubProcess::PIPE);
+
+  ASSERT_EQ(cat.spawn(), 0);
+  cat.kill();
+  ASSERT_EQ(cat.join(), 128 + SIGTERM);
+  std::cerr << "err: " << cat.err() << std::endl;
+  ASSERT_FALSE(cat.err().c_str()[0] == '\0');
+}
+
+#ifndef _WIN32
+TEST(SubProcess, CatWithArgs)
+{
+  SubProcess cat("cat", SubProcess::PIPE, SubProcess::PIPE, SubProcess::PIPE);
+  cat.add_cmd_args("/dev/stdin", "/dev/null", "/NOTEXIST", NULL);
+
+  ASSERT_EQ(cat.spawn(), 0);
+  std::string msg("Hello, Word!");
+  int n = write(cat.get_stdin(), msg.c_str(), msg.size());
+  ASSERT_EQ(n, (int)msg.size());
+  cat.close_stdin();
+  std::string buf;
+  ASSERT_TRUE(read_from_fd(cat.get_stdout(), buf));
+  std::cerr << "stdout: " << buf << std::endl;
+  ASSERT_EQ(buf, msg);
+  ASSERT_TRUE(read_from_fd(cat.get_stderr(), buf));
+  std::cerr << "stderr: " << buf;
+  ASSERT_FALSE(buf.empty());
+  ASSERT_EQ(cat.join(), 1);
+  std::cerr << "err: " << cat.err() << std::endl;
+  ASSERT_FALSE(cat.err().c_str()[0] == '\0');
+}
+#endif
+
+TEST(SubProcess, Subshell)
+{
+  SubProcess sh(SHELL, SubProcess::PIPE, SubProcess::PIPE, SubProcess::PIPE);
+  sh.add_cmd_args("-c",
+      "sleep 0; "
+      "cat; "
+      "echo 'error from subshell' >&2; "
+      SHELL " -c 'exit 13'", NULL);
+  ASSERT_EQ(sh.spawn(), 0);
+  std::string msg("hello via subshell");
+  int n = write(sh.get_stdin(), msg.c_str(), msg.size());
+  ASSERT_EQ(n, (int)msg.size());
+  sh.close_stdin();
+  std::string buf;
+  ASSERT_TRUE(read_from_fd(sh.get_stdout(), buf));
+  std::cerr << "stdout: " << buf << std::endl;
+  ASSERT_EQ(buf, msg);
+  ASSERT_TRUE(read_from_fd(sh.get_stderr(), buf));
+  std::cerr << "stderr: " << buf;
+  ASSERT_EQ(buf, "error from subshell\n");
+  ASSERT_EQ(sh.join(), 13);
+  std::cerr << "err: " << sh.err() << std::endl;
+  ASSERT_FALSE(sh.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcessTimed, True)
+{
+  SubProcessTimed p("true", SubProcess::CLOSE, SubProcess::CLOSE, SubProcess::CLOSE, 10);
+  ASSERT_EQ(p.spawn(), 0);
+  ASSERT_EQ(p.join(), 0);
+  ASSERT_TRUE(p.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcessTimed, SleepNoTimeout)
+{
+  SubProcessTimed sleep("sleep", SubProcess::CLOSE, SubProcess::CLOSE, SubProcess::CLOSE, 0);
+  sleep.add_cmd_arg("1");
+
+  ASSERT_EQ(sleep.spawn(), 0);
+  ASSERT_EQ(sleep.join(), 0);
+  ASSERT_TRUE(sleep.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcessTimed, Killed)
+{
+  SubProcessTimed cat("cat", SubProcess::PIPE, SubProcess::PIPE, SubProcess::PIPE, 5);
+
+  ASSERT_EQ(cat.spawn(), 0);
+  cat.kill();
+  std::string buf;
+  ASSERT_TRUE(read_from_fd(cat.get_stdout(), buf));
+  ASSERT_TRUE(buf.empty());
+  ASSERT_TRUE(read_from_fd(cat.get_stderr(), buf));
+  ASSERT_TRUE(buf.empty());
+  ASSERT_EQ(cat.join(), 128 + SIGTERM);
+  std::cerr << "err: " << cat.err() << std::endl;
+  ASSERT_FALSE(cat.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcessTimed, SleepTimedout)
+{
+  SubProcessTimed sleep("sleep", SubProcess::CLOSE, SubProcess::CLOSE, SubProcess::PIPE, 1);
+  sleep.add_cmd_arg("10");
+
+  ASSERT_EQ(sleep.spawn(), 0);
+  std::string buf;
+  ASSERT_TRUE(read_from_fd(sleep.get_stderr(), buf));
+  #ifndef _WIN32
+  std::cerr << "stderr: " << buf;
+  ASSERT_FALSE(buf.empty());
+  #endif
+  ASSERT_EQ(sleep.join(), 128 + SIGKILL);
+  std::cerr << "err: " << sleep.err() << std::endl;
+  ASSERT_FALSE(sleep.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcessTimed, SubshellNoTimeout)
+{
+  SubProcessTimed sh(SHELL, SubProcess::PIPE, SubProcess::PIPE, SubProcess::PIPE, 0);
+  sh.add_cmd_args("-c", "cat >&2", NULL);
+  ASSERT_EQ(sh.spawn(), 0);
+  std::string msg("the quick brown fox jumps over the lazy dog");
+  int n = write(sh.get_stdin(), msg.c_str(), msg.size());
+  ASSERT_EQ(n, (int)msg.size());
+  sh.close_stdin();
+  std::string buf;
+  ASSERT_TRUE(read_from_fd(sh.get_stdout(), buf));
+  std::cerr << "stdout: " << buf << std::endl;
+  ASSERT_TRUE(buf.empty());
+  ASSERT_TRUE(read_from_fd(sh.get_stderr(), buf));
+  std::cerr << "stderr: " << buf << std::endl;
+  ASSERT_EQ(buf, msg);
+  ASSERT_EQ(sh.join(), 0);
+  ASSERT_TRUE(sh.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcessTimed, SubshellKilled)
+{
+  SubProcessTimed sh(SHELL, SubProcess::PIPE, SubProcess::PIPE, SubProcess::PIPE, 10);
+  sh.add_cmd_args("-c", SHELL "-c cat", NULL);
+  ASSERT_EQ(sh.spawn(), 0);
+  std::string msg("etaoin shrdlu");
+  int n = write(sh.get_stdin(), msg.c_str(), msg.size());
+  ASSERT_EQ(n, (int)msg.size());
+  sh.kill();
+  std::string buf;
+  ASSERT_TRUE(read_from_fd(sh.get_stderr(), buf));
+  ASSERT_TRUE(buf.empty());
+  ASSERT_EQ(sh.join(), 128 + SIGTERM);
+  std::cerr << "err: " << sh.err() << std::endl;
+  ASSERT_FALSE(sh.err().c_str()[0] == '\0');
+}
+
+TEST(SubProcessTimed, SubshellTimedout)
+{
+  SubProcessTimed sh(SHELL, SubProcess::PIPE, SubProcess::PIPE, SubProcess::PIPE, 1, SIGTERM);
+  sh.add_cmd_args("-c", "sleep 1000& cat; NEVER REACHED", NULL);
+  ASSERT_EQ(sh.spawn(), 0);
+  std::string buf;
+  #ifndef _WIN32
+  ASSERT_TRUE(read_from_fd(sh.get_stderr(), buf));
+  std::cerr << "stderr: " << buf;
+  ASSERT_FALSE(buf.empty());
+  #endif
+  ASSERT_EQ(sh.join(), 128 + SIGTERM);
+  std::cerr << "err: " << sh.err() << std::endl;
+  ASSERT_FALSE(sh.err().c_str()[0] == '\0');
+}
+
+#ifndef _WIN32
+TEST(fork_function, normal)
+{
+  ASSERT_EQ(0, fork_function(10, std::cerr, [&]() { return 0; }));
+  ASSERT_EQ(1, fork_function(10, std::cerr, [&]() { return 1; }));
+  ASSERT_EQ(13, fork_function(10, std::cerr, [&]() { return 13; }));
+  ASSERT_EQ(-1, fork_function(10, std::cerr, [&]() { return -1; }));
+  ASSERT_EQ(-13, fork_function(10, std::cerr, [&]() { return -13; }));
+  ASSERT_EQ(-ETIMEDOUT,
+	    fork_function(10, std::cerr, [&]() { return -ETIMEDOUT; }));
+}
+
+TEST(fork_function, timeout)
+{
+  ASSERT_EQ(-ETIMEDOUT, fork_function(2, std::cerr, [&]() {
+	sleep(60);
+	return 0; }));
+  ASSERT_EQ(-ETIMEDOUT, fork_function(2, std::cerr, [&]() {
+	sleep(60);
+	return 1; }));
+  ASSERT_EQ(-ETIMEDOUT, fork_function(2, std::cerr, [&]() {
+	sleep(60);
+	return -111; }));
+}
+#endif