blob: 2677e9dd1d52f6889db2b09940eb2ceb928bf4ea [file] [log] [blame]
Yingdi Yu3bf91f52015-06-12 19:39:40 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
Yingdi Yu6ee2d362015-07-16 21:48:05 -07003 * Copyright (c) 2013-2017 Regents of the University of California.
Yingdi Yu3bf91f52015-06-12 19:39:40 -07004 *
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
Alexander Afanasyev97709c02016-08-25 19:58:30 -070022#include "security/pib/pib-memory.hpp"
23#include "security/pib/pib-sqlite3.hpp"
24#include "security/pib/pib.hpp"
Yingdi Yu6ee2d362015-07-16 21:48:05 -070025#include "security/security-common.hpp"
Davide Pesaventoeee3e822016-11-26 19:19:34 +010026
27#include "boost-test.hpp"
Yingdi Yu3bf91f52015-06-12 19:39:40 -070028#include "pib-data-fixture.hpp"
29
Mickey Sweatt11314b72015-06-10 17:20:19 -070030#include <boost/filesystem.hpp>
Yingdi Yu3bf91f52015-06-12 19:39:40 -070031#include <boost/mpl/list.hpp>
Yingdi Yu3bf91f52015-06-12 19:39:40 -070032
33namespace ndn {
34namespace security {
Yingdi Yu6ee2d362015-07-16 21:48:05 -070035namespace pib {
Yingdi Yu3bf91f52015-06-12 19:39:40 -070036namespace tests {
37
Yingdi Yu6ee2d362015-07-16 21:48:05 -070038using namespace ndn::security::tests;
39
Davide Pesaventoeee3e822016-11-26 19:19:34 +010040BOOST_AUTO_TEST_SUITE(Security)
Yingdi Yu6ee2d362015-07-16 21:48:05 -070041BOOST_AUTO_TEST_SUITE(Pib)
Davide Pesaventoeee3e822016-11-26 19:19:34 +010042BOOST_AUTO_TEST_SUITE(TestPibImpl)
Yingdi Yu3bf91f52015-06-12 19:39:40 -070043
Yingdi Yu6ee2d362015-07-16 21:48:05 -070044using pib::Pib;
45
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070046class PibMemoryFixture : public PibDataFixture
Mickey Sweatt11314b72015-06-10 17:20:19 -070047{
48public:
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070049 PibMemory pib;
Mickey Sweatt11314b72015-06-10 17:20:19 -070050};
51
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070052class PibSqlite3Fixture : public PibDataFixture
Mickey Sweatt11314b72015-06-10 17:20:19 -070053{
54public:
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070055 PibSqlite3Fixture()
Mickey Sweatt11314b72015-06-10 17:20:19 -070056 : tmpPath(boost::filesystem::path(UNIT_TEST_CONFIG_PATH) / "DbTest")
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070057 , pib(tmpPath.c_str())
Mickey Sweatt11314b72015-06-10 17:20:19 -070058 {
59 }
60
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070061 ~PibSqlite3Fixture()
Mickey Sweatt11314b72015-06-10 17:20:19 -070062 {
63 boost::filesystem::remove_all(tmpPath);
64 }
65
Davide Pesaventoeee3e822016-11-26 19:19:34 +010066public:
Mickey Sweatt11314b72015-06-10 17:20:19 -070067 boost::filesystem::path tmpPath;
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070068 PibSqlite3 pib;
Mickey Sweatt11314b72015-06-10 17:20:19 -070069};
70
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070071typedef boost::mpl::list<PibMemoryFixture,
72 PibSqlite3Fixture> PibImpls;
Yingdi Yu3bf91f52015-06-12 19:39:40 -070073
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070074BOOST_FIXTURE_TEST_CASE_TEMPLATE(TpmLocator, T, PibImpls, T)
Yingdi Yu3bf91f52015-06-12 19:39:40 -070075{
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070076 // Basic getting and setting
77 BOOST_CHECK_NO_THROW(this->pib.getTpmLocator());
Yingdi Yu3bf91f52015-06-12 19:39:40 -070078
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070079 BOOST_CHECK_NO_THROW(this->pib.setTpmLocator("tpmLocator"));
80 BOOST_CHECK_EQUAL(this->pib.getTpmLocator(), "tpmLocator");
Yingdi Yu3bf91f52015-06-12 19:39:40 -070081
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070082 // Add cert, and do not change TPM locator
83 this->pib.addCertificate(this->id1Key1Cert1);
84 BOOST_CHECK(this->pib.hasIdentity(this->id1));
85 BOOST_CHECK(this->pib.hasKey(this->id1Key1Name));
86 BOOST_CHECK(this->pib.hasCertificate(this->id1Key1Cert1.getName()));
Yingdi Yu3bf91f52015-06-12 19:39:40 -070087
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070088 // Set TPM locator to the same value, nothing should change
89 this->pib.setTpmLocator("tpmLocator");
90 BOOST_CHECK(this->pib.hasIdentity(this->id1));
91 BOOST_CHECK(this->pib.hasKey(this->id1Key1Name));
92 BOOST_CHECK(this->pib.hasCertificate(this->id1Key1Cert1.getName()));
Yingdi Yu3bf91f52015-06-12 19:39:40 -070093
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070094 // Change TPM locator (contents of PIB should not change)
95 this->pib.setTpmLocator("newTpmLocator");
96 BOOST_CHECK(this->pib.hasIdentity(this->id1));
97 BOOST_CHECK(this->pib.hasKey(this->id1Key1Name));
98 BOOST_CHECK(this->pib.hasCertificate(this->id1Key1Cert1.getName()));
Yingdi Yu3bf91f52015-06-12 19:39:40 -070099}
100
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700101BOOST_FIXTURE_TEST_CASE_TEMPLATE(IdentityManagement, T, PibImpls, T)
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700102{
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700103 // no default setting, throw Error
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700104 BOOST_CHECK_THROW(this->pib.getDefaultIdentity(), Pib::Error);
105
106 // check id1, which should not exist
107 BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), false);
108
109 // add id1, should be default
110 this->pib.addIdentity(this->id1);
111 BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), true);
112 BOOST_CHECK_NO_THROW(this->pib.getDefaultIdentity());
113 BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id1);
114
115 // add id2, should not be default
116 this->pib.addIdentity(this->id2);
117 BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id2), true);
118 BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id1);
119
120 // set id2 explicitly as default
121 this->pib.setDefaultIdentity(this->id2);
122 BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id2);
123
124 // remove id2, should not have default identity
125 this->pib.removeIdentity(this->id2);
126 BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id2), false);
127 BOOST_CHECK_THROW(this->pib.getDefaultIdentity(), Pib::Error);
128
129 // add id2 again, should be default
130 this->pib.addIdentity(this->id2);
131 BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id2);
132
133 // get all identities, should contain id1 and id2
134 std::set<Name> idNames = this->pib.getIdentities();
135 BOOST_CHECK_EQUAL(idNames.size(), 2);
136 BOOST_CHECK_EQUAL(idNames.count(this->id1), 1);
137 BOOST_CHECK_EQUAL(idNames.count(this->id2), 1);
138}
139
140BOOST_FIXTURE_TEST_CASE_TEMPLATE(ClearIdentities, T, PibImpls, T)
141{
142 this->pib.setTpmLocator("tpmLocator");
143
144 // Add id, key, and cert
145 this->pib.addCertificate(this->id1Key1Cert1);
146 BOOST_CHECK(this->pib.hasIdentity(this->id1));
147 BOOST_CHECK(this->pib.hasKey(this->id1Key1Name));
148 BOOST_CHECK(this->pib.hasCertificate(this->id1Key1Cert1.getName()));
149
150 // Clear identities
151 this->pib.clearIdentities();
152 BOOST_CHECK_EQUAL(this->pib.getIdentities().size(), 0);
153 BOOST_CHECK_EQUAL(this->pib.getKeysOfIdentity(this->id1).size(), 0);
154 BOOST_CHECK_EQUAL(this->pib.getCertificatesOfKey(this->id1Key1Name).size(), 0);
155 BOOST_CHECK_EQUAL(this->pib.getTpmLocator(), "tpmLocator");
156}
157
158BOOST_FIXTURE_TEST_CASE_TEMPLATE(KeyManagement, T, PibImpls, T)
159{
160 // no default setting, throw Error
161 BOOST_CHECK_THROW(this->pib.getDefaultKeyOfIdentity(this->id1), Pib::Error);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700162
163 // check id1Key1, should not exist, neither should id1.
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700164 BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), false);
165 BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), false);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700166
167 // add id1Key1, should be default, id1 should be added implicitly
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700168 this->pib.addKey(this->id1, this->id1Key1Name, this->id1Key1.buf(), this->id1Key1.size());
169 BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), true);
170 BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), true);
171 const Buffer& keyBits = this->pib.getKeyBits(this->id1Key1Name);
Yingdi Yucbe72b02015-11-25 17:35:37 -0800172 BOOST_CHECK(keyBits == this->id1Key1);
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700173 BOOST_CHECK_NO_THROW(this->pib.getDefaultKeyOfIdentity(this->id1));
174 BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id1), this->id1Key1Name);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700175
176 // add id1Key2, should not be default
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700177 this->pib.addKey(this->id1, this->id1Key2Name, this->id1Key2.buf(), this->id1Key2.size());
178 BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key2Name), true);
179 BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id1), this->id1Key1Name);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700180
181 // set id1Key2 explicitly as default
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700182 this->pib.setDefaultKeyOfIdentity(this->id1, this->id1Key2Name);
183 BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id1), this->id1Key2Name);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700184
185 // set a non-existing key as default, throw Error
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700186 BOOST_CHECK_THROW(this->pib.setDefaultKeyOfIdentity(this->id1, Name("/non-existing")),
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700187 Pib::Error);
188
189 // remove id1Key2, should not have default key
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700190 this->pib.removeKey(this->id1Key2Name);
191 BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key2Name), false);
192 BOOST_CHECK_THROW(this->pib.getKeyBits(this->id1Key2Name), Pib::Error);
193 BOOST_CHECK_THROW(this->pib.getDefaultKeyOfIdentity(this->id1), Pib::Error);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700194
195 // add id1Key2 back, should be default
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700196 this->pib.addKey(this->id1, this->id1Key2Name, this->id1Key2.buf(), this->id1Key2.size());
197 BOOST_CHECK_NO_THROW(this->pib.getKeyBits(this->id1Key2Name));
198 BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id1), this->id1Key2Name);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700199
200 // get all the keys: id1Key1 and id1Key2
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700201 std::set<Name> keyNames = this->pib.getKeysOfIdentity(this->id1);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700202 BOOST_CHECK_EQUAL(keyNames.size(), 2);
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700203 BOOST_CHECK_EQUAL(keyNames.count(this->id1Key1Name), 1);
204 BOOST_CHECK_EQUAL(keyNames.count(this->id1Key2Name), 1);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700205
206 // remove id1, should remove all the keys
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700207 this->pib.removeIdentity(this->id1);
208 keyNames = this->pib.getKeysOfIdentity(this->id1);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700209 BOOST_CHECK_EQUAL(keyNames.size(), 0);
210}
211
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700212BOOST_FIXTURE_TEST_CASE_TEMPLATE(CertificateManagement, T, PibImpls, T)
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700213{
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700214 // no default setting, throw Error
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700215 BOOST_CHECK_THROW(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), Pib::Error);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700216
217 // check id1Key1Cert1, should not exist, neither should id1 and id1Key1
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700218 BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert1.getName()), false);
219 BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), false);
220 BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), false);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700221
222 // add id1Key1Cert1, should be default, id1 and id1Key1 should be added implicitly
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700223 this->pib.addCertificate(this->id1Key1Cert1);
224 BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert1.getName()), true);
225 BOOST_CHECK_EQUAL(this->pib.hasIdentity(this->id1), true);
226 BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), true);
Yingdi Yu03997682015-11-23 16:41:38 -0800227 BOOST_CHECK(this->pib.getCertificate(this->id1Key1Cert1.getName()).wireEncode() == this->id1Key1Cert1.wireEncode());
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700228 BOOST_CHECK_NO_THROW(this->pib.getDefaultCertificateOfKey(this->id1Key1Name));
229 BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), this->id1Key1Cert1);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700230
231 // add id1Key1Cert2, should not be default
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700232 this->pib.addCertificate(this->id1Key1Cert2);
233 BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert2.getName()), true);
234 BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), this->id1Key1Cert1);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700235
236 // set id1Key1Cert2 explicitly as default
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700237 this->pib.setDefaultCertificateOfKey(this->id1Key1Name, this->id1Key1Cert2.getName());
238 BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), this->id1Key1Cert2);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700239
240 // set a non-existing cert as default, throw Error
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700241 BOOST_CHECK_THROW(this->pib.setDefaultCertificateOfKey(this->id1Key1Name, Name("/non-existing")),
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700242 Pib::Error);
243
244 // remove id1Key1Cert2, should not have default cert
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700245 this->pib.removeCertificate(this->id1Key1Cert2.getName());
246 BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert2.getName()), false);
247 BOOST_CHECK_THROW(this->pib.getCertificate(this->id1Key1Cert2.getName()), Pib::Error);
248 BOOST_CHECK_THROW(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), Pib::Error);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700249
250 // add id1Key1Cert2, should be default
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700251 this->pib.addCertificate(this->id1Key1Cert2);
252 BOOST_CHECK_NO_THROW(this->pib.getCertificate(this->id1Key1Cert1.getName()));
253 BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id1Key1Name), this->id1Key1Cert2);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700254
255 // get all certificates: id1Key1Cert1 and id1Key1Cert2
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700256 std::set<Name> certNames = this->pib.getCertificatesOfKey(this->id1Key1Name);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700257 BOOST_CHECK_EQUAL(certNames.size(), 2);
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700258 BOOST_CHECK_EQUAL(certNames.count(this->id1Key1Cert1.getName()), 1);
259 BOOST_CHECK_EQUAL(certNames.count(this->id1Key1Cert2.getName()), 1);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700260
261 // remove id1Key1, should remove all the certs
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700262 this->pib.removeKey(this->id1Key1Name);
263 certNames = this->pib.getCertificatesOfKey(this->id1Key1Name);
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700264 BOOST_CHECK_EQUAL(certNames.size(), 0);
265}
266
Yingdi Yu03997682015-11-23 16:41:38 -0800267BOOST_FIXTURE_TEST_CASE_TEMPLATE(DefaultsManagement, T, PibImpls, T)
268{
269 this->pib.addIdentity(this->id1);
270 BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id1);
271
272 this->pib.addIdentity(this->id2);
273 BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id1);
274
275 this->pib.removeIdentity(this->id1);
276 BOOST_CHECK_THROW(this->pib.getDefaultIdentity(), Pib::Error);
277
278 this->pib.addKey(this->id2, this->id2Key1Name, this->id2Key1.buf(), this->id2Key1.size());
279 BOOST_CHECK_EQUAL(this->pib.getDefaultIdentity(), this->id2);
280 BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id2), this->id2Key1Name);
281
282 this->pib.addKey(this->id2, this->id2Key2Name, this->id2Key2.buf(), this->id2Key2.size());
283 BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id2), this->id2Key1Name);
284
285 this->pib.removeKey(this->id2Key1Name);
286 BOOST_CHECK_THROW(this->pib.getDefaultKeyOfIdentity(this->id2), Pib::Error);
287
288 this->pib.addCertificate(this->id2Key2Cert1);
289 BOOST_CHECK_EQUAL(this->pib.getDefaultKeyOfIdentity(this->id2), this->id2Key2Name);
290 BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id2Key2Name).getName(), this->id2Key2Cert1.getName());
291
292 this->pib.addCertificate(this->id2Key2Cert2);
293 BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id2Key2Name).getName(), this->id2Key2Cert1.getName());
294
295 this->pib.removeCertificate(this->id2Key2Cert2.getName());
296 BOOST_CHECK_EQUAL(this->pib.getDefaultCertificateOfKey(this->id2Key2Name).getName(), this->id2Key2Cert1.getName());
297}
298
299BOOST_FIXTURE_TEST_CASE_TEMPLATE(Overwrite, T, PibImpls, T)
300{
301 // check id1Key1, should not exist
302 this->pib.removeIdentity(this->id1);
303 BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), false);
304
305 // add id1Key1
306 this->pib.addKey(this->id1, this->id1Key1Name, this->id1Key1.buf(), this->id1Key1.size());
307 BOOST_CHECK_EQUAL(this->pib.hasKey(this->id1Key1Name), true);
308 const Buffer& keyBits = this->pib.getKeyBits(this->id1Key1Name);
309 BOOST_CHECK(keyBits == this->id1Key1);
310
311 // check overwrite, add a key with the same name.
312 this->pib.addKey(this->id1, this->id1Key1Name, this->id1Key2.buf(), this->id1Key2.size());
313 const Buffer& keyBits2 = this->pib.getKeyBits(this->id1Key1Name);
314 BOOST_CHECK(keyBits2 == this->id1Key2);
315
316 // check id1Key1Cert1, should not exist
317 this->pib.removeIdentity(this->id1);
318 BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert1.getName()), false);
319
320 // add id1Key1Cert1
321 this->pib.addKey(this->id1, this->id1Key1Name, this->id1Key1.buf(), this->id1Key1.size());
322 this->pib.addCertificate(this->id1Key1Cert1);
323 BOOST_CHECK_EQUAL(this->pib.hasCertificate(this->id1Key1Cert1.getName()), true);
324
325 auto cert = this->pib.getCertificate(this->id1Key1Cert1.getName());
326 BOOST_CHECK(cert.wireEncode() == this->id1Key1Cert1.wireEncode());
327
328 // Create a fake cert with the same name
329 auto cert2 = this->id1Key2Cert1;
330 cert2.setName(this->id1Key1Cert1.getName());
331 cert2.setSignature(this->id1Key2Cert1.getSignature());
332 this->pib.addCertificate(cert2);
333
334 auto cert3 = this->pib.getCertificate(this->id1Key1Cert1.getName());
335 BOOST_CHECK(cert3.wireEncode() == cert2.wireEncode());
336
337 // both key and certificate are overwritten
338 Buffer keyBits3 = this->pib.getKeyBits(this->id1Key1Name);
339 BOOST_CHECK(keyBits3 == this->id1Key2);
340}
341
Davide Pesaventoeee3e822016-11-26 19:19:34 +0100342BOOST_AUTO_TEST_SUITE_END() // TestPibImpl
Yingdi Yu6ee2d362015-07-16 21:48:05 -0700343BOOST_AUTO_TEST_SUITE_END() // Pib
Davide Pesaventoeee3e822016-11-26 19:19:34 +0100344BOOST_AUTO_TEST_SUITE_END() // Security
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700345
346} // namespace tests
Yingdi Yu6ee2d362015-07-16 21:48:05 -0700347} // namespace pib
Yingdi Yu3bf91f52015-06-12 19:39:40 -0700348} // namespace security
349} // namespace ndn