Revision control

Line Code
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
/* 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;
}