mirror of https://github.com/postgres/postgres
connections, increase robustness, add NLS, and prepare for Windows port. (vacuumdb and clusterdb will follow later.)WIN32_DEV
Peter Eisentraut
23 years ago
parent
cf1cf89649
commit
9e0ab7126d
@ -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: /cvsroot/pgsql/src/bin/scripts/common.c,v 1.1 2003/03/18 22:19:46 petere Exp $ |
||||
* |
||||
*------------------------------------------------------------------------- |
||||
*/ |
||||
|
||||
#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; |
||||
} |
||||
@ -0,0 +1,35 @@ |
||||
#include "postgres_fe.h" |
||||
|
||||
#ifdef HAVE_GETOPT_H |
||||
# include <getopt.h> |
||||
#endif |
||||
|
||||
#ifndef HAVE_GETOPT_LONG |
||||
#include "getopt_long.h" |
||||
#endif |
||||
|
||||
#include "libpq-fe.h" |
||||
#include "pqexpbuffer.h" |
||||
|
||||
#ifndef HAVE_OPTRESET |
||||
int optreset; |
||||
#endif |
||||
|
||||
const char *get_user_name(const char *progname); |
||||
char *get_progname(char *argv0); |
||||
|
||||
#define _(x) gettext((x)) |
||||
void init_nls(void); |
||||
|
||||
typedef void (*help_handler)(const char *); |
||||
|
||||
void handle_help_version_opts(int argc, char *argv[], const char *fixed_progname, help_handler hlp); |
||||
|
||||
extern char *simple_prompt(const char *prompt, int maxlen, bool echo); |
||||
|
||||
PGconn * |
||||
connectDatabase(const char *dbname, const char *pghost, const char *pgport, |
||||
const char *pguser, bool require_password, const char *progname); |
||||
|
||||
PGresult * |
||||
executeQuery(PGconn *conn, const char *command, const char *progname, bool echo); |
||||
@ -1,212 +0,0 @@ |
||||
#!/bin/sh |
||||
#------------------------------------------------------------------------- |
||||
# |
||||
# createdb-- |
||||
# create a postgres database |
||||
# |
||||
# This program runs psql with the "-c" option to create |
||||
# the requested database. |
||||
# |
||||
# Portions Copyright (c) 1996-2002, PostgreSQL Global Development Group |
||||
# Portions Copyright (c) 1994, Regents of the University of California |
||||
# |
||||
# |
||||
# IDENTIFICATION |
||||
# $Header: /cvsroot/pgsql/src/bin/scripts/Attic/createdb,v 1.28 2002/10/18 22:05:36 petere Exp $ |
||||
# |
||||
#------------------------------------------------------------------------- |
||||
|
||||
CMDNAME=`basename "$0"` |
||||
PATHNAME=`echo "$0" | sed "s,$CMDNAME\$,,"` |
||||
|
||||
MB= |
||||
TEMPLATE= |
||||
PSQLOPT= |
||||
dbname= |
||||
dbowner= |
||||
dbcomment= |
||||
dbpath= |
||||
|
||||
while [ "$#" -gt 0 ] |
||||
do |
||||
case "$1" in |
||||
--help|-\?) |
||||
usage=t |
||||
break |
||||
;; |
||||
# options passed on to psql |
||||
--host|-h) |
||||
PSQLOPT="$PSQLOPT -h $2" |
||||
shift;; |
||||
-h*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--host=*) |
||||
PSQLOPT="$PSQLOPT -h `echo \"$1\" | sed 's/^--host=//'`" |
||||
;; |
||||
--port|-p) |
||||
PSQLOPT="$PSQLOPT -p $2" |
||||
shift;; |
||||
-p*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--port=*) |
||||
PSQLOPT="$PSQLOPT -p `echo \"$1\" | sed 's/^--port=//'`" |
||||
;; |
||||
--username|-U) |
||||
PSQLOPT="$PSQLOPT -U $2" |
||||
shift;; |
||||
-U*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--username=*) |
||||
PSQLOPT="$PSQLOPT -U `echo \"$1\" | sed 's/^--username=//'`" |
||||
;; |
||||
--password|-W) |
||||
PSQLOPT="$PSQLOPT -W" |
||||
;; |
||||
--echo|-e) |
||||
PSQLOPT="$PSQLOPT -e" |
||||
;; |
||||
--quiet|-q) |
||||
PSQLOPT="$PSQLOPT -o /dev/null" |
||||
;; |
||||
# options converted into SQL command |
||||
--owner|-O) |
||||
dbowner="$2" |
||||
shift;; |
||||
-O*) |
||||
dbowner=`echo "$1" | sed 's/^-O//'` |
||||
;; |
||||
--owner=*) |
||||
dbowner=`echo "$1" | sed 's/^--owner=//'` |
||||
;; |
||||
--location|-D) |
||||
dbpath="$2" |
||||
shift;; |
||||
-D*) |
||||
dbpath=`echo "$1" | sed 's/^-D//'` |
||||
;; |
||||
--location=*) |
||||
dbpath=`echo "$1" | sed 's/^--location=//'` |
||||
;; |
||||
--template|-T) |
||||
TEMPLATE="$2" |
||||
shift;; |
||||
-T*) |
||||
TEMPLATE=`echo "$1" | sed 's/^-T//'` |
||||
;; |
||||
--template=*) |
||||
TEMPLATE=`echo "$1" | sed 's/^--template=//'` |
||||
;; |
||||
--encoding|-E) |
||||
MB="$2" |
||||
shift;; |
||||
-E*) |
||||
MB=`echo "$1" | sed 's/^-E//'` |
||||
;; |
||||
--encoding=*) |
||||
MB=`echo "$1" | sed 's/^--encoding=//'` |
||||
;; |
||||
-*) |
||||
echo "$CMDNAME: invalid option: $1" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
;; |
||||
*) |
||||
dbname="$1" |
||||
if [ "$2" ] |
||||
then |
||||
shift |
||||
dbcomment="$1" |
||||
fi |
||||
if [ "$#" -ne 1 ]; then |
||||
echo "$CMDNAME: invalid option: $2" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
fi |
||||
;; |
||||
esac |
||||
shift |
||||
done |
||||
|
||||
if [ "$usage" ]; then |
||||
echo "$CMDNAME creates a PostgreSQL database." |
||||
echo |
||||
echo "Usage:" |
||||
echo " $CMDNAME [OPTION]... [DBNAME] [DESCRIPTION]" |
||||
echo |
||||
echo "Options:" |
||||
echo " -D, --location=PATH alternative place to store the database" |
||||
echo " -E, --encoding=ENCODING encoding for the database" |
||||
echo " -O, --owner=OWNER database user to own the new database" |
||||
echo " -T, --template=TEMPLATE template database to copy" |
||||
echo " -e, --echo show the query being sent to the backend" |
||||
echo " -q, --quiet don't write any messages" |
||||
echo " --help show this help, then exit" |
||||
echo |
||||
echo "Connection options:" |
||||
echo " -h, --host=HOSTNAME database server host" |
||||
echo " -p, --port=PORT database server port" |
||||
echo " -U, --username=USERNAME user name to connect as" |
||||
echo " -W, --password prompt for password" |
||||
echo |
||||
echo "By default, a database with the same name as the current user is created." |
||||
echo |
||||
echo "Report bugs to <pgsql-bugs@postgresql.org>." |
||||
exit 0 |
||||
fi |
||||
|
||||
|
||||
if [ -n "$MB" ] |
||||
then |
||||
mbcode=`${PATHNAME}pg_encoding "$MB"` |
||||
if [ -z "$mbcode" ] |
||||
then |
||||
echo "$CMDNAME: \"$MB\" is not a valid encoding name" 1>&2 |
||||
exit 1 |
||||
fi |
||||
fi |
||||
|
||||
if [ -z "$dbname" ]; then |
||||
if [ "$PGDATABASE" ]; then |
||||
dbname="$PGDATABASE" |
||||
elif [ "$PGUSER" ]; then |
||||
dbname="$PGUSER" |
||||
else |
||||
dbname=`${PATHNAME}pg_id -u -n` |
||||
fi |
||||
[ "$?" -ne 0 ] && exit 1 |
||||
fi |
||||
|
||||
|
||||
# escape the quotes |
||||
dbpath=`echo "$dbpath" | sed "s/'/\\\\\'/g"` |
||||
dbname=`echo "$dbname" | sed 's/\"/\\\"/g'` |
||||
TEMPLATE=`echo "$TEMPLATE" | sed 's/\"/\"\"/g'` |
||||
|
||||
withstring= |
||||
[ "$dbowner" ] && withstring="$withstring OWNER = \"$dbowner\"" |
||||
[ "$dbpath" ] && withstring="$withstring LOCATION = '$dbpath'" |
||||
[ "$MB" ] && withstring="$withstring ENCODING = '$MB'" |
||||
[ "$TEMPLATE" ] && withstring="$withstring TEMPLATE = \"$TEMPLATE\"" |
||||
[ "$withstring" ] && withstring=" WITH$withstring" |
||||
|
||||
${PATHNAME}psql $PSQLOPT -d template1 -c "SET autocommit TO 'on';CREATE DATABASE \"$dbname\"$withstring" |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: database creation failed" 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
# Insert comment as well, if requested |
||||
[ -z "$dbcomment" ] && exit 0 |
||||
|
||||
dbcomment=`echo "$dbcomment" | sed "s/'/\\\\\'/g"` |
||||
|
||||
${PATHNAME}psql $PSQLOPT -d "$dbname" -c "SET autocommit TO 'on';COMMENT ON DATABASE \"$dbname\" IS '$dbcomment'" |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: comment creation failed (database was created)" 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
exit 0 |
||||
@ -0,0 +1,232 @@ |
||||
/*-------------------------------------------------------------------------
|
||||
* |
||||
* createdb |
||||
* |
||||
* Portions Copyright (c) 1996-2003, PostgreSQL Global Development Group |
||||
* Portions Copyright (c) 1994, Regents of the University of California |
||||
* |
||||
* $Header: /cvsroot/pgsql/src/bin/scripts/createdb.c,v 1.1 2003/03/18 22:19:46 petere Exp $ |
||||
* |
||||
*------------------------------------------------------------------------- |
||||
*/ |
||||
|
||||
#include "postgres_fe.h" |
||||
#include "common.h" |
||||
#include "dumputils.h" |
||||
|
||||
#include "mb/pg_wchar.h" |
||||
|
||||
|
||||
static void help(const char *progname); |
||||
|
||||
|
||||
int |
||||
main(int argc, char *argv[]) |
||||
{ |
||||
static struct option long_options[] = { |
||||
{"host", required_argument, NULL, 'h'}, |
||||
{"port", required_argument, NULL, 'p'}, |
||||
{"username", required_argument, NULL, 'U'}, |
||||
{"password", no_argument, NULL, 'W'}, |
||||
{"echo", no_argument, NULL, 'e'}, |
||||
{"quiet", no_argument, NULL, 'q'}, |
||||
{"owner", required_argument, NULL, 'O'}, |
||||
{"location", required_argument, NULL, 'D'}, |
||||
{"template", required_argument, NULL, 'T'}, |
||||
{"encoding", required_argument, NULL, 'E'}, |
||||
{NULL, 0, NULL, 0} |
||||
}; |
||||
|
||||
char *progname; |
||||
int optindex; |
||||
int c; |
||||
|
||||
const char *dbname = NULL; |
||||
char *comment = NULL; |
||||
char *host = NULL; |
||||
char *port = NULL; |
||||
char *username = NULL; |
||||
bool password = false; |
||||
bool echo = false; |
||||
bool quiet = false; |
||||
char *owner = NULL; |
||||
char *location = NULL; |
||||
char *template = NULL; |
||||
char *encoding = NULL; |
||||
|
||||
PQExpBufferData sql; |
||||
|
||||
PGconn *conn; |
||||
PGresult *result; |
||||
|
||||
progname = get_progname(argv[0]); |
||||
init_nls(); |
||||
handle_help_version_opts(argc, argv, "createdb", help); |
||||
|
||||
while ((c = getopt_long(argc, argv, "h:p:U:WeqO:D:T:E:", long_options, &optindex)) != -1) |
||||
{ |
||||
switch (c) |
||||
{ |
||||
case 'h': |
||||
host = optarg; |
||||
break; |
||||
case 'p': |
||||
port = optarg; |
||||
break; |
||||
case 'U': |
||||
username = optarg; |
||||
break; |
||||
case 'W': |
||||
password = true; |
||||
break; |
||||
case 'e': |
||||
echo = true; |
||||
break; |
||||
case 'q': |
||||
quiet = true; |
||||
break; |
||||
case 'O': |
||||
owner = optarg; |
||||
break; |
||||
case 'D': |
||||
location = optarg; |
||||
break; |
||||
case 'T': |
||||
template = optarg; |
||||
break; |
||||
case 'E': |
||||
encoding = optarg; |
||||
break; |
||||
default: |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
} |
||||
|
||||
switch (argc - optind) |
||||
{ |
||||
case 0: |
||||
break; |
||||
case 1: |
||||
dbname = argv[optind]; |
||||
break; |
||||
case 2: |
||||
dbname = argv[optind]; |
||||
comment = argv[optind + 1]; |
||||
break; |
||||
default: |
||||
fprintf(stderr, _("%s: too many command-line arguments (first is '%s')\n"), |
||||
progname, argv[optind + 2]); |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
|
||||
if (encoding) |
||||
{ |
||||
if (pg_char_to_encoding(encoding) < 0) |
||||
{ |
||||
fprintf(stderr, _("%s: \"%s\" is not a valid encoding name\n"), |
||||
progname, encoding); |
||||
exit(1); |
||||
} |
||||
} |
||||
|
||||
if (dbname == NULL) |
||||
{ |
||||
if (getenv("PGDATABASE")) |
||||
dbname = getenv("PGDATABASE"); |
||||
else if (getenv("PGUSER")) |
||||
dbname = getenv("PGUSER"); |
||||
else |
||||
dbname = get_user_name(progname); |
||||
} |
||||
|
||||
initPQExpBuffer(&sql); |
||||
|
||||
appendPQExpBuffer(&sql, "SET autocommit TO on;\nCREATE DATABASE %s", |
||||
fmtId(dbname)); |
||||
|
||||
if (owner) |
||||
appendPQExpBuffer(&sql, " OWNER %s", fmtId(owner)); |
||||
if (location) |
||||
{ |
||||
appendPQExpBuffer(&sql, " LOCATION "); |
||||
appendStringLiteral(&sql, location, false); |
||||
} |
||||
if (encoding) |
||||
appendPQExpBuffer(&sql, " ENCODING '%s'", encoding); |
||||
if (template) |
||||
appendPQExpBuffer(&sql, " TEMPLATE %s", fmtId(template)); |
||||
appendPQExpBuffer(&sql, ";\n"); |
||||
|
||||
conn = connectDatabase("template1", host, port, username, password, progname); |
||||
|
||||
if (echo) |
||||
printf("%s", sql.data); |
||||
result = PQexec(conn, sql.data); |
||||
|
||||
if (PQresultStatus(result) != PGRES_COMMAND_OK) |
||||
{ |
||||
fprintf(stderr, _("%s: database creation failed: %s"), |
||||
progname, PQerrorMessage(conn)); |
||||
PQfinish(conn); |
||||
exit(1); |
||||
} |
||||
|
||||
PQclear(result); |
||||
PQfinish(conn); |
||||
|
||||
if (!quiet) |
||||
puts("CREATE DATABASE"); |
||||
|
||||
if (comment) |
||||
{ |
||||
printfPQExpBuffer(&sql, "SET autocommit TO on;\nCOMMENT ON DATABASE %s IS ", fmtId(dbname)); |
||||
appendStringLiteral(&sql, comment, false); |
||||
appendPQExpBuffer(&sql, ";\n"); |
||||
|
||||
conn = connectDatabase(dbname, host, port, username, password, progname); |
||||
if (echo) |
||||
printf("%s", sql.data); |
||||
result = PQexec(conn, sql.data); |
||||
|
||||
if (PQresultStatus(result) != PGRES_COMMAND_OK) |
||||
{ |
||||
fprintf(stderr, _("%s: comment creation failed (database was created): %s"), |
||||
progname, PQerrorMessage(conn)); |
||||
PQfinish(conn); |
||||
exit(1); |
||||
} |
||||
|
||||
PQfinish(conn); |
||||
if (!quiet) |
||||
puts("COMMENT"); |
||||
} |
||||
|
||||
exit(0); |
||||
} |
||||
|
||||
|
||||
static void |
||||
help(const char *progname) |
||||
{ |
||||
printf(_("%s creates a PostgreSQL database.\n\n"), progname); |
||||
printf(_("Usage:\n")); |
||||
printf(_(" %s [OPTION]... [DBNAME] [DESCRIPTION]\n"), progname); |
||||
printf(_("\nOptions:\n")); |
||||
printf(_(" -D, --location=PATH alternative place to store the database\n")); |
||||
printf(_(" -E, --encoding=ENCODING encoding for the database\n")); |
||||
printf(_(" -O, --owner=OWNER database user to own the new database\n")); |
||||
printf(_(" -T, --template=TEMPLATE template database to copy\n")); |
||||
printf(_(" -e, --echo show the commands being sent to the server\n")); |
||||
printf(_(" -q, --quiet don't write any messages\n")); |
||||
printf(_(" --help show this help, then exit\n")); |
||||
printf(_(" --version output version information, then exit\n")); |
||||
printf(_("\nConnection options:\n")); |
||||
printf(_(" -h, --host=HOSTNAME database server host\n")); |
||||
printf(_(" -p, --port=PORT database server port\n")); |
||||
printf(_(" -U, --username=USERNAME user name to connect as\n")); |
||||
printf(_(" -W, --password prompt for password\n")); |
||||
printf(_("\nBy default, a database with the same name as the current user is created.\n")); |
||||
printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n")); |
||||
} |
||||
@ -0,0 +1,286 @@ |
||||
/*-------------------------------------------------------------------------
|
||||
* |
||||
* createlang |
||||
* |
||||
* Portions Copyright (c) 1996-2003, PostgreSQL Global Development Group |
||||
* Portions Copyright (c) 1994, Regents of the University of California |
||||
* |
||||
* $Header: /cvsroot/pgsql/src/bin/scripts/createlang.c,v 1.1 2003/03/18 22:19:46 petere Exp $ |
||||
* |
||||
*------------------------------------------------------------------------- |
||||
*/ |
||||
|
||||
#include "postgres_fe.h" |
||||
#include "common.h" |
||||
#include "print.h" |
||||
|
||||
|
||||
static void help(const char *progname); |
||||
|
||||
|
||||
int |
||||
main(int argc, char *argv[]) |
||||
{ |
||||
static struct option long_options[] = { |
||||
{"list", no_argument, NULL, 'l'}, |
||||
{"host", required_argument, NULL, 'h'}, |
||||
{"port", required_argument, NULL, 'p'}, |
||||
{"username", required_argument, NULL, 'U'}, |
||||
{"password", no_argument, NULL, 'W'}, |
||||
{"dbname", required_argument, NULL, 'd'}, |
||||
{"pglib", required_argument, NULL, 'L'}, |
||||
{"echo", no_argument, NULL, 'e'}, |
||||
{NULL, 0, NULL, 0} |
||||
}; |
||||
|
||||
char *progname; |
||||
int optindex; |
||||
int c; |
||||
|
||||
bool listlangs = false; |
||||
const char *dbname = NULL; |
||||
char *host = NULL; |
||||
char *port = NULL; |
||||
char *username = NULL; |
||||
bool password = false; |
||||
bool echo = false; |
||||
char *pglib = NULL; |
||||
char *langname = NULL; |
||||
|
||||
char *p; |
||||
bool handlerexists; |
||||
bool trusted; |
||||
char *handler; |
||||
char *object; |
||||
|
||||
PQExpBufferData sql; |
||||
|
||||
PGconn *conn; |
||||
PGresult *result; |
||||
|
||||
progname = get_progname(argv[0]); |
||||
init_nls(); |
||||
handle_help_version_opts(argc, argv, "createlang", help); |
||||
|
||||
while ((c = getopt_long(argc, argv, "lh:p:U:Wd:L:e", long_options, &optindex)) != -1) |
||||
{ |
||||
switch (c) |
||||
{ |
||||
case 'l': |
||||
listlangs = true; |
||||
break; |
||||
case 'h': |
||||
host = optarg; |
||||
break; |
||||
case 'p': |
||||
port = optarg; |
||||
break; |
||||
case 'U': |
||||
username = optarg; |
||||
break; |
||||
case 'W': |
||||
password = true; |
||||
break; |
||||
case 'd': |
||||
dbname = optarg; |
||||
break; |
||||
case 'L': |
||||
pglib = optarg; |
||||
break; |
||||
case 'e': |
||||
echo = true; |
||||
break; |
||||
default: |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
} |
||||
|
||||
if (argc - optind > 0) |
||||
{ |
||||
if (listlangs) |
||||
dbname = argv[optind++]; |
||||
else |
||||
{ |
||||
langname = argv[optind++]; |
||||
if (argc - optind > 0) |
||||
dbname = argv[optind++]; |
||||
} |
||||
} |
||||
|
||||
if (argc - optind > 0) |
||||
{ |
||||
fprintf(stderr, _("%s: too many command-line arguments (first is '%s')\n"), |
||||
progname, argv[optind]); |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
|
||||
if (dbname == NULL) |
||||
{ |
||||
if (getenv("PGDATABASE")) |
||||
dbname = getenv("PGDATABASE"); |
||||
else if (getenv("PGUSER")) |
||||
dbname = getenv("PGUSER"); |
||||
else |
||||
dbname = get_user_name(progname); |
||||
} |
||||
|
||||
initPQExpBuffer(&sql); |
||||
|
||||
/*
|
||||
* List option |
||||
*/ |
||||
if (listlangs) |
||||
{ |
||||
printQueryOpt popt; |
||||
|
||||
conn = connectDatabase(dbname, host, port, username, password, progname); |
||||
|
||||
printfPQExpBuffer(&sql, "SELECT lanname as \"%s\", (CASE WHEN lanpltrusted THEN '%s' ELSE '%s' END) as \"%s\" FROM pg_language WHERE lanispl IS TRUE;", _("Name"), _("yes"), _("no"), _("Trusted?")); |
||||
result = executeQuery(conn, sql.data, progname, echo); |
||||
|
||||
memset(&popt, 0, sizeof(popt)); |
||||
popt.topt.format = PRINT_ALIGNED; |
||||
popt.topt.border = 1; |
||||
popt.topt.encoding = PQclientEncoding(conn); |
||||
popt.title = _("Procedural Languages"); |
||||
printQuery(result, &popt, stdout); |
||||
|
||||
PQfinish(conn); |
||||
exit(0); |
||||
} |
||||
|
||||
if (langname == NULL) |
||||
{ |
||||
fprintf(stderr, _("%s: missing required argument language name\n"), progname); |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
|
||||
if (!pglib) |
||||
pglib = "$libdir"; |
||||
|
||||
for (p = langname; *p; p++) |
||||
if (*p >= 'A' && *p <= 'Z') |
||||
*p += ('a' - 'A'); |
||||
|
||||
if (strcmp(langname, "plpgsql")==0) |
||||
{ |
||||
trusted = true; |
||||
handler = "plpgsql_call_handler"; |
||||
object = "plpgsql"; |
||||
} |
||||
else if (strcmp(langname, "pltcl")==0) |
||||
{ |
||||
trusted = true; |
||||
handler = "pltcl_call_handler"; |
||||
object = "pltcl"; |
||||
} |
||||
else if (strcmp(langname, "pltclu")==0) |
||||
{ |
||||
trusted = false; |
||||
handler = "pltclu_call_handler"; |
||||
object = "pltcl"; |
||||
} |
||||
else if (strcmp(langname, "plperl")==0) |
||||
{ |
||||
trusted = true; |
||||
handler = "plperl_call_handler"; |
||||
object = "plperl"; |
||||
} |
||||
else if (strcmp(langname, "plperlu")==0) |
||||
{ |
||||
trusted = false; |
||||
handler = "plperl_call_handler"; |
||||
object = "plperl"; |
||||
} |
||||
else if (strcmp(langname, "plpython")==0) |
||||
{ |
||||
trusted = true; |
||||
handler = "plpython_call_handler"; |
||||
object = "plpython"; |
||||
} |
||||
else |
||||
{ |
||||
fprintf(stderr, _("%s: unsupported language \"%s\"\n"), progname, langname); |
||||
fprintf(stderr, _("Supported languages are plpgsql, pltcl, pltclu, plperl, plperlu, and plpython.\n")); |
||||
exit(1); |
||||
} |
||||
|
||||
conn = connectDatabase(dbname, host, port, username, password, progname); |
||||
|
||||
/*
|
||||
* Make sure the language isn't already installed |
||||
*/ |
||||
printfPQExpBuffer(&sql, "SELECT oid FROM pg_language WHERE lanname = '%s';", langname); |
||||
result = executeQuery(conn, sql.data, progname, echo); |
||||
if (PQntuples(result) > 0) |
||||
{ |
||||
PQfinish(conn); |
||||
fprintf(stderr, |
||||
_("%s: language \"%s\" is already installed in database \"%s\"\n"), |
||||
progname, langname, dbname); |
||||
/* separate exit status for "already installed" */ |
||||
exit(2); |
||||
} |
||||
PQclear(result); |
||||
|
||||
/*
|
||||
* Check whether the call handler exists |
||||
*/ |
||||
printfPQExpBuffer(&sql, "SELECT oid FROM pg_proc WHERE proname = '%s' AND prorettype = (SELECT oid FROM pg_type WHERE typname = 'language_handler') AND pronargs = 0;", handler); |
||||
result = executeQuery(conn, sql.data, progname, echo); |
||||
handlerexists = (PQntuples(result) > 0); |
||||
PQclear(result); |
||||
|
||||
/*
|
||||
* Create the call handler and the language |
||||
*/ |
||||
printfPQExpBuffer(&sql, "SET autocommit TO on;\n"); |
||||
|
||||
if (!handlerexists) |
||||
appendPQExpBuffer(&sql, |
||||
"CREATE FUNCTION \"%s\" () RETURNS language_handler AS '%s/%s' LANGUAGE C;\n", |
||||
handler, pglib, object); |
||||
|
||||
appendPQExpBuffer(&sql, |
||||
"CREATE %sLANGUAGE \"%s\" HANDLER \"%s\";\n", |
||||
(trusted ? "TRUSTED " : ""), langname, handler); |
||||
|
||||
if (echo) |
||||
printf("%s", sql.data); |
||||
result = PQexec(conn, sql.data); |
||||
if (PQresultStatus(result) != PGRES_COMMAND_OK) |
||||
{ |
||||
fprintf(stderr, _("%s: language installation failed: %s"), |
||||
progname, PQerrorMessage(conn)); |
||||
PQfinish(conn); |
||||
exit(1); |
||||
} |
||||
|
||||
PQfinish(conn); |
||||
exit(0); |
||||
} |
||||
|
||||
|
||||
|
||||
static void |
||||
help(const char *progname) |
||||
{ |
||||
printf(_("%s installs a procedural language into a PostgreSQL database.\n\n"), progname); |
||||
printf(_("Usage:\n")); |
||||
printf(_(" %s [OPTION]... LANGNAME [DBNAME]\n"), progname); |
||||
printf(_("\nOptions:\n")); |
||||
printf(_(" -d, --dbname=DBNAME database to install language in\n")); |
||||
printf(_(" -e, --echo show the commands being sent to the server\n")); |
||||
printf(_(" -l, --list show a list of currently installed languages\n")); |
||||
printf(_(" -L, --pglib=DIRECTORY find language interpreter file in DIRECTORY\n")); |
||||
printf(_(" -h, --host=HOSTNAME database server host\n")); |
||||
printf(_(" -p, --port=PORT database server port\n")); |
||||
printf(_(" -U, --username=USERNAME user name to connect as\n")); |
||||
printf(_(" -W, --password prompt for password\n")); |
||||
printf(_(" --help show this help, then exit\n")); |
||||
printf(_(" --version output version information, then exit\n")); |
||||
printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n")); |
||||
} |
||||
@ -1,312 +0,0 @@ |
||||
#! /bin/sh |
||||
#------------------------------------------------------------------------- |
||||
# |
||||
# createlang -- |
||||
# Install a procedural language in a database |
||||
# |
||||
# Portions Copyright (c) 1996-2002, PostgreSQL Global Development Group |
||||
# Portions Copyright (c) 1994, Regents of the University of California |
||||
# |
||||
# $Header: /cvsroot/pgsql/src/bin/scripts/Attic/createlang.sh,v 1.42 2002/10/18 22:05:36 petere Exp $ |
||||
# |
||||
#------------------------------------------------------------------------- |
||||
|
||||
CMDNAME=`basename "$0"` |
||||
PATHNAME=`echo "$0" | sed "s,$CMDNAME\$,,"` |
||||
|
||||
PSQLOPT= |
||||
dbname= |
||||
langname= |
||||
list= |
||||
showsql= |
||||
PGLIB= |
||||
|
||||
# Check for echo -n vs echo \c |
||||
|
||||
if echo '\c' | grep -s c >/dev/null 2>&1 |
||||
then |
||||
ECHO_N="echo -n" |
||||
ECHO_C="" |
||||
else |
||||
ECHO_N="echo" |
||||
ECHO_C='\c' |
||||
fi |
||||
|
||||
|
||||
|
||||
# ---------- |
||||
# Get options, language name and dbname |
||||
# ---------- |
||||
while [ "$#" -gt 0 ] |
||||
do |
||||
case "$1" in |
||||
--help|-\?) |
||||
usage=t |
||||
break |
||||
;; |
||||
--list|-l) |
||||
list=t |
||||
;; |
||||
# options passed on to psql |
||||
--host|-h) |
||||
PSQLOPT="$PSQLOPT -h $2" |
||||
shift;; |
||||
-h*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--host=*) |
||||
PSQLOPT="$PSQLOPT -h `echo \"$1\" | sed 's/^--host=//'`" |
||||
;; |
||||
--port|-p) |
||||
PSQLOPT="$PSQLOPT -p $2" |
||||
shift;; |
||||
-p*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--port=*) |
||||
PSQLOPT="$PSQLOPT -p `echo \"$1\" | sed 's/^--port=//'`" |
||||
;; |
||||
--username|-U) |
||||
PSQLOPT="$PSQLOPT -U $2" |
||||
shift;; |
||||
-U*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--username=*) |
||||
PSQLOPT="$PSQLOPT -U `echo \"$1\" | sed 's/^--username=//'`" |
||||
;; |
||||
--password|-W) |
||||
PSQLOPT="$PSQLOPT -W" |
||||
;; |
||||
--dbname|-d) |
||||
dbname="$2" |
||||
shift;; |
||||
-d*) |
||||
dbname=`echo "$1" | sed 's/^-d//'` |
||||
;; |
||||
--dbname=*) |
||||
dbname=`echo "$1" | sed 's/^--dbname=//'` |
||||
;; |
||||
# misc options |
||||
--pglib|-L) |
||||
PGLIB="$2" |
||||
shift;; |
||||
-L*) |
||||
PGLIB=`echo "$1" | sed 's/^-L//'` |
||||
;; |
||||
--pglib=*) |
||||
PGLIB=`echo "$1" | sed 's/^--pglib=//'` |
||||
;; |
||||
--echo|-e) |
||||
showsql=yes |
||||
;; |
||||
|
||||
-*) |
||||
echo "$CMDNAME: invalid option: $1" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
;; |
||||
*) |
||||
if [ "$list" != "t" ] |
||||
then langname="$1" |
||||
if [ "$2" ] |
||||
then |
||||
shift |
||||
dbname="$1" |
||||
fi |
||||
else dbname="$1" |
||||
fi |
||||
if [ "$#" -ne 1 ]; then |
||||
echo "$CMDNAME: invalid option: $2" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
fi |
||||
;; |
||||
esac |
||||
shift |
||||
done |
||||
|
||||
if [ -n "$usage" ]; then |
||||
echo "$CMDNAME installs a procedural language into a PostgreSQL database." |
||||
echo |
||||
echo "Usage:" |
||||
echo " $CMDNAME [OPTION]... LANGNAME [DBNAME]" |
||||
echo |
||||
echo "Options:" |
||||
echo " -d, --dbname=DBNAME database to install language in" |
||||
echo " -l, --list show a list of currently installed languages" |
||||
echo " -L, --pglib=DIRECTORY find language interpreter file in DIRECTORY" |
||||
echo " -h, --host=HOSTNAME database server host" |
||||
echo " -p, --port=PORT database server port" |
||||
echo " -U, --username=USERNAME user name to connect as" |
||||
echo " -W, --password prompt for password" |
||||
echo " --help show this help, then exit" |
||||
echo |
||||
echo "Report bugs to <pgsql-bugs@postgresql.org>." |
||||
exit 0 |
||||
fi |
||||
|
||||
|
||||
if [ -z "$dbname" ]; then |
||||
if [ "$PGDATABASE" ]; then |
||||
dbname="$PGDATABASE" |
||||
elif [ "$PGUSER" ]; then |
||||
dbname="$PGUSER" |
||||
else |
||||
dbname=`${PATHNAME}pg_id -u -n` |
||||
fi |
||||
[ "$?" -ne 0 ] && exit 1 |
||||
fi |
||||
|
||||
|
||||
# ---------- |
||||
# List option, doesn't need langname |
||||
# ---------- |
||||
if [ "$list" ]; then |
||||
sqlcmd="SELECT lanname as \"Name\", lanpltrusted as \"Trusted?\" FROM pg_language WHERE lanispl = TRUE;" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
${PATHNAME}psql $PSQLOPT -d "$dbname" -P 'title=Procedural languages' -c "$sqlcmd" |
||||
exit $? |
||||
fi |
||||
|
||||
|
||||
# ---------- |
||||
# We can't go any farther without a langname |
||||
# ---------- |
||||
if [ -z "$langname" ]; then |
||||
echo "$CMDNAME: missing required argument language name" 1>&2 |
||||
echo "Try '$CMDNAME --help' for help." 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
# ---------- |
||||
# Check that we have PGLIB |
||||
# ---------- |
||||
if [ -z "$PGLIB" ]; then |
||||
PGLIB='$libdir' |
||||
fi |
||||
|
||||
# ---------- |
||||
# Check if supported and set related values |
||||
# ---------- |
||||
|
||||
langname=`echo "$langname" | tr 'ABCDEFGHIJKLMNOPQRSTUVWXYZ' 'abcdefghijklmnopqrstuvwxyz'` |
||||
|
||||
case "$langname" in |
||||
plpgsql) |
||||
trusted="TRUSTED " |
||||
handler="plpgsql_call_handler" |
||||
object="plpgsql" |
||||
;; |
||||
pltcl) |
||||
trusted="TRUSTED " |
||||
handler="pltcl_call_handler" |
||||
object="pltcl" |
||||
;; |
||||
pltclu) |
||||
trusted="" |
||||
handler="pltclu_call_handler" |
||||
object="pltcl" |
||||
;; |
||||
plperl) |
||||
trusted="TRUSTED " |
||||
handler="plperl_call_handler" |
||||
object="plperl" |
||||
;; |
||||
plperlu) |
||||
trusted="" |
||||
handler="plperl_call_handler" |
||||
object="plperl" |
||||
;; |
||||
plpython) |
||||
trusted="TRUSTED " |
||||
handler="plpython_call_handler" |
||||
object="plpython" |
||||
;; |
||||
*) |
||||
echo "$CMDNAME: unsupported language \"$langname\"" 1>&2 |
||||
echo "Supported languages are plpgsql, pltcl, pltclu, plperl, plperlu, and plpython." 1>&2 |
||||
exit 1 |
||||
;; |
||||
esac |
||||
|
||||
|
||||
PSQL="${PATHNAME}psql -A -t -q $PSQLOPT -d $dbname -c" |
||||
|
||||
# ---------- |
||||
# Make sure the language isn't already installed |
||||
# ---------- |
||||
sqlcmd="SELECT oid FROM pg_language WHERE lanname = '$langname';" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
res=`$PSQL "$sqlcmd"` |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: external error" 1>&2 |
||||
exit 1 |
||||
fi |
||||
if [ -n "$res" ]; then |
||||
echo "$CMDNAME: language \"$langname\" is already installed in database $dbname" 1>&2 |
||||
# separate exit status for "already installed" |
||||
exit 2 |
||||
fi |
||||
|
||||
# ---------- |
||||
# Check whether the call handler exists |
||||
# ---------- |
||||
sqlcmd="SELECT oid FROM pg_proc WHERE proname = '$handler' AND prorettype = 0 AND pronargs = 0;" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
res=`$PSQL "$sqlcmd"` |
||||
if [ -n "$res" ]; then |
||||
handlerexists=yes |
||||
else |
||||
handlerexists=no |
||||
fi |
||||
|
||||
# ---------- |
||||
# Create the call handler and the language |
||||
# ---------- |
||||
if [ "$handlerexists" = no ]; then |
||||
sqlcmd="SET autocommit TO 'on';CREATE FUNCTION \"$handler\" () RETURNS LANGUAGE_HANDLER AS '$PGLIB/${object}' LANGUAGE C;" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
$PSQL "$sqlcmd" |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: language installation failed" 1>&2 |
||||
exit 1 |
||||
fi |
||||
fi |
||||
|
||||
sqlcmd="SET autocommit TO 'on';CREATE ${trusted}LANGUAGE \"$langname\" HANDLER \"$handler\";" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
$PSQL "$sqlcmd" |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: language installation failed" 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
# ---------- |
||||
# Grant privileges. As of 7.3 the default privileges for a language include |
||||
# public USAGE, so we need not change them for a trusted language; but it |
||||
# seems best to disable public USAGE for an untrusted one. |
||||
# ---------- |
||||
if test -z "$trusted"; then |
||||
sqlcmd="SET autocommit TO 'on';REVOKE ALL ON LANGUAGE \"$langname\" FROM PUBLIC;" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
$PSQL "$sqlcmd" |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: language installation failed" 1>&2 |
||||
exit 1 |
||||
fi |
||||
fi |
||||
|
||||
exit 0 |
||||
@ -1,265 +0,0 @@ |
||||
#!/bin/sh |
||||
#------------------------------------------------------------------------- |
||||
# |
||||
# createuser-- |
||||
# Utility for creating a user in the PostgreSQL database |
||||
# |
||||
# Portions Copyright (c) 1996-2002, PostgreSQL Global Development Group |
||||
# Portions Copyright (c) 1994, Regents of the University of California |
||||
# |
||||
# |
||||
# IDENTIFICATION |
||||
# $Header: /cvsroot/pgsql/src/bin/scripts/Attic/createuser,v 1.31 2003/02/19 03:52:57 momjian Exp $ |
||||
# |
||||
# Note - this should NOT be setuid. |
||||
# |
||||
#------------------------------------------------------------------------- |
||||
|
||||
CMDNAME=`basename "$0"` |
||||
PATHNAME=`echo "$0" | sed "s,$CMDNAME\$,,"` |
||||
|
||||
NewUser= |
||||
SysID= |
||||
CanAddUser= |
||||
CanCreateDb= |
||||
PwPrompt= |
||||
Password= |
||||
PSQLOPT= |
||||
Encrypted= # blank uses default |
||||
|
||||
# These handle spaces/tabs in identifiers |
||||
_IFS="$IFS" |
||||
NL=" |
||||
" |
||||
# Check for echo -n vs echo \c |
||||
|
||||
if echo '\c' | grep -s c >/dev/null 2>&1 |
||||
then |
||||
ECHO_N="echo -n" |
||||
ECHO_C="" |
||||
else |
||||
ECHO_N="echo" |
||||
ECHO_C='\c' |
||||
fi |
||||
|
||||
|
||||
while [ "$#" -gt 0 ] |
||||
do |
||||
case "$1" in |
||||
--help|-\?) |
||||
usage=t |
||||
break |
||||
;; |
||||
# options passed on to psql |
||||
--host|-h) |
||||
PSQLOPT="$PSQLOPT -h $2" |
||||
shift;; |
||||
-h*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--host=*) |
||||
PSQLOPT="$PSQLOPT -h `echo \"$1\" | sed 's/^--host=//'`" |
||||
;; |
||||
--port|-p) |
||||
PSQLOPT="$PSQLOPT -p $2" |
||||
shift;; |
||||
-p*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--port=*) |
||||
PSQLOPT="$PSQLOPT -p `echo \"$1\" | sed 's/^--port=//'`" |
||||
;; |
||||
# Note: These two specify the user to connect as (like in psql), |
||||
# not the user you're creating. |
||||
--username|-U) |
||||
PSQLOPT="$PSQLOPT -U $2" |
||||
shift;; |
||||
-U*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--username=*) |
||||
PSQLOPT="$PSQLOPT -U `echo \"$1\" | sed 's/^--username=//'`" |
||||
;; |
||||
--password|-W) |
||||
PSQLOPT="$PSQLOPT -W" |
||||
;; |
||||
--echo|-e) |
||||
PSQLOPT="$PSQLOPT -e" |
||||
;; |
||||
--quiet|-q) |
||||
PSQLOPT="$PSQLOPT -o /dev/null" |
||||
;; |
||||
# options converted into SQL command |
||||
--createdb|-d) |
||||
CanCreateDb=t |
||||
;; |
||||
--no-createdb|-D) |
||||
CanCreateDb=f |
||||
;; |
||||
--adduser|-a) |
||||
CanAddUser=t |
||||
;; |
||||
--no-adduser|-A) |
||||
CanAddUser=f |
||||
;; |
||||
--sysid|-i) |
||||
SysID="$2" |
||||
shift;; |
||||
--sysid=*) |
||||
SysID=`echo "$1" | sed 's/^--sysid=//'` |
||||
;; |
||||
--encrypted|-E) |
||||
Encrypted=t |
||||
;; |
||||
--unencrypted|-N) |
||||
Encrypted=f |
||||
;; |
||||
-i*) |
||||
SysID=`echo "$1" | sed 's/^-i//'` |
||||
;; |
||||
--pwprompt|--pw|-P) |
||||
PwPrompt=t |
||||
;; |
||||
-*) |
||||
echo "$CMDNAME: invalid option: $1" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
;; |
||||
*) |
||||
NewUser="$1" |
||||
if [ "$#" -ne 1 ]; then |
||||
echo "$CMDNAME: invalid option: $2" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
fi |
||||
;; |
||||
esac |
||||
shift; |
||||
done |
||||
|
||||
if [ "$usage" ]; then |
||||
echo "$CMDNAME creates a new PostgreSQL user." |
||||
echo |
||||
echo "Usage:" |
||||
echo " $CMDNAME [OPTION]... [USERNAME]" |
||||
echo |
||||
echo "Options:" |
||||
echo " -a, --adduser user can add new users" |
||||
echo " -A, --no-adduser user cannot add new users" |
||||
echo " -d, --createdb user can create new databases" |
||||
echo " -D, --no-createdb user cannot create databases" |
||||
echo " -P, --pwprompt assign a password to new user" |
||||
echo " -E, --encrypted encrypt stored password" |
||||
echo " -N, --unencrypted do no encrypt stored password" |
||||
echo " -i, --sysid=SYSID select sysid for new user" |
||||
echo " -e, --echo show the query being sent to the backend" |
||||
echo " -q, --quiet don't write any messages" |
||||
echo " --help show this help, then exit" |
||||
echo |
||||
echo "Connection options:" |
||||
echo " -h, --host=HOSTNAME database server host" |
||||
echo " -p, --port=PORT database server port" |
||||
echo " -U, --username=USERNAME user name to connect as (not the one to create)" |
||||
echo " -W, --password prompt for password to connect" |
||||
echo |
||||
echo "If one of -a, -A, -d, -D, and USERNAME is not specified, you will" |
||||
echo "be prompted interactively." |
||||
echo |
||||
echo "Report bugs to <pgsql-bugs@postgresql.org>." |
||||
exit 0 |
||||
fi |
||||
|
||||
if [ "$SysID" ]; then |
||||
if [ "$SysID" != "`echo $SysID | sed 's/[^0-9]//g'`" ]; then |
||||
echo "$CMDNAME: user sysid must be a positive number" 1>&2 |
||||
exit 1 |
||||
fi |
||||
fi |
||||
|
||||
# Don't want to leave the user blind if he breaks |
||||
# during password entry. |
||||
|
||||
trap 'stty echo >/dev/null 2>&1; echo; exit 1' 1 2 3 15 |
||||
|
||||
# Get missing user attributes |
||||
|
||||
if [ -z "$NewUser" ]; then |
||||
$ECHO_N "Enter name of user to add: "$ECHO_C |
||||
IFS="$NL" |
||||
read NewUser |
||||
IFS="$_IFS" |
||||
[ "$?" -ne 0 ] && exit 1 |
||||
fi |
||||
|
||||
if [ "$PwPrompt" ]; then |
||||
$ECHO_N "Enter password for user \"$NewUser\": "$ECHO_C |
||||
stty -echo >/dev/null 2>&1 |
||||
IFS="$NL" |
||||
read FirstPw |
||||
IFS="$_IFS" |
||||
stty echo >/dev/null 2>&1 |
||||
echo |
||||
$ECHO_N "Enter it again: "$ECHO_C |
||||
stty -echo >/dev/null 2>&1 |
||||
IFS="$NL" |
||||
read SecondPw |
||||
IFS="$_IFS" |
||||
stty echo >/dev/null 2>&1 |
||||
echo |
||||
if [ "$FirstPw" != "$SecondPw" ]; then |
||||
echo "Passwords didn't match." 1>&2 |
||||
exit 1 |
||||
fi |
||||
Password="$FirstPw" |
||||
fi |
||||
|
||||
if [ -z "$CanCreateDb" ]; then |
||||
$ECHO_N "Shall the new user be allowed to create databases? (y/n) "$ECHO_C |
||||
read REPLY |
||||
[ "$?" -ne 0 ] && exit 1 |
||||
if [ "$REPLY" = "y" -o "$REPLY" = "Y" ]; then |
||||
CanCreateDb=t |
||||
else |
||||
CanCreateDb=f |
||||
fi |
||||
fi |
||||
|
||||
if [ -z "$CanAddUser" ]; then |
||||
$ECHO_N "Shall the new user be allowed to create more new users? (y/n) "$ECHO_C |
||||
read REPLY |
||||
[ "$?" -ne 0 ] && exit 1 |
||||
if [ "$REPLY" = "y" -o "$REPLY" = "Y" ]; then |
||||
CanAddUser=t |
||||
else |
||||
CanAddUser=f |
||||
fi |
||||
fi |
||||
|
||||
|
||||
# |
||||
# build SQL command |
||||
# |
||||
NewUser=`echo "$NewUser" | sed 's/\"/\\\"/g'` |
||||
Password=`echo "$Password" | sed 's/\"/\\\"/g'` |
||||
|
||||
QUERY="CREATE USER \"$NewUser\"" |
||||
|
||||
SUBQUERY= |
||||
[ "$SysID" ] && SUBQUERY="$SUBQUERY SYSID $SysID" |
||||
[ "$Encrypted" = t ] && SUBQUERY="$SUBQUERY ENCRYPTED" |
||||
[ "$Encrypted" = f ] && SUBQUERY="$SUBQUERY UNENCRYPTED" |
||||
[ "$Password" ] && SUBQUERY="$SUBQUERY PASSWORD '$Password'" |
||||
[ "$SUBQUERY" ] && QUERY="$QUERY WITH $SUBQUERY" |
||||
|
||||
[ "$CanCreateDb" = t ] && QUERY="$QUERY CREATEDB" |
||||
[ "$CanCreateDb" = f ] && QUERY="$QUERY NOCREATEDB" |
||||
[ "$CanAddUser" = t ] && QUERY="$QUERY CREATEUSER" |
||||
[ "$CanAddUser" = f ] && QUERY="$QUERY NOCREATEUSER" |
||||
|
||||
${PATHNAME}psql -c "SET autocommit TO 'on';$QUERY" -d template1 $PSQLOPT |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: creation of user \"$NewUser\" failed" 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
exit 0 |
||||
@ -0,0 +1,258 @@ |
||||
/*-------------------------------------------------------------------------
|
||||
* |
||||
* createuser |
||||
* |
||||
* Portions Copyright (c) 1996-2003, PostgreSQL Global Development Group |
||||
* Portions Copyright (c) 1994, Regents of the University of California |
||||
* |
||||
* $Header: /cvsroot/pgsql/src/bin/scripts/createuser.c,v 1.1 2003/03/18 22:19:46 petere Exp $ |
||||
* |
||||
*------------------------------------------------------------------------- |
||||
*/ |
||||
|
||||
#include "postgres_fe.h" |
||||
#include "common.h" |
||||
#include "dumputils.h" |
||||
|
||||
|
||||
static void help(const char *progname); |
||||
|
||||
|
||||
int |
||||
main(int argc, char *argv[]) |
||||
{ |
||||
static struct option long_options[] = { |
||||
{"host", required_argument, NULL, 'h'}, |
||||
{"port", required_argument, NULL, 'p'}, |
||||
{"username", required_argument, NULL, 'U'}, |
||||
{"password", no_argument, NULL, 'W'}, |
||||
{"echo", no_argument, NULL, 'e'}, |
||||
{"quiet", no_argument, NULL, 'q'}, |
||||
{"createdb", no_argument, NULL, 'd'}, |
||||
{"no-createdb", no_argument, NULL, 'D'}, |
||||
{"adduser", no_argument, NULL, 'a'}, |
||||
{"no-adduser", no_argument, NULL, 'A'}, |
||||
{"sysid", required_argument, NULL, 'i'}, |
||||
{"pwprompt", no_argument, NULL, 'P'}, |
||||
{"encrypted", no_argument, NULL, 'E'}, |
||||
{"unencrypted", no_argument, NULL, 'N'}, |
||||
{NULL, 0, NULL, 0} |
||||
}; |
||||
|
||||
char *progname; |
||||
int optindex; |
||||
int c; |
||||
|
||||
char *newuser = NULL; |
||||
char *host = NULL; |
||||
char *port = NULL; |
||||
char *username = NULL; |
||||
bool password = false; |
||||
bool echo = false; |
||||
bool quiet = false; |
||||
int createdb = 0; |
||||
int adduser = 0; |
||||
char *sysid = NULL; |
||||
bool pwprompt = false; |
||||
int encrypted = 0; /* 0 uses server default */ |
||||
char *newpassword = NULL; |
||||
|
||||
PQExpBufferData sql; |
||||
|
||||
PGconn *conn; |
||||
PGresult *result; |
||||
|
||||
progname = get_progname(argv[0]); |
||||
init_nls(); |
||||
handle_help_version_opts(argc, argv, "createuser", help); |
||||
|
||||
while ((c = getopt_long(argc, argv, "h:p:U:WeqaAdDi:PEN", long_options, &optindex)) != -1) |
||||
{ |
||||
switch (c) |
||||
{ |
||||
case 'h': |
||||
host = optarg; |
||||
break; |
||||
case 'p': |
||||
port = optarg; |
||||
break; |
||||
case 'U': |
||||
username = optarg; |
||||
break; |
||||
case 'W': |
||||
password = true; |
||||
break; |
||||
case 'e': |
||||
echo = true; |
||||
break; |
||||
case 'q': |
||||
quiet = true; |
||||
break; |
||||
case 'a': |
||||
adduser = +1; |
||||
break; |
||||
case 'A': |
||||
adduser = -1; |
||||
break; |
||||
case 'd': |
||||
createdb = +1; |
||||
break; |
||||
case 'D': |
||||
createdb = -1; |
||||
break; |
||||
case 'i': |
||||
sysid = optarg; |
||||
break; |
||||
case 'P': |
||||
pwprompt = true; |
||||
break; |
||||
case 'E': |
||||
encrypted = +1; |
||||
break; |
||||
case 'N': |
||||
encrypted = -1; |
||||
break; |
||||
default: |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
} |
||||
|
||||
switch (argc - optind) |
||||
{ |
||||
case 0: |
||||
break; |
||||
case 1: |
||||
newuser = argv[optind]; |
||||
break; |
||||
default: |
||||
fprintf(stderr, _("%s: too many command-line arguments (first is '%s')\n"), |
||||
progname, argv[optind + 1]); |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
|
||||
if (sysid) |
||||
{ |
||||
char *endptr; |
||||
|
||||
if (strtol(sysid, &endptr, 10) <= 0 || *endptr != '\0') |
||||
{ |
||||
fprintf(stderr, _("%s: user ID must be a positive number\n"), progname); |
||||
exit(1); |
||||
} |
||||
} |
||||
|
||||
if (newuser == NULL) |
||||
newuser = simple_prompt("Enter name of user to add: ", 128, true); |
||||
|
||||
if (pwprompt) |
||||
{ |
||||
char *pw1, *pw2; |
||||
|
||||
pw1 = simple_prompt("Enter password for new user: ", 100, false); |
||||
pw2 = simple_prompt("Enter it again: ", 100, false); |
||||
if (strcmp(pw1, pw2)!=0) |
||||
{ |
||||
fprintf(stderr, _("Passwords didn't match.\n")); |
||||
exit(1); |
||||
} |
||||
newpassword = pw1; |
||||
free(pw2); |
||||
} |
||||
|
||||
if (createdb == 0) |
||||
{ |
||||
char *reply; |
||||
|
||||
reply = simple_prompt("Shall the new user be allowed to create databases? (y/n) ", 1, true); |
||||
if (reply[0] == 'y' || reply[0] == 'Y') |
||||
createdb = +1; |
||||
else |
||||
createdb = -1; |
||||
} |
||||
|
||||
if (adduser == 0) |
||||
{ |
||||
char *reply; |
||||
|
||||
reply = simple_prompt("Shall the new user be allowed to create more new users? (y/n) ", 1, true); |
||||
if (reply[0] == 'y' || reply[0] == 'Y') |
||||
adduser = +1; |
||||
else |
||||
adduser = -1; |
||||
} |
||||
|
||||
initPQExpBuffer(&sql); |
||||
|
||||
printfPQExpBuffer(&sql, "SET autocommit TO on;\nCREATE USER %s", fmtId(newuser)); |
||||
if (sysid) |
||||
appendPQExpBuffer(&sql, " SYSID %s", sysid); |
||||
if (encrypted == +1) |
||||
appendPQExpBuffer(&sql, " ENCRYPTED"); |
||||
if (encrypted == -1) |
||||
appendPQExpBuffer(&sql, " UNENCRYPTED"); |
||||
if (newpassword) |
||||
{ |
||||
appendPQExpBuffer(&sql, " PASSWORD "); |
||||
appendStringLiteral(&sql, newpassword, false); |
||||
} |
||||
if (createdb == +1) |
||||
appendPQExpBuffer(&sql, " CREATEDB"); |
||||
if (createdb == -1) |
||||
appendPQExpBuffer(&sql, " NOCREATEDB"); |
||||
if (adduser == +1) |
||||
appendPQExpBuffer(&sql, " CREATEUSER"); |
||||
if (adduser == -1) |
||||
appendPQExpBuffer(&sql, " NOCREATEUSER"); |
||||
appendPQExpBuffer(&sql, ";\n"); |
||||
|
||||
conn = connectDatabase("template1", host, port, username, password, progname); |
||||
|
||||
if (echo) |
||||
printf("%s", sql.data); |
||||
result = PQexec(conn, sql.data); |
||||
|
||||
if (PQresultStatus(result) != PGRES_COMMAND_OK) |
||||
{ |
||||
fprintf(stderr, _("%s: creation of new user failed: %s"), |
||||
progname, PQerrorMessage(conn)); |
||||
PQfinish(conn); |
||||
exit(1); |
||||
} |
||||
|
||||
PQfinish(conn); |
||||
if (!quiet) |
||||
puts("CREATE USER"); |
||||
exit(0); |
||||
} |
||||
|
||||
|
||||
static void |
||||
help(const char *progname) |
||||
{ |
||||
printf(_("%s creates a new PostgreSQL user.\n\n"), progname); |
||||
printf(_("Usage:\n")); |
||||
printf(_(" %s [OPTION]... [USERNAME]\n"), progname); |
||||
printf(_("\nOptions:\n")); |
||||
printf(_(" -a, --adduser user can add new users\n")); |
||||
printf(_(" -A, --no-adduser user cannot add new users\n")); |
||||
printf(_(" -d, --createdb user can create new databases\n")); |
||||
printf(_(" -D, --no-createdb user cannot create databases\n")); |
||||
printf(_(" -P, --pwprompt assign a password to new user\n")); |
||||
printf(_(" -E, --encrypted encrypt stored password\n")); |
||||
printf(_(" -N, --unencrypted do no encrypt stored password\n")); |
||||
printf(_(" -i, --sysid=SYSID select sysid for new user\n" )); |
||||
printf(_(" -e, --echo show the commands being sent to the server\n")); |
||||
printf(_(" -q, --quiet don't write any messages\n")); |
||||
printf(_(" --help show this help, then exit\n")); |
||||
printf(_(" --version output version information, then exit\n")); |
||||
printf(_("\nConnection options:\n")); |
||||
printf(_(" -h, --host=HOSTNAME database server host\n")); |
||||
printf(_(" -p, --port=PORT database server port\n")); |
||||
printf(_(" -U, --username=USERNAME user name to connect as (not the one to create)\n")); |
||||
printf(_(" -W, --password prompt for password to connect\n")); |
||||
printf(_("\nIf one of -a, -A, -d, -D, and USERNAME is not specified, you will\n" |
||||
"be prompted interactively.\n")); |
||||
printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n")); |
||||
} |
||||
@ -1,148 +0,0 @@ |
||||
#!/bin/sh |
||||
#------------------------------------------------------------------------- |
||||
# |
||||
# dropdb-- |
||||
# destroy a postgres database |
||||
# |
||||
# this program runs psql to drop the requested database. |
||||
# |
||||
# Portions Copyright (c) 1996-2002, PostgreSQL Global Development Group |
||||
# Portions Copyright (c) 1994, Regents of the University of California |
||||
# |
||||
# |
||||
# IDENTIFICATION |
||||
# $Header: /cvsroot/pgsql/src/bin/scripts/Attic/dropdb,v 1.19 2002/10/18 22:05:36 petere Exp $ |
||||
# |
||||
#------------------------------------------------------------------------- |
||||
|
||||
CMDNAME=`basename "$0"` |
||||
PATHNAME=`echo "$0" | sed "s,$CMDNAME\$,,"` |
||||
|
||||
PSQLOPT= |
||||
dbname= |
||||
forcedel=t |
||||
|
||||
# Check for echo -n vs echo \c |
||||
|
||||
if echo '\c' | grep -s c >/dev/null 2>&1 |
||||
then |
||||
ECHO_N="echo -n" |
||||
ECHO_C="" |
||||
else |
||||
ECHO_N="echo" |
||||
ECHO_C='\c' |
||||
fi |
||||
|
||||
|
||||
while [ "$#" -gt 0 ] |
||||
do |
||||
case "$1" in |
||||
--help|-\?) |
||||
usage=t |
||||
break |
||||
;; |
||||
# options passed on to psql |
||||
--host|-h) |
||||
PSQLOPT="$PSQLOPT -h $2" |
||||
shift;; |
||||
-h*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--host=*) |
||||
PSQLOPT="$PSQLOPT -h `echo \"$1\" | sed 's/^--host=//'`" |
||||
;; |
||||
--port|-p) |
||||
PSQLOPT="$PSQLOPT -p $2" |
||||
shift;; |
||||
-p*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--port=*) |
||||
PSQLOPT="$PSQLOPT -p `echo \"$1\" | sed 's/^--port=//'`" |
||||
;; |
||||
--username|-U) |
||||
PSQLOPT="$PSQLOPT -U $2" |
||||
shift;; |
||||
-U*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--username=*) |
||||
PSQLOPT="$PSQLOPT -U `echo \"$1\" | sed 's/^--username=//'`" |
||||
;; |
||||
--password|-W) |
||||
PSQLOPT="$PSQLOPT -W" |
||||
;; |
||||
--echo|-e) |
||||
PSQLOPT="$PSQLOPT -e" |
||||
;; |
||||
--quiet|-q) |
||||
PSQLOPT="$PSQLOPT -o /dev/null" |
||||
;; |
||||
# other options |
||||
--interactive|-i) |
||||
forcedel=f |
||||
;; |
||||
-*) |
||||
echo "$CMDNAME: invalid option: $1" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
;; |
||||
*) |
||||
dbname="$1" |
||||
if [ "$#" -ne 1 ]; then |
||||
echo "$CMDNAME: invalid option: $2" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
fi |
||||
;; |
||||
esac |
||||
shift |
||||
done |
||||
|
||||
|
||||
if [ "$usage" ]; then |
||||
echo "$CMDNAME removes a PostgreSQL database." |
||||
echo |
||||
echo "Usage:" |
||||
echo " $CMDNAME [OPTION]... DBNAME" |
||||
echo |
||||
echo "Options:" |
||||
echo " -e, --echo show the query being sent to the backend" |
||||
echo " -i, --interactive prompt before deleting anything" |
||||
echo " -q, --quiet don't write any messages" |
||||
echo " -h, --host=HOSTNAME database server host" |
||||
echo " -p, --port=PORT database server port" |
||||
echo " -U, --username=USERNAME user name to connect as" |
||||
echo " -W, --password prompt for password" |
||||
echo " --help show this help, then exit" |
||||
echo |
||||
echo "Report bugs to <pgsql-bugs@postgresql.org>." |
||||
exit 0 |
||||
fi |
||||
|
||||
if [ -z "$dbname" ]; then |
||||
echo "$CMDNAME: missing required argument database name" 1>&2 |
||||
echo "Try '$CMDNAME -?' for help." 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
|
||||
if [ "$forcedel" = f ]; then |
||||
echo "Database \"$dbname\" will be permanently deleted." |
||||
$ECHO_N "Are you sure? (y/n) "$ECHO_C |
||||
read REPLY |
||||
|
||||
[ "$?" -eq 1 ] && exit 1 |
||||
[ "$REPLY" != "y" -a "$REPLY" != "Y" ] && exit 0 |
||||
fi |
||||
|
||||
|
||||
dbname=`echo "$dbname" | sed 's/\"/\\\"/g'` |
||||
|
||||
${PATHNAME}psql $PSQLOPT -d template1 -c "SET autocommit TO 'on';DROP DATABASE \"$dbname\"" |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: database removal failed" 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
exit 0 |
||||
@ -0,0 +1,156 @@ |
||||
/*-------------------------------------------------------------------------
|
||||
* |
||||
* dropdb |
||||
* |
||||
* Portions Copyright (c) 1996-2003, PostgreSQL Global Development Group |
||||
* Portions Copyright (c) 1994, Regents of the University of California |
||||
* |
||||
* $Header: /cvsroot/pgsql/src/bin/scripts/dropdb.c,v 1.1 2003/03/18 22:19:46 petere Exp $ |
||||
* |
||||
*------------------------------------------------------------------------- |
||||
*/ |
||||
|
||||
#include "postgres_fe.h" |
||||
#include "common.h" |
||||
#include "dumputils.h" |
||||
|
||||
|
||||
static void help(const char *progname); |
||||
|
||||
|
||||
int |
||||
main(int argc, char *argv[]) |
||||
{ |
||||
static struct option long_options[] = { |
||||
{"host", required_argument, NULL, 'h'}, |
||||
{"port", required_argument, NULL, 'p'}, |
||||
{"username", required_argument, NULL, 'U'}, |
||||
{"password", no_argument, NULL, 'W'}, |
||||
{"echo", no_argument, NULL, 'e'}, |
||||
{"quiet", no_argument, NULL, 'q'}, |
||||
{"interactive", no_argument, NULL, 'i'}, |
||||
{NULL, 0, NULL, 0} |
||||
}; |
||||
|
||||
char *progname; |
||||
int optindex; |
||||
int c; |
||||
|
||||
char *dbname = NULL; |
||||
char *host = NULL; |
||||
char *port = NULL; |
||||
char *username = NULL; |
||||
bool password = false; |
||||
bool echo = false; |
||||
bool quiet = false; |
||||
bool interactive = false; |
||||
|
||||
PQExpBufferData sql; |
||||
|
||||
PGconn *conn; |
||||
PGresult *result; |
||||
|
||||
progname = get_progname(argv[0]); |
||||
init_nls(); |
||||
handle_help_version_opts(argc, argv, "dropdb", help); |
||||
|
||||
while ((c = getopt_long(argc, argv, "h:p:U:Weqi", long_options, &optindex)) != -1) |
||||
{ |
||||
switch (c) |
||||
{ |
||||
case 'h': |
||||
host = optarg; |
||||
break; |
||||
case 'p': |
||||
port = optarg; |
||||
break; |
||||
case 'U': |
||||
username = optarg; |
||||
break; |
||||
case 'W': |
||||
password = true; |
||||
break; |
||||
case 'e': |
||||
echo = true; |
||||
break; |
||||
case 'q': |
||||
quiet = true; |
||||
break; |
||||
case 'i': |
||||
interactive = true; |
||||
break; |
||||
default: |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
} |
||||
|
||||
switch (argc - optind) |
||||
{ |
||||
case 0: |
||||
fprintf(stderr, _("%s: missing required argument database name\n"), progname); |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
case 1: |
||||
dbname = argv[optind]; |
||||
break; |
||||
default: |
||||
fprintf(stderr, _("%s: too many command-line arguments (first is '%s')\n"), |
||||
progname, argv[optind + 1]); |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
|
||||
if (interactive) |
||||
{ |
||||
char *reply; |
||||
|
||||
printf(_("Database \"%s\" will be permanently deleted.\n"), dbname); |
||||
reply = simple_prompt("Are you sure? (y/n) ", 1, true); |
||||
if (reply[0] != 'y' && reply[0] != 'Y') |
||||
exit(0); |
||||
} |
||||
|
||||
initPQExpBuffer(&sql); |
||||
|
||||
appendPQExpBuffer(&sql, "SET autocommit TO on;\nDROP DATABASE %s;\n", |
||||
fmtId(dbname)); |
||||
|
||||
conn = connectDatabase("template1", host, port, username, password, progname); |
||||
|
||||
if (echo) |
||||
printf("%s", sql.data); |
||||
result = PQexec(conn, sql.data); |
||||
if (PQresultStatus(result) != PGRES_COMMAND_OK) |
||||
{ |
||||
fprintf(stderr, _("%s: database removal failed: %s"), |
||||
progname, PQerrorMessage(conn)); |
||||
PQfinish(conn); |
||||
exit(1); |
||||
} |
||||
|
||||
PQfinish(conn); |
||||
if (!quiet) |
||||
puts("DROP DATABASE"); |
||||
exit(0); |
||||
} |
||||
|
||||
|
||||
static void |
||||
help(const char *progname) |
||||
{ |
||||
printf(_("%s removes a PostgreSQL database.\n\n"), progname); |
||||
printf(_("Usage:\n")); |
||||
printf(_(" %s [OPTION]... DBNAME\n"), progname); |
||||
printf(_("\nOptions:\n")); |
||||
printf(_(" -e, --echo show the commands being sent to the server\n")); |
||||
printf(_(" -i, --interactive prompt before deleting anything\n")); |
||||
printf(_(" -q, --quiet don't write any messages\n")); |
||||
printf(_(" -h, --host=HOSTNAME database server host\n")); |
||||
printf(_(" -p, --port=PORT database server port\n")); |
||||
printf(_(" -U, --username=USERNAME user name to connect as\n")); |
||||
printf(_(" -W, --password prompt for password\n")); |
||||
printf(_(" --help show this help, then exit\n")); |
||||
printf(_(" --version output version information, then exit\n")); |
||||
printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n")); |
||||
} |
||||
@ -1,270 +0,0 @@ |
||||
#! /bin/sh |
||||
#------------------------------------------------------------------------- |
||||
# |
||||
# droplang -- |
||||
# Remove a procedural language from a database |
||||
# |
||||
# Portions Copyright (c) 1996-2002, PostgreSQL Global Development Group |
||||
# Portions Copyright (c) 1994, Regents of the University of California |
||||
# |
||||
# $Header: /cvsroot/pgsql/src/bin/scripts/Attic/droplang,v 1.27 2002/10/18 22:05:36 petere Exp $ |
||||
# |
||||
#------------------------------------------------------------------------- |
||||
|
||||
CMDNAME=`basename "$0"` |
||||
PATHNAME=`echo "$0" | sed "s,$CMDNAME\$,,"` |
||||
|
||||
PSQLOPT= |
||||
dbname= |
||||
langname= |
||||
echo= |
||||
list= |
||||
showsql= |
||||
|
||||
# Check for echo -n vs echo \c |
||||
|
||||
if echo '\c' | grep -s c >/dev/null 2>&1 |
||||
then |
||||
ECHO_N="echo -n" |
||||
ECHO_C="" |
||||
else |
||||
ECHO_N="echo" |
||||
ECHO_C='\c' |
||||
fi |
||||
|
||||
|
||||
# ---------- |
||||
# Get options, language name and dbname |
||||
# ---------- |
||||
while [ "$#" -gt 0 ] |
||||
do |
||||
case "$1" in |
||||
--help|-\?) |
||||
usage=t |
||||
break |
||||
;; |
||||
--list|-l) |
||||
list=t |
||||
;; |
||||
# options passed on to psql |
||||
--host|-h) |
||||
PSQLOPT="$PSQLOPT -h $2" |
||||
shift;; |
||||
-h*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--host=*) |
||||
PSQLOPT="$PSQLOPT -h `echo \"$1\" | sed 's/^--host=//'`" |
||||
;; |
||||
--port|-p) |
||||
PSQLOPT="$PSQLOPT -p $2" |
||||
shift;; |
||||
-p*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--port=*) |
||||
PSQLOPT="$PSQLOPT -p `echo \"$1\" | sed 's/^--port=//'`" |
||||
;; |
||||
--username|-U) |
||||
PSQLOPT="$PSQLOPT -U $2" |
||||
shift;; |
||||
-U*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--username=*) |
||||
PSQLOPT="$PSQLOPT -U `echo \"$1\" | sed 's/^--username=//'`" |
||||
;; |
||||
--password|-W) |
||||
PSQLOPT="$PSQLOPT -W" |
||||
;; |
||||
--dbname|-d) |
||||
dbname="$2" |
||||
shift;; |
||||
-d*) |
||||
dbname=`echo "$1" | sed 's/^-d//'` |
||||
;; |
||||
--dbname=*) |
||||
dbname=`echo "$1" | sed 's/^--dbname=//'` |
||||
;; |
||||
--echo|-e) |
||||
showsql=yes |
||||
;; |
||||
|
||||
-*) |
||||
echo "$CMDNAME: invalid option: $1" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
;; |
||||
*) |
||||
if [ "$list" != "t" ] |
||||
then langname="$1" |
||||
if [ "$2" ] |
||||
then |
||||
shift |
||||
dbname="$1" |
||||
fi |
||||
else dbname="$1" |
||||
fi |
||||
if [ "$#" -ne 1 ]; then |
||||
echo "$CMDNAME: invalid option: $2" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
fi |
||||
;; |
||||
esac |
||||
shift |
||||
done |
||||
|
||||
|
||||
if [ "$usage" ]; then |
||||
echo "$CMDNAME removes a procedural language from a database." |
||||
echo |
||||
echo "Usage:" |
||||
echo " $CMDNAME [OPTION]... LANGNAME [DBNAME]" |
||||
echo |
||||
echo "Options:" |
||||
echo " -d, --dbname=DBNAME database to remove language from" |
||||
echo " -l, --list show a list of currently installed languages" |
||||
echo " -h, --host=HOSTNAME database server host" |
||||
echo " -p, --port=PORT database server port" |
||||
echo " -U, --username=USERNAME user name to connect as" |
||||
echo " -W, --password prompt for password" |
||||
echo " --help show this help, then exit" |
||||
echo |
||||
echo "Report bugs to <pgsql-bugs@postgresql.org>." |
||||
exit 0 |
||||
fi |
||||
|
||||
|
||||
if [ -z "$dbname" ]; then |
||||
if [ "$PGDATABASE" ]; then |
||||
dbname="$PGDATABASE" |
||||
elif [ "$PGUSER" ]; then |
||||
dbname="$PGUSER" |
||||
else |
||||
dbname=`${PATHNAME}pg_id -u -n` |
||||
fi |
||||
[ "$?" -ne 0 ] && exit 1 |
||||
fi |
||||
|
||||
|
||||
# ---------- |
||||
# List option, doesn't need langname |
||||
# ---------- |
||||
if [ "$list" ]; then |
||||
sqlcmd="SELECT lanname as \"Name\", lanpltrusted as \"Trusted?\" FROM pg_language WHERE lanispl = TRUE" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
${PATHNAME}psql $PSQLOPT -d "$dbname" -P 'title=Procedural languages' -c "$sqlcmd" |
||||
exit $? |
||||
fi |
||||
|
||||
|
||||
# ---------- |
||||
# We can't go any farther without a langname |
||||
# ---------- |
||||
if [ -z "$langname" ]; then |
||||
echo "$CMDNAME: missing required argument language name" 1>&2 |
||||
echo "Try '$CMDNAME --help' for help." 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
PSQL="${PATHNAME}psql -A -t -q $PSQLOPT -d $dbname -c" |
||||
|
||||
|
||||
# ---------- |
||||
# Make sure the language is installed and find the oid of the handler function |
||||
# ---------- |
||||
sqlcmd="SELECT lanplcallfoid FROM pg_language WHERE lanname = '$langname' AND lanispl;" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
lanplcallfoid=`$PSQL "$sqlcmd"` |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: external error" 1>&2 |
||||
exit 1 |
||||
fi |
||||
if [ -z "$lanplcallfoid" ]; then |
||||
echo "$CMDNAME: language \"$langname\" is not installed in database $dbname" 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
|
||||
# ---------- |
||||
# Check that there are no functions left defined in that language |
||||
# ---------- |
||||
sqlcmd="SELECT COUNT(proname) FROM pg_proc P, pg_language L WHERE P.prolang = L.oid AND L.lanname = '$langname';" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
res=`$PSQL "$sqlcmd"` |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: external error" 1>&2 |
||||
exit 1 |
||||
fi |
||||
if [ "$res" -ne 0 ]; then |
||||
echo "$CMDNAME: There are $res functions/trigger procedures declared in language" 1>&2 |
||||
echo "$langname. Language not removed." 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
# ---------- |
||||
# Check that the handler function isn't used by some other language |
||||
# ---------- |
||||
sqlcmd="SELECT count(*) FROM pg_language WHERE lanplcallfoid = $lanplcallfoid AND lanname <> '$langname';" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
res=`$PSQL "$sqlcmd"` |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: external error" 1>&2 |
||||
exit 1 |
||||
fi |
||||
if [ "$res" -eq 0 ]; then |
||||
keephandler=no |
||||
else |
||||
keephandler=yes |
||||
fi |
||||
|
||||
# ---------- |
||||
# Drop the language |
||||
# ---------- |
||||
sqlcmd="SET autocommit TO 'on';DROP LANGUAGE \"$langname\";" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
$PSQL "$sqlcmd" |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: language removal failed" 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
# ---------- |
||||
# Drop the call handler |
||||
# ---------- |
||||
if [ "$keephandler" = yes ]; then |
||||
exit 0 |
||||
fi |
||||
|
||||
sqlcmd="SELECT proname FROM pg_proc WHERE oid = $lanplcallfoid;" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
handler=`$PSQL "$sqlcmd"` |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: external error" 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
sqlcmd="SET autocommit TO 'on';DROP FUNCTION \"$handler\" ();" |
||||
if [ "$showsql" = yes ]; then |
||||
echo "$sqlcmd" |
||||
fi |
||||
$PSQL "$sqlcmd" |
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: language removal failed" 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
exit 0 |
||||
@ -0,0 +1,255 @@ |
||||
/*-------------------------------------------------------------------------
|
||||
* |
||||
* droplang |
||||
* |
||||
* Portions Copyright (c) 1996-2003, PostgreSQL Global Development Group |
||||
* Portions Copyright (c) 1994, Regents of the University of California |
||||
* |
||||
* $Header: /cvsroot/pgsql/src/bin/scripts/droplang.c,v 1.1 2003/03/18 22:19:47 petere Exp $ |
||||
* |
||||
*------------------------------------------------------------------------- |
||||
*/ |
||||
|
||||
#include "postgres_fe.h" |
||||
#include "common.h" |
||||
#include "print.h" |
||||
|
||||
|
||||
static void help(const char *progname); |
||||
|
||||
|
||||
int |
||||
main(int argc, char *argv[]) |
||||
{ |
||||
static struct option long_options[] = { |
||||
{"list", no_argument, NULL, 'l'}, |
||||
{"host", required_argument, NULL, 'h'}, |
||||
{"port", required_argument, NULL, 'p'}, |
||||
{"username", required_argument, NULL, 'U'}, |
||||
{"password", no_argument, NULL, 'W'}, |
||||
{"dbname", required_argument, NULL, 'd'}, |
||||
{"echo", no_argument, NULL, 'e'}, |
||||
{NULL, 0, NULL, 0} |
||||
}; |
||||
|
||||
char *progname; |
||||
int optindex; |
||||
int c; |
||||
|
||||
bool listlangs = false; |
||||
const char *dbname = NULL; |
||||
char *host = NULL; |
||||
char *port = NULL; |
||||
char *username = NULL; |
||||
bool password = false; |
||||
bool echo = false; |
||||
char *langname = NULL; |
||||
|
||||
char *p; |
||||
char *lanplcallfoid; |
||||
char *handler; |
||||
bool keephandler; |
||||
|
||||
PQExpBufferData sql; |
||||
|
||||
PGconn *conn; |
||||
PGresult *result; |
||||
|
||||
progname = get_progname(argv[0]); |
||||
init_nls(); |
||||
handle_help_version_opts(argc, argv, "droplang", help); |
||||
|
||||
while ((c = getopt_long(argc, argv, "lh:p:U:Wd:e", long_options, &optindex)) != -1) |
||||
{ |
||||
switch (c) |
||||
{ |
||||
case 'l': |
||||
listlangs = true; |
||||
break; |
||||
case 'h': |
||||
host = optarg; |
||||
break; |
||||
case 'p': |
||||
port = optarg; |
||||
break; |
||||
case 'U': |
||||
username = optarg; |
||||
break; |
||||
case 'W': |
||||
password = true; |
||||
break; |
||||
case 'd': |
||||
dbname = optarg; |
||||
break; |
||||
case 'e': |
||||
echo = true; |
||||
break; |
||||
default: |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
} |
||||
|
||||
if (argc - optind > 0) |
||||
{ |
||||
if (listlangs) |
||||
dbname = argv[optind++]; |
||||
else |
||||
{ |
||||
langname = argv[optind++]; |
||||
if (argc - optind > 0) |
||||
dbname = argv[optind++]; |
||||
} |
||||
} |
||||
|
||||
if (argc - optind > 0) |
||||
{ |
||||
fprintf(stderr, _("%s: too many command line options (first is '%s')\n"), |
||||
progname, argv[optind]); |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
|
||||
if (dbname == NULL) |
||||
{ |
||||
if (getenv("PGDATABASE")) |
||||
dbname = getenv("PGDATABASE"); |
||||
else if (getenv("PGUSER")) |
||||
dbname = getenv("PGUSER"); |
||||
else |
||||
dbname = get_user_name(progname); |
||||
} |
||||
|
||||
initPQExpBuffer(&sql); |
||||
|
||||
/*
|
||||
* List option |
||||
*/ |
||||
if (listlangs) |
||||
{ |
||||
printQueryOpt popt; |
||||
|
||||
conn = connectDatabase(dbname, host, port, username, password, progname); |
||||
|
||||
printfPQExpBuffer(&sql, "SELECT lanname as \"%s\", (CASE WHEN lanpltrusted THEN '%s' ELSE '%s' END) as \"%s\" FROM pg_language WHERE lanispl IS TRUE;", _("Name"), _("yes"), _("no"), _("Trusted?")); |
||||
result = executeQuery(conn, sql.data, progname, echo); |
||||
|
||||
memset(&popt, 0, sizeof(popt)); |
||||
popt.topt.format = PRINT_ALIGNED; |
||||
popt.topt.border = 1; |
||||
popt.topt.encoding = PQclientEncoding(conn); |
||||
popt.title = _("Procedural Languages"); |
||||
printQuery(result, &popt, stdout); |
||||
|
||||
PQfinish(conn); |
||||
exit(0); |
||||
} |
||||
|
||||
if (langname == NULL) |
||||
{ |
||||
fprintf(stderr, _("%s: missing required argument language name\n"), progname); |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
|
||||
for (p = langname; *p; p++) |
||||
if (*p >= 'A' && *p <= 'Z') |
||||
*p += ('a' - 'A'); |
||||
|
||||
conn = connectDatabase(dbname, host, port, username, password, progname); |
||||
|
||||
/*
|
||||
* Make sure the language is installed and find the OID of the handler function |
||||
*/ |
||||
printfPQExpBuffer(&sql, "SELECT lanplcallfoid FROM pg_language WHERE lanname = '%s' AND lanispl;", langname); |
||||
result = executeQuery(conn, sql.data, progname, echo); |
||||
if (PQntuples(result) == 0) |
||||
{ |
||||
PQfinish(conn); |
||||
fprintf(stderr, _("%s: language \"%s\" is not installed in database \"%s\"\n"), |
||||
progname, langname, dbname); |
||||
exit(1); |
||||
} |
||||
lanplcallfoid = PQgetvalue(result, 0, 0); |
||||
/* result not cleared! */ |
||||
|
||||
/*
|
||||
* Check that there are no functions left defined in that language |
||||
*/ |
||||
printfPQExpBuffer(&sql, "SELECT count(proname) FROM pg_proc P, pg_language L WHERE P.prolang = L.oid AND L.lanname = '%s';", langname); |
||||
result = executeQuery(conn, sql.data, progname, echo); |
||||
if (strcmp(PQgetvalue(result, 0, 0), "0")!=0) |
||||
{ |
||||
PQfinish(conn); |
||||
fprintf(stderr, |
||||
_("%s: There are %s functions declared in language \"%s\". Language not removed.\n"), |
||||
progname, PQgetvalue(result, 0, 0), langname); |
||||
exit(1); |
||||
} |
||||
PQclear(result); |
||||
|
||||
/*
|
||||
* Check that the handler function isn't used by some other language |
||||
*/ |
||||
printfPQExpBuffer(&sql, "SELECT count(*) FROM pg_language WHERE lanplcallfoid = %s AND lanname <> '%s';", lanplcallfoid, langname); |
||||
result = executeQuery(conn, sql.data, progname, echo); |
||||
if (strcmp(PQgetvalue(result, 0, 0), "0")==0) |
||||
keephandler = false; |
||||
else |
||||
keephandler = true; |
||||
PQclear(result); |
||||
|
||||
/*
|
||||
* Find the handler name |
||||
*/ |
||||
if (!keephandler) |
||||
{ |
||||
printfPQExpBuffer(&sql, "SELECT proname FROM pg_proc WHERE oid = %s;", lanplcallfoid); |
||||
result = executeQuery(conn, sql.data, progname, echo); |
||||
handler = PQgetvalue(result, 0, 0); |
||||
/* result not cleared! */ |
||||
} |
||||
else |
||||
handler = NULL; |
||||
|
||||
/*
|
||||
* Drop the language |
||||
*/ |
||||
printfPQExpBuffer(&sql, "SET autocommit TO on;\n"); |
||||
appendPQExpBuffer(&sql, "DROP LANGUAGE \"%s\";\n", langname); |
||||
if (!keephandler) |
||||
appendPQExpBuffer(&sql, "DROP FUNCTION \"%s\" ();\n", handler); |
||||
if (echo) |
||||
printf("%s", sql.data); |
||||
result = PQexec(conn, sql.data); |
||||
if (PQresultStatus(result) != PGRES_COMMAND_OK) |
||||
{ |
||||
fprintf(stderr, _("%s: language removal failed: %s"), |
||||
progname, PQerrorMessage(conn)); |
||||
PQfinish(conn); |
||||
exit(1); |
||||
} |
||||
|
||||
PQfinish(conn); |
||||
exit(0); |
||||
} |
||||
|
||||
|
||||
static void |
||||
help(const char *progname) |
||||
{ |
||||
printf(_("%s removes a procedural language from a database.\n\n"), progname); |
||||
printf(_("Usage:\n")); |
||||
printf(_(" %s [OPTION]... LANGNAME [DBNAME]\n"), progname); |
||||
printf(_("\nOptions:\n")); |
||||
printf(_(" -d, --dbname=DBNAME database to install language in\n")); |
||||
printf(_(" -e, --echo show the commands being sent to the server\n")); |
||||
printf(_(" -l, --list show a list of currently installed languages\n")); |
||||
printf(_(" -h, --host=HOSTNAME database server host\n")); |
||||
printf(_(" -p, --port=PORT database server port\n")); |
||||
printf(_(" -U, --username=USERNAME user name to connect as\n")); |
||||
printf(_(" -W, --password prompt for password\n")); |
||||
printf(_(" --help show this help, then exit\n")); |
||||
printf(_(" --version output version information, then exit\n")); |
||||
printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n")); |
||||
} |
||||
@ -1,159 +0,0 @@ |
||||
#!/bin/sh |
||||
#------------------------------------------------------------------------- |
||||
# |
||||
# dropuser-- |
||||
# Utility for removing a user from the PostgreSQL database. |
||||
# |
||||
# Portions Copyright (c) 1996-2002, PostgreSQL Global Development Group |
||||
# Portions Copyright (c) 1994, Regents of the University of California |
||||
# |
||||
# |
||||
# IDENTIFICATION |
||||
# $Header: /cvsroot/pgsql/src/bin/scripts/Attic/dropuser,v 1.21 2002/10/18 22:05:36 petere Exp $ |
||||
# |
||||
# Note - this should NOT be setuid. |
||||
# |
||||
#------------------------------------------------------------------------- |
||||
|
||||
CMDNAME=`basename "$0"` |
||||
PATHNAME=`echo "$0" | sed "s,$CMDNAME\$,,"` |
||||
|
||||
PSQLOPT= |
||||
forcedel=t |
||||
DelUser= |
||||
|
||||
# These handle spaces/tabs in identifiers |
||||
_IFS="$IFS" |
||||
NL=" |
||||
" |
||||
# Check for echo -n vs echo \c |
||||
|
||||
if echo '\c' | grep -s c >/dev/null 2>&1 |
||||
then |
||||
ECHO_N="echo -n" |
||||
ECHO_C="" |
||||
else |
||||
ECHO_N="echo" |
||||
ECHO_C='\c' |
||||
fi |
||||
|
||||
|
||||
while [ "$#" -gt 0 ] |
||||
do |
||||
case "$1" in |
||||
--help|-\?) |
||||
usage=t |
||||
break |
||||
;; |
||||
# options passed on to psql |
||||
--host|-h) |
||||
PSQLOPT="$PSQLOPT -h $2" |
||||
shift;; |
||||
-h*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--host=*) |
||||
PSQLOPT="$PSQLOPT -h `echo \"$1\" | sed 's/^--host=//'`" |
||||
;; |
||||
--port|-p) |
||||
PSQLOPT="$PSQLOPT -p $2" |
||||
shift;; |
||||
-p*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--port=*) |
||||
PSQLOPT="$PSQLOPT -p `echo \"$1\" | sed 's/^--port=//'`" |
||||
;; |
||||
# Note: These two specify the user to connect as (like in psql), |
||||
# not the user you're dropping. |
||||
--username|-U) |
||||
PSQLOPT="$PSQLOPT -U $2" |
||||
shift;; |
||||
-U*) |
||||
PSQLOPT="$PSQLOPT $1" |
||||
;; |
||||
--username=*) |
||||
PSQLOPT="$PSQLOPT -U `echo \"$1\" | sed 's/^--username=//'`" |
||||
;; |
||||
--password|-W) |
||||
PSQLOPT="$PSQLOPT -W" |
||||
;; |
||||
--echo|-e) |
||||
PSQLOPT="$PSQLOPT -e" |
||||
;; |
||||
--quiet|-q) |
||||
PSQLOPT="$PSQLOPT -o /dev/null" |
||||
;; |
||||
# other options |
||||
--interactive|-i) |
||||
forcedel=f |
||||
;; |
||||
-*) |
||||
echo "$CMDNAME: invalid option: $1" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
;; |
||||
*) |
||||
DelUser="$1" |
||||
if [ "$#" -ne 1 ]; then |
||||
echo "$CMDNAME: invalid option: $2" 1>&2 |
||||
echo "Try '$CMDNAME --help' for more information." 1>&2 |
||||
exit 1 |
||||
fi |
||||
;; |
||||
esac |
||||
shift; |
||||
done |
||||
|
||||
|
||||
if [ "$usage" ]; then |
||||
echo "$CMDNAME removes a PostgreSQL user." |
||||
echo |
||||
echo "Usage:" |
||||
echo " $CMDNAME [OPTION]... [USERNAME]" |
||||
echo |
||||
echo "Options:" |
||||
echo " -e, --echo show the query being sent to the backend" |
||||
echo " -i, --interactive prompt before deleting anything" |
||||
echo " -q, --quiet don't write any messages" |
||||
echo " -h, --host=HOSTNAME database server host" |
||||
echo " -p, --port=PORT database server port" |
||||
echo " -U, --username=USERNAME user name to connect as (not the one to drop)" |
||||
echo " -W, --password prompt for password to connect" |
||||
echo " --help show this help, then exit" |
||||
echo |
||||
echo "Report bugs to <pgsql-bugs@postgresql.org>." |
||||
exit 0 |
||||
fi |
||||
|
||||
# Prompt for username if missing |
||||
|
||||
if [ -z "$DelUser" ]; then |
||||
$ECHO_N "Enter name of user to delete: "$ECHO_C |
||||
IFS="$NL" |
||||
read DelUser |
||||
IFS="$_IFS" |
||||
[ "$?" -ne 0 ] && exit 1 |
||||
fi |
||||
|
||||
|
||||
if [ "$forcedel" = f ]; then |
||||
echo "User \"$DelUser\" will be permanently deleted." |
||||
$ECHO_N "Are you sure? (y/n) "$ECHO_C |
||||
read REPLY |
||||
|
||||
[ "$?" -eq 1 ] && exit 1 |
||||
[ "$REPLY" != "y" -a "$REPLY" != "Y" ] && exit 0 |
||||
fi |
||||
|
||||
|
||||
DelUser=`echo "$DelUser" | sed 's/\"/\\\"/g'` |
||||
|
||||
${PATHNAME}psql $PSQLOPT -d template1 -c "SET autocommit TO 'on';DROP USER \"$DelUser\"" |
||||
|
||||
if [ "$?" -ne 0 ]; then |
||||
echo "$CMDNAME: deletion of user \"$DelUser\" failed" 1>&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
exit 0 |
||||
@ -0,0 +1,156 @@ |
||||
/*-------------------------------------------------------------------------
|
||||
* |
||||
* dropuser |
||||
* |
||||
* Portions Copyright (c) 1996-2003, PostgreSQL Global Development Group |
||||
* Portions Copyright (c) 1994, Regents of the University of California |
||||
* |
||||
* $Header: /cvsroot/pgsql/src/bin/scripts/dropuser.c,v 1.1 2003/03/18 22:19:47 petere Exp $ |
||||
* |
||||
*------------------------------------------------------------------------- |
||||
*/ |
||||
|
||||
#include "postgres_fe.h" |
||||
#include "common.h" |
||||
#include "dumputils.h" |
||||
|
||||
|
||||
static void help(const char *progname); |
||||
|
||||
|
||||
int |
||||
main(int argc, char *argv[]) |
||||
{ |
||||
static struct option long_options[] = { |
||||
{"host", required_argument, NULL, 'h'}, |
||||
{"port", required_argument, NULL, 'p'}, |
||||
{"username", required_argument, NULL, 'U'}, |
||||
{"password", no_argument, NULL, 'W'}, |
||||
{"echo", no_argument, NULL, 'e'}, |
||||
{"quiet", no_argument, NULL, 'q'}, |
||||
{"interactive", no_argument, NULL, 'i'}, |
||||
{NULL, 0, NULL, 0} |
||||
}; |
||||
|
||||
char *progname; |
||||
int optindex; |
||||
int c; |
||||
|
||||
char *dropuser = NULL; |
||||
char *host = NULL; |
||||
char *port = NULL; |
||||
char *username = NULL; |
||||
bool password = false; |
||||
bool echo = false; |
||||
bool quiet = false; |
||||
bool interactive = false; |
||||
|
||||
PQExpBufferData sql; |
||||
|
||||
PGconn *conn; |
||||
PGresult *result; |
||||
|
||||
progname = get_progname(argv[0]); |
||||
init_nls(); |
||||
handle_help_version_opts(argc, argv, "dropuser", help); |
||||
|
||||
while ((c = getopt_long(argc, argv, "h:p:U:Weqi", long_options, &optindex)) != -1) |
||||
{ |
||||
switch (c) |
||||
{ |
||||
case 'h': |
||||
host = optarg; |
||||
break; |
||||
case 'p': |
||||
port = optarg; |
||||
break; |
||||
case 'U': |
||||
username = optarg; |
||||
break; |
||||
case 'W': |
||||
password = true; |
||||
break; |
||||
case 'e': |
||||
echo = true; |
||||
break; |
||||
case 'q': |
||||
quiet = true; |
||||
break; |
||||
case 'i': |
||||
interactive = true; |
||||
break; |
||||
default: |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
} |
||||
|
||||
switch (argc - optind) |
||||
{ |
||||
case 0: |
||||
break; |
||||
case 1: |
||||
dropuser = argv[optind]; |
||||
break; |
||||
default: |
||||
fprintf(stderr, _("%s: too many command-line arguments (first is '%s')\n"), |
||||
progname, argv[optind + 1]); |
||||
fprintf(stderr, _("Try '%s --help' for more information.\n"), progname); |
||||
exit(1); |
||||
} |
||||
|
||||
if (dropuser == NULL) |
||||
dropuser = simple_prompt("Enter name of user to drop: ", 128, true); |
||||
|
||||
if (interactive) |
||||
{ |
||||
char *reply; |
||||
|
||||
printf(_("User \"%s\" will be permanently deleted.\n"), dropuser); |
||||
reply = simple_prompt("Are you sure? (y/n) ", 1, true); |
||||
if (reply[0] != 'y' && reply[0] != 'Y') |
||||
exit(0); |
||||
} |
||||
|
||||
initPQExpBuffer(&sql); |
||||
appendPQExpBuffer(&sql, "SET autocommit TO on;\nDROP USER %s;\n", fmtId(dropuser)); |
||||
|
||||
conn = connectDatabase("template1", host, port, username, password, progname); |
||||
|
||||
if (echo) |
||||
printf("%s", sql.data); |
||||
result = PQexec(conn, sql.data); |
||||
|
||||
if (PQresultStatus(result) != PGRES_COMMAND_OK) |
||||
{ |
||||
fprintf(stderr, _("%s: deletion of user %s failed: %s"), |
||||
progname, dropuser, PQerrorMessage(conn)); |
||||
PQfinish(conn); |
||||
exit(1); |
||||
} |
||||
|
||||
PQfinish(conn); |
||||
if (!quiet) |
||||
puts("DROP USER"); |
||||
exit(0); |
||||
} |
||||
|
||||
|
||||
static void |
||||
help(const char *progname) |
||||
{ |
||||
printf(_("%s removes a PostgreSQL user.\n\n"), progname); |
||||
printf(_("Usage:\n")); |
||||
printf(_(" %s [OPTION]... [USERNAME]\n"), progname); |
||||
printf(_("\nOptions:\n")); |
||||
printf(_(" -e, --echo show the commands being sent to the server\n")); |
||||
printf(_(" -i, --interactive prompt before deleting anything\n")); |
||||
printf(_(" -q, --quiet don't write any messages\n")); |
||||
printf(_(" -h, --host=HOSTNAME database server host\n")); |
||||
printf(_(" -p, --port=PORT database server port\n")); |
||||
printf(_(" -U, --username=USERNAME user name to connect as (not the one to drop)\n")); |
||||
printf(_(" -W, --password prompt for password to connect\n")); |
||||
printf(_(" --help show this help, then exit\n")); |
||||
printf(_(" --version output version information, then exit\n")); |
||||
printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n")); |
||||
} |
||||
@ -0,0 +1,7 @@ |
||||
# $Header: /cvsroot/pgsql/src/bin/scripts/nls.mk,v 1.1 2003/03/18 22:19:47 petere Exp $
|
||||
CATALOG_NAME := pgscripts
|
||||
AVAIL_LANGUAGES :=
|
||||
GETTEXT_FILES := createdb.c createlang.c createuser.c \
|
||||
dropdb.c droplang.c dropuser.c \
|
||||
common.c
|
||||
GETTEXT_TRIGGERS:= _ simple_prompt
|
||||
Loading…
Reference in new issue