granicus.if.org Git - postgresql/blob - src/bin/scripts/droplang.c

   1 /*-------------------------------------------------------------------------
   2  *
   3  * droplang
   4  *
   5  * Portions Copyright (c) 1996-2007, 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.22 2007/01/05 22:19:50 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\", "
 144                                 "(CASE WHEN lanpltrusted THEN '%s' ELSE '%s' END) as \"%s\" "
 145                                                   "FROM pg_catalog.pg_language WHERE lanispl;",
 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          * Force schema search path to be just pg_catalog, so that we don't have
 177          * to be paranoid about search paths below.
 178          */
 179         executeCommand(conn, "SET search_path = pg_catalog;", progname, echo);
 180
 181         /*
 182          * Make sure the language is installed and find the OIDs of the handler
 183          * and validator functions
 184          */
 185         printfPQExpBuffer(&sql, "SELECT lanplcallfoid, lanvalidator "
 186                                           "FROM pg_language WHERE lanname = '%s' AND lanispl;",
 187                                           langname);
 188         result = executeQuery(conn, sql.data, progname, echo);
 189         if (PQntuples(result) == 0)
 190         {
 191                 PQfinish(conn);
 192                 fprintf(stderr, _("%s: language \"%s\" is not installed in "
 193                                                   "database \"%s\"\n"),
 194                                 progname, langname, dbname);
 195                 exit(1);
 196         }
 197         lanplcallfoid = atooid(PQgetvalue(result, 0, 0));
 198         lanvalidator = atooid(PQgetvalue(result, 0, 1));
 199         PQclear(result);
 200
 201         /*
 202          * Check that there are no functions left defined in that language
 203          */
 204         printfPQExpBuffer(&sql, "SELECT count(proname) FROM pg_proc P, "
 205                                           "pg_language L WHERE P.prolang = L.oid "
 206                                           "AND L.lanname = '%s';", langname);
 207         result = executeQuery(conn, sql.data, progname, echo);
 208         if (strcmp(PQgetvalue(result, 0, 0), "0") != 0)
 209         {
 210                 PQfinish(conn);
 211                 fprintf(stderr,
 212                                 _("%s: still %s functions declared in language \"%s\"; "
 213                                   "language not removed\n"),
 214                                 progname, PQgetvalue(result, 0, 0), langname);
 215                 exit(1);
 216         }
 217         PQclear(result);
 218
 219         /*
 220          * Check that the handler function isn't used by some other language
 221          */
 222         printfPQExpBuffer(&sql, "SELECT count(*) FROM pg_language "
 223                                           "WHERE lanplcallfoid = %u AND lanname <> '%s';",
 224                                           lanplcallfoid, langname);
 225         result = executeQuery(conn, sql.data, progname, echo);
 226         if (strcmp(PQgetvalue(result, 0, 0), "0") == 0)
 227                 keephandler = false;
 228         else
 229                 keephandler = true;
 230         PQclear(result);
 231
 232         /*
 233          * Find the handler name
 234          */
 235         if (!keephandler)
 236         {
 237                 printfPQExpBuffer(&sql, "SELECT proname, (SELECT nspname "
 238                                                   "FROM pg_namespace ns WHERE ns.oid = pronamespace) "
 239                                                   "AS prons FROM pg_proc WHERE oid = %u;",
 240                                                   lanplcallfoid);
 241                 result = executeQuery(conn, sql.data, progname, echo);
 242                 handler = strdup(PQgetvalue(result, 0, 0));
 243                 handler_ns = strdup(PQgetvalue(result, 0, 1));
 244                 PQclear(result);
 245         }
 246         else
 247         {
 248                 handler = NULL;
 249                 handler_ns = NULL;
 250         }
 251
 252         /*
 253          * Check that the validator function isn't used by some other language
 254          */
 255         if (OidIsValid(lanvalidator))
 256         {
 257                 printfPQExpBuffer(&sql, "SELECT count(*) FROM pg_language "
 258                                                   "WHERE lanvalidator = %u AND lanname <> '%s';",
 259                                                   lanvalidator, langname);
 260                 result = executeQuery(conn, sql.data, progname, echo);
 261                 if (strcmp(PQgetvalue(result, 0, 0), "0") == 0)
 262                         keepvalidator = false;
 263                 else
 264                         keepvalidator = true;
 265                 PQclear(result);
 266         }
 267         else
 268                 keepvalidator = true;   /* don't try to delete it */
 269
 270         /*
 271          * Find the validator name
 272          */
 273         if (!keepvalidator)
 274         {
 275                 printfPQExpBuffer(&sql, "SELECT proname, (SELECT nspname "
 276                                                   "FROM pg_namespace ns WHERE ns.oid = pronamespace) "
 277                                                   "AS prons FROM pg_proc WHERE oid = %u;",
 278                                                   lanvalidator);
 279                 result = executeQuery(conn, sql.data, progname, echo);
 280                 validator = strdup(PQgetvalue(result, 0, 0));
 281                 validator_ns = strdup(PQgetvalue(result, 0, 1));
 282                 PQclear(result);
 283         }
 284         else
 285         {
 286                 validator = NULL;
 287                 validator_ns = NULL;
 288         }
 289
 290         /*
 291          * Drop the language and the functions
 292          */
 293         printfPQExpBuffer(&sql, "DROP LANGUAGE \"%s\";\n", langname);
 294         if (!keephandler)
 295                 appendPQExpBuffer(&sql, "DROP FUNCTION \"%s\".\"%s\" ();\n",
 296                                                   handler_ns, handler);
 297         if (!keepvalidator)
 298                 appendPQExpBuffer(&sql, "DROP FUNCTION \"%s\".\"%s\" (oid);\n",
 299                                                   validator_ns, validator);
 300         if (echo)
 301                 printf("%s", sql.data);
 302         result = PQexec(conn, sql.data);
 303         if (PQresultStatus(result) != PGRES_COMMAND_OK)
 304         {
 305                 fprintf(stderr, _("%s: language removal failed: %s"),
 306                                 progname, PQerrorMessage(conn));
 307                 PQfinish(conn);
 308                 exit(1);
 309         }
 310
 311         PQclear(result);
 312         PQfinish(conn);
 313         exit(0);
 314 }
 315
 316
 317 static void
 318 help(const char *progname)
 319 {
 320         printf(_("%s removes a procedural language from a database.\n\n"), progname);
 321         printf(_("Usage:\n"));
 322         printf(_("  %s [OPTION]... LANGNAME [DBNAME]\n"), progname);
 323         printf(_("\nOptions:\n"));
 324         printf(_("  -d, --dbname=DBNAME       database from which to remove the language\n"));
 325         printf(_("  -e, --echo                show the commands being sent to the server\n"));
 326         printf(_("  -l, --list                show a list of currently installed languages\n"));
 327         printf(_("  -h, --host=HOSTNAME       database server host or socket directory\n"));
 328         printf(_("  -p, --port=PORT           database server port\n"));
 329         printf(_("  -U, --username=USERNAME   user name to connect as\n"));
 330         printf(_("  -W, --password            prompt for password\n"));
 331         printf(_("  --help                    show this help, then exit\n"));
 332         printf(_("  --version                 output version information, then exit\n"));
 333         printf(_("\nReport bugs to <pgsql-bugs@postgresql.org>.\n"));
 334 }