blob: 1a9b43bb3826b330fe161e428d4ace3841155e87 [file] [log] [blame]
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
* Copyright (c) 2014-2024, Regents of the University of California,
* Arizona Board of Regents,
* Colorado State University,
* University Pierre & Marie Curie, Sorbonne University,
* Washington University in St. Louis,
* Beijing Institute of Technology,
* The University of Memphis.
*
* This file is part of NFD (Named Data Networking Forwarding Daemon).
* See AUTHORS.md for complete list of NFD authors and contributors.
*
* NFD is free software: you can redistribute it and/or modify it under the terms
* of the GNU General Public License as published by the Free Software Foundation,
* either version 3 of the License, or (at your option) any later version.
*
* NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
* PURPOSE. See the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along with
* NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
*/
#include "tests/key-chain-fixture.hpp"
#include <ndn-cxx/util/io.hpp>
#include <filesystem>
#include <system_error>
namespace nfd::tests {
using namespace ndn::security;
KeyChainFixture::KeyChainFixture()
: m_keyChain("pib-memory:", "tpm-memory:")
{
}
KeyChainFixture::~KeyChainFixture()
{
std::error_code ec;
for (const auto& certFile : m_certFiles) {
std::filesystem::remove(certFile, ec); // ignore error
}
}
bool
KeyChainFixture::saveCert(const Data& cert, const std::string& filename)
{
m_certFiles.push_back(filename);
try {
ndn::io::save(cert, filename);
return true;
}
catch (const ndn::io::Error&) {
return false;
}
}
bool
KeyChainFixture::saveIdentityCert(const Identity& identity, const std::string& filename)
{
Certificate cert;
try {
cert = identity.getDefaultKey().getDefaultCertificate();
}
catch (const Pib::Error&) {
return false;
}
return saveCert(cert, filename);
}
bool
KeyChainFixture::saveIdentityCert(const Name& identityName, const std::string& filename,
bool allowCreate)
{
Identity id;
try {
id = m_keyChain.getPib().getIdentity(identityName);
}
catch (const Pib::Error&) {
if (allowCreate) {
id = m_keyChain.createIdentity(identityName);
}
}
if (!id) {
return false;
}
return saveIdentityCert(id, filename);
}
} // namespace nfd::tests