blob: 128ae190dc4b7323ba5db0e9dac531d200381328 [file] [log] [blame]
Alexander Afanasyev82c359c2017-01-04 14:48:07 -08001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
Davide Pesaventob310efb2019-04-11 22:10:24 -04002/*
Davide Pesaventof2cae612021-03-24 18:47:05 -04003 * Copyright (c) 2013-2021 Regents of the University of California.
Alexander Afanasyev82c359c2017-01-04 14:48:07 -08004 *
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
22#include "ndnsec.hpp"
23#include "util.hpp"
24
25namespace ndn {
26namespace ndnsec {
27
28int
29ndnsec_set_default(int argc, char** argv)
30{
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080031 namespace po = boost::program_options;
32
Alexander Afanasyev35109a12017-01-04 15:39:06 -080033 Name name;
Davide Pesaventob310efb2019-04-11 22:10:24 -040034 bool wantSetDefaultKey = false;
35 bool wantSetDefaultCert = false;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080036
Davide Pesaventob310efb2019-04-11 22:10:24 -040037 po::options_description description(
38 "Usage: ndnsec set-default [-h] [-k|-c] [-n] NAME\n"
39 "\n"
40 "Options");
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080041 description.add_options()
42 ("help,h", "produce help message")
Davide Pesaventob310efb2019-04-11 22:10:24 -040043 ("name,n", po::value<Name>(&name), "the identity/key/certificate name to set")
44 ("default-key,k", po::bool_switch(&wantSetDefaultKey), "set default key of the identity")
45 ("default-cert,c", po::bool_switch(&wantSetDefaultCert), "set default certificate of the key")
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080046 ;
47
48 po::positional_options_description p;
49 p.add("name", 1);
Davide Pesaventob310efb2019-04-11 22:10:24 -040050
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080051 po::variables_map vm;
52 try {
53 po::store(po::command_line_parser(argc, argv).options(description).positional(p).run(), vm);
54 po::notify(vm);
55 }
56 catch (const std::exception& e) {
Davide Pesaventob310efb2019-04-11 22:10:24 -040057 std::cerr << "ERROR: " << e.what() << "\n\n"
58 << description << std::endl;
59 return 2;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080060 }
61
Davide Pesaventob310efb2019-04-11 22:10:24 -040062 if (vm.count("help") > 0) {
63 std::cout << description << std::endl;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080064 return 0;
65 }
66
67 if (vm.count("name") == 0) {
Davide Pesaventob310efb2019-04-11 22:10:24 -040068 std::cerr << "ERROR: you must specify a name" << std::endl;
69 return 2;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080070 }
71
Davide Pesaventob310efb2019-04-11 22:10:24 -040072 if (wantSetDefaultKey && wantSetDefaultCert) {
73 std::cerr << "ERROR: cannot specify both '--default-key' and '--default-cert'" << std::endl;
74 return 2;
75 }
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080076
Davide Pesaventof2cae612021-03-24 18:47:05 -040077 KeyChain keyChain;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080078
Davide Pesaventob310efb2019-04-11 22:10:24 -040079 if (wantSetDefaultKey) {
Davide Pesaventof2cae612021-03-24 18:47:05 -040080 auto identity = keyChain.getPib().getIdentity(security::extractIdentityFromKeyName(name));
Davide Pesaventob310efb2019-04-11 22:10:24 -040081 auto key = identity.getKey(name);
Alexander Afanasyev35109a12017-01-04 15:39:06 -080082 keyChain.setDefaultKey(identity, key);
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080083 }
Davide Pesaventob310efb2019-04-11 22:10:24 -040084 else if (wantSetDefaultCert) {
Davide Pesaventof2cae612021-03-24 18:47:05 -040085 auto identity = keyChain.getPib().getIdentity(security::extractIdentityFromCertName(name));
86 auto key = identity.getKey(security::extractKeyNameFromCertName(name));
Davide Pesaventob310efb2019-04-11 22:10:24 -040087 auto cert = key.getCertificate(name);
Alexander Afanasyev35109a12017-01-04 15:39:06 -080088 keyChain.setDefaultCertificate(key, cert);
Davide Pesaventob310efb2019-04-11 22:10:24 -040089 }
90 else {
91 auto identity = keyChain.getPib().getIdentity(name);
92 keyChain.setDefaultIdentity(identity);
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080093 }
94
Davide Pesaventob310efb2019-04-11 22:10:24 -040095 return 0;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080096}
97
98} // namespace ndnsec
99} // namespace ndn