thomas/mittos64
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Handle null pointers in string comparisons

Browse Source
This commit is contained in:
Thomas Lovén 2017-12-04 22:04:41 +01:00
parent cfa1a15076
commit ad74123882
1 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/kernel/include/ttest.h
View File

@ -42,21 +42,28 @@ extern tt_test tt_tests[];
}while(0); }while(0);
#define ASSERT_STRN(lhs, rhs, n) do { \ #define ASSERT_STRN(lhs, rhs, n) do { \
size_t tt_n = (size_t)(n); \ char *tt_lhs = (char *)(lhs); \
char *tt_lhs = malloc(tt_n+1); \ char *tt_rhs = (char *)(rhs); \
char *tt_rhs = malloc(tt_n+1); \ if(!tt_lhs || !tt_rhs) \
memcpy(tt_lhs, (lhs), tt_n); \ { \
memcpy(tt_rhs, (rhs), tt_n); \ TT_FAIL("Expected string, got null pointer\n", 0); \
tt_rhs[tt_n] = tt_lhs[tt_n] = '\0'; \
if(strncmp(tt_lhs, tt_rhs, tt_n)) { \
TT_FAIL("Expected <%s> got <%s>\n", tt_rhs, tt_lhs); \
free(tt_lhs); free(tt_rhs); \
exit(1); \ exit(1); \
} \ } \
free(tt_lhs); free(tt_rhs); \ size_t tt_n = (size_t)(n); \
char *tt_lhs_c = malloc(tt_n+1); \
char *tt_rhs_c = malloc(tt_n+1); \
memcpy(tt_lhs_c, tt_lhs, tt_n); \
memcpy(tt_rhs_c, tt_rhs, tt_n); \
tt_rhs_c[tt_n] = tt_lhs_c[tt_n] = '\0'; \
if(strncmp(tt_lhs_c, tt_rhs_c, tt_n)) { \
TT_FAIL("Expected <%s> got <%s>\n", tt_rhs_c, tt_lhs_c); \
free(tt_lhs_c); free(tt_rhs_c); \
exit(1); \
} \
free(tt_lhs_c); free(tt_rhs_c); \
}while(0); }while(0);
#define TEST(name, body) void ttt_##name() { tt_current_test = #name; if(tt_before) tt_before(); body } #define TEST(name, ...) void ttt_##name() { tt_current_test = #name; if(tt_before) tt_before(); __VA_ARGS__ }
#define BEFORE(body) void tt_before() { body } #define BEFORE(body) void tt_before() { body }