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 <stddef.h>
6#include <stdint.h>
7
8#include <string>
9#include <utility>
10
11#include "base/at_exit.h"
12#include "base/files/file_path.h"
13#include "base/files/file_util.h"
14#include "base/macros.h"
15#include "base/message_loop/message_loop.h"
16#include "base/run_loop.h"
17#include "base/strings/utf_string_conversions.h"
18#include "gin/array_buffer.h"
19#include "gin/public/isolate_holder.h"
20#include "gin/v8_initializer.h"
21#include "mojo/common/data_pipe_utils.h"
22#include "mojo/edk/js/mojo_runner_delegate.h"
23#include "mojo/edk/js/tests/js_to_cpp.mojom.h"
24#include "mojo/edk/test/test_utils.h"
25#include "mojo/public/cpp/bindings/binding.h"
26#include "mojo/public/cpp/system/core.h"
27#include "testing/gtest/include/gtest/gtest.h"
28
29namespace mojo {
30namespace edk {
31namespace js {
32
33// Global value updated by some checks to prevent compilers from optimizing
34// reads out of existence.
35uint32_t g_waste_accumulator = 0;
36
37namespace {
38
39// Negative numbers with different values in each byte, the last of
40// which can survive promotion to double and back.
41const int8_t kExpectedInt8Value = -65;
42const int16_t kExpectedInt16Value = -16961;
43const int32_t kExpectedInt32Value = -1145258561;
44const int64_t kExpectedInt64Value = -77263311946305LL;
45
46// Positive numbers with different values in each byte, the last of
47// which can survive promotion to double and back.
48const uint8_t kExpectedUInt8Value = 65;
49const uint16_t kExpectedUInt16Value = 16961;
50const uint32_t kExpectedUInt32Value = 1145258561;
51const uint64_t kExpectedUInt64Value = 77263311946305LL;
52
53// Double/float values, including special case constants.
54const double kExpectedDoubleVal = 3.14159265358979323846;
55const double kExpectedDoubleInf = std::numeric_limits<double>::infinity();
56const double kExpectedDoubleNan = std::numeric_limits<double>::quiet_NaN();
57const float kExpectedFloatVal = static_cast<float>(kExpectedDoubleVal);
58const float kExpectedFloatInf = std::numeric_limits<float>::infinity();
59const float kExpectedFloatNan = std::numeric_limits<float>::quiet_NaN();
60
61// NaN has the property that it is not equal to itself.
62#define EXPECT_NAN(x) EXPECT_NE(x, x)
63
64void CheckDataPipe(ScopedDataPipeConsumerHandle data_pipe_handle) {
65  std::string buffer;
66  bool result = common::BlockingCopyToString(std::move(data_pipe_handle),
67                                             &buffer);
68  EXPECT_TRUE(result);
69  EXPECT_EQ(64u, buffer.size());
70  for (int i = 0; i < 64; ++i) {
71    EXPECT_EQ(i, buffer[i]);
72  }
73}
74
75void CheckMessagePipe(MessagePipeHandle message_pipe_handle) {
76  unsigned char buffer[100];
77  uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer));
78  MojoResult result = Wait(
79      message_pipe_handle, MOJO_HANDLE_SIGNAL_READABLE,
80      MOJO_DEADLINE_INDEFINITE, nullptr);
81  EXPECT_EQ(MOJO_RESULT_OK, result);
82  result = ReadMessageRaw(
83      message_pipe_handle, buffer, &buffer_size, 0, 0, 0);
84  EXPECT_EQ(MOJO_RESULT_OK, result);
85  EXPECT_EQ(64u, buffer_size);
86  for (int i = 0; i < 64; ++i) {
87    EXPECT_EQ(255 - i, buffer[i]);
88  }
89}
90
91js_to_cpp::EchoArgsPtr BuildSampleEchoArgs() {
92  js_to_cpp::EchoArgsPtr args(js_to_cpp::EchoArgs::New());
93  args->si64 = kExpectedInt64Value;
94  args->si32 = kExpectedInt32Value;
95  args->si16 = kExpectedInt16Value;
96  args->si8 = kExpectedInt8Value;
97  args->ui64 = kExpectedUInt64Value;
98  args->ui32 = kExpectedUInt32Value;
99  args->ui16 = kExpectedUInt16Value;
100  args->ui8 = kExpectedUInt8Value;
101  args->float_val = kExpectedFloatVal;
102  args->float_inf = kExpectedFloatInf;
103  args->float_nan = kExpectedFloatNan;
104  args->double_val = kExpectedDoubleVal;
105  args->double_inf = kExpectedDoubleInf;
106  args->double_nan = kExpectedDoubleNan;
107  args->name = "coming";
108  Array<String> string_array(3);
109  string_array[0] = "one";
110  string_array[1] = "two";
111  string_array[2] = "three";
112  args->string_array = std::move(string_array);
113  return args;
114}
115
116void CheckSampleEchoArgs(js_to_cpp::EchoArgsPtr arg) {
117  EXPECT_EQ(kExpectedInt64Value, arg->si64);
118  EXPECT_EQ(kExpectedInt32Value, arg->si32);
119  EXPECT_EQ(kExpectedInt16Value, arg->si16);
120  EXPECT_EQ(kExpectedInt8Value, arg->si8);
121  EXPECT_EQ(kExpectedUInt64Value, arg->ui64);
122  EXPECT_EQ(kExpectedUInt32Value, arg->ui32);
123  EXPECT_EQ(kExpectedUInt16Value, arg->ui16);
124  EXPECT_EQ(kExpectedUInt8Value, arg->ui8);
125  EXPECT_EQ(kExpectedFloatVal, arg->float_val);
126  EXPECT_EQ(kExpectedFloatInf, arg->float_inf);
127  EXPECT_NAN(arg->float_nan);
128  EXPECT_EQ(kExpectedDoubleVal, arg->double_val);
129  EXPECT_EQ(kExpectedDoubleInf, arg->double_inf);
130  EXPECT_NAN(arg->double_nan);
131  EXPECT_EQ(std::string("coming"), arg->name.get());
132  EXPECT_EQ(std::string("one"), arg->string_array[0].get());
133  EXPECT_EQ(std::string("two"), arg->string_array[1].get());
134  EXPECT_EQ(std::string("three"), arg->string_array[2].get());
135  CheckDataPipe(std::move(arg->data_handle));
136  CheckMessagePipe(arg->message_handle.get());
137}
138
139void CheckSampleEchoArgsList(const js_to_cpp::EchoArgsListPtr& list) {
140  if (list.is_null())
141    return;
142  CheckSampleEchoArgs(std::move(list->item));
143  CheckSampleEchoArgsList(list->next);
144}
145
146// More forgiving checks are needed in the face of potentially corrupt
147// messages. The values don't matter so long as all accesses are within
148// bounds.
149void CheckCorruptedString(const String& arg) {
150  if (arg.is_null())
151    return;
152  for (size_t i = 0; i < arg.size(); ++i)
153    g_waste_accumulator += arg[i];
154}
155
156void CheckCorruptedStringArray(const Array<String>& string_array) {
157  if (string_array.is_null())
158    return;
159  for (size_t i = 0; i < string_array.size(); ++i)
160    CheckCorruptedString(string_array[i]);
161}
162
163void CheckCorruptedDataPipe(MojoHandle data_pipe_handle) {
164  unsigned char buffer[100];
165  uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer));
166  MojoResult result = MojoReadData(
167      data_pipe_handle, buffer, &buffer_size, MOJO_READ_DATA_FLAG_NONE);
168  if (result != MOJO_RESULT_OK)
169    return;
170  for (uint32_t i = 0; i < buffer_size; ++i)
171    g_waste_accumulator += buffer[i];
172}
173
174void CheckCorruptedMessagePipe(MojoHandle message_pipe_handle) {
175  unsigned char buffer[100];
176  uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer));
177  MojoResult result = MojoReadMessage(
178      message_pipe_handle, buffer, &buffer_size, 0, 0, 0);
179  if (result != MOJO_RESULT_OK)
180    return;
181  for (uint32_t i = 0; i < buffer_size; ++i)
182    g_waste_accumulator += buffer[i];
183}
184
185void CheckCorruptedEchoArgs(const js_to_cpp::EchoArgsPtr& arg) {
186  if (arg.is_null())
187    return;
188  CheckCorruptedString(arg->name);
189  CheckCorruptedStringArray(arg->string_array);
190  if (arg->data_handle.is_valid())
191    CheckCorruptedDataPipe(arg->data_handle.get().value());
192  if (arg->message_handle.is_valid())
193    CheckCorruptedMessagePipe(arg->message_handle.get().value());
194}
195
196void CheckCorruptedEchoArgsList(const js_to_cpp::EchoArgsListPtr& list) {
197  if (list.is_null())
198    return;
199  CheckCorruptedEchoArgs(list->item);
200  CheckCorruptedEchoArgsList(list->next);
201}
202
203// Base Provider implementation class. It's expected that tests subclass and
204// override the appropriate Provider functions. When test is done quit the
205// run_loop().
206class CppSideConnection : public js_to_cpp::CppSide {
207 public:
208  CppSideConnection()
209      : run_loop_(nullptr),
210        js_side_(nullptr),
211        mishandled_messages_(0),
212        binding_(this) {}
213  ~CppSideConnection() override {}
214
215  void set_run_loop(base::RunLoop* run_loop) { run_loop_ = run_loop; }
216  base::RunLoop* run_loop() { return run_loop_; }
217
218  void set_js_side(js_to_cpp::JsSide* js_side) { js_side_ = js_side; }
219  js_to_cpp::JsSide* js_side() { return js_side_; }
220
221  void Bind(InterfaceRequest<js_to_cpp::CppSide> request) {
222    binding_.Bind(std::move(request));
223    // Keep the pipe open even after validation errors.
224    binding_.EnableTestingMode();
225  }
226
227  // js_to_cpp::CppSide:
228  void StartTest() override { NOTREACHED(); }
229
230  void TestFinished() override { NOTREACHED(); }
231
232  void PingResponse() override { mishandled_messages_ += 1; }
233
234  void EchoResponse(js_to_cpp::EchoArgsListPtr list) override {
235    mishandled_messages_ += 1;
236  }
237
238  void BitFlipResponse(js_to_cpp::EchoArgsListPtr list) override {
239    mishandled_messages_ += 1;
240  }
241
242  void BackPointerResponse(js_to_cpp::EchoArgsListPtr list) override {
243    mishandled_messages_ += 1;
244  }
245
246 protected:
247  base::RunLoop* run_loop_;
248  js_to_cpp::JsSide* js_side_;
249  int mishandled_messages_;
250  mojo::Binding<js_to_cpp::CppSide> binding_;
251
252 private:
253  DISALLOW_COPY_AND_ASSIGN(CppSideConnection);
254};
255
256// Trivial test to verify a message sent from JS is received.
257class PingCppSideConnection : public CppSideConnection {
258 public:
259  PingCppSideConnection() : got_message_(false) {}
260  ~PingCppSideConnection() override {}
261
262  // js_to_cpp::CppSide:
263  void StartTest() override { js_side_->Ping(); }
264
265  void PingResponse() override {
266    got_message_ = true;
267    run_loop()->Quit();
268  }
269
270  bool DidSucceed() {
271    return got_message_ && !mishandled_messages_;
272  }
273
274 private:
275  bool got_message_;
276  DISALLOW_COPY_AND_ASSIGN(PingCppSideConnection);
277};
278
279// Test that parameters are passed with correct values.
280class EchoCppSideConnection : public CppSideConnection {
281 public:
282  EchoCppSideConnection() :
283      message_count_(0),
284      termination_seen_(false) {
285  }
286  ~EchoCppSideConnection() override {}
287
288  // js_to_cpp::CppSide:
289  void StartTest() override {
290    js_side_->Echo(kExpectedMessageCount, BuildSampleEchoArgs());
291  }
292
293  void EchoResponse(js_to_cpp::EchoArgsListPtr list) override {
294    const js_to_cpp::EchoArgsPtr& special_arg = list->item;
295    message_count_ += 1;
296    EXPECT_EQ(-1, special_arg->si64);
297    EXPECT_EQ(-1, special_arg->si32);
298    EXPECT_EQ(-1, special_arg->si16);
299    EXPECT_EQ(-1, special_arg->si8);
300    EXPECT_EQ(std::string("going"), special_arg->name.To<std::string>());
301    CheckSampleEchoArgsList(list->next);
302  }
303
304  void TestFinished() override {
305    termination_seen_ = true;
306    run_loop()->Quit();
307  }
308
309  bool DidSucceed() {
310    return termination_seen_ &&
311        !mishandled_messages_ &&
312        message_count_ == kExpectedMessageCount;
313  }
314
315 private:
316  static const int kExpectedMessageCount = 10;
317  int message_count_;
318  bool termination_seen_;
319  DISALLOW_COPY_AND_ASSIGN(EchoCppSideConnection);
320};
321
322// Test that corrupted messages don't wreak havoc.
323class BitFlipCppSideConnection : public CppSideConnection {
324 public:
325  BitFlipCppSideConnection() : termination_seen_(false) {}
326  ~BitFlipCppSideConnection() override {}
327
328  // js_to_cpp::CppSide:
329  void StartTest() override { js_side_->BitFlip(BuildSampleEchoArgs()); }
330
331  void BitFlipResponse(js_to_cpp::EchoArgsListPtr list) override {
332    CheckCorruptedEchoArgsList(list);
333  }
334
335  void TestFinished() override {
336    termination_seen_ = true;
337    run_loop()->Quit();
338  }
339
340  bool DidSucceed() {
341    return termination_seen_;
342  }
343
344 private:
345  bool termination_seen_;
346  DISALLOW_COPY_AND_ASSIGN(BitFlipCppSideConnection);
347};
348
349// Test that severely random messages don't wreak havoc.
350class BackPointerCppSideConnection : public CppSideConnection {
351 public:
352  BackPointerCppSideConnection() : termination_seen_(false) {}
353  ~BackPointerCppSideConnection() override {}
354
355  // js_to_cpp::CppSide:
356  void StartTest() override { js_side_->BackPointer(BuildSampleEchoArgs()); }
357
358  void BackPointerResponse(js_to_cpp::EchoArgsListPtr list) override {
359    CheckCorruptedEchoArgsList(list);
360  }
361
362  void TestFinished() override {
363    termination_seen_ = true;
364    run_loop()->Quit();
365  }
366
367  bool DidSucceed() {
368    return termination_seen_;
369  }
370
371 private:
372  bool termination_seen_;
373  DISALLOW_COPY_AND_ASSIGN(BackPointerCppSideConnection);
374};
375
376}  // namespace
377
378class JsToCppTest : public testing::Test {
379 public:
380  JsToCppTest() {}
381
382  void RunTest(const std::string& test, CppSideConnection* cpp_side) {
383    cpp_side->set_run_loop(&run_loop_);
384
385    js_to_cpp::JsSidePtr js_side;
386    auto js_side_proxy = GetProxy(&js_side);
387
388    cpp_side->set_js_side(js_side.get());
389    js_to_cpp::CppSidePtr cpp_side_ptr;
390    cpp_side->Bind(GetProxy(&cpp_side_ptr));
391
392    js_side->SetCppSide(std::move(cpp_side_ptr));
393
394#ifdef V8_USE_EXTERNAL_STARTUP_DATA
395    gin::V8Initializer::LoadV8Snapshot();
396    gin::V8Initializer::LoadV8Natives();
397#endif
398
399    gin::IsolateHolder::Initialize(gin::IsolateHolder::kStrictMode,
400                                   gin::IsolateHolder::kStableV8Extras,
401                                   gin::ArrayBufferAllocator::SharedInstance());
402    gin::IsolateHolder instance;
403    MojoRunnerDelegate delegate;
404    gin::ShellRunner runner(&delegate, instance.isolate());
405    delegate.Start(&runner, js_side_proxy.PassMessagePipe().release().value(),
406                   test);
407
408    run_loop_.Run();
409  }
410
411 private:
412  base::ShadowingAtExitManager at_exit_;
413  base::MessageLoop loop;
414  base::RunLoop run_loop_;
415
416  DISALLOW_COPY_AND_ASSIGN(JsToCppTest);
417};
418
419TEST_F(JsToCppTest, Ping) {
420  PingCppSideConnection cpp_side_connection;
421  RunTest("mojo/edk/js/tests/js_to_cpp_tests", &cpp_side_connection);
422  EXPECT_TRUE(cpp_side_connection.DidSucceed());
423}
424
425TEST_F(JsToCppTest, Echo) {
426  EchoCppSideConnection cpp_side_connection;
427  RunTest("mojo/edk/js/tests/js_to_cpp_tests", &cpp_side_connection);
428  EXPECT_TRUE(cpp_side_connection.DidSucceed());
429}
430
431TEST_F(JsToCppTest, BitFlip) {
432  BitFlipCppSideConnection cpp_side_connection;
433  RunTest("mojo/edk/js/tests/js_to_cpp_tests", &cpp_side_connection);
434  EXPECT_TRUE(cpp_side_connection.DidSucceed());
435}
436
437TEST_F(JsToCppTest, BackPointer) {
438  BackPointerCppSideConnection cpp_side_connection;
439  RunTest("mojo/edk/js/tests/js_to_cpp_tests", &cpp_side_connection);
440  EXPECT_TRUE(cpp_side_connection.DidSucceed());
441}
442
443}  // namespace js
444}  // namespace edk
445}  // namespace mojo
446