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 3821 of file command.c.

3822 {
3823  if (!pset.quiet && !pset.notty)
3824  {
3825  int client_ver = PG_VERSION_NUM;
3826  char cverbuf[32];
3827  char sverbuf[32];
3828 
3829  if (pset.sversion != client_ver)
3830  {
3831  const char *server_version;
3832 
3833  /* Try to get full text form, might include "devel" etc */
3834  server_version = PQparameterStatus(pset.db, "server_version");
3835  /* Otherwise fall back on pset.sversion */
3836  if (!server_version)
3837  {
3839  sverbuf, sizeof(sverbuf));
3840  server_version = sverbuf;
3841  }
3842 
3843  printf(_("%s (%s, server %s)\n"),
3844  pset.progname, PG_VERSION, server_version);
3845  }
3846  /* For version match, only print psql banner on startup. */
3847  else if (in_startup)
3848  printf("%s (%s)\n", pset.progname, PG_VERSION);
3849 
3850  /*
3851  * Warn if server's major version is newer than ours, or if server
3852  * predates our support cutoff (currently 9.2).
3853  */
3854  if (pset.sversion / 100 > client_ver / 100 ||
3855  pset.sversion < 90200)
3856  printf(_("WARNING: %s major version %s, server major version %s.\n"
3857  " Some psql features might not work.\n"),
3858  pset.progname,
3859  formatPGVersionNumber(client_ver, false,
3860  cverbuf, sizeof(cverbuf)),
3862  sverbuf, sizeof(sverbuf)));
3863 
3864 #ifdef WIN32
3865  if (in_startup)
3866  checkWin32Codepage();
3867 #endif
3868  printSSLInfo();
3869  printGSSInfo();
3870  }
3871 }
static void printSSLInfo(void)
Definition: command.c:3880
static void printGSSInfo(void)
Definition: command.c:3908
#define _(x)
Definition: elog.c:90
const char * PQparameterStatus(const PGconn *conn, const char *paramName)
Definition: fe-connect.c:7112
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:113

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 4440 of file command.c.

4441 {
4442  size_t vallen = 0;
4443 
4444  Assert(param != NULL);
4445 
4446  if (value)
4447  vallen = strlen(value);
4448 
4449  /* set format */
4450  if (strcmp(param, "format") == 0)
4451  {
4452  static const struct fmt
4453  {
4454  const char *name;
4455  enum printFormat number;
4456  } formats[] =
4457  {
4458  /* remember to update error message below when adding more */
4459  {"aligned", PRINT_ALIGNED},
4460  {"asciidoc", PRINT_ASCIIDOC},
4461  {"csv", PRINT_CSV},
4462  {"html", PRINT_HTML},
4463  {"latex", PRINT_LATEX},
4464  {"troff-ms", PRINT_TROFF_MS},
4465  {"unaligned", PRINT_UNALIGNED},
4466  {"wrapped", PRINT_WRAPPED}
4467  };
4468 
4469  if (!value)
4470  ;
4471  else
4472  {
4473  int match_pos = -1;
4474 
4475  for (int i = 0; i < lengthof(formats); i++)
4476  {
4477  if (pg_strncasecmp(formats[i].name, value, vallen) == 0)
4478  {
4479  if (match_pos < 0)
4480  match_pos = i;
4481  else
4482  {
4483  pg_log_error("\\pset: ambiguous abbreviation \"%s\" matches both \"%s\" and \"%s\"",
4484  value,
4485  formats[match_pos].name, formats[i].name);
4486  return false;
4487  }
4488  }
4489  }
4490  if (match_pos >= 0)
4491  popt->topt.format = formats[match_pos].number;
4492  else if (pg_strncasecmp("latex-longtable", value, vallen) == 0)
4493  {
4494  /*
4495  * We must treat latex-longtable specially because latex is a
4496  * prefix of it; if both were in the table above, we'd think
4497  * "latex" is ambiguous.
4498  */
4500  }
4501  else
4502  {
4503  pg_log_error("\\pset: allowed formats are aligned, asciidoc, csv, html, latex, latex-longtable, troff-ms, unaligned, wrapped");
4504  return false;
4505  }
4506  }
4507  }
4508 
4509  /* set table line style */
4510  else if (strcmp(param, "linestyle") == 0)
4511  {
4512  if (!value)
4513  ;
4514  else if (pg_strncasecmp("ascii", value, vallen) == 0)
4515  popt->topt.line_style = &pg_asciiformat;
4516  else if (pg_strncasecmp("old-ascii", value, vallen) == 0)
4518  else if (pg_strncasecmp("unicode", value, vallen) == 0)
4519  popt->topt.line_style = &pg_utf8format;
4520  else
4521  {
4522  pg_log_error("\\pset: allowed line styles are ascii, old-ascii, unicode");
4523  return false;
4524  }
4525  }
4526 
4527  /* set unicode border line style */
4528  else if (strcmp(param, "unicode_border_linestyle") == 0)
4529  {
4530  if (!value)
4531  ;
4532  else if (set_unicode_line_style(value, vallen,
4534  refresh_utf8format(&(popt->topt));
4535  else
4536  {
4537  pg_log_error("\\pset: allowed Unicode border line styles are single, double");
4538  return false;
4539  }
4540  }
4541 
4542  /* set unicode column line style */
4543  else if (strcmp(param, "unicode_column_linestyle") == 0)
4544  {
4545  if (!value)
4546  ;
4547  else if (set_unicode_line_style(value, vallen,
4549  refresh_utf8format(&(popt->topt));
4550  else
4551  {
4552  pg_log_error("\\pset: allowed Unicode column line styles are single, double");
4553  return false;
4554  }
4555  }
4556 
4557  /* set unicode header line style */
4558  else if (strcmp(param, "unicode_header_linestyle") == 0)
4559  {
4560  if (!value)
4561  ;
4562  else if (set_unicode_line_style(value, vallen,
4564  refresh_utf8format(&(popt->topt));
4565  else
4566  {
4567  pg_log_error("\\pset: allowed Unicode header line styles are single, double");
4568  return false;
4569  }
4570  }
4571 
4572  /* set border style/width */
4573  else if (strcmp(param, "border") == 0)
4574  {
4575  if (value)
4576  popt->topt.border = atoi(value);
4577  }
4578 
4579  /* set expanded/vertical mode */
4580  else if (strcmp(param, "x") == 0 ||
4581  strcmp(param, "expanded") == 0 ||
4582  strcmp(param, "vertical") == 0)
4583  {
4584  if (value && pg_strcasecmp(value, "auto") == 0)
4585  popt->topt.expanded = 2;
4586  else if (value)
4587  {
4588  bool on_off;
4589 
4590  if (ParseVariableBool(value, NULL, &on_off))
4591  popt->topt.expanded = on_off ? 1 : 0;
4592  else
4593  {
4594  PsqlVarEnumError(param, value, "on, off, auto");
4595  return false;
4596  }
4597  }
4598  else
4599  popt->topt.expanded = !popt->topt.expanded;
4600  }
4601 
4602  /* header line width in expanded mode */
4603  else if (strcmp(param, "xheader_width") == 0)
4604  {
4605  if (!value)
4606  ;
4607  else if (pg_strcasecmp(value, "full") == 0)
4609  else if (pg_strcasecmp(value, "column") == 0)
4611  else if (pg_strcasecmp(value, "page") == 0)
4613  else
4614  {
4615  int intval = atoi(value);
4616 
4617  if (intval == 0)
4618  {
4619  pg_log_error("\\pset: allowed xheader_width values are \"%s\" (default), \"%s\", \"%s\", or a number specifying the exact width", "full", "column", "page");
4620  return false;
4621  }
4622 
4624  popt->topt.expanded_header_exact_width = intval;
4625  }
4626  }
4627 
4628  /* field separator for CSV format */
4629  else if (strcmp(param, "csv_fieldsep") == 0)
4630  {
4631  if (value)
4632  {
4633  /* CSV separator has to be a one-byte character */
4634  if (strlen(value) != 1)
4635  {
4636  pg_log_error("\\pset: csv_fieldsep must be a single one-byte character");
4637  return false;
4638  }
4639  if (value[0] == '"' || value[0] == '\n' || value[0] == '\r')
4640  {
4641  pg_log_error("\\pset: csv_fieldsep cannot be a double quote, a newline, or a carriage return");
4642  return false;
4643  }
4644  popt->topt.csvFieldSep[0] = value[0];
4645  }
4646  }
4647 
4648  /* locale-aware numeric output */
4649  else if (strcmp(param, "numericlocale") == 0)
4650  {
4651  if (value)
4652  return ParseVariableBool(value, param, &popt->topt.numericLocale);
4653  else
4654  popt->topt.numericLocale = !popt->topt.numericLocale;
4655  }
4656 
4657  /* null display */
4658  else if (strcmp(param, "null") == 0)
4659  {
4660  if (value)
4661  {
4662  free(popt->nullPrint);
4663  popt->nullPrint = pg_strdup(value);
4664  }
4665  }
4666 
4667  /* field separator for unaligned text */
4668  else if (strcmp(param, "fieldsep") == 0)
4669  {
4670  if (value)
4671  {
4672  free(popt->topt.fieldSep.separator);
4674  popt->topt.fieldSep.separator_zero = false;
4675  }
4676  }
4677 
4678  else if (strcmp(param, "fieldsep_zero") == 0)
4679  {
4680  free(popt->topt.fieldSep.separator);
4681  popt->topt.fieldSep.separator = NULL;
4682  popt->topt.fieldSep.separator_zero = true;
4683  }
4684 
4685  /* record separator for unaligned text */
4686  else if (strcmp(param, "recordsep") == 0)
4687  {
4688  if (value)
4689  {
4690  free(popt->topt.recordSep.separator);
4692  popt->topt.recordSep.separator_zero = false;
4693  }
4694  }
4695 
4696  else if (strcmp(param, "recordsep_zero") == 0)
4697  {
4698  free(popt->topt.recordSep.separator);
4699  popt->topt.recordSep.separator = NULL;
4700  popt->topt.recordSep.separator_zero = true;
4701  }
4702 
4703  /* toggle between full and tuples-only format */
4704  else if (strcmp(param, "t") == 0 || strcmp(param, "tuples_only") == 0)
4705  {
4706  if (value)
4707  return ParseVariableBool(value, param, &popt->topt.tuples_only);
4708  else
4709  popt->topt.tuples_only = !popt->topt.tuples_only;
4710  }
4711 
4712  /* set title override */
4713  else if (strcmp(param, "C") == 0 || strcmp(param, "title") == 0)
4714  {
4715  free(popt->title);
4716  if (!value)
4717  popt->title = NULL;
4718  else
4719  popt->title = pg_strdup(value);
4720  }
4721 
4722  /* set HTML table tag options */
4723  else if (strcmp(param, "T") == 0 || strcmp(param, "tableattr") == 0)
4724  {
4725  free(popt->topt.tableAttr);
4726  if (!value)
4727  popt->topt.tableAttr = NULL;
4728  else
4729  popt->topt.tableAttr = pg_strdup(value);
4730  }
4731 
4732  /* toggle use of pager */
4733  else if (strcmp(param, "pager") == 0)
4734  {
4735  if (value && pg_strcasecmp(value, "always") == 0)
4736  popt->topt.pager = 2;
4737  else if (value)
4738  {
4739  bool on_off;
4740 
4741  if (!ParseVariableBool(value, NULL, &on_off))
4742  {
4743  PsqlVarEnumError(param, value, "on, off, always");
4744  return false;
4745  }
4746  popt->topt.pager = on_off ? 1 : 0;
4747  }
4748  else if (popt->topt.pager == 1)
4749  popt->topt.pager = 0;
4750  else
4751  popt->topt.pager = 1;
4752  }
4753 
4754  /* set minimum lines for pager use */
4755  else if (strcmp(param, "pager_min_lines") == 0)
4756  {
4757  if (value &&
4758  !ParseVariableNum(value, "pager_min_lines", &popt->topt.pager_min_lines))
4759  return false;
4760  }
4761 
4762  /* disable "(x rows)" footer */
4763  else if (strcmp(param, "footer") == 0)
4764  {
4765  if (value)
4766  return ParseVariableBool(value, param, &popt->topt.default_footer);
4767  else
4768  popt->topt.default_footer = !popt->topt.default_footer;
4769  }
4770 
4771  /* set border style/width */
4772  else if (strcmp(param, "columns") == 0)
4773  {
4774  if (value)
4775  popt->topt.columns = atoi(value);
4776  }
4777  else
4778  {
4779  pg_log_error("\\pset: unknown option: %s", param);
4780  return false;
4781  }
4782 
4783  if (!quiet)
4784  printPsetInfo(param, &pset.popt);
4785 
4786  return true;
4787 }
#define Assert(condition)
Definition: c.h:858
#define lengthof(array)
Definition: c.h:788
static bool set_unicode_line_style(const char *value, size_t vallen, unicode_linestyle *linestyle)
Definition: command.c:4397
static bool printPsetInfo(const char *param, printQueryOpt *popt)
Definition: command.c:4793
char * pg_strdup(const char *in)
Definition: fe_memutils.c:85
void refresh_utf8format(const printTableOpt *opt)
Definition: print.c:3691
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:78
@ PRINT_XHEADER_PAGE
Definition: print.h:76
@ PRINT_XHEADER_COLUMN
Definition: print.h:74
@ 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 @155 value
int i
Definition: isn.c:73
static void const char * fmt
#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:185
char * nullPrint
Definition: print.h:186
char * title
Definition: print.h:187
unsigned short int expanded
Definition: print.h:114
unicode_linestyle unicode_border_linestyle
Definition: print.h:141
bool tuples_only
Definition: print.h:126
int columns
Definition: print.h:140
enum printFormat format
Definition: print.h:113
struct separator fieldSep
Definition: print.h:132
int expanded_header_exact_width
Definition: print.h:118
struct separator recordSep
Definition: print.h:133
printXheaderWidthType expanded_header_width_type
Definition: print.h:116
char csvFieldSep[2]
Definition: print.h:134
const printTextFormat * line_style
Definition: print.h:131
bool default_footer
Definition: print.h:129
int pager_min_lines
Definition: print.h:124
unsigned short int pager
Definition: print.h:122
char * tableAttr
Definition: print.h:137
bool numericLocale
Definition: print.h:135
unsigned short int border
Definition: print.h:120
unicode_linestyle unicode_header_linestyle
Definition: print.h:143
unicode_linestyle unicode_column_linestyle
Definition: print.h:142
bool separator_zero
Definition: print.h:108
char * separator
Definition: print.h:107
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
bool ParseVariableNum(const char *value, const char *name, int *result)
Definition: variables.c:156
const char * name

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(), ParseVariableNum(), 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 215 of file command.c.

219 {
220  backslashResult status;
221  char *cmd;
222  char *arg;
223 
224  Assert(scan_state != NULL);
225  Assert(cstack != NULL);
226 
227  /* Parse off the command name */
228  cmd = psql_scan_slash_command(scan_state);
229 
230  /* And try to execute it */
231  status = exec_command(cmd, scan_state, cstack, query_buf, previous_buf);
232 
233  if (status == PSQL_CMD_UNKNOWN)
234  {
235  pg_log_error("invalid command \\%s", cmd);
237  pg_log_error_hint("Try \\? for help.");
238  status = PSQL_CMD_ERROR;
239  }
240 
241  if (status != PSQL_CMD_ERROR)
242  {
243  /*
244  * Eat any remaining arguments after a valid command. We want to
245  * suppress evaluation of backticks in this situation, so transiently
246  * push an inactive conditional-stack entry.
247  */
248  bool active_branch = conditional_active(cstack);
249 
251  while ((arg = psql_scan_slash_option(scan_state,
252  OT_NORMAL, NULL, false)))
253  {
254  if (active_branch)
255  pg_log_warning("\\%s: extra argument \"%s\" ignored", cmd, arg);
256  free(arg);
257  }
258  conditional_stack_pop(cstack);
259  }
260  else
261  {
262  /* silently throw away rest of line after an erroneous command */
263  while ((arg = psql_scan_slash_option(scan_state,
264  OT_WHOLE_LINE, NULL, false)))
265  free(arg);
266  }
267 
268  /* if there is a trailing \\, swallow it */
269  psql_scan_slash_command_end(scan_state);
270 
271  free(cmd);
272 
273  /* some commands write to queryFout, so make sure output is sent */
275 
276  return status;
277 }
static backslashResult exec_command(const char *cmd, PsqlScanState scan_state, ConditionalStack cstack, PQExpBuffer query_buf, PQExpBuffer previous_buf)
Definition: command.c:288
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
#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:111

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(), and _psqlSettings::queryFout.

Referenced by main(), and MainLoop().

◆ process_file()

int process_file ( char *  filename,
bool  use_relative_path 
)

Definition at line 4287 of file command.c.

4288 {
4289  FILE *fd;
4290  int result;
4291  char *oldfilename;
4292  char relpath[MAXPGPATH];
4293 
4294  if (!filename)
4295  {
4296  fd = stdin;
4297  filename = NULL;
4298  }
4299  else if (strcmp(filename, "-") != 0)
4300  {
4302 
4303  /*
4304  * If we were asked to resolve the pathname relative to the location
4305  * of the currently executing script, and there is one, and this is a
4306  * relative pathname, then prepend all but the last pathname component
4307  * of the current script to this pathname.
4308  */
4309  if (use_relative_path && pset.inputfile &&
4311  {
4312  strlcpy(relpath, pset.inputfile, sizeof(relpath));
4316 
4317  filename = relpath;
4318  }
4319 
4320  fd = fopen(filename, PG_BINARY_R);
4321 
4322  if (!fd)
4323  {
4324  pg_log_error("%s: %m", filename);
4325  return EXIT_FAILURE;
4326  }
4327  }
4328  else
4329  {
4330  fd = stdin;
4331  filename = "<stdin>"; /* for future error messages */
4332  }
4333 
4334  oldfilename = pset.inputfile;
4336 
4338 
4339  result = MainLoop(fd);
4340 
4341  if (fd != stdin)
4342  fclose(fd);
4343 
4344  pset.inputfile = oldfilename;
4345 
4347 
4348  return result;
4349 }
#define PG_BINARY_R
Definition: c.h:1275
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:976
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:167
char * inputfile
Definition: settings.h:114

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 5029 of file command.c.

5030 {
5031  /* Free all the old data we're about to overwrite the pointers to. */
5032 
5033  /* topt.line_style points to const data that need not be duplicated */
5034  free(popt->topt.fieldSep.separator);
5035  free(popt->topt.recordSep.separator);
5036  free(popt->topt.tableAttr);
5037  free(popt->nullPrint);
5038  free(popt->title);
5039 
5040  /*
5041  * footers and translate_columns are never set in psql's print settings,
5042  * so we needn't write code to duplicate them.
5043  */
5044  Assert(popt->footers == NULL);
5045  Assert(popt->translate_columns == NULL);
5046 
5047  /* Now we may flat-copy all the fields, including pointers. */
5048  memcpy(popt, save, sizeof(printQueryOpt));
5049 
5050  /* Lastly, free "save" ... but its sub-structures now belong to popt. */
5051  free(save);
5052 }
const bool * translate_columns
Definition: print.h:190
char ** footers
Definition: print.h:188

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 4993 of file command.c.

4994 {
4995  printQueryOpt *save;
4996 
4997  save = (printQueryOpt *) pg_malloc(sizeof(printQueryOpt));
4998 
4999  /* Flat-copy all the scalar fields, then duplicate sub-structures. */
5000  memcpy(save, popt, sizeof(printQueryOpt));
5001 
5002  /* topt.line_style points to const data that need not be duplicated */
5003  if (popt->topt.fieldSep.separator)
5005  if (popt->topt.recordSep.separator)
5007  if (popt->topt.tableAttr)
5008  save->topt.tableAttr = pg_strdup(popt->topt.tableAttr);
5009  if (popt->nullPrint)
5010  save->nullPrint = pg_strdup(popt->nullPrint);
5011  if (popt->title)
5012  save->title = pg_strdup(popt->title);
5013 
5014  /*
5015  * footers and translate_columns are never set in psql's print settings,
5016  * so we needn't write code to duplicate them.
5017  */
5018  Assert(popt->footers == NULL);
5019  Assert(popt->translate_columns == NULL);
5020 
5021  return save;
5022 }
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 3949 of file command.c.

3950 {
3951  char vbuf[32];
3952  const char *server_version;
3953 
3954  /* get stuff from connection */
3958 
3959  SetVariable(pset.vars, "DBNAME", PQdb(pset.db));
3960  SetVariable(pset.vars, "USER", PQuser(pset.db));
3961  SetVariable(pset.vars, "HOST", PQhost(pset.db));
3962  SetVariable(pset.vars, "PORT", PQport(pset.db));
3964 
3965  /* this bit should match connection_warnings(): */
3966  /* Try to get full text form of version, might include "devel" etc */
3967  server_version = PQparameterStatus(pset.db, "server_version");
3968  /* Otherwise fall back on pset.sversion */
3969  if (!server_version)
3970  {
3971  formatPGVersionNumber(pset.sversion, true, vbuf, sizeof(vbuf));
3972  server_version = vbuf;
3973  }
3974  SetVariable(pset.vars, "SERVER_VERSION_NAME", server_version);
3975 
3976  snprintf(vbuf, sizeof(vbuf), "%d", pset.sversion);
3977  SetVariable(pset.vars, "SERVER_VERSION_NUM", vbuf);
3978 
3979  /* send stuff to it, too */
3982 }
int PQserverVersion(const PGconn *conn)
Definition: fe-connect.c:7137
char * PQhost(const PGconn *conn)
Definition: fe-connect.c:7026
char * PQdb(const PGconn *conn)
Definition: fe-connect.c:6993
int PQclientEncoding(const PGconn *conn)
Definition: fe-connect.c:7235
PGContextVisibility PQsetErrorContextVisibility(PGconn *conn, PGContextVisibility show_context)
Definition: fe-connect.c:7297
char * PQuser(const PGconn *conn)
Definition: fe-connect.c:7001
char * PQport(const PGconn *conn)
Definition: fe-connect.c:7062
PGVerbosity PQsetErrorVerbosity(PGconn *conn, PGVerbosity verbosity)
Definition: fe-connect.c:7285
#define pg_encoding_to_char
Definition: pg_wchar.h:630
#define snprintf
Definition: port.h:238
VariableSpace vars
Definition: settings.h:122
int encoding
Definition: settings.h:83
PGVerbosity verbosity
Definition: settings.h:154
PGContextVisibility show_context
Definition: settings.h:156
int encoding
Definition: print.h:138
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 3990 of file command.c.

3991 {
3992  SetVariable(pset.vars, "DBNAME", NULL);
3993  SetVariable(pset.vars, "USER", NULL);
3994  SetVariable(pset.vars, "HOST", NULL);
3995  SetVariable(pset.vars, "PORT", NULL);
3996  SetVariable(pset.vars, "ENCODING", NULL);
3997  SetVariable(pset.vars, "SERVER_VERSION_NAME", NULL);
3998  SetVariable(pset.vars, "SERVER_VERSION_NUM", NULL);
3999 }

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

Referenced by CheckConnection(), and do_connect().