aboutsummaryrefslogtreecommitdiff
path: root/toolsrc/src/commands_hash.cpp
blob: 20e960d02f39b5224ffe80a6f7b989c735bfe824 (plain)
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
#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);
	}
}