sha1.c 2.59 KB
Newer Older
1 2 3 4 5
#include "clar_libgit2.h"
#include "hash.h"

#define FIXTURE_DIR "sha1"

6 7 8 9
#ifdef GIT_SHA1_WIN32
static git_hash_win32_provider_t orig_provider;
#endif

10
void test_sha1__initialize(void)
11
{
12 13 14 15
#ifdef GIT_SHA1_WIN32
	orig_provider = git_hash_win32_provider();
#endif

16 17 18
	cl_fixture_sandbox(FIXTURE_DIR);
}

19
void test_sha1__cleanup(void)
20
{
21 22 23 24
#ifdef GIT_SHA1_WIN32
	git_hash_win32_set_provider(orig_provider);
#endif

25 26 27
	cl_fixture_cleanup(FIXTURE_DIR);
}

28
static int sha1_file(unsigned char *out, const char *filename)
29 30 31 32 33 34 35 36 37
{
	git_hash_ctx ctx;
	char buf[2048];
	int fd, ret;
	ssize_t read_len;

	fd = p_open(filename, O_RDONLY);
	cl_assert(fd >= 0);

38
	cl_git_pass(git_hash_ctx_init(&ctx, GIT_HASH_ALGORITHM_SHA1));
39 40 41 42 43 44 45

	while ((read_len = p_read(fd, buf, 2048)) > 0)
		cl_git_pass(git_hash_update(&ctx, buf, (size_t)read_len));

	cl_assert_equal_i(0, read_len);
	p_close(fd);

46
	ret = git_hash_final(out, &ctx);
47 48 49 50 51
	git_hash_ctx_cleanup(&ctx);

	return ret;
}

52
void test_sha1__sum(void)
53
{
54 55 56 57 58
	unsigned char expected[GIT_HASH_SHA1_SIZE] = {
		0x4e, 0x72, 0x67, 0x9e, 0x3e, 0xa4, 0xd0, 0x4e, 0x0c, 0x64,
		0x2f, 0x02, 0x9e, 0x61, 0xeb, 0x80, 0x56, 0xc7, 0xed, 0x94
	};
	unsigned char actual[GIT_HASH_SHA1_SIZE];
59

60 61
	cl_git_pass(sha1_file(actual, FIXTURE_DIR "/hello_c"));
	cl_assert_equal_i(0, memcmp(expected, actual, GIT_HASH_SHA1_SIZE));
62 63 64
}

/* test that sha1 collision detection works when enabled */
65
void test_sha1__detect_collision_attack(void)
66
{
67 68 69 70 71
	unsigned char actual[GIT_HASH_SHA1_SIZE];
	unsigned char expected[GIT_HASH_SHA1_SIZE] = {
		0x38, 0x76, 0x2c, 0xf7, 0xf5, 0x59, 0x34, 0xb3, 0x4d, 0x17,
		0x9a, 0xe6, 0xa4, 0xc8, 0x0c, 0xad, 0xcc, 0xbb, 0x7f, 0x0a
	};
72 73

#ifdef GIT_SHA1_COLLISIONDETECT
74
	GIT_UNUSED(&expected);
75
	cl_git_fail(sha1_file(actual, FIXTURE_DIR "/shattered-1.pdf"));
76
	cl_assert_equal_s("SHA1 collision attack detected", git_error_last()->message);
77
#else
78 79
	cl_git_pass(sha1_file(actual, FIXTURE_DIR "/shattered-1.pdf"));
	cl_assert_equal_i(0, memcmp(expected, actual, GIT_HASH_SHA1_SIZE));
80 81 82
#endif
}

83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
void test_sha1__win32_providers(void)
{
#ifdef GIT_SHA1_WIN32
	unsigned char expected[GIT_HASH_SHA1_SIZE] = {
		0x38, 0x76, 0x2c, 0xf7, 0xf5, 0x59, 0x34, 0xb3, 0x4d, 0x17,
		0x9a, 0xe6, 0xa4, 0xc8, 0x0c, 0xad, 0xcc, 0xbb, 0x7f, 0x0a
	};
	unsigned char actual[GIT_HASH_SHA1_SIZE];

	git_hash_win32_set_provider(GIT_HASH_WIN32_CRYPTOAPI);
	cl_git_pass(sha1_file(actual, FIXTURE_DIR "/shattered-1.pdf"));
	cl_assert_equal_i(0, memcmp(expected, actual, GIT_HASH_SHA1_SIZE));

	git_hash_win32_set_provider(GIT_HASH_WIN32_CNG);
	cl_git_pass(sha1_file(actual, FIXTURE_DIR "/shattered-1.pdf"));
	cl_assert_equal_i(0, memcmp(expected, actual, GIT_HASH_SHA1_SIZE));
#endif
}