blob: c40cddcdd1daebc185c657aaec98535a7952577c [file] [log] [blame]
Yingdi Yub8f8b342015-04-27 11:06:42 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
Davide Pesavento5d0b0102017-10-07 13:43:16 -04002/*
Davide Pesavento4fb35d82019-10-31 19:33:10 -04003 * Copyright (c) 2013-2019 Regents of the University of California.
Yingdi Yub8f8b342015-04-27 11:06:42 -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
Davide Pesavento7e780642018-11-24 15:51:34 -050022#include "ndn-cxx/security/pib/pib.hpp"
Davide Pesavento4fb35d82019-10-31 19:33:10 -040023#include "ndn-cxx/security/pib/impl/pib-memory.hpp"
Yingdi Yub8f8b342015-04-27 11:06:42 -070024
Davide Pesavento7e780642018-11-24 15:51:34 -050025#include "tests/boost-test.hpp"
26#include "tests/unit/security/pib/pib-data-fixture.hpp"
Yingdi Yub8f8b342015-04-27 11:06:42 -070027
28namespace ndn {
29namespace security {
Yingdi Yu6ee2d362015-07-16 21:48:05 -070030namespace pib {
Yingdi Yub8f8b342015-04-27 11:06:42 -070031namespace tests {
32
Yingdi Yu6ee2d362015-07-16 21:48:05 -070033using namespace ndn::security::tests;
34
Davide Pesaventoeee3e822016-11-26 19:19:34 +010035BOOST_AUTO_TEST_SUITE(Security)
Yingdi Yu6ee2d362015-07-16 21:48:05 -070036BOOST_AUTO_TEST_SUITE(Pib)
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070037BOOST_FIXTURE_TEST_SUITE(TestPib, PibDataFixture)
Yingdi Yub8f8b342015-04-27 11:06:42 -070038
Yingdi Yu6ee2d362015-07-16 21:48:05 -070039using pib::Pib;
40
Davide Pesaventoeee3e822016-11-26 19:19:34 +010041BOOST_AUTO_TEST_CASE(ValidityChecking)
Yingdi Yub8f8b342015-04-27 11:06:42 -070042{
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070043 Pib pib("pib-memory", "", make_shared<PibMemory>());
Yingdi Yub8f8b342015-04-27 11:06:42 -070044
45 Identity id = pib.addIdentity(id1);
Davide Pesavento4fb35d82019-10-31 19:33:10 -040046 BOOST_CHECK(id);
Yingdi Yub8f8b342015-04-27 11:06:42 -070047 BOOST_CHECK_EQUAL(!id, false);
48
Davide Pesavento5d0b0102017-10-07 13:43:16 -040049 Key key = id.addKey(id1Key1.data(), id1Key1.size(), id1Key1Name);
Davide Pesavento4fb35d82019-10-31 19:33:10 -040050 BOOST_CHECK(key);
Yingdi Yub8f8b342015-04-27 11:06:42 -070051 BOOST_CHECK_EQUAL(!key, false);
Yingdi Yub8f8b342015-04-27 11:06:42 -070052}
53
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070054BOOST_AUTO_TEST_CASE(TpmLocator)
55{
56 Pib pib("pib-memory", "", make_shared<PibMemory>());
57
58 BOOST_CHECK_EQUAL(pib.getPibLocator(), "pib-memory:");
59 BOOST_CHECK_THROW(pib.getTpmLocator(), Pib::Error);
60
61 pib.setTpmLocator("test-tpm-locator");
62 BOOST_CHECK_NO_THROW(pib.getTpmLocator());
63
64 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
65 pib.addIdentity(id1);
66 BOOST_CHECK_NO_THROW(pib.getIdentity(id1));
67
68 pib.setTpmLocator("another-tpm-locator");
69 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
70
71 pib.addIdentity(id1);
72 BOOST_CHECK_NO_THROW(pib.getIdentity(id1));
73 pib.reset();
74 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
75 BOOST_CHECK_THROW(pib.getTpmLocator(), Pib::Error);
76}
77
Davide Pesaventoeee3e822016-11-26 19:19:34 +010078BOOST_AUTO_TEST_CASE(IdentityOperations)
Yingdi Yub8f8b342015-04-27 11:06:42 -070079{
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070080 Pib pib("pib-memory", "", make_shared<PibMemory>());
Yingdi Yucbe72b02015-11-25 17:35:37 -080081 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 0);
Yingdi Yub8f8b342015-04-27 11:06:42 -070082
Yingdi Yucbe72b02015-11-25 17:35:37 -080083 // get non-existing identity, throw Pib::Error
Yingdi Yub8f8b342015-04-27 11:06:42 -070084 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
Yingdi Yucbe72b02015-11-25 17:35:37 -080085 // get default identity when it is not set yet, throw Pib::Error
86 BOOST_CHECK_THROW(pib.getDefaultIdentity(), Pib::Error);
87
88 // add identity
89 pib.addIdentity(id1);
Yingdi Yub8f8b342015-04-27 11:06:42 -070090 BOOST_CHECK_NO_THROW(pib.getIdentity(id1));
Yingdi Yucbe72b02015-11-25 17:35:37 -080091 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 1);
92
93 // new key becomes default key when there was no default key
94 BOOST_REQUIRE_NO_THROW(pib.getDefaultIdentity());
95 BOOST_CHECK_EQUAL(pib.getDefaultIdentity().getName(), id1);
96
97 // remove identity
Yingdi Yub8f8b342015-04-27 11:06:42 -070098 pib.removeIdentity(id1);
99 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
Yingdi Yub8f8b342015-04-27 11:06:42 -0700100 BOOST_CHECK_THROW(pib.getDefaultIdentity(), Pib::Error);
Yingdi Yucbe72b02015-11-25 17:35:37 -0800101 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 0);
102
103 // set default identity
Yingdi Yub8f8b342015-04-27 11:06:42 -0700104 BOOST_REQUIRE_NO_THROW(pib.setDefaultIdentity(id1));
105 BOOST_REQUIRE_NO_THROW(pib.getDefaultIdentity());
106 BOOST_CHECK_EQUAL(pib.getDefaultIdentity().getName(), id1);
Yingdi Yucbe72b02015-11-25 17:35:37 -0800107 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 1);
108 BOOST_REQUIRE_NO_THROW(pib.setDefaultIdentity(id2));
109 BOOST_REQUIRE_NO_THROW(pib.getDefaultIdentity());
110 BOOST_CHECK_EQUAL(pib.getDefaultIdentity().getName(), id2);
111 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 2);
112
113 // remove default identity
114 pib.removeIdentity(id2);
115 BOOST_CHECK_THROW(pib.getIdentity(id2), Pib::Error);
116 BOOST_CHECK_THROW(pib.getDefaultIdentity(), Pib::Error);
117 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 1);
Yingdi Yub8f8b342015-04-27 11:06:42 -0700118 pib.removeIdentity(id1);
119 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
Yingdi Yucbe72b02015-11-25 17:35:37 -0800120 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 0);
Yingdi Yub8f8b342015-04-27 11:06:42 -0700121}
122
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700123BOOST_AUTO_TEST_SUITE_END() // TestPib
Yingdi Yu6ee2d362015-07-16 21:48:05 -0700124BOOST_AUTO_TEST_SUITE_END() // Pib
Davide Pesaventoeee3e822016-11-26 19:19:34 +0100125BOOST_AUTO_TEST_SUITE_END() // Security
Yingdi Yub8f8b342015-04-27 11:06:42 -0700126
127} // namespace tests
Yingdi Yu6ee2d362015-07-16 21:48:05 -0700128} // namespace pib
Yingdi Yub8f8b342015-04-27 11:06:42 -0700129} // namespace security
130} // namespace ndn