oom.c 1.28 KB
Newer Older
1 2
#include "clar_libgit2.h"

3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
/* Override default allocators with ones that will fail predictably. */

static git_allocator std_alloc;
static git_allocator oom_alloc;

static void *oom_malloc(size_t n, const char *file, int line)
{
	/* Reject any allocation of more than 100 bytes */
	return (n > 100) ? NULL : std_alloc.gmalloc(n, file, line);
}

static void *oom_realloc(void *p, size_t n, const char *file, int line)
{
	/* Reject any allocation of more than 100 bytes */
	return (n > 100) ? NULL : std_alloc.grealloc(p, n, file, line);
}

20
void test_str_oom__initialize(void)
21 22 23 24 25 26 27
{
	git_stdalloc_init_allocator(&std_alloc);
	git_stdalloc_init_allocator(&oom_alloc);

	oom_alloc.gmalloc = oom_malloc;
	oom_alloc.grealloc = oom_realloc;

28
	cl_git_pass(git_allocator_setup(&oom_alloc));
29 30
}

31
void test_str_oom__cleanup(void)
32
{
33
	cl_git_pass(git_allocator_setup(NULL));
34 35
}

36
void test_str_oom__grow(void)
37
{
38
	git_str buf = GIT_STR_INIT;
39

40 41
	cl_git_pass(git_str_grow(&buf, 42));
	cl_assert(!git_str_oom(&buf));
42

43 44
	cl_assert(git_str_grow(&buf, 101) == -1);
	cl_assert(git_str_oom(&buf));
45

46
	git_str_dispose(&buf);
47
}
48

49
void test_str_oom__grow_by(void)
50
{
51
	git_str buf = GIT_STR_INIT;
52

53 54
	cl_git_pass(git_str_grow_by(&buf, 42));
	cl_assert(!git_str_oom(&buf));
55

56 57
	cl_assert(git_str_grow_by(&buf, 101) == -1);
	cl_assert(git_str_oom(&buf));
58
}