|
|
|
/*
|
|
|
|
* psql - the PostgreSQL interactive terminal
|
|
|
|
*
|
|
|
|
* Copyright (c) 2000-2014, PostgreSQL Global Development Group
|
|
|
|
*
|
|
|
|
* src/bin/psql/startup.c
|
|
|
|
*/
|
|
|
|
#include "postgres_fe.h"
|
|
|
|
|
|
|
|
#include <sys/types.h>
|
|
|
|
|
|
|
|
#ifndef WIN32
|
|
|
|
#include <unistd.h>
|
|
|
|
#else /* WIN32 */
|
|
|
|
#include <io.h>
|
|
|
|
#include <win32.h>
|
|
|
|
#endif /* WIN32 */
|
|
|
|
|
|
|
|
#include "getopt_long.h"
|
|
|
|
|
|
|
|
#include <locale.h>
|
|
|
|
|
|
|
|
#include "command.h"
|
|
|
|
#include "common.h"
|
|
|
|
#include "describe.h"
|
|
|
|
#include "help.h"
|
|
|
|
#include "input.h"
|
|
|
|
#include "mainloop.h"
|
|
|
|
#include "print.h"
|
|
|
|
#include "settings.h"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Global psql options
|
|
|
|
*/
|
|
|
|
PsqlSettings pset;
|
|
|
|
|
|
|
|
#ifndef WIN32
|
|
|
|
#define SYSPSQLRC "psqlrc"
|
|
|
|
#define PSQLRC ".psqlrc"
|
|
|
|
#else
|
|
|
|
#define SYSPSQLRC "psqlrc"
|
|
|
|
#define PSQLRC "psqlrc.conf"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Structures to pass information between the option parsing routine
|
|
|
|
* and the main function
|
|
|
|
*/
|
|
|
|
enum _actions
|
|
|
|
{
|
|
|
|
ACT_NOTHING = 0,
|
|
|
|
ACT_SINGLE_SLASH,
|
|
|
|
ACT_LIST_DB,
|
|
|
|
ACT_SINGLE_QUERY,
|
|
|
|
ACT_FILE
|
|
|
|
};
|
|
|
|
|
|
|
|
struct adhoc_opts
|
|
|
|
{
|
|
|
|
char *dbname;
|
|
|
|
char *host;
|
|
|
|
char *port;
|
|
|
|
char *username;
|
|
|
|
char *logfilename;
|
|
|
|
enum _actions action;
|
|
|
|
char *action_string;
|
|
|
|
bool no_readline;
|
|
|
|
bool no_psqlrc;
|
|
|
|
bool single_txn;
|
|
|
|
};
|
|
|
|
|
|
|
|
static void parse_psql_options(int argc, char *argv[],
|
|
|
|
struct adhoc_opts * options);
|
|
|
|
static void process_psqlrc(char *argv0);
|
|
|
|
static void process_psqlrc_file(char *filename);
|
|
|
|
static void showVersion(void);
|
|
|
|
static void EstablishVariableSpace(void);
|
|
|
|
|
Add new psql help topics, accessible to both --help and \?.
Add --help=<topic> for the commandline, and \? <topic> as a backslash
command, to show more help than the invocations without parameters
do. "commands", "variables" and "options" currently exist as help
topics describing, respectively, backslash commands, psql variables,
and commandline switches. Without parameters the help commands show
their previous topic.
Some further wordsmithing or extending of the added help content might
be needed; but there seems little benefit delaying the overall feature
further.
Author: Pavel Stehule, editorialized by many
Reviewed-By: Andres Freund, Petr Jelinek, Fujii Masao, MauMau, Abhijit
Menon-Sen and Erik Rijkers.
Discussion: CAFj8pRDVGuC-nXBfe2CK8vpyzd2Dsr9GVpbrATAnZO=2YQ0s2Q@mail.gmail.com,
CAFj8pRA54AbTv2RXDTRxiAd8hy8wxmoVLqhJDRCwEnhdd7OUkw@mail.gmail.com
11 years ago
|
|
|
#define NOPAGER 0
|
|
|
|
|
|
|
|
/*
|
|
|
|
*
|
|
|
|
* main
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
int
|
|
|
|
main(int argc, char *argv[])
|
|
|
|
{
|
|
|
|
struct adhoc_opts options;
|
|
|
|
int successResult;
|
|
|
|
char *password = NULL;
|
|
|
|
char *password_prompt = NULL;
|
|
|
|
bool new_pass;
|
|
|
|
|
|
|
|
set_pglocale_pgservice(argv[0], PG_TEXTDOMAIN("psql"));
|
|
|
|
|
|
|
|
if (argc > 1)
|
|
|
|
{
|
Add new psql help topics, accessible to both --help and \?.
Add --help=<topic> for the commandline, and \? <topic> as a backslash
command, to show more help than the invocations without parameters
do. "commands", "variables" and "options" currently exist as help
topics describing, respectively, backslash commands, psql variables,
and commandline switches. Without parameters the help commands show
their previous topic.
Some further wordsmithing or extending of the added help content might
be needed; but there seems little benefit delaying the overall feature
further.
Author: Pavel Stehule, editorialized by many
Reviewed-By: Andres Freund, Petr Jelinek, Fujii Masao, MauMau, Abhijit
Menon-Sen and Erik Rijkers.
Discussion: CAFj8pRDVGuC-nXBfe2CK8vpyzd2Dsr9GVpbrATAnZO=2YQ0s2Q@mail.gmail.com,
CAFj8pRA54AbTv2RXDTRxiAd8hy8wxmoVLqhJDRCwEnhdd7OUkw@mail.gmail.com
11 years ago
|
|
|
if ((strcmp(argv[1], "-?") == 0) || (argc == 2 && (strcmp(argv[1], "--help") == 0)))
|
|
|
|
{
|
Add new psql help topics, accessible to both --help and \?.
Add --help=<topic> for the commandline, and \? <topic> as a backslash
command, to show more help than the invocations without parameters
do. "commands", "variables" and "options" currently exist as help
topics describing, respectively, backslash commands, psql variables,
and commandline switches. Without parameters the help commands show
their previous topic.
Some further wordsmithing or extending of the added help content might
be needed; but there seems little benefit delaying the overall feature
further.
Author: Pavel Stehule, editorialized by many
Reviewed-By: Andres Freund, Petr Jelinek, Fujii Masao, MauMau, Abhijit
Menon-Sen and Erik Rijkers.
Discussion: CAFj8pRDVGuC-nXBfe2CK8vpyzd2Dsr9GVpbrATAnZO=2YQ0s2Q@mail.gmail.com,
CAFj8pRA54AbTv2RXDTRxiAd8hy8wxmoVLqhJDRCwEnhdd7OUkw@mail.gmail.com
11 years ago
|
|
|
usage(NOPAGER);
|
|
|
|
exit(EXIT_SUCCESS);
|
|
|
|
}
|
|
|
|
if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
|
|
|
|
{
|
|
|
|
showVersion();
|
|
|
|
exit(EXIT_SUCCESS);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef WIN32
|
|
|
|
setvbuf(stderr, NULL, _IONBF, 0);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
pset.progname = get_progname(argv[0]);
|
|
|
|
|
|
|
|
pset.db = NULL;
|
|
|
|
setDecimalLocale();
|
|
|
|
pset.encoding = PQenv2encoding();
|
|
|
|
pset.queryFout = stdout;
|
|
|
|
pset.queryFoutPipe = false;
|
|
|
|
pset.copyStream = NULL;
|
|
|
|
pset.cur_cmd_source = stdin;
|
|
|
|
pset.cur_cmd_interactive = false;
|
|
|
|
|
|
|
|
/* We rely on unmentioned fields of pset.popt to start out 0/false/NULL */
|
|
|
|
pset.popt.topt.format = PRINT_ALIGNED;
|
|
|
|
pset.popt.topt.border = 1;
|
|
|
|
pset.popt.topt.pager = 1;
|
|
|
|
pset.popt.topt.start_table = true;
|
|
|
|
pset.popt.topt.stop_table = true;
|
|
|
|
pset.popt.topt.default_footer = true;
|
|
|
|
|
|
|
|
pset.popt.topt.unicode_border_linestyle = UNICODE_LINESTYLE_SINGLE;
|
|
|
|
pset.popt.topt.unicode_column_linestyle = UNICODE_LINESTYLE_SINGLE;
|
|
|
|
pset.popt.topt.unicode_header_linestyle = UNICODE_LINESTYLE_SINGLE;
|
|
|
|
|
|
|
|
refresh_utf8format(&(pset.popt.topt));
|
|
|
|
|
|
|
|
/* We must get COLUMNS here before readline() sets it */
|
|
|
|
pset.popt.topt.env_columns = getenv("COLUMNS") ? atoi(getenv("COLUMNS")) : 0;
|
|
|
|
|
|
|
|
pset.notty = (!isatty(fileno(stdin)) || !isatty(fileno(stdout)));
|
|
|
|
|
|
|
|
pset.getPassword = TRI_DEFAULT;
|
|
|
|
|
|
|
|
EstablishVariableSpace();
|
|
|
|
|
|
|
|
SetVariable(pset.vars, "VERSION", PG_VERSION_STR);
|
|
|
|
|
|
|
|
/* Default values for variables */
|
|
|
|
SetVariableBool(pset.vars, "AUTOCOMMIT");
|
|
|
|
SetVariable(pset.vars, "VERBOSITY", "default");
|
|
|
|
SetVariable(pset.vars, "PROMPT1", DEFAULT_PROMPT1);
|
|
|
|
SetVariable(pset.vars, "PROMPT2", DEFAULT_PROMPT2);
|
|
|
|
SetVariable(pset.vars, "PROMPT3", DEFAULT_PROMPT3);
|
|
|
|
|
|
|
|
parse_psql_options(argc, argv, &options);
|
|
|
|
|
|
|
|
/*
|
|
|
|
* If no action was specified and we're in non-interactive mode, treat it
|
|
|
|
* as if the user had specified "-f -". This lets single-transaction mode
|
|
|
|
* work in this case.
|
|
|
|
*/
|
|
|
|
if (options.action == ACT_NOTHING && pset.notty)
|
|
|
|
{
|
|
|
|
options.action = ACT_FILE;
|
|
|
|
options.action_string = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Bail out if -1 was specified but will be ignored. */
|
|
|
|
if (options.single_txn && options.action != ACT_FILE && options.action == ACT_NOTHING)
|
|
|
|
{
|
|
|
|
fprintf(stderr, _("%s: -1 can only be used in non-interactive mode\n"), pset.progname);
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!pset.popt.topt.fieldSep.separator &&
|
|
|
|
!pset.popt.topt.fieldSep.separator_zero)
|
|
|
|
{
|
|
|
|
pset.popt.topt.fieldSep.separator = pg_strdup(DEFAULT_FIELD_SEP);
|
|
|
|
pset.popt.topt.fieldSep.separator_zero = false;
|
|
|
|
}
|
|
|
|
if (!pset.popt.topt.recordSep.separator &&
|
|
|
|
!pset.popt.topt.recordSep.separator_zero)
|
|
|
|
{
|
|
|
|
pset.popt.topt.recordSep.separator = pg_strdup(DEFAULT_RECORD_SEP);
|
|
|
|
pset.popt.topt.recordSep.separator_zero = false;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (options.username == NULL)
|
|
|
|
password_prompt = pg_strdup(_("Password: "));
|
|
|
|
else
|
|
|
|
password_prompt = psprintf(_("Password for user %s: "),
|
|
|
|
options.username);
|
|
|
|
|
|
|
|
if (pset.getPassword == TRI_YES)
|
|
|
|
password = simple_prompt(password_prompt, 100, false);
|
|
|
|
|
|
|
|
/* loop until we have a password if requested by backend */
|
|
|
|
do
|
|
|
|
{
|
|
|
|
#define PARAMS_ARRAY_SIZE 8
|
|
|
|
const char **keywords = pg_malloc(PARAMS_ARRAY_SIZE * sizeof(*keywords));
|
|
|
|
const char **values = pg_malloc(PARAMS_ARRAY_SIZE * sizeof(*values));
|
|
|
|
|
|
|
|
keywords[0] = "host";
|
|
|
|
values[0] = options.host;
|
|
|
|
keywords[1] = "port";
|
|
|
|
values[1] = options.port;
|
|
|
|
keywords[2] = "user";
|
|
|
|
values[2] = options.username;
|
|
|
|
keywords[3] = "password";
|
|
|
|
values[3] = password;
|
|
|
|
keywords[4] = "dbname";
|
|
|
|
values[4] = (options.action == ACT_LIST_DB &&
|
|
|
|
options.dbname == NULL) ?
|
|
|
|
"postgres" : options.dbname;
|
|
|
|
keywords[5] = "fallback_application_name";
|
|
|
|
values[5] = pset.progname;
|
|
|
|
keywords[6] = "client_encoding";
|
|
|
|
values[6] = (pset.notty || getenv("PGCLIENTENCODING")) ? NULL : "auto";
|
|
|
|
keywords[7] = NULL;
|
|
|
|
values[7] = NULL;
|
|
|
|
|
|
|
|
new_pass = false;
|
|
|
|
pset.db = PQconnectdbParams(keywords, values, true);
|
|
|
|
free(keywords);
|
|
|
|
free(values);
|
|
|
|
|
|
|
|
if (PQstatus(pset.db) == CONNECTION_BAD &&
|
|
|
|
PQconnectionNeedsPassword(pset.db) &&
|
|
|
|
password == NULL &&
|
|
|
|
pset.getPassword != TRI_NO)
|
|
|
|
{
|
|
|
|
PQfinish(pset.db);
|
|
|
|
password = simple_prompt(password_prompt, 100, false);
|
|
|
|
new_pass = true;
|
|
|
|
}
|
|
|
|
} while (new_pass);
|
|
|
|
|
|
|
|
free(password);
|
|
|
|
free(password_prompt);
|
|
|
|
|
|
|
|
if (PQstatus(pset.db) == CONNECTION_BAD)
|
|
|
|
{
|
|
|
|
fprintf(stderr, "%s: %s", pset.progname, PQerrorMessage(pset.db));
|
|
|
|
PQfinish(pset.db);
|
|
|
|
exit(EXIT_BADCONN);
|
|
|
|
}
|
|
|
|
|
|
|
|
setup_cancel_handler();
|
|
|
|
|
|
|
|
PQsetNoticeProcessor(pset.db, NoticeProcessor, NULL);
|
|
|
|
|
|
|
|
SyncVariables();
|
|
|
|
|
|
|
|
if (options.action == ACT_LIST_DB)
|
|
|
|
{
|
|
|
|
int success;
|
|
|
|
|
|
|
|
if (!options.no_psqlrc)
|
|
|
|
process_psqlrc(argv[0]);
|
|
|
|
|
|
|
|
success = listAllDbs(NULL, false);
|
|
|
|
PQfinish(pset.db);
|
|
|
|
exit(success ? EXIT_SUCCESS : EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (options.logfilename)
|
|
|
|
{
|
|
|
|
pset.logfile = fopen(options.logfilename, "a");
|
|
|
|
if (!pset.logfile)
|
|
|
|
fprintf(stderr, _("%s: could not open log file \"%s\": %s\n"),
|
|
|
|
pset.progname, options.logfilename, strerror(errno));
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Now find something to do
|
|
|
|
*/
|
|
|
|
|
|
|
|
/*
|
|
|
|
* process file given by -f
|
|
|
|
*/
|
|
|
|
if (options.action == ACT_FILE)
|
|
|
|
{
|
|
|
|
if (!options.no_psqlrc)
|
|
|
|
process_psqlrc(argv[0]);
|
|
|
|
|
|
|
|
successResult = process_file(options.action_string, options.single_txn, false);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* process slash command if one was given to -c
|
|
|
|
*/
|
|
|
|
else if (options.action == ACT_SINGLE_SLASH)
|
|
|
|
{
|
|
|
|
PsqlScanState scan_state;
|
|
|
|
|
|
|
|
if (pset.echo == PSQL_ECHO_ALL)
|
|
|
|
puts(options.action_string);
|
|
|
|
|
|
|
|
scan_state = psql_scan_create();
|
|
|
|
psql_scan_setup(scan_state,
|
|
|
|
options.action_string,
|
|
|
|
strlen(options.action_string));
|
|
|
|
|
|
|
|
successResult = HandleSlashCmds(scan_state, NULL) != PSQL_CMD_ERROR
|
|
|
|
? EXIT_SUCCESS : EXIT_FAILURE;
|
|
|
|
|
|
|
|
psql_scan_destroy(scan_state);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* If the query given to -c was a normal one, send it
|
|
|
|
*/
|
|
|
|
else if (options.action == ACT_SINGLE_QUERY)
|
|
|
|
{
|
|
|
|
if (pset.echo == PSQL_ECHO_ALL)
|
|
|
|
puts(options.action_string);
|
|
|
|
|
|
|
|
successResult = SendQuery(options.action_string)
|
|
|
|
? EXIT_SUCCESS : EXIT_FAILURE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* or otherwise enter interactive main loop
|
|
|
|
*/
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if (!options.no_psqlrc)
|
|
|
|
process_psqlrc(argv[0]);
|
|
|
|
|
|
|
|
connection_warnings(true);
|
|
|
|
if (!pset.quiet)
|
|
|
|
printf(_("Type \"help\" for help.\n\n"));
|
|
|
|
initializeInput(options.no_readline ? 0 : 1);
|
|
|
|
successResult = MainLoop(stdin);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* clean up */
|
|
|
|
if (pset.logfile)
|
|
|
|
fclose(pset.logfile);
|
|
|
|
PQfinish(pset.db);
|
|
|
|
setQFout(NULL);
|
|
|
|
|
|
|
|
return successResult;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Parse command line options
|
|
|
|
*/
|
|
|
|
|
|
|
|
static void
|
|
|
|
parse_psql_options(int argc, char *argv[], struct adhoc_opts * options)
|
|
|
|
{
|
|
|
|
static struct option long_options[] =
|
|
|
|
{
|
|
|
|
{"echo-all", no_argument, NULL, 'a'},
|
|
|
|
{"no-align", no_argument, NULL, 'A'},
|
|
|
|
{"command", required_argument, NULL, 'c'},
|
|
|
|
{"dbname", required_argument, NULL, 'd'},
|
|
|
|
{"echo-queries", no_argument, NULL, 'e'},
|
|
|
|
{"echo-errors", no_argument, NULL, 'b'},
|
|
|
|
{"echo-hidden", no_argument, NULL, 'E'},
|
|
|
|
{"file", required_argument, NULL, 'f'},
|
|
|
|
{"field-separator", required_argument, NULL, 'F'},
|
|
|
|
{"field-separator-zero", no_argument, NULL, 'z'},
|
|
|
|
{"host", required_argument, NULL, 'h'},
|
|
|
|
{"html", no_argument, NULL, 'H'},
|
|
|
|
{"list", no_argument, NULL, 'l'},
|
|
|
|
{"log-file", required_argument, NULL, 'L'},
|
|
|
|
{"no-readline", no_argument, NULL, 'n'},
|
|
|
|
{"single-transaction", no_argument, NULL, '1'},
|
|
|
|
{"output", required_argument, NULL, 'o'},
|
|
|
|
{"port", required_argument, NULL, 'p'},
|
|
|
|
{"pset", required_argument, NULL, 'P'},
|
|
|
|
{"quiet", no_argument, NULL, 'q'},
|
|
|
|
{"record-separator", required_argument, NULL, 'R'},
|
|
|
|
{"record-separator-zero", no_argument, NULL, '0'},
|
|
|
|
{"single-step", no_argument, NULL, 's'},
|
|
|
|
{"single-line", no_argument, NULL, 'S'},
|
|
|
|
{"tuples-only", no_argument, NULL, 't'},
|
|
|
|
{"table-attr", required_argument, NULL, 'T'},
|
|
|
|
{"username", required_argument, NULL, 'U'},
|
|
|
|
{"set", required_argument, NULL, 'v'},
|
|
|
|
{"variable", required_argument, NULL, 'v'},
|
|
|
|
{"version", no_argument, NULL, 'V'},
|
|
|
|
{"no-password", no_argument, NULL, 'w'},
|
|
|
|
{"password", no_argument, NULL, 'W'},
|
|
|
|
{"expanded", no_argument, NULL, 'x'},
|
|
|
|
{"no-psqlrc", no_argument, NULL, 'X'},
|
Add new psql help topics, accessible to both --help and \?.
Add --help=<topic> for the commandline, and \? <topic> as a backslash
command, to show more help than the invocations without parameters
do. "commands", "variables" and "options" currently exist as help
topics describing, respectively, backslash commands, psql variables,
and commandline switches. Without parameters the help commands show
their previous topic.
Some further wordsmithing or extending of the added help content might
be needed; but there seems little benefit delaying the overall feature
further.
Author: Pavel Stehule, editorialized by many
Reviewed-By: Andres Freund, Petr Jelinek, Fujii Masao, MauMau, Abhijit
Menon-Sen and Erik Rijkers.
Discussion: CAFj8pRDVGuC-nXBfe2CK8vpyzd2Dsr9GVpbrATAnZO=2YQ0s2Q@mail.gmail.com,
CAFj8pRA54AbTv2RXDTRxiAd8hy8wxmoVLqhJDRCwEnhdd7OUkw@mail.gmail.com
11 years ago
|
|
|
{"help", optional_argument, NULL, 1},
|
|
|
|
{NULL, 0, NULL, 0}
|
|
|
|
};
|
|
|
|
|
|
|
|
int optindex;
|
|
|
|
int c;
|
|
|
|
|
* Includes tab completion. It's not magic, but it's very cool. At any
rate
it's better than what used to be there.
* Does proper SQL "host variable" substitution as pointed out by Andreas
Zeugwetter (thanks): select * from :foo; Also some changes in how ':'
and ';' are treated (escape with \ to send to backend). This does
_not_
affect the '::' cast operator, but perhaps others that contain : or ;
(but there are none right now).
* To show description with a <something> listing, append '?' to command
name, e.g., \df?. This seemed to be the convenient and logical
solution.
Or append a '+' to see more useless information, e.g., \df+.
* Fixed fflush()'ing bug pointed out by Jan during the regression test
discussion.
* Added LastOid variable. This ought to take care of TODO item "Add a
function to return the last inserted oid, for use in psql scripts"
(under CLIENTS)
E.g.,
insert into foo values(...);
insert into bar values(..., :LastOid);
\echo $LastOid
* \d command shows constraints, rules, and triggers defined on the table
(in addition to indices)
* Various fixes, optimizations, corrections
* Documentation update as well
Note: This now requires snprintf(), which, if necessary, is taken from
src/backend/port. This is certainly a little weird, but it should
suffice
until a source tree cleanup is done.
Enjoy.
--
Peter Eisentraut Sernanders väg 10:115
26 years ago
|
|
|
memset(options, 0, sizeof *options);
|
|
|
|
|
|
|
|
while ((c = getopt_long(argc, argv, "aAbc:d:eEf:F:h:HlL:no:p:P:qR:sStT:U:v:VwWxXz?01",
|
|
|
|
long_options, &optindex)) != -1)
|
|
|
|
{
|
|
|
|
switch (c)
|
|
|
|
{
|
|
|
|
case 'a':
|
|
|
|
SetVariable(pset.vars, "ECHO", "all");
|
|
|
|
break;
|
|
|
|
case 'A':
|
|
|
|
pset.popt.topt.format = PRINT_UNALIGNED;
|
|
|
|
break;
|
|
|
|
case 'b':
|
|
|
|
SetVariable(pset.vars, "ECHO", "errors");
|
|
|
|
break;
|
|
|
|
case 'c':
|
|
|
|
options->action_string = pg_strdup(optarg);
|
|
|
|
if (optarg[0] == '\\')
|
|
|
|
{
|
|
|
|
options->action = ACT_SINGLE_SLASH;
|
|
|
|
options->action_string++;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
options->action = ACT_SINGLE_QUERY;
|
|
|
|
break;
|
|
|
|
case 'd':
|
|
|
|
options->dbname = pg_strdup(optarg);
|
|
|
|
break;
|
|
|
|
case 'e':
|
|
|
|
SetVariable(pset.vars, "ECHO", "queries");
|
|
|
|
break;
|
|
|
|
case 'E':
|
|
|
|
SetVariableBool(pset.vars, "ECHO_HIDDEN");
|
|
|
|
break;
|
|
|
|
case 'f':
|
|
|
|
options->action = ACT_FILE;
|
|
|
|
options->action_string = pg_strdup(optarg);
|
|
|
|
break;
|
|
|
|
case 'F':
|
|
|
|
pset.popt.topt.fieldSep.separator = pg_strdup(optarg);
|
|
|
|
pset.popt.topt.fieldSep.separator_zero = false;
|
|
|
|
break;
|
|
|
|
case 'h':
|
|
|
|
options->host = pg_strdup(optarg);
|
|
|
|
break;
|
|
|
|
case 'H':
|
|
|
|
pset.popt.topt.format = PRINT_HTML;
|
|
|
|
break;
|
|
|
|
case 'l':
|
|
|
|
options->action = ACT_LIST_DB;
|
|
|
|
break;
|
|
|
|
case 'L':
|
|
|
|
options->logfilename = pg_strdup(optarg);
|
|
|
|
break;
|
|
|
|
case 'n':
|
|
|
|
options->no_readline = true;
|
|
|
|
break;
|
|
|
|
case 'o':
|
|
|
|
setQFout(optarg);
|
|
|
|
break;
|
|
|
|
case 'p':
|
|
|
|
options->port = pg_strdup(optarg);
|
|
|
|
break;
|
|
|
|
case 'P':
|
|
|
|
{
|
|
|
|
char *value;
|
|
|
|
char *equal_loc;
|
|
|
|
bool result;
|
|
|
|
|
|
|
|
value = pg_strdup(optarg);
|
|
|
|
equal_loc = strchr(value, '=');
|
|
|
|
if (!equal_loc)
|
|
|
|
result = do_pset(value, NULL, &pset.popt, true);
|
|
|
|
else
|
|
|
|
{
|
|
|
|
*equal_loc = '\0';
|
|
|
|
result = do_pset(value, equal_loc + 1, &pset.popt, true);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!result)
|
|
|
|
{
|
|
|
|
fprintf(stderr, _("%s: could not set printing parameter \"%s\"\n"), pset.progname, value);
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
free(value);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case 'q':
|
|
|
|
SetVariableBool(pset.vars, "QUIET");
|
|
|
|
break;
|
|
|
|
case 'R':
|
|
|
|
pset.popt.topt.recordSep.separator = pg_strdup(optarg);
|
|
|
|
pset.popt.topt.recordSep.separator_zero = false;
|
|
|
|
break;
|
|
|
|
case 's':
|
|
|
|
SetVariableBool(pset.vars, "SINGLESTEP");
|
|
|
|
break;
|
|
|
|
case 'S':
|
|
|
|
SetVariableBool(pset.vars, "SINGLELINE");
|
|
|
|
break;
|
|
|
|
case 't':
|
|
|
|
pset.popt.topt.tuples_only = true;
|
|
|
|
break;
|
|
|
|
case 'T':
|
|
|
|
pset.popt.topt.tableAttr = pg_strdup(optarg);
|
|
|
|
break;
|
|
|
|
case 'U':
|
|
|
|
options->username = pg_strdup(optarg);
|
|
|
|
break;
|
|
|
|
case 'v':
|
|
|
|
{
|
|
|
|
char *value;
|
|
|
|
char *equal_loc;
|
|
|
|
|
|
|
|
value = pg_strdup(optarg);
|
|
|
|
equal_loc = strchr(value, '=');
|
|
|
|
if (!equal_loc)
|
|
|
|
{
|
|
|
|
if (!DeleteVariable(pset.vars, value))
|
|
|
|
{
|
|
|
|
fprintf(stderr, _("%s: could not delete variable \"%s\"\n"),
|
|
|
|
pset.progname, value);
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
*equal_loc = '\0';
|
|
|
|
if (!SetVariable(pset.vars, value, equal_loc + 1))
|
|
|
|
{
|
|
|
|
fprintf(stderr, _("%s: could not set variable \"%s\"\n"),
|
|
|
|
pset.progname, value);
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
free(value);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case 'V':
|
|
|
|
showVersion();
|
|
|
|
exit(EXIT_SUCCESS);
|
|
|
|
case 'w':
|
|
|
|
pset.getPassword = TRI_NO;
|
|
|
|
break;
|
|
|
|
case 'W':
|
|
|
|
pset.getPassword = TRI_YES;
|
|
|
|
break;
|
|
|
|
case 'x':
|
|
|
|
pset.popt.topt.expanded = true;
|
|
|
|
break;
|
|
|
|
case 'X':
|
|
|
|
options->no_psqlrc = true;
|
|
|
|
break;
|
|
|
|
case 'z':
|
|
|
|
pset.popt.topt.fieldSep.separator_zero = true;
|
|
|
|
break;
|
|
|
|
case '0':
|
|
|
|
pset.popt.topt.recordSep.separator_zero = true;
|
|
|
|
break;
|
|
|
|
case '1':
|
|
|
|
options->single_txn = true;
|
|
|
|
break;
|
|
|
|
case '?':
|
|
|
|
/* Actual help option given */
|
Add new psql help topics, accessible to both --help and \?.
Add --help=<topic> for the commandline, and \? <topic> as a backslash
command, to show more help than the invocations without parameters
do. "commands", "variables" and "options" currently exist as help
topics describing, respectively, backslash commands, psql variables,
and commandline switches. Without parameters the help commands show
their previous topic.
Some further wordsmithing or extending of the added help content might
be needed; but there seems little benefit delaying the overall feature
further.
Author: Pavel Stehule, editorialized by many
Reviewed-By: Andres Freund, Petr Jelinek, Fujii Masao, MauMau, Abhijit
Menon-Sen and Erik Rijkers.
Discussion: CAFj8pRDVGuC-nXBfe2CK8vpyzd2Dsr9GVpbrATAnZO=2YQ0s2Q@mail.gmail.com,
CAFj8pRA54AbTv2RXDTRxiAd8hy8wxmoVLqhJDRCwEnhdd7OUkw@mail.gmail.com
11 years ago
|
|
|
if (strcmp(argv[optind - 1], "-?") == 0)
|
|
|
|
{
|
Add new psql help topics, accessible to both --help and \?.
Add --help=<topic> for the commandline, and \? <topic> as a backslash
command, to show more help than the invocations without parameters
do. "commands", "variables" and "options" currently exist as help
topics describing, respectively, backslash commands, psql variables,
and commandline switches. Without parameters the help commands show
their previous topic.
Some further wordsmithing or extending of the added help content might
be needed; but there seems little benefit delaying the overall feature
further.
Author: Pavel Stehule, editorialized by many
Reviewed-By: Andres Freund, Petr Jelinek, Fujii Masao, MauMau, Abhijit
Menon-Sen and Erik Rijkers.
Discussion: CAFj8pRDVGuC-nXBfe2CK8vpyzd2Dsr9GVpbrATAnZO=2YQ0s2Q@mail.gmail.com,
CAFj8pRA54AbTv2RXDTRxiAd8hy8wxmoVLqhJDRCwEnhdd7OUkw@mail.gmail.com
11 years ago
|
|
|
usage(NOPAGER);
|
|
|
|
exit(EXIT_SUCCESS);
|
|
|
|
}
|
|
|
|
/* unknown option reported by getopt */
|
|
|
|
else
|
Add new psql help topics, accessible to both --help and \?.
Add --help=<topic> for the commandline, and \? <topic> as a backslash
command, to show more help than the invocations without parameters
do. "commands", "variables" and "options" currently exist as help
topics describing, respectively, backslash commands, psql variables,
and commandline switches. Without parameters the help commands show
their previous topic.
Some further wordsmithing or extending of the added help content might
be needed; but there seems little benefit delaying the overall feature
further.
Author: Pavel Stehule, editorialized by many
Reviewed-By: Andres Freund, Petr Jelinek, Fujii Masao, MauMau, Abhijit
Menon-Sen and Erik Rijkers.
Discussion: CAFj8pRDVGuC-nXBfe2CK8vpyzd2Dsr9GVpbrATAnZO=2YQ0s2Q@mail.gmail.com,
CAFj8pRA54AbTv2RXDTRxiAd8hy8wxmoVLqhJDRCwEnhdd7OUkw@mail.gmail.com
11 years ago
|
|
|
goto unknown_option;
|
|
|
|
break;
|
|
|
|
case 1:
|
|
|
|
{
|
Add new psql help topics, accessible to both --help and \?.
Add --help=<topic> for the commandline, and \? <topic> as a backslash
command, to show more help than the invocations without parameters
do. "commands", "variables" and "options" currently exist as help
topics describing, respectively, backslash commands, psql variables,
and commandline switches. Without parameters the help commands show
their previous topic.
Some further wordsmithing or extending of the added help content might
be needed; but there seems little benefit delaying the overall feature
further.
Author: Pavel Stehule, editorialized by many
Reviewed-By: Andres Freund, Petr Jelinek, Fujii Masao, MauMau, Abhijit
Menon-Sen and Erik Rijkers.
Discussion: CAFj8pRDVGuC-nXBfe2CK8vpyzd2Dsr9GVpbrATAnZO=2YQ0s2Q@mail.gmail.com,
CAFj8pRA54AbTv2RXDTRxiAd8hy8wxmoVLqhJDRCwEnhdd7OUkw@mail.gmail.com
11 years ago
|
|
|
if (!optarg || strcmp(optarg, "options") == 0)
|
|
|
|
usage(NOPAGER);
|
|
|
|
else if (optarg && strcmp(optarg, "commands") == 0)
|
|
|
|
slashUsage(NOPAGER);
|
|
|
|
else if (optarg && strcmp(optarg, "variables") == 0)
|
|
|
|
helpVariables(NOPAGER);
|
|
|
|
else
|
|
|
|
goto unknown_option;
|
|
|
|
|
|
|
|
exit(EXIT_SUCCESS);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
Add new psql help topics, accessible to both --help and \?.
Add --help=<topic> for the commandline, and \? <topic> as a backslash
command, to show more help than the invocations without parameters
do. "commands", "variables" and "options" currently exist as help
topics describing, respectively, backslash commands, psql variables,
and commandline switches. Without parameters the help commands show
their previous topic.
Some further wordsmithing or extending of the added help content might
be needed; but there seems little benefit delaying the overall feature
further.
Author: Pavel Stehule, editorialized by many
Reviewed-By: Andres Freund, Petr Jelinek, Fujii Masao, MauMau, Abhijit
Menon-Sen and Erik Rijkers.
Discussion: CAFj8pRDVGuC-nXBfe2CK8vpyzd2Dsr9GVpbrATAnZO=2YQ0s2Q@mail.gmail.com,
CAFj8pRA54AbTv2RXDTRxiAd8hy8wxmoVLqhJDRCwEnhdd7OUkw@mail.gmail.com
11 years ago
|
|
|
unknown_option:
|
|
|
|
fprintf(stderr, _("Try \"%s --help\" for more information.\n"),
|
|
|
|
pset.progname);
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* if we still have arguments, use it as the database name and username
|
|
|
|
*/
|
|
|
|
while (argc - optind >= 1)
|
|
|
|
{
|
|
|
|
if (!options->dbname)
|
|
|
|
options->dbname = argv[optind];
|
|
|
|
else if (!options->username)
|
|
|
|
options->username = argv[optind];
|
|
|
|
else if (!pset.quiet)
|
|
|
|
fprintf(stderr, _("%s: warning: extra command-line argument \"%s\" ignored\n"),
|
|
|
|
pset.progname, argv[optind]);
|
|
|
|
|
|
|
|
optind++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Load .psqlrc file, if found.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
process_psqlrc(char *argv0)
|
|
|
|
{
|
|
|
|
char home[MAXPGPATH];
|
|
|
|
char rc_file[MAXPGPATH];
|
|
|
|
char my_exec_path[MAXPGPATH];
|
|
|
|
char etc_path[MAXPGPATH];
|
|
|
|
char *envrc = getenv("PSQLRC");
|
|
|
|
|
|
|
|
if (find_my_exec(argv0, my_exec_path) < 0)
|
|
|
|
{
|
|
|
|
fprintf(stderr, _("%s: could not find own program executable\n"), argv0);
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
get_etc_path(my_exec_path, etc_path);
|
|
|
|
|
|
|
|
snprintf(rc_file, MAXPGPATH, "%s/%s", etc_path, SYSPSQLRC);
|
|
|
|
process_psqlrc_file(rc_file);
|
|
|
|
|
|
|
|
if (envrc != NULL && strlen(envrc) > 0)
|
|
|
|
{
|
|
|
|
/* might need to free() this */
|
|
|
|
char *envrc_alloc = pstrdup(envrc);
|
|
|
|
|
|
|
|
expand_tilde(&envrc_alloc);
|
|
|
|
process_psqlrc_file(envrc_alloc);
|
|
|
|
}
|
|
|
|
else if (get_home_path(home))
|
|
|
|
{
|
|
|
|
snprintf(rc_file, MAXPGPATH, "%s/%s", home, PSQLRC);
|
|
|
|
process_psqlrc_file(rc_file);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static void
|
|
|
|
process_psqlrc_file(char *filename)
|
|
|
|
{
|
|
|
|
char *psqlrc_minor,
|
|
|
|
*psqlrc_major;
|
|
|
|
|
|
|
|
#if defined(WIN32) && (!defined(__MINGW32__))
|
|
|
|
#define R_OK 4
|
|
|
|
#endif
|
|
|
|
|
|
|
|
psqlrc_minor = psprintf("%s-%s", filename, PG_VERSION);
|
|
|
|
psqlrc_major = psprintf("%s-%s", filename, PG_MAJORVERSION);
|
|
|
|
|
|
|
|
/* check for minor version first, then major, then no version */
|
|
|
|
if (access(psqlrc_minor, R_OK) == 0)
|
|
|
|
(void) process_file(psqlrc_minor, false, false);
|
|
|
|
else if (access(psqlrc_major, R_OK) == 0)
|
|
|
|
(void) process_file(psqlrc_major, false, false);
|
|
|
|
else if (access(filename, R_OK) == 0)
|
|
|
|
(void) process_file(filename, false, false);
|
|
|
|
|
|
|
|
free(psqlrc_minor);
|
|
|
|
free(psqlrc_major);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* showVersion
|
|
|
|
*
|
|
|
|
* This output format is intended to match GNU standards.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
showVersion(void)
|
|
|
|
{
|
|
|
|
puts("psql (PostgreSQL) " PG_VERSION);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Assign hooks for psql variables.
|
|
|
|
*
|
|
|
|
* This isn't an amazingly good place for them, but neither is anywhere else.
|
|
|
|
*/
|
|
|
|
|
|
|
|
static void
|
|
|
|
autocommit_hook(const char *newval)
|
|
|
|
{
|
|
|
|
pset.autocommit = ParseVariableBool(newval);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
on_error_stop_hook(const char *newval)
|
|
|
|
{
|
|
|
|
pset.on_error_stop = ParseVariableBool(newval);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
quiet_hook(const char *newval)
|
|
|
|
{
|
|
|
|
pset.quiet = ParseVariableBool(newval);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
singleline_hook(const char *newval)
|
|
|
|
{
|
|
|
|
pset.singleline = ParseVariableBool(newval);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
singlestep_hook(const char *newval)
|
|
|
|
{
|
|
|
|
pset.singlestep = ParseVariableBool(newval);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
fetch_count_hook(const char *newval)
|
|
|
|
{
|
|
|
|
pset.fetch_count = ParseVariableNum(newval, -1, -1, false);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
echo_hook(const char *newval)
|
|
|
|
{
|
|
|
|
if (newval == NULL)
|
|
|
|
pset.echo = PSQL_ECHO_NONE;
|
|
|
|
else if (strcmp(newval, "queries") == 0)
|
|
|
|
pset.echo = PSQL_ECHO_QUERIES;
|
|
|
|
else if (strcmp(newval, "errors") == 0)
|
|
|
|
pset.echo = PSQL_ECHO_ERRORS;
|
|
|
|
else if (strcmp(newval, "all") == 0)
|
|
|
|
pset.echo = PSQL_ECHO_ALL;
|
|
|
|
else
|
|
|
|
pset.echo = PSQL_ECHO_NONE;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
echo_hidden_hook(const char *newval)
|
|
|
|
{
|
|
|
|
if (newval == NULL)
|
|
|
|
pset.echo_hidden = PSQL_ECHO_HIDDEN_OFF;
|
|
|
|
else if (strcmp(newval, "noexec") == 0)
|
|
|
|
pset.echo_hidden = PSQL_ECHO_HIDDEN_NOEXEC;
|
|
|
|
else if (pg_strcasecmp(newval, "off") == 0)
|
|
|
|
pset.echo_hidden = PSQL_ECHO_HIDDEN_OFF;
|
|
|
|
else
|
|
|
|
pset.echo_hidden = PSQL_ECHO_HIDDEN_ON;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
on_error_rollback_hook(const char *newval)
|
|
|
|
{
|
|
|
|
if (newval == NULL)
|
|
|
|
pset.on_error_rollback = PSQL_ERROR_ROLLBACK_OFF;
|
|
|
|
else if (pg_strcasecmp(newval, "interactive") == 0)
|
|
|
|
pset.on_error_rollback = PSQL_ERROR_ROLLBACK_INTERACTIVE;
|
|
|
|
else if (pg_strcasecmp(newval, "off") == 0)
|
|
|
|
pset.on_error_rollback = PSQL_ERROR_ROLLBACK_OFF;
|
|
|
|
else
|
|
|
|
pset.on_error_rollback = PSQL_ERROR_ROLLBACK_ON;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
histcontrol_hook(const char *newval)
|
|
|
|
{
|
|
|
|
if (newval == NULL)
|
|
|
|
pset.histcontrol = hctl_none;
|
|
|
|
else if (strcmp(newval, "ignorespace") == 0)
|
|
|
|
pset.histcontrol = hctl_ignorespace;
|
|
|
|
else if (strcmp(newval, "ignoredups") == 0)
|
|
|
|
pset.histcontrol = hctl_ignoredups;
|
|
|
|
else if (strcmp(newval, "ignoreboth") == 0)
|
|
|
|
pset.histcontrol = hctl_ignoreboth;
|
|
|
|
else
|
|
|
|
pset.histcontrol = hctl_none;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
prompt1_hook(const char *newval)
|
|
|
|
{
|
|
|
|
pset.prompt1 = newval ? newval : "";
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
prompt2_hook(const char *newval)
|
|
|
|
{
|
|
|
|
pset.prompt2 = newval ? newval : "";
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
prompt3_hook(const char *newval)
|
|
|
|
{
|
|
|
|
pset.prompt3 = newval ? newval : "";
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
verbosity_hook(const char *newval)
|
|
|
|
{
|
|
|
|
if (newval == NULL)
|
|
|
|
pset.verbosity = PQERRORS_DEFAULT;
|
|
|
|
else if (strcmp(newval, "default") == 0)
|
|
|
|
pset.verbosity = PQERRORS_DEFAULT;
|
|
|
|
else if (strcmp(newval, "terse") == 0)
|
|
|
|
pset.verbosity = PQERRORS_TERSE;
|
|
|
|
else if (strcmp(newval, "verbose") == 0)
|
|
|
|
pset.verbosity = PQERRORS_VERBOSE;
|
|
|
|
else
|
|
|
|
pset.verbosity = PQERRORS_DEFAULT;
|
|
|
|
|
|
|
|
if (pset.db)
|
|
|
|
PQsetErrorVerbosity(pset.db, pset.verbosity);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static void
|
|
|
|
EstablishVariableSpace(void)
|
|
|
|
{
|
|
|
|
pset.vars = CreateVariableSpace();
|
|
|
|
|
|
|
|
SetVariableAssignHook(pset.vars, "AUTOCOMMIT", autocommit_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "ON_ERROR_STOP", on_error_stop_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "QUIET", quiet_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "SINGLELINE", singleline_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "SINGLESTEP", singlestep_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "FETCH_COUNT", fetch_count_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "ECHO", echo_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "ECHO_HIDDEN", echo_hidden_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "ON_ERROR_ROLLBACK", on_error_rollback_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "HISTCONTROL", histcontrol_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "PROMPT1", prompt1_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "PROMPT2", prompt2_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "PROMPT3", prompt3_hook);
|
|
|
|
SetVariableAssignHook(pset.vars, "VERBOSITY", verbosity_hook);
|
|
|
|
}
|