+++ /dev/null
-// -*- 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"
-
-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);
- 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');
-}
-
-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');
-}
-
-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');
-}
-
-TEST(SubProcess, Subshell)
-{
- SubProcess sh("/bin/sh", SubProcess::PIPE, SubProcess::PIPE, SubProcess::PIPE);
- sh.add_cmd_args("-c",
- "sleep 0; "
- "cat; "
- "echo 'error from subshell' >&2; "
- "/bin/sh -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));
- std::cerr << "stderr: " << buf;
- ASSERT_FALSE(buf.empty());
- 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("/bin/sh", 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("/bin/sh", SubProcess::PIPE, SubProcess::PIPE, SubProcess::PIPE, 10);
- sh.add_cmd_args("-c", "sh -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("/bin/sh", 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;
- ASSERT_TRUE(read_from_fd(sh.get_stderr(), buf));
- std::cerr << "stderr: " << buf;
- ASSERT_FALSE(buf.empty());
- ASSERT_EQ(sh.join(), 128 + SIGTERM);
- std::cerr << "err: " << sh.err() << std::endl;
- ASSERT_FALSE(sh.err().c_str()[0] == '\0');
-}
-
-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; }));
-}