alloc.c 1.29 KB
Newer Older
1 2 3 4 5 6 7 8
/*
 * 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 "alloc.h"
9
#include "runtime.h"
10

11
#include "allocators/failalloc.h"
12
#include "allocators/stdalloc.h"
13
#include "allocators/win32_leakcheck.h"
14

15 16 17 18 19 20 21 22 23 24 25 26
/* Fail any allocation until git_libgit2_init is called. */
git_allocator git__allocator = {
	git_failalloc_malloc,
	git_failalloc_calloc,
	git_failalloc_strdup,
	git_failalloc_strndup,
	git_failalloc_substrdup,
	git_failalloc_realloc,
	git_failalloc_reallocarray,
	git_failalloc_mallocarray,
	git_failalloc_free
};
27

28 29
static int setup_default_allocator(void)
{
30
#if defined(GIT_WIN32_LEAKCHECK)
31
	return git_win32_leakcheck_init_allocator(&git__allocator);
32 33 34 35 36
#else
	return git_stdalloc_init_allocator(&git__allocator);
#endif
}

37 38
int git_allocator_global_init(void)
{
39
	/*
40 41
	 * We don't want to overwrite any allocator which has been set
	 * before the init function is called.
42
	 */
43
	if (git__allocator.gmalloc != git_failalloc_malloc)
44 45
		return 0;

46
	return setup_default_allocator();
47 48 49 50
}

int git_allocator_setup(git_allocator *allocator)
{
51 52 53
	if (!allocator)
		return setup_default_allocator();

54 55 56
	memcpy(&git__allocator, allocator, sizeof(*allocator));
	return 0;
}