blob: 7e27beec2df47f766258b5506b635189f44bd4f1 [file] [log] [blame]
Alexander Afanasyev82c359c2017-01-04 14:48:07 -08001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
Alexander Afanasyev22ee0892017-09-02 12:29:16 -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
Davide Pesaventob310efb2019-04-11 22:10:24 -040025#include "ndn-cxx/security/impl/openssl.hpp"
Davide Pesavento25d4f1c2020-04-29 23:31:04 -040026#include "ndn-cxx/util/io.hpp"
Davide Pesavento97ee8112020-08-11 21:38:34 -040027#include "ndn-cxx/util/scope.hpp"
Davide Pesaventob310efb2019-04-11 22:10:24 -040028
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080029namespace ndn {
30namespace ndnsec {
31
32int
33ndnsec_import(int argc, char** argv)
34{
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080035 namespace po = boost::program_options;
36
Davide Pesaventob310efb2019-04-11 22:10:24 -040037 std::string input;
38 std::string password;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080039
Davide Pesavento97ee8112020-08-11 21:38:34 -040040 auto guard = make_scope_exit([&password] {
Davide Pesaventob310efb2019-04-11 22:10:24 -040041 OPENSSL_cleanse(&password.front(), password.size());
Davide Pesavento97ee8112020-08-11 21:38:34 -040042 });
Davide Pesaventob310efb2019-04-11 22:10:24 -040043
44 po::options_description description(
45 "Usage: ndnsec import [-h] [-P PASSPHRASE] [-i] FILE\n"
46 "\n"
47 "Options");
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080048 description.add_options()
49 ("help,h", "produce help message")
Davide Pesaventob310efb2019-04-11 22:10:24 -040050 ("input,i", po::value<std::string>(&input)->default_value("-"),
51 "input file, '-' for stdin (the default)")
52 ("password,P", po::value<std::string>(&password),
53 "passphrase, will prompt if empty or not specified")
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080054 ;
55
56 po::positional_options_description p;
57 p.add("input", 1);
58
59 po::variables_map vm;
60 try {
61 po::store(po::command_line_parser(argc, argv).options(description).positional(p).run(), vm);
62 po::notify(vm);
63 }
64 catch (const std::exception& e) {
Davide Pesaventob310efb2019-04-11 22:10:24 -040065 std::cerr << "ERROR: " << e.what() << "\n\n"
66 << description << std::endl;
67 return 2;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080068 }
69
Davide Pesaventob310efb2019-04-11 22:10:24 -040070 if (vm.count("help") > 0) {
71 std::cout << description << std::endl;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080072 return 0;
73 }
74
Davide Pesaventof2cae612021-03-24 18:47:05 -040075 KeyChain keyChain;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080076
Davide Pesaventob310efb2019-04-11 22:10:24 -040077 shared_ptr<security::SafeBag> safeBag;
78 if (input == "-")
79 safeBag = io::load<security::SafeBag>(std::cin);
80 else
81 safeBag = io::load<security::SafeBag>(input);
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080082
Davide Pesaventob310efb2019-04-11 22:10:24 -040083 if (password.empty()) {
84 int count = 3;
85 while (!getPassword(password, "Passphrase for the private key: ", false)) {
86 count--;
87 if (count <= 0) {
88 std::cerr << "ERROR: invalid password" << std::endl;
89 return 1;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080090 }
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080091 }
Davide Pesaventob310efb2019-04-11 22:10:24 -040092 }
Alexander Afanasyev634a62b2018-06-15 16:55:26 -040093
Davide Pesaventob310efb2019-04-11 22:10:24 -040094 keyChain.importSafeBag(*safeBag, password.data(), password.size());
95
96 return 0;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080097}
98
99} // namespace ndnsec
100} // namespace ndn