blob: b6e9f12376610a4afb3a9c4b2692054e3fecb165 [file] [log] [blame]
Yingdi Yub8f8b342015-04-27 11:06:42 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
Davide Pesavento74daf742018-11-23 18:14:13 -05002/*
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/certificate-container.hpp"
Davide Pesavento4fb35d82019-10-31 19:33:10 -040023#include "ndn-cxx/security/pib/impl/pib-memory.hpp"
Davide Pesavento7e780642018-11-24 15:51:34 -050024#include "ndn-cxx/security/pib/pib.hpp"
Yingdi Yub8f8b342015-04-27 11:06:42 -070025
Davide Pesavento7e780642018-11-24 15:51:34 -050026#include "tests/boost-test.hpp"
27#include "tests/unit/security/pib/pib-data-fixture.hpp"
Yingdi Yub8f8b342015-04-27 11:06:42 -070028
29namespace ndn {
30namespace security {
Yingdi Yu6ee2d362015-07-16 21:48:05 -070031namespace pib {
Yingdi Yub8f8b342015-04-27 11:06:42 -070032namespace tests {
33
Yingdi Yu6ee2d362015-07-16 21:48:05 -070034using namespace ndn::security::tests;
35
Davide Pesaventoeee3e822016-11-26 19:19:34 +010036BOOST_AUTO_TEST_SUITE(Security)
Yingdi Yu6ee2d362015-07-16 21:48:05 -070037BOOST_AUTO_TEST_SUITE(Pib)
Yingdi Yucbe72b02015-11-25 17:35:37 -080038BOOST_FIXTURE_TEST_SUITE(TestCertificateContainer, PibDataFixture)
Yingdi Yub8f8b342015-04-27 11:06:42 -070039
Yingdi Yu6ee2d362015-07-16 21:48:05 -070040using pib::Pib;
41
Yingdi Yucbe72b02015-11-25 17:35:37 -080042BOOST_AUTO_TEST_CASE(Basic)
Yingdi Yub8f8b342015-04-27 11:06:42 -070043{
Yingdi Yu3bf91f52015-06-12 19:39:40 -070044 auto pibImpl = make_shared<PibMemory>();
Yingdi Yub8f8b342015-04-27 11:06:42 -070045
Yingdi Yucbe72b02015-11-25 17:35:37 -080046 // start with an empty container
47 CertificateContainer container(id1Key1Name, pibImpl);
48 BOOST_CHECK_EQUAL(container.size(), 0);
49 BOOST_CHECK_EQUAL(container.getCache().size(), 0);
Yingdi Yub8f8b342015-04-27 11:06:42 -070050
Yingdi Yucbe72b02015-11-25 17:35:37 -080051 // add one cert
52 container.add(id1Key1Cert1);
53 BOOST_CHECK_EQUAL(container.size(), 1);
54 BOOST_CHECK_EQUAL(container.getCache().size(), 1);
55 BOOST_CHECK(container.find(id1Key1Cert1.getName()) != container.end());
56
57 // add the same cert again
58 container.add(id1Key1Cert1);
59 BOOST_CHECK_EQUAL(container.size(), 1);
60 BOOST_CHECK_EQUAL(container.getCache().size(), 1);
61 BOOST_CHECK(container.find(id1Key1Cert1.getName()) != container.end());
62
63 // add another cert
64 container.add(id1Key1Cert2);
Yingdi Yub8f8b342015-04-27 11:06:42 -070065 BOOST_CHECK_EQUAL(container.size(), 2);
Yingdi Yucbe72b02015-11-25 17:35:37 -080066 BOOST_CHECK_EQUAL(container.getCache().size(), 2);
Yingdi Yub8f8b342015-04-27 11:06:42 -070067 BOOST_CHECK(container.find(id1Key1Cert1.getName()) != container.end());
68 BOOST_CHECK(container.find(id1Key1Cert2.getName()) != container.end());
69
Yingdi Yucbe72b02015-11-25 17:35:37 -080070 // get certs
71 BOOST_REQUIRE_NO_THROW(container.get(id1Key1Cert1.getName()));
72 BOOST_REQUIRE_NO_THROW(container.get(id1Key1Cert2.getName()));
73 Name id1Key1Cert3Name = id1Key1Name;
74 id1Key1Cert3Name.append("issuer").appendVersion(3);
75 BOOST_CHECK_THROW(container.get(id1Key1Cert3Name), Pib::Error);
76
77 // check cert
78 v2::Certificate cert1 = container.get(id1Key1Cert1.getName());
79 v2::Certificate cert2 = container.get(id1Key1Cert2.getName());
80 BOOST_CHECK_EQUAL(cert1, id1Key1Cert1);
81 BOOST_CHECK_EQUAL(cert2, id1Key1Cert2);
82
83 // create another container from the same PibImpl
84 // cache should be empty
85 CertificateContainer container2(id1Key1Name, pibImpl);
86 BOOST_CHECK_EQUAL(container2.size(), 2);
87 BOOST_CHECK_EQUAL(container2.getCache().size(), 0);
88
89 // get certificate, cache should be filled
90 BOOST_REQUIRE_NO_THROW(container2.get(id1Key1Cert1.getName()));
91 BOOST_CHECK_EQUAL(container2.size(), 2);
92 BOOST_CHECK_EQUAL(container2.getCache().size(), 1);
93
94 BOOST_REQUIRE_NO_THROW(container2.get(id1Key1Cert2.getName()));
95 BOOST_CHECK_EQUAL(container2.size(), 2);
96 BOOST_CHECK_EQUAL(container2.getCache().size(), 2);
97
98 // remove a certificate
99 container2.remove(id1Key1Cert1.getName());
100 BOOST_CHECK_EQUAL(container2.size(), 1);
101 BOOST_CHECK_EQUAL(container2.getCache().size(), 1);
102 BOOST_CHECK(container2.find(id1Key1Cert1.getName()) == container2.end());
103 BOOST_CHECK(container2.find(id1Key1Cert2.getName()) != container2.end());
104
105 // remove another certificate
106 container2.remove(id1Key1Cert2.getName());
107 BOOST_CHECK_EQUAL(container2.size(), 0);
108 BOOST_CHECK_EQUAL(container2.getCache().size(), 0);
109 BOOST_CHECK(container2.find(id1Key1Cert2.getName()) == container2.end());
110}
111
112BOOST_AUTO_TEST_CASE(Errors)
113{
114 auto pibImpl = make_shared<PibMemory>();
115
116 CertificateContainer container(id1Key1Name, pibImpl);
117
118 BOOST_CHECK_THROW(container.add(id1Key2Cert1), std::invalid_argument);
119 BOOST_CHECK_THROW(container.remove(id1Key2Cert1.getName()), std::invalid_argument);
120 BOOST_CHECK_THROW(container.get(id1Key2Cert1.getName()), std::invalid_argument);
121}
122
123BOOST_AUTO_TEST_CASE(Iterator)
124{
125 auto pibImpl = make_shared<PibMemory>();
126
127 // start with an empty container
128 CertificateContainer container(id1Key1Name, pibImpl);
129 container.add(id1Key1Cert1);
130 container.add(id1Key1Cert2);
131
Yingdi Yub8f8b342015-04-27 11:06:42 -0700132 std::set<Name> certNames;
133 certNames.insert(id1Key1Cert1.getName());
134 certNames.insert(id1Key1Cert2.getName());
135
136 CertificateContainer::const_iterator it = container.begin();
137 std::set<Name>::const_iterator testIt = certNames.begin();
138 BOOST_CHECK_EQUAL((*it).getName(), *testIt);
139 it++;
140 testIt++;
141 BOOST_CHECK_EQUAL((*it).getName(), *testIt);
142 ++it;
143 testIt++;
144 BOOST_CHECK(it == container.end());
145
146 size_t count = 0;
147 testIt = certNames.begin();
148 for (const auto& cert : container) {
149 BOOST_CHECK_EQUAL(cert.getName(), *testIt);
150 testIt++;
151 count++;
152 }
153 BOOST_CHECK_EQUAL(count, 2);
Yingdi Yucbe72b02015-11-25 17:35:37 -0800154
155 BOOST_CHECK(CertificateContainer::const_iterator() == CertificateContainer::const_iterator());
156 BOOST_CHECK(CertificateContainer::const_iterator() == container.end());
157 BOOST_CHECK(container.end() == CertificateContainer::const_iterator());
Yingdi Yub8f8b342015-04-27 11:06:42 -0700158}
159
Davide Pesaventoeee3e822016-11-26 19:19:34 +0100160BOOST_AUTO_TEST_SUITE_END() // TestCertificateContainer
Yingdi Yu6ee2d362015-07-16 21:48:05 -0700161BOOST_AUTO_TEST_SUITE_END() // Pib
Davide Pesaventoeee3e822016-11-26 19:19:34 +0100162BOOST_AUTO_TEST_SUITE_END() // Security
Yingdi Yub8f8b342015-04-27 11:06:42 -0700163
164} // namespace tests
Yingdi Yu6ee2d362015-07-16 21:48:05 -0700165} // namespace pib
Yingdi Yub8f8b342015-04-27 11:06:42 -0700166} // namespace security
167} // namespace ndn