Fix overflow bug when comparing strings

2017-12-03 23:05:22 +01:00
parent e50fbff2d1
commit 1b80c00658
1 changed files with 2 additions and 2 deletions
--- a/src/kernel/include/ttest.h
+++ b/src/kernel/include/ttest.h
@@ -41,8 +41,8 @@ extern tt_test tt_tests[];

 #define ASSERT_STRN(lhs, rhs, n) do { \
  size_t tt_n = (size_t)(n); \
-  char *tt_lhs = malloc(tt_n); \
-  char *tt_rhs = malloc(tt_n); \
+  char *tt_lhs = malloc(tt_n+1); \
+  char *tt_rhs = malloc(tt_n+1); \
  memcpy(tt_lhs, (lhs), tt_n); \
  memcpy(tt_rhs, (rhs), tt_n); \
  tt_rhs[tt_n] = tt_lhs[tt_n] = '\0'; \