Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 <limits.h> | 5 #include <limits.h> |
| 6 | 6 |
| 7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
| 8 #include "base/files/memory_mapped_file.h" | 8 #include "base/files/memory_mapped_file.h" |
| 9 #include "base/logging.h" | 9 #include "base/logging.h" |
| 10 #include "base/strings/string_piece.h" | 10 #include "base/strings/string_piece.h" |
| (...skipping 90 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 101 } | 101 } |
| 102 const char* data = reinterpret_cast<const char*>(mapped_file_.data()); | 102 const char* data = reinterpret_cast<const char*>(mapped_file_.data()); |
| 103 file_data_.set(data, mapped_file_.length()); | 103 file_data_.set(data, mapped_file_.length()); |
| 104 return true; | 104 return true; |
| 105 } | 105 } |
| 106 | 106 |
| 107 bool Reader::ReadMessages() { | 107 bool Reader::ReadMessages() { |
| 108 for (size_t i = 0; i < header_->message_count; ++i) { | 108 for (size_t i = 0; i < header_->message_count; ++i) { |
| 109 const char* begin = file_data_.begin(); | 109 const char* begin = file_data_.begin(); |
| 110 const char* end = file_data_.end(); | 110 const char* end = file_data_.end(); |
| 111 const char* message_tail = IPC::Message::FindNext(begin, end); | 111 IPC::Message::NextMessageInfo info = IPC::Message::FindNext(begin, end); |
|
Tom Sepez
2015/09/01 18:01:54
nit: the assignment may not be cheap; a copy of th
erikchen
2015/09/04 02:13:39
I changed the signature of FindNext to take a Next
| |
| 112 if (!message_tail) { | 112 if (!info.message_found) { |
| 113 LOG(ERROR) << "Failed to parse message."; | 113 LOG(ERROR) << "Failed to parse message."; |
| 114 return false; | 114 return false; |
| 115 } | 115 } |
| 116 | 116 |
| 117 size_t msglen = message_tail - begin; | 117 CHECK_EQ(info.message_end, info.pickle_end); |
| 118 size_t msglen = info.message_end - begin; | |
| 118 if (msglen > INT_MAX) { | 119 if (msglen > INT_MAX) { |
| 119 LOG(ERROR) << "Message too large."; | 120 LOG(ERROR) << "Message too large."; |
| 120 return false; | 121 return false; |
| 121 } | 122 } |
| 122 | 123 |
| 123 // Copy is necessary to fix message type later. | 124 // Copy is necessary to fix message type later. |
| 124 IPC::Message const_message(begin, msglen); | 125 IPC::Message const_message(begin, msglen); |
| 125 IPC::Message* message = new IPC::Message(const_message); | 126 IPC::Message* message = new IPC::Message(const_message); |
| 126 messages_->push_back(message); | 127 messages_->push_back(message); |
| 127 file_data_.remove_prefix(msglen); | 128 file_data_.remove_prefix(msglen); |
| (...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 221 } | 222 } |
| 222 | 223 |
| 223 } // namespace | 224 } // namespace |
| 224 | 225 |
| 225 bool MessageFile::Read(const base::FilePath& path, MessageVector* messages) { | 226 bool MessageFile::Read(const base::FilePath& path, MessageVector* messages) { |
| 226 Reader reader(path); | 227 Reader reader(path); |
| 227 return reader.Read(messages); | 228 return reader.Read(messages); |
| 228 } | 229 } |
| 229 | 230 |
| 230 } // namespace ipc_fuzzer | 231 } // namespace ipc_fuzzer |
| OLD | NEW |