/*-------------------------------------------------------------------------
 *
 * fe-connect.c--
 *	  functions related to setting up a connection to the backend
 *
 * Copyright (c) 1994, Regents of the University of California
 *
 *
 * IDENTIFICATION
 *	  $Header: /cvsroot/pgsql/src/interfaces/libpq/fe-connect.c,v 1.62 1998/01/31 21:27:28 scrappy Exp $
 *
 *-------------------------------------------------------------------------
 */

#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>
#include <ctype.h>
#include <string.h>
#include <netdb.h>
#include <sys/un.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <errno.h>
#include <signal.h>
#include <ctype.h>				/* for isspace() */

#include "postgres.h"
#include "fe-auth.h"
#include "fe-connect.h"
#include "libpq-fe.h"

#ifndef HAVE_STRDUP
#include "strdup.h"
#endif
#ifdef HAVE_CRYPT_H
#include <crypt.h>
#endif


/* use a local version instead of the one found in pqpacket.c */
static ConnStatusType connectDB(PGconn *conn);

static void freePGconn(PGconn *conn);
static void closePGconn(PGconn *conn);
static int	conninfo_parse(const char *conninfo, char *errorMessage);
static char *conninfo_getval(char *keyword);
static void conninfo_free(void);
void PQsetenv(PGconn *conn);

#define NOTIFYLIST_INITIAL_SIZE 10
#define NOTIFYLIST_GROWBY 10


/* ----------
 * Definition of the conninfo parameters and their fallback resources.
 * If Environment-Var and Compiled-in are specified as NULL, no
 * fallback is available. If after all no value can be determined
 * for an option, an error is returned.
 *
 * The values for dbname and user are treated special in conninfo_parse.
 * If the Compiled-in resource is specified as a NULL value, the
 * user is determined by fe_getauthname() and for dbname the user
 * name is copied.
 *
 * The Label and Disp-Char entries are provided for applications that
 * want to use PQconndefaults() to create a generic database connection
 * dialog. Disp-Char is defined as follows:
 *	   ""		Normal input field
 * ----------
 */
static PQconninfoOption PQconninfoOptions[] = {
/* ----------------------------------------------------------------- */
/*	  Option-name		Environment-Var Compiled-in		Current value	*/
/*						Label							Disp-Char		*/
/* ----------------- --------------- --------------- --------------- */
	/* "authtype" is ignored as it is no longer used. */
	{"authtype", "PGAUTHTYPE", DefaultAuthtype, NULL,
	"Database-Authtype", "", 20},

	{"user", "PGUSER", NULL, NULL,
	"Database-User", "", 20},

	{"password", "PGPASSWORD", DefaultPassword, NULL,
	"Database-Password", "", 20},

	{"dbname", "PGDATABASE", NULL, NULL,
	"Database-Name", "", 20},

	{"host", "PGHOST", NULL, NULL,
	"Database-Host", "", 40},

	{"port", "PGPORT", DEF_PGPORT, NULL,
	"Database-Port", "", 6},

	{"tty", "PGTTY", DefaultTty, NULL,
	"Backend-Debug-TTY", "D", 40},

	{"options", "PGOPTIONS", DefaultOption, NULL,
	"Backend-Debug-Options", "D", 40},
/* ----------------- --------------- --------------- --------------- */
	{NULL, NULL, NULL, NULL,
	NULL, NULL, 0}
};

struct EnvironmentOptions
{
	const char *envName,
			   *pgName;
}			EnvironmentOptions[] =

{
	/* common user-interface settings */
	{	"PGDATESTYLE",	"datestyle" },
	{	"PGTZ",			"timezone" },

	/* internal performance-related settings */
	{	"PGCOSTHEAP",	"cost_heap" },
	{	"PGCOSTINDEX",	"cost_index" },
	{	"PGRPLANS",		"r_plans" },
	{	"PGGEQO",		"geqo" },
	{	NULL }
};

/* ----------------
 *		PQconnectdb
 *
 * establishes a connection to a postgres backend through the postmaster
 * using connection information in a string.
 *
 * The conninfo string is a list of
 *
 *	   option = value
 *
 * definitions. Value might be a single value containing no whitespaces
 * or a single quoted string. If a single quote should appear everywhere
 * in the value, it must be escaped with a backslash like \'
 *
 * Returns a PGconn* which is needed for all subsequent libpq calls
 * if the status field of the connection returned is CONNECTION_BAD,
 * then some fields may be null'ed out instead of having valid values
 * ----------------
 */
PGconn *
PQconnectdb(const char *conninfo)
{
	PGconn	   *conn;
	char		errorMessage[ERROR_MSG_LENGTH];
	char* tmp;
	/* ----------
	 * Allocate memory for the conn structure
	 * ----------
	 */
	conn = (PGconn *) malloc(sizeof(PGconn));
	if (conn == NULL)
	{
		fprintf(stderr,
		   "FATAL: PQsetdb() -- unable to allocate memory for a PGconn");
		return (PGconn *) NULL;
	}
	MemSet((char *) conn, 0, sizeof(PGconn));

	/* ----------
	 * Parse the conninfo string and get the fallback resources
	 * ----------
	 */
	if (conninfo_parse(conninfo, errorMessage) < 0)
	{
		conn->status = CONNECTION_BAD;
		strcpy(conn->errorMessage, errorMessage);
		conninfo_free();
		return conn;
	}

	/* ----------
	 * Setup the conn structure
	 * ----------
	 */
	conn->lobjfuncs = (PGlobjfuncs *) NULL;
	conn->Pfout = NULL;
	conn->Pfin = NULL;
	conn->Pfdebug = NULL;
	conn->notifyList = DLNewList();

	tmp = conninfo_getval("host");
	conn->pghost = tmp ? strdup(tmp) : NULL;
	tmp = conninfo_getval("port");
	conn->pgport = tmp ? strdup(tmp) : NULL;
	tmp = conninfo_getval("tty");
	conn->pgtty  = tmp ? strdup(tmp) : NULL;
	tmp = conninfo_getval("options");
	conn->pgoptions = tmp ? strdup(tmp) : NULL;
	tmp = conninfo_getval("user");
	conn->pguser = tmp ? strdup(tmp) : NULL;
	tmp = conninfo_getval("password");
	conn->pgpass = tmp ? strdup(tmp) : NULL;
	tmp = conninfo_getval("dbname");
	conn->dbName = tmp ? strdup(tmp) : NULL;

	/* ----------
	 * Free the connection info - all is in conn now
	 * ----------
	 */
	conninfo_free();

	/* ----------
	 * Connect to the database
	 * ----------
	 */
	conn->status = connectDB(conn);
	if (conn->status == CONNECTION_OK)
	{
		PGresult   *res;

		/*
		 * Send a blank query to make sure everything works; in
		 * particular, that the database exists.
		 */
		res = PQexec(conn, " ");
		if (res == NULL || res->resultStatus != PGRES_EMPTY_QUERY)
		{
			/* PQexec has put error message in conn->errorMessage */
			closePGconn(conn);
		}
		PQclear(res);
	}

	PQsetenv(conn);

	return conn;
}

/* ----------------
 *		PQconndefaults
 *
 * Parse an empty string like PQconnectdb() would do and return the
 * address of the connection options structure. Using this function
 * an application might determine all possible options and their
 * current default values.
 * ----------------
 */
PQconninfoOption *
PQconndefaults(void)
{
	char		errorMessage[ERROR_MSG_LENGTH];

	conninfo_parse("", errorMessage);
	return PQconninfoOptions;
}

/* ----------------
 *		PQsetdbLogin
 *
 * establishes a connection to a postgres backend through the postmaster
 * at the specified host and port.
 *
 * returns a PGconn* which is needed for all subsequent libpq calls
 * if the status field of the connection returned is CONNECTION_BAD,
 * then some fields may be null'ed out instead of having valid values
 *
 *	Uses these environment variables:
 *
 *	  PGHOST	   identifies host to which to connect if <pghost> argument
 *				   is NULL or a null string.
 *
 *	  PGPORT	   identifies TCP port to which to connect if <pgport> argument
 *				   is NULL or a null string.
 *
 *	  PGTTY		   identifies tty to which to send messages if <pgtty> argument
 *				   is NULL or a null string.
 *
 *	  PGOPTIONS    identifies connection options if <pgoptions> argument is
 *				   NULL or a null string.
 *
 *	  PGUSER	   Postgres username to associate with the connection.
 *
 *	  PGPASSWORD   The user's password.
 *
 *	  PGDATABASE   name of database to which to connect if <pgdatabase>
 *				   argument is NULL or a null string
 *
 *	  None of the above need be defined.  There are defaults for all of them.
 *
 * To support "delimited identifiers" for database names, only convert
 * the database name to lower case if it is not surrounded by double quotes.
 * Otherwise, strip the double quotes but leave the reset of the string intact.
 * - thomas 1997-11-08
 *
 * ----------------
 */
PGconn	   *
PQsetdbLogin(const char *pghost, const char *pgport, const char *pgoptions, const char *pgtty, const char *dbName, const char *login, const char *pwd)
{
	PGconn	   *conn;
	char	   *tmp;
	char		errorMessage[ERROR_MSG_LENGTH];

	/* An error message from some service we call. */
	bool		error;

	/* We encountered an error that prevents successful completion */
	int			i;

	conn = (PGconn *) malloc(sizeof(PGconn));

	if (conn == NULL)
		fprintf(stderr,
		   "FATAL: PQsetdb() -- unable to allocate memory for a PGconn");
	else
	{
		conn->lobjfuncs = (PGlobjfuncs *) NULL;
		conn->Pfout = NULL;
		conn->Pfin = NULL;
		conn->Pfdebug = NULL;
		conn->notifyList = DLNewList();

		if ((pghost == NULL) || pghost[0] == '\0')
		{
			conn->pghost = NULL;
			if ((tmp = getenv("PGHOST")) != NULL)
				conn->pghost = strdup(tmp);
		}
		else
			conn->pghost = strdup(pghost);

		if ((pgport == NULL) || pgport[0] == '\0')
		{
			if ((tmp = getenv("PGPORT")) == NULL)
				tmp = DEF_PGPORT;
			conn->pgport = strdup(tmp);
		}
		else
			conn->pgport = strdup(pgport);

		if ((pgtty == NULL) || pgtty[0] == '\0')
		{
			if ((tmp = getenv("PGTTY")) == NULL)
				tmp = DefaultTty;
			conn->pgtty = strdup(tmp);
		}
		else
			conn->pgtty = strdup(pgtty);

		if ((pgoptions == NULL) || pgoptions[0] == '\0')
		{
			if ((tmp = getenv("PGOPTIONS")) == NULL)
				tmp = DefaultOption;
			conn->pgoptions = strdup(tmp);
		}
		else
			conn->pgoptions = strdup(pgoptions);

		if (login)
		{
			error = FALSE;
			conn->pguser = strdup(login);
		}
		else if ((tmp = getenv("PGUSER")) != NULL)
		{
			error = FALSE;
			conn->pguser = strdup(tmp);
		}
		else
		{
			tmp = fe_getauthname(errorMessage);
			if (tmp == 0)
			{
				error = TRUE;
				sprintf(conn->errorMessage,
						"FATAL: PQsetdb: Unable to determine a Postgres username!\n");
			}
			else
			{
				error = FALSE;
				conn->pguser = tmp;
			}
		}

		if (pwd)
		{
			conn->pgpass = strdup(pwd);
		}
		else if ((tmp = getenv("PGPASSWORD")) != NULL)
		{
			conn->pgpass = strdup(tmp);
		}
		else
			conn->pgpass = strdup(DefaultPassword);

		if (!error)
		{
			if ((((tmp = (char *) dbName) != NULL) && (dbName[0] != '\0'))
				|| ((tmp = getenv("PGDATABASE"))))
				conn->dbName = strdup(tmp);
			else
				conn->dbName = strdup(conn->pguser);

			/*
			 * if the database name is surrounded by double-quotes,
			 *  then don't convert case
			 */
			if (*conn->dbName == '"')
			{
				strcpy(conn->dbName, conn->dbName + 1);
				*(conn->dbName + strlen(conn->dbName) - 1) = '\0';
			}
			else
				for (i = 0; conn->dbName[i]; i++)
					if (isupper(conn->dbName[i]))
						conn->dbName[i] = tolower(conn->dbName[i]);
		}
		else
			conn->dbName = NULL;

		if (error)
			conn->status = CONNECTION_BAD;
		else
		{
			conn->status = connectDB(conn);
			/* Puts message in conn->errorMessage */
			if (conn->status == CONNECTION_OK)
			{
				PGresult   *res;

				/*
				 * Send a blank query to make sure everything works; in
				 * particular, that the database exists.
				 */
				res = PQexec(conn, " ");
				if (res == NULL || res->resultStatus != PGRES_EMPTY_QUERY)
				{
					/* PQexec has put error message in conn->errorMessage */
					closePGconn(conn);
				}
				PQclear(res);
			}
			PQsetenv(conn);
		}
	}
	return conn;
}


/*
 * connectDB -
 * make a connection to the backend so it is ready to receive queries.
 * return CONNECTION_OK if successful, CONNECTION_BAD if not.
 *
 */
static ConnStatusType
connectDB(PGconn *conn)
{
	struct hostent *hp;

	StartupPacket	sp;
	AuthRequest	areq;
	int			laddrlen = sizeof(SockAddr);
	int			portno,
				family,
				len;

	/*
	 * Initialize the startup packet.
	 */

	MemSet((char *)&sp, 0, sizeof (StartupPacket));

	sp.protoVersion = (ProtocolVersion)htonl(PG_PROTOCOL_LATEST);

	strncpy(sp.user, conn->pguser, SM_USER);
	strncpy(sp.database, conn->dbName, SM_DATABASE);
	strncpy(sp.tty, conn->pgtty, SM_TTY);

	if (conn->pgoptions)
		strncpy(sp.options, conn->pgoptions, SM_OPTIONS);

	/*
	 * Open a connection to postmaster/backend.
	 */

	if (conn->pghost != NULL)
	{
		hp = gethostbyname(conn->pghost);
		if ((hp == NULL) || (hp->h_addrtype != AF_INET))
		{
			(void) sprintf(conn->errorMessage,
						   "connectDB() --  unknown hostname: %s\n",
						   conn->pghost);
			goto connect_errReturn;
		}
	} else
		hp = NULL;

#if FALSE
	MemSet((char *) &port->raddr, 0, sizeof(port->raddr));
#endif
	portno = atoi(conn->pgport);
	family = (conn->pghost != NULL) ? AF_INET : AF_UNIX;
	conn->raddr.sa.sa_family = family;
	if (family == AF_INET)
	{
		memmove((char *) &(conn->raddr.in.sin_addr),
				(char *) hp->h_addr,
				hp->h_length);
		conn->raddr.in.sin_port = htons((unsigned short) (portno));
		len = sizeof(struct sockaddr_in);
	}
	else
	{
		len = UNIXSOCK_PATH(conn->raddr.un, portno);
	}
	/* connect to the server  */
	if ((conn->sock = socket(family, SOCK_STREAM, 0)) < 0)
	{
		(void) sprintf(conn->errorMessage,
					   "connectDB() -- socket() failed: errno=%d\n%s\n",
					   errno, strerror(errno));
		goto connect_errReturn;
	}
	if (connect(conn->sock, &conn->raddr.sa, len) < 0)
	{
		(void) sprintf(conn->errorMessage,
					   "connectDB() failed: Is the postmaster running and accepting%s connections at '%s' on port '%s'?\n",
					   conn->pghost ? " TCP/IP(with -i)" : "",
					   conn->pghost ? conn->pghost : "UNIX Socket",
					   conn->pgport);
		goto connect_errReturn;
	}
	if (family == AF_INET)
	{
		struct protoent *pe;
		int			on = 1;

		pe = getprotobyname("TCP");
		if (pe == NULL)
		{
			(void) sprintf(conn->errorMessage,
						   "connectDB(): getprotobyname failed\n");
			goto connect_errReturn;
		}
		if (setsockopt(conn->sock, pe->p_proto, TCP_NODELAY,
					   &on, sizeof(on)) < 0)
		{
			(void) sprintf(conn->errorMessage,
						   "connectDB(): setsockopt failed\n");
			goto connect_errReturn;
		}
	}

	/* fill in the client address */
	if (getsockname(conn->sock, &conn->laddr.sa, &laddrlen) < 0)
	{
		(void) sprintf(conn->errorMessage,
				   "connectDB() -- getsockname() failed: errno=%d\n%s\n",
					   errno, strerror(errno));
		goto connect_errReturn;
	}

	/* set up the socket file descriptors */
	conn->Pfout = fdopen(conn->sock, "w");
	conn->Pfin = fdopen(dup(conn->sock), "r");
	if ((conn->Pfout == NULL) || (conn->Pfin == NULL))
	{
		(void) sprintf(conn->errorMessage,
					   "connectDB() -- fdopen() failed: errno=%d\n%s\n",
					   errno, strerror(errno));
		goto connect_errReturn;
	}

	/* Send the startup packet. */

	if (packetSend(conn, (char *)&sp, sizeof(StartupPacket)) != STATUS_OK)
	{
		sprintf(conn->errorMessage,
				"connectDB() --  couldn't send complete packet: errno=%d\n%s\n", errno, strerror(errno));
		goto connect_errReturn;
	}

	/*
	 * Get the response from the backend, either an error message or an
	 * authentication request.
	 */

	do
	{
		int beresp;

		if ((beresp = pqGetc(conn->Pfin, conn->Pfdebug)) == EOF)
		{
			(void)sprintf(conn->errorMessage,
					"connectDB() -- error getting authentication request\n");

			goto connect_errReturn;
		}

		/* Handle errors. */

		if (beresp == 'E')
		{
			pqGets(conn->errorMessage, sizeof (conn->errorMessage),
				conn->Pfin, conn->Pfdebug);

			goto connect_errReturn;
		}

		/* Check it was an authentication request. */

		if (beresp != 'R')
		{
			(void)sprintf(conn->errorMessage,
					"connectDB() -- expected authentication request\n");

			goto connect_errReturn;
		}

		/* Get the type of request. */

		if (pqGetInt((int *)&areq, 4, conn->Pfin, conn->Pfdebug))
		{
			(void)sprintf(conn->errorMessage,
					"connectDB() -- error getting authentication request type\n");

			goto connect_errReturn;
		}

		/* Get the password salt if there is one. */

		if (areq == AUTH_REQ_CRYPT &&
			pqGetnchar(conn->salt, sizeof (conn->salt),
					conn->Pfin, conn->Pfdebug))
		{
			(void)sprintf(conn->errorMessage,
					"connectDB() -- error getting password salt\n");

			goto connect_errReturn;
		}


		/* Respond to the request. */

		if (fe_sendauth(areq, conn, conn->pghost, conn->pgpass,
					conn->errorMessage) != STATUS_OK)
			goto connect_errReturn;
	}
	while (areq != AUTH_REQ_OK);

	/* free the password so it's not hanging out in memory forever */
	if (conn->pgpass != NULL)
	{
		free(conn->pgpass);
	}

	return CONNECTION_OK;

connect_errReturn:
	return CONNECTION_BAD;

}

void
PQsetenv(PGconn *conn)
{
	struct EnvironmentOptions *eo;
	char setQuery[80];		/* mjl: size okay? XXX */

	for (eo = EnvironmentOptions; eo->envName; eo++)
	{
		const char *val;

		if ((val = getenv(eo->envName)))
		{
			PGresult   *res;

			if (strcasecmp(val, "default") == 0)
				sprintf(setQuery, "SET %s = %.60s", eo->pgName, val);
			else
				sprintf(setQuery, "SET %s = '%.60s'", eo->pgName, val);
#ifdef CONNECTDEBUG
printf("Use environment variable %s to send %s\n", eo->envName, setQuery);
#endif
			res = PQexec(conn, setQuery);
			PQclear(res);	/* Don't care? */
		}
	}
} /* PQsetenv() */

/*
 * freePGconn
 *	 - free the PGconn data structure
 *
 */
static void
freePGconn(PGconn *conn)
{
	if (!conn)
		return;
	if (conn->pghost)
		free(conn->pghost);
	if (conn->pgtty)
		free(conn->pgtty);
	if (conn->pgoptions)
		free(conn->pgoptions);
	if (conn->pgport)
		free(conn->pgport);
	if (conn->dbName)
		free(conn->dbName);
	if (conn->pguser)
		free(conn->pguser);
	if (conn->notifyList)
		DLFreeList(conn->notifyList);
	free(conn);
}

/*
   closePGconn
	 - properly close a connection to the backend
*/
static void
closePGconn(PGconn *conn)
{
/* GH: What to do for !USE_POSIX_SIGNALS ? */
#if defined(USE_POSIX_SIGNALS)
	struct sigaction ignore_action;

	/*
	 * This is used as a constant, but not declared as such because the
	 * sigaction structure is defined differently on different systems
	 */
	struct sigaction oldaction;

	/*
	 * If connection is already gone, that's cool.  No reason for kernel
	 * to kill us when we try to write to it.  So ignore SIGPIPE signals.
	 */
	ignore_action.sa_handler = SIG_IGN;
	sigemptyset(&ignore_action.sa_mask);
	ignore_action.sa_flags = 0;
	sigaction(SIGPIPE, (struct sigaction *) & ignore_action, &oldaction);

	fputs("X\0", conn->Pfout);
	fflush(conn->Pfout);
	sigaction(SIGPIPE, &oldaction, NULL);
#else
				signal(SIGPIPE, SIG_IGN);
	fputs("X\0", conn->Pfout);
	fflush(conn->Pfout);
	signal(SIGPIPE, SIG_DFL);
#endif
	if (conn->Pfout)
		fclose(conn->Pfout);
	if (conn->Pfin)
		fclose(conn->Pfin);
	if (conn->Pfdebug)
		fclose(conn->Pfdebug);
	conn->status = CONNECTION_BAD;		/* Well, not really _bad_ - just
										 * absent */
}

/*
   PQfinish:
	  properly close a connection to the backend
	  also frees the PGconn data structure so it shouldn't be re-used
	  after this
*/
void
PQfinish(PGconn *conn)
{
	if (!conn)
	{
		fprintf(stderr, "PQfinish() -- pointer to PGconn is null\n");
	}
	else
	{
		if (conn->status == CONNECTION_OK)
			closePGconn(conn);
		freePGconn(conn);
	}
}

/* PQreset :
   resets the connection to the backend
   closes the existing connection and makes a new one
*/
void
PQreset(PGconn *conn)
{
	if (!conn)
	{
		fprintf(stderr, "PQreset() -- pointer to PGconn is null\n");
	}
	else
	{
		closePGconn(conn);
		conn->status = connectDB(conn);
	}
}

/*
 * PacketSend()
 *
 this is just like PacketSend(), defined in backend/libpq/pqpacket.c
 but we define it here to avoid linking in all of libpq.a

 * packetSend -- send a single-packet message.
 *
 * RETURNS: STATUS_ERROR if the write fails, STATUS_OK otherwise.
 * SIDE_EFFECTS: may block.
*/
int
packetSend(PGconn *conn, const char *buf, size_t len)
{
	/* Send the total packet size. */

	if (pqPutInt(4 + len, 4, conn->Pfout, conn->Pfdebug))
		return STATUS_ERROR;

	/* Send the packet itself. */

	if (pqPutnchar(buf, len, conn->Pfout, conn->Pfdebug))
		return STATUS_ERROR;

	pqFlush(conn->Pfout, conn->Pfdebug);

	return STATUS_OK;
}


/* ----------------
 * Conninfo parser routine
 * ----------------
 */
static int
conninfo_parse(const char *conninfo, char *errorMessage)
{
	char	   *pname;
	char	   *pval;
	char	   *buf;
	char	   *tmp;
	char	   *cp;
	char	   *cp2;
	PQconninfoOption *option;
	char		errortmp[ERROR_MSG_LENGTH];

	conninfo_free();

	if ((buf = strdup(conninfo)) == NULL)
	{
		strcpy(errorMessage,
		  "FATAL: cannot allocate memory for copy of conninfo string\n");
		return -1;
	}
	cp = buf;

	while (*cp)
	{
		/* Skip blanks before the parameter name */
		if (isspace(*cp))
		{
			cp++;
			continue;
		}

		/* Get the parameter name */
		pname = cp;
		while (*cp)
		{
			if (*cp == '=')
			{
				break;
			}
			if (isspace(*cp))
			{
				*cp++ = '\0';
				while (*cp)
				{
					if (!isspace(*cp))
					{
						break;
					}
					cp++;
				}
				break;
			}
			cp++;
		}

		/* Check that there is a following '=' */
		if (*cp != '=')
		{
			sprintf(errorMessage,
			"ERROR: PQconnectdb() - Missing '=' after '%s' in conninfo\n",
					pname);
			free(buf);
			return -1;
		}
		*cp++ = '\0';

		/* Skip blanks after the '=' */
		while (*cp)
		{
			if (!isspace(*cp))
			{
				break;
			}
			cp++;
		}

		pval = cp;

		if (*cp != '\'')
		{
			cp2 = pval;
			while (*cp)
			{
				if (isspace(*cp))
				{
					*cp++ = '\0';
					break;
				}
				if (*cp == '\\')
				{
					cp++;
					if (*cp != '\0')
					{
						*cp2++ = *cp++;
					}
				}
				else
				{
					*cp2++ = *cp++;
				}
			}
			*cp2 = '\0';
		}
		else
		{
			cp2 = pval;
			cp++;
			for (;;)
			{
				if (*cp == '\0')
				{
					sprintf(errorMessage,
							"ERROR: PQconnectdb() - unterminated quoted string in conninfo\n");
					free(buf);
					return -1;
				}
				if (*cp == '\\')
				{
					cp++;
					if (*cp != '\0')
					{
						*cp2++ = *cp++;
					}
					continue;
				}
				if (*cp == '\'')
				{
					*cp2 = '\0';
					cp++;
					break;
				}
				*cp2++ = *cp++;
			}
		}

		/* ----------
		 * Now we have the name and the value. Search
		 * for the param record.
		 * ----------
		 */
		for (option = PQconninfoOptions; option->keyword != NULL; option++)
		{
			if (!strcmp(option->keyword, pname))
			{
				break;
			}
		}
		if (option->keyword == NULL)
		{
			sprintf(errorMessage,
					"ERROR: PQconnectdb() - unknown option '%s'\n",
					pname);
			free(buf);
			return -1;
		}

		/* ----------
		 * Store the value
		 * ----------
		 */
		option->val = strdup(pval);
	}

	free(buf);

	/* ----------
	 * Get the fallback resources for parameters not specified
	 * in the conninfo string.
	 * ----------
	 */
	for (option = PQconninfoOptions; option->keyword != NULL; option++)
	{
		if (option->val != NULL)
			continue;			/* Value was in conninfo */

		/* ----------
		 * Try to get the environment variable fallback
		 * ----------
		 */
		if (option->environ != NULL)
		{
			if ((tmp = getenv(option->environ)) != NULL)
			{
				option->val = strdup(tmp);
				continue;
			}
		}

		/* ----------
		 * No environment variable specified or this one isn't set -
		 * try compiled in
		 * ----------
		 */
		if (option->compiled != NULL)
		{
			option->val = strdup(option->compiled);
			continue;
		}

		/* ----------
		 * Special handling for user
		 * ----------
		 */
		if (!strcmp(option->keyword, "user"))
		{
			tmp = fe_getauthname(errortmp);
			if (tmp)
			{
				option->val = strdup(tmp);
			}
		}

		/* ----------
		 * Special handling for dbname
		 * ----------
		 */
		if (!strcmp(option->keyword, "dbname"))
		{
			tmp = conninfo_getval("user");
			if (tmp)
			{
				option->val = strdup(tmp);
			}
		}
	}

	return 0;
}


static char *
conninfo_getval(char *keyword)
{
	PQconninfoOption *option;

	for (option = PQconninfoOptions; option->keyword != NULL; option++)
	{
		if (!strcmp(option->keyword, keyword))
		{
			return option->val;
		}
	}

	return NULL;
}


static void
conninfo_free()
{
	PQconninfoOption *option;

	for (option = PQconninfoOptions; option->keyword != NULL; option++)
	{
		if (option->val != NULL)
		{
			free(option->val);
			option->val = NULL;
		}
	}
}

/* =========== accessor functions for PGconn ========= */
char	   *
PQdb(PGconn *conn)
{
	if (!conn)
	{
		fprintf(stderr, "PQdb() -- pointer to PGconn is null\n");
		return (char *) NULL;
	}
	return conn->dbName;
}

char	   *
PQuser(PGconn *conn)
{
	if (!conn)
	{
		fprintf(stderr, "PQuser() -- pointer to PGconn is null\n");
		return (char *) NULL;
	}
	return conn->pguser;
}

char	   *
PQhost(PGconn *conn)
{
	if (!conn)
	{
		fprintf(stderr, "PQhost() -- pointer to PGconn is null\n");
		return (char *) NULL;
	}

	return conn->pghost;
}

char	   *
PQoptions(PGconn *conn)
{
	if (!conn)
	{
		fprintf(stderr, "PQoptions() -- pointer to PGconn is null\n");
		return (char *) NULL;
	}
	return conn->pgoptions;
}

char	   *
PQtty(PGconn *conn)
{
	if (!conn)
	{
		fprintf(stderr, "PQtty() -- pointer to PGconn is null\n");
		return (char *) NULL;
	}
	return conn->pgtty;
}

char	   *
PQport(PGconn *conn)
{
	if (!conn)
	{
		fprintf(stderr, "PQport() -- pointer to PGconn is null\n");
		return (char *) NULL;
	}
	return conn->pgport;
}

ConnStatusType
PQstatus(PGconn *conn)
{
	if (!conn)
	{
		fprintf(stderr, "PQstatus() -- pointer to PGconn is null\n");
		return CONNECTION_BAD;
	}
	return conn->status;
}

char	   *
PQerrorMessage(PGconn *conn)
{
	if (!conn)
	{
		fprintf(stderr, "PQerrorMessage() -- pointer to PGconn is null\n");
		return (char *) NULL;
	}
	return conn->errorMessage;
}

void
PQtrace(PGconn *conn, FILE *debug_port)
{
	if (conn == NULL ||
		conn->status == CONNECTION_BAD)
	{
		return;
	}
	PQuntrace(conn);
	conn->Pfdebug = debug_port;
}

void
PQuntrace(PGconn *conn)
{
	if (conn == NULL ||
		conn->status == CONNECTION_BAD)
	{
		return;
	}
	if (conn->Pfdebug)
	{
		fflush(conn->Pfdebug);
		conn->Pfdebug = NULL;
	}
}
