]> granicus.if.org Git - postgresql/commitdiff
My first cut at libpq revision didn't handle MULTIBYTE correctly,
authorTom Lane <tgl@sss.pgh.pa.us>
Sun, 25 Apr 1999 19:27:47 +0000 (19:27 +0000)
committerTom Lane <tgl@sss.pgh.pa.us>
Sun, 25 Apr 1999 19:27:47 +0000 (19:27 +0000)
but I think it's OK now...

src/backend/access/common/printtup.c
src/backend/commands/async.c
src/backend/libpq/pqformat.c
src/backend/tcop/dest.c
src/include/libpq/pqformat.h

index 5ce4830ebd4a7caa5043d67ff7b99845c31f21f2..0d2aeeecd1c289929055187675fe321ebb118fa4 100644 (file)
@@ -8,7 +8,7 @@
  *
  *
  * IDENTIFICATION
- *       $Header: /cvsroot/pgsql/src/backend/access/common/printtup.c,v 1.43 1999/04/25 03:19:23 tgl Exp $
+ *       $Header: /cvsroot/pgsql/src/backend/access/common/printtup.c,v 1.44 1999/04/25 19:27:43 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
 #include "libpq/pqformat.h"
 #include "utils/syscache.h"
 
-#ifdef MULTIBYTE
-#include "mb/pg_wchar.h"
-#endif
-
 static void printtup_setup(DestReceiver* self, TupleDesc typeinfo);
 static void printtup(HeapTuple tuple, TupleDesc typeinfo, DestReceiver* self);
 static void printtup_cleanup(DestReceiver* self);
@@ -157,14 +153,10 @@ printtup(HeapTuple tuple, TupleDesc typeinfo, DestReceiver* self)
        StringInfoData buf;
        int                     i,
                                j,
-                               k,
-                               outputlen;
+                               k;
        char       *outputstr;
        Datum           attr;
        bool            isnull;
-#ifdef MULTIBYTE
-       unsigned char *p;
-#endif
 
        /* Set or update my derived attribute info, if needed */
        if (myState->attrinfo != typeinfo ||
@@ -213,24 +205,13 @@ printtup(HeapTuple tuple, TupleDesc typeinfo, DestReceiver* self)
                {
                        outputstr = (char *) (*fmgr_faddr(&thisState->finfo))
                                (attr, thisState->typelem, typeinfo->attrs[i]->atttypmod);
-#ifdef MULTIBYTE
-                       p = pg_server_to_client(outputstr, strlen(outputstr));
-                       outputlen = strlen(p);
-                       pq_sendint(&buf, outputlen + VARHDRSZ, VARHDRSZ);
-                       pq_sendbytes(&buf, p, outputlen);
-#else
-                       outputlen = strlen(outputstr);
-                       pq_sendint(&buf, outputlen + VARHDRSZ, VARHDRSZ);
-                       pq_sendbytes(&buf, outputstr, outputlen);
-#endif
+                       pq_sendcountedtext(&buf, outputstr, strlen(outputstr));
                        pfree(outputstr);
                }
                else
                {
                        outputstr = "<unprintable>";
-                       outputlen = strlen(outputstr);
-                       pq_sendint(&buf, outputlen + VARHDRSZ, VARHDRSZ);
-                       pq_sendbytes(&buf, outputstr, outputlen);
+                       pq_sendcountedtext(&buf, outputstr, strlen(outputstr));
                }
        }
 
index 3d5cf92f7d1299d61e78e698eeba8dd48e7cd732..fb1c116bc476ec486107f30ab734441af634ab24 100644 (file)
@@ -6,7 +6,7 @@
  * Copyright (c) 1994, Regents of the University of California
  *
  * IDENTIFICATION
- *       $Header: /cvsroot/pgsql/src/backend/commands/async.c,v 1.45 1999/04/25 03:19:08 tgl Exp $
+ *       $Header: /cvsroot/pgsql/src/backend/commands/async.c,v 1.46 1999/04/25 19:27:43 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -803,7 +803,7 @@ NotifyMyFrontEnd(char *relname, int32 listenerPID)
                pq_beginmessage(&buf);
                pq_sendbyte(&buf, 'A');
                pq_sendint(&buf, listenerPID, sizeof(int32));
-               pq_sendstring(&buf, relname, strlen(relname));
+               pq_sendstring(&buf, relname);
                pq_endmessage(&buf);
                /* NOTE: we do not do pq_flush() here.  For a self-notify, it will
                 * happen at the end of the transaction, and for incoming notifies
index 06f55d648bc9896016e0ecfb4ddc532f43b40ad6..ad6f045a944737f0be4a98e2cb09500a7636e3b4 100644 (file)
@@ -15,7 +15,7 @@
  *
  * Copyright (c) 1994, Regents of the University of California
  *
- *  $Id: pqformat.c,v 1.1 1999/04/25 03:19:22 tgl Exp $
+ *  $Id: pqformat.c,v 1.2 1999/04/25 19:27:44 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -26,7 +26,7 @@
  *             pq_sendbyte             - append a raw byte to a StringInfo buffer
  *             pq_sendint              - append a binary integer to a StringInfo buffer
  *             pq_sendbytes    - append raw data to a StringInfo buffer
- *             pq_sendtext             - append a text string (with MULTIBYTE conversion)
+ *             pq_sendcountedtext - append a text string (with MULTIBYTE conversion)
  *             pq_sendstring   - append a null-terminated text string (with MULTIBYTE)
  *             pq_endmessage   - send the completed message to the frontend
  * Note: it is also possible to append data to the StringInfo buffer using
@@ -36,9 +36,8 @@
  * Message input:
  *             pq_getint               - get an integer from connection
  *             pq_getstr               - get a null terminated string from connection
- *             pq_getnchar             - get n characters from connection, and null-terminate
- * pq_getstr and pq_getnchar perform MULTIBYTE conversion on the collected
- * string.  Use the raw pqcomm.c routines pq_getstring and pq_getbytes
+ * pq_getstr performs MULTIBYTE conversion on the collected string.
+ * Use the raw pqcomm.c routines pq_getstring or pq_getbytes
  * to fetch data without conversion.
  */
 #include "postgres.h"
@@ -48,6 +47,7 @@
 #ifdef MULTIBYTE
 #include "mb/pg_wchar.h"
 #endif
+#include <string.h>
 #ifdef HAVE_ENDIAN_H
 #include <endian.h>
 #endif
@@ -110,36 +110,49 @@ pq_sendbytes(StringInfo buf, const char *data, int datalen)
 }
 
 /* --------------------------------
- *             pq_sendtext             - append a text string (with MULTIBYTE conversion)
+ *             pq_sendcountedtext - append a text string (with MULTIBYTE conversion)
  *
- * NB: passed text string must be null-terminated, even though we expect
- * the caller to hand us the length (this is just because the caller
- * usually knows the length anyway).  In this routine, the data sent to
- * the frontend is NOT null-terminated.
+ * The data sent to the frontend by this routine is a 4-byte count field
+ * (the count includes itself, by convention) followed by the string.
+ * The passed text string need not be null-terminated, and the data sent
+ * to the frontend isn't either.
  * --------------------------------
  */
 void
-pq_sendtext(StringInfo buf, const char *str, int slen)
+pq_sendcountedtext(StringInfo buf, const char *str, int slen)
 {
 #ifdef MULTIBYTE
-       str = (const char *) pg_server_to_client(str, slen);
+       const char *p;
+       p = (const char *) pg_server_to_client((unsigned char *) str, slen);
+       if (p != str)                           /* actual conversion has been done? */
+       {
+               str = p;
+               slen = strlen(str);
+       }
 #endif
+       pq_sendint(buf, slen + 4, 4);
        appendBinaryStringInfo(buf, str, slen);
 }
 
 /* --------------------------------
  *             pq_sendstring   - append a null-terminated text string (with MULTIBYTE)
  *
- * NB: passed text string must be null-terminated, even though we expect
- * the caller to hand us the length (this is just because the caller
- * usually knows the length anyway).
+ * NB: passed text string must be null-terminated, and so is the data
+ * sent to the frontend.
  * --------------------------------
  */
 void
-pq_sendstring(StringInfo buf, const char *str, int slen)
+pq_sendstring(StringInfo buf, const char *str)
 {
+       int slen = strlen(str);
 #ifdef MULTIBYTE
-       str = (const char *) pg_server_to_client(str, slen);
+       const char *p;
+       p = (const char *) pg_server_to_client((unsigned char *) str, slen);
+       if (p != str)                           /* actual conversion has been done? */
+       {
+               str = p;
+               slen = strlen(str);
+       }
 #endif
        appendBinaryStringInfo(buf, str, slen+1);
 }
@@ -258,35 +271,18 @@ pq_getstr(char *s, int maxlen)
        c = pq_getstring(s, maxlen);
 
 #ifdef MULTIBYTE
-       p = (char*) pg_client_to_server((unsigned char *) s, maxlen);
-       if (s != p)                                     /* actual conversion has been done? */
-               strcpy(s, p);
-#endif
-
-       return c;
-}
-
-/* --------------------------------
- *             pq_getnchar - get n characters from connection, and null-terminate
- *
- *             returns 0 if OK, EOF if trouble
- * --------------------------------
- */
-int
-pq_getnchar(char *s, int len)
-{
-       int                     c;
-#ifdef MULTIBYTE
-       char       *p;
-#endif
-
-       c = pq_getbytes(s, len);
-       s[len] = '\0';
-
-#ifdef MULTIBYTE
-       p = (char*) pg_client_to_server((unsigned char *) s, len+1);
-       if (s != p)                                     /* actual conversion has been done? */
-               strcpy(s, p);
+       p = (char*) pg_client_to_server((unsigned char *) s, strlen(s));
+       if (p != s)                                     /* actual conversion has been done? */
+       {
+               int newlen = strlen(p);
+               if (newlen < maxlen)
+                       strcpy(s, p);
+               else
+               {
+                       strncpy(s, p, maxlen);
+                       s[maxlen-1] = '\0';
+               }
+       }
 #endif
 
        return c;
index d6f29aa8afcd4195b942c8a3fa522e32ece28b5c..26e04f5f855b15f7956b02383aac7c9b0e28f83d 100644 (file)
@@ -7,7 +7,7 @@
  *
  *
  * IDENTIFICATION
- *       $Header: /cvsroot/pgsql/src/backend/tcop/dest.c,v 1.26 1999/04/25 03:19:09 tgl Exp $
+ *       $Header: /cvsroot/pgsql/src/backend/tcop/dest.c,v 1.27 1999/04/25 19:27:45 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -155,8 +155,7 @@ BeginCommand(char *pname,
 
                                for (i = 0; i < natts; ++i)
                                {
-                                       pq_sendstring(&buf, attrs[i]->attname.data,
-                                                                 strlen(attrs[i]->attname.data));
+                                       pq_sendstring(&buf, attrs[i]->attname.data);
                                        pq_sendint(&buf, (int) attrs[i]->atttypid,
                                                           sizeof(attrs[i]->atttypid));
                                        pq_sendint(&buf, attrs[i]->attlen,
index 2fe48539a2d2ed7a050c57aa1abd59535cd73db6..6404583a103e58911e7160bd851d464245e20d53 100644 (file)
@@ -5,7 +5,7 @@
  *
  * Copyright (c) 1994, Regents of the University of California
  *
- * $Id: pqformat.h,v 1.1 1999/04/25 03:19:14 tgl Exp $
+ * $Id: pqformat.h,v 1.2 1999/04/25 19:27:47 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
 
 extern void    pq_sendbyte(StringInfo buf, int byt);
 extern void    pq_sendbytes(StringInfo buf, const char *data, int datalen);
-extern void    pq_sendtext(StringInfo buf, const char *str, int slen);
-extern void    pq_sendstring(StringInfo buf, const char *str, int slen);
+extern void    pq_sendcountedtext(StringInfo buf, const char *str, int slen);
+extern void    pq_sendstring(StringInfo buf, const char *str);
 extern void    pq_sendint(StringInfo buf, int i, int b);
 extern void    pq_endmessage(StringInfo buf);
 
 extern int     pq_getint(int *result, int b);
 extern int     pq_getstr(char *s, int maxlen);
-extern int     pq_getnchar(char *s, int len);
 
 #endif  /* PQFORMAT_H */