X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=src%2Fmain.cpp;h=03d11737f596ce50d7aced80fce1c7441c7818ab;hb=fb173a9728e915fc7b3f99bc41f55034757105df;hp=ac75df1f1a7ab658ee3163590a67a779f13553cf;hpb=7975d02c6c1ee679a236087e86955c086f1a9a8e;p=crypto_lab3.git

diff --git a/src/main.cpp b/src/main.cpp
index ac75df1..03d1173 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -6,35 +6,34 @@
   * Author: Grégory Burri
   */
 
+#include <vector>
 #include <iostream>
 using namespace std;
 
 #include <gmpxx.h>
 
-#include "Rsa.h"
-#include "RsaCrt.h"
+#include "Tests.h"
 
 const uint KEY_SIZE_BITS = 1024;
 const uint RSA_PUBLIC_EXPONENT = 65537;
 
+void printUsage(const string& progName)
+{
+   cout << "Usage: " << progName << " [tests|time-measures]" << endl;
+}
+
 int main(int argc, char** argv)
 {
    vector<string> args;
    for (int i = 0; i < argc; i++)
       args.push_back(string(argv[i]));
 
-   const auto& keys = Rsa::generateRSAKeys(RSA_PUBLIC_EXPONENT, KEY_SIZE_BITS);
-   const auto& kPub = keys.first;
-   const auto& kPriv = keys.second;
-
-   mpz_class message(42);
-   mpz_class signature = Rsa::sign(message, kPriv);
-
-   mpz_class message2(42);
-   cout << "verify: " << Rsa::verifySignature(message2, signature, kPub) << endl;
-
-   mpz_class message3(43);
-   cout << "verify: " << Rsa::verifySignature(message3, signature, kPub) << endl;
+   if (args.size() >= 2 && args[1] == "tests")
+      Tests(KEY_SIZE_BITS, RSA_PUBLIC_EXPONENT).runTests();
+   else if (args.size() >= 2 && args[1] == "time-measures")
+      Tests(KEY_SIZE_BITS, RSA_PUBLIC_EXPONENT).runTimeMeasures();
+   else
+      printUsage(args[0]);
 
    return 0;
 }