> > I belive, we need proper includes. > Given that wait_error.c already seems to have the right includes worked out for WEXITSTATUS/WIFSTOPPED/etc, I decided to just add a function there.
I named it wait_result_to_exit_code(), but I welcome suggestions of a better name.
From 9e2827a6f955e7cebf87ca538fab113a359951b4 Mon Sep 17 00:00:00 2001 From: coreyhuinker <corey.huin...@gmail.com> Date: Wed, 11 Jan 2023 17:27:23 -0500 Subject: [PATCH v6 1/3] Add PG_OS_TARGET environment variable to enable OS-specific changes to regression tests. --- src/test/regress/pg_regress.c | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/src/test/regress/pg_regress.c b/src/test/regress/pg_regress.c index 6cd5998b9d..bf0ada4560 100644 --- a/src/test/regress/pg_regress.c +++ b/src/test/regress/pg_regress.c @@ -594,6 +594,17 @@ initialize_environment(void) #endif } + /* + * Regression tests that invoke OS commands must occasionally use + * OS-specific syntax (example: NUL vs /dev/null). This env var allows + * those tests to adjust commands accordingly. + */ +#if defined(WIN32) + setenv("PG_OS_TARGET", "WIN32", 1); +#else + setenv("PG_OS_TARGET", "OTHER", 1); +#endif + /* * Set translation-related settings to English; otherwise psql will * produce translated messages and produce diffs. (XXX If we ever support -- 2.39.0
From 2c73156874bd0c222f414a2988271c4870d4293b Mon Sep 17 00:00:00 2001 From: coreyhuinker <corey.huin...@gmail.com> Date: Thu, 12 Jan 2023 23:04:31 -0500 Subject: [PATCH v6 2/3] Add wait_result_to_exit_code(). --- src/common/wait_error.c | 15 +++++++++++++++ src/include/port.h | 1 + 2 files changed, 16 insertions(+) diff --git a/src/common/wait_error.c b/src/common/wait_error.c index 4a3c3c61af..1eb7ff3fa2 100644 --- a/src/common/wait_error.c +++ b/src/common/wait_error.c @@ -127,3 +127,18 @@ wait_result_is_any_signal(int exit_status, bool include_command_not_found) return true; return false; } + +/* + * Return the POSIX exit code (0 to 255) that corresponds to the argument. + * The argument is a return code returned by wait(2) or waitpid(2), which + * also applies to pclose(3) and system(3). + */ +int +wait_result_to_exit_code(int exit_status) +{ + if (WIFEXITED(exit_status)) + return WEXITSTATUS(exit_status); + if (WIFSIGNALED(exit_status)) + return WTERMSIG(exit_status); + return 0; +} diff --git a/src/include/port.h b/src/include/port.h index e66193bed9..1b119a3c56 100644 --- a/src/include/port.h +++ b/src/include/port.h @@ -495,6 +495,7 @@ extern char *escape_single_quotes_ascii(const char *src); extern char *wait_result_to_str(int exitstatus); extern bool wait_result_is_signal(int exit_status, int signum); extern bool wait_result_is_any_signal(int exit_status, bool include_command_not_found); +extern int wait_result_to_exit_code(int exit_status); /* * Interfaces that we assume all Unix system have. We retain individual macros -- 2.39.0
From 9a969b50eec614cc181f020e25bd8dd00e7802f8 Mon Sep 17 00:00:00 2001 From: coreyhuinker <corey.huin...@gmail.com> Date: Thu, 12 Jan 2023 23:28:14 -0500 Subject: [PATCH v6 3/3] Add psql variables SHELL_ERROR and SHELL_EXIT_CODE which report the success or failure of the most recent psql OS-command executed via the \! command or `backticks`. These variables are roughly analogous to ERROR and SQLSTATE, but for OS level operations instead of SQL commands. --- doc/src/sgml/ref/psql-ref.sgml | 21 ++++++++++++++++++ src/bin/psql/command.c | 14 ++++++++++++ src/bin/psql/help.c | 4 ++++ src/bin/psql/psqlscanslash.l | 35 +++++++++++++++++++++++++----- src/test/regress/expected/psql.out | 25 +++++++++++++++++++++ src/test/regress/sql/psql.sql | 21 ++++++++++++++++++ 6 files changed, 114 insertions(+), 6 deletions(-) diff --git a/doc/src/sgml/ref/psql-ref.sgml b/doc/src/sgml/ref/psql-ref.sgml index dc6528dc11..3d8a7353b3 100644 --- a/doc/src/sgml/ref/psql-ref.sgml +++ b/doc/src/sgml/ref/psql-ref.sgml @@ -4264,6 +4264,27 @@ bar </listitem> </varlistentry> + <varlistentry id="app-psql-variables-shell-error"> + <term><varname>SHELL_ERROR</varname></term> + <listitem> + <para> + <literal>true</literal> if the last shell command failed, <literal>false</literal> if + it succeeded. This applies to shell commands invoked via either <literal>\!</literal> + or <literal>`</literal>. See also <varname>SHELL_EXIT_CODE</varname>. + </para> + </listitem> + </varlistentry> + + <varlistentry id="app-psql-variables-exit-code"> + <term><varname>SHELL_EXIT_CODE</varname></term> + <listitem> + <para> + The exit code return by the last shell command, invoked via either <literal>\!</literal> or <literal>`</literal>. + 0 means no error. See also <varname>SHELL_ERROR</varname>. + </para> + </listitem> + </varlistentry> + <varlistentry id="app-psql-variables-show-all-results"> <term><varname>SHOW_ALL_RESULTS</varname></term> <listitem> diff --git a/src/bin/psql/command.c b/src/bin/psql/command.c index b5201edf55..ea79d4fe57 100644 --- a/src/bin/psql/command.c +++ b/src/bin/psql/command.c @@ -5032,6 +5032,20 @@ do_shell(const char *command) else result = system(command); + if (result == 0) + { + SetVariable(pset.vars, "SHELL_EXIT_CODE", "0"); + SetVariable(pset.vars, "SHELL_ERROR", "false"); + } + else + { + int exit_code = wait_result_to_exit_code(result); + char buf[32]; + snprintf(buf, sizeof(buf), "%d", exit_code); + SetVariable(pset.vars, "SHELL_EXIT_CODE", buf); + SetVariable(pset.vars, "SHELL_ERROR", "true"); + } + if (result == 127 || result == -1) { pg_log_error("\\!: failed"); diff --git a/src/bin/psql/help.c b/src/bin/psql/help.c index e45c4aaca5..6d5226f793 100644 --- a/src/bin/psql/help.c +++ b/src/bin/psql/help.c @@ -455,6 +455,10 @@ helpVariables(unsigned short int pager) " show all results of a combined query (\\;) instead of only the last\n"); HELP0(" SHOW_CONTEXT\n" " controls display of message context fields [never, errors, always]\n"); + HELP0(" SHELL_ERROR\n" + " true if last shell command failed, else false\n"); + HELP0(" SHELL_EXIT_CODE\n" + " Exit code of the last shell command\n"); HELP0(" SINGLELINE\n" " if set, end of line terminates SQL commands (same as -S option)\n"); HELP0(" SINGLESTEP\n" diff --git a/src/bin/psql/psqlscanslash.l b/src/bin/psql/psqlscanslash.l index 8449ee1a52..3545787d31 100644 --- a/src/bin/psql/psqlscanslash.l +++ b/src/bin/psql/psqlscanslash.l @@ -23,6 +23,7 @@ #include "fe_utils/conditional.h" #include "libpq-fe.h" +#include "settings.h" } %{ @@ -774,6 +775,7 @@ evaluate_backtick(PsqlScanState state) bool error = false; char buf[512]; size_t result; + int exit_code = 0; initPQExpBuffer(&cmd_output); @@ -783,6 +785,7 @@ evaluate_backtick(PsqlScanState state) { pg_log_error("%s: %m", cmd); error = true; + exit_code = -1; } if (!error) @@ -790,7 +793,8 @@ evaluate_backtick(PsqlScanState state) do { result = fread(buf, 1, sizeof(buf), fd); - if (ferror(fd)) + exit_code = ferror(fd); + if (exit_code) { pg_log_error("%s: %m", cmd); error = true; @@ -800,10 +804,21 @@ evaluate_backtick(PsqlScanState state) } while (!feof(fd)); } - if (fd && pclose(fd) == -1) + if (fd) { - pg_log_error("%s: %m", cmd); - error = true; + /* + * Capture exit code for SHELL_EXIT_CODE, but to not erase an existing + * nonzero exit_code. + */ + int close_exit_code = pclose(fd); + + if (close_exit_code && !exit_code) + { + error = true; + exit_code = close_exit_code; + if (close_exit_code == -1) + pg_log_error("%s: %m", cmd); + } } if (PQExpBufferDataBroken(cmd_output)) @@ -824,7 +839,15 @@ evaluate_backtick(PsqlScanState state) cmd_output.data[cmd_output.len - 1] == '\n') cmd_output.len--; appendBinaryPQExpBuffer(output_buf, cmd_output.data, cmd_output.len); + SetVariable(pset.vars, "SHELL_EXIT_CODE", "0"); + SetVariable(pset.vars, "SHELL_ERROR", "false"); + } + else + { + char exit_code_buf[32]; + snprintf(exit_code_buf, sizeof(exit_code_buf), "%d", + wait_result_to_exit_code(exit_code)); + SetVariable(pset.vars, "SHELL_EXIT_CODE", exit_code_buf); + SetVariable(pset.vars, "SHELL_ERROR", "true"); } - - termPQExpBuffer(&cmd_output); } diff --git a/src/test/regress/expected/psql.out b/src/test/regress/expected/psql.out index 8fc62cebd2..b2addffa70 100644 --- a/src/test/regress/expected/psql.out +++ b/src/test/regress/expected/psql.out @@ -1306,6 +1306,31 @@ execute q; +----+-------------+ deallocate q; +-- test SHELL_EXIT_CODE +\getenv pg_os_target PG_OS_TARGET +SELECT :'pg_os_target' = 'WIN32' AS windows_target \gset +\if :windows_target + \set bad_cmd 'nosuchcommand 2> NUL' +\else + \set bad_cmd 'nosuchcommand 2> /dev/null' +\endif +\set bad_shbang '\\! ' :bad_cmd +:bad_shbang +\echo :SHELL_EXIT_CODE +127 +\echo :SHELL_ERROR +true +\set SHELL_EXIT_CODE 'clear' +\set SHELL_ERROR 'clear' +\echo :SHELL_EXIT_CODE +clear +\echo :SHELL_ERROR +clear +\set nosuchvar `:bad_cmd` +\echo :SHELL_EXIT_CODE +127 +\echo :SHELL_ERROR +true -- test single-line header and data prepare q as select repeat('x',2*n) as "0123456789abcdef", repeat('y',20-2*n) as "0123456789" from generate_series(1,10) as n; \pset linestyle ascii diff --git a/src/test/regress/sql/psql.sql b/src/test/regress/sql/psql.sql index 2da9665a19..be86ef6984 100644 --- a/src/test/regress/sql/psql.sql +++ b/src/test/regress/sql/psql.sql @@ -291,6 +291,27 @@ execute q; deallocate q; +-- test SHELL_EXIT_CODE +\getenv pg_os_target PG_OS_TARGET +SELECT :'pg_os_target' = 'WIN32' AS windows_target \gset +\if :windows_target + \set bad_cmd 'nosuchcommand 2> NUL' +\else + \set bad_cmd 'nosuchcommand 2> /dev/null' +\endif +\set bad_shbang '\\! ' :bad_cmd + +:bad_shbang +\echo :SHELL_EXIT_CODE +\echo :SHELL_ERROR +\set SHELL_EXIT_CODE 'clear' +\set SHELL_ERROR 'clear' +\echo :SHELL_EXIT_CODE +\echo :SHELL_ERROR +\set nosuchvar `:bad_cmd` +\echo :SHELL_EXIT_CODE +\echo :SHELL_ERROR + -- test single-line header and data prepare q as select repeat('x',2*n) as "0123456789abcdef", repeat('y',20-2*n) as "0123456789" from generate_series(1,10) as n; -- 2.39.0