adiatron/main.cpp at main · daffukk/adiatron · GitHub

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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#include <iostream>
#include "headers.h"

// General flags

Config parseArgs(int argc, char** argv) {
  Config cfg;

  if(argc < 2) exit(1);

  cfg.mode = argv[1];

  if(cfg.mode == "keygen") return cfg;
  if(argc < 3) {
    std::cerr << "File is not selected.\n";
    exit(1);
  }

  cfg.file = argv[2];

  int i=3;

  for(; i < argc; i++) {
    std::string arg = argv[i];

    if(arg == "--dir" || arg == "-d") {
      cfg.isDir = true;
    }

    else if((arg == "--filename" || arg == "-o") && i+1 < argc) {
      cfg.filename = argv[++i];
    }

    else if(arg == "--keydir" && i+1 < argc) {
      cfg.keysDir = argv[++i];
    }

    else if(arg == "--pkey" && i+1 < argc) {
      cfg.pubDir = argv[++i];
    }

    else if(arg == "--skey" && i+1 < argc) {
      cfg.secDir = argv[++i];
    }

    else if(arg.find("--filename=") == 0) {
      if(arg.substr(11).length() < 1) {
        std::cerr << "Invalid filename.\n";
        exit(1);
      } else {
        cfg.filename = arg.substr(11);
      }
    }

    else if(arg.find("-o=") == 0) {
      if(arg.substr(3).length() < 1) {
        std::cerr << "Invalid filename.\n";
        exit(1);
      } else {
        cfg.filename = arg.substr(3);
      }
    }

    else if(arg.find("--keydir=") == 0) {
      if(arg.substr(9).length() < 1) {
        std::cerr << "Invalid keydir.\n";
        exit(1);
      } else {
        cfg.keysDir = arg.substr(11);
      }
    }

    else if(arg.find("--pkey=") == 0) {
      if(arg.substr(7).length() < 1) {
        std::cerr << "Invalid public key.\n";
        exit(1);
      } else {
        cfg.pubDir = arg.substr(7);
      }
    }

    else if(arg.find("--skey=") == 0) {
      if(arg.substr(7).length() < 1) {
        std::cerr << "Invalid secret key.\n";
        exit(1);
      } else {
        cfg.secDir = arg.substr(7);
      }
    }

    else {
      std::cout << "Unknown argument: " << arg << "\n";
      exit(1);
    }
  }

  return cfg;
}


int main(int argc, char* argv[]) {

  if(argc < 2) {
    printHelp(argc, argv);
    return 1;
  }

  Config cfg = parseArgs(argc, argv);

  if(cfg.mode == "keygen") generateKeypair();

  if(argc < 3 && cfg.mode != "keygen") {
    printHelp(argc, argv);
    return 1;
  }


  if(cfg.mode == "encrypt") encrypt(cfg);
  else if(cfg.mode == "decrypt") decrypt(cfg);

  return 0;
}