libchromeos: Support setting uid/gid of child processes in process.h

chromeos/process_test.cc

 // Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <chromeos/process.h>
 #include <chromeos/process_mock.h>
 #include <chromeos/test_helpers.h>
 
 #include <base/file_util.h>
 #include <gtest/gtest.h>
@@ skipping 74 matching lines @@
 TEST_F(ProcessTest, NonZeroReturnValue) {
   process_.AddArg(kBinFalse);
   EXPECT_EQ(1, process_.Run());
   ExpectFileEquals("", output_file_.c_str());
   EXPECT_EQ("", GetLog());
 }
 
 TEST_F(ProcessTest, BadOutputFile) {
   process_.AddArg(kBinEcho);
   process_.RedirectOutput("/bad/path");
-  EXPECT_EQ(127, process_.Run());
+  EXPECT_EQ(Process::kErrorExitStatus, process_.Run());
 }
 
 TEST_F(ProcessTest, ExistingOutputFile) {
   process_.AddArg(kBinEcho);
   process_.AddArg("hello world");
   EXPECT_FALSE(file_util::PathExists(FilePath(output_file_)));
   EXPECT_EQ(0, process_.Run());
   EXPECT_TRUE(file_util::PathExists(FilePath(output_file_)));
-  EXPECT_EQ(127, process_.Run());
+  EXPECT_EQ(Process::kErrorExitStatus, process_.Run());
 }
 
 TEST_F(ProcessTest, BadExecutable) {
   process_.AddArg("false");
-  EXPECT_EQ(127, process_.Run());
+  EXPECT_EQ(Process::kErrorExitStatus, process_.Run());
 }
 
 void ProcessTest::CheckStderrCaptured() {
   std::string contents;
   process_.AddArg(kBinCp);
   EXPECT_EQ(1, process_.Run());
   EXPECT_TRUE(file_util::ReadFileToString(FilePath(output_file_),
                                                    &contents));
   EXPECT_NE(std::string::npos, contents.find("missing file operand"));
   EXPECT_EQ("", GetLog());
@@ skipping 69 matching lines @@
   EXPECT_TRUE(process_.Start());
   int write_fd = process_.GetPipe(STDIN_FILENO);
   EXPECT_EQ(-1, process_.GetPipe(STDERR_FILENO));
   EXPECT_TRUE(file_util::WriteFile(GetFdPath(write_fd), kMessage,
                                    strlen(kMessage)));
   close(write_fd);
   EXPECT_EQ(0, process_.Wait());
   ExpectFileEquals(kMessage, output_file_.c_str());
 }
 
+TEST_F(ProcessTest, WithSameUid) {
+  gid_t uid = geteuid();
+  process_.AddArg(kBinEcho);
+  process_.SetUid(uid);
+  EXPECT_EQ(0, process_.Run());
+}
+
+TEST_F(ProcessTest, WithSameGid) {
+  gid_t gid = getegid();
+  process_.AddArg(kBinEcho);
+  process_.SetGid(gid);
+  EXPECT_EQ(0, process_.Run());
+}
+
+TEST_F(ProcessTest, WithIllegalUid) {
+  ASSERT_NE(0, geteuid());
+  process_.AddArg(kBinEcho);
+  process_.SetUid(0);
+  EXPECT_EQ(Process::kErrorExitStatus, process_.Run());
+  std::string contents;
+  EXPECT_TRUE(file_util::ReadFileToString(FilePath(output_file_),
+                                          &contents));
+  EXPECT_NE(std::string::npos,
+            contents.find("Unable to set UID to 0: 1\n"));
+}
+
+TEST_F(ProcessTest, WithIllegalGid) {
+  ASSERT_NE(0, getegid());
+  process_.AddArg(kBinEcho);
+  process_.SetGid(0);
+  EXPECT_EQ(Process::kErrorExitStatus, process_.Run());
+  std::string contents;
+  EXPECT_TRUE(file_util::ReadFileToString(FilePath(output_file_),
+                                          &contents));
+  EXPECT_NE(std::string::npos,
+            contents.find("Unable to set GID to 0: 1\n"));
+}
+
 TEST_F(ProcessTest, NoParams) {
-  EXPECT_EQ(127, process_.Run());
+  EXPECT_EQ(Process::kErrorExitStatus, process_.Run());
 }
 
 TEST_F(ProcessTest, SegFaultHandling) {
   process_.AddArg(kBinBash);
   process_.AddArg("-c");
   process_.AddArg("kill -SEGV $$");
   EXPECT_EQ(-1, process_.Run());
   EXPECT_TRUE(FindLog("did not exit normally: 11"));
 }
 
@@ skipping 25 matching lines @@
   EXPECT_TRUE(process_.Kill(SIGTERM, 1));
   EXPECT_EQ(0, process_.pid());
 }
 
 TEST_F(ProcessTest, Reset) {
   process_.AddArg(kBinFalse);
   process_.Reset(0);
   process_.AddArg(kBinEcho);
   EXPECT_EQ(0, process_.Run());
 }