diff options
Diffstat (limited to 'dependencies/uthash/tests/test93.c')
-rw-r--r-- | dependencies/uthash/tests/test93.c | 116 |
1 files changed, 116 insertions, 0 deletions
diff --git a/dependencies/uthash/tests/test93.c b/dependencies/uthash/tests/test93.c new file mode 100644 index 000000000..81103a1a7 --- /dev/null +++ b/dependencies/uthash/tests/test93.c @@ -0,0 +1,116 @@ +#include <stdio.h> +#include <setjmp.h> + +#define HASH_BLOOM 16 + +#include "uthash.h" + +#undef uthash_malloc +#undef uthash_fatal +#define uthash_malloc(sz) alt_malloc(sz) +#define uthash_fatal(s) alt_fatal(s) + +typedef struct example_user_t { + int id; + int cookie; + UT_hash_handle hh; +} example_user_t; + +static int malloc_cnt = 0; +static int malloc_failed; +static int is_fatal; +static jmp_buf j_buf; +static example_user_t * users; +static int user_id = 0; + +static void *alt_malloc(size_t sz) +{ + if (--malloc_cnt <= 0) { + malloc_failed = 1; + return 0; + } + malloc_failed = 0; + return malloc(sz); +} + +static void alt_fatal(char const * s) { + (void)s; + is_fatal = 1; + longjmp(j_buf, 1); +} + +static void init_users(int need_malloc_cnt) { + users = NULL; + example_user_t * user = (example_user_t*)malloc(sizeof(example_user_t)); + user->id = user_id; + is_fatal = 0; + malloc_cnt = need_malloc_cnt; + if (!setjmp(j_buf)) { + HASH_ADD_INT(users, id, user); + } else { + free(user); + } +} + +int main() +{ + example_user_t *user; + + init_users(3); /* bloom filter must fail */ + if (!is_fatal) { + printf("fatal not called after bloom failure\n"); + } + + init_users(2); /* bucket creation must fail */ + if (!is_fatal) { + printf("fatal not called after bucket creation failure\n"); + } + + init_users(1); /* table creation must fail */ + if (!is_fatal) { + printf("fatal not called after table creation failure\n"); + } + + init_users(4); /* hash must create OK */ + if (is_fatal) { + printf("fatal error when creating hash normally\n"); + /* bad idea to continue running */ + return 1; + } + + /* let's add users until expansion fails */ + users = NULL; + malloc_cnt = 4; + while (1) { + if (user_id++ == 1000) { + printf("there is no way 1000 iterations didn't require realloc\n"); + break; + } + user = (example_user_t*)malloc(sizeof(example_user_t)); + user->id = user_id; + if (!setjmp(j_buf)) { + HASH_ADD_INT(users, id, user); + } else { + free(user); + } + if (malloc_failed) { + + if (!is_fatal) { + printf("fatal not called after bucket not extended\n"); + } + if (user_id < 10) { + printf("there is no way your bucket size is 10\n"); + } + + /* we can't really do anything, the hash is not in consistent + * state, so assume this is a success. */ + break; + } + malloc_cnt = 0; + } + + HASH_CLEAR(hh, users); + + printf("End\n"); + return 0; +} |