blob: 5ae91b50ecd7882ff82ae8187a11f258307e5eac [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/*
Yingdi Yu6ee2d362015-07-16 21:48:05 -07003 * Copyright (c) 2013-2017 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
Alexander Afanasyev97709c02016-08-25 19:58:30 -070022#include "security/pib/pib.hpp"
23#include "security/pib/pib-memory.hpp"
Yingdi Yub8f8b342015-04-27 11:06:42 -070024
25#include "boost-test.hpp"
Davide Pesaventoeee3e822016-11-26 19:19:34 +010026#include "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);
46
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070047 BOOST_CHECK_EQUAL(static_cast<bool>(id), true);
Yingdi Yub8f8b342015-04-27 11:06:42 -070048 BOOST_CHECK_EQUAL(!id, false);
49
50 if (id)
51 BOOST_CHECK(true);
52 else
53 BOOST_CHECK(false);
54
55 // key
Davide Pesavento5d0b0102017-10-07 13:43:16 -040056 Key key = id.addKey(id1Key1.data(), id1Key1.size(), id1Key1Name);
Yingdi Yub8f8b342015-04-27 11:06:42 -070057
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070058 BOOST_CHECK_EQUAL(static_cast<bool>(key), true);
Yingdi Yub8f8b342015-04-27 11:06:42 -070059 BOOST_CHECK_EQUAL(!key, false);
60
61 if (key)
62 BOOST_CHECK(true);
63 else
64 BOOST_CHECK(false);
65}
66
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070067BOOST_AUTO_TEST_CASE(TpmLocator)
68{
69 Pib pib("pib-memory", "", make_shared<PibMemory>());
70
71 BOOST_CHECK_EQUAL(pib.getPibLocator(), "pib-memory:");
72 BOOST_CHECK_THROW(pib.getTpmLocator(), Pib::Error);
73
74 pib.setTpmLocator("test-tpm-locator");
75 BOOST_CHECK_NO_THROW(pib.getTpmLocator());
76
77 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
78 pib.addIdentity(id1);
79 BOOST_CHECK_NO_THROW(pib.getIdentity(id1));
80
81 pib.setTpmLocator("another-tpm-locator");
82 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
83
84 pib.addIdentity(id1);
85 BOOST_CHECK_NO_THROW(pib.getIdentity(id1));
86 pib.reset();
87 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
88 BOOST_CHECK_THROW(pib.getTpmLocator(), Pib::Error);
89}
90
Davide Pesaventoeee3e822016-11-26 19:19:34 +010091BOOST_AUTO_TEST_CASE(IdentityOperations)
Yingdi Yub8f8b342015-04-27 11:06:42 -070092{
Yingdi Yu7b3b5e92015-08-13 19:52:35 -070093 Pib pib("pib-memory", "", make_shared<PibMemory>());
Yingdi Yucbe72b02015-11-25 17:35:37 -080094 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 0);
Yingdi Yub8f8b342015-04-27 11:06:42 -070095
Yingdi Yucbe72b02015-11-25 17:35:37 -080096 // get non-existing identity, throw Pib::Error
Yingdi Yub8f8b342015-04-27 11:06:42 -070097 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
Yingdi Yucbe72b02015-11-25 17:35:37 -080098 // get default identity when it is not set yet, throw Pib::Error
99 BOOST_CHECK_THROW(pib.getDefaultIdentity(), Pib::Error);
100
101 // add identity
102 pib.addIdentity(id1);
Yingdi Yub8f8b342015-04-27 11:06:42 -0700103 BOOST_CHECK_NO_THROW(pib.getIdentity(id1));
Yingdi Yucbe72b02015-11-25 17:35:37 -0800104 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 1);
105
106 // new key becomes default key when there was no default key
107 BOOST_REQUIRE_NO_THROW(pib.getDefaultIdentity());
108 BOOST_CHECK_EQUAL(pib.getDefaultIdentity().getName(), id1);
109
110 // remove identity
Yingdi Yub8f8b342015-04-27 11:06:42 -0700111 pib.removeIdentity(id1);
112 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
Yingdi Yub8f8b342015-04-27 11:06:42 -0700113 BOOST_CHECK_THROW(pib.getDefaultIdentity(), Pib::Error);
Yingdi Yucbe72b02015-11-25 17:35:37 -0800114 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 0);
115
116 // set default identity
Yingdi Yub8f8b342015-04-27 11:06:42 -0700117 BOOST_REQUIRE_NO_THROW(pib.setDefaultIdentity(id1));
118 BOOST_REQUIRE_NO_THROW(pib.getDefaultIdentity());
119 BOOST_CHECK_EQUAL(pib.getDefaultIdentity().getName(), id1);
Yingdi Yucbe72b02015-11-25 17:35:37 -0800120 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 1);
121 BOOST_REQUIRE_NO_THROW(pib.setDefaultIdentity(id2));
122 BOOST_REQUIRE_NO_THROW(pib.getDefaultIdentity());
123 BOOST_CHECK_EQUAL(pib.getDefaultIdentity().getName(), id2);
124 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 2);
125
126 // remove default identity
127 pib.removeIdentity(id2);
128 BOOST_CHECK_THROW(pib.getIdentity(id2), Pib::Error);
129 BOOST_CHECK_THROW(pib.getDefaultIdentity(), Pib::Error);
130 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 1);
Yingdi Yub8f8b342015-04-27 11:06:42 -0700131 pib.removeIdentity(id1);
132 BOOST_CHECK_THROW(pib.getIdentity(id1), Pib::Error);
Yingdi Yucbe72b02015-11-25 17:35:37 -0800133 BOOST_CHECK_EQUAL(pib.getIdentities().size(), 0);
Yingdi Yub8f8b342015-04-27 11:06:42 -0700134}
135
Yingdi Yu7b3b5e92015-08-13 19:52:35 -0700136BOOST_AUTO_TEST_SUITE_END() // TestPib
Yingdi Yu6ee2d362015-07-16 21:48:05 -0700137BOOST_AUTO_TEST_SUITE_END() // Pib
Davide Pesaventoeee3e822016-11-26 19:19:34 +0100138BOOST_AUTO_TEST_SUITE_END() // Security
Yingdi Yub8f8b342015-04-27 11:06:42 -0700139
140} // namespace tests
Yingdi Yu6ee2d362015-07-16 21:48:05 -0700141} // namespace pib
Yingdi Yub8f8b342015-04-27 11:06:42 -0700142} // namespace security
143} // namespace ndn