| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "mojo/public/cpp/bindings/lib/pipe_control_message_handler.h" | 5 #include "mojo/public/cpp/bindings/lib/pipe_control_message_handler.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "mojo/public/cpp/bindings/lib/message_builder.h" | 8 #include "mojo/public/cpp/bindings/lib/message_builder.h" |
| 9 #include "mojo/public/cpp/bindings/lib/pipe_control_message_handler_delegate.h" | 9 #include "mojo/public/cpp/bindings/lib/pipe_control_message_handler_delegate.h" |
| 10 #include "mojo/public/cpp/bindings/lib/serialization.h" | 10 #include "mojo/public/cpp/bindings/lib/serialization.h" |
| 11 #include "mojo/public/cpp/bindings/lib/validation_context.h" |
| 11 #include "mojo/public/cpp/bindings/lib/validation_util.h" | 12 #include "mojo/public/cpp/bindings/lib/validation_util.h" |
| 12 #include "mojo/public/interfaces/bindings/pipe_control_messages.mojom.h" | 13 #include "mojo/public/interfaces/bindings/pipe_control_messages.mojom.h" |
| 13 | 14 |
| 14 namespace mojo { | 15 namespace mojo { |
| 15 namespace internal { | 16 namespace internal { |
| 16 | 17 |
| 17 PipeControlMessageHandler::PipeControlMessageHandler( | 18 PipeControlMessageHandler::PipeControlMessageHandler( |
| 18 PipeControlMessageHandlerDelegate* delegate) | 19 PipeControlMessageHandlerDelegate* delegate) |
| 19 : delegate_(delegate) {} | 20 : delegate_(delegate) {} |
| 20 | 21 |
| 21 PipeControlMessageHandler::~PipeControlMessageHandler() {} | 22 PipeControlMessageHandler::~PipeControlMessageHandler() {} |
| 22 | 23 |
| 24 void PipeControlMessageHandler::SetDescription(const std::string& description) { |
| 25 description_ = description; |
| 26 } |
| 27 |
| 23 // static | 28 // static |
| 24 bool PipeControlMessageHandler::IsPipeControlMessage(const Message* message) { | 29 bool PipeControlMessageHandler::IsPipeControlMessage(const Message* message) { |
| 25 return !IsValidInterfaceId(message->interface_id()); | 30 return !IsValidInterfaceId(message->interface_id()); |
| 26 } | 31 } |
| 27 | 32 |
| 28 bool PipeControlMessageHandler::Accept(Message* message) { | 33 bool PipeControlMessageHandler::Accept(Message* message) { |
| 29 if (!Validate(message)) | 34 if (!Validate(message)) |
| 30 return false; | 35 return false; |
| 31 | 36 |
| 32 if (message->name() == pipe_control::kRunOrClosePipeMessageId) | 37 if (message->name() == pipe_control::kRunOrClosePipeMessageId) |
| 33 return RunOrClosePipe(message); | 38 return RunOrClosePipe(message); |
| 34 | 39 |
| 35 NOTREACHED(); | 40 NOTREACHED(); |
| 36 return false; | 41 return false; |
| 37 } | 42 } |
| 38 | 43 |
| 39 bool PipeControlMessageHandler::Validate(const Message* message) { | 44 bool PipeControlMessageHandler::Validate(Message* message) { |
| 45 ValidationContext validation_context( |
| 46 message->data(), message->data_num_bytes(), 0, message, description_); |
| 47 |
| 40 if (message->name() == pipe_control::kRunOrClosePipeMessageId) { | 48 if (message->name() == pipe_control::kRunOrClosePipeMessageId) { |
| 41 if (!ValidateMessageIsRequestWithoutResponse(message)) | 49 if (!ValidateMessageIsRequestWithoutResponse(message, &validation_context)) |
| 42 return false; | 50 return false; |
| 43 return ValidateMessagePayload< | 51 return ValidateMessagePayload< |
| 44 pipe_control::internal::RunOrClosePipeMessageParams_Data>(message); | 52 pipe_control::internal::RunOrClosePipeMessageParams_Data>( |
| 53 message, &validation_context); |
| 45 } | 54 } |
| 46 | 55 |
| 47 return false; | 56 return false; |
| 48 } | 57 } |
| 49 | 58 |
| 50 bool PipeControlMessageHandler::RunOrClosePipe(Message* message) { | 59 bool PipeControlMessageHandler::RunOrClosePipe(Message* message) { |
| 51 pipe_control::internal::RunOrClosePipeMessageParams_Data* params = | 60 pipe_control::internal::RunOrClosePipeMessageParams_Data* params = |
| 52 reinterpret_cast< | 61 reinterpret_cast< |
| 53 pipe_control::internal::RunOrClosePipeMessageParams_Data*>( | 62 pipe_control::internal::RunOrClosePipeMessageParams_Data*>( |
| 54 message->mutable_payload()); | 63 message->mutable_payload()); |
| (...skipping 13 matching lines...) Expand all Loading... |
| 68 ->id); | 77 ->id); |
| 69 } | 78 } |
| 70 | 79 |
| 71 DVLOG(1) << "Unsupported command in a RunOrClosePipe message pipe control " | 80 DVLOG(1) << "Unsupported command in a RunOrClosePipe message pipe control " |
| 72 << "message. Closing the pipe."; | 81 << "message. Closing the pipe."; |
| 73 return false; | 82 return false; |
| 74 } | 83 } |
| 75 | 84 |
| 76 } // namespace internal | 85 } // namespace internal |
| 77 } // namespace mojo | 86 } // namespace mojo |
| OLD | NEW |