diff options
| author | Peter Eisentraut | 2003-03-18 22:19:47 +0000 |
|---|---|---|
| committer | Peter Eisentraut | 2003-03-18 22:19:47 +0000 |
| commit | 0bd055405d6cf42741f870959b6ac2f1a4906116 (patch) | |
| tree | c0ac0d61b8f83554f0d65c16d28f7b985fe7c04a /src/bin/scripts/common.c | |
| parent | 02d13330f32b86b9914d5e596720a30dd368fde3 (diff) | |
Reimplement create and drop scripts in C, to reduce repetitive
connections, increase robustness, add NLS, and prepare for Windows port.
(vacuumdb and clusterdb will follow later.)
Diffstat (limited to 'src/bin/scripts/common.c')
| -rw-r--r-- | src/bin/scripts/common.c | 168 |
1 files changed, 168 insertions, 0 deletions
diff --git a/src/bin/scripts/common.c b/src/bin/scripts/common.c new file mode 100644 index 0000000000..da71836562 --- /dev/null +++ b/src/bin/scripts/common.c @@ -0,0 +1,168 @@ +/*------------------------------------------------------------------------- + * + * Miscellaneous shared code + * + * Portions Copyright (c) 1996-2003, PostgreSQL Global Development Group + * Portions Copyright (c) 1994, Regents of the University of California + * + * $Header$ + * + *------------------------------------------------------------------------- + */ + +#include "postgres_fe.h" +#include "common.h" + +#include <pwd.h> +#include <unistd.h> + + +/* + * Returns the current user name. + */ +const char * +get_user_name(const char *progname) +{ + struct passwd *pw; + + pw = getpwuid(getuid()); + if (!pw) + { + perror(progname); + exit(1); + } + return pw->pw_name; +} + + +/* + * Extracts the actual name of the program as called. + */ +char * +get_progname(char *argv0) +{ + if (!strrchr(argv0, '/')) + return argv0; + else + return strrchr(argv0, '/') + 1; +} + + +/* + * Initialized NLS if enabled. + */ +void +init_nls(void) +{ +#ifdef ENABLE_NLS + setlocale(LC_ALL, ""); + bindtextdomain("pgscripts", LOCALEDIR); + textdomain("pgscripts"); +#endif +} + + +/* + * Provide strictly harmonized handling of --help and --version + * options. + */ +void +handle_help_version_opts(int argc, char *argv[], const char *fixed_progname, help_handler hlp) +{ + if (argc > 1) + { + if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0) + { + hlp(get_progname(argv[0])); + exit(0); + } + if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0) + { + printf("%s (PostgreSQL) " PG_VERSION "\n", fixed_progname); + exit(0); + } + } +} + + +/* + * Make a database connection with the given parameters. An + * interactive password prompt is automatically issued if required. + */ +PGconn * +connectDatabase(const char *dbname, const char *pghost, const char *pgport, + const char *pguser, bool require_password, const char *progname) +{ + PGconn *conn; + char *password = NULL; + bool need_pass = false; + + if (require_password) + password = simple_prompt("Password: ", 100, false); + + /* + * Start the connection. Loop until we have a password if requested + * by backend. + */ + do + { + need_pass = false; + conn = PQsetdbLogin(pghost, pgport, NULL, NULL, dbname, pguser, password); + + if (!conn) + { + fprintf(stderr, _("%s: could not connect to database %s\n"), + progname, dbname); + exit(1); + } + + if (PQstatus(conn) == CONNECTION_BAD && + strcmp(PQerrorMessage(conn), "fe_sendauth: no password supplied\n") == 0 && + !feof(stdin)) + { + PQfinish(conn); + need_pass = true; + free(password); + password = NULL; + password = simple_prompt("Password: ", 100, false); + } + } while (need_pass); + + if (password) + free(password); + + /* check to see that the backend connection was successfully made */ + if (PQstatus(conn) == CONNECTION_BAD) + { + fprintf(stderr, _("%s: could not connect to database %s: %s"), + progname, dbname, PQerrorMessage(conn)); + exit(1); + } + + return conn; +} + + +/* + * Run a query, return the results, exit program on failure. + */ +PGresult * +executeQuery(PGconn *conn, const char *query, const char *progname, bool echo) +{ + PGresult *res; + + if (echo) + printf("%s\n", query); + + res = PQexec(conn, query); + if (!res || + PQresultStatus(res) != PGRES_TUPLES_OK) + { + fprintf(stderr, _("%s: query failed: %s"), progname, PQerrorMessage(conn)); + fprintf(stderr, _("%s: query was: %s\n"), progname, query); + PQfinish(conn); + exit(1); + } + + return res; +} |