1 files changed, 312 insertions, 0 deletions
diff --git a/libtransport/src/test/test_core_manifest.cc b/libtransport/src/test/test_core_manifest.cc
new file mode 100644
index 000000000..99c71a56c
--- /dev/null
+++ b/libtransport/src/test/test_core_manifest.cc
@@ -0,0 +1,312 @@
+/*
+ * Copyright (c) 2021-2022 Cisco and/or its affiliates.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <core/manifest.h>
+#include <core/manifest_format_fixed.h>
+#include <gtest/gtest.h>
+#include <hicn/transport/auth/crypto_hash.h>
+#include <hicn/transport/auth/signer.h>
+#include <hicn/transport/auth/verifier.h>
+#include <test/packet_samples.h>
+
+#include <climits>
+#include <random>
+#include <vector>
+
+namespace transport {
+
+namespace core {
+
+namespace {
+// The fixture for testing class Foo.
+class ManifestTest : public ::testing::Test {
+ protected:
+  using ContentObjectManifest = Manifest<Fixed>;
+
+  ManifestTest()
+      : format_(HICN_PACKET_FORMAT_IPV6_TCP_AH),
+        name_("b001::123|321"),
+        signature_size_(0) {
+    manifest_ = ContentObjectManifest::createContentManifest(format_, name_,
+                                                             signature_size_);
+  }
+
+  virtual ~ManifestTest() {
+    // You can do clean-up work that doesn't throw exceptions here.
+  }
+
+  // If the constructor and destructor are not enough for setting up
+  // and cleaning up each test, you can define the following methods:
+
+  virtual void SetUp() {
+    // Code here will be called immediately after the constructor (right
+    // before each test).
+  }
+
+  virtual void TearDown() {
+    // Code here will be called immediately after each test (right
+    // before the destructor).
+  }
+
+  Packet::Format format_;
+  Name name_;
+  std::size_t signature_size_;
+  std::shared_ptr<ContentObjectManifest> manifest_;
+  std::vector<uint8_t> manifest_payload_ = {
+      0x11, 0x11, 0x01, 0x00, 0xb0, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00,
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xde, 0xad  // , 0x00, 0x00,
+                                                            // 0x00, 0x45, 0xa3,
+                                                            // 0xd1, 0xf2, 0x2b,
+                                                            // 0x94, 0x41, 0x22,
+                                                            // 0xc9, 0x00, 0x00,
+                                                            // 0x00, 0x44, 0xa3,
+                                                            // 0xd1, 0xf2, 0x2b,
+                                                            // 0x94, 0x41, 0x22,
+                                                            // 0xc8
+  };
+};
+
+}  // namespace
+
+TEST_F(ManifestTest, ManifestConstructor) {
+  // Create content object with manifest in payload
+  ContentObject::Ptr co =
+      core::PacketManager<>::getInstance().getPacket<ContentObject>(
+          format_, signature_size_);
+  co->setName(name_);
+  co->appendPayload(manifest_payload_.data(), manifest_payload_.size());
+
+  uint8_t buffer[256] = {0};
+  co->appendPayload(buffer, 256);
+
+  // Copy packet payload
+  uint8_t packet[1500];
+  auto length = co->getPayload()->length();
+  std::memcpy(packet, co->getPayload()->data(), length);
+
+  // Create manifest
+  ContentObjectManifest manifest(co);
+
+  // Check manifest payload is exactly the same of content object
+  ASSERT_EQ(length, manifest.getPacket()->getPayload()->length());
+  auto ret =
+      std::memcmp(packet, manifest.getPacket()->getPayload()->data(), length);
+  ASSERT_EQ(ret, 0);
+}
+
+TEST_F(ManifestTest, SetManifestType) {
+  manifest_->Encoder::clear();
+
+  ManifestType type1 = ManifestType::INLINE_MANIFEST;
+  ManifestType type2 = ManifestType::FLIC_MANIFEST;
+
+  manifest_->setType(type1);
+  ManifestType type_returned1 = manifest_->getType();
+
+  manifest_->Encoder::clear();
+
+  manifest_->setType(type2);
+  ManifestType type_returned2 = manifest_->getType();
+
+  ASSERT_EQ(type1, type_returned1);
+  ASSERT_EQ(type2, type_returned2);
+}
+
+TEST_F(ManifestTest, SetMaxCapacity) {
+  manifest_->Encoder::clear();
+
+  uint8_t max_capacity1 = 0;
+  uint8_t max_capacity2 = 20;
+
+  manifest_->setMaxCapacity(max_capacity1);
+  uint8_t max_capacity_returned1 = manifest_->getMaxCapacity();
+
+  manifest_->Encoder::clear();
+
+  manifest_->setMaxCapacity(max_capacity2);
+  uint8_t max_capacity_returned2 = manifest_->getMaxCapacity();
+
+  ASSERT_EQ(max_capacity1, max_capacity_returned1);
+  ASSERT_EQ(max_capacity2, max_capacity_returned2);
+}
+
+TEST_F(ManifestTest, SetHashAlgorithm) {
+  manifest_->Encoder::clear();
+
+  auth::CryptoHashType hash1 = auth::CryptoHashType::SHA256;
+  auth::CryptoHashType hash2 = auth::CryptoHashType::SHA512;
+  auth::CryptoHashType hash3 = auth::CryptoHashType::BLAKE2B512;
+
+  manifest_->setHashAlgorithm(hash1);
+  auto type_returned1 = manifest_->getHashAlgorithm();
+
+  manifest_->Encoder::clear();
+
+  manifest_->setHashAlgorithm(hash2);
+  auto type_returned2 = manifest_->getHashAlgorithm();
+
+  manifest_->Encoder::clear();
+
+  manifest_->setHashAlgorithm(hash3);
+  auto type_returned3 = manifest_->getHashAlgorithm();
+
+  ASSERT_EQ(hash1, type_returned1);
+  ASSERT_EQ(hash2, type_returned2);
+  ASSERT_EQ(hash3, type_returned3);
+}
+
+TEST_F(ManifestTest, SetLastManifest) {
+  manifest_->Encoder::clear();
+
+  manifest_->setIsLast(true);
+  bool is_last = manifest_->getIsLast();
+
+  ASSERT_TRUE(is_last);
+}
+
+TEST_F(ManifestTest, SetBaseName) {
+  manifest_->Encoder::clear();
+
+  core::Name base_name("b001::dead");
+
+  manifest_->setBaseName(base_name);
+  core::Name ret_name = manifest_->getBaseName();
+
+  ASSERT_EQ(base_name, ret_name);
+}
+
+TEST_F(ManifestTest, setParamsBytestream) {
+  manifest_->Encoder::clear();
+
+  ParamsBytestream params{
+      .final_segment = 0x0a,
+  };
+
+  manifest_->setParamsBytestream(params);
+  auth::CryptoHash hash(auth::CryptoHashType::SHA256);
+  hash.computeDigest({0x01, 0x02, 0x03, 0x04});
+  manifest_->addEntry(1, hash);
+
+  manifest_->encode();
+  manifest_->decode();
+
+  auto transport_type_returned = manifest_->getTransportType();
+  auto params_returned = manifest_->getParamsBytestream();
+
+  ASSERT_EQ(interface::ProductionProtocolAlgorithms::BYTE_STREAM,
+            transport_type_returned);
+  ASSERT_EQ(params, params_returned);
+}
+
+TEST_F(ManifestTest, SetParamsRTC) {
+  manifest_->Encoder::clear();
+
+  ParamsRTC params{
+      .timestamp = 0x0a,
+      .prod_rate = 0x0b,
+      .prod_seg = 0x0c,
+      .fec_type = protocol::fec::FECType::UNKNOWN,
+  };
+
+  manifest_->setParamsRTC(params);
+  auth::CryptoHash hash(auth::CryptoHashType::SHA256);
+  hash.computeDigest({0x01, 0x02, 0x03, 0x04});
+  manifest_->addEntry(1, hash);
+
+  manifest_->encode();
+  manifest_->decode();
+
+  auto transport_type_returned = manifest_->getTransportType();
+  auto params_returned = manifest_->getParamsRTC();
+
+  ASSERT_EQ(interface::ProductionProtocolAlgorithms::RTC_PROD,
+            transport_type_returned);
+  ASSERT_EQ(params, params_returned);
+}
+
+TEST_F(ManifestTest, SignManifest) {
+  auto signer = std::make_shared<auth::SymmetricSigner>(
+      auth::CryptoSuite::HMAC_SHA256, "hunter2");
+  auto verifier = std::make_shared<auth::SymmetricVerifier>("hunter2");
+
+  // Instantiate manifest
+  uint8_t max_capacity = 30;
+  std::shared_ptr<ContentObjectManifest> manifest =
+      ContentObjectManifest::createContentManifest(
+          format_, name_, signer->getSignatureFieldSize());
+  manifest->setHeaders(ManifestType::INLINE_MANIFEST, max_capacity,
+                       signer->getHashType(), false /* is_last */, name_);
+
+  // Add manifest entry
+  auth::CryptoHash hash(signer->getHashType());
+  hash.computeDigest({0x01, 0x02, 0x03, 0x04});
+  manifest->addEntry(1, hash);
+
+  // Encode manifest
+  manifest->encode();
+  auto manifest_co =
+      std::dynamic_pointer_cast<ContentObject>(manifest->getPacket());
+
+  // Sign manifest
+  signer->signPacket(manifest_co.get());
+
+  // Check size
+  ASSERT_EQ(manifest_co->payloadSize(), manifest->Encoder::manifestSize());
+  ASSERT_EQ(manifest_co->length(),
+            manifest_co->headerSize() + manifest_co->payloadSize());
+
+  // Verify manifest
+  auth::VerificationPolicy policy = verifier->verifyPackets(manifest_co.get());
+  ASSERT_EQ(auth::VerificationPolicy::ACCEPT, policy);
+}
+
+TEST_F(ManifestTest, SetSuffixList) {
+  manifest_->Encoder::clear();
+
+  using random_bytes_engine =
+      std::independent_bits_engine<std::default_random_engine, CHAR_BIT,
+                                   unsigned char>;
+  random_bytes_engine rbe;
+
+  std::default_random_engine eng((std::random_device())());
+  std::uniform_int_distribution<uint64_t> idis(
+      0, std::numeric_limits<uint32_t>::max());
+
+  auto entries = new std::pair<uint32_t, auth::CryptoHash>[3];
+  uint32_t suffixes[3];
+  std::vector<unsigned char> data[3];
+
+  for (int i = 0; i < 3; i++) {
+    data[i].resize(32);
+    std::generate(std::begin(data[i]), std::end(data[i]), std::ref(rbe));
+    suffixes[i] = idis(eng);
+    entries[i] = std::make_pair(suffixes[i],
+                                auth::CryptoHash(data[i].data(), data[i].size(),
+                                                 auth::CryptoHashType::SHA256));
+    manifest_->addEntry(entries[i].first, entries[i].second);
+  }
+
+  core::Name base_name("b001::dead");
+  manifest_->setBaseName(base_name);
+
+  core::Name ret_name = manifest_->getBaseName();
+  ASSERT_EQ(base_name, ret_name);
+
+  delete[] entries;
+}
+
+}  // namespace core
+
+}  // namespace transport