@@ -19,7 +19,7 @@ struct TestCase
19
19
PGconn * connect_database (
20
20
const char * db_host , const char * db_port , const char * db_user , const char * db_password , const char * db_name )
21
21
{
22
- const int const_db_uri_chars = strlen ("host= port= user= dbname= sslmode=disable password=" );
22
+ const int const_db_uri_chars = strlen ("host= port= user= dbname= sslmode=disable password=" ) + 1 ;
23
23
char * db_uri = malloc (strlen (db_host ) + strlen (db_port ) + strlen (db_user ) + strlen (db_password ) + strlen (db_name )
24
24
+ const_db_uri_chars );
25
25
sprintf (db_uri ,
@@ -44,7 +44,7 @@ PGconn *connect_database(
44
44
int recreate_database (PGconn * root_conn , const char * test_db_name )
45
45
{
46
46
char * statement = "DROP DATABASE IF EXISTS " ;
47
- char * full_statement = malloc (strlen (statement ) + strlen (test_db_name ));
47
+ char * full_statement = malloc (strlen (statement ) + strlen (test_db_name ) + 1 );
48
48
sprintf (full_statement , "%s%s" , statement , test_db_name );
49
49
PGresult * res = PQexec (root_conn , full_statement );
50
50
free (full_statement );
@@ -56,7 +56,7 @@ int recreate_database(PGconn *root_conn, const char *test_db_name)
56
56
}
57
57
58
58
statement = "CREATE DATABASE " ;
59
- full_statement = malloc (strlen (statement ) + strlen (test_db_name ));
59
+ full_statement = malloc (strlen (statement ) + strlen (test_db_name ) + 1 );
60
60
sprintf (full_statement , "%s%s" , statement , test_db_name );
61
61
res = PQexec (root_conn , full_statement );
62
62
free (full_statement );
0 commit comments