properly zero-terminate input buffer - Bug #605

http://bugs.mariadb.com/show_bug.cgi?id=605
2014-11-02 14:18:22 +01:00
parent ff30e2f206
commit d23cb087ec
1 changed files with 2 additions and 1 deletions
--- a/query_classifier/test/canonical_tests/canonizer.c
+++ b/query_classifier/test/canonical_tests/canonizer.c
@ -51,12 +51,13 @@ int main(int argc, char** argv)
  fsz = lseek(fdin,0,SEEK_END);
  lseek(fdin,0,SEEK_SET);

-  if(!(buffer = malloc(sizeof(char)*fsz))){
+  if(!(buffer = malloc(sizeof(char)*fsz + 1))){
    printf("Error: Failed to allocate memory.");
    return 1;
  }

  read(fdin,buffer,fsz);
+  buffer[fsz] = '\0';