Support SQL92 delimited identifiers by checking some attribute names
authorThomas G. Lockhart
Thu, 30 Oct 1997 16:48:03 +0000 (16:48 +0000)
committerThomas G. Lockhart
Thu, 30 Oct 1997 16:48:03 +0000 (16:48 +0000)
 for mixed-case and surrounding with double quotes.

src/bin/pg_dump/common.c
src/bin/pg_dump/pg_dump.c
src/bin/pg_dump/pg_dump.h

index 8e1a7709cc34d4b5119dad217dbf92868212d615..5f4dc96a8e2c426ac218999d85323eb4f8169344 100644 (file)
@@ -7,7 +7,7 @@
  *
  *
  * IDENTIFICATION
- *   $Header: /cvsroot/pgsql/src/bin/pg_dump/common.c,v 1.17 1997/10/02 13:57:03 vadim Exp $
+ *   $Header: /cvsroot/pgsql/src/bin/pg_dump/common.c,v 1.18 1997/10/30 16:47:57 thomas Exp $
  *
  * Modifications - 6/12/96 - [email protected] - version 1.13.dhb.2
  *
@@ -22,6 +22,7 @@
 #include 
 #include 
 #include 
+#include 
 #include          /* for MAXHOSTNAMELEN on most */
 #ifdef sparc_solaris
 #include              /* for MAXHOSTNAMELEN on some */
@@ -478,3 +479,29 @@ isArchiveName(const char *relname)
 {
    return (strlen(relname) > 1 && relname[1] == ',');
 }
+
+/*
+ * fmtId
+ *
+ * checks input string for non-lowercase characters
+ * returns pointer to input string or string surrounded by double quotes
+ */
+const char *
+fmtId(const char *rawid)
+{
+   const char *cp;
+   static char id[MAXQUERYLEN];
+
+   for (cp = rawid; *cp != '\0'; cp++)
+       if (! (islower(*cp) || isdigit(*cp) || (*cp == '_'))) break;
+
+   if (*cp != '\0') {
+       strcpy(id, "\"");
+       strcat(id, rawid);
+       strcat(id, "\"");
+       cp = id;
+   } else {
+       cp = rawid;
+   }
+   return(cp);
+} /* fmtId() */
index 197f71897e78b21ed1e920ecbb12700ebab718d8..f6f68135226d10b8fb387a81b455f91e9edb8eac 100644 (file)
@@ -21,7 +21,7 @@
  *
  *
  * IDENTIFICATION
- *   $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_dump.c,v 1.51 1997/10/30 03:59:46 momjian Exp $
+ *   $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_dump.c,v 1.52 1997/10/30 16:47:59 thomas Exp $
  *
  * Modifications - 6/10/96 - [email protected] - version 1.13.dhb
  *
@@ -205,14 +205,14 @@ dumpClasses_nodumpData(FILE *fout, const char *classname, const bool oids)
    if (oids)
    {
        fprintf(fout, "COPY %s WITH OIDS FROM stdin;\n",
-               classname);
+               fmtId(classname));
        sprintf(query, "COPY %s WITH OIDS TO stdout;\n",
-               classname);
+               fmtId(classname));
    }
    else
    {
-       fprintf(fout, "COPY %s FROM stdin;\n", classname);
-       sprintf(query, "COPY %s TO stdout;\n", classname);
+       fprintf(fout, "COPY %s FROM stdin;\n", fmtId(classname));
+       sprintf(query, "COPY %s TO stdout;\n", fmtId(classname));
    }
    res = PQexec(g_conn, query);
    if (!res)
@@ -309,7 +309,7 @@ dumpClasses_dumpData(FILE *fout, const char *classname,
    tuple = 0;
    while (tuple < PQntuples(res))
    {
-       fprintf(fout, "insert into %s ", classname);
+       fprintf(fout, "insert into %s ", fmtId(classname));
        if (attrNames)
        {
            int         j;
@@ -323,7 +323,7 @@ dumpClasses_dumpData(FILE *fout, const char *classname,
                    sprintf(q, "%s%s%s",
                            q,
                            (actual_atts > 0) ? "," : "",
-                           tblinfo.attnames[j]);
+                           fmtId(tblinfo.attnames[j]));
                    actual_atts++;
                }
            }
@@ -1944,12 +1944,12 @@ dumpOneFunc(FILE *fout, FuncInfo *finfo, int i,
        sprintf(q, "%s%s%s",
                q,
                (j > 0) ? "," : "",
-               typname);
+               fmtId(typname));
    }
    sprintf(q, "%s ) RETURNS %s%s AS '%s' LANGUAGE '%s';\n",
            q,
            (finfo[i].retset) ? " SETOF " : "",
-           findTypeByOid(tinfo, numTypes, finfo[i].prorettype),
+           fmtId(findTypeByOid(tinfo, numTypes, finfo[i].prorettype)),
            (finfo[i].lang == INTERNALlanguageId) ? finfo[i].prosrc :
                (finfo[i].lang == ClanguageId) ? finfo[i].probin :
                    (finfo[i].lang == SQLlanguageId) ? finfo[i].prosrc : "unknown",
@@ -2005,13 +2005,13 @@ dumpOprs(FILE *fout, OprInfo *oprinfo, int numOperators,
            strcmp(oprinfo[i].oprkind, "b") == 0)
        {
            sprintf(leftarg, ", LEFTARG = %s ",
-                   findTypeByOid(tinfo, numTypes, oprinfo[i].oprleft));
+                   fmtId(findTypeByOid(tinfo, numTypes, oprinfo[i].oprleft)));
        }
        if (strcmp(oprinfo[i].oprkind, "l") == 0 ||
            strcmp(oprinfo[i].oprkind, "b") == 0)
        {
            sprintf(rightarg, ", RIGHTARG = %s ",
-                   findTypeByOid(tinfo, numTypes, oprinfo[i].oprright));
+                   fmtId(findTypeByOid(tinfo, numTypes, oprinfo[i].oprright)));
        }
        if (strcmp(oprinfo[i].oprcom, "0") == 0)
            commutator[0] = '\0';
@@ -2094,7 +2094,7 @@ dumpAggs(FILE *fout, AggInfo *agginfo, int numAggs,
        
        sprintf(basetype,
                "BASETYPE = %s, ",
-             findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype));
+             fmtId(findTypeByOid(tinfo, numTypes, agginfo[i].aggbasetype)));
 
        if (strcmp(agginfo[i].aggtransfn1, "-") == 0)
            sfunc1[0] = '\0';
@@ -2103,7 +2103,7 @@ dumpAggs(FILE *fout, AggInfo *agginfo, int numAggs,
            sprintf(sfunc1,
                    "SFUNC1 = %s, STYPE1 = %s",
                    agginfo[i].aggtransfn1,
-                   findTypeByOid(tinfo, numTypes, agginfo[i].aggtranstype1));
+                   fmtId(findTypeByOid(tinfo, numTypes, agginfo[i].aggtranstype1)));
            if (agginfo[i].agginitval1)
                sprintf(sfunc1, "%s, INITCOND1 = '%s'",
                        sfunc1, agginfo[i].agginitval1);
@@ -2117,7 +2117,7 @@ dumpAggs(FILE *fout, AggInfo *agginfo, int numAggs,
            sprintf(sfunc2,
                    "SFUNC2 = %s, STYPE2 = %s",
                    agginfo[i].aggtransfn2,
-              findTypeByOid(tinfo, numTypes, agginfo[i].aggtranstype2));
+              fmtId(findTypeByOid(tinfo, numTypes, agginfo[i].aggtranstype2)));
            if (agginfo[i].agginitval2)
                sprintf(sfunc2, "%s, INITCOND2 = '%s'",
                        sfunc2, agginfo[i].agginitval2);
@@ -2213,7 +2213,7 @@ dumpTables(FILE *fout, TableInfo *tblinfo, int numTables,
 
            fprintf(fout, "\\connect - %s\n", tblinfo[i].usename);
 
-           sprintf(q, "CREATE TABLE %s (", tblinfo[i].relname);
+           sprintf(q, "CREATE TABLE %s (", fmtId(tblinfo[i].relname));
            actual_atts = 0;
            for (j = 0; j < tblinfo[i].numatts; j++)
            {
@@ -2226,7 +2226,7 @@ dumpTables(FILE *fout, TableInfo *tblinfo, int numTables,
                        sprintf(q, "%s%s%s char",
                                q,
                                (actual_atts > 0) ? ", " : "",
-                               tblinfo[i].attnames[j]);
+                               fmtId(tblinfo[i].attnames[j]));
 
                        /* stored length can be -1 (variable) */
                        if (tblinfo[i].attlen[j] > 0)
@@ -2240,7 +2240,7 @@ dumpTables(FILE *fout, TableInfo *tblinfo, int numTables,
                        sprintf(q, "%s%s%s %s",
                                q,
                                (actual_atts > 0) ? ", " : "",
-                               tblinfo[i].attnames[j],
+                               fmtId(tblinfo[i].attnames[j]),
                                tblinfo[i].typnames[j]);
 
                        /* stored length can be -1 (variable) */
@@ -2255,8 +2255,8 @@ dumpTables(FILE *fout, TableInfo *tblinfo, int numTables,
                        sprintf(q, "%s%s%s %s",
                                q,
                                (actual_atts > 0) ? ", " : "",
-                               tblinfo[i].attnames[j],
-                               tblinfo[i].typnames[j]);
+                               fmtId(tblinfo[i].attnames[j]),
+                               fmtId(tblinfo[i].typnames[j]));
                        actual_atts++;
                    }
                    if (tblinfo[i].adef_expr[j] != NULL)
@@ -2347,7 +2347,7 @@ dumpIndices(FILE *fout, IndInfo *indinfo, int numIndices,
    for (i = 0; i < numIndices; i++)
    {
        tableInd = findTableByName(tblinfo, numTables,
-                                  indinfo[i].indrelname);
+                                  fmtId(indinfo[i].indrelname));
 
        if (strcmp(indinfo[i].indproc, "0") == 0)
        {
@@ -2420,7 +2420,7 @@ dumpIndices(FILE *fout, IndInfo *indinfo, int numIndices,
                attname = tblinfo[tableInd].attnames[indkey];
            if (funcname)
                sprintf(attlist + strlen(attlist), "%s%s",
-                       (k == 0) ? "" : ", ", attname);
+                       (k == 0) ? "" : ", ", fmtId(attname));
            else
            {
                if (k >= nclass)
@@ -2431,7 +2431,7 @@ dumpIndices(FILE *fout, IndInfo *indinfo, int numIndices,
                    exit_nicely(g_conn);
                }
                sprintf(attlist + strlen(attlist), "%s%s %s",
-                       (k == 0) ? "" : ", ", attname, classname[k]);
+                       (k == 0) ? "" : ", ", fmtId(attname), fmtId(classname[k]));
                free(classname[k]);
            }
        }
@@ -2441,13 +2441,13 @@ dumpIndices(FILE *fout, IndInfo *indinfo, int numIndices,
 
            sprintf(q, "CREATE %s INDEX %s on %s using %s (",
              (strcmp(indinfo[i].indisunique, "t") == 0) ? "UNIQUE" : "",
-                   indinfo[i].indexrelname,
-                   indinfo[i].indrelname,
+                   fmtId(indinfo[i].indexrelname),
+                   fmtId(indinfo[i].indrelname),
                    indinfo[i].indamname);
            if (funcname)
            {
                sprintf(q, "%s %s (%s) %s );\n",
-                       q, funcname, attlist, classname[0]);
+                       q, funcname, attlist, fmtId(classname[0]));
                free(funcname);
                free(classname[0]);
            }
@@ -2666,7 +2666,7 @@ dumpSequence(FILE *fout, TableInfo tbinfo)
    sprintf(query,
            "SELECT sequence_name, last_value, increment_by, max_value, "
            "min_value, cache_value, is_cycled, is_called from %s",
-           tbinfo.relname);
+           fmtId(tbinfo.relname));
 
    res = PQexec(g_conn, query);
    if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
@@ -2706,7 +2706,7 @@ dumpSequence(FILE *fout, TableInfo tbinfo)
    sprintf(query,
            "CREATE SEQUENCE %s start %d increment %d maxvalue %d "
            "minvalue %d  cache %d %s;\n",
-           tbinfo.relname, last, incby, maxv, minv, cache,
+           fmtId(tbinfo.relname), last, incby, maxv, minv, cache,
            (cycled == 't') ? "cycle" : "");
 
    fputs(query, fout);
index 16b1ccd60991975eaec07865a78748a14166958a..995f195c46c5f6ce2b530ab75277a082518e13b9 100644 (file)
@@ -5,7 +5,7 @@
  *
  * Copyright (c) 1994, Regents of the University of California
  *
- * $Id: pg_dump.h,v 1.25 1997/10/02 13:57:07 vadim Exp $
+ * $Id: pg_dump.h,v 1.26 1997/10/30 16:48:03 thomas Exp $
  *
  * Modifications - 6/12/96 - [email protected] - version 1.13.dhb.2
  *
@@ -233,5 +233,8 @@ extern void
 dumpIndices(FILE *fout, IndInfo *indinfo, int numIndices,
            TableInfo *tbinfo, int numTables, const char *tablename);
 
+extern const char *
+fmtId(const char *identifier);
+
 /* largest query string size */
 #define MAXQUERYLEN  5000