1
0
mirror of https://github.com/postgres/postgres.git synced 2025-07-31 22:04:40 +03:00

Fix uninitialized memory bug

Add support for PGHOST, PGPORT, PGUSER environment variables
This commit is contained in:
Tatsuo Ishii
2003-06-10 09:07:15 +00:00
parent cdfb3d9981
commit 412893b416
3 changed files with 23 additions and 4 deletions

View File

@ -1,5 +1,5 @@
/*
* $Header: /cvsroot/pgsql/contrib/pgbench/pgbench.c,v 1.23 2003/05/14 03:25:56 tgl Exp $
* $Header: /cvsroot/pgsql/contrib/pgbench/pgbench.c,v 1.24 2003/06/10 09:07:15 ishii Exp $
*
* pgbench: a simple TPC-B like benchmark program for PostgreSQL
* written by Tatsuo Ishii
@ -694,6 +694,14 @@ main(int argc, char **argv)
PGconn *con;
PGresult *res;
char *env;
if ((env = getenv("PGHOST")) != NULL && *env != '\0')
pghost = env;
if ((env = getenv("PGPORT")) != NULL && *env != '\0')
pgport = env;
else if ((env = getenv("PGUSER")) != NULL && *env != '\0')
login = env;
while ((c = getopt(argc, argv, "ih:nvp:dc:t:s:U:P:CNSl")) != -1)
{
@ -788,8 +796,11 @@ main(int argc, char **argv)
dbName = argv[optind];
else
{
dbName = getenv("USER");
if (dbName == NULL)
if ((env = getenv("PGDATABASE")) != NULL && *env != '\0')
dbName = env;
else if (login != NULL && *login != '\0')
dbName = login;
else
dbName = "";
}
@ -802,7 +813,7 @@ main(int argc, char **argv)
remains = nclients;
state = (CState *) malloc(sizeof(*state) * nclients);
memset(state, 0, sizeof(*state));
memset(state, 0, sizeof(*state) * nclients);
if (use_log)
{