Add checks for valid encoding of query strings passed to SPI by plperl.

diff --git a/src/pl/plperl/plperl.c b/src/pl/plperl/plperl.c
index 449b283462e51ea72b187e95da83ca888528cafb..08e88838c8a4cf99e96d372eb44fa7dc6f2f92f7 100644 (file)
--- a/src/pl/plperl/plperl.c
+++ b/src/pl/plperl/plperl.c
@@ -1,7 +1,7 @@
 /**********************************************************************
  * plperl.c - perl as a procedural language for PostgreSQL
  *
- *   $PostgreSQL: pgsql/src/pl/plperl/plperl.c,v 1.169 2010/02/26 02:01:33 momjian Exp $
+ *   $PostgreSQL: pgsql/src/pl/plperl/plperl.c,v 1.170 2010/03/09 02:48:33 adunstan Exp $
  *
  **********************************************************************/
 
@@ -2102,6 +2102,8 @@ plperl_spi_exec(char *query, int limit)
    {
        int         spi_rv;
 
+       pg_verifymbstr(query, strlen(query), false);
+
        spi_rv = SPI_execute(query, current_call_data->prodesc->fn_readonly,
                             limit);
        ret_hv = plperl_spi_execute_fetch_result(SPI_tuptable, SPI_processed,
@@ -2343,6 +2345,9 @@ plperl_spi_query(char *query)
        void       *plan;
        Portal      portal;
 
+       /* Make sure the query is validly encoded */
+       pg_verifymbstr(query, strlen(query), false);
+
        /* Create a cursor for the query */
        plan = SPI_prepare(query, 0, NULL);
        if (plan == NULL)
@@ -2549,6 +2554,9 @@ plperl_spi_prepare(char *query, int argc, SV **argv)
            qdesc->argtypioparams[i] = typIOParam;
        }
 
+       /* Make sure the query is validly encoded */
+       pg_verifymbstr(query, strlen(query), false);
+
        /************************************************************
         * Prepare the plan and check for errors
         ************************************************************/