OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <stdio.h> | 5 #include <stdio.h> |
6 #include <string> | 6 #include <string> |
7 #include <sstream> | 7 #include <sstream> |
8 | 8 |
9 #include "base/message_loop.h" | 9 #include "base/message_loop.h" |
10 #include "base/process_util.h" | 10 #include "base/process_util.h" |
(...skipping 337 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
348 | 348 |
349 EXPECT_TRUE(base::WaitForSingleProcess(server_process, 5000)); | 349 EXPECT_TRUE(base::WaitForSingleProcess(server_process, 5000)); |
350 base::CloseProcessHandle(server_process); | 350 base::CloseProcessHandle(server_process); |
351 } | 351 } |
352 | 352 |
353 // This class is for testing the IPC_BEGIN_MESSAGE_MAP_EX macros. | 353 // This class is for testing the IPC_BEGIN_MESSAGE_MAP_EX macros. |
354 class ServerMacroExTest { | 354 class ServerMacroExTest { |
355 public: | 355 public: |
356 ServerMacroExTest() : unhandled_msgs_(0) { | 356 ServerMacroExTest() : unhandled_msgs_(0) { |
357 } | 357 } |
| 358 |
358 virtual ~ServerMacroExTest() { | 359 virtual ~ServerMacroExTest() { |
359 } | 360 } |
| 361 |
360 virtual bool OnMessageReceived(const IPC::Message& msg) { | 362 virtual bool OnMessageReceived(const IPC::Message& msg) { |
361 bool msg_is_ok = false; | 363 bool msg_is_ok = false; |
362 IPC_BEGIN_MESSAGE_MAP_EX(ServerMacroExTest, msg, msg_is_ok) | 364 IPC_BEGIN_MESSAGE_MAP_EX(ServerMacroExTest, msg, msg_is_ok) |
363 IPC_MESSAGE_HANDLER(MsgClassIS, OnMsgClassISMessage) | 365 IPC_MESSAGE_HANDLER(MsgClassIS, OnMsgClassISMessage) |
364 IPC_MESSAGE_HANDLER(MsgClassSI, OnMsgClassSIMessage) | 366 IPC_MESSAGE_HANDLER(MsgClassSI, OnMsgClassSIMessage) |
365 IPC_MESSAGE_UNHANDLED(++unhandled_msgs_) | 367 IPC_MESSAGE_UNHANDLED(++unhandled_msgs_) |
366 IPC_END_MESSAGE_MAP_EX() | 368 IPC_END_MESSAGE_MAP_EX() |
367 return msg_is_ok; | 369 return msg_is_ok; |
368 } | 370 } |
369 | 371 |
370 int unhandled_msgs() const { | 372 int unhandled_msgs() const { |
371 return unhandled_msgs_; | 373 return unhandled_msgs_; |
372 } | 374 } |
373 | 375 |
374 private: | 376 private: |
375 void OnMsgClassISMessage(int value, const std::wstring& text) { | 377 void OnMsgClassISMessage(int value, const std::wstring& text) { |
376 } | 378 } |
377 void OnMsgClassSIMessage(const std::wstring& text, int value) { | 379 void OnMsgClassSIMessage(const std::wstring& text, int value) { |
378 } | 380 } |
379 | 381 |
380 int unhandled_msgs_; | 382 int unhandled_msgs_; |
| 383 |
| 384 DISALLOW_COPY_AND_ASSIGN(ServerMacroExTest); |
381 }; | 385 }; |
382 | 386 |
383 TEST_F(IPCFuzzingTest, MsgMapExMacro) { | 387 TEST_F(IPCFuzzingTest, MsgMapExMacro) { |
384 IPC::Message* msg = NULL; | 388 IPC::Message* msg = NULL; |
385 ServerMacroExTest server; | 389 ServerMacroExTest server; |
386 | 390 |
387 // Test the regular messages. | 391 // Test the regular messages. |
388 msg = new MsgClassIS(3, L"text3"); | 392 msg = new MsgClassIS(3, L"text3"); |
389 EXPECT_TRUE(server.OnMessageReceived(*msg)); | 393 EXPECT_TRUE(server.OnMessageReceived(*msg)); |
390 delete msg; | 394 delete msg; |
(...skipping 12 matching lines...) Expand all Loading... |
403 msg = new IPC::Message(MSG_ROUTING_CONTROL, MsgClassIS::ID, | 407 msg = new IPC::Message(MSG_ROUTING_CONTROL, MsgClassIS::ID, |
404 IPC::Message::PRIORITY_NORMAL); | 408 IPC::Message::PRIORITY_NORMAL); |
405 msg->WriteInt(0x64); | 409 msg->WriteInt(0x64); |
406 msg->WriteInt(0x32); | 410 msg->WriteInt(0x32); |
407 EXPECT_FALSE(server.OnMessageReceived(*msg)); | 411 EXPECT_FALSE(server.OnMessageReceived(*msg)); |
408 delete msg; | 412 delete msg; |
409 | 413 |
410 EXPECT_EQ(0, server.unhandled_msgs()); | 414 EXPECT_EQ(0, server.unhandled_msgs()); |
411 #endif | 415 #endif |
412 } | 416 } |
OLD | NEW |