Change schema query to involve the use of an optional schema parameter.
authorTilghman Lesher <tilghman@meg.abyt.es>
Tue, 6 Oct 2009 19:31:39 +0000 (19:31 +0000)
committerTilghman Lesher <tilghman@meg.abyt.es>
Tue, 6 Oct 2009 19:31:39 +0000 (19:31 +0000)
This change is done in such a way as to allow the driver to continue to
function with older databases which don't have these features.
(closes issue #16000)
 Reported by: jamicque
 Patches:
       20091002__issue16000.diff.txt uploaded by tilghman (license 14)
       20091002__issue16000__1.6.1.diff.txt uploaded by tilghman (license 14)
 Tested by: jamicque

git-svn-id: https://origsvn.digium.com/svn/asterisk/trunk@222309 65c4cc65-6c06-0410-ace0-fbb531ad65f3

cdr/cdr_pgsql.c
res/res_config_pgsql.c

index c5b2b93..fe92bcb 100644 (file)
@@ -466,22 +466,58 @@ static int config_module(int reload)
 
        conn = PQsetdbLogin(pghostname, pgdbport, NULL, NULL, pgdbname, pgdbuser, pgpassword);
        if (PQstatus(conn) != CONNECTION_BAD) {
-               char sqlcmd[512];
+               char sqlcmd[768];
                char *fname, *ftype, *flen, *fnotnull, *fdef;
-               char *tableptr;
-               int i, rows;
+               int i, rows, version;
                ast_debug(1, "Successfully connected to PostgreSQL database.\n");
                connected = 1;
+               version = PQserverVersion(conn);
+
+               if (version >= 70300) {
+                       char *schemaname, *tablename;
+                       if (strchr(table, '.')) {
+                               schemaname = ast_strdupa(table);
+                               tablename = strchr(schemaname, '.');
+                               *tablename++ = '\0';
+                       } else {
+                               schemaname = "";
+                               tablename = table;
+                       }
+
+                       /* Escape special characters in schemaname */
+                       if (strchr(schemaname, '\\') || strchr(schemaname, '\'')) {
+                               char *tmp = schemaname, *ptr;
+
+                               ptr = schemaname = alloca(strlen(tmp) * 2 + 1);
+                               for (; *tmp; tmp++) {
+                                       if (strchr("\\'", *tmp)) {
+                                               *ptr++ = *tmp;
+                                       }
+                                       *ptr++ = *tmp;
+                               }
+                               *ptr = '\0';
+                       }
+                       /* Escape special characters in tablename */
+                       if (strchr(tablename, '\\') || strchr(tablename, '\'')) {
+                               char *tmp = tablename, *ptr;
+
+                               ptr = tablename = alloca(strlen(tmp) * 2 + 1);
+                               for (; *tmp; tmp++) {
+                                       if (strchr("\\'", *tmp)) {
+                                               *ptr++ = *tmp;
+                                       }
+                                       *ptr++ = *tmp;
+                               }
+                               *ptr = '\0';
+                       }
 
-               /* Remove any schema name from the table */
-               if ((tableptr = strrchr(table, '.'))) {
-                       tableptr++;
+                       snprintf(sqlcmd, sizeof(sqlcmd), "SELECT a.attname, t.typname, a.attlen, a.attnotnull, d.adsrc, a.atttypmod FROM (((pg_catalog.pg_class c INNER JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace AND c.relname = '%s' AND n.nspname = %s%s%s) INNER JOIN pg_catalog.pg_attribute a ON (NOT a.attisdropped) AND a.attnum > 0 AND a.attrelid = c.oid) INNER JOIN pg_catalog.pg_type t ON t.oid = a.atttypid) LEFT OUTER JOIN pg_attrdef d ON a.atthasdef AND d.adrelid = a.attrelid AND d.adnum = a.attnum ORDER BY n.nspname, c.relname, attnum",
+                               tablename,
+                               ast_strlen_zero(schemaname) ? "" : "'", ast_strlen_zero(schemaname) ? "current_schema()" : schemaname, ast_strlen_zero(schemaname) ? "" : "'");
                } else {
-                       tableptr = table;
+                       snprintf(sqlcmd, sizeof(sqlcmd), "SELECT a.attname, t.typname, a.attlen, a.attnotnull, d.adsrc, a.atttypmod FROM pg_class c, pg_type t, pg_attribute a LEFT OUTER JOIN pg_attrdef d ON a.atthasdef AND d.adrelid = a.attrelid AND d.adnum = a.attnum WHERE c.oid = a.attrelid AND a.atttypid = t.oid AND (a.attnum > 0) AND c.relname = '%s' ORDER BY c.relname, attnum", table);
                }
-
                /* Query the columns */
-               snprintf(sqlcmd, sizeof(sqlcmd), "select a.attname, t.typname, a.attlen, a.attnotnull, d.adsrc from pg_class c, pg_type t, pg_attribute a left outer join pg_attrdef d on a.atthasdef and d.adrelid = a.attrelid and d.adnum = a.attnum where c.oid = a.attrelid and a.atttypid = t.oid and (a.attnum > 0) and c.relname = '%s' order by c.relname, attnum", tableptr);
                result = PQexec(conn, sqlcmd);
                if (PQresultStatus(result) != PGRES_TUPLES_OK) {
                        pgerror = PQresultErrorMessage(result);
@@ -498,6 +534,10 @@ static int config_module(int reload)
                        flen = PQgetvalue(result, i, 2);
                        fnotnull = PQgetvalue(result, i, 3);
                        fdef = PQgetvalue(result, i, 4);
+                       if (atoi(flen) == -1) {
+                               /* For varchar columns, the maximum length is encoded in a different field */
+                               flen = PQgetvalue(result, i, 5);
+                       }
                        ast_verb(4, "Found column '%s' of type '%s'\n", fname, ftype);
                        cur = ast_calloc(1, sizeof(*cur) + strlen(fname) + strlen(ftype) + 2);
                        if (cur) {
index 9ba6e97..2fa5a23 100644 (file)
@@ -50,6 +50,8 @@ AST_THREADSTORAGE(escapebuf_buf);
 #define RES_CONFIG_PGSQL_CONF "res_pgsql.conf"
 
 static PGconn *pgsqlConn = NULL;
+static int version;
+#define has_schema_support     (version > 70300 ? 1 : 0)
 
 #define MAX_DB_OPTION_SIZE 64
 
@@ -112,7 +114,7 @@ static void destroy_table(struct tables *table)
        ast_free(table);
 }
 
-static struct tables *find_table(const char *tablename)
+static struct tables *find_table(const char *orig_tablename)
 {
        struct columns *column;
        struct tables *table;
@@ -124,7 +126,7 @@ static struct tables *find_table(const char *tablename)
 
        AST_LIST_LOCK(&psql_tables);
        AST_LIST_TRAVERSE(&psql_tables, table, list) {
-               if (!strcasecmp(table->name, tablename)) {
+               if (!strcasecmp(table->name, orig_tablename)) {
                        ast_debug(1, "Found table in cache; now locking\n");
                        ast_rwlock_rdlock(&table->lock);
                        ast_debug(1, "Lock cached table; now returning\n");
@@ -133,10 +135,69 @@ static struct tables *find_table(const char *tablename)
                }
        }
 
-       ast_debug(1, "Table '%s' not found in cache, querying now\n", tablename);
+       ast_debug(1, "Table '%s' not found in cache, querying now\n", orig_tablename);
 
        /* Not found, scan the table */
-       ast_str_set(&sql, 0, "SELECT a.attname, t.typname, a.attlen, a.attnotnull, d.adsrc, a.atttypmod FROM pg_class c, pg_type t, pg_attribute a LEFT OUTER JOIN pg_attrdef d ON a.atthasdef AND d.adrelid = a.attrelid AND d.adnum = a.attnum WHERE c.oid = a.attrelid AND a.atttypid = t.oid AND (a.attnum > 0) AND c.relname = '%s' ORDER BY c.relname, attnum", tablename);
+       if (has_schema_support) {
+               char *schemaname, *tablename;
+               if (strchr(orig_tablename, '.')) {
+                       schemaname = ast_strdupa(orig_tablename);
+                       tablename = strchr(schemaname, '.');
+                       *tablename++ = '\0';
+               } else {
+                       schemaname = "";
+                       tablename = ast_strdupa(orig_tablename);
+               }
+
+               /* Escape special characters in schemaname */
+               if (strchr(schemaname, '\\') || strchr(schemaname, '\'')) {
+                       char *tmp = schemaname, *ptr;
+
+                       ptr = schemaname = alloca(strlen(tmp) * 2 + 1);
+                       for (; *tmp; tmp++) {
+                               if (strchr("\\'", *tmp)) {
+                                       *ptr++ = *tmp;
+                               }
+                               *ptr++ = *tmp;
+                       }
+                       *ptr = '\0';
+               }
+               /* Escape special characters in tablename */
+               if (strchr(tablename, '\\') || strchr(tablename, '\'')) {
+                       char *tmp = tablename, *ptr;
+
+                       ptr = tablename = alloca(strlen(tmp) * 2 + 1);
+                       for (; *tmp; tmp++) {
+                               if (strchr("\\'", *tmp)) {
+                                       *ptr++ = *tmp;
+                               }
+                               *ptr++ = *tmp;
+                       }
+                       *ptr = '\0';
+               }
+
+               ast_str_set(&sql, 0, "SELECT a.attname, t.typname, a.attlen, a.attnotnull, d.adsrc, a.atttypmod FROM (((pg_catalog.pg_class c INNER JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace AND c.relname = '%s' AND n.nspname = %s%s%s) INNER JOIN pg_catalog.pg_attribute a ON (NOT a.attisdropped) AND a.attnum > 0 AND a.attrelid = c.oid) INNER JOIN pg_catalog.pg_type t ON t.oid = a.atttypid) LEFT OUTER JOIN pg_attrdef d ON a.atthasdef AND d.adrelid = a.attrelid AND d.adnum = a.attnum ORDER BY n.nspname, c.relname, attnum",
+                       tablename,
+                       ast_strlen_zero(schemaname) ? "" : "'", ast_strlen_zero(schemaname) ? "current_schema()" : schemaname, ast_strlen_zero(schemaname) ? "" : "'");
+       } else {
+               /* Escape special characters in tablename */
+               if (strchr(orig_tablename, '\\') || strchr(orig_tablename, '\'')) {
+                       const char *tmp = orig_tablename;
+                       char *ptr;
+
+                       orig_tablename = ptr = alloca(strlen(tmp) * 2 + 1);
+                       for (; *tmp; tmp++) {
+                               if (strchr("\\'", *tmp)) {
+                                       *ptr++ = *tmp;
+                               }
+                               *ptr++ = *tmp;
+                       }
+                       *ptr = '\0';
+               }
+
+               ast_str_set(&sql, 0, "SELECT a.attname, t.typname, a.attlen, a.attnotnull, d.adsrc, a.atttypmod FROM pg_class c, pg_type t, pg_attribute a LEFT OUTER JOIN pg_attrdef d ON a.atthasdef AND d.adrelid = a.attrelid AND d.adnum = a.attnum WHERE c.oid = a.attrelid AND a.atttypid = t.oid AND (a.attnum > 0) AND c.relname = '%s' ORDER BY c.relname, attnum", orig_tablename);
+       }
+
        result = PQexec(pgsqlConn, ast_str_buffer(sql));
        ast_debug(1, "Query of table structure complete.  Now retrieving results.\n");
        if (PQresultStatus(result) != PGRES_TUPLES_OK) {
@@ -147,12 +208,12 @@ static struct tables *find_table(const char *tablename)
                return NULL;
        }
 
-       if (!(table = ast_calloc(1, sizeof(*table) + strlen(tablename) + 1))) {
+       if (!(table = ast_calloc(1, sizeof(*table) + strlen(orig_tablename) + 1))) {
                ast_log(LOG_ERROR, "Unable to allocate memory for new table structure\n");
                AST_LIST_UNLOCK(&psql_tables);
                return NULL;
        }
-       strcpy(table->name, tablename); /* SAFE */
+       strcpy(table->name, orig_tablename); /* SAFE */
        ast_rwlock_init(&table->lock);
        AST_LIST_HEAD_INIT_NOLOCK(&table->columns);
 
@@ -166,7 +227,7 @@ static struct tables *find_table(const char *tablename)
                ast_verb(4, "Found column '%s' of type '%s'\n", fname, ftype);
 
                if (!(column = ast_calloc(1, sizeof(*column) + strlen(fname) + strlen(ftype) + 2))) {
-                       ast_log(LOG_ERROR, "Unable to allocate column element for %s, %s\n", tablename, fname);
+                       ast_log(LOG_ERROR, "Unable to allocate column element for %s, %s\n", orig_tablename, fname);
                        destroy_table(table);
                        AST_LIST_UNLOCK(&psql_tables);
                        return NULL;
@@ -1409,6 +1470,7 @@ static int pgsql_reconnect(const char *database)
                if (pgsqlConn && PQstatus(pgsqlConn) == CONNECTION_OK) {
                        ast_debug(1, "PostgreSQL RealTime: Successfully connected to database.\n");
                        connect_time = time(NULL);
+                       version = PQserverVersion(pgsqlConn);
                        return 1;
                } else {
                        ast_log(LOG_ERROR,