Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(70)

Side by Side Diff: mojo/public/cpp/bindings/lib/message_buffer.cc

Issue 2207763003: Make mojo::internal::Buffer a non-virtual class. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@85_3_inline_validation_context
Patch Set: Created 4 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2016 The Chromium Authors. All rights reserved. 1 // Copyright 2016 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/message_buffer.h" 5 #include "mojo/public/cpp/bindings/lib/message_buffer.h"
6 6
7 #include <limits> 7 #include <limits>
8 8
9 #include "mojo/public/cpp/bindings/lib/serialization_util.h" 9 #include "mojo/public/cpp/bindings/lib/serialization_util.h"
10 10
11 namespace mojo { 11 namespace mojo {
12 namespace internal { 12 namespace internal {
13 13
14 MessageBuffer::MessageBuffer(size_t capacity, bool zero_initialized) { 14 MessageBuffer::MessageBuffer(size_t capacity, bool zero_initialized) {
15 DCHECK_LE(capacity, std::numeric_limits<uint32_t>::max()); 15 DCHECK_LE(capacity, std::numeric_limits<uint32_t>::max());
16 data_num_bytes_ = static_cast<uint32_t>(capacity);
17 16
18 MojoResult rv = AllocMessage(capacity, nullptr, 0, 17 MojoResult rv = AllocMessage(capacity, nullptr, 0,
19 MOJO_ALLOC_MESSAGE_FLAG_NONE, &message_); 18 MOJO_ALLOC_MESSAGE_FLAG_NONE, &message_);
20 CHECK_EQ(rv, MOJO_RESULT_OK); 19 CHECK_EQ(rv, MOJO_RESULT_OK);
21 20
22 if (capacity == 0) { 21 void* buffer = nullptr;
23 buffer_ = nullptr; 22 if (capacity != 0) {
24 } else { 23 rv = GetMessageBuffer(message_.get(), &buffer);
25 rv = GetMessageBuffer(message_.get(), &buffer_);
26 CHECK_EQ(rv, MOJO_RESULT_OK); 24 CHECK_EQ(rv, MOJO_RESULT_OK);
27 25
28 if (zero_initialized) 26 if (zero_initialized)
29 memset(buffer_, 0, capacity); 27 memset(buffer, 0, capacity);
30 } 28 }
29 Initialize(buffer, capacity);
31 } 30 }
32 31
33 MessageBuffer::MessageBuffer(ScopedMessageHandle message, uint32_t num_bytes) { 32 MessageBuffer::MessageBuffer(ScopedMessageHandle message, uint32_t num_bytes) {
34 message_ = std::move(message); 33 message_ = std::move(message);
35 data_num_bytes_ = num_bytes;
36 34
37 if (num_bytes == 0) { 35 void* buffer = nullptr;
38 buffer_ = nullptr; 36 if (num_bytes != 0) {
39 } else { 37 MojoResult rv = GetMessageBuffer(message_.get(), &buffer);
40 MojoResult rv = GetMessageBuffer(message_.get(), &buffer_);
41 CHECK_EQ(rv, MOJO_RESULT_OK); 38 CHECK_EQ(rv, MOJO_RESULT_OK);
42 } 39 }
40 Initialize(buffer, num_bytes);
43 } 41 }
44 42
45 MessageBuffer::~MessageBuffer() {} 43 MessageBuffer::~MessageBuffer() {}
46 44
47 void* MessageBuffer::Allocate(size_t delta) {
48 delta = internal::Align(delta);
49
50 DCHECK_LE(delta, static_cast<size_t>(data_num_bytes_));
51 DCHECK_GT(bytes_claimed_ + static_cast<uint32_t>(delta), bytes_claimed_);
52
53 uint32_t new_bytes_claimed = bytes_claimed_ + static_cast<uint32_t>(delta);
54 if (new_bytes_claimed > data_num_bytes_) {
55 NOTREACHED();
56 return nullptr;
57 }
58
59 char* start = static_cast<char*>(buffer_) + bytes_claimed_;
60 bytes_claimed_ = new_bytes_claimed;
61 return static_cast<void*>(start);
62 }
63
64 void MessageBuffer::NotifyBadMessage(const std::string& error) { 45 void MessageBuffer::NotifyBadMessage(const std::string& error) {
65 DCHECK(message_.is_valid()); 46 DCHECK(message_.is_valid());
66 MojoResult result = mojo::NotifyBadMessage(message_.get(), error); 47 MojoResult result = mojo::NotifyBadMessage(message_.get(), error);
67 DCHECK_EQ(result, MOJO_RESULT_OK); 48 DCHECK_EQ(result, MOJO_RESULT_OK);
68 } 49 }
69 50
70 } // namespace internal 51 } // namespace internal
71 } // namespace mojo 52 } // namespace mojo
OLDNEW
« no previous file with comments | « mojo/public/cpp/bindings/lib/message_buffer.h ('k') | mojo/public/cpp/bindings/lib/serialization.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698