PostgreSQL Source Code  git master
command.h File Reference
Include dependency graph for command.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Typedefs

typedef enum _backslashResult backslashResult
 

Enumerations

enum  _backslashResult {
  PSQL_CMD_UNKNOWN = 0 , PSQL_CMD_SEND , PSQL_CMD_SKIP_LINE , PSQL_CMD_TERMINATE ,
  PSQL_CMD_NEWEDIT , PSQL_CMD_ERROR
}
 

Functions

backslashResult HandleSlashCmds (PsqlScanState scan_state, ConditionalStack cstack, PQExpBuffer query_buf, PQExpBuffer previous_buf)
 
int process_file (char *filename, bool use_relative_path)
 
bool do_pset (const char *param, const char *value, printQueryOpt *popt, bool quiet)
 
printQueryOptsavePsetInfo (const printQueryOpt *popt)
 
void restorePsetInfo (printQueryOpt *popt, printQueryOpt *save)
 
void connection_warnings (bool in_startup)
 
void SyncVariables (void)
 
void UnsyncVariables (void)
 

Typedef Documentation

◆ backslashResult

Enumeration Type Documentation

◆ _backslashResult

Enumerator
PSQL_CMD_UNKNOWN 
PSQL_CMD_SEND 
PSQL_CMD_SKIP_LINE 
PSQL_CMD_TERMINATE 
PSQL_CMD_NEWEDIT 
PSQL_CMD_ERROR 

Definition at line 15 of file command.h.

16 {
17  PSQL_CMD_UNKNOWN = 0, /* not done parsing yet (internal only) */
18  PSQL_CMD_SEND, /* query complete; send off */
19  PSQL_CMD_SKIP_LINE, /* keep building query */
20  PSQL_CMD_TERMINATE, /* quit program */
21  PSQL_CMD_NEWEDIT, /* query buffer was changed (e.g., via \e) */
22  PSQL_CMD_ERROR /* the execution of the backslash command
23  * resulted in an error */
@ PSQL_CMD_TERMINATE
Definition: command.h:20
@ PSQL_CMD_UNKNOWN
Definition: command.h:17
@ PSQL_CMD_NEWEDIT
Definition: command.h:21
@ PSQL_CMD_ERROR
Definition: command.h:22
@ PSQL_CMD_SEND
Definition: command.h:18
@ PSQL_CMD_SKIP_LINE
Definition: command.h:19
enum _backslashResult backslashResult

Function Documentation

◆ connection_warnings()

void connection_warnings ( bool  in_startup)

Definition at line 3638 of file command.c.

3639 {
3640  if (!pset.quiet && !pset.notty)
3641  {
3642  int client_ver = PG_VERSION_NUM;
3643  char cverbuf[32];
3644  char sverbuf[32];
3645 
3646  if (pset.sversion != client_ver)
3647  {
3648  const char *server_version;
3649 
3650  /* Try to get full text form, might include "devel" etc */
3651  server_version = PQparameterStatus(pset.db, "server_version");
3652  /* Otherwise fall back on pset.sversion */
3653  if (!server_version)
3654  {
3656  sverbuf, sizeof(sverbuf));
3657  server_version = sverbuf;
3658  }
3659 
3660  printf(_("%s (%s, server %s)\n"),
3661  pset.progname, PG_VERSION, server_version);
3662  }
3663  /* For version match, only print psql banner on startup. */
3664  else if (in_startup)
3665  printf("%s (%s)\n", pset.progname, PG_VERSION);
3666 
3667  /*
3668  * Warn if server's major version is newer than ours, or if server
3669  * predates our support cutoff (currently 9.2).
3670  */
3671  if (pset.sversion / 100 > client_ver / 100 ||
3672  pset.sversion < 90200)
3673  printf(_("WARNING: %s major version %s, server major version %s.\n"
3674  " Some psql features might not work.\n"),
3675  pset.progname,
3676  formatPGVersionNumber(client_ver, false,
3677  cverbuf, sizeof(cverbuf)),
3679  sverbuf, sizeof(sverbuf)));
3680 
3681 #ifdef WIN32
3682  if (in_startup)
3683  checkWin32Codepage();
3684 #endif
3685  printSSLInfo();
3686  printGSSInfo();
3687  }
3688 }
static void printSSLInfo(void)
Definition: command.c:3697
static void printGSSInfo(void)
Definition: command.c:3722
#define _(x)
Definition: elog.c:90
const char * PQparameterStatus(const PGconn *conn, const char *paramName)
Definition: fe-connect.c:6708
static int server_version
Definition: pg_dumpall.c:110
#define printf(...)
Definition: port.h:244
PsqlSettings pset
Definition: startup.c:32
char * formatPGVersionNumber(int version_number, bool include_minor, char *buf, size_t buflen)
Definition: string_utils.c:177
PGconn * db
Definition: settings.h:82
const char * progname
Definition: settings.h:112

References _, _psqlSettings::db, formatPGVersionNumber(), _psqlSettings::notty, PQparameterStatus(), printf, printGSSInfo(), printSSLInfo(), _psqlSettings::progname, pset, _psqlSettings::quiet, server_version, and _psqlSettings::sversion.

Referenced by CheckConnection(), do_connect(), and main().

◆ do_pset()

bool do_pset ( const char *  param,
const char *  value,
printQueryOpt popt,
bool  quiet 
)

Definition at line 4254 of file command.c.

4255 {
4256  size_t vallen = 0;
4257 
4258  Assert(param != NULL);
4259 
4260  if (value)
4261  vallen = strlen(value);
4262 
4263  /* set format */
4264  if (strcmp(param, "format") == 0)
4265  {
4266  static const struct fmt
4267  {
4268  const char *name;
4269  enum printFormat number;
4270  } formats[] =
4271  {
4272  /* remember to update error message below when adding more */
4273  {"aligned", PRINT_ALIGNED},
4274  {"asciidoc", PRINT_ASCIIDOC},
4275  {"csv", PRINT_CSV},
4276  {"html", PRINT_HTML},
4277  {"latex", PRINT_LATEX},
4278  {"troff-ms", PRINT_TROFF_MS},
4279  {"unaligned", PRINT_UNALIGNED},
4280  {"wrapped", PRINT_WRAPPED}
4281  };
4282 
4283  if (!value)
4284  ;
4285  else
4286  {
4287  int match_pos = -1;
4288 
4289  for (int i = 0; i < lengthof(formats); i++)
4290  {
4291  if (pg_strncasecmp(formats[i].name, value, vallen) == 0)
4292  {
4293  if (match_pos < 0)
4294  match_pos = i;
4295  else
4296  {
4297  pg_log_error("\\pset: ambiguous abbreviation \"%s\" matches both \"%s\" and \"%s\"",
4298  value,
4299  formats[match_pos].name, formats[i].name);
4300  return false;
4301  }
4302  }
4303  }
4304  if (match_pos >= 0)
4305  popt->topt.format = formats[match_pos].number;
4306  else if (pg_strncasecmp("latex-longtable", value, vallen) == 0)
4307  {
4308  /*
4309  * We must treat latex-longtable specially because latex is a
4310  * prefix of it; if both were in the table above, we'd think
4311  * "latex" is ambiguous.
4312  */
4314  }
4315  else
4316  {
4317  pg_log_error("\\pset: allowed formats are aligned, asciidoc, csv, html, latex, latex-longtable, troff-ms, unaligned, wrapped");
4318  return false;
4319  }
4320  }
4321  }
4322 
4323  /* set table line style */
4324  else if (strcmp(param, "linestyle") == 0)
4325  {
4326  if (!value)
4327  ;
4328  else if (pg_strncasecmp("ascii", value, vallen) == 0)
4329  popt->topt.line_style = &pg_asciiformat;
4330  else if (pg_strncasecmp("old-ascii", value, vallen) == 0)
4332  else if (pg_strncasecmp("unicode", value, vallen) == 0)
4333  popt->topt.line_style = &pg_utf8format;
4334  else
4335  {
4336  pg_log_error("\\pset: allowed line styles are ascii, old-ascii, unicode");
4337  return false;
4338  }
4339  }
4340 
4341  /* set unicode border line style */
4342  else if (strcmp(param, "unicode_border_linestyle") == 0)
4343  {
4344  if (!value)
4345  ;
4346  else if (set_unicode_line_style(value, vallen,
4348  refresh_utf8format(&(popt->topt));
4349  else
4350  {
4351  pg_log_error("\\pset: allowed Unicode border line styles are single, double");
4352  return false;
4353  }
4354  }
4355 
4356  /* set unicode column line style */
4357  else if (strcmp(param, "unicode_column_linestyle") == 0)
4358  {
4359  if (!value)
4360  ;
4361  else if (set_unicode_line_style(value, vallen,
4363  refresh_utf8format(&(popt->topt));
4364  else
4365  {
4366  pg_log_error("\\pset: allowed Unicode column line styles are single, double");
4367  return false;
4368  }
4369  }
4370 
4371  /* set unicode header line style */
4372  else if (strcmp(param, "unicode_header_linestyle") == 0)
4373  {
4374  if (!value)
4375  ;
4376  else if (set_unicode_line_style(value, vallen,
4378  refresh_utf8format(&(popt->topt));
4379  else
4380  {
4381  pg_log_error("\\pset: allowed Unicode header line styles are single, double");
4382  return false;
4383  }
4384  }
4385 
4386  /* set border style/width */
4387  else if (strcmp(param, "border") == 0)
4388  {
4389  if (value)
4390  popt->topt.border = atoi(value);
4391  }
4392 
4393  /* set expanded/vertical mode */
4394  else if (strcmp(param, "x") == 0 ||
4395  strcmp(param, "expanded") == 0 ||
4396  strcmp(param, "vertical") == 0)
4397  {
4398  if (value && pg_strcasecmp(value, "auto") == 0)
4399  popt->topt.expanded = 2;
4400  else if (value)
4401  {
4402  bool on_off;
4403 
4404  if (ParseVariableBool(value, NULL, &on_off))
4405  popt->topt.expanded = on_off ? 1 : 0;
4406  else
4407  {
4408  PsqlVarEnumError(param, value, "on, off, auto");
4409  return false;
4410  }
4411  }
4412  else
4413  popt->topt.expanded = !popt->topt.expanded;
4414  }
4415 
4416  /* header line width in expanded mode */
4417  else if (strcmp(param, "xheader_width") == 0)
4418  {
4419  if (! value)
4420  ;
4421  else if (pg_strcasecmp(value, "full") == 0)
4423  else if (pg_strcasecmp(value, "column") == 0)
4425  else if (pg_strcasecmp(value, "page") == 0)
4427  else
4428  {
4430  popt->topt.expanded_header_exact_width = atoi(value);
4431  if (popt->topt.expanded_header_exact_width == 0)
4432  {
4433  pg_log_error("\\pset: allowed xheader_width values are full (default), column, page, or a number specifying the exact width.");
4434  return false;
4435  }
4436  }
4437  }
4438 
4439  /* field separator for CSV format */
4440  else if (strcmp(param, "csv_fieldsep") == 0)
4441  {
4442  if (value)
4443  {
4444  /* CSV separator has to be a one-byte character */
4445  if (strlen(value) != 1)
4446  {
4447  pg_log_error("\\pset: csv_fieldsep must be a single one-byte character");
4448  return false;
4449  }
4450  if (value[0] == '"' || value[0] == '\n' || value[0] == '\r')
4451  {
4452  pg_log_error("\\pset: csv_fieldsep cannot be a double quote, a newline, or a carriage return");
4453  return false;
4454  }
4455  popt->topt.csvFieldSep[0] = value[0];
4456  }
4457  }
4458 
4459  /* locale-aware numeric output */
4460  else if (strcmp(param, "numericlocale") == 0)
4461  {
4462  if (value)
4463  return ParseVariableBool(value, param, &popt->topt.numericLocale);
4464  else
4465  popt->topt.numericLocale = !popt->topt.numericLocale;
4466  }
4467 
4468  /* null display */
4469  else if (strcmp(param, "null") == 0)
4470  {
4471  if (value)
4472  {
4473  free(popt->nullPrint);
4474  popt->nullPrint = pg_strdup(value);
4475  }
4476  }
4477 
4478  /* field separator for unaligned text */
4479  else if (strcmp(param, "fieldsep") == 0)
4480  {
4481  if (value)
4482  {
4483  free(popt->topt.fieldSep.separator);
4485  popt->topt.fieldSep.separator_zero = false;
4486  }
4487  }
4488 
4489  else if (strcmp(param, "fieldsep_zero") == 0)
4490  {
4491  free(popt->topt.fieldSep.separator);
4492  popt->topt.fieldSep.separator = NULL;
4493  popt->topt.fieldSep.separator_zero = true;
4494  }
4495 
4496  /* record separator for unaligned text */
4497  else if (strcmp(param, "recordsep") == 0)
4498  {
4499  if (value)
4500  {
4501  free(popt->topt.recordSep.separator);
4503  popt->topt.recordSep.separator_zero = false;
4504  }
4505  }
4506 
4507  else if (strcmp(param, "recordsep_zero") == 0)
4508  {
4509  free(popt->topt.recordSep.separator);
4510  popt->topt.recordSep.separator = NULL;
4511  popt->topt.recordSep.separator_zero = true;
4512  }
4513 
4514  /* toggle between full and tuples-only format */
4515  else if (strcmp(param, "t") == 0 || strcmp(param, "tuples_only") == 0)
4516  {
4517  if (value)
4518  return ParseVariableBool(value, param, &popt->topt.tuples_only);
4519  else
4520  popt->topt.tuples_only = !popt->topt.tuples_only;
4521  }
4522 
4523  /* set title override */
4524  else if (strcmp(param, "C") == 0 || strcmp(param, "title") == 0)
4525  {
4526  free(popt->title);
4527  if (!value)
4528  popt->title = NULL;
4529  else
4530  popt->title = pg_strdup(value);
4531  }
4532 
4533  /* set HTML table tag options */
4534  else if (strcmp(param, "T") == 0 || strcmp(param, "tableattr") == 0)
4535  {
4536  free(popt->topt.tableAttr);
4537  if (!value)
4538  popt->topt.tableAttr = NULL;
4539  else
4540  popt->topt.tableAttr = pg_strdup(value);
4541  }
4542 
4543  /* toggle use of pager */
4544  else if (strcmp(param, "pager") == 0)
4545  {
4546  if (value && pg_strcasecmp(value, "always") == 0)
4547  popt->topt.pager = 2;
4548  else if (value)
4549  {
4550  bool on_off;
4551 
4552  if (!ParseVariableBool(value, NULL, &on_off))
4553  {
4554  PsqlVarEnumError(param, value, "on, off, always");
4555  return false;
4556  }
4557  popt->topt.pager = on_off ? 1 : 0;
4558  }
4559  else if (popt->topt.pager == 1)
4560  popt->topt.pager = 0;
4561  else
4562  popt->topt.pager = 1;
4563  }
4564 
4565  /* set minimum lines for pager use */
4566  else if (strcmp(param, "pager_min_lines") == 0)
4567  {
4568  if (value)
4569  popt->topt.pager_min_lines = atoi(value);
4570  }
4571 
4572  /* disable "(x rows)" footer */
4573  else if (strcmp(param, "footer") == 0)
4574  {
4575  if (value)
4576  return ParseVariableBool(value, param, &popt->topt.default_footer);
4577  else
4578  popt->topt.default_footer = !popt->topt.default_footer;
4579  }
4580 
4581  /* set border style/width */
4582  else if (strcmp(param, "columns") == 0)
4583  {
4584  if (value)
4585  popt->topt.columns = atoi(value);
4586  }
4587  else
4588  {
4589  pg_log_error("\\pset: unknown option: %s", param);
4590  return false;
4591  }
4592 
4593  if (!quiet)
4594  printPsetInfo(param, &pset.popt);
4595 
4596  return true;
4597 }
#define lengthof(array)
Definition: c.h:724
static bool set_unicode_line_style(const char *value, size_t vallen, unicode_linestyle *linestyle)
Definition: command.c:4211
static bool printPsetInfo(const char *param, printQueryOpt *popt)
Definition: command.c:4603
const char * name
Definition: encode.c:561
char * pg_strdup(const char *in)
Definition: fe_memutils.c:85
void refresh_utf8format(const printTableOpt *opt)
Definition: print.c:3676
const printTextFormat pg_asciiformat
Definition: print.c:56
const printTextFormat pg_asciiformat_old
Definition: print.c:77
printTextFormat pg_utf8format
Definition: print.c:99
@ PRINT_XHEADER_EXACT_WIDTH
Definition: print.h:75
@ PRINT_XHEADER_PAGE
Definition: print.h:74
@ PRINT_XHEADER_COLUMN
Definition: print.h:73
@ PRINT_XHEADER_FULL
Definition: print.h:72
printFormat
Definition: print.h:29
@ PRINT_LATEX_LONGTABLE
Definition: print.h:36
@ PRINT_CSV
Definition: print.h:33
@ PRINT_UNALIGNED
Definition: print.h:38
@ PRINT_ALIGNED
Definition: print.h:31
@ PRINT_TROFF_MS
Definition: print.h:37
@ PRINT_ASCIIDOC
Definition: print.h:32
@ PRINT_LATEX
Definition: print.h:35
@ PRINT_HTML
Definition: print.h:34
@ PRINT_WRAPPED
Definition: print.h:39
#define free(a)
Definition: header.h:65
static struct @143 value
int i
Definition: isn.c:73
static void const char * fmt
Assert(fmt[strlen(fmt) - 1] !='\n')
#define pg_log_error(...)
Definition: logging.h:106
int pg_strcasecmp(const char *s1, const char *s2)
Definition: pgstrcasecmp.c:36
int pg_strncasecmp(const char *s1, const char *s2, size_t n)
Definition: pgstrcasecmp.c:69
printQueryOpt popt
Definition: settings.h:91
printTableOpt topt
Definition: print.h:180
char * nullPrint
Definition: print.h:181
char * title
Definition: print.h:182
unsigned short int expanded
Definition: print.h:111
unicode_linestyle unicode_border_linestyle
Definition: print.h:136
bool tuples_only
Definition: print.h:121
int columns
Definition: print.h:135
enum printFormat format
Definition: print.h:110
struct separator fieldSep
Definition: print.h:127
int expanded_header_exact_width
Definition: print.h:114
struct separator recordSep
Definition: print.h:128
printXheaderWidthType expanded_header_width_type
Definition: print.h:113
char csvFieldSep[2]
Definition: print.h:129
const printTextFormat * line_style
Definition: print.h:126
bool default_footer
Definition: print.h:124
int pager_min_lines
Definition: print.h:119
unsigned short int pager
Definition: print.h:117
char * tableAttr
Definition: print.h:132
bool numericLocale
Definition: print.h:130
unsigned short int border
Definition: print.h:115
unicode_linestyle unicode_header_linestyle
Definition: print.h:138
unicode_linestyle unicode_column_linestyle
Definition: print.h:137
bool separator_zero
Definition: print.h:105
char * separator
Definition: print.h:104
void PsqlVarEnumError(const char *name, const char *value, const char *suggestions)
Definition: variables.c:416
bool ParseVariableBool(const char *value, const char *name, bool *result)
Definition: variables.c:107

References Assert(), printTableOpt::border, printTableOpt::columns, printTableOpt::csvFieldSep, printTableOpt::default_footer, printTableOpt::expanded, printTableOpt::expanded_header_exact_width, printTableOpt::expanded_header_width_type, printTableOpt::fieldSep, fmt, printTableOpt::format, free, i, lengthof, printTableOpt::line_style, name, printQueryOpt::nullPrint, printTableOpt::numericLocale, printTableOpt::pager, printTableOpt::pager_min_lines, ParseVariableBool(), pg_asciiformat, pg_asciiformat_old, pg_log_error, pg_strcasecmp(), pg_strdup(), pg_strncasecmp(), pg_utf8format, _psqlSettings::popt, PRINT_ALIGNED, PRINT_ASCIIDOC, PRINT_CSV, PRINT_HTML, PRINT_LATEX, PRINT_LATEX_LONGTABLE, PRINT_TROFF_MS, PRINT_UNALIGNED, PRINT_WRAPPED, PRINT_XHEADER_COLUMN, PRINT_XHEADER_EXACT_WIDTH, PRINT_XHEADER_FULL, PRINT_XHEADER_PAGE, printPsetInfo(), pset, PsqlVarEnumError(), printTableOpt::recordSep, refresh_utf8format(), separator::separator, separator::separator_zero, set_unicode_line_style(), printTableOpt::tableAttr, printQueryOpt::title, printQueryOpt::topt, printTableOpt::tuples_only, printTableOpt::unicode_border_linestyle, printTableOpt::unicode_column_linestyle, printTableOpt::unicode_header_linestyle, and value.

Referenced by exec_command_a(), exec_command_C(), exec_command_f(), exec_command_html(), exec_command_pset(), exec_command_t(), exec_command_T(), exec_command_x(), parse_psql_options(), and process_command_g_options().

◆ HandleSlashCmds()

backslashResult HandleSlashCmds ( PsqlScanState  scan_state,
ConditionalStack  cstack,
PQExpBuffer  query_buf,
PQExpBuffer  previous_buf 
)

Definition at line 213 of file command.c.

217 {
219  char *cmd;
220  char *arg;
221 
222  Assert(scan_state != NULL);
223  Assert(cstack != NULL);
224 
225  /* Parse off the command name */
226  cmd = psql_scan_slash_command(scan_state);
227 
228  /* And try to execute it */
229  status = exec_command(cmd, scan_state, cstack, query_buf, previous_buf);
230 
231  if (status == PSQL_CMD_UNKNOWN)
232  {
233  pg_log_error("invalid command \\%s", cmd);
235  pg_log_error_hint("Try \\? for help.");
237  }
238 
239  if (status != PSQL_CMD_ERROR)
240  {
241  /*
242  * Eat any remaining arguments after a valid command. We want to
243  * suppress evaluation of backticks in this situation, so transiently
244  * push an inactive conditional-stack entry.
245  */
246  bool active_branch = conditional_active(cstack);
247 
249  while ((arg = psql_scan_slash_option(scan_state,
250  OT_NORMAL, NULL, false)))
251  {
252  if (active_branch)
253  pg_log_warning("\\%s: extra argument \"%s\" ignored", cmd, arg);
254  free(arg);
255  }
256  conditional_stack_pop(cstack);
257  }
258  else
259  {
260  /* silently throw away rest of line after an erroneous command */
261  while ((arg = psql_scan_slash_option(scan_state,
262  OT_WHOLE_LINE, NULL, false)))
263  free(arg);
264  }
265 
266  /* if there is a trailing \\, swallow it */
267  psql_scan_slash_command_end(scan_state);
268 
269  free(cmd);
270 
271  /* some commands write to queryFout, so make sure output is sent */
273 
274  return status;
275 }
static backslashResult exec_command(const char *cmd, PsqlScanState scan_state, ConditionalStack cstack, PQExpBuffer query_buf, PQExpBuffer previous_buf)
Definition: command.c:286
void conditional_stack_push(ConditionalStack cstack, ifState new_state)
Definition: conditional.c:53
bool conditional_stack_pop(ConditionalStack cstack)
Definition: conditional.c:69
bool conditional_active(ConditionalStack cstack)
Definition: conditional.c:140
@ IFSTATE_IGNORED
Definition: conditional.h:37
static void const char fflush(stdout)
#define pg_log_error_hint(...)
Definition: logging.h:112
void * arg
static void static void status(const char *fmt,...) pg_attribute_printf(1
Definition: pg_regress.c:225
#define pg_log_warning(...)
Definition: pgfnames.c:24
char * psql_scan_slash_command(PsqlScanState state)
void psql_scan_slash_command_end(PsqlScanState state)
@ OT_NORMAL
Definition: psqlscanslash.h:17
@ OT_WHOLE_LINE
Definition: psqlscanslash.h:21
char * psql_scan_slash_option(PsqlScanState state, enum slash_option_type type, char *quote, bool semicolon)
FILE * queryFout
Definition: settings.h:84
bool cur_cmd_interactive
Definition: settings.h:110

References arg, Assert(), conditional_active(), conditional_stack_pop(), conditional_stack_push(), _psqlSettings::cur_cmd_interactive, exec_command(), fflush(), free, IFSTATE_IGNORED, OT_NORMAL, OT_WHOLE_LINE, pg_log_error, pg_log_error_hint, pg_log_warning, pset, PSQL_CMD_ERROR, PSQL_CMD_UNKNOWN, psql_scan_slash_command(), psql_scan_slash_command_end(), psql_scan_slash_option(), _psqlSettings::queryFout, and status().

Referenced by main(), and MainLoop().

◆ process_file()

int process_file ( char *  filename,
bool  use_relative_path 
)

Definition at line 4101 of file command.c.

4102 {
4103  FILE *fd;
4104  int result;
4105  char *oldfilename;
4106  char relpath[MAXPGPATH];
4107 
4108  if (!filename)
4109  {
4110  fd = stdin;
4111  filename = NULL;
4112  }
4113  else if (strcmp(filename, "-") != 0)
4114  {
4116 
4117  /*
4118  * If we were asked to resolve the pathname relative to the location
4119  * of the currently executing script, and there is one, and this is a
4120  * relative pathname, then prepend all but the last pathname component
4121  * of the current script to this pathname.
4122  */
4123  if (use_relative_path && pset.inputfile &&
4125  {
4126  strlcpy(relpath, pset.inputfile, sizeof(relpath));
4130 
4131  filename = relpath;
4132  }
4133 
4134  fd = fopen(filename, PG_BINARY_R);
4135 
4136  if (!fd)
4137  {
4138  pg_log_error("%s: %m", filename);
4139  return EXIT_FAILURE;
4140  }
4141  }
4142  else
4143  {
4144  fd = stdin;
4145  filename = "<stdin>"; /* for future error messages */
4146  }
4147 
4148  oldfilename = pset.inputfile;
4150 
4152 
4153  result = MainLoop(fd);
4154 
4155  if (fd != stdin)
4156  fclose(fd);
4157 
4158  pset.inputfile = oldfilename;
4159 
4161 
4162  return result;
4163 }
#define PG_BINARY_R
Definition: c.h:1211
void pg_logging_config(int new_flags)
Definition: logging.c:163
#define PG_LOG_FLAG_TERSE
Definition: logging.h:86
int MainLoop(FILE *source)
Definition: mainloop.c:33
#define MAXPGPATH
static char * filename
Definition: pg_dumpall.c:119
void join_path_components(char *ret_path, const char *head, const char *tail)
Definition: path.c:219
#define is_absolute_path(filename)
Definition: port.h:103
void canonicalize_path(char *path)
Definition: path.c:264
void get_parent_directory(char *path)
Definition: path.c:977
bool has_drive_prefix(const char *path)
Definition: path.c:88
size_t strlcpy(char *dst, const char *src, size_t siz)
Definition: strlcpy.c:45
static int fd(const char *x, int i)
Definition: preproc-init.c:105
#define relpath(rlocator, forknum)
Definition: relpath.h:94
#define EXIT_FAILURE
Definition: settings.h:166
char * inputfile
Definition: settings.h:113

References canonicalize_path(), EXIT_FAILURE, fd(), filename, get_parent_directory(), has_drive_prefix(), _psqlSettings::inputfile, is_absolute_path, join_path_components(), MainLoop(), MAXPGPATH, PG_BINARY_R, pg_log_error, PG_LOG_FLAG_TERSE, pg_logging_config(), pset, relpath, and strlcpy().

Referenced by exec_command_include().

◆ restorePsetInfo()

void restorePsetInfo ( printQueryOpt popt,
printQueryOpt save 
)

Definition at line 4839 of file command.c.

4840 {
4841  /* Free all the old data we're about to overwrite the pointers to. */
4842 
4843  /* topt.line_style points to const data that need not be duplicated */
4844  free(popt->topt.fieldSep.separator);
4845  free(popt->topt.recordSep.separator);
4846  free(popt->topt.tableAttr);
4847  free(popt->nullPrint);
4848  free(popt->title);
4849 
4850  /*
4851  * footers and translate_columns are never set in psql's print settings,
4852  * so we needn't write code to duplicate them.
4853  */
4854  Assert(popt->footers == NULL);
4855  Assert(popt->translate_columns == NULL);
4856 
4857  /* Now we may flat-copy all the fields, including pointers. */
4858  memcpy(popt, save, sizeof(printQueryOpt));
4859 
4860  /* Lastly, free "save" ... but its sub-structures now belong to popt. */
4861  free(save);
4862 }
const bool * translate_columns
Definition: print.h:185
char ** footers
Definition: print.h:183

References Assert(), printTableOpt::fieldSep, printQueryOpt::footers, free, printQueryOpt::nullPrint, printTableOpt::recordSep, separator::separator, printTableOpt::tableAttr, printQueryOpt::title, printQueryOpt::topt, and printQueryOpt::translate_columns.

Referenced by process_command_g_options(), and SendQuery().

◆ savePsetInfo()

printQueryOpt* savePsetInfo ( const printQueryOpt popt)

Definition at line 4803 of file command.c.

4804 {
4805  printQueryOpt *save;
4806 
4807  save = (printQueryOpt *) pg_malloc(sizeof(printQueryOpt));
4808 
4809  /* Flat-copy all the scalar fields, then duplicate sub-structures. */
4810  memcpy(save, popt, sizeof(printQueryOpt));
4811 
4812  /* topt.line_style points to const data that need not be duplicated */
4813  if (popt->topt.fieldSep.separator)
4815  if (popt->topt.recordSep.separator)
4817  if (popt->topt.tableAttr)
4818  save->topt.tableAttr = pg_strdup(popt->topt.tableAttr);
4819  if (popt->nullPrint)
4820  save->nullPrint = pg_strdup(popt->nullPrint);
4821  if (popt->title)
4822  save->title = pg_strdup(popt->title);
4823 
4824  /*
4825  * footers and translate_columns are never set in psql's print settings,
4826  * so we needn't write code to duplicate them.
4827  */
4828  Assert(popt->footers == NULL);
4829  Assert(popt->translate_columns == NULL);
4830 
4831  return save;
4832 }
void * pg_malloc(size_t size)
Definition: fe_memutils.c:47

References Assert(), printTableOpt::fieldSep, printQueryOpt::footers, printQueryOpt::nullPrint, pg_malloc(), pg_strdup(), printTableOpt::recordSep, separator::separator, printTableOpt::tableAttr, printQueryOpt::title, printQueryOpt::topt, and printQueryOpt::translate_columns.

Referenced by exec_command_g(), and process_command_g_options().

◆ SyncVariables()

void SyncVariables ( void  )

Definition at line 3763 of file command.c.

3764 {
3765  char vbuf[32];
3766  const char *server_version;
3767 
3768  /* get stuff from connection */
3772 
3773  SetVariable(pset.vars, "DBNAME", PQdb(pset.db));
3774  SetVariable(pset.vars, "USER", PQuser(pset.db));
3775  SetVariable(pset.vars, "HOST", PQhost(pset.db));
3776  SetVariable(pset.vars, "PORT", PQport(pset.db));
3778 
3779  /* this bit should match connection_warnings(): */
3780  /* Try to get full text form of version, might include "devel" etc */
3781  server_version = PQparameterStatus(pset.db, "server_version");
3782  /* Otherwise fall back on pset.sversion */
3783  if (!server_version)
3784  {
3785  formatPGVersionNumber(pset.sversion, true, vbuf, sizeof(vbuf));
3786  server_version = vbuf;
3787  }
3788  SetVariable(pset.vars, "SERVER_VERSION_NAME", server_version);
3789 
3790  snprintf(vbuf, sizeof(vbuf), "%d", pset.sversion);
3791  SetVariable(pset.vars, "SERVER_VERSION_NUM", vbuf);
3792 
3793  /* send stuff to it, too */
3796 }
const char * pg_encoding_to_char(int encoding)
Definition: encnames.c:588
int PQserverVersion(const PGconn *conn)
Definition: fe-connect.c:6733
char * PQhost(const PGconn *conn)
Definition: fe-connect.c:6622
char * PQdb(const PGconn *conn)
Definition: fe-connect.c:6589
int PQclientEncoding(const PGconn *conn)
Definition: fe-connect.c:6820
PGContextVisibility PQsetErrorContextVisibility(PGconn *conn, PGContextVisibility show_context)
Definition: fe-connect.c:6882
char * PQuser(const PGconn *conn)
Definition: fe-connect.c:6597
char * PQport(const PGconn *conn)
Definition: fe-connect.c:6658
PGVerbosity PQsetErrorVerbosity(PGconn *conn, PGVerbosity verbosity)
Definition: fe-connect.c:6870
#define snprintf
Definition: port.h:238
VariableSpace vars
Definition: settings.h:121
int encoding
Definition: settings.h:83
PGVerbosity verbosity
Definition: settings.h:153
PGContextVisibility show_context
Definition: settings.h:155
int encoding
Definition: print.h:133
bool SetVariable(VariableSpace space, const char *name, const char *value)
Definition: variables.c:211

References _psqlSettings::db, _psqlSettings::encoding, printTableOpt::encoding, formatPGVersionNumber(), pg_encoding_to_char(), _psqlSettings::popt, PQclientEncoding(), PQdb(), PQhost(), PQparameterStatus(), PQport(), PQserverVersion(), PQsetErrorContextVisibility(), PQsetErrorVerbosity(), PQuser(), pset, server_version, SetVariable(), _psqlSettings::show_context, snprintf, _psqlSettings::sversion, printQueryOpt::topt, _psqlSettings::vars, and _psqlSettings::verbosity.

Referenced by CheckConnection(), do_connect(), and main().

◆ UnsyncVariables()

void UnsyncVariables ( void  )

Definition at line 3804 of file command.c.

3805 {
3806  SetVariable(pset.vars, "DBNAME", NULL);
3807  SetVariable(pset.vars, "USER", NULL);
3808  SetVariable(pset.vars, "HOST", NULL);
3809  SetVariable(pset.vars, "PORT", NULL);
3810  SetVariable(pset.vars, "ENCODING", NULL);
3811  SetVariable(pset.vars, "SERVER_VERSION_NAME", NULL);
3812  SetVariable(pset.vars, "SERVER_VERSION_NUM", NULL);
3813 }

References pset, SetVariable(), and _psqlSettings::vars.

Referenced by CheckConnection(), and do_connect().