src/bin/scripts/droplang.c

   1 /*-------------------------------------------------------------------------
   2  *
   3  * droplang
   4  *
   5  * Portions Copyright (c) 1996-2005, PostgreSQL Global Development Group
   6  * Portions Copyright (c) 1994, Regents of the University of California
   7  *
   8  * $PostgreSQL: pgsql/src/bin/scripts/droplang.c,v 1.16 2005/07/10 14:26:30 momjian Exp $
   9  *
  10  *-------------------------------------------------------------------------
  11  */
  12
  13 #include "postgres_fe.h"
  14 #include "common.h"
  15 #include "print.h"
  16
  17 #define atooid(x)  ((Oid) strtoul((x), NULL, 10))
  18
  19
  20 static void help(const char *progname);
  21
  22
  23 int
  24 main(int argc, char *argv[])
  25 {
  26         static struct option long_options[] = {
  27                 {"list", no_argument, NULL, 'l'},
  28                 {"host", required_argument, NULL, 'h'},
  29                 {"port", required_argument, NULL, 'p'},
  30                 {"username", required_argument, NULL, 'U'},
  31                 {"password", no_argument, NULL, 'W'},
  32                 {"dbname", required_argument, NULL, 'd'},
  33                 {"echo", no_argument, NULL, 'e'},
  34                 {NULL, 0, NULL, 0}
  35         };
  36
  37         const char *progname;
  38         int                     optindex;
  39         int                     c;
  40
  41         bool            listlangs = false;
  42         const char *dbname = NULL;
  43         char       *host = NULL;
  44         char       *port = NULL;
  45         char       *username = NULL;
  46         bool            password = false;
  47         bool            echo = false;
  48         char       *langname = NULL;
  49
  50         char       *p;
  51         Oid                     lanplcallfoid;
  52         Oid                     lanvalidator;
  53         char       *handler;
  54         char       *validator;
  55         char       *handler_ns;
  56         char       *validator_ns;
  57         bool            keephandler;
  58         bool            keepvalidator;
  59
  60         PQExpBufferData sql;
  61
  62         PGconn     *conn;
  63         PGresult   *result;
  64
  65         progname = get_progname(argv[0]);
  66         set_pglocale_pgservice(argv[0], "pgscripts");
  67
  68         handle_help_version_opts(argc, argv, "droplang", help);
  69
  70         while ((c = getopt_long(argc, argv, "lh:p:U:Wd:e", long_options, &optindex)) != -1)
  71         {
  72                 switch (c)
  73                 {
  74                         case 'l':
  75                                 listlangs = true;
  76                                 break;
  77                         case 'h':
  78                                 host = optarg;
  79                                 break;
  80                         case 'p':
  81                                 port = optarg;
  82                                 break;
  83                         case 'U':
  84                                 username = optarg;
  85                                 break;
  86                         case 'W':
  87                                 password = true;
  88                                 break;
  89                         case 'd':
  90                                 dbname = optarg;
  91                                 break;
  92                         case 'e':
  93                                 echo = true;
  94                                 break;
  95                         default:
  96                                 fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
  97                                 exit(1);
  98                 }
  99         }
 100
 101         if (argc - optind > 0)
 102         {
 103                 if (listlangs)
 104                         dbname = argv[optind++];
 105                 else
 106                 {
 107                         langname = argv[optind++];
 108                         if (argc - optind > 0)
 109                                 dbname = argv[optind++];
 110                 }
 111         }
 112
 113         if (argc - optind > 0)
 114         {
 115                 fprintf(stderr, _("%s: too many command-line arguments (first is \"%s\")\n"),
 116                                 progname, argv[optind]);
 117                 fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
 118                 exit(1);
 119         }
 120
 121         if (dbname == NULL)
 122         {
 123                 if (getenv("PGDATABASE"))
 124                         dbname = getenv("PGDATABASE");
 125                 else if (getenv("PGUSER"))
 126                         dbname = getenv("PGUSER");
 127                 else
 128                         dbname = get_user_name(progname);
 129         }
 130
 131         initPQExpBuffer(&sql);
 132
 133         /*
 134          * List option
 135          */
 136         if (listlangs)
 137         {
 138                 printQueryOpt popt;
 139
 140                 conn = connectDatabase(dbname, host, port, username, password,
 141                                                            progname);
 142
 143                 printfPQExpBuffer(&sql, "SELECT lanname as \"%s\", (CASE "
 144                                                   "WHEN lanpltrusted THEN '%s' ELSE '%s' END) "
 145                                                   "as \"%s\" FROM pg_language WHERE lanispl IS TRUE;",
 146                                                   _("Name"), _("yes"), _("no"), _("Trusted?"));
 147                 result = executeQuery(conn, sql.data, progname, echo);
 148
 149                 memset(&popt, 0, sizeof(popt));
 150                 popt.topt.format = PRINT_ALIGNED;
 151                 popt.topt.border = 1;
 152                 popt.topt.encoding = PQclientEncoding(conn);
 153                 popt.title = _("Procedural Languages");
 154                 printQuery(result, &popt, stdout, NULL);
 155
 156                 PQfinish(conn);
 157                 exit(0);
 158         }
 159
 160         if (langname == NULL)
 161         {
 162                 fprintf(stderr, _("%s: missing required argument language name\n"),
 163                                 progname);
 164                 fprintf(stderr, _("Try \"%s --help\" for more information.\n"),
 165                                 progname);
 166                 exit(1);
 167         }
 168
 169         for (p = langname; *p; p++)
 170                 if (*p >= 'A' && *p <= 'Z')
 171                         *p += ('a' - 'A');
 172
 173         conn = connectDatabase(dbname, host, port, username, password, progname);
 174
 175         /*
 176          * Make sure the language is installed and find the OIDs of the
 177          * handler and validator functions
 178          */
 179         printfPQExpBuffer(&sql, "SELECT lanplcallfoid, lanvalidator "
 180                                           "FROM pg_language WHERE lanname = '%s' AND lanispl;",
 181                                           langname);
 182         result = executeQuery(conn, sql.data, progname, echo);
 183         if (PQntuples(result) == 0)
 184         {
 185                 PQfinish(conn);
 186                 fprintf(stderr, _("%s: language \"%s\" is not installed in "
 187                                                   "database \"%s\"\n"),
 188                                 progname, langname, dbname);
 189                 exit(1);
 190         }
 191         lanplcallfoid = atooid(PQgetvalue(result, 0, 0));
 192         lanvalidator = atooid(PQgetvalue(result, 0, 1));
 193         PQclear(result);
 194
 195         /*
 196          * Check that there are no functions left defined in that language
 197          */
 198         printfPQExpBuffer(&sql, "SELECT count(proname) FROM pg_proc P, "
 199                                           "pg_language L WHERE P.prolang = L.oid "
 200                                           "AND L.lanname = '%s';", langname);
 201         result = executeQuery(conn, sql.data, progname, echo);
 202         if (strcmp(PQgetvalue(result, 0, 0), "0") != 0)
 203         {
 204                 PQfinish(conn);
 205                 fprintf(stderr,
 206                                 _("%s: still %s functions declared in language \"%s\"; "
 207                                   "language not removed\n"),
 208                                 progname, PQgetvalue(result, 0, 0), langname);
 209                 exit(1);
 210         }
 211         PQclear(result);
 212
 213         /*
 214          * Check that the handler function isn't used by some other language
 215          */
 216         printfPQExpBuffer(&sql, "SELECT count(*) FROM pg_language "
 217                                           "WHERE lanplcallfoid = %u AND lanname <> '%s';",
 218                                           lanplcallfoid, langname);
 219         result = executeQuery(conn, sql.data, progname, echo);
 220         if (strcmp(PQgetvalue(result, 0, 0), "0") == 0)
 221                 keephandler = false;
 222         else
 223                 keephandler = true;
 224         PQclear(result);
 225
 226         /*
 227          * Find the handler name
 228          */
 229         if (!keephandler)
 230         {
 231                 printfPQExpBuffer(&sql, "SELECT proname, (SELECT nspname "
 232                                                   "FROM pg_namespace ns WHERE ns.oid = pronamespace) "
 233                                                   "AS prons FROM pg_proc WHERE oid = %u;",
 234                                                   lanplcallfoid);
 235                 result = executeQuery(conn, sql.data, progname, echo);
 236                 handler = strdup(PQgetvalue(result, 0, 0));
 237                 handler_ns = strdup(PQgetvalue(result, 0, 1));
 238                 PQclear(result);
 239         }
 240         else
 241         {
 242                 handler = NULL;
 243                 handler_ns = NULL;
 244         }
 245
 246         /*
 247          * Check that the validator function isn't used by some other language
 248          */
 249         if (OidIsValid(lanvalidator))
 250         {
 251                 printfPQExpBuffer(&sql, "SELECT count(*) FROM pg_language WHERE "
 252                                                   "lanvalidator = %u AND lanname <> '%s';",
 253                                                   lanvalidator, langname);
 254                 result = executeQuery(conn, sql.data, progname, echo);
 255                 if (strcmp(PQgetvalue(result, 0, 0), "0") == 0)
 256                         keepvalidator = false;
 257                 else
 258                         keepvalidator = true;
 259                 PQclear(result);
 260         }
 261         else
 262                 keepvalidator = true;   /* don't try to delete it */
 263
 264         /*
 265          * Find the validator name
 266          */
 267         if (!keepvalidator)
 268         {
 269                 printfPQExpBuffer(&sql, "SELECT proname, (SELECT nspname "
 270                                                   "FROM pg_namespace ns WHERE ns.oid = pronamespace) "
 271                                                   "AS prons FROM pg_proc WHERE oid = %u;",
 272                                                   lanvalidator);
 273                 result = executeQuery(conn, sql.data, progname, echo);
 274                 validator = strdup(PQgetvalue(result, 0, 0));
 275                 validator_ns = strdup(PQgetvalue(result, 0, 1));
 276                 PQclear(result);
 277         }
 278         else
 279         {
 280                 validator = NULL;
 281                 validator_ns = NULL;
 282         }
 283
 284         /*
 285          * Drop the language and the functions
 286          */
 287         printfPQExpBuffer(&sql, "DROP LANGUAGE \"%s\";\n", langname);
 288         if (!keephandler)
 289                 appendPQExpBuffer(&sql, "DROP FUNCTION \"%s\".\"%s\" ();\n",
 290                                                   handler_ns, handler);
 291         if (!keepvalidator)
 292                 appendPQExpBuffer(&sql, "DROP FUNCTION \"%s\".\"%s\" (oid);\n",
 293                                                   validator_ns, validator);
 294         if (echo)
 295                 printf("%s", sql.data);
 296         result = PQexec(conn, sql.data);
 297         if (PQresultStatus(result) != PGRES_COMMAND_OK)
 298         {
 299                 fprintf(stderr, _("%s: language removal failed: %s"),
 300                                 progname, PQerrorMessage(conn));
 301                 PQfinish(conn);
 302                 exit(1);
 303         }
 304
 305         PQfinish(conn);
 306         exit(0);
 307 }
 308
 309
 310 static void
 311 help(const char *progname)
 312 {
 313         printf(_("%s removes a procedural language from a database.\n\n"), progname);
 314         printf(_("Usage:\n"));
 315         printf(_("  %s [OPTION]... LANGNAME [DBNAME]\n"), progname);
 316         printf(_("\nOptions:\n"));
 317         printf(_("  -d, --dbname=DBNAME       database from which to remove the language\n"));
 318         printf(_("  -e, --echo                show the commands being sent to the server\n"));
 319         printf(_("  -l, --list                show a list of currently installed languages\n"));
 320         printf(_("  -h, --host=HOSTNAME       database server host or socket directory\n"));
 321         printf(_("  -p, --port=PORT           database server port\n"));
 322         printf(_("  -U, --username=USERNAME   user name to connect as\n"));
 323         printf(_("  -W, --password            prompt for password\n"));
 324         printf(_("  --help                    show this help, then exit\n"));
 325         printf(_("  --version                 output version information, then exit\n"));
 326         printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n"));
 327 }