зеркало из https://github.com/mozilla/gecko-dev.git
65 строки
1.6 KiB
C++
65 строки
1.6 KiB
C++
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include <algorithm>
|
|
#include <cstring>
|
|
#include <iostream>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <prinit.h>
|
|
|
|
#include "argparse.h"
|
|
#include "db/dbtool.h"
|
|
#include "enc/enctool.h"
|
|
#include "tool.h"
|
|
|
|
static void Usage() {
|
|
std::cerr << "Usage: nss <command> <subcommand> [options]" << std::endl;
|
|
std::cerr << " nss db [--path <directory>] <commands>" << std::endl;
|
|
std::cerr << " nss encrypt <options>" << std::endl;
|
|
std::cerr << " nss decrypt <options>" << std::endl;
|
|
}
|
|
|
|
static const std::string kDbCommand = "db";
|
|
static const std::string kEncryptCommand = "encrypt";
|
|
static const std::string kDecryptCommand = "decrypt";
|
|
|
|
int main(int argc, char **argv) {
|
|
if (argc < 2) {
|
|
Usage();
|
|
return 1;
|
|
}
|
|
std::vector<std::string> arguments(argv + 2, argv + argc);
|
|
|
|
std::unique_ptr<Tool> tool = nullptr;
|
|
if (argv[1] == kDbCommand) {
|
|
tool = std::unique_ptr<Tool>(new DBTool());
|
|
}
|
|
if (argv[1] == kEncryptCommand) {
|
|
tool = std::unique_ptr<Tool>(new EncTool());
|
|
arguments.push_back("--encrypt");
|
|
}
|
|
if (argv[1] == kDecryptCommand) {
|
|
tool = std::unique_ptr<Tool>(new EncTool());
|
|
arguments.push_back("--decrypt");
|
|
}
|
|
if (!tool) {
|
|
Usage();
|
|
return 1;
|
|
}
|
|
|
|
int exit_code = 0;
|
|
PR_Init(PR_SYSTEM_THREAD, PR_PRIORITY_NORMAL, 1);
|
|
|
|
if (!tool->Run(arguments)) {
|
|
exit_code = 1;
|
|
}
|
|
|
|
PR_Cleanup();
|
|
|
|
return exit_code;
|
|
}
|