Add FIDO U2F message and packet classes

device/u2f/u2f_message_unittest.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 "device/u2f/u2f_message.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+class U2fMessageTest : public testing::Test {};
+
+// Packets should be 64 bytes + 1 report ID byte
+TEST_F(U2fMessageTest, TestPacketSize) {
+  char channel_id[4] = {5, 6, 7, 8};
+  std::vector<uint8_t> data;
+
+  scoped_refptr<device::U2fPacket> init_packet(
+      new device::U2fInitPacket(channel_id, 0, data, data.size()));
+  ASSERT_EQ(init_packet->GetBuffer()->size(), 65);
+
+  scoped_refptr<device::U2fPacket> cont_packet(
+      new device::U2fContPacket(channel_id, 0, data));
+  ASSERT_EQ(init_packet->GetBuffer()->size(), 65);
+}
+
+/*
+ * U2f Init Packets are of the format:
+ * Byte 0:    0
+ * Byte 1-4:  Channel ID
+ * Byte 5:    Command byte
+ * Byte 6-7:  Big Endian size of data
+ * Byte 8-n:  Data block
+ *
+ * Remaining buffer is padded with 0
+ */
+TEST_F(U2fMessageTest, TestPacketData) {
+  char channel_id[4] = {5, 6, 7, 8};
+  std::vector<uint8_t> data{10, 11};
+  char cmd = 0x88;
+  scoped_refptr<device::U2fPacket> init_packet(
+      new device::U2fInitPacket(channel_id, cmd, data, data.size()));
+  int index = 0;
+
+  ASSERT_EQ(init_packet->GetBuffer()->data()[index++], 0);
+  ASSERT_EQ(init_packet->GetBuffer()->data()[index++], channel_id[0]);
+  ASSERT_EQ(init_packet->GetBuffer()->data()[index++], channel_id[1]);
+  ASSERT_EQ(init_packet->GetBuffer()->data()[index++], channel_id[2]);
+  ASSERT_EQ(init_packet->GetBuffer()->data()[index++], channel_id[3]);
+  ASSERT_EQ(init_packet->GetBuffer()->data()[index++], cmd);
+  ASSERT_EQ(init_packet->GetBuffer()->data()[index++],
+            (char)(data.size() >> 8));
+  ASSERT_EQ(init_packet->GetBuffer()->data()[index++],
+            (char)(data.size() & 0xff));
+  ASSERT_EQ(init_packet->GetBuffer()->data()[index++], data.at(0));
+  ASSERT_EQ(init_packet->GetBuffer()->data()[index++], data.at(1));
+  for (; index < init_packet->GetBuffer()->size(); index++) {
+    ASSERT_EQ(init_packet->GetBuffer()->data()[index], 0);
+  }
+}
+
+TEST_F(U2fMessageTest, TestMessagePartioning) {
+  char channel_id[4] = {1, 1, 2, 3};
+  std::vector<uint8_t> data(device::U2fMessage::kInitPacketDataSize + 1);
+  scoped_refptr<device::U2fMessage> two_packet_message =
+      device::U2fMessage::Create(channel_id, device::U2fMessage::Type::CMD_PING,
+                                 data);
+  ASSERT_EQ(two_packet_message->NumPackets(), 2U);
+
+  data.resize(device::U2fMessage::kInitPacketDataSize);
+  scoped_refptr<device::U2fMessage> one_packet_message =
+      device::U2fMessage::Create(channel_id, device::U2fMessage::Type::CMD_PING,
+                                 data);
+  ASSERT_EQ(one_packet_message->NumPackets(), 1U);
+
+  data.resize(device::U2fMessage::kInitPacketDataSize +
+              device::U2fMessage::kContPacketDataSize + 1);
+  scoped_refptr<device::U2fMessage> three_packet_message =
+      device::U2fMessage::Create(channel_id, device::U2fMessage::Type::CMD_PING,
+                                 data);
+  ASSERT_EQ(three_packet_message->NumPackets(), 3U);
+}
+
+TEST_F(U2fMessageTest, TestMaxSize) {
+  char channel_id[4] = {0, 1, 2, 3};
+  std::vector<uint8_t> data(device::U2fMessage::kMaxMessageSize + 1);
+  scoped_refptr<device::U2fMessage> oversize_message =
+      device::U2fMessage::Create(channel_id, device::U2fMessage::Type::CMD_PING,
+                                 data);
+  ASSERT_EQ(oversize_message, nullptr);
+}