test-threads.cc 3.33 KB
Newer Older
iposva@chromium.org's avatar
iposva@chromium.org committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
// Copyright 2008 the V8 project authors. All rights reserved.
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
//     * Redistributions of source code must retain the above copyright
//       notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above
//       copyright notice, this list of conditions and the following
//       disclaimer in the documentation and/or other materials provided
//       with the distribution.
//     * Neither the name of Google Inc. nor the names of its
//       contributors may be used to endorse or promote products derived
//       from this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

28
#include "test/cctest/cctest.h"
iposva@chromium.org's avatar
iposva@chromium.org committed
29

30
#include "src/execution/thread-id.h"
iposva@chromium.org's avatar
iposva@chromium.org committed
31

Clemens Hammacher's avatar
Clemens Hammacher committed
32 33 34 35 36 37 38 39
namespace v8 {
namespace internal {

// {ThreadId} must be trivially copyable to be stored in {std::atomic}.
ASSERT_TRIVIALLY_COPYABLE(i::ThreadId);
using AtomicThreadId = std::atomic<i::ThreadId>;

class ThreadIdValidationThread : public base::Thread {
40
 public:
Clemens Hammacher's avatar
Clemens Hammacher committed
41 42
  ThreadIdValidationThread(base::Thread* thread_to_start, AtomicThreadId* refs,
                           unsigned int thread_no, base::Semaphore* semaphore)
43 44 45 46 47
      : Thread(Options("ThreadRefValidationThread")),
        refs_(refs),
        thread_no_(thread_no),
        thread_to_start_(thread_to_start),
        semaphore_(semaphore) {}
48

49
  void Run() override {
50
    i::ThreadId thread_id = i::ThreadId::Current();
Clemens Hammacher's avatar
Clemens Hammacher committed
51
    CHECK(thread_id.IsValid());
52
    for (int i = 0; i < thread_no_; i++) {
Clemens Hammacher's avatar
Clemens Hammacher committed
53
      CHECK_NE(refs_[i].load(std::memory_order_relaxed), thread_id);
54
    }
Clemens Hammacher's avatar
Clemens Hammacher committed
55
    refs_[thread_no_].store(thread_id, std::memory_order_relaxed);
56
    if (thread_to_start_ != nullptr) {
57
      CHECK(thread_to_start_->Start());
58 59 60
    }
    semaphore_->Signal();
  }
61

62
 private:
Clemens Hammacher's avatar
Clemens Hammacher committed
63 64 65 66
  AtomicThreadId* const refs_;
  const int thread_no_;
  base::Thread* const thread_to_start_;
  base::Semaphore* const semaphore_;
67 68 69
};

TEST(ThreadIdValidation) {
Clemens Hammacher's avatar
Clemens Hammacher committed
70 71 72 73
  constexpr int kNThreads = 100;
  std::unique_ptr<ThreadIdValidationThread> threads[kNThreads];
  AtomicThreadId refs[kNThreads];
  base::Semaphore semaphore(0);
74
  for (int i = kNThreads - 1; i >= 0; i--) {
Clemens Hammacher's avatar
Clemens Hammacher committed
75 76 77
    ThreadIdValidationThread* prev =
        i == kNThreads - 1 ? nullptr : threads[i + 1].get();
    threads[i] =
78
        std::make_unique<ThreadIdValidationThread>(prev, refs, i, &semaphore);
79
  }
80
  CHECK(threads[0]->Start());
81
  for (int i = 0; i < kNThreads; i++) {
82
    semaphore.Wait();
83 84
  }
}
Clemens Hammacher's avatar
Clemens Hammacher committed
85 86 87

}  // namespace internal
}  // namespace v8