Index: sandbox/linux/services/thread_helpers_unittests.cc |
diff --git a/sandbox/linux/services/thread_helpers_unittests.cc b/sandbox/linux/services/thread_helpers_unittests.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a36fd2975f81da18dc32f53ae221da9475f65483 |
--- /dev/null |
+++ b/sandbox/linux/services/thread_helpers_unittests.cc |
@@ -0,0 +1,111 @@ |
+// Copyright 2014 The Chromium 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 "sandbox/linux/services/thread_helpers.h" |
+ |
+#include <errno.h> |
+#include <fcntl.h> |
+#include <sys/stat.h> |
+#include <sys/types.h> |
+#include <unistd.h> |
+ |
+#include "base/basictypes.h" |
+#include "base/logging.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "base/posix/eintr_wrapper.h" |
+#include "base/process/process_metrics.h" |
+#include "base/threading/platform_thread.h" |
+#include "base/threading/thread.h" |
+#include "build/build_config.h" |
+#include "sandbox/linux/tests/unit_tests.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+using base::PlatformThread; |
+ |
+namespace sandbox { |
+ |
+namespace { |
+ |
+int GetRaceTestIterations() { |
+ if (IsRunningOnValgrind()) { |
+ return 2; |
+ } else { |
+ return 1000; |
+ } |
+} |
+ |
+class ScopedProcSelfTask { |
+ public: |
+ ScopedProcSelfTask() : fd_(-1) { |
+ fd_ = open("/proc/self/task/", O_RDONLY | O_DIRECTORY); |
+ CHECK_LE(0, fd_); |
+ } |
+ |
+ ~ScopedProcSelfTask() { PCHECK(0 == IGNORE_EINTR(close(fd_))); } |
+ |
+ int fd() { return fd_; } |
+ |
+ private: |
+ int fd_; |
+ DISALLOW_COPY_AND_ASSIGN(ScopedProcSelfTask); |
+}; |
+ |
+#if defined(THREAD_SANITIZER) |
+// These tests fail under ThreadSanitizer, see http://crbug.com/342305 |
+#define MAYBE_IsSingleThreadedBasic DISABLED_IsSingleThreadedBasic |
+#define MAYBE_IsSingleThreadedIterated DISABLED_IsSingleThreadedIterated |
+#define MAYBE_IsSingleThreadedStartAndStop DISABLED_IsSingleThreadedStartAndStop |
+#else |
+#define MAYBE_IsSingleThreadedBasic IsSingleThreadedBasic |
+#define MAYBE_IsSingleThreadedIterated IsSingleThreadedIterated |
+#define MAYBE_IsSingleThreadedStartAndStop IsSingleThreadedStartAndStop |
+#endif |
+ |
+TEST(ThreadHelpers, MAYBE_IsSingleThreadedBasic) { |
+ ScopedProcSelfTask task; |
+ ASSERT_TRUE(ThreadHelpers::IsSingleThreaded(task.fd())); |
+ ASSERT_TRUE(ThreadHelpers::IsSingleThreaded(-1)); |
+ |
+ base::Thread thread("sandbox_tests"); |
+ ASSERT_TRUE(thread.Start()); |
+ ASSERT_FALSE(ThreadHelpers::IsSingleThreaded(task.fd())); |
+ ASSERT_FALSE(ThreadHelpers::IsSingleThreaded(-1)); |
+ // Explicitly stop the thread here to not pollute the next test. |
+ ASSERT_TRUE(ThreadHelpers::StopThreadAndWatchProcFS(task.fd(), &thread)); |
+} |
+ |
+TEST(ThreadHelpers, MAYBE_IsSingleThreadedIterated) { |
+ ScopedProcSelfTask task; |
+ ASSERT_TRUE(ThreadHelpers::IsSingleThreaded(task.fd())); |
+ |
+ // Iterate to check for race conditions. |
+ for (int i = 0; i < GetRaceTestIterations(); ++i) { |
+ base::Thread thread("sandbox_tests"); |
+ ASSERT_TRUE(thread.Start()); |
+ ASSERT_FALSE(ThreadHelpers::IsSingleThreaded(task.fd())); |
+ // Explicitly stop the thread here to not pollute the next test. |
+ ASSERT_TRUE(ThreadHelpers::StopThreadAndWatchProcFS(task.fd(), &thread)); |
+ } |
+} |
+ |
+TEST(ThreadHelpers, MAYBE_IsSingleThreadedStartAndStop) { |
+ ScopedProcSelfTask task; |
+ ASSERT_TRUE(ThreadHelpers::IsSingleThreaded(task.fd())); |
+ |
+ base::Thread thread("sandbox_tests"); |
+ // This is testing for a race condition, so iterate. |
+ // Manually, this has been tested with more that 1M iterations. |
+ for (int i = 0; i < GetRaceTestIterations(); ++i) { |
+ ASSERT_TRUE(thread.Start()); |
+ ASSERT_FALSE(ThreadHelpers::IsSingleThreaded(task.fd())); |
+ |
+ ASSERT_TRUE(ThreadHelpers::StopThreadAndWatchProcFS(task.fd(), &thread)); |
+ ASSERT_TRUE(ThreadHelpers::IsSingleThreaded(task.fd())); |
+ ASSERT_EQ(1, base::GetNumberOfThreads(base::GetCurrentProcessHandle())); |
+ } |
+} |
+ |
+} // namespace |
+ |
+} // namespace sandbox |