Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 1 | /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
Davide Pesavento | 5d0b010 | 2017-10-07 13:43:16 -0400 | [diff] [blame] | 2 | /* |
Eric Newberry | a3c8bd1 | 2020-05-15 17:27:07 -0700 | [diff] [blame] | 3 | * Copyright (c) 2013-2020 Regents of the University of California. |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 4 | * |
| 5 | * This file is part of ndn-cxx library (NDN C++ library with eXperimental eXtensions). |
| 6 | * |
| 7 | * ndn-cxx library is free software: you can redistribute it and/or modify it under the |
| 8 | * terms of the GNU Lesser General Public License as published by the Free Software |
| 9 | * Foundation, either version 3 of the License, or (at your option) any later version. |
| 10 | * |
| 11 | * ndn-cxx library is distributed in the hope that it will be useful, but WITHOUT ANY |
| 12 | * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A |
| 13 | * PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. |
| 14 | * |
| 15 | * You should have received copies of the GNU General Public License and GNU Lesser |
| 16 | * General Public License along with ndn-cxx, e.g., in COPYING.md file. If not, see |
| 17 | * <http://www.gnu.org/licenses/>. |
| 18 | * |
| 19 | * See AUTHORS.md for complete list of ndn-cxx authors and contributors. |
| 20 | */ |
| 21 | |
Davide Pesavento | 4fb35d8 | 2019-10-31 19:33:10 -0400 | [diff] [blame] | 22 | #include "ndn-cxx/security/pib/impl/pib-memory.hpp" |
| 23 | #include "ndn-cxx/security/pib/impl/pib-sqlite3.hpp" |
Davide Pesavento | 7e78064 | 2018-11-24 15:51:34 -0500 | [diff] [blame] | 24 | #include "ndn-cxx/security/pib/pib.hpp" |
| 25 | #include "ndn-cxx/security/security-common.hpp" |
Davide Pesavento | eee3e82 | 2016-11-26 19:19:34 +0100 | [diff] [blame] | 26 | |
Davide Pesavento | 7e78064 | 2018-11-24 15:51:34 -0500 | [diff] [blame] | 27 | #include "tests/boost-test.hpp" |
| 28 | #include "tests/unit/security/pib/pib-data-fixture.hpp" |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 29 | |
Mickey Sweatt | 11314b7 | 2015-06-10 17:20:19 -0700 | [diff] [blame] | 30 | #include <boost/filesystem.hpp> |
Davide Pesavento | 77d9e81 | 2019-06-03 22:05:54 -0400 | [diff] [blame] | 31 | #include <boost/mpl/vector.hpp> |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 32 | |
| 33 | namespace ndn { |
| 34 | namespace security { |
Yingdi Yu | 6ee2d36 | 2015-07-16 21:48:05 -0700 | [diff] [blame] | 35 | namespace pib { |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 36 | namespace tests { |
| 37 | |
Yingdi Yu | 6ee2d36 | 2015-07-16 21:48:05 -0700 | [diff] [blame] | 38 | using namespace ndn::security::tests; |
| 39 | |
Davide Pesavento | eee3e82 | 2016-11-26 19:19:34 +0100 | [diff] [blame] | 40 | BOOST_AUTO_TEST_SUITE(Security) |
Yingdi Yu | 6ee2d36 | 2015-07-16 21:48:05 -0700 | [diff] [blame] | 41 | BOOST_AUTO_TEST_SUITE(Pib) |
Davide Pesavento | eee3e82 | 2016-11-26 19:19:34 +0100 | [diff] [blame] | 42 | BOOST_AUTO_TEST_SUITE(TestPibImpl) |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 43 | |
Yingdi Yu | 6ee2d36 | 2015-07-16 21:48:05 -0700 | [diff] [blame] | 44 | using pib::Pib; |
| 45 | |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 46 | class PibMemoryFixture : public PibDataFixture |
Mickey Sweatt | 11314b7 | 2015-06-10 17:20:19 -0700 | [diff] [blame] | 47 | { |
| 48 | public: |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 49 | PibMemory pib; |
Mickey Sweatt | 11314b7 | 2015-06-10 17:20:19 -0700 | [diff] [blame] | 50 | }; |
| 51 | |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 52 | class PibSqlite3Fixture : public PibDataFixture |
Mickey Sweatt | 11314b7 | 2015-06-10 17:20:19 -0700 | [diff] [blame] | 53 | { |
| 54 | public: |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 55 | ~PibSqlite3Fixture() |
Mickey Sweatt | 11314b7 | 2015-06-10 17:20:19 -0700 | [diff] [blame] | 56 | { |
Davide Pesavento | 4c1ad4c | 2020-11-16 21:12:02 -0500 | [diff] [blame] | 57 | boost::filesystem::remove_all(m_path); |
Mickey Sweatt | 11314b7 | 2015-06-10 17:20:19 -0700 | [diff] [blame] | 58 | } |
| 59 | |
Davide Pesavento | 4c1ad4c | 2020-11-16 21:12:02 -0500 | [diff] [blame] | 60 | private: |
| 61 | const boost::filesystem::path m_path{boost::filesystem::path(UNIT_TESTS_TMPDIR) / "TestPibImpl"}; |
| 62 | |
Davide Pesavento | eee3e82 | 2016-11-26 19:19:34 +0100 | [diff] [blame] | 63 | public: |
Davide Pesavento | 4c1ad4c | 2020-11-16 21:12:02 -0500 | [diff] [blame] | 64 | PibSqlite3 pib{m_path.string()}; |
Mickey Sweatt | 11314b7 | 2015-06-10 17:20:19 -0700 | [diff] [blame] | 65 | }; |
| 66 | |
Davide Pesavento | 77d9e81 | 2019-06-03 22:05:54 -0400 | [diff] [blame] | 67 | using PibImpls = boost::mpl::vector<PibMemoryFixture, PibSqlite3Fixture>; |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 68 | |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 69 | BOOST_FIXTURE_TEST_CASE_TEMPLATE(TpmLocator, T, PibImpls, T) |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 70 | { |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 71 | // Basic getting and setting |
| 72 | BOOST_CHECK_NO_THROW(this->pib.getTpmLocator()); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 73 | |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 74 | BOOST_CHECK_NO_THROW(this->pib.setTpmLocator("tpmLocator")); |
| 75 | BOOST_CHECK_EQUAL(this->pib.getTpmLocator(), "tpmLocator"); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 76 | |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 77 | // Add cert, and do not change TPM locator |
| 78 | this->pib.addCertificate(this->id1Key1Cert1); |
| 79 | BOOST_CHECK(this->pib.hasIdentity(this->id1)); |
| 80 | BOOST_CHECK(this->pib.hasKey(this->id1Key1Name)); |
| 81 | BOOST_CHECK(this->pib.hasCertificate(this->id1Key1Cert1.getName())); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 82 | |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 83 | // Set TPM locator to the same value, nothing should change |
| 84 | this->pib.setTpmLocator("tpmLocator"); |
| 85 | BOOST_CHECK(this->pib.hasIdentity(this->id1)); |
| 86 | BOOST_CHECK(this->pib.hasKey(this->id1Key1Name)); |
| 87 | BOOST_CHECK(this->pib.hasCertificate(this->id1Key1Cert1.getName())); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 88 | |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 89 | // Change TPM locator (contents of PIB should not change) |
| 90 | this->pib.setTpmLocator("newTpmLocator"); |
| 91 | BOOST_CHECK(this->pib.hasIdentity(this->id1)); |
| 92 | BOOST_CHECK(this->pib.hasKey(this->id1Key1Name)); |
| 93 | BOOST_CHECK(this->pib.hasCertificate(this->id1Key1Cert1.getName())); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 94 | } |
| 95 | |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 96 | BOOST_FIXTURE_TEST_CASE_TEMPLATE(IdentityManagement, T, PibImpls, T) |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 97 | { |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 98 | // no default setting, throw Error |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 99 | BOOST_CHECK_THROW(this->pib.getDefaultIdentity(), Pib::Error); |
| 100 | |
| 101 | // check id1, which should not exist |
| 102 | BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), false); |
| 103 | |
| 104 | // add id1, should be default |
| 105 | this->pib.addIdentity(this->id1); |
| 106 | BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), true); |
| 107 | BOOST_CHECK_NO_THROW(this->pib.getDefaultIdentity()); |
| 108 | BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id1); |
| 109 | |
| 110 | // add id2, should not be default |
| 111 | this->pib.addIdentity(this->id2); |
| 112 | BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id2), true); |
| 113 | BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id1); |
| 114 | |
| 115 | // set id2 explicitly as default |
| 116 | this->pib.setDefaultIdentity(this->id2); |
| 117 | BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id2); |
| 118 | |
| 119 | // remove id2, should not have default identity |
| 120 | this->pib.removeIdentity(this->id2); |
| 121 | BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id2), false); |
| 122 | BOOST_CHECK_THROW(this->pib.getDefaultIdentity(), Pib::Error); |
| 123 | |
| 124 | // add id2 again, should be default |
| 125 | this->pib.addIdentity(this->id2); |
| 126 | BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id2); |
| 127 | |
Alexander Afanasyev | 355fd78 | 2020-06-10 16:40:33 -0400 | [diff] [blame] | 128 | // try to set non-existing identity as a default |
| 129 | BOOST_CHECK_THROW(this->pib.setDefaultIdentity("/non-existing-identity"), Pib::Error); |
| 130 | |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 131 | // get all identities, should contain id1 and id2 |
| 132 | std::set<Name> idNames = this->pib.getIdentities(); |
| 133 | BOOST_CHECK_EQUAL(idNames.size(), 2); |
| 134 | BOOST_CHECK_EQUAL(idNames.count(this->id1), 1); |
| 135 | BOOST_CHECK_EQUAL(idNames.count(this->id2), 1); |
| 136 | } |
| 137 | |
| 138 | BOOST_FIXTURE_TEST_CASE_TEMPLATE(ClearIdentities, T, PibImpls, T) |
| 139 | { |
| 140 | this->pib.setTpmLocator("tpmLocator"); |
| 141 | |
| 142 | // Add id, key, and cert |
| 143 | this->pib.addCertificate(this->id1Key1Cert1); |
| 144 | BOOST_CHECK(this->pib.hasIdentity(this->id1)); |
| 145 | BOOST_CHECK(this->pib.hasKey(this->id1Key1Name)); |
| 146 | BOOST_CHECK(this->pib.hasCertificate(this->id1Key1Cert1.getName())); |
| 147 | |
| 148 | // Clear identities |
| 149 | this->pib.clearIdentities(); |
| 150 | BOOST_CHECK_EQUAL(this->pib.getIdentities().size(), 0); |
| 151 | BOOST_CHECK_EQUAL(this->pib.getKeysOfIdentity(this->id1).size(), 0); |
| 152 | BOOST_CHECK_EQUAL(this->pib.getCertificatesOfKey(this->id1Key1Name).size(), 0); |
| 153 | BOOST_CHECK_EQUAL(this->pib.getTpmLocator(), "tpmLocator"); |
| 154 | } |
| 155 | |
| 156 | BOOST_FIXTURE_TEST_CASE_TEMPLATE(KeyManagement, T, PibImpls, T) |
| 157 | { |
| 158 | // no default setting, throw Error |
| 159 | BOOST_CHECK_THROW(this->pib.getDefaultKeyOfIdentity(this->id1), Pib::Error); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 160 | |
| 161 | // check id1Key1, should not exist, neither should id1. |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 162 | BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), false); |
| 163 | BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), false); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 164 | |
| 165 | // add id1Key1, should be default, id1 should be added implicitly |
Davide Pesavento | 5d0b010 | 2017-10-07 13:43:16 -0400 | [diff] [blame] | 166 | this->pib.addKey(this->id1, this->id1Key1Name, this->id1Key1.data(), this->id1Key1.size()); |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 167 | BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), true); |
| 168 | BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), true); |
| 169 | const Buffer& keyBits = this->pib.getKeyBits(this->id1Key1Name); |
Yingdi Yu | cbe72b0 | 2015-11-25 17:35:37 -0800 | [diff] [blame] | 170 | BOOST_CHECK(keyBits == this->id1Key1); |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 171 | BOOST_CHECK_NO_THROW(this->pib.getDefaultKeyOfIdentity(this->id1)); |
| 172 | BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id1), this->id1Key1Name); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 173 | |
| 174 | // add id1Key2, should not be default |
Davide Pesavento | 5d0b010 | 2017-10-07 13:43:16 -0400 | [diff] [blame] | 175 | this->pib.addKey(this->id1, this->id1Key2Name, this->id1Key2.data(), this->id1Key2.size()); |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 176 | BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key2Name), true); |
| 177 | BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id1), this->id1Key1Name); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 178 | |
| 179 | // set id1Key2 explicitly as default |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 180 | this->pib.setDefaultKeyOfIdentity(this->id1, this->id1Key2Name); |
| 181 | BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id1), this->id1Key2Name); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 182 | |
| 183 | // set a non-existing key as default, throw Error |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 184 | BOOST_CHECK_THROW(this->pib.setDefaultKeyOfIdentity(this->id1, Name("/non-existing")), |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 185 | Pib::Error); |
| 186 | |
| 187 | // remove id1Key2, should not have default key |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 188 | this->pib.removeKey(this->id1Key2Name); |
| 189 | BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key2Name), false); |
| 190 | BOOST_CHECK_THROW(this->pib.getKeyBits(this->id1Key2Name), Pib::Error); |
| 191 | BOOST_CHECK_THROW(this->pib.getDefaultKeyOfIdentity(this->id1), Pib::Error); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 192 | |
| 193 | // add id1Key2 back, should be default |
Davide Pesavento | 5d0b010 | 2017-10-07 13:43:16 -0400 | [diff] [blame] | 194 | this->pib.addKey(this->id1, this->id1Key2Name, this->id1Key2.data(), this->id1Key2.size()); |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 195 | BOOST_CHECK_NO_THROW(this->pib.getKeyBits(this->id1Key2Name)); |
| 196 | BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id1), this->id1Key2Name); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 197 | |
| 198 | // get all the keys: id1Key1 and id1Key2 |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 199 | std::set<Name> keyNames = this->pib.getKeysOfIdentity(this->id1); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 200 | BOOST_CHECK_EQUAL(keyNames.size(), 2); |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 201 | BOOST_CHECK_EQUAL(keyNames.count(this->id1Key1Name), 1); |
| 202 | BOOST_CHECK_EQUAL(keyNames.count(this->id1Key2Name), 1); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 203 | |
| 204 | // remove id1, should remove all the keys |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 205 | this->pib.removeIdentity(this->id1); |
| 206 | keyNames = this->pib.getKeysOfIdentity(this->id1); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 207 | BOOST_CHECK_EQUAL(keyNames.size(), 0); |
| 208 | } |
| 209 | |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 210 | BOOST_FIXTURE_TEST_CASE_TEMPLATE(CertificateManagement, T, PibImpls, T) |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 211 | { |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 212 | // no default setting, throw Error |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 213 | BOOST_CHECK_THROW(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), Pib::Error); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 214 | |
| 215 | // check id1Key1Cert1, should not exist, neither should id1 and id1Key1 |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 216 | BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert1.getName()), false); |
| 217 | BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), false); |
| 218 | BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), false); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 219 | |
| 220 | // add id1Key1Cert1, should be default, id1 and id1Key1 should be added implicitly |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 221 | this->pib.addCertificate(this->id1Key1Cert1); |
| 222 | BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert1.getName()), true); |
| 223 | BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), true); |
| 224 | BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), true); |
Junxiao Shi | 72c0c64 | 2018-04-20 15:41:09 +0000 | [diff] [blame] | 225 | BOOST_CHECK_EQUAL(this->pib.getCertificate(this->id1Key1Cert1.getName()).wireEncode(), |
| 226 | this->id1Key1Cert1.wireEncode()); |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 227 | BOOST_CHECK_NO_THROW(this->pib.getDefaultCertificateOfKey(this->id1Key1Name)); |
| 228 | BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), this->id1Key1Cert1); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 229 | |
| 230 | // add id1Key1Cert2, should not be default |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 231 | this->pib.addCertificate(this->id1Key1Cert2); |
| 232 | BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert2.getName()), true); |
| 233 | BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), this->id1Key1Cert1); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 234 | |
| 235 | // set id1Key1Cert2 explicitly as default |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 236 | this->pib.setDefaultCertificateOfKey(this->id1Key1Name, this->id1Key1Cert2.getName()); |
| 237 | BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), this->id1Key1Cert2); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 238 | |
| 239 | // set a non-existing cert as default, throw Error |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 240 | BOOST_CHECK_THROW(this->pib.setDefaultCertificateOfKey(this->id1Key1Name, Name("/non-existing")), |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 241 | Pib::Error); |
| 242 | |
| 243 | // remove id1Key1Cert2, should not have default cert |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 244 | this->pib.removeCertificate(this->id1Key1Cert2.getName()); |
| 245 | BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert2.getName()), false); |
| 246 | BOOST_CHECK_THROW(this->pib.getCertificate(this->id1Key1Cert2.getName()), Pib::Error); |
| 247 | BOOST_CHECK_THROW(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), Pib::Error); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 248 | |
| 249 | // add id1Key1Cert2, should be default |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 250 | this->pib.addCertificate(this->id1Key1Cert2); |
| 251 | BOOST_CHECK_NO_THROW(this->pib.getCertificate(this->id1Key1Cert1.getName())); |
| 252 | BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), this->id1Key1Cert2); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 253 | |
| 254 | // get all certificates: id1Key1Cert1 and id1Key1Cert2 |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 255 | std::set<Name> certNames = this->pib.getCertificatesOfKey(this->id1Key1Name); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 256 | BOOST_CHECK_EQUAL(certNames.size(), 2); |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 257 | BOOST_CHECK_EQUAL(certNames.count(this->id1Key1Cert1.getName()), 1); |
| 258 | BOOST_CHECK_EQUAL(certNames.count(this->id1Key1Cert2.getName()), 1); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 259 | |
| 260 | // remove id1Key1, should remove all the certs |
Yingdi Yu | 7b3b5e9 | 2015-08-13 19:52:35 -0700 | [diff] [blame] | 261 | this->pib.removeKey(this->id1Key1Name); |
| 262 | certNames = this->pib.getCertificatesOfKey(this->id1Key1Name); |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 263 | BOOST_CHECK_EQUAL(certNames.size(), 0); |
| 264 | } |
| 265 | |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 266 | BOOST_FIXTURE_TEST_CASE_TEMPLATE(DefaultsManagement, T, PibImpls, T) |
| 267 | { |
| 268 | this->pib.addIdentity(this->id1); |
| 269 | BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id1); |
| 270 | |
| 271 | this->pib.addIdentity(this->id2); |
| 272 | BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id1); |
| 273 | |
| 274 | this->pib.removeIdentity(this->id1); |
| 275 | BOOST_CHECK_THROW(this->pib.getDefaultIdentity(), Pib::Error); |
| 276 | |
Davide Pesavento | 5d0b010 | 2017-10-07 13:43:16 -0400 | [diff] [blame] | 277 | this->pib.addKey(this->id2, this->id2Key1Name, this->id2Key1.data(), this->id2Key1.size()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 278 | BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id2); |
| 279 | BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id2), this->id2Key1Name); |
| 280 | |
Davide Pesavento | 5d0b010 | 2017-10-07 13:43:16 -0400 | [diff] [blame] | 281 | this->pib.addKey(this->id2, this->id2Key2Name, this->id2Key2.data(), this->id2Key2.size()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 282 | BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id2), this->id2Key1Name); |
| 283 | |
| 284 | this->pib.removeKey(this->id2Key1Name); |
| 285 | BOOST_CHECK_THROW(this->pib.getDefaultKeyOfIdentity(this->id2), Pib::Error); |
| 286 | |
| 287 | this->pib.addCertificate(this->id2Key2Cert1); |
| 288 | BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id2), this->id2Key2Name); |
Davide Pesavento | 77d9e81 | 2019-06-03 22:05:54 -0400 | [diff] [blame] | 289 | BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id2Key2Name).getName(), |
| 290 | this->id2Key2Cert1.getName()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 291 | |
| 292 | this->pib.addCertificate(this->id2Key2Cert2); |
Davide Pesavento | 77d9e81 | 2019-06-03 22:05:54 -0400 | [diff] [blame] | 293 | BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id2Key2Name).getName(), |
| 294 | this->id2Key2Cert1.getName()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 295 | |
| 296 | this->pib.removeCertificate(this->id2Key2Cert2.getName()); |
Davide Pesavento | 77d9e81 | 2019-06-03 22:05:54 -0400 | [diff] [blame] | 297 | BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id2Key2Name).getName(), |
| 298 | this->id2Key2Cert1.getName()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 299 | } |
| 300 | |
| 301 | BOOST_FIXTURE_TEST_CASE_TEMPLATE(Overwrite, T, PibImpls, T) |
| 302 | { |
| 303 | // check id1Key1, should not exist |
| 304 | this->pib.removeIdentity(this->id1); |
| 305 | BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), false); |
| 306 | |
| 307 | // add id1Key1 |
Davide Pesavento | 5d0b010 | 2017-10-07 13:43:16 -0400 | [diff] [blame] | 308 | this->pib.addKey(this->id1, this->id1Key1Name, this->id1Key1.data(), this->id1Key1.size()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 309 | BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), true); |
| 310 | const Buffer& keyBits = this->pib.getKeyBits(this->id1Key1Name); |
| 311 | BOOST_CHECK(keyBits == this->id1Key1); |
| 312 | |
| 313 | // check overwrite, add a key with the same name. |
Davide Pesavento | 5d0b010 | 2017-10-07 13:43:16 -0400 | [diff] [blame] | 314 | this->pib.addKey(this->id1, this->id1Key1Name, this->id1Key2.data(), this->id1Key2.size()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 315 | const Buffer& keyBits2 = this->pib.getKeyBits(this->id1Key1Name); |
| 316 | BOOST_CHECK(keyBits2 == this->id1Key2); |
| 317 | |
| 318 | // check id1Key1Cert1, should not exist |
| 319 | this->pib.removeIdentity(this->id1); |
| 320 | BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert1.getName()), false); |
| 321 | |
| 322 | // add id1Key1Cert1 |
Davide Pesavento | 5d0b010 | 2017-10-07 13:43:16 -0400 | [diff] [blame] | 323 | this->pib.addKey(this->id1, this->id1Key1Name, this->id1Key1.data(), this->id1Key1.size()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 324 | this->pib.addCertificate(this->id1Key1Cert1); |
| 325 | BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert1.getName()), true); |
| 326 | |
| 327 | auto cert = this->pib.getCertificate(this->id1Key1Cert1.getName()); |
Junxiao Shi | 72c0c64 | 2018-04-20 15:41:09 +0000 | [diff] [blame] | 328 | BOOST_CHECK_EQUAL(cert.wireEncode(), this->id1Key1Cert1.wireEncode()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 329 | |
| 330 | // Create a fake cert with the same name |
| 331 | auto cert2 = this->id1Key2Cert1; |
| 332 | cert2.setName(this->id1Key1Cert1.getName()); |
Davide Pesavento | 14c56cd | 2020-05-21 01:44:03 -0400 | [diff] [blame] | 333 | BOOST_CHECK_EQUAL(cert2.getSignatureInfo(), this->id1Key2Cert1.getSignatureInfo()); |
| 334 | BOOST_CHECK_EQUAL(cert2.getSignatureValue(), this->id1Key2Cert1.getSignatureValue()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 335 | this->pib.addCertificate(cert2); |
| 336 | |
| 337 | auto cert3 = this->pib.getCertificate(this->id1Key1Cert1.getName()); |
Junxiao Shi | 72c0c64 | 2018-04-20 15:41:09 +0000 | [diff] [blame] | 338 | BOOST_CHECK_EQUAL(cert3.wireEncode(), cert2.wireEncode()); |
Yingdi Yu | 0399768 | 2015-11-23 16:41:38 -0800 | [diff] [blame] | 339 | |
| 340 | // both key and certificate are overwritten |
| 341 | Buffer keyBits3 = this->pib.getKeyBits(this->id1Key1Name); |
| 342 | BOOST_CHECK(keyBits3 == this->id1Key2); |
| 343 | } |
| 344 | |
Davide Pesavento | eee3e82 | 2016-11-26 19:19:34 +0100 | [diff] [blame] | 345 | BOOST_AUTO_TEST_SUITE_END() // TestPibImpl |
Yingdi Yu | 6ee2d36 | 2015-07-16 21:48:05 -0700 | [diff] [blame] | 346 | BOOST_AUTO_TEST_SUITE_END() // Pib |
Davide Pesavento | eee3e82 | 2016-11-26 19:19:34 +0100 | [diff] [blame] | 347 | BOOST_AUTO_TEST_SUITE_END() // Security |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 348 | |
| 349 | } // namespace tests |
Yingdi Yu | 6ee2d36 | 2015-07-16 21:48:05 -0700 | [diff] [blame] | 350 | } // namespace pib |
Yingdi Yu | 3bf91f5 | 2015-06-12 19:39:40 -0700 | [diff] [blame] | 351 | } // namespace security |
| 352 | } // namespace ndn |