Fix bogus failure-return value from lo_create, per report from Gavin

Sherry.  Also clean up leakage of open files and LOs in failure exits
from lo_import and lo_export.

diff --git a/src/interfaces/libpq/fe-lobj.c b/src/interfaces/libpq/fe-lobj.c
index 873095ce9956be1689641ab59d001e44cd2416e2..7be372d6a33098454377032c580a65c2246053d2 100644 (file)
--- a/src/interfaces/libpq/fe-lobj.c
+++ b/src/interfaces/libpq/fe-lobj.c
@@ -8,7 +8,7 @@
  *
  *
  * IDENTIFICATION
- *   $Header: /cvsroot/pgsql/src/interfaces/libpq/fe-lobj.c,v 1.38 2001/08/21 20:39:54 momjian Exp $
+ *   $Header: /cvsroot/pgsql/src/interfaces/libpq/fe-lobj.c,v 1.39 2001/09/17 20:05:47 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -18,9 +18,6 @@
 #include 
 #include 
 
-#include "libpq-fe.h"
-#include "libpq-int.h"
-
 #ifdef WIN32
 #include "win32.h"
 #include "io.h"
@@ -28,12 +25,16 @@
 #include 
 #endif
 
+#include "libpq-fe.h"
+#include "libpq-int.h"
 #include "libpq/libpq-fs.h"        /* must come after sys/stat.h */
 
+
 #define LO_BUFSIZE       8192
 
 static int lo_initialize(PGconn *conn);
 
+
 /*
  * lo_open
  *   opens an existing large object
@@ -123,7 +124,7 @@ lo_close(PGconn *conn, int fd)
  * lo_read
  *   read len bytes of the large object into buf
  *
- * returns the length of bytes read.
+ * returns the number of bytes read, or -1 on failure.
  * the CALLER must have allocated enough space to hold the result returned
  */
 
@@ -166,6 +167,7 @@ lo_read(PGconn *conn, int fd, char *buf, size_t len)
  * lo_write
  *   write len bytes of buf into the large object fd
  *
+ * returns the number of bytes written, or -1 on failure.
  */
 int
 lo_write(PGconn *conn, int fd, char *buf, size_t len)
@@ -273,7 +275,7 @@ lo_creat(PGconn *conn, int mode)
    if (conn->lobjfuncs == (PGlobjfuncs *) NULL)
    {
        if (lo_initialize(conn) < 0)
-           return -1;
+           return InvalidOid;
    }
 
    argv[0].isint = 1;
@@ -373,9 +375,9 @@ lo_unlink(PGconn *conn, Oid lobjId)
 /*
  * lo_import -
  *   imports a file as an (inversion) large object.
- *     returns the oid of that object upon success,
- * returns InvalidOid upon failure
  *
+ * returns the oid of that object upon success,
+ * returns InvalidOid upon failure
  */
 
 Oid
@@ -409,6 +411,7 @@ lo_import(PGconn *conn, const char *filename)
        printfPQExpBuffer(&conn->errorMessage,
                          libpq_gettext("could not create large object for file \"%s\"\n"),
                          filename);
+       (void) close(fd);
        return InvalidOid;
    }
 
@@ -418,6 +421,7 @@ lo_import(PGconn *conn, const char *filename)
        printfPQExpBuffer(&conn->errorMessage,
                          libpq_gettext("could not open large object %u\n"),
                          lobjOid);
+       (void) close(fd);
        return InvalidOid;
    }
 
@@ -432,6 +436,8 @@ lo_import(PGconn *conn, const char *filename)
            printfPQExpBuffer(&conn->errorMessage,
                              libpq_gettext("error while reading file \"%s\"\n"),
                              filename);
+           (void) close(fd);
+           (void) lo_close(conn, lobj);
            return InvalidOid;
        }
    }
@@ -457,7 +463,7 @@ lo_export(PGconn *conn, Oid lobjId, const char *filename)
    int         lobj;
 
    /*
-    * create an inversion "object"
+    * open the large object.
     */
    lobj = lo_open(conn, lobjId, INV_READ);
    if (lobj == -1)
@@ -468,7 +474,7 @@ lo_export(PGconn *conn, Oid lobjId, const char *filename)
    }
 
    /*
-    * open the file to be written to
+    * create the file to be written to
     */
    fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC | PG_BINARY, 0666);
    if (fd < 0)
@@ -476,7 +482,8 @@ lo_export(PGconn *conn, Oid lobjId, const char *filename)
        printfPQExpBuffer(&conn->errorMessage,
                          libpq_gettext("could not open file \"%s\": %s\n"),
                          filename, strerror(errno));
-       return 0;
+       (void) lo_close(conn, lobj);
+       return -1;
    }
 
    /*
@@ -490,6 +497,8 @@ lo_export(PGconn *conn, Oid lobjId, const char *filename)
            printfPQExpBuffer(&conn->errorMessage,
                              libpq_gettext("error while writing to file \"%s\"\n"),
                              filename);
+           (void) lo_close(conn, lobj);
+           (void) close(fd);
            return -1;
        }
    }