blob: c25b0c7c4fe485cf32aa8eb8d8c30e396c4f7fde [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 Pesavento80d671f2022-06-08 04:04:52 -04003 * Copyright (c) 2013-2022 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 Pesavento97ee8112020-08-11 21:38:34 -040025#include "ndn-cxx/util/scope.hpp"
Davide Pesaventob310efb2019-04-11 22:10:24 -040026
Davide Pesavento80d671f2022-06-08 04:04:52 -040027#include <openssl/crypto.h>
28
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 Pesavento949075a2021-10-17 22:07:07 -040077 auto safeBag = loadFromFile<security::SafeBag>(input);
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080078
Davide Pesaventob310efb2019-04-11 22:10:24 -040079 if (password.empty()) {
80 int count = 3;
81 while (!getPassword(password, "Passphrase for the private key: ", false)) {
82 count--;
83 if (count <= 0) {
84 std::cerr << "ERROR: invalid password" << std::endl;
85 return 1;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080086 }
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080087 }
Davide Pesaventob310efb2019-04-11 22:10:24 -040088 }
Alexander Afanasyev634a62b2018-06-15 16:55:26 -040089
Davide Pesavento949075a2021-10-17 22:07:07 -040090 keyChain.importSafeBag(safeBag, password.data(), password.size());
Davide Pesaventob310efb2019-04-11 22:10:24 -040091
92 return 0;
Alexander Afanasyev82c359c2017-01-04 14:48:07 -080093}
94
95} // namespace ndnsec
96} // namespace ndn