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

Unified Diff: device/u2f/u2f_message.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 side-by-side diff with in-line comments
Download patch
« no previous file with comments | « device/u2f/u2f_message.h ('k') | device/u2f/u2f_message_fuzzer.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: device/u2f/u2f_message.cc
diff --git a/device/u2f/u2f_message.cc b/device/u2f/u2f_message.cc
index e33b9bff21e3900eb340ad95f9abd3c0367dda3a..8cf2281ef24504e9ab54280a9c78550b066a0ca1 100644
--- a/device/u2f/u2f_message.cc
+++ b/device/u2f/u2f_message.cc
@@ -2,24 +2,27 @@
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
-#include "device/u2f/u2f_message.h"
+#include "base/memory/ptr_util.h"
#include "device/u2f/u2f_packet.h"
#include "net/base/io_buffer.h"
+#include "u2f_message.h"
+
namespace device {
// static
-scoped_refptr<U2fMessage> U2fMessage::Create(uint32_t channel_id,
- Type type,
- const std::vector<uint8_t>& data) {
+std::unique_ptr<U2fMessage> U2fMessage::Create(
+ uint32_t channel_id,
+ Type type,
+ const std::vector<uint8_t>& data) {
if (data.size() > kMaxMessageSize)
return nullptr;
- return make_scoped_refptr(new U2fMessage(channel_id, type, data));
+ return base::MakeUnique<U2fMessage>(channel_id, type, data);
}
// static
-scoped_refptr<U2fMessage> U2fMessage::CreateFromSerializedData(
+std::unique_ptr<U2fMessage> U2fMessage::CreateFromSerializedData(
scoped_refptr<net::IOBufferWithSize> buf) {
size_t remaining_size = 0;
if (buf == nullptr ||
@@ -27,19 +30,19 @@ scoped_refptr<U2fMessage> U2fMessage::CreateFromSerializedData(
static_cast<size_t>(buf->size()) < kInitPacketHeader)
return nullptr;
- scoped_refptr<U2fInitPacket> init_packet =
+ std::unique_ptr<U2fInitPacket> init_packet =
U2fInitPacket::CreateFromSerializedData(buf, &remaining_size);
if (init_packet == nullptr)
return nullptr;
- return make_scoped_refptr(new U2fMessage(init_packet, remaining_size));
+ return base::MakeUnique<U2fMessage>(std::move(init_packet), remaining_size);
}
-U2fMessage::U2fMessage(scoped_refptr<U2fInitPacket> init_packet,
+U2fMessage::U2fMessage(std::unique_ptr<U2fInitPacket> init_packet,
size_t remaining_size)
: remaining_size_(remaining_size) {
channel_id_ = init_packet->channel_id();
- packets_.push_back(init_packet);
+ packets_.push_back(std::move(init_packet));
}
U2fMessage::U2fMessage(uint32_t channel_id,
@@ -60,9 +63,9 @@ U2fMessage::U2fMessage(uint32_t channel_id,
remaining_bytes = 0;
}
- packets_.push_back(make_scoped_refptr(
- new U2fInitPacket(channel_id, static_cast<uint8_t>(type),
- std::vector<uint8_t>(first, last), data.size())));
+ packets_.push_back(base::MakeUnique<U2fInitPacket>(
+ channel_id, static_cast<uint8_t>(type), std::vector<uint8_t>(first, last),
+ data.size()));
while (remaining_bytes > 0) {
first = last;
@@ -74,19 +77,19 @@ U2fMessage::U2fMessage(uint32_t channel_id,
remaining_bytes = 0;
}
- packets_.push_back(make_scoped_refptr(new U2fContinuationPacket(
- channel_id, sequence, std::vector<uint8_t>(first, last))));
+ packets_.push_back(base::MakeUnique<U2fContinuationPacket>(
+ channel_id, sequence, std::vector<uint8_t>(first, last)));
sequence++;
}
}
U2fMessage::~U2fMessage() {}
-std::list<scoped_refptr<U2fPacket>>::const_iterator U2fMessage::begin() {
+std::list<std::unique_ptr<U2fPacket>>::const_iterator U2fMessage::begin() {
return packets_.cbegin();
}
-std::list<scoped_refptr<U2fPacket>>::const_iterator U2fMessage::end() {
+std::list<std::unique_ptr<U2fPacket>>::const_iterator U2fMessage::end() {
return packets_.cend();
}
@@ -103,7 +106,7 @@ scoped_refptr<net::IOBufferWithSize> U2fMessage::PopNextPacket() {
bool U2fMessage::AddContinuationPacket(
scoped_refptr<net::IOBufferWithSize> buf) {
size_t remaining_size = remaining_size_;
- scoped_refptr<U2fContinuationPacket> cont_packet =
+ std::unique_ptr<U2fContinuationPacket> cont_packet =
U2fContinuationPacket::CreateFromSerializedData(buf, &remaining_size);
// Reject packets with a different channel id
@@ -111,7 +114,7 @@ bool U2fMessage::AddContinuationPacket(
return false;
remaining_size_ = remaining_size;
- packets_.push_back(cont_packet);
+ packets_.push_back(std::move(cont_packet));
return true;
}
« no previous file with comments | « device/u2f/u2f_message.h ('k') | device/u2f/u2f_message_fuzzer.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698