Skip to content
Snippets Groups Projects
Commit 3ce4f61c authored by Sven M. Hallberg's avatar Sven M. Hallberg
Browse files

add test for out-of-memory handling

parent 2309bd6d
No related branches found
No related tags found
No related merge requests found
#include <glib.h> #include <glib.h>
#include <string.h> #include <string.h>
#include <sys/resource.h>
#include "test_suite.h" #include "test_suite.h"
#include "hammer.h" #include "hammer.h"
...@@ -29,7 +30,43 @@ static void test_tt_registry(void) { ...@@ -29,7 +30,43 @@ static void test_tt_registry(void) {
g_check_cmp_int32(h_get_token_type_number("com.upstandinghackers.test.unkown_token_type"), ==, 0); g_check_cmp_int32(h_get_token_type_number("com.upstandinghackers.test.unkown_token_type"), ==, 0);
} }
// perform a big allocation during parsing to trigger out-of-memory handling
static HParsedToken *act_big_alloc(const HParseResult *r, void *user) {
void *buf = h_arena_malloc(r->arena, 1024*1024*1024);
assert(buf != NULL);
return NULL;
}
static void test_oom(void) {
HParser *p = h_action(h_ch('x'), act_big_alloc, NULL);
// this should always fail, but never crash
struct rlimit bak, lim;
int i;
i = getrlimit(RLIMIT_DATA, &bak);
assert(i == 0);
lim.rlim_cur = 1000*1024*1024; // never enough
if(lim.rlim_cur > bak.rlim_max)
lim.rlim_cur = bak.rlim_max;
lim.rlim_max = bak.rlim_max;
i = setrlimit(RLIMIT_DATA, &lim);
assert(i == 0);
g_check_parse_failed(p, PB_PACKRAT, "x",1);
g_check_parse_failed(p, PB_REGULAR, "x",1);
g_check_parse_failed(p, PB_LLk, "x",1);
g_check_parse_failed(p, PB_LALR, "x",1);
g_check_parse_failed(p, PB_GLR, "x",1);
g_check_parse_chunks_failed(p, PB_LLk, "",0, "x",1);
//g_check_parse_chunks_failed(p, PB_LALR, "",0, "x",1);
//g_check_parse_chunks_failed(p, PB_GLR, "",0, "x",1);
i = setrlimit(RLIMIT_DATA, &bak);
assert(i == 0);
}
void register_misc_tests(void) { void register_misc_tests(void) {
g_test_add_func("/core/misc/tt_user", test_tt_user); g_test_add_func("/core/misc/tt_user", test_tt_user);
g_test_add_func("/core/misc/tt_registry", test_tt_registry); g_test_add_func("/core/misc/tt_registry", test_tt_registry);
g_test_add_func("/core/misc/oom", test_oom);
} }
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment