Ticket #32165: 32165.3.diff

src/wp-includes/wp-db.php

@@ -2710 +2710 @@
                         $queries = array();
                         foreach ( $data as $col => $value ) {
                                 if ( ! empty( $value['db'] ) ) {
-                                        if ( ! isset( $queries[ $value['charset'] ] ) ) {
-                                                $queries[ $value['charset'] ] = array();
-                                        }
-
                                         // We're going to need to truncate by characters or bytes, depending on the length value we have.
                                         if ( 'byte' === $value['length']['type'] ) {
                                                 // Split the CONVERT() calls by charset, so we can make sure the connection is right
-                                                $queries[ $value['charset'] ][ $col ] = $this->prepare( "CONVERT( LEFT( CONVERT( %s USING binary ), %.0f ) USING {$value['charset']} )", $value['value'], $value['length']['length'] );
+                                                $queries[ $col ] = $this->prepare( "CONVERT( LEFT( CONVERT( %s USING binary ), %.0f ) USING {$value['charset']} )", $value['value'], $value['length']['length'] );
                                         } else {
-                                                $queries[ $value['charset'] ][ $col ] = $this->prepare( "LEFT( CONVERT( %s USING {$value['charset']} ), %.0f )", $value['value'], $value['length']['length'] );
+                                                $queries[ $col ] = $this->prepare( "LEFT( CONVERT( %s USING {$value['charset']} ), %.0f )", $value['value'], $value['length']['length'] );
                                         }
 
                                         unset( $data[ $col ]['db'] );
@@ -2726 +2722 @@
                                 }
                         }
 
-                        $connection_charset = $this->charset;
-                        foreach ( $queries as $charset => $query ) {
+                        $this->check_current_query = false;
+
+                        $sql = array();
+                        foreach ( $queries as $column => $query ) {
                                 if ( ! $query ) {
                                         continue;
                                 }
 
-                                // Change the charset to match the string(s) we're converting
-                                if ( $charset !== $connection_charset ) {
-                                        $connection_charset = $charset;
-                                        $this->set_charset( $this->dbh, $charset );
-                                }
+                                $sql[] = $query . " AS x_$column";
+                        }
 
-                                $this->check_current_query = false;
-
-                                $sql = array();
-                                foreach ( $query as $column => $column_query ) {
-                                        $sql[] = $column_query . " AS x_$column";
-                                }
-
-                                $row = $this->get_row( "SELECT " . implode( ', ', $sql ), ARRAY_A );
-                                if ( ! $row ) {
-                                        $this->set_charset( $this->dbh, $connection_charset );
-                                        return new WP_Error( 'wpdb_strip_invalid_text_failure' );
-                                }
-
-                                foreach ( array_keys( $query ) as $column ) {
-                                        $data[ $column ]['value'] = $row["x_$column"];
-                                }
+                        $row = $this->get_row( "SELECT " . implode( ', ', $sql ), ARRAY_A );
+                        if ( ! $row ) {
+                                return new WP_Error( 'wpdb_strip_invalid_text_failure' );
                         }
 
-                        // Don't forget to change the charset back!
-                        if ( $connection_charset !== $this->charset ) {
-                                $this->set_charset( $this->dbh );
+                        foreach ( array_keys( $data ) as $column ) {
+                                $data[ $column ]['value'] = $row["x_$column"];
                         }
                 }
 

tests/phpunit/tests/db/charset.php

@@ -303 +303 @@
                         $data_provider[] = array( $data, $expected, $test_case );
                 }
 
-                // Time for our test of multiple fields at once.
-                $data_provider[] = array( $multiple, $multiple_expected, 'multiple fields/charsets' );
-
                 return $data_provider;
         }
 
@@ -318 +315 @@
                         $this->markTestSkipped( 'This test fails in PHP 5.2 on Windows. See https://core.trac.wordpress.org/ticket/31262' );
                 }
 
+                $charset = self::$_wpdb->charset;
+                self::$_wpdb->set_charset( self::$_wpdb->dbh, $data[0]['charset'] );
+
                 $actual = self::$_wpdb->strip_invalid_text( $data );
+
+                self::$_wpdb->set_charset( self::$_wpdb->dbh, $charset );
+
                 $this->assertSame( $expected, $actual, $message );
         }