midx_fuzzer.c 1.87 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
/*
 * libgit2 multi-pack-index fuzzer target.
 *
 * Copyright (C) the libgit2 contributors. All rights reserved.
 *
 * This file is part of libgit2, distributed under the GNU GPL v2 with
 * a Linking Exception. For full terms see the included COPYING file.
 */

#include <stdio.h>

#include "git2.h"

#include "common.h"
#include "futils.h"
#include "hash.h"
#include "midx.h"

19 20
#include "standalone_driver.h"

21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
int LLVMFuzzerInitialize(int *argc, char ***argv)
{
	GIT_UNUSED(argc);
	GIT_UNUSED(argv);

	if (git_libgit2_init() < 0) {
		fprintf(stderr, "Failed to initialize libgit2\n");
		abort();
	}
	return 0;
}

int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
{
	git_midx_file idx = {{0}};
	git_midx_entry e;
37
	git_str midx_buf = GIT_STR_INIT;
38
	unsigned char hash[GIT_HASH_SHA1_SIZE];
Edward Thomson committed
39
	git_oid oid = GIT_OID_NONE;
40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
	bool append_hash = false;

	if (size < 4)
		return 0;

	/*
	 * If the first byte in the stream has the high bit set, append the
	 * SHA1 hash so that the packfile is somewhat valid.
	 */
	append_hash = *data & 0x80;
	/* Keep a 4-byte alignment to avoid unaligned accesses. */
	data += 4;
	size -= 4;

	if (append_hash) {
55
		if (git_str_init(&midx_buf, size + GIT_HASH_SHA1_SIZE) < 0)
56
			goto cleanup;
57
		if (git_hash_buf(hash, data, size, GIT_HASH_ALGORITHM_SHA1) < 0) {
58 59 60 61
			fprintf(stderr, "Failed to compute the SHA1 hash\n");
			abort();
		}
		memcpy(midx_buf.ptr, data, size);
62 63
		memcpy(midx_buf.ptr + size, hash, GIT_HASH_SHA1_SIZE);

64
		memcpy(oid.id, hash, GIT_OID_SHA1_SIZE);
65
	} else {
66
		git_str_attach_notowned(&midx_buf, (char *)data, size);
67 68
	}

69
	if (git_midx_parse(&idx, (const unsigned char *)git_str_cstr(&midx_buf), git_str_len(&midx_buf)) < 0)
70 71 72
		goto cleanup;

	/* Search for any oid, just to exercise that codepath. */
73
	if (git_midx_entry_find(&e, &idx, &oid, GIT_OID_SHA1_HEXSIZE) < 0)
74 75 76 77
		goto cleanup;

cleanup:
	git_midx_close(&idx);
78
	git_str_dispose(&midx_buf);
79 80
	return 0;
}