sandbox/linux/tests/test_utils.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "sandbox/linux/tests/test_utils.h"
   6
   7 #include <errno.h>
   8 #include <sys/stat.h>
   9 #include <sys/types.h>
  10 #include <sys/wait.h>
  11 #include <unistd.h>
  12
  13 #include "base/logging.h"
  14 #include "base/posix/eintr_wrapper.h"
  15
  16 namespace sandbox {
  17
  18 bool TestUtils::CurrentProcessHasChildren() {
  19   siginfo_t process_info;
  20   int ret = HANDLE_EINTR(
  21       waitid(P_ALL, 0, &process_info, WEXITED | WNOHANG | WNOWAIT));
  22   if (-1 == ret) {
  23     PCHECK(ECHILD == errno);
  24     return false;
  25   } else {
  26     return true;
  27   }
  28 }
  29
  30 void TestUtils::HandlePostForkReturn(pid_t pid) {
  31   const int kChildExitCode = 1;
  32   if (pid > 0) {
  33     int status = 0;
  34     PCHECK(pid == HANDLE_EINTR(waitpid(pid, &status, 0)));
  35     CHECK(WIFEXITED(status));
  36     CHECK_EQ(kChildExitCode, WEXITSTATUS(status));
  37   } else if (pid == 0) {
  38     _exit(kChildExitCode);
  39   }
  40 }
  41
  42 }  // namespace sandbox