Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(356)

Unified Diff: mojo/edk/system/message_pipe_dispatcher_unittest.cc

Issue 814543006: Move //mojo/{public, edk} underneath //third_party (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Rebase Created 5 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « mojo/edk/system/message_pipe_dispatcher.cc ('k') | mojo/edk/system/message_pipe_endpoint.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: mojo/edk/system/message_pipe_dispatcher_unittest.cc
diff --git a/mojo/edk/system/message_pipe_dispatcher_unittest.cc b/mojo/edk/system/message_pipe_dispatcher_unittest.cc
deleted file mode 100644
index b5562b00ff7dadc65287cfa6f05c1e9399ff50e8..0000000000000000000000000000000000000000
--- a/mojo/edk/system/message_pipe_dispatcher_unittest.cc
+++ /dev/null
@@ -1,691 +0,0 @@
-// Copyright 2013 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.
-
-// NOTE(vtl): Some of these tests are inherently flaky (e.g., if run on a
-// heavily-loaded system). Sorry. |test::EpsilonTimeout()| may be increased to
-// increase tolerance and reduce observed flakiness (though doing so reduces the
-// meaningfulness of the test).
-
-#include "mojo/edk/system/message_pipe_dispatcher.h"
-
-#include <string.h>
-
-#include <limits>
-
-#include "base/memory/ref_counted.h"
-#include "base/memory/scoped_vector.h"
-#include "base/rand_util.h"
-#include "base/threading/platform_thread.h" // For |Sleep()|.
-#include "base/threading/simple_thread.h"
-#include "base/time/time.h"
-#include "mojo/edk/system/message_pipe.h"
-#include "mojo/edk/system/test_utils.h"
-#include "mojo/edk/system/waiter.h"
-#include "mojo/edk/system/waiter_test_utils.h"
-#include "testing/gtest/include/gtest/gtest.h"
-
-namespace mojo {
-namespace system {
-namespace {
-
-const MojoHandleSignals kAllSignals = MOJO_HANDLE_SIGNAL_READABLE |
- MOJO_HANDLE_SIGNAL_WRITABLE |
- MOJO_HANDLE_SIGNAL_PEER_CLOSED;
-
-TEST(MessagePipeDispatcherTest, Basic) {
- test::Stopwatch stopwatch;
- int32_t buffer[1];
- const uint32_t kBufferSize = static_cast<uint32_t>(sizeof(buffer));
- uint32_t buffer_size;
-
- // Run this test both with |d0| as port 0, |d1| as port 1 and vice versa.
- for (unsigned i = 0; i < 2; i++) {
- scoped_refptr<MessagePipeDispatcher> d0(new MessagePipeDispatcher(
- MessagePipeDispatcher::kDefaultCreateOptions));
- EXPECT_EQ(Dispatcher::kTypeMessagePipe, d0->GetType());
- scoped_refptr<MessagePipeDispatcher> d1(new MessagePipeDispatcher(
- MessagePipeDispatcher::kDefaultCreateOptions));
- {
- scoped_refptr<MessagePipe> mp(MessagePipe::CreateLocalLocal());
- d0->Init(mp, i); // 0, 1.
- d1->Init(mp, i ^ 1); // 1, 0.
- }
- Waiter w;
- uint32_t context = 0;
- HandleSignalsState hss;
-
- // Try adding a writable waiter when already writable.
- w.Init();
- hss = HandleSignalsState();
- EXPECT_EQ(MOJO_RESULT_ALREADY_EXISTS,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_WRITABLE, 0, &hss));
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_WRITABLE, hss.satisfied_signals);
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals);
- // Shouldn't need to remove the waiter (it was not added).
-
- // Add a readable waiter to |d0|, then make it readable (by writing to
- // |d1|), then wait.
- w.Init();
- ASSERT_EQ(MOJO_RESULT_OK,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_READABLE, 1, nullptr));
- buffer[0] = 123456789;
- EXPECT_EQ(MOJO_RESULT_OK,
- d1->WriteMessage(UserPointer<const void>(buffer), kBufferSize,
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE));
- stopwatch.Start();
- EXPECT_EQ(MOJO_RESULT_OK, w.Wait(MOJO_DEADLINE_INDEFINITE, &context));
- EXPECT_EQ(1u, context);
- EXPECT_LT(stopwatch.Elapsed(), test::EpsilonTimeout());
- hss = HandleSignalsState();
- d0->RemoveAwakable(&w, &hss);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE,
- hss.satisfied_signals);
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals);
-
- // Try adding a readable waiter when already readable (from above).
- w.Init();
- hss = HandleSignalsState();
- EXPECT_EQ(MOJO_RESULT_ALREADY_EXISTS,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_READABLE, 2, &hss));
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE,
- hss.satisfied_signals);
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals);
- // Shouldn't need to remove the waiter (it was not added).
-
- // Make |d0| no longer readable (by reading from it).
- buffer[0] = 0;
- buffer_size = kBufferSize;
- EXPECT_EQ(MOJO_RESULT_OK,
- d0->ReadMessage(UserPointer<void>(buffer),
- MakeUserPointer(&buffer_size), 0, nullptr,
- MOJO_READ_MESSAGE_FLAG_NONE));
- EXPECT_EQ(kBufferSize, buffer_size);
- EXPECT_EQ(123456789, buffer[0]);
-
- // Wait for zero time for readability on |d0| (will time out).
- w.Init();
- ASSERT_EQ(MOJO_RESULT_OK,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_READABLE, 3, nullptr));
- stopwatch.Start();
- EXPECT_EQ(MOJO_RESULT_DEADLINE_EXCEEDED, w.Wait(0, nullptr));
- EXPECT_LT(stopwatch.Elapsed(), test::EpsilonTimeout());
- hss = HandleSignalsState();
- d0->RemoveAwakable(&w, &hss);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_WRITABLE, hss.satisfied_signals);
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals);
-
- // Wait for non-zero, finite time for readability on |d0| (will time out).
- w.Init();
- ASSERT_EQ(MOJO_RESULT_OK,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_READABLE, 3, nullptr));
- stopwatch.Start();
- EXPECT_EQ(MOJO_RESULT_DEADLINE_EXCEEDED,
- w.Wait(2 * test::EpsilonTimeout().InMicroseconds(), nullptr));
- base::TimeDelta elapsed = stopwatch.Elapsed();
- EXPECT_GT(elapsed, (2 - 1) * test::EpsilonTimeout());
- EXPECT_LT(elapsed, (2 + 1) * test::EpsilonTimeout());
- hss = HandleSignalsState();
- d0->RemoveAwakable(&w, &hss);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_WRITABLE, hss.satisfied_signals);
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals);
-
- // Check the peer closed signal.
- w.Init();
- ASSERT_EQ(MOJO_RESULT_OK,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_PEER_CLOSED, 12, nullptr));
-
- // Close the peer.
- EXPECT_EQ(MOJO_RESULT_OK, d1->Close());
-
- // It should be signaled.
- EXPECT_EQ(MOJO_RESULT_OK, w.Wait(1000, &context));
- EXPECT_EQ(12u, context);
- hss = HandleSignalsState();
- d0->RemoveAwakable(&w, &hss);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfied_signals);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfiable_signals);
-
- EXPECT_EQ(MOJO_RESULT_OK, d0->Close());
- }
-}
-
-TEST(MessagePipeDispatcherTest, InvalidParams) {
- char buffer[1];
-
- scoped_refptr<MessagePipeDispatcher> d0(
- new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions));
- scoped_refptr<MessagePipeDispatcher> d1(
- new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions));
- {
- scoped_refptr<MessagePipe> mp(MessagePipe::CreateLocalLocal());
- d0->Init(mp, 0);
- d1->Init(mp, 1);
- }
-
- // |WriteMessage|:
- // Huge buffer size.
- EXPECT_EQ(MOJO_RESULT_RESOURCE_EXHAUSTED,
- d0->WriteMessage(UserPointer<const void>(buffer),
- std::numeric_limits<uint32_t>::max(), nullptr,
- MOJO_WRITE_MESSAGE_FLAG_NONE));
-
- EXPECT_EQ(MOJO_RESULT_OK, d0->Close());
- EXPECT_EQ(MOJO_RESULT_OK, d1->Close());
-}
-
-// These test invalid arguments that should cause death if we're being paranoid
-// about checking arguments (which we would want to do if, e.g., we were in a
-// true "kernel" situation, but we might not want to do otherwise for
-// performance reasons). Probably blatant errors like passing in null pointers
-// (for required pointer arguments) will still cause death, but perhaps not
-// predictably.
-TEST(MessagePipeDispatcherTest, InvalidParamsDeath) {
- const char kMemoryCheckFailedRegex[] = "Check failed";
-
- scoped_refptr<MessagePipeDispatcher> d0(
- new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions));
- scoped_refptr<MessagePipeDispatcher> d1(
- new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions));
- {
- scoped_refptr<MessagePipe> mp(MessagePipe::CreateLocalLocal());
- d0->Init(mp, 0);
- d1->Init(mp, 1);
- }
-
- // |WriteMessage|:
- // Null buffer with nonzero buffer size.
- EXPECT_DEATH_IF_SUPPORTED(d0->WriteMessage(NullUserPointer(), 1, nullptr,
- MOJO_WRITE_MESSAGE_FLAG_NONE),
- kMemoryCheckFailedRegex);
-
- // |ReadMessage|:
- // Null buffer with nonzero buffer size.
- // First write something so that we actually have something to read.
- EXPECT_EQ(MOJO_RESULT_OK,
- d1->WriteMessage(UserPointer<const void>("x"), 1, nullptr,
- MOJO_WRITE_MESSAGE_FLAG_NONE));
- uint32_t buffer_size = 1;
- EXPECT_DEATH_IF_SUPPORTED(
- d0->ReadMessage(NullUserPointer(), MakeUserPointer(&buffer_size), 0,
- nullptr, MOJO_READ_MESSAGE_FLAG_NONE),
- kMemoryCheckFailedRegex);
-
- EXPECT_EQ(MOJO_RESULT_OK, d0->Close());
- EXPECT_EQ(MOJO_RESULT_OK, d1->Close());
-}
-
-// Test what happens when one end is closed (single-threaded test).
-TEST(MessagePipeDispatcherTest, BasicClosed) {
- int32_t buffer[1];
- const uint32_t kBufferSize = static_cast<uint32_t>(sizeof(buffer));
- uint32_t buffer_size;
-
- // Run this test both with |d0| as port 0, |d1| as port 1 and vice versa.
- for (unsigned i = 0; i < 2; i++) {
- scoped_refptr<MessagePipeDispatcher> d0(new MessagePipeDispatcher(
- MessagePipeDispatcher::kDefaultCreateOptions));
- scoped_refptr<MessagePipeDispatcher> d1(new MessagePipeDispatcher(
- MessagePipeDispatcher::kDefaultCreateOptions));
- {
- scoped_refptr<MessagePipe> mp(MessagePipe::CreateLocalLocal());
- d0->Init(mp, i); // 0, 1.
- d1->Init(mp, i ^ 1); // 1, 0.
- }
- Waiter w;
- HandleSignalsState hss;
-
- // Write (twice) to |d1|.
- buffer[0] = 123456789;
- EXPECT_EQ(MOJO_RESULT_OK,
- d1->WriteMessage(UserPointer<const void>(buffer), kBufferSize,
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE));
- buffer[0] = 234567890;
- EXPECT_EQ(MOJO_RESULT_OK,
- d1->WriteMessage(UserPointer<const void>(buffer), kBufferSize,
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE));
-
- // Try waiting for readable on |d0|; should fail (already satisfied).
- w.Init();
- hss = HandleSignalsState();
- EXPECT_EQ(MOJO_RESULT_ALREADY_EXISTS,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_READABLE, 0, &hss));
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE,
- hss.satisfied_signals);
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals);
-
- // Try reading from |d1|; should fail (nothing to read).
- buffer[0] = 0;
- buffer_size = kBufferSize;
- EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT,
- d1->ReadMessage(UserPointer<void>(buffer),
- MakeUserPointer(&buffer_size), 0, nullptr,
- MOJO_READ_MESSAGE_FLAG_NONE));
-
- // Close |d1|.
- EXPECT_EQ(MOJO_RESULT_OK, d1->Close());
-
- // Try waiting for readable on |d0|; should fail (already satisfied).
- w.Init();
- hss = HandleSignalsState();
- EXPECT_EQ(MOJO_RESULT_ALREADY_EXISTS,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_READABLE, 1, &hss));
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_PEER_CLOSED,
- hss.satisfied_signals);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_PEER_CLOSED,
- hss.satisfiable_signals);
-
- // Read from |d0|.
- buffer[0] = 0;
- buffer_size = kBufferSize;
- EXPECT_EQ(MOJO_RESULT_OK,
- d0->ReadMessage(UserPointer<void>(buffer),
- MakeUserPointer(&buffer_size), 0, nullptr,
- MOJO_READ_MESSAGE_FLAG_NONE));
- EXPECT_EQ(kBufferSize, buffer_size);
- EXPECT_EQ(123456789, buffer[0]);
-
- // Try waiting for readable on |d0|; should fail (already satisfied).
- w.Init();
- hss = HandleSignalsState();
- EXPECT_EQ(MOJO_RESULT_ALREADY_EXISTS,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_READABLE, 2, &hss));
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_PEER_CLOSED,
- hss.satisfied_signals);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_PEER_CLOSED,
- hss.satisfiable_signals);
-
- // Read again from |d0|.
- buffer[0] = 0;
- buffer_size = kBufferSize;
- EXPECT_EQ(MOJO_RESULT_OK,
- d0->ReadMessage(UserPointer<void>(buffer),
- MakeUserPointer(&buffer_size), 0, nullptr,
- MOJO_READ_MESSAGE_FLAG_NONE));
- EXPECT_EQ(kBufferSize, buffer_size);
- EXPECT_EQ(234567890, buffer[0]);
-
- // Try waiting for readable on |d0|; should fail (unsatisfiable).
- w.Init();
- hss = HandleSignalsState();
- EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_READABLE, 3, &hss));
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfied_signals);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfiable_signals);
-
- // Try waiting for writable on |d0|; should fail (unsatisfiable).
- w.Init();
- hss = HandleSignalsState();
- EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION,
- d0->AddAwakable(&w, MOJO_HANDLE_SIGNAL_WRITABLE, 4, &hss));
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfied_signals);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfiable_signals);
-
- // Try reading from |d0|; should fail (nothing to read and other end
- // closed).
- buffer[0] = 0;
- buffer_size = kBufferSize;
- EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION,
- d0->ReadMessage(UserPointer<void>(buffer),
- MakeUserPointer(&buffer_size), 0, nullptr,
- MOJO_READ_MESSAGE_FLAG_NONE));
-
- // Try writing to |d0|; should fail (other end closed).
- buffer[0] = 345678901;
- EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION,
- d0->WriteMessage(UserPointer<const void>(buffer), kBufferSize,
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE));
-
- EXPECT_EQ(MOJO_RESULT_OK, d0->Close());
- }
-}
-
-#if defined(OS_WIN)
-// http://crbug.com/396386
-#define MAYBE_BasicThreaded DISABLED_BasicThreaded
-#else
-#define MAYBE_BasicThreaded BasicThreaded
-#endif
-TEST(MessagePipeDispatcherTest, MAYBE_BasicThreaded) {
- test::Stopwatch stopwatch;
- int32_t buffer[1];
- const uint32_t kBufferSize = static_cast<uint32_t>(sizeof(buffer));
- uint32_t buffer_size;
- base::TimeDelta elapsed;
- bool did_wait;
- MojoResult result;
- uint32_t context;
- HandleSignalsState hss;
-
- // Run this test both with |d0| as port 0, |d1| as port 1 and vice versa.
- for (unsigned i = 0; i < 2; i++) {
- scoped_refptr<MessagePipeDispatcher> d0(new MessagePipeDispatcher(
- MessagePipeDispatcher::kDefaultCreateOptions));
- scoped_refptr<MessagePipeDispatcher> d1(new MessagePipeDispatcher(
- MessagePipeDispatcher::kDefaultCreateOptions));
- {
- scoped_refptr<MessagePipe> mp(MessagePipe::CreateLocalLocal());
- d0->Init(mp, i); // 0, 1.
- d1->Init(mp, i ^ 1); // 1, 0.
- }
-
- // Wait for readable on |d1|, which will become readable after some time.
- {
- test::WaiterThread thread(d1, MOJO_HANDLE_SIGNAL_READABLE,
- MOJO_DEADLINE_INDEFINITE, 1, &did_wait, &result,
- &context, &hss);
- stopwatch.Start();
- thread.Start();
- base::PlatformThread::Sleep(2 * test::EpsilonTimeout());
- // Wake it up by writing to |d0|.
- buffer[0] = 123456789;
- EXPECT_EQ(MOJO_RESULT_OK,
- d0->WriteMessage(UserPointer<const void>(buffer), kBufferSize,
- nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE));
- } // Joins the thread.
- elapsed = stopwatch.Elapsed();
- EXPECT_GT(elapsed, (2 - 1) * test::EpsilonTimeout());
- EXPECT_LT(elapsed, (2 + 1) * test::EpsilonTimeout());
- EXPECT_TRUE(did_wait);
- EXPECT_EQ(MOJO_RESULT_OK, result);
- EXPECT_EQ(1u, context);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE,
- hss.satisfied_signals);
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals);
-
- // Now |d1| is already readable. Try waiting for it again.
- {
- test::WaiterThread thread(d1, MOJO_HANDLE_SIGNAL_READABLE,
- MOJO_DEADLINE_INDEFINITE, 2, &did_wait, &result,
- &context, &hss);
- stopwatch.Start();
- thread.Start();
- } // Joins the thread.
- EXPECT_LT(stopwatch.Elapsed(), test::EpsilonTimeout());
- EXPECT_FALSE(did_wait);
- EXPECT_EQ(MOJO_RESULT_ALREADY_EXISTS, result);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE,
- hss.satisfied_signals);
- EXPECT_EQ(kAllSignals, hss.satisfiable_signals);
-
- // Consume what we wrote to |d0|.
- buffer[0] = 0;
- buffer_size = kBufferSize;
- EXPECT_EQ(MOJO_RESULT_OK,
- d1->ReadMessage(UserPointer<void>(buffer),
- MakeUserPointer(&buffer_size), 0, nullptr,
- MOJO_READ_MESSAGE_FLAG_NONE));
- EXPECT_EQ(kBufferSize, buffer_size);
- EXPECT_EQ(123456789, buffer[0]);
-
- // Wait for readable on |d1| and close |d0| after some time, which should
- // cancel that wait.
- {
- test::WaiterThread thread(d1, MOJO_HANDLE_SIGNAL_READABLE,
- MOJO_DEADLINE_INDEFINITE, 3, &did_wait, &result,
- &context, &hss);
- stopwatch.Start();
- thread.Start();
- base::PlatformThread::Sleep(2 * test::EpsilonTimeout());
- EXPECT_EQ(MOJO_RESULT_OK, d0->Close());
- } // Joins the thread.
- elapsed = stopwatch.Elapsed();
- EXPECT_GT(elapsed, (2 - 1) * test::EpsilonTimeout());
- EXPECT_LT(elapsed, (2 + 1) * test::EpsilonTimeout());
- EXPECT_TRUE(did_wait);
- EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, result);
- EXPECT_EQ(3u, context);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfied_signals);
- EXPECT_EQ(MOJO_HANDLE_SIGNAL_PEER_CLOSED, hss.satisfiable_signals);
-
- EXPECT_EQ(MOJO_RESULT_OK, d1->Close());
- }
-
- for (unsigned i = 0; i < 2; i++) {
- scoped_refptr<MessagePipeDispatcher> d0(new MessagePipeDispatcher(
- MessagePipeDispatcher::kDefaultCreateOptions));
- scoped_refptr<MessagePipeDispatcher> d1(new MessagePipeDispatcher(
- MessagePipeDispatcher::kDefaultCreateOptions));
- {
- scoped_refptr<MessagePipe> mp(MessagePipe::CreateLocalLocal());
- d0->Init(mp, i); // 0, 1.
- d1->Init(mp, i ^ 1); // 1, 0.
- }
-
- // Wait for readable on |d1| and close |d1| after some time, which should
- // cancel that wait.
- {
- test::WaiterThread thread(d1, MOJO_HANDLE_SIGNAL_READABLE,
- MOJO_DEADLINE_INDEFINITE, 4, &did_wait, &result,
- &context, &hss);
- stopwatch.Start();
- thread.Start();
- base::PlatformThread::Sleep(2 * test::EpsilonTimeout());
- EXPECT_EQ(MOJO_RESULT_OK, d1->Close());
- } // Joins the thread.
- elapsed = stopwatch.Elapsed();
- EXPECT_GT(elapsed, (2 - 1) * test::EpsilonTimeout());
- EXPECT_LT(elapsed, (2 + 1) * test::EpsilonTimeout());
- EXPECT_TRUE(did_wait);
- EXPECT_EQ(MOJO_RESULT_CANCELLED, result);
- EXPECT_EQ(4u, context);
- EXPECT_EQ(0u, hss.satisfied_signals);
- EXPECT_EQ(0u, hss.satisfiable_signals);
-
- EXPECT_EQ(MOJO_RESULT_OK, d0->Close());
- }
-}
-
-// Stress test -----------------------------------------------------------------
-
-const size_t kMaxMessageSize = 2000;
-
-class WriterThread : public base::SimpleThread {
- public:
- // |*messages_written| and |*bytes_written| belong to the thread while it's
- // alive.
- WriterThread(scoped_refptr<Dispatcher> write_dispatcher,
- size_t* messages_written,
- size_t* bytes_written)
- : base::SimpleThread("writer_thread"),
- write_dispatcher_(write_dispatcher),
- messages_written_(messages_written),
- bytes_written_(bytes_written) {
- *messages_written_ = 0;
- *bytes_written_ = 0;
- }
-
- ~WriterThread() override { Join(); }
-
- private:
- void Run() override {
- // Make some data to write.
- unsigned char buffer[kMaxMessageSize];
- for (size_t i = 0; i < kMaxMessageSize; i++)
- buffer[i] = static_cast<unsigned char>(i);
-
- // Number of messages to write.
- *messages_written_ = static_cast<size_t>(base::RandInt(1000, 6000));
-
- // Write messages.
- for (size_t i = 0; i < *messages_written_; i++) {
- uint32_t bytes_to_write = static_cast<uint32_t>(
- base::RandInt(1, static_cast<int>(kMaxMessageSize)));
- EXPECT_EQ(MOJO_RESULT_OK,
- write_dispatcher_->WriteMessage(UserPointer<const void>(buffer),
- bytes_to_write, nullptr,
- MOJO_WRITE_MESSAGE_FLAG_NONE));
- *bytes_written_ += bytes_to_write;
- }
-
- // Write one last "quit" message.
- EXPECT_EQ(MOJO_RESULT_OK, write_dispatcher_->WriteMessage(
- UserPointer<const void>("quit"), 4, nullptr,
- MOJO_WRITE_MESSAGE_FLAG_NONE));
- }
-
- const scoped_refptr<Dispatcher> write_dispatcher_;
- size_t* const messages_written_;
- size_t* const bytes_written_;
-
- DISALLOW_COPY_AND_ASSIGN(WriterThread);
-};
-
-class ReaderThread : public base::SimpleThread {
- public:
- // |*messages_read| and |*bytes_read| belong to the thread while it's alive.
- ReaderThread(scoped_refptr<Dispatcher> read_dispatcher,
- size_t* messages_read,
- size_t* bytes_read)
- : base::SimpleThread("reader_thread"),
- read_dispatcher_(read_dispatcher),
- messages_read_(messages_read),
- bytes_read_(bytes_read) {
- *messages_read_ = 0;
- *bytes_read_ = 0;
- }
-
- ~ReaderThread() override { Join(); }
-
- private:
- void Run() override {
- unsigned char buffer[kMaxMessageSize];
- Waiter w;
- HandleSignalsState hss;
- MojoResult result;
-
- // Read messages.
- for (;;) {
- // Wait for it to be readable.
- w.Init();
- hss = HandleSignalsState();
- result = read_dispatcher_->AddAwakable(&w, MOJO_HANDLE_SIGNAL_READABLE, 0,
- &hss);
- EXPECT_TRUE(result == MOJO_RESULT_OK ||
- result == MOJO_RESULT_ALREADY_EXISTS)
- << "result: " << result;
- if (result == MOJO_RESULT_OK) {
- // Actually need to wait.
- EXPECT_EQ(MOJO_RESULT_OK, w.Wait(MOJO_DEADLINE_INDEFINITE, nullptr));
- read_dispatcher_->RemoveAwakable(&w, &hss);
- }
- // We may not actually be readable, since we're racing with other threads.
- EXPECT_TRUE((hss.satisfiable_signals & MOJO_HANDLE_SIGNAL_READABLE));
-
- // Now, try to do the read.
- // Clear the buffer so that we can check the result.
- memset(buffer, 0, sizeof(buffer));
- uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer));
- result = read_dispatcher_->ReadMessage(
- UserPointer<void>(buffer), MakeUserPointer(&buffer_size), 0, nullptr,
- MOJO_READ_MESSAGE_FLAG_NONE);
- EXPECT_TRUE(result == MOJO_RESULT_OK || result == MOJO_RESULT_SHOULD_WAIT)
- << "result: " << result;
- // We're racing with others to read, so maybe we failed.
- if (result == MOJO_RESULT_SHOULD_WAIT)
- continue; // In which case, try again.
- // Check for quit.
- if (buffer_size == 4 && memcmp("quit", buffer, 4) == 0)
- return;
- EXPECT_GE(buffer_size, 1u);
- EXPECT_LE(buffer_size, kMaxMessageSize);
- EXPECT_TRUE(IsValidMessage(buffer, buffer_size));
-
- (*messages_read_)++;
- *bytes_read_ += buffer_size;
- }
- }
-
- static bool IsValidMessage(const unsigned char* buffer,
- uint32_t message_size) {
- size_t i;
- for (i = 0; i < message_size; i++) {
- if (buffer[i] != static_cast<unsigned char>(i))
- return false;
- }
- // Check that the remaining bytes weren't stomped on.
- for (; i < kMaxMessageSize; i++) {
- if (buffer[i] != 0)
- return false;
- }
- return true;
- }
-
- const scoped_refptr<Dispatcher> read_dispatcher_;
- size_t* const messages_read_;
- size_t* const bytes_read_;
-
- DISALLOW_COPY_AND_ASSIGN(ReaderThread);
-};
-
-TEST(MessagePipeDispatcherTest, Stress) {
- static const size_t kNumWriters = 30;
- static const size_t kNumReaders = kNumWriters;
-
- scoped_refptr<MessagePipeDispatcher> d_write(
- new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions));
- scoped_refptr<MessagePipeDispatcher> d_read(
- new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions));
- {
- scoped_refptr<MessagePipe> mp(MessagePipe::CreateLocalLocal());
- d_write->Init(mp, 0);
- d_read->Init(mp, 1);
- }
-
- size_t messages_written[kNumWriters];
- size_t bytes_written[kNumWriters];
- size_t messages_read[kNumReaders];
- size_t bytes_read[kNumReaders];
- {
- // Make writers.
- ScopedVector<WriterThread> writers;
- for (size_t i = 0; i < kNumWriters; i++) {
- writers.push_back(
- new WriterThread(d_write, &messages_written[i], &bytes_written[i]));
- }
-
- // Make readers.
- ScopedVector<ReaderThread> readers;
- for (size_t i = 0; i < kNumReaders; i++) {
- readers.push_back(
- new ReaderThread(d_read, &messages_read[i], &bytes_read[i]));
- }
-
- // Start writers.
- for (size_t i = 0; i < kNumWriters; i++)
- writers[i]->Start();
-
- // Start readers.
- for (size_t i = 0; i < kNumReaders; i++)
- readers[i]->Start();
-
- // TODO(vtl): Maybe I should have an event that triggers all the threads to
- // start doing stuff for real (so that the first ones created/started aren't
- // advantaged).
- } // Joins all the threads.
-
- size_t total_messages_written = 0;
- size_t total_bytes_written = 0;
- for (size_t i = 0; i < kNumWriters; i++) {
- total_messages_written += messages_written[i];
- total_bytes_written += bytes_written[i];
- }
- size_t total_messages_read = 0;
- size_t total_bytes_read = 0;
- for (size_t i = 0; i < kNumReaders; i++) {
- total_messages_read += messages_read[i];
- total_bytes_read += bytes_read[i];
- // We'd have to be really unlucky to have read no messages on a thread.
- EXPECT_GT(messages_read[i], 0u) << "reader: " << i;
- EXPECT_GE(bytes_read[i], messages_read[i]) << "reader: " << i;
- }
- EXPECT_EQ(total_messages_written, total_messages_read);
- EXPECT_EQ(total_bytes_written, total_bytes_read);
-
- EXPECT_EQ(MOJO_RESULT_OK, d_write->Close());
- EXPECT_EQ(MOJO_RESULT_OK, d_read->Close());
-}
-
-} // namespace
-} // namespace system
-} // namespace mojo
« no previous file with comments | « mojo/edk/system/message_pipe_dispatcher.cc ('k') | mojo/edk/system/message_pipe_endpoint.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698