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 "mojo/edk/system/message_in_transit.h" | 5 #include "mojo/edk/system/message_in_transit.h" |
6 | 6 |
7 #include <string.h> | 7 #include <string.h> |
8 | 8 |
9 #include "base/compiler_specific.h" | 9 #include "base/compiler_specific.h" |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
(...skipping 25 matching lines...) Expand all Loading... |
36 | 36 |
37 struct MessageInTransit::PrivateStructForCompileAsserts { | 37 struct MessageInTransit::PrivateStructForCompileAsserts { |
38 // The size of |Header| must be a multiple of the alignment. | 38 // The size of |Header| must be a multiple of the alignment. |
39 static_assert(sizeof(Header) % kMessageAlignment == 0, | 39 static_assert(sizeof(Header) % kMessageAlignment == 0, |
40 "sizeof(MessageInTransit::Header) invalid"); | 40 "sizeof(MessageInTransit::Header) invalid"); |
41 }; | 41 }; |
42 | 42 |
43 MessageInTransit::View::View(size_t message_size, const void* buffer) | 43 MessageInTransit::View::View(size_t message_size, const void* buffer) |
44 : buffer_(buffer) { | 44 : buffer_(buffer) { |
45 size_t next_message_size = 0; | 45 size_t next_message_size = 0; |
46 DCHECK(MessageInTransit::GetNextMessageSize( | 46 DCHECK(MessageInTransit::GetNextMessageSize(buffer_, message_size, |
47 buffer_, message_size, &next_message_size)); | 47 &next_message_size)); |
48 DCHECK_EQ(message_size, next_message_size); | 48 DCHECK_EQ(message_size, next_message_size); |
49 // This should be equivalent. | 49 // This should be equivalent. |
50 DCHECK_EQ(message_size, total_size()); | 50 DCHECK_EQ(message_size, total_size()); |
51 } | 51 } |
52 | 52 |
53 bool MessageInTransit::View::IsValid(size_t serialized_platform_handle_size, | 53 bool MessageInTransit::View::IsValid(size_t serialized_platform_handle_size, |
54 const char** error_message) const { | 54 const char** error_message) const { |
55 size_t max_message_num_bytes = GetConfiguration().max_message_num_bytes; | 55 size_t max_message_num_bytes = GetConfiguration().max_message_num_bytes; |
56 // Avoid dangerous situations, but making sure that the size of the "header" + | 56 // Avoid dangerous situations, but making sure that the size of the "header" + |
57 // the size of the data fits into a 31-bit number. | 57 // the size of the data fits into a 31-bit number. |
58 DCHECK_LE(static_cast<uint64_t>(sizeof(Header)) + max_message_num_bytes, | 58 DCHECK_LE(static_cast<uint64_t>(sizeof(Header)) + max_message_num_bytes, |
59 0x7fffffffULL) | 59 0x7fffffffULL) |
60 << "GetConfiguration().max_message_num_bytes too big"; | 60 << "GetConfiguration().max_message_num_bytes too big"; |
61 | 61 |
62 // We assume (to avoid extra rounding code) that the maximum message (data) | 62 // We assume (to avoid extra rounding code) that the maximum message (data) |
63 // size is a multiple of the alignment. | 63 // size is a multiple of the alignment. |
64 DCHECK_EQ(max_message_num_bytes % kMessageAlignment, 0U) | 64 DCHECK_EQ(max_message_num_bytes % kMessageAlignment, 0U) |
65 << "GetConfiguration().max_message_num_bytes not a multiple of alignment"; | 65 << "GetConfiguration().max_message_num_bytes not a multiple of alignment"; |
66 | 66 |
67 // Note: This also implies a check on the |main_buffer_size()|, which is just | 67 // Note: This also implies a check on the |main_buffer_size()|, which is just |
68 // |RoundUpMessageAlignment(sizeof(Header) + num_bytes())|. | 68 // |RoundUpMessageAlignment(sizeof(Header) + num_bytes())|. |
69 if (num_bytes() > max_message_num_bytes) { | 69 if (num_bytes() > max_message_num_bytes) { |
70 *error_message = "Message data payload too large"; | 70 *error_message = "Message data payload too large"; |
71 return false; | 71 return false; |
72 } | 72 } |
73 | 73 |
74 if (transport_data_buffer_size() > 0) { | 74 if (transport_data_buffer_size() > 0) { |
75 const char* e = | 75 const char* e = TransportData::ValidateBuffer( |
76 TransportData::ValidateBuffer(serialized_platform_handle_size, | 76 serialized_platform_handle_size, transport_data_buffer(), |
77 transport_data_buffer(), | 77 transport_data_buffer_size()); |
78 transport_data_buffer_size()); | |
79 if (e) { | 78 if (e) { |
80 *error_message = e; | 79 *error_message = e; |
81 return false; | 80 return false; |
82 } | 81 } |
83 } | 82 } |
84 | 83 |
85 return true; | 84 return true; |
86 } | 85 } |
87 | 86 |
88 MessageInTransit::MessageInTransit(Type type, | 87 MessageInTransit::MessageInTransit(Type type, |
89 Subtype subtype, | 88 Subtype subtype, |
90 uint32_t num_bytes, | 89 uint32_t num_bytes, |
91 const void* bytes) | 90 const void* bytes) |
92 : main_buffer_size_(RoundUpMessageAlignment(sizeof(Header) + num_bytes)), | 91 : main_buffer_size_(RoundUpMessageAlignment(sizeof(Header) + num_bytes)), |
93 main_buffer_(static_cast<char*>( | 92 main_buffer_(static_cast<char*>( |
94 base::AlignedAlloc(main_buffer_size_, kMessageAlignment))) { | 93 base::AlignedAlloc(main_buffer_size_, kMessageAlignment))) { |
95 ConstructorHelper(type, subtype, num_bytes); | 94 ConstructorHelper(type, subtype, num_bytes); |
96 if (bytes) { | 95 if (bytes) { |
97 memcpy(MessageInTransit::bytes(), bytes, num_bytes); | 96 memcpy(MessageInTransit::bytes(), bytes, num_bytes); |
98 memset(static_cast<char*>(MessageInTransit::bytes()) + num_bytes, | 97 memset(static_cast<char*>(MessageInTransit::bytes()) + num_bytes, 0, |
99 0, | |
100 main_buffer_size_ - sizeof(Header) - num_bytes); | 98 main_buffer_size_ - sizeof(Header) - num_bytes); |
101 } else { | 99 } else { |
102 memset(MessageInTransit::bytes(), 0, main_buffer_size_ - sizeof(Header)); | 100 memset(MessageInTransit::bytes(), 0, main_buffer_size_ - sizeof(Header)); |
103 } | 101 } |
104 } | 102 } |
105 | 103 |
106 MessageInTransit::MessageInTransit(Type type, | 104 MessageInTransit::MessageInTransit(Type type, |
107 Subtype subtype, | 105 Subtype subtype, |
108 uint32_t num_bytes, | 106 uint32_t num_bytes, |
109 UserPointer<const void> bytes) | 107 UserPointer<const void> bytes) |
(...skipping 105 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
215 DCHECK_EQ(main_buffer_size_ % kMessageAlignment, 0u); | 213 DCHECK_EQ(main_buffer_size_ % kMessageAlignment, 0u); |
216 header()->total_size = static_cast<uint32_t>(main_buffer_size_); | 214 header()->total_size = static_cast<uint32_t>(main_buffer_size_); |
217 if (transport_data_) { | 215 if (transport_data_) { |
218 header()->total_size += | 216 header()->total_size += |
219 static_cast<uint32_t>(transport_data_->buffer_size()); | 217 static_cast<uint32_t>(transport_data_->buffer_size()); |
220 } | 218 } |
221 } | 219 } |
222 | 220 |
223 } // namespace system | 221 } // namespace system |
224 } // namespace mojo | 222 } // namespace mojo |
OLD | NEW |