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 |