Changeset 32368

Timestamp:

05/06/2015 05:00:39 AM (10 years ago)

Author:

pento

Message:

WPDB: Allow queries to reference tables in the dbname.tablename format, and allow table names to contain any valid character, rather than just ASCII.

Props pento, willstedt for the initial patch.

See #32090.

Location:

trunk

Files:

• src/wp-includes/wp-db.php (modified) (4 diffs)
• tests/phpunit/tests/db.php (modified) (2 diffs)

trunk/src/wp-includes/wp-db.php

@@ -2267 +2267 @@
 
         $charsets = $columns = array();
-        $results = $this->get_results( "SHOW FULL COLUMNS FROM `$table`" );
+
+        $table_parts = explode( '.', $table );
+        $table = '`' . implode( '`.`', $table_parts ) . '`';
+        $results = $this->get_results( "SHOW FULL COLUMNS FROM $table" );
         if ( ! $results ) {
             return new WP_Error( 'wpdb_get_table_charset_failure' );
@@ -2839 +2842 @@
                 . '|UPDATE(?:\s+LOW_PRIORITY)?(?:\s+IGNORE)?'
                 . '|DELETE(?:\s+LOW_PRIORITY|\s+QUICK|\s+IGNORE)*(?:\s+FROM)?'
-                . ')\s+`?([\w-]+)`?/is', $query, $maybe ) ) {
-            return $maybe[1];
+                . ')\s+((?:[0-9a-zA-Z$_.`]|[\xC2-\xDF][\x80-\xBF])+)/is', $query, $maybe ) ) {
+            return str_replace( '`', '', $maybe[1] );
         }
 
@@ -2847 +2850 @@
                 . 'SHOW\s+TABLE\s+STATUS.+(?:LIKE\s+|WHERE\s+Name\s*=\s*)'
                 . '|SHOW\s+(?:FULL\s+)?TABLES.+(?:LIKE\s+|WHERE\s+Name\s*=\s*)'
-                . ')\W([\w-]+)\W/is', $query, $maybe ) ) {
-            return $maybe[1];
+                . ')\W((?:[0-9a-zA-Z$_.`]|[\xC2-\xDF][\x80-\xBF])+)\W/is', $query, $maybe ) ) {
+            return str_replace( '`', '', $maybe[1] );
         }
 
@@ -2866 +2869 @@
                 . '|(?:GRANT|REVOKE).*ON\s+TABLE'
                 . '|SHOW\s+(?:.*FROM|.*TABLE)'
-                . ')\s+\(*\s*`?([\w-]+)`?\s*\)*/is', $query, $maybe ) ) {
-            return $maybe[1];
+                . ')\s+\(*\s*((?:[0-9a-zA-Z$_.`]|[\xC2-\xDF][\x80-\xBF])+)\s*\)*/is', $query, $maybe ) ) {
+            return str_replace( '`', '', $maybe[1] );
         }
 

trunk/tests/phpunit/tests/db.php

@@ -527 +527 @@
     function data_get_table_from_query() {
         $table = 'a_test_table_name';
+        $db_table = '`a_test_db`.`another_test_table`';
 
         $queries = array(
@@ -629 +630 @@
         );
 
-        foreach ( $queries as &$query ) {
-            $query = array( $query, $table );
+        $querycount = count( $queries );
+        for ( $ii = 0; $ii < $querycount; $ii++ ) {
+            $db_query = str_replace( $table, $db_table, $queries[ $ii ] );
+            $expected_db_table = str_replace( '`', '', $db_table );
+
+            $queries[ $ii ] = array( $queries[ $ii ], $table );
+            $queries[] = array( $db_query, $expected_db_table );
         }
         return $queries;