Merge branch 'sdcb-hash'

1 files changed, 50 insertions, 0 deletions

diff --git a/toolsrc/src/commands_hash.cpp b/toolsrc/src/commands_hash.cpp
new file mode 100644
index 000000000..07ce9d6e7
--- /dev/null
+++ b/toolsrc/src/commands_hash.cpp
@@ -0,0 +1,50 @@
+#include "vcpkg_Commands.h"
+#include "vcpkg_System.h"
+#include "vcpkg.h"
+#include <iostream>
+#include <iomanip>
+#include <Windows.h>
+
+namespace fs = std::tr2::sys;
+
+namespace vcpkg
+{
+    void file_hash_sha512(fs::path const& path, std::wstring const& hashType)
+    {
+        auto cmd_line = Strings::wformat(LR"(CertUtil.exe -hashfile "%s" %s)",
+                                         path.c_str(), hashType.c_str());
+        auto ec_data = System::cmd_execute_and_capture_output(cmd_line);
+        Checks::check_exit(ec_data.exit_code == 0, "Running command:\n   %s\n failed", Strings::utf16_to_utf8(cmd_line));
+
+        std::string const& output = ec_data.output;
+
+        auto start = output.find_first_of("\r\n");
+        Checks::check_exit(start != std::string::npos, "Unexpected output format from command: %s", Strings::utf16_to_utf8(cmd_line));
+
+        auto end = output.find_first_of("\r\n", start + 1);
+        Checks::check_exit(end != std::string::npos, "Unexpected output format from command: %s", Strings::utf16_to_utf8(cmd_line));
+
+        auto hash = output.substr(start, end - start);
+        hash.erase(std::remove_if(hash.begin(), hash.end(), isspace), hash.end());
+        System::println(hash.c_str());
+    }
+
+    void hash_command(const vcpkg_cmd_arguments& args)
+    {
+        static const std::string example = Strings::format(
+            "The argument should be a file path\n%s", create_example_string("hash boost_1_62_0.tar.bz2"));
+        args.check_min_arg_count(1, example.c_str());
+        args.check_max_arg_count(2, example.c_str());
+
+        if (args.command_arguments.size() == 1)
+        {
+            file_hash_sha512(args.command_arguments[0], L"SHA512");
+        }
+        if (args.command_arguments.size() == 2)
+        {
+            file_hash_sha512(args.command_arguments[0], Strings::utf8_to_utf16(args.command_arguments[1]));
+        }
+
+        exit(EXIT_SUCCESS);
+    }
+}