Weave Packet Generator

components/proximity_auth/ble/bluetooth_low_energy_weave_packet_generator.h

+// 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.
+
+#ifndef COMPONENTS_PROXIMITY_AUTH_BLE_BLUETOOTH_LOW_ENERGY_WEAVE_PACKET_GENERATOR_H_
+#define COMPONENTS_PROXIMITY_AUTH_BLE_BLUETOOTH_LOW_ENERGY_WEAVE_PACKET_GENERATOR_H_
+
+#include <stddef.h>
+#include <stdint.h>
+
+#include <memory>
+#include <string>
+#include <vector>
+
+namespace proximity_auth {
+// Generates the messages sent using the uWeave protocol.
+class BluetoothLowEnergyWeavePacketGenerator {
+ public:
+  class Factory {
+   public:
+    static std::unique_ptr<BluetoothLowEnergyWeavePacketGenerator>
+    NewInstance();
+
+    // Exposed for testing.
+    static void SetInstanceForTesting(Factory* factory);
+
+   protected:
+    // Exposed for testing.
+    BluetoothLowEnergyWeavePacketGenerator* BuildInstance();
+
+   private:
+    static Factory* factory_instance_;
+  };
+
+  // Determine whether a packet is a control packet or a data packet.
+  enum PacketType { DATA = 0x00, CONTROL = 0x01 };
+
+  // Identify the action intended by the control packet.
+  enum ControlPacketCommand {
+    CONNECTION_REQUEST = 0x00,
+    CONNECTION_RESPONSE = 0x01,
+    CONNECTION_CLOSE = 0x02
+  };
+
+  // Sent with the ConnectionClose control packet.
+  // Identify why the client/server wished to close the connection.
+  enum ReasonForClose {
+    CLOSE_WITHOUT_ERROR = 0x00,
+    UNKNOWN_ERROR = 0x01,
+    NO_COMMON_VERSION_SUPPORTED = 0x02,
+    RECEIVED_PACKET_OUT_OF_SEQUENCE = 0x03,
+    APPLICATION_ERROR = 0x80
+  };
+
+  typedef std::vector<uint8_t> Packet;
+
+  std::unique_ptr<Packet> CreateConnectionRequest();
+  std::unique_ptr<Packet> CreateConnectionResponse();
+  std::unique_ptr<Packet> CreateConnectionClose(
+      ReasonForClose reason_for_close);
+
+  void SetDataPacketSize(uint32_t size);
+
+  std::unique_ptr<std::vector<Packet>> EncodeDataMessage(std::string message);
+
+ protected:
+  BluetoothLowEnergyWeavePacketGenerator();
+
+ private:
+  Packet* CreateControlPacket();
+
+  // For error checking.
+  bool IsBit(uint8_t val);
+
+  void SetShortField(Packet* packet, uint32_t index, uint16_t val);
+  void SetPacketTypeBit(Packet* packet, PacketType val);
+  void SetControlCommand(Packet* packet, ControlPacketCommand val);
+  void SetPacketCounter(Packet* packet);
+  void SetDataFirstBit(Packet* packet);
+  void SetDataLastBit(Packet* packet);
+
+  // The default data packet length is 20 unless setDataPacketLength() is called
+  // and specified otherwise.
+  uint32_t packet_size_;
+
+  // Counter for the number of packets sent, starting at 0.
+  uint32_t packet_number_;
+};
+
+}  // namespace proximity_auth
+
+#endif  // COMPONENTS_PROXIMITY_AUTH_BLE_BLUETOOTH_LOW_ENERGY_WEAVE_PACKET_GENERATOR_H_