Weave Packet Generator

components/proximity_auth/ble/bluetooth_low_energy_weave_packet_generator.cc

+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "components/proximity_auth/ble/bluetooth_low_energy_weave_packet_generator.h"
+
+#include <string.h>
+
+#include <algorithm>
+
+#include "base/logging.h"
+
+namespace proximity_auth {
+
+BluetoothLowEnergyWeavePacketGenerator::Factory*
+    BluetoothLowEnergyWeavePacketGenerator::Factory::factory_instance_ = NULL;

[Kyle Horimoto, 2016/06/08 00:06:29]

Initialize to nullptr.

[jingxuy, 2016/06/08 21:30:36]

Done.

+
+std::unique_ptr<BluetoothLowEnergyWeavePacketGenerator>
+BluetoothLowEnergyWeavePacketGenerator::Factory::NewInstance() {
+  if (!factory_instance_) {

[Kyle Horimoto, 2016/06/08 00:06:30]

Check for equality with nullptr.

[jingxuy, 2016/06/08 21:30:36]

Done.

+    factory_instance_ = new Factory();
+  }
+  return std::unique_ptr<BluetoothLowEnergyWeavePacketGenerator>(
+      factory_instance_->BuildInstance());
+}
+
+void BluetoothLowEnergyWeavePacketGenerator::Factory::SetInstanceForTesting(
+    Factory* factory) {
+  factory_instance_ = factory;
+}
+
+BluetoothLowEnergyWeavePacketGenerator*
+BluetoothLowEnergyWeavePacketGenerator::Factory::BuildInstance() {
+  return new BluetoothLowEnergyWeavePacketGenerator();
+}
+
+BluetoothLowEnergyWeavePacketGenerator::BluetoothLowEnergyWeavePacketGenerator()
+    : packet_size_(kDefaultDataPacketSize), packet_number_(0) {}
+
+std::unique_ptr<BluetoothLowEnergyWeavePacketGenerator::Packet>
+BluetoothLowEnergyWeavePacketGenerator::CreateConnectionRequest() {
+  Packet* packet = CreateControlPacket();
+
+  SetControlCmd(packet, ControlPacketCommand::REQUEST);
+  SetIntField(packet, 1, kMinSupportedWeaveVersion);
+  SetIntField(packet, 3, kMaxSupportedWeaveVersion);
+  SetIntField(packet, 5, kMaxSupportedPacketSize);
+
+  return std::unique_ptr<Packet>(packet);
+}
+
+std::unique_ptr<BluetoothLowEnergyWeavePacketGenerator::Packet>
+BluetoothLowEnergyWeavePacketGenerator::CreateConnectionResponse() {
+  Packet* packet = CreateControlPacket();
+
+  SetControlCmd(packet, ControlPacketCommand::RESPONSE);
+  SetIntField(packet, 1, kServerWeaveVersion);
+  SetIntField(packet, 3, packet_size_);
+
+  return std::unique_ptr<Packet>(packet);
+}
+
+std::unique_ptr<BluetoothLowEnergyWeavePacketGenerator::Packet>
+BluetoothLowEnergyWeavePacketGenerator::CreateConnectionClose(
+    ReasonForClose reason_for_close) {
+  Packet* packet = CreateControlPacket();
+
+  SetControlCmd(packet, ControlPacketCommand::CLOSE);
+  SetIntField(packet, 1, reason_for_close);
+
+  return std::unique_ptr<Packet>(packet);
+}
+
+BluetoothLowEnergyWeavePacketGenerator::Packet*
+BluetoothLowEnergyWeavePacketGenerator::CreateControlPacket() {
+  Packet* packet = new Packet(kControlPacketSize, 0);
+
+  // Packet is a control packet.
+  SetPacketTypeBit(packet, PacketType::CONTROL);
+  SetPacketCounter(packet);
+
+  return packet;
+}
+
+void BluetoothLowEnergyWeavePacketGenerator::SetDataPacketSize(uint32_t size) {
+  packet_size_ = size;
+}
+
+std::unique_ptr<std::vector<BluetoothLowEnergyWeavePacketGenerator::Packet>>
+BluetoothLowEnergyWeavePacketGenerator::EncodeDataMessage(std::string message) {
+  uint32_t payload_size = packet_size_ - 1;
+  // (message.length() + 1) + (payload_size - 1) / (payload_size).
+  uint32_t num_packets = (message.length() + payload_size) / payload_size;
+  std::vector<Packet>* weave_message = new std::vector<Packet>(num_packets);
+
+  uint32_t last_byte = message.length() + 1;
+  const char* byte_message = message.c_str();
+
+  for (uint32_t i = 0; i < num_packets; ++i) {
+    Packet* packet = &weave_message->at(i);
+    uint32_t begin = payload_size * i;
+    uint32_t end = std::min(begin + payload_size, last_byte);
+
+    packet->push_back(0);
+
+    SetPacketTypeBit(packet, PacketType::DATA);
+    SetPacketCounter(packet);
+
+    for (uint32_t i = begin; i < end; ++i) {
+      packet->push_back(byte_message[i]);
+    }
+  }
+
+  // Since even empty string has \0, weave_message is guaranteed to be nonempty.
+  SetDataFirstBit(&weave_message->at(0));
+  SetDataLastBit(&weave_message->at(num_packets - 1));
+
+  return std::unique_ptr<std::vector<Packet>>(weave_message);
+}
+
+void BluetoothLowEnergyWeavePacketGenerator::SetIntField(Packet* packet,

[Kyle Horimoto, 2016/06/08 00:06:30]

nit: Rename this to SetShortField().

+                                                         uint32_t index,
+                                                         uint16_t val) {
+  DCHECK(packet);
+  DCHECK_LT(index, packet->size());
+  DCHECK_LT(index + 1, packet->size());
+
+  uint8_t upper = val >> 8 & 0xFF;
+  uint8_t lower = val & 0xFF;
+
+  packet->at(index) = lower;
+  packet->at(index + 1) = upper;
+}
+
+void BluetoothLowEnergyWeavePacketGenerator::SetPacketTypeBit(Packet* packet,
+                                                              bool val) {
+  DCHECK(packet);
+  DCHECK(!packet->empty());
+  packet->at(0) = (packet->at(0) & 0x7F) | (val << 7);
+}
+
+void BluetoothLowEnergyWeavePacketGenerator::SetControlCmd(Packet* packet,
+                                                           uint8_t val) {
+  DCHECK(packet);
+  DCHECK(!packet->empty());
+  // Currently only commands supported are 0, 1, and 2.
+  DCHECK_LT(val, 3);
+  packet->at(0) = (packet->at(0) & 0xF0) | val;
+}
+
+void BluetoothLowEnergyWeavePacketGenerator::SetPacketCounter(Packet* packet) {
+  DCHECK(packet);
+  DCHECK(!packet->empty());
+  uint8_t counter = packet_number_ % kMaxPacketCounter;
+  packet->at(0) = (packet->at(0) & 0x8F) | (counter << 4);
+  packet_number_++;
+}
+
+void BluetoothLowEnergyWeavePacketGenerator::SetDataFirstBit(Packet* packet) {
+  DCHECK(packet);
+  DCHECK(!packet->empty());
+  packet->at(0) = (packet->at(0) & 0xF7) | (1 << 3);
+}
+
+void BluetoothLowEnergyWeavePacketGenerator::SetDataLastBit(Packet* packet) {
+  DCHECK(packet);
+  DCHECK(!packet->empty());
+  packet->at(0) = (packet->at(0) & 0xFB) | (1 << 2);
+}
+
+}  // namespace proximity_auth