1// Copyright 2011 The Kyua Authors. 2// All rights reserved. 3// 4// Redistribution and use in source and binary forms, with or without 5// modification, are permitted provided that the following conditions are 6// met: 7// 8// * Redistributions of source code must retain the above copyright 9// notice, this list of conditions and the following disclaimer. 10// * Redistributions in binary form must reproduce the above copyright 11// notice, this list of conditions and the following disclaimer in the 12// documentation and/or other materials provided with the distribution. 13// * Neither the name of Google Inc. nor the names of its contributors 14// may be used to endorse or promote products derived from this software 15// without specific prior written permission. 16// 17// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 29extern "C" { 30#include <sys/stat.h> 31 32#include <unistd.h> 33 34extern char** environ; 35} 36 37#include <cstdlib> 38#include <cstring> 39#include <fstream> 40#include <iostream> 41#include <sstream> 42 43#include "utils/env.hpp" 44#include "utils/format/containers.ipp" 45#include "utils/format/macros.hpp" 46#include "utils/fs/operations.hpp" 47#include "utils/fs/path.hpp" 48#include "utils/optional.ipp" 49#include "utils/test_utils.ipp" 50 51namespace fs = utils::fs; 52 53using utils::optional; 54 55 56namespace { 57 58 59/// Gets the name of the test case to run. 60/// 61/// We use the value of the TEST_CASE environment variable if present, or 62/// else the basename of the test program. 63/// 64/// \param arg0 Value of argv[0] as passed to main(). 65/// 66/// \return A test case name. The name may not be valid. 67static std::string 68guess_test_case_name(const char* arg0) 69{ 70 const optional< std::string > test_case_env = utils::getenv("TEST_CASE"); 71 if (test_case_env) { 72 return test_case_env.get(); 73 } else { 74 return fs::path(arg0).leaf_name(); 75 } 76} 77 78 79/// Logs an error message and exits the test with an error code. 80/// 81/// \param str The error message to log. 82static void 83fail(const std::string& str) 84{ 85 std::cerr << str << '\n'; 86 std::exit(EXIT_FAILURE); 87} 88 89 90/// A test case that validates the TEST_ENV_* variables. 91static void 92test_check_configuration_variables(void) 93{ 94 std::set< std::string > vars; 95 char** iter; 96 for (iter = environ; *iter != NULL; ++iter) { 97 if (std::strstr(*iter, "TEST_ENV_") == *iter) { 98 vars.insert(*iter); 99 } 100 } 101 102 std::set< std::string > exp_vars; 103 exp_vars.insert("TEST_ENV_first=some value"); 104 exp_vars.insert("TEST_ENV_second=some other value"); 105 if (vars != exp_vars) { 106 fail(F("Expected: %s\nFound: %s\n") % exp_vars % vars); 107 } 108} 109 110 111/// A test case that crashes. 112static void 113test_crash(void) 114{ 115 utils::abort_without_coredump(); 116} 117 118 119/// A test case that exits with a non-zero exit code, and not 1. 120static void 121test_fail(void) 122{ 123 std::exit(8); 124} 125 126 127/// A test case that passes. 128static void 129test_pass(void) 130{ 131} 132 133 134/// A test case that spawns a subchild that gets stuck. 135/// 136/// This test case is used by the caller to validate that the whole process tree 137/// is terminated when the test case is killed. 138static void 139test_spawn_blocking_child(void) 140{ 141 pid_t pid = ::fork(); 142 if (pid == -1) 143 fail("Cannot fork subprocess"); 144 else if (pid == 0) { 145 for (;;) 146 ::pause(); 147 } else { 148 const fs::path name = fs::path(utils::getenv("CONTROL_DIR").get()) / 149 "pid"; 150 std::ofstream pidfile(name.c_str()); 151 if (!pidfile) 152 fail("Failed to create the pidfile"); 153 pidfile << pid; 154 pidfile.close(); 155 } 156} 157 158 159/// A test case that times out. 160/// 161/// Note that the timeout is defined in the Kyuafile, as the plain interface has 162/// no means for test programs to specify this by themselves. 163static void 164test_timeout(void) 165{ 166 ::sleep(10); 167 const fs::path control_dir = fs::path(utils::getenv("CONTROL_DIR").get()); 168 std::ofstream file((control_dir / "cookie").c_str()); 169 if (!file) 170 fail("Failed to create the control cookie"); 171 file.close(); 172} 173 174 175/// A test case that performs basic checks on the runtime environment. 176/// 177/// If the runtime environment does not look clean (according to the rules in 178/// the Kyua runtime properties), the test fails. 179static void 180test_validate_isolation(void) 181{ 182 if (utils::getenv("HOME").get() == "fake-value") 183 fail("HOME not reset"); 184 if (utils::getenv("LANG")) 185 fail("LANG not unset"); 186} 187 188 189} // anonymous namespace 190 191 192/// Entry point to the test program. 193/// 194/// The caller can select which test case to run by defining the TEST_CASE 195/// environment variable. This is not "standard", in the sense this is not a 196/// generic property of the plain test case interface. 197/// 198/// \todo It may be worth to split this binary into separate, smaller binaries, 199/// one for every "test case". We use this program as a dispatcher for 200/// different "main"s, the only reason being to keep the amount of helper test 201/// programs to a minimum. However, putting this each function in its own 202/// binary could simplify many other things. 203/// 204/// \param argc The number of CLI arguments. 205/// \param argv The CLI arguments themselves. These are not used because 206/// Kyua will not pass any arguments to the plain test program. 207int 208main(int argc, char** argv) 209{ 210 if (argc != 1) { 211 std::cerr << "No arguments allowed; select the test case with the " 212 "TEST_CASE variable"; 213 return EXIT_FAILURE; 214 } 215 216 const std::string& test_case = guess_test_case_name(argv[0]); 217 218 if (test_case == "check_configuration_variables") 219 test_check_configuration_variables(); 220 else if (test_case == "crash") 221 test_crash(); 222 else if (test_case == "fail") 223 test_fail(); 224 else if (test_case == "pass") 225 test_pass(); 226 else if (test_case == "spawn_blocking_child") 227 test_spawn_blocking_child(); 228 else if (test_case == "timeout") 229 test_timeout(); 230 else if (test_case == "validate_isolation") 231 test_validate_isolation(); 232 else { 233 std::cerr << "Unknown test case"; 234 return EXIT_FAILURE; 235 } 236 237 return EXIT_SUCCESS; 238} 239