Ready for adding other hash algorithms.

magnetrwn · Nov 15, 2023 · b396f57 · b396f57
1 parent 7f7f54c
commit b396f57
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/include/hash.hpp b/include/hash.hpp
@@ -11,7 +11,7 @@
 
 class HashTools {
 public:
-    static std::string md5(const std::string &src);
+    static std::string digest(const std::string &algorithm, const std::string &src);
 };
 
 #endif
diff --git a/src/actions.cpp b/src/actions.cpp
@@ -48,7 +48,7 @@ ActionTransform::transform(const std::vector<std::string> &input, const Actions
             break;
         case MD5:
             output.reserve(1);
-            hexCiphertext = HashTools::md5(input[0]);
+            hexCiphertext = HashTools::digest("md5", input[0]);
             output.emplace_back(std::pair<std::string, std::string>("Hex Hash", hexCiphertext));
             break;
         default:

diff --git a/src/hash.cpp b/src/hash.cpp
@@ -1,16 +1,18 @@
 #include "hash.hpp"
 
-std::string HashTools::md5(const std::string &src) {
+std::string HashTools::digest(const std::string &algorithm, const std::string &src) {
     EVP_MD_CTX *mdctx;
     const EVP_MD *md;
     unsigned char digest[EVP_MAX_MD_SIZE];
     unsigned int mdLen;
 
     OpenSSL_add_all_digests();
-    md = EVP_get_digestbyname("md5");
+    md = EVP_get_digestbyname(algorithm.c_str());
 
-    if (!md)
-        throw std::runtime_error("Unable to initialize HashTools::md5 digest.");
+    if (!md) {
+        std::string msg = "Unable to find digest for algorithm " + algorithm + ".";
+        throw std::runtime_error(msg);
+    }
 
     mdctx = EVP_MD_CTX_new();
     EVP_DigestInit_ex(mdctx, md, NULL);