Be a little bit more careful about using sqlstate, in case libpq returned
authorPeter Eisentraut <[email protected]>
Fri, 8 Aug 2003 13:16:20 +0000 (13:16 +0000)
committerPeter Eisentraut <[email protected]>
Fri, 8 Aug 2003 13:16:20 +0000 (13:16 +0000)
NULL for it.

src/interfaces/ecpg/ecpglib/connect.c
src/interfaces/ecpg/ecpglib/error.c

index 679efb6466b6cb38528ff6540108fa892dbe08dd..6574216a6068c3019bee5199882dde3479b357a0 100644 (file)
@@ -1,4 +1,4 @@
-/* $Header: /cvsroot/pgsql/src/interfaces/ecpg/ecpglib/connect.c,v 1.15 2003/08/04 00:43:32 momjian Exp $ */
+/* $Header: /cvsroot/pgsql/src/interfaces/ecpg/ecpglib/connect.c,v 1.16 2003/08/08 13:16:20 petere Exp $ */
 
 #define POSTGRES_ECPG_INTERNAL
 #include "postgres_fe.h"
@@ -164,6 +164,9 @@ ECPGnoticeReceiver(void *arg, const PGresult *result)
 
    int         sqlcode;
 
+   if (sqlstate == NULL)
+       sqlstate = ECPG_SQLSTATE_ECPG_INTERNAL_ERROR;
+
    /* these are not warnings */
    if (strncmp(sqlstate, "00", 2) == 0)
        return;
index 33bac6e892d417f22fced09125eb3001bd002872..82f2ccca3665c08371315d704065bd31a37634b6 100644 (file)
@@ -1,4 +1,4 @@
-/* $Header: /cvsroot/pgsql/src/interfaces/ecpg/ecpglib/error.c,v 1.6 2003/08/04 00:43:32 momjian Exp $ */
+/* $Header: /cvsroot/pgsql/src/interfaces/ecpg/ecpglib/error.c,v 1.7 2003/08/08 13:16:20 petere Exp $ */
 
 #define POSTGRES_ECPG_INTERNAL
 #include "postgres_fe.h"
@@ -159,18 +159,29 @@ void
 ECPGraise_backend(int line, PGresult *result, PGconn *conn, int compat)
 {
    struct sqlca_t *sqlca = ECPGget_sqlca();
+   char       *sqlstate;
+   char       *message;
+
+   if (result)
+   {
+       sqlstate = PQresultErrorField(result, 'C');
+       if (sqlstate == NULL)
+           sqlstate = ECPG_SQLSTATE_ECPG_INTERNAL_ERROR;
+       message = PQresultErrorField(result, 'M');
+   }
+   else
+   {
+       sqlstate = ECPG_SQLSTATE_ECPG_INTERNAL_ERROR;
+       message = PQerrorMessage(conn);
+   }
 
    /* copy error message */
    snprintf(sqlca->sqlerrm.sqlerrmc, sizeof(sqlca->sqlerrm.sqlerrmc),
-            "'%s' in line %d.",
-        result ? PQresultErrorField(result, 'M') : PQerrorMessage(conn),
-            line);
+            "'%s' in line %d.", message, line);
    sqlca->sqlerrm.sqlerrml = strlen(sqlca->sqlerrm.sqlerrmc);
 
    /* copy SQLSTATE */
-   strncpy(sqlca->sqlstate,
-           result ? PQresultErrorField(result, 'C') : ECPG_SQLSTATE_ECPG_INTERNAL_ERROR,
-           sizeof(sqlca->sqlstate));
+   strncpy(sqlca->sqlstate, sqlstate, sizeof(sqlca->sqlstate));
 
    /* assign SQLCODE for backward compatibility */
    if (strncmp(sqlca->sqlstate, "23505", sizeof(sqlca->sqlstate)) == 0)