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

Side by Side Diff: device/u2f/u2f_packet.cc

Issue 2766723003: Use unique pointers for U2fPacket and U2fMessage (Closed)
Patch Set: Change fuzzer to use the new unique_ptr constructor Created 3 years, 9 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
« no previous file with comments | « device/u2f/u2f_packet.h ('k') | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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 "device/u2f/u2f_packet.h" 5 #include "base/memory/ptr_util.h"
6 #include "net/base/io_buffer.h" 6 #include "net/base/io_buffer.h"
7 7
8 #include "u2f_packet.h"
9
8 namespace device { 10 namespace device {
9 11
10 U2fPacket::U2fPacket(const std::vector<uint8_t> data, const uint32_t channel_id) 12 U2fPacket::U2fPacket(const std::vector<uint8_t> data, const uint32_t channel_id)
11 : data_(data), channel_id_(channel_id) {} 13 : data_(data), channel_id_(channel_id) {}
12 14
13 U2fPacket::U2fPacket() {} 15 U2fPacket::U2fPacket() {}
14 16
15 U2fPacket::~U2fPacket() {} 17 U2fPacket::~U2fPacket() {}
16 18
17 scoped_refptr<net::IOBufferWithSize> U2fPacket::GetSerializedBuffer() { 19 scoped_refptr<net::IOBufferWithSize> U2fPacket::GetSerializedBuffer() {
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after
51 payload_length_ = payload_length; 53 payload_length_ = payload_length;
52 serialized_->data()[index++] = (payload_length >> 8) & 0xff; 54 serialized_->data()[index++] = (payload_length >> 8) & 0xff;
53 serialized_->data()[index++] = payload_length & 0xff; 55 serialized_->data()[index++] = payload_length & 0xff;
54 for (size_t data_idx = 0; data_idx < data_.size(); ++data_idx) 56 for (size_t data_idx = 0; data_idx < data_.size(); ++data_idx)
55 serialized_->data()[index++] = data_.at(data_idx); 57 serialized_->data()[index++] = data_.at(data_idx);
56 while (static_cast<int>(index) < serialized_->size()) 58 while (static_cast<int>(index) < serialized_->size())
57 serialized_->data()[index++] = 0; 59 serialized_->data()[index++] = 0;
58 } 60 }
59 61
60 // static 62 // static
61 scoped_refptr<U2fInitPacket> U2fInitPacket::CreateFromSerializedData( 63 std::unique_ptr<U2fInitPacket> U2fInitPacket::CreateFromSerializedData(
62 scoped_refptr<net::IOBufferWithSize> buf, 64 scoped_refptr<net::IOBufferWithSize> buf,
63 size_t* remaining_size) { 65 size_t* remaining_size) {
64 if (buf == nullptr || remaining_size == nullptr || buf->size() != kPacketSize) 66 if (buf == nullptr || remaining_size == nullptr || buf->size() != kPacketSize)
65 return nullptr; 67 return nullptr;
66 68
67 return make_scoped_refptr(new U2fInitPacket(buf, remaining_size)); 69 return base::MakeUnique<U2fInitPacket>(buf, remaining_size);
68 } 70 }
69 71
70 U2fInitPacket::U2fInitPacket(scoped_refptr<net::IOBufferWithSize> buf, 72 U2fInitPacket::U2fInitPacket(scoped_refptr<net::IOBufferWithSize> buf,
71 size_t* remaining_size) { 73 size_t* remaining_size) {
72 // Report ID is at index 0, so start at index 1 for channel ID 74 // Report ID is at index 0, so start at index 1 for channel ID
73 size_t index = 1; 75 size_t index = 1;
74 uint16_t payload_size = 0; 76 uint16_t payload_size = 0;
75 uint16_t data_size = 0; 77 uint16_t data_size = 0;
76 78
77 channel_id_ = (buf->data()[index++] & 0xff) << 24; 79 channel_id_ = (buf->data()[index++] & 0xff) << 24;
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after
120 122
121 serialized_->data()[index++] = sequence_; 123 serialized_->data()[index++] = sequence_;
122 for (size_t idx = 0; idx < data_.size(); ++idx) 124 for (size_t idx = 0; idx < data_.size(); ++idx)
123 serialized_->data()[index++] = data_.at(idx); 125 serialized_->data()[index++] = data_.at(idx);
124 126
125 while (static_cast<int>(index) < serialized_->size()) 127 while (static_cast<int>(index) < serialized_->size())
126 serialized_->data()[index++] = 0; 128 serialized_->data()[index++] = 0;
127 } 129 }
128 130
129 // static 131 // static
130 scoped_refptr<U2fContinuationPacket> 132 std::unique_ptr<U2fContinuationPacket>
131 U2fContinuationPacket::CreateFromSerializedData( 133 U2fContinuationPacket::CreateFromSerializedData(
132 scoped_refptr<net::IOBufferWithSize> buf, 134 scoped_refptr<net::IOBufferWithSize> buf,
133 size_t* remaining_size) { 135 size_t* remaining_size) {
134 if (buf == nullptr || remaining_size == nullptr || buf->size() != kPacketSize) 136 if (buf == nullptr || remaining_size == nullptr || buf->size() != kPacketSize)
135 return nullptr; 137 return nullptr;
136 138
137 return make_scoped_refptr(new U2fContinuationPacket(buf, remaining_size)); 139 return base::MakeUnique<U2fContinuationPacket>(buf, remaining_size);
138 } 140 }
139 141
140 U2fContinuationPacket::U2fContinuationPacket( 142 U2fContinuationPacket::U2fContinuationPacket(
141 scoped_refptr<net::IOBufferWithSize> buf, 143 scoped_refptr<net::IOBufferWithSize> buf,
142 size_t* remaining_size) { 144 size_t* remaining_size) {
143 // Report ID is at index 0, so start at index 1 for channel ID 145 // Report ID is at index 0, so start at index 1 for channel ID
144 size_t index = 1; 146 size_t index = 1;
145 size_t data_size; 147 size_t data_size;
146 148
147 channel_id_ = (buf->data()[index++] & 0xff) << 24; 149 channel_id_ = (buf->data()[index++] & 0xff) << 24;
148 channel_id_ |= (buf->data()[index++] & 0xff) << 16; 150 channel_id_ |= (buf->data()[index++] & 0xff) << 16;
149 channel_id_ |= (buf->data()[index++] & 0xff) << 8; 151 channel_id_ |= (buf->data()[index++] & 0xff) << 8;
150 channel_id_ |= buf->data()[index++] & 0xff; 152 channel_id_ |= buf->data()[index++] & 0xff;
151 sequence_ = buf->data()[index++]; 153 sequence_ = buf->data()[index++];
152 154
153 // Check to see if packet payload is less than maximum size and padded with 0s 155 // Check to see if packet payload is less than maximum size and padded with 0s
154 data_size = std::min(*remaining_size, kPacketSize - index); 156 data_size = std::min(*remaining_size, kPacketSize - index);
155 *remaining_size -= data_size; 157 *remaining_size -= data_size;
156 data_.insert(std::end(data_), &buf->data()[index], 158 data_.insert(std::end(data_), &buf->data()[index],
157 &buf->data()[index + data_size]); 159 &buf->data()[index + data_size]);
158 160
159 // Incoming buffer may not be padded with 0's, so manually update buffer 161 // Incoming buffer may not be padded with 0's, so manually update buffer
160 for (int i = index + data_size; i < buf->size(); ++i) 162 for (int i = index + data_size; i < buf->size(); ++i)
161 buf->data()[i] = 0; 163 buf->data()[i] = 0;
162 serialized_ = buf; 164 serialized_ = buf;
163 } 165 }
164 166
165 U2fContinuationPacket::~U2fContinuationPacket() {} 167 U2fContinuationPacket::~U2fContinuationPacket() {}
168
166 } // namespace device 169 } // namespace device
OLDNEW
« no previous file with comments | « device/u2f/u2f_packet.h ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698