Add FIDO U2F message and packet classes

device/u2f/u2f_message_unittest.cc

Index: device/u2f/u2f_message_unittest.cc
diff --git a/device/u2f/u2f_message_unittest.cc b/device/u2f/u2f_message_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..6dbd0c18a19278f75f0843710dfd9fe2d5594f7d
--- /dev/null
+++ b/device/u2f/u2f_message_unittest.cc
@@ -0,0 +1,76 @@
+// 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<char> data;
+
+  scoped_refptr<device::U2fPacket> init_packet(
+      new device::U2fInitPacket(channel_id, 0, data));
+  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<char> data{10, 11};
+  char cmd = 0x88;
+  scoped_refptr<device::U2fPacket> init_packet(
+      new device::U2fInitPacket(channel_id, cmd, data));
+  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<char> data(device::U2fMessage::kInitPacketDataSize + 1);
+  scoped_refptr<device::U2fMessage> two_packet_message(
+      new device::U2fMessage(channel_id, device::U2fMessage::CMD_PING, data));
+  ASSERT_EQ(two_packet_message->NumPackets(), 2U);
+
+  data.resize(device::U2fMessage::kInitPacketDataSize);
+  scoped_refptr<device::U2fMessage> one_packet_message(
+      new device::U2fMessage(channel_id, device::U2fMessage::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(
+      new device::U2fMessage(channel_id, device::U2fMessage::CMD_PING, data));
+  ASSERT_EQ(three_packet_message->NumPackets(), 3U);
+}